chore(weave): Add gpt-4.5-preview and deepseek to playground (#3830)

* Add deepseek to playground * update model_providers: * add 4.5 * remove apikey field
wandb · Mar 4, 2025 · 3dffee0 · 3dffee0
1 parent 3cc7f6f
commit 3dffee0
Show file tree

Hide file tree

Showing 3 changed files with 74 additions and 1 deletion.
diff --git a/...e-js/src/components/PagePanelComponents/Home/Browse3/pages/PlaygroundPage/llmMaxTokens.ts b/...e-js/src/components/PagePanelComponents/Home/Browse3/pages/PlaygroundPage/llmMaxTokens.ts
@@ -107,6 +107,16 @@ export const LLM_MAX_TOKENS = {
     max_tokens: 100000,
     supports_function_calling: true,
   },
+  'gpt-4.5-preview': {
+    provider: 'openai',
+    max_tokens: 16384,
+    supports_function_calling: true,
+  },
+  'gpt-4.5-preview-2025-02-27': {
+    provider: 'openai',
+    max_tokens: 16384,
+    supports_function_calling: true,
+  },
 
   // Anthropic models
   'claude-3-7-sonnet-20250219': {
@@ -484,6 +494,18 @@ export const LLM_MAX_TOKENS = {
     provider: 'xai',
     supports_function_calling: true,
   },
+
+  // DeepSeek models
+  'deepseek/deepseek-reasoner': {
+    provider: 'deepseek',
+    max_tokens: 8192,
+    supports_function_calling: true,
+  },
+  'deepseek/deepseek-chat': {
+    provider: 'deepseek',
+    max_tokens: 8192,
+    supports_function_calling: true,
+  },
 };
 
 export type LLMMaxTokensKey = keyof typeof LLM_MAX_TOKENS;
@@ -500,6 +522,7 @@ export const LLM_PROVIDERS = [
   'groq',
   'bedrock',
   'xai',
+  'deepseek',
 ];
 
 export const LLM_PROVIDER_LABELS: Record<
@@ -513,4 +536,5 @@ export const LLM_PROVIDER_LABELS: Record<
   groq: 'Groq',
   bedrock: 'AWS Bedrock',
   xai: 'xAI',
+  deepseek: 'DeepSeek',
 };
diff --git a/weave/trace_server/model_providers/model_providers.json b/weave/trace_server/model_providers/model_providers.json
@@ -19,6 +19,14 @@
     "litellm_provider": "openai",
     "api_key_name": "OPENAI_API_KEY"
   },
+  "gpt-4.5-preview": {
+    "litellm_provider": "openai",
+    "api_key_name": "OPENAI_API_KEY"
+  },
+  "gpt-4.5-preview-2025-02-27": {
+    "litellm_provider": "openai",
+    "api_key_name": "OPENAI_API_KEY"
+  },
   "gpt-4o-audio-preview": {
     "litellm_provider": "openai",
     "api_key_name": "OPENAI_API_KEY"
@@ -555,6 +563,10 @@
     "litellm_provider": "azure_ai",
     "api_key_name": "AZURE_API_KEY"
   },
+  "azure_ai/Phi-4": {
+    "litellm_provider": "azure_ai",
+    "api_key_name": "AZURE_API_KEY"
+  },
   "azure_ai/Phi-3.5-mini-instruct": {
     "litellm_provider": "azure_ai",
     "api_key_name": "AZURE_API_KEY"
@@ -611,6 +623,14 @@
     "litellm_provider": "anthropic",
     "api_key_name": "ANTHROPIC_API_KEY"
   },
+  "deepseek/deepseek-reasoner": {
+    "litellm_provider": "deepseek",
+    "api_key_name": "DEEPSEEK_API_KEY"
+  },
+  "deepseek/deepseek-chat": {
+    "litellm_provider": "deepseek",
+    "api_key_name": "DEEPSEEK_API_KEY"
+  },
   "xai/grok-beta": {
     "litellm_provider": "xai",
     "api_key_name": "XAI_API_KEY"
@@ -643,6 +663,10 @@
     "litellm_provider": "xai",
     "api_key_name": "XAI_API_KEY"
   },
+  "deepseek/deepseek-coder": {
+    "litellm_provider": "deepseek",
+    "api_key_name": "DEEPSEEK_API_KEY"
+  },
   "groq/deepseek-r1-distill-llama-70b": {
     "litellm_provider": "groq",
     "api_key_name": "GEMMA_API_KEY"
@@ -743,6 +767,14 @@
     "litellm_provider": "anthropic",
     "api_key_name": "ANTHROPIC_API_KEY"
   },
+  "claude-3-5-haiku-latest": {
+    "litellm_provider": "anthropic",
+    "api_key_name": "ANTHROPIC_API_KEY"
+  },
+  "claude-3-opus-latest": {
+    "litellm_provider": "anthropic",
+    "api_key_name": "ANTHROPIC_API_KEY"
+  },
   "claude-3-opus-20240229": {
     "litellm_provider": "anthropic",
     "api_key_name": "ANTHROPIC_API_KEY"
@@ -751,10 +783,18 @@
     "litellm_provider": "anthropic",
     "api_key_name": "ANTHROPIC_API_KEY"
   },
+  "claude-3-5-sonnet-latest": {
+    "litellm_provider": "anthropic",
+    "api_key_name": "ANTHROPIC_API_KEY"
+  },
   "claude-3-5-sonnet-20240620": {
     "litellm_provider": "anthropic",
     "api_key_name": "ANTHROPIC_API_KEY"
   },
+  "claude-3-7-sonnet-latest": {
+    "litellm_provider": "anthropic",
+    "api_key_name": "ANTHROPIC_API_KEY"
+  },
   "claude-3-7-sonnet-20250219": {
     "litellm_provider": "anthropic",
     "api_key_name": "ANTHROPIC_API_KEY"
@@ -995,12 +1035,16 @@
     "litellm_provider": "bedrock",
     "api_key_name": "BEDROCK_API_KEY"
   },
+  "bedrock/invoke/anthropic.claude-3-5-sonnet-20240620-v1:0": {
+    "litellm_provider": "bedrock",
+    "api_key_name": "BEDROCK_API_KEY"
+  },
   "anthropic.claude-3-5-sonnet-20240620-v1:0": {
     "litellm_provider": "bedrock",
     "api_key_name": "BEDROCK_API_KEY"
   },
   "anthropic.claude-3-7-sonnet-20250219-v1:0": {
-    "litellm_provider": "bedrock",
+    "litellm_provider": "bedrock_converse",
     "api_key_name": "BEDROCK_API_KEY"
   },
   "anthropic.claude-3-5-sonnet-20241022-v2:0": {
@@ -1031,6 +1075,10 @@
     "litellm_provider": "bedrock",
     "api_key_name": "BEDROCK_API_KEY"
   },
+  "us.anthropic.claude-3-7-sonnet-20250219-v1:0": {
+    "litellm_provider": "bedrock_converse",
+    "api_key_name": "BEDROCK_API_KEY"
+  },
   "us.anthropic.claude-3-haiku-20240307-v1:0": {
     "litellm_provider": "bedrock",
     "api_key_name": "BEDROCK_API_KEY"

diff --git a/weave/trace_server/model_providers/model_providers.py b/weave/trace_server/model_providers/model_providers.py
@@ -15,6 +15,7 @@
     "azure_ai": "AZURE_API_KEY",
     "bedrock": "BEDROCK_API_KEY",
     "bedrock_converse": "BEDROCK_API_KEY",
+    "deepseek": "DEEPSEEK_API_KEY",
     "fireworks": "FIREWORKS_API_KEY",
     "gemini": "GOOGLE_API_KEY",
     "groq": "GEMMA_API_KEY",