Merge branch 'main' of github.com:l3vels/L3AGI

l3vels · Mar 11, 2024 · b227aba · b227aba
2 parents 33ebcea + 89af6a1
commit b227aba
Show file tree

Hide file tree

Showing 4 changed files with 60 additions and 0 deletions.
diff --git a/apps/server/typings/model.py b/apps/server/typings/model.py
@@ -8,6 +8,7 @@ class ModelProviders(Enum):
     OPEN_AI = "OpenAI"
     HUGGING_FACE = "Hugging Face"
     REPLICATE = "Replicate"
+    CLAUDE = "Claude"
 
     def __str__(self) -> str:
         return self.value

diff --git a/apps/server/utils/model.py b/apps/server/utils/model.py
@@ -30,6 +30,22 @@
         "fine_tuning": True,
         "is_voice": True,
     },
+    {
+        "id": "171e01f0-ca4a-41fb-b00d-3c49b0b7f579",
+        "provider": ModelProviders.OPEN_AI,
+        "name": "GPT-3.5 0125",
+        "value": "gpt-3.5-turbo-0125",
+        "fine_tuning": False,
+        "is_voice": True,
+    },
+    {
+        "id": "d7531988-ed64-4801-b8f6-60cbf778897a",
+        "provider": ModelProviders.OPEN_AI,
+        "name": "GPT-3.5 1106",
+        "value": "gpt-3.5-turbo-1106",
+        "fine_tuning": False,
+        "is_voice": True,
+    },
     {
         "id": "f4fc991a-8ef8-4ae7-a395-a22057f33d4a",
         "provider": ModelProviders.OPEN_AI,
@@ -46,6 +62,30 @@
         "fine_tuning": False,
         "is_voice": True,
     },
+    {
+        "id": "1191a1ca-d716-41c1-997c-30eaa5dbb7b4",
+        "provider": ModelProviders.OPEN_AI,
+        "name": "GPT-4 0125",
+        "value": "gpt-4-0125-preview",
+        "fine_tuning": False,
+        "is_voice": True,
+    },
+    {
+        "id": "3300eec2-1426-49ce-a2af-bb3a371b8fd4",
+        "provider": ModelProviders.OPEN_AI,
+        "name": "GPT-4 1106",
+        "value": "gpt-4-1106-preview",
+        "fine_tuning": False,
+        "is_voice": True,
+    },
+    {
+        "id": "09cdc79f-3e8d-49f8-a4d2-544656a23b74",
+        "provider": ModelProviders.OPEN_AI,
+        "name": "GPT-4 32k",
+        "value": "gpt-4-32k",
+        "fine_tuning": False,
+        "is_voice": True,
+    },
     {
         "id": "308346e9-80d1-44c9-b2e1-5c261a0c5931",
         "provider": ModelProviders.HUGGING_FACE,
@@ -74,6 +114,20 @@
         "value": "meta/llama-2-70b-chat:latest",
         "fine_tuning": False,
     },
+    {
+        "id": "56b4eaae-f74e-4fd7-9b4b-ffd7d1e8d90b",
+        "provider": ModelProviders.CLAUDE,
+        "name": "Claude 3 Opus",
+        "value": "claude-3-opus-20240229",
+        "fine_tuning": False,
+    },
+    {
+        "id": "1ed94d6a-a0a4-4256-8239-57fe35773a1a",
+        "provider": ModelProviders.CLAUDE,
+        "name": "Claude 3 Sonnet",
+        "value": "claude-3-sonnet-20240229",
+        "fine_tuning": False,
+    },
 ]
 
 

diff --git a/apps/ui/src/assets/models/claude-ai-icon.webp b/apps/ui/src/assets/models/claude-ai-icon.webp
diff --git a/apps/ui/src/pages/Models/constants.ts b/apps/ui/src/pages/Models/constants.ts
@@ -1,6 +1,7 @@
 import GPTLogo from 'assets/models/openai.jpg'
 import HuggingFaceLogo from 'assets/models/hugging-face.png'
 import ReplicateLogo from 'assets/models/replicate.png'
+import ClaudeLogo from 'assets/models/claude-ai-icon.webp'
 
 export const MODEL_PROVIDER_LOGOS = [
   {
@@ -15,4 +16,8 @@ export const MODEL_PROVIDER_LOGOS = [
     logoSrc: ReplicateLogo,
     provider: 'Replicate',
   },
+  {
+    logoSrc: ClaudeLogo,
+    provider: 'Claude',
+  },
 ]