30 lines
796 B
TypeScript
30 lines
796 B
TypeScript
import { customProvider, gateway } from "ai";
|
|
import { isTestEnvironment } from "../constants";
|
|
import { titleModel } from "./models";
|
|
|
|
export const myProvider = isTestEnvironment
|
|
? (() => {
|
|
const { chatModel, titleModel } = require("./models.mock");
|
|
return customProvider({
|
|
languageModels: {
|
|
"chat-model": chatModel,
|
|
"title-model": titleModel,
|
|
},
|
|
});
|
|
})()
|
|
: null;
|
|
|
|
export function getLanguageModel(modelId: string) {
|
|
if (isTestEnvironment && myProvider) {
|
|
return myProvider.languageModel(modelId);
|
|
}
|
|
|
|
return gateway.languageModel(modelId);
|
|
}
|
|
|
|
export function getTitleModel() {
|
|
if (isTestEnvironment && myProvider) {
|
|
return myProvider.languageModel("title-model");
|
|
}
|
|
return gateway.languageModel(titleModel.id);
|
|
}
|