Function chatModelProvider

server/src/utils/models.ts:11–119

(
  provider: string,
  modelName: string,
  temperature: number,
  otherFields?: any
)

import { Replicate } from "@langchain/community/llms/replicate";

export const chatModelProvider = (
  provider: string,
  modelName: string,
  temperature: number,
  otherFields?: any
) => {
  modelName = modelName.replace("-dbase", "");
  modelName = modelName.replace(/_dialoqbase_[0-9]+$/, "");

  switch (provider.toLowerCase()) {
    case "openai":
      return new ChatOpenAI({
        modelName: modelName,
        temperature: temperature,
        ...otherFields,
        configuration: {
          ...otherFields.configuration,
          baseURL: process.env.OPENAI_API_URL,
        },
      });
    case "anthropic":
      return new ChatAnthropic({
        modelName: modelName,
        temperature: temperature,
        ...otherFields,
      }) as any;
    case "google-bison":
      return new ChatGooglePaLM({
        temperature: temperature,
        apiKey: process.env.GOOGLE_API_KEY,
        ...otherFields,
      });
    case "huggingface-api":
      return new HuggingFaceInference({
        modelName: modelName,
        temperature: temperature,
        ...otherFields,
      });
    case "fireworks":
      return new DialoqbaseFireworksModel({
        model: modelName,
        temperature: temperature,
        is_chat: !notChatModels.includes(modelName),
        ...otherFields,
      });
    case "openai-instruct":
      return new OpenAI({
        modelName: modelName,
        temperature: temperature,
        ...otherFields,
        configuration: {
          baseURL: process.env.OPENAI_API_URL,
        },
      });
    case "local":
      return new ChatOpenAI({
        modelName: modelName,
        temperature: temperature,