diff --git a/weave-js/src/components/PagePanelComponents/Home/Browse3/pages/PlaygroundPage/llmMaxTokens.ts b/weave-js/src/components/PagePanelComponents/Home/Browse3/pages/PlaygroundPage/llmMaxTokens.ts index 1c623e4b2d85..421b30428bc1 100644 --- a/weave-js/src/components/PagePanelComponents/Home/Browse3/pages/PlaygroundPage/llmMaxTokens.ts +++ b/weave-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 index df387ebeea71..625b0491f1d8 100644 --- 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 index 7fe44b6fdc08..57e430d962cc 100644 --- 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",