feat(providers): add anthropic

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;
     }
   }