Merge pull request #515 from dcSpark/bence/add-claude-models

feat: add claude models
dcSpark · Nov 6, 2024 · 050cb67 · 050cb67
2 parents e73ab8c + dd56d14
commit 050cb67
Show file tree

Hide file tree

Showing 7 changed files with 43 additions and 0 deletions.
diff --git a/apps/shinkai-desktop/src/components/chat/constants.ts b/apps/shinkai-desktop/src/components/chat/constants.ts
@@ -5,6 +5,7 @@ export const streamingSupportedModels = [
   Models.Exo,
   Models.Gemini,
   Models.OpenRouter,
+  Models.Claude,
 ];
 
 export const ARTIFACTS_SYSTEM_PROMPT = `

diff --git a/apps/shinkai-desktop/src/pages/create-agent.tsx b/apps/shinkai-desktop/src/pages/create-agent.tsx
@@ -67,6 +67,10 @@ const modelOptions: { value: Models; label: string }[] = [
     value: Models.Exo,
     label: 'Exo',
   },
+  {
+    value: Models.Claude,
+    label: 'Claude',
+  },
 ];
 
 export const getModelObject = (
@@ -84,6 +88,8 @@ export const getModelObject = (
       return { Gemini: { model_type: modelType } };
     case Models.Exo:
       return { Exo: { model_type: modelType } };
+    case Models.Claude:
+      return { Claude: { model_type: modelType } };
     default:
       return { [model]: { model_type: modelType } };
   }

diff --git a/apps/shinkai-visor/src/components/agents/add-agent.tsx b/apps/shinkai-visor/src/components/agents/add-agent.tsx
@@ -51,6 +51,8 @@ export const getModelObject = (
       return { Groq: { model_type: modelType } };
     case Models.OpenRouter:
       return { OpenRouter: { model_type: modelType } };
+    case Models.Claude:
+      return { Claude: { model_type: modelType } };
     default:
       return { [model]: { model_type: modelType } };
   }

diff --git a/libs/shinkai-message-ts/src/api/jobs/index.ts b/libs/shinkai-message-ts/src/api/jobs/index.ts
@@ -216,6 +216,8 @@ function getModelString(model: LLMProviderInterface): string {
     return 'openrouter:' + model.OpenRouter.model_type;
   } else if (model?.Exo?.model_type) {
     return 'exo:' + model.Exo.model_type;
+  } else if (model?.Claude?.model_type) {
+    return 'claude:' + model.Claude.model_type;
   } else if (Object.keys(model).length > 0) {
     const customModelProvider = Object.keys(model)[0];
     return `${customModelProvider}:${model[customModelProvider].model_type}`;

diff --git a/libs/shinkai-message-ts/src/api/jobs/types.ts b/libs/shinkai-message-ts/src/api/jobs/types.ts
@@ -184,6 +184,7 @@ export type LLMProviderInterface = {
   Exo?: Exo;
   Groq?: Groq;
   OpenRouter?: OpenRouter;
+  Claude?: Claude;
 } & {
   [model: string]: ModelType;
 };
@@ -219,6 +220,10 @@ export interface Exo {
   model_type: string;
 }
 
+export interface Claude {
+  model_type: string;
+}
+
 export type SerializedLLMProvider = {
   id: string;
   full_identity_name: string;

diff --git a/libs/shinkai-message-ts/src/models/SchemaTypes.ts b/libs/shinkai-message-ts/src/models/SchemaTypes.ts
@@ -268,6 +268,7 @@ export type AgentAPIModel = {
   Ollama?: Ollama;
   Gemini?: Gemini;
   Exo?: Exo;
+  Claude?: Claude;
 } & {
   [model: string]: ModelType;
 };
@@ -303,6 +304,10 @@ export interface Exo {
   model_type: string;
 }
 
+export interface Claude {
+  model_type: string;
+}
+
 export interface APIAddAgentRequest {
   agent: SerializedLLMProvider;
 }

diff --git a/libs/shinkai-node-state/src/lib/utils/models.ts b/libs/shinkai-node-state/src/lib/utils/models.ts
@@ -6,6 +6,7 @@ export enum Models {
   Exo = 'exo',
   Groq = 'groq',
   OpenRouter = 'openrouter',
+  Claude = 'claude',
 }
 
 export const modelsConfig = {
@@ -107,4 +108,25 @@ export const modelsConfig = {
     apiUrl: 'https://openrouter.ai',
     modelTypes: [],
   },
+  [Models.Claude]: {
+    apiUrl: 'https://api.anthropic.com',
+    modelTypes: [
+      {
+        name: 'Claude 3.5 Sonnet',
+        value: 'claude-3-5-sonnet-latest',
+      },
+      {
+        name: 'Claude 3 Opus',
+        value: 'claude-3-opus-latest',
+      },
+      {
+        name: 'Claude 3 Sonnet',
+        value: 'claude-3-sonnet-20240229',
+      },
+      {
+        name: 'Claude 3 Haiku',
+        value: 'claude-3-haiku-20240307',
+      },
+    ],
+  },
 };