feat(providers): add anthropic

src/config.ts

@@ -12,6 +12,7 @@ interface Config {
   API_KEYS: {
     OPENAI: string;
     GROQ: string;
+    ANTHROPIC: string;
   };
   API_ENDPOINTS: {
     SEARXNG: string;
@@ -37,6 +38,8 @@ export const getOpenaiApiKey = () => loadConfig().API_KEYS.OPENAI;
 
 export const getGroqApiKey = () => loadConfig().API_KEYS.GROQ;
 
+export const getAnthropicApiKey = () => loadConfig().API_KEYS.ANTHROPIC;
+
 export const getSearxngApiEndpoint = () => loadConfig().API_ENDPOINTS.SEARXNG;
 
 export const getOllamaApiEndpoint = () => loadConfig().API_ENDPOINTS.OLLAMA;

src/lib/providers/anthropic.ts

@@ -0,0 +1,39 @@
+import { ChatAnthropic } from '@langchain/anthropic';
+import { getAnthropicApiKey } from '../../config';
+import logger from '../../utils/logger';
+
+export const loadAnthropicChatModels = async () => {
+  const anthropicApiKey = getAnthropicApiKey();
+
+  if (!anthropicApiKey) return {};
+
+  try {
+    const chatModels = {
+      'Claude 3.5 Sonnet': new ChatAnthropic({
+        temperature: 0.7,
+        anthropicApiKey: anthropicApiKey,
+        model: 'claude-3-5-sonnet-20240620',
+      }),
+      'Claude 3 Opus': new ChatAnthropic({
+        temperature: 0.7,
+        anthropicApiKey: anthropicApiKey,
+        model: 'claude-3-opus-20240229',
+      }),
+      'Claude 3 Sonnet': new ChatAnthropic({
+        temperature: 0.7,
+        anthropicApiKey: anthropicApiKey,
+        model: 'claude-3-sonnet-20240229',
+      }),
+      'Claude 3 Haiku': new ChatAnthropic({
+        temperature: 0.7,
+        anthropicApiKey: anthropicApiKey,
+        model: 'claude-3-haiku-20240307',
+      }),
+    };
+
+    return chatModels;
+  } catch (err) {
+    logger.error(`Error loading Anthropic models: ${err}`);
+    return {};
+  }
+};

src/lib/providers/index.ts

@@ -1,12 +1,14 @@
 import { loadGroqChatModels } from './groq';
 import { loadOllamaChatModels, loadOllamaEmbeddingsModels } from './ollama';
 import { loadOpenAIChatModels, loadOpenAIEmbeddingsModels } from './openai';
+import { loadAnthropicChatModels } from './anthropic';
 import { loadTransformersEmbeddingsModels } from './transformers';
 
 const chatModelProviders = {
   openai: loadOpenAIChatModels,
   groq: loadGroqChatModels,
   ollama: loadOllamaChatModels,
+  anthropic: loadAnthropicChatModels,
 };
 
 const embeddingModelProviders = {
@@ -21,11 +23,11 @@ export const getAvailableChatModelProviders = async () => {
   for (const provider in chatModelProviders) {
     const providerModels = await chatModelProviders[provider]();
     if (Object.keys(providerModels).length > 0) {
-      models[provider] = providerModels
+      models[provider] = providerModels;
     }
   }
 
-  models['custom_openai'] = {}
+  models['custom_openai'] = {};
 
   return models;
 };
@@ -36,7 +38,7 @@ export const getAvailableEmbeddingModelProviders = async () => {
   for (const provider in embeddingModelProviders) {
     const providerModels = await embeddingModelProviders[provider]();
     if (Object.keys(providerModels).length > 0) {
-      models[provider] = providerModels
+      models[provider] = providerModels;
     }
   }
 

src/routes/config.ts

@@ -6,6 +6,7 @@ import {
 import {
   getGroqApiKey,
   getOllamaApiEndpoint,
+  getAnthropicApiKey,
   getOpenaiApiKey,
   updateConfig,
 } from '../config';
@@ -37,6 +38,7 @@ router.get('/', async (_, res) => {
 
   config['openaiApiKey'] = getOpenaiApiKey();
   config['ollamaApiUrl'] = getOllamaApiEndpoint();
+  config['anthropicApiKey'] = getAnthropicApiKey();
   config['groqApiKey'] = getGroqApiKey();
 
   res.status(200).json(config);
@@ -49,6 +51,7 @@ router.post('/', async (req, res) => {
     API_KEYS: {
       OPENAI: config.openaiApiKey,
       GROQ: config.groqApiKey,
+      ANTHROPIC: config.anthropicApiKey,
     },
     API_ENDPOINTS: {
       OLLAMA: config.ollamaApiUrl,