feat(app): remove backend

src/lib/providers/anthropic.ts

@@ -1,6 +1,38 @@
-import { ChatAnthropic } from '@langchain/anthropic';
-import { getAnthropicApiKey } from '../../config';
-import logger from '../../utils/logger';
+import { ChatOpenAI } from '@langchain/openai';
+import { ChatModel } from '.';
+import { getAnthropicApiKey } from '../config';
+import { BaseChatModel } from '@langchain/core/language_models/chat_models';
+
+const anthropicChatModels: Record<string, string>[] = [
+  {
+    displayName: 'Claude 3.7 Sonnet',
+    key: 'claude-3-7-sonnet-20250219',
+  },
+  {
+    displayName: 'Claude 3.5 Haiku',
+    key: 'claude-3-5-haiku-20241022',
+  },
+  {
+    displayName: 'Claude 3.5 Sonnet v2',
+    key: 'claude-3-5-sonnet-20241022',
+  },
+  {
+    displayName: 'Claude 3.5 Sonnet',
+    key: 'claude-3-5-sonnet-20240620',
+  },
+  {
+    displayName: 'Claude 3 Opus',
+    key: 'claude-3-opus-20240229',
+  },
+  {
+    displayName: 'Claude 3 Sonnet',
+    key: 'claude-3-sonnet-20240229',
+  },
+  {
+    displayName: 'Claude 3 Haiku',
+    key: 'claude-3-haiku-20240307',
+  },
+];
 
 export const loadAnthropicChatModels = async () => {
   const anthropicApiKey = getAnthropicApiKey();
@@ -8,52 +40,25 @@ export const loadAnthropicChatModels = async () => {
   if (!anthropicApiKey) return {};
 
   try {
-    const chatModels = {
-      'claude-3-5-sonnet-20241022': {
-        displayName: 'Claude 3.5 Sonnet',
-        model: new ChatAnthropic({
+    const chatModels: Record<string, ChatModel> = {};
+
+    anthropicChatModels.forEach((model) => {
+      chatModels[model.key] = {
+        displayName: model.displayName,
+        model: new ChatOpenAI({
+          openAIApiKey: anthropicApiKey,
+          modelName: model.key,
           temperature: 0.7,
-          anthropicApiKey: anthropicApiKey,
-          model: 'claude-3-5-sonnet-20241022',
-        }),
-      },
-      'claude-3-5-haiku-20241022': {
-        displayName: 'Claude 3.5 Haiku',
-        model: new ChatAnthropic({
-          temperature: 0.7,
-          anthropicApiKey: anthropicApiKey,
-          model: 'claude-3-5-haiku-20241022',
-        }),
-      },
-      'claude-3-opus-20240229': {
-        displayName: 'Claude 3 Opus',
-        model: new ChatAnthropic({
-          temperature: 0.7,
-          anthropicApiKey: anthropicApiKey,
-          model: 'claude-3-opus-20240229',
-        }),
-      },
-      'claude-3-sonnet-20240229': {
-        displayName: 'Claude 3 Sonnet',
-        model: new ChatAnthropic({
-          temperature: 0.7,
-          anthropicApiKey: anthropicApiKey,
-          model: 'claude-3-sonnet-20240229',
-        }),
-      },
-      'claude-3-haiku-20240307': {
-        displayName: 'Claude 3 Haiku',
-        model: new ChatAnthropic({
-          temperature: 0.7,
-          anthropicApiKey: anthropicApiKey,
-          model: 'claude-3-haiku-20240307',
-        }),
-      },
-    };
+          configuration: {
+            baseURL: 'https://api.anthropic.com/v1/',
+          },
+        }) as unknown as BaseChatModel,
+      };
+    });
 
     return chatModels;
   } catch (err) {
-    logger.error(`Error loading Anthropic models: ${err}`);
+    console.error(`Error loading Anthropic models: ${err}`);
     return {};
   }
 };

src/lib/providers/gemini.ts

@@ -1,9 +1,42 @@
-import {
-  ChatGoogleGenerativeAI,
-  GoogleGenerativeAIEmbeddings,
-} from '@langchain/google-genai';
-import { getGeminiApiKey } from '../../config';
-import logger from '../../utils/logger';
+import { ChatOpenAI, OpenAIEmbeddings } from '@langchain/openai';
+import { getGeminiApiKey } from '../config';
+import { ChatModel, EmbeddingModel } from '.';
+import { BaseChatModel } from '@langchain/core/language_models/chat_models';
+import { Embeddings } from '@langchain/core/embeddings';
+
+const geminiChatModels: Record<string, string>[] = [
+  {
+    displayName: 'Gemini 2.0 Flash',
+    key: 'gemini-2.0-flash',
+  },
+  {
+    displayName: 'Gemini 2.0 Flash-Lite',
+    key: 'gemini-2.0-flash-lite',
+  },
+  {
+    displayName: 'Gemini 2.0 Pro Experimental',
+    key: 'gemini-2.0-pro-exp-02-05',
+  },
+  {
+    displayName: 'Gemini 1.5 Flash',
+    key: 'gemini-1.5-flash',
+  },
+  {
+    displayName: 'Gemini 1.5 Flash-8B',
+    key: 'gemini-1.5-flash-8b',
+  },
+  {
+    displayName: 'Gemini 1.5 Pro',
+    key: 'gemini-1.5-pro',
+  },
+];
+
+const geminiEmbeddingModels: Record<string, string>[] = [
+  {
+    displayName: 'Gemini Embedding',
+    key: 'gemini-embedding-exp',
+  },
+];
 
 export const loadGeminiChatModels = async () => {
   const geminiApiKey = getGeminiApiKey();
@@ -11,75 +44,53 @@ export const loadGeminiChatModels = async () => {
   if (!geminiApiKey) return {};
 
   try {
-    const chatModels = {
-      'gemini-1.5-flash': {
-        displayName: 'Gemini 1.5 Flash',
-        model: new ChatGoogleGenerativeAI({
-          modelName: 'gemini-1.5-flash',
+    const chatModels: Record<string, ChatModel> = {};
+
+    geminiChatModels.forEach((model) => {
+      chatModels[model.key] = {
+        displayName: model.displayName,
+        model: new ChatOpenAI({
+          openAIApiKey: geminiApiKey,
+          modelName: model.key,
           temperature: 0.7,
-          apiKey: geminiApiKey,
-        }),
-      },
-      'gemini-1.5-flash-8b': {
-        displayName: 'Gemini 1.5 Flash 8B',
-        model: new ChatGoogleGenerativeAI({
-          modelName: 'gemini-1.5-flash-8b',
-          temperature: 0.7,
-          apiKey: geminiApiKey,
-        }),
-      },
-      'gemini-1.5-pro': {
-        displayName: 'Gemini 1.5 Pro',
-        model: new ChatGoogleGenerativeAI({
-          modelName: 'gemini-1.5-pro',
-          temperature: 0.7,
-          apiKey: geminiApiKey,
-        }),
-      },
-      'gemini-2.0-flash-exp': {
-        displayName: 'Gemini 2.0 Flash Exp',
-        model: new ChatGoogleGenerativeAI({
-          modelName: 'gemini-2.0-flash-exp',
-          temperature: 0.7,
-          apiKey: geminiApiKey,
-        }),
-      },
-      'gemini-2.0-flash-thinking-exp-01-21': {
-        displayName: 'Gemini 2.0 Flash Thinking Exp 01-21',
-        model: new ChatGoogleGenerativeAI({
-          modelName: 'gemini-2.0-flash-thinking-exp-01-21',
-          temperature: 0.7,
-          apiKey: geminiApiKey,
-        }),
-      },
-    };
+          configuration: {
+            baseURL: 'https://generativelanguage.googleapis.com/v1beta/openai/',
+          },
+        }) as unknown as BaseChatModel,
+      };
+    });
 
     return chatModels;
   } catch (err) {
-    logger.error(`Error loading Gemini models: ${err}`);
+    console.error(`Error loading Gemini models: ${err}`);
     return {};
   }
 };
 
-export const loadGeminiEmbeddingsModels = async () => {
+export const loadGeminiEmbeddingModels = async () => {
   const geminiApiKey = getGeminiApiKey();
 
   if (!geminiApiKey) return {};
 
   try {
-    const embeddingModels = {
-      'text-embedding-004': {
-        displayName: 'Text Embedding',
-        model: new GoogleGenerativeAIEmbeddings({
-          apiKey: geminiApiKey,
-          modelName: 'text-embedding-004',
-        }),
-      },
-    };
+    const embeddingModels: Record<string, EmbeddingModel> = {};
+
+    geminiEmbeddingModels.forEach((model) => {
+      embeddingModels[model.key] = {
+        displayName: model.displayName,
+        model: new OpenAIEmbeddings({
+          openAIApiKey: geminiApiKey,
+          modelName: model.key,
+          configuration: {
+            baseURL: 'https://generativelanguage.googleapis.com/v1beta/openai/',
+          },
+        }) as unknown as Embeddings,
+      };
+    });
 
     return embeddingModels;
   } catch (err) {
-    logger.error(`Error loading Gemini embeddings model: ${err}`);
+    console.error(`Error loading OpenAI embeddings models: ${err}`);
     return {};
   }
 };

src/lib/providers/groq.ts

@@ -1,6 +1,82 @@
 import { ChatOpenAI } from '@langchain/openai';
-import { getGroqApiKey } from '../../config';
-import logger from '../../utils/logger';
+import { getGroqApiKey } from '../config';
+import { ChatModel } from '.';
+import { BaseChatModel } from '@langchain/core/language_models/chat_models';
+
+const groqChatModels: Record<string, string>[] = [
+  {
+    displayName: 'Gemma2 9B IT',
+    key: 'gemma2-9b-it',
+  },
+  {
+    displayName: 'Llama 3.3 70B Versatile',
+    key: 'llama-3.3-70b-versatile',
+  },
+  {
+    displayName: 'Llama 3.1 8B Instant',
+    key: 'llama-3.1-8b-instant',
+  },
+  {
+    displayName: 'Llama3 70B 8192',
+    key: 'llama3-70b-8192',
+  },
+  {
+    displayName: 'Llama3 8B 8192',
+    key: 'llama3-8b-8192',
+  },
+  {
+    displayName: 'Mixtral 8x7B 32768',
+    key: 'mixtral-8x7b-32768',
+  },
+  {
+    displayName: 'Qwen QWQ 32B (Preview)',
+    key: 'qwen-qwq-32b',
+  },
+  {
+    displayName: 'Mistral Saba 24B (Preview)',
+    key: 'mistral-saba-24b',
+  },
+  {
+    displayName: 'Qwen 2.5 Coder 32B (Preview)',
+    key: 'qwen-2.5-coder-32b',
+  },
+  {
+    displayName: 'Qwen 2.5 32B (Preview)',
+    key: 'qwen-2.5-32b',
+  },
+  {
+    displayName: 'DeepSeek R1 Distill Qwen 32B (Preview)',
+    key: 'deepseek-r1-distill-qwen-32b',
+  },
+  {
+    displayName: 'DeepSeek R1 Distill Llama 70B SpecDec (Preview)',
+    key: 'deepseek-r1-distill-llama-70b-specdec',
+  },
+  {
+    displayName: 'DeepSeek R1 Distill Llama 70B (Preview)',
+    key: 'deepseek-r1-distill-llama-70b',
+  },
+  {
+    displayName: 'Llama 3.3 70B SpecDec (Preview)',
+    key: 'llama-3.3-70b-specdec',
+  },
+  {
+    displayName: 'Llama 3.2 1B Preview (Preview)',
+    key: 'llama-3.2-1b-preview',
+  },
+  {
+    displayName: 'Llama 3.2 3B Preview (Preview)',
+    key: 'llama-3.2-3b-preview',
+  },
+  {
+    displayName: 'Llama 3.2 11B Vision Preview (Preview)',
+    key: 'llama-3.2-11b-vision-preview',
+  },
+  {
+    displayName: 'Llama 3.2 90B Vision Preview (Preview)',
+    key: 'llama-3.2-90b-vision-preview',
+  },
+];
 
 export const loadGroqChatModels = async () => {
   const groqApiKey = getGroqApiKey();
@@ -8,129 +84,25 @@ export const loadGroqChatModels = async () => {
   if (!groqApiKey) return {};
 
   try {
-    const chatModels = {
-      'llama-3.3-70b-versatile': {
-        displayName: 'Llama 3.3 70B',
-        model: new ChatOpenAI(
-          {
-            openAIApiKey: groqApiKey,
-            modelName: 'llama-3.3-70b-versatile',
-            temperature: 0.7,
-          },
-          {
+    const chatModels: Record<string, ChatModel> = {};
+
+    groqChatModels.forEach((model) => {
+      chatModels[model.key] = {
+        displayName: model.displayName,
+        model: new ChatOpenAI({
+          openAIApiKey: groqApiKey,
+          modelName: model.key,
+          temperature: 0.7,
+          configuration: {
             baseURL: 'https://api.groq.com/openai/v1',
           },
-        ),
-      },
-      'llama-3.2-3b-preview': {
-        displayName: 'Llama 3.2 3B',
-        model: new ChatOpenAI(
-          {
-            openAIApiKey: groqApiKey,
-            modelName: 'llama-3.2-3b-preview',
-            temperature: 0.7,
-          },
-          {
-            baseURL: 'https://api.groq.com/openai/v1',
-          },
-        ),
-      },
-      'llama-3.2-11b-vision-preview': {
-        displayName: 'Llama 3.2 11B Vision',
-        model: new ChatOpenAI(
-          {
-            openAIApiKey: groqApiKey,
-            modelName: 'llama-3.2-11b-vision-preview',
-            temperature: 0.7,
-          },
-          {
-            baseURL: 'https://api.groq.com/openai/v1',
-          },
-        ),
-      },
-      'llama-3.2-90b-vision-preview': {
-        displayName: 'Llama 3.2 90B Vision',
-        model: new ChatOpenAI(
-          {
-            openAIApiKey: groqApiKey,
-            modelName: 'llama-3.2-90b-vision-preview',
-            temperature: 0.7,
-          },
-          {
-            baseURL: 'https://api.groq.com/openai/v1',
-          },
-        ),
-      },
-      'llama-3.1-8b-instant': {
-        displayName: 'Llama 3.1 8B',
-        model: new ChatOpenAI(
-          {
-            openAIApiKey: groqApiKey,
-            modelName: 'llama-3.1-8b-instant',
-            temperature: 0.7,
-          },
-          {
-            baseURL: 'https://api.groq.com/openai/v1',
-          },
-        ),
-      },
-      'llama3-8b-8192': {
-        displayName: 'LLaMA3 8B',
-        model: new ChatOpenAI(
-          {
-            openAIApiKey: groqApiKey,
-            modelName: 'llama3-8b-8192',
-            temperature: 0.7,
-          },
-          {
-            baseURL: 'https://api.groq.com/openai/v1',
-          },
-        ),
-      },
-      'llama3-70b-8192': {
-        displayName: 'LLaMA3 70B',
-        model: new ChatOpenAI(
-          {
-            openAIApiKey: groqApiKey,
-            modelName: 'llama3-70b-8192',
-            temperature: 0.7,
-          },
-          {
-            baseURL: 'https://api.groq.com/openai/v1',
-          },
-        ),
-      },
-      'mixtral-8x7b-32768': {
-        displayName: 'Mixtral 8x7B',
-        model: new ChatOpenAI(
-          {
-            openAIApiKey: groqApiKey,
-            modelName: 'mixtral-8x7b-32768',
-            temperature: 0.7,
-          },
-          {
-            baseURL: 'https://api.groq.com/openai/v1',
-          },
-        ),
-      },
-      'gemma2-9b-it': {
-        displayName: 'Gemma2 9B',
-        model: new ChatOpenAI(
-          {
-            openAIApiKey: groqApiKey,
-            modelName: 'gemma2-9b-it',
-            temperature: 0.7,
-          },
-          {
-            baseURL: 'https://api.groq.com/openai/v1',
-          },
-        ),
-      },
-    };
+        }) as unknown as BaseChatModel,
+      };
+    });
 
     return chatModels;
   } catch (err) {
-    logger.error(`Error loading Groq models: ${err}`);
+    console.error(`Error loading Groq models: ${err}`);
     return {};
   }
 };

src/lib/providers/index.ts

@@ -1,33 +1,49 @@
-import { loadGroqChatModels } from './groq';
-import { loadOllamaChatModels, loadOllamaEmbeddingsModels } from './ollama';
-import { loadOpenAIChatModels, loadOpenAIEmbeddingsModels } from './openai';
-import { loadAnthropicChatModels } from './anthropic';
-import { loadTransformersEmbeddingsModels } from './transformers';
-import { loadGeminiChatModels, loadGeminiEmbeddingsModels } from './gemini';
+import { Embeddings } from '@langchain/core/embeddings';
+import { BaseChatModel } from '@langchain/core/language_models/chat_models';
+import { loadOpenAIChatModels, loadOpenAIEmbeddingModels } from './openai';
 import {
   getCustomOpenaiApiKey,
   getCustomOpenaiApiUrl,
   getCustomOpenaiModelName,
-} from '../../config';
+} from '../config';
 import { ChatOpenAI } from '@langchain/openai';
+import { loadOllamaChatModels, loadOllamaEmbeddingModels } from './ollama';
+import { loadGroqChatModels } from './groq';
+import { loadAnthropicChatModels } from './anthropic';
+import { loadGeminiChatModels, loadGeminiEmbeddingModels } from './gemini';
 
-const chatModelProviders = {
+export interface ChatModel {
+  displayName: string;
+  model: BaseChatModel;
+}
+
+export interface EmbeddingModel {
+  displayName: string;
+  model: Embeddings;
+}
+
+export const chatModelProviders: Record<
+  string,
+  () => Promise<Record<string, ChatModel>>
+> = {
   openai: loadOpenAIChatModels,
-  groq: loadGroqChatModels,
   ollama: loadOllamaChatModels,
+  groq: loadGroqChatModels,
   anthropic: loadAnthropicChatModels,
   gemini: loadGeminiChatModels,
 };
 
-const embeddingModelProviders = {
-  openai: loadOpenAIEmbeddingsModels,
-  local: loadTransformersEmbeddingsModels,
-  ollama: loadOllamaEmbeddingsModels,
-  gemini: loadGeminiEmbeddingsModels,
+export const embeddingModelProviders: Record<
+  string,
+  () => Promise<Record<string, EmbeddingModel>>
+> = {
+  openai: loadOpenAIEmbeddingModels,
+  ollama: loadOllamaEmbeddingModels,
+  gemini: loadGeminiEmbeddingModels,
 };
 
 export const getAvailableChatModelProviders = async () => {
-  const models = {};
+  const models: Record<string, Record<string, ChatModel>> = {};
 
   for (const provider in chatModelProviders) {
     const providerModels = await chatModelProviders[provider]();
@@ -52,7 +68,7 @@ export const getAvailableChatModelProviders = async () => {
               configuration: {
                 baseURL: customOpenAiApiUrl,
               },
-            }),
+            }) as unknown as BaseChatModel,
           },
         }
       : {}),
@@ -62,7 +78,7 @@ export const getAvailableChatModelProviders = async () => {
 };
 
 export const getAvailableEmbeddingModelProviders = async () => {
-  const models = {};
+  const models: Record<string, Record<string, EmbeddingModel>> = {};
 
   for (const provider in embeddingModelProviders) {
     const providerModels = await embeddingModelProviders[provider]();

src/lib/providers/ollama.ts

@@ -1,74 +1,73 @@
-import { OllamaEmbeddings } from '@langchain/community/embeddings/ollama';
-import { getKeepAlive, getOllamaApiEndpoint } from '../../config';
-import logger from '../../utils/logger';
-import { ChatOllama } from '@langchain/community/chat_models/ollama';
 import axios from 'axios';
+import { getKeepAlive, getOllamaApiEndpoint } from '../config';
+import { ChatModel, EmbeddingModel } from '.';
+import { ChatOllama } from '@langchain/community/chat_models/ollama';
+import { OllamaEmbeddings } from '@langchain/community/embeddings/ollama';
 
 export const loadOllamaChatModels = async () => {
-  const ollamaEndpoint = getOllamaApiEndpoint();
-  const keepAlive = getKeepAlive();
+  const ollamaApiEndpoint = getOllamaApiEndpoint();
 
-  if (!ollamaEndpoint) return {};
+  if (!ollamaApiEndpoint) return {};
 
   try {
-    const response = await axios.get(`${ollamaEndpoint}/api/tags`, {
+    const res = await axios.get(`${ollamaApiEndpoint}/api/tags`, {
       headers: {
         'Content-Type': 'application/json',
       },
     });
 
-    const { models: ollamaModels } = response.data;
+    const { models } = res.data;
 
-    const chatModels = ollamaModels.reduce((acc, model) => {
-      acc[model.model] = {
+    const chatModels: Record<string, ChatModel> = {};
+
+    models.forEach((model: any) => {
+      chatModels[model.model] = {
         displayName: model.name,
         model: new ChatOllama({
-          baseUrl: ollamaEndpoint,
+          baseUrl: ollamaApiEndpoint,
           model: model.model,
           temperature: 0.7,
-          keepAlive: keepAlive,
+          keepAlive: getKeepAlive(),
         }),
       };
-
-      return acc;
-    }, {});
+    });
 
     return chatModels;
   } catch (err) {
-    logger.error(`Error loading Ollama models: ${err}`);
+    console.error(`Error loading Ollama models: ${err}`);
     return {};
   }
 };
 
-export const loadOllamaEmbeddingsModels = async () => {
-  const ollamaEndpoint = getOllamaApiEndpoint();
+export const loadOllamaEmbeddingModels = async () => {
+  const ollamaApiEndpoint = getOllamaApiEndpoint();
 
-  if (!ollamaEndpoint) return {};
+  if (!ollamaApiEndpoint) return {};
 
   try {
-    const response = await axios.get(`${ollamaEndpoint}/api/tags`, {
+    const res = await axios.get(`${ollamaApiEndpoint}/api/tags`, {
       headers: {
         'Content-Type': 'application/json',
       },
     });
 
-    const { models: ollamaModels } = response.data;
+    const { models } = res.data;
 
-    const embeddingsModels = ollamaModels.reduce((acc, model) => {
-      acc[model.model] = {
+    const embeddingModels: Record<string, EmbeddingModel> = {};
+
+    models.forEach((model: any) => {
+      embeddingModels[model.model] = {
         displayName: model.name,
         model: new OllamaEmbeddings({
-          baseUrl: ollamaEndpoint,
+          baseUrl: ollamaApiEndpoint,
           model: model.model,
         }),
       };
+    });
 
-      return acc;
-    }, {});
-
-    return embeddingsModels;
+    return embeddingModels;
   } catch (err) {
-    logger.error(`Error loading Ollama embeddings model: ${err}`);
+    console.error(`Error loading Ollama embeddings models: ${err}`);
     return {};
   }
 };

src/lib/providers/openai.ts

@@ -1,89 +1,90 @@
 import { ChatOpenAI, OpenAIEmbeddings } from '@langchain/openai';
-import { getOpenaiApiKey } from '../../config';
-import logger from '../../utils/logger';
+import { getOpenaiApiKey } from '../config';
+import { ChatModel, EmbeddingModel } from '.';
+import { BaseChatModel } from '@langchain/core/language_models/chat_models';
+import { Embeddings } from '@langchain/core/embeddings';
+
+const openaiChatModels: Record<string, string>[] = [
+  {
+    displayName: 'GPT-3.5 Turbo',
+    key: 'gpt-3.5-turbo',
+  },
+  {
+    displayName: 'GPT-4',
+    key: 'gpt-4',
+  },
+  {
+    displayName: 'GPT-4 turbo',
+    key: 'gpt-4-turbo',
+  },
+  {
+    displayName: 'GPT-4 omni',
+    key: 'gpt-4o',
+  },
+  {
+    displayName: 'GPT-4 omni mini',
+    key: 'gpt-4o-mini',
+  },
+];
+
+const openaiEmbeddingModels: Record<string, string>[] = [
+  {
+    displayName: 'Text Embedding 3 Small',
+    key: 'text-embedding-3-small',
+  },
+  {
+    displayName: 'Text Embedding 3 Large',
+    key: 'text-embedding-3-large',
+  },
+];
 
 export const loadOpenAIChatModels = async () => {
-  const openAIApiKey = getOpenaiApiKey();
+  const openaiApiKey = getOpenaiApiKey();
 
-  if (!openAIApiKey) return {};
+  if (!openaiApiKey) return {};
 
   try {
-    const chatModels = {
-      'gpt-3.5-turbo': {
-        displayName: 'GPT-3.5 Turbo',
+    const chatModels: Record<string, ChatModel> = {};
+
+    openaiChatModels.forEach((model) => {
+      chatModels[model.key] = {
+        displayName: model.displayName,
         model: new ChatOpenAI({
-          openAIApiKey,
-          modelName: 'gpt-3.5-turbo',
+          openAIApiKey: openaiApiKey,
+          modelName: model.key,
           temperature: 0.7,
-        }),
-      },
-      'gpt-4': {
-        displayName: 'GPT-4',
-        model: new ChatOpenAI({
-          openAIApiKey,
-          modelName: 'gpt-4',
-          temperature: 0.7,
-        }),
-      },
-      'gpt-4-turbo': {
-        displayName: 'GPT-4 turbo',
-        model: new ChatOpenAI({
-          openAIApiKey,
-          modelName: 'gpt-4-turbo',
-          temperature: 0.7,
-        }),
-      },
-      'gpt-4o': {
-        displayName: 'GPT-4 omni',
-        model: new ChatOpenAI({
-          openAIApiKey,
-          modelName: 'gpt-4o',
-          temperature: 0.7,
-        }),
-      },
-      'gpt-4o-mini': {
-        displayName: 'GPT-4 omni mini',
-        model: new ChatOpenAI({
-          openAIApiKey,
-          modelName: 'gpt-4o-mini',
-          temperature: 0.7,
-        }),
-      },
-    };
+        }) as unknown as BaseChatModel,
+      };
+    });
 
     return chatModels;
   } catch (err) {
-    logger.error(`Error loading OpenAI models: ${err}`);
+    console.error(`Error loading OpenAI models: ${err}`);
     return {};
   }
 };
 
-export const loadOpenAIEmbeddingsModels = async () => {
-  const openAIApiKey = getOpenaiApiKey();
+export const loadOpenAIEmbeddingModels = async () => {
+  const openaiApiKey = getOpenaiApiKey();
 
-  if (!openAIApiKey) return {};
+  if (!openaiApiKey) return {};
 
   try {
-    const embeddingModels = {
-      'text-embedding-3-small': {
-        displayName: 'Text Embedding 3 Small',
+    const embeddingModels: Record<string, EmbeddingModel> = {};
+
+    openaiEmbeddingModels.forEach((model) => {
+      embeddingModels[model.key] = {
+        displayName: model.displayName,
         model: new OpenAIEmbeddings({
-          openAIApiKey,
-          modelName: 'text-embedding-3-small',
-        }),
-      },
-      'text-embedding-3-large': {
-        displayName: 'Text Embedding 3 Large',
-        model: new OpenAIEmbeddings({
-          openAIApiKey,
-          modelName: 'text-embedding-3-large',
-        }),
-      },
-    };
+          openAIApiKey: openaiApiKey,
+          modelName: model.key,
+        }) as unknown as Embeddings,
+      };
+    });
 
     return embeddingModels;
   } catch (err) {
-    logger.error(`Error loading OpenAI embeddings model: ${err}`);
+    console.error(`Error loading OpenAI embeddings models: ${err}`);
     return {};
   }
 };

src/lib/providers/transformers.ts

@@ -1,32 +0,0 @@
-import logger from '../../utils/logger';
-import { HuggingFaceTransformersEmbeddings } from '../huggingfaceTransformer';
-
-export const loadTransformersEmbeddingsModels = async () => {
-  try {
-    const embeddingModels = {
-      'xenova-bge-small-en-v1.5': {
-        displayName: 'BGE Small',
-        model: new HuggingFaceTransformersEmbeddings({
-          modelName: 'Xenova/bge-small-en-v1.5',
-        }),
-      },
-      'xenova-gte-small': {
-        displayName: 'GTE Small',
-        model: new HuggingFaceTransformersEmbeddings({
-          modelName: 'Xenova/gte-small',
-        }),
-      },
-      'xenova-bert-base-multilingual-uncased': {
-        displayName: 'Bert Multilingual',
-        model: new HuggingFaceTransformersEmbeddings({
-          modelName: 'Xenova/bert-base-multilingual-uncased',
-        }),
-      },
-    };
-
-    return embeddingModels;
-  } catch (err) {
-    logger.error(`Error loading Transformers embeddings model: ${err}`);
-    return {};
-  }
-};