feat(providers): separate each provider

src/lib/providers/ollama.ts

@@ -0,0 +1,59 @@
+import { OllamaEmbeddings } from '@langchain/community/embeddings/ollama';
+import { getOllamaApiEndpoint } from '../../config';
+import logger from '../../utils/logger';
+import { ChatOllama } from '@langchain/community/chat_models/ollama';
+
+export const loadOllamaChatModels = async () => {
+  const ollamaEndpoint = getOllamaApiEndpoint();
+
+  try {
+    const response = await fetch(`${ollamaEndpoint}/api/tags`, {
+      headers: {
+        'Content-Type': 'application/json',
+      },
+    });
+
+    const { models: ollamaModels } = (await response.json()) as any;
+
+    const chatModels = ollamaModels.reduce((acc, model) => {
+      acc[model.model] = new ChatOllama({
+        baseUrl: ollamaEndpoint,
+        model: model.model,
+        temperature: 0.7,
+      });
+      return acc;
+    }, {});
+
+    return chatModels;
+  } catch (err) {
+    logger.error(`Error loading Ollama models: ${err}`);
+    return {};
+  }
+};
+
+export const loadOpenAIEmbeddingsModel = async () => {
+  const ollamaEndpoint = getOllamaApiEndpoint();
+
+  try {
+    const response = await fetch(`${ollamaEndpoint}/api/tags`, {
+      headers: {
+        'Content-Type': 'application/json',
+      },
+    });
+
+    const { models: ollamaModels } = (await response.json()) as any;
+
+    const embeddingsModels = ollamaModels.reduce((acc, model) => {
+      acc[model.model] = new OllamaEmbeddings({
+        baseUrl: ollamaEndpoint,
+        model: model.model,
+      });
+      return acc;
+    }, {});
+
+    return embeddingsModels;
+  } catch (err) {
+    logger.error(`Error loading Ollama embeddings model: ${err}`);
+    return {};
+  }
+};