Parcourir la source

update default models and parameters for providers

alexchenzl il y a 3 mois
Parent
commit
74c7102926
1 fichiers modifiés avec 47 ajouts et 50 suppressions
  1. 47 50
      packages/storage/lib/settings/types.ts

+ 47 - 50
packages/storage/lib/settings/types.ts

@@ -22,28 +22,25 @@ export enum ProviderTypeEnum {
 
 // Default supported models for each built-in provider
 export const llmProviderModelNames = {
-  [ProviderTypeEnum.OpenAI]: ['gpt-4o', 'gpt-4o-mini', 'o1', 'o3-mini'],
+  [ProviderTypeEnum.OpenAI]: ['gpt-4.1', 'gpt-4.1-mini', 'gpt-4.1-nano', 'gpt-4o', 'gpt-4o-mini', 'o4-mini', 'o3'],
   [ProviderTypeEnum.Anthropic]: ['claude-3-7-sonnet-latest', 'claude-3-5-sonnet-latest', 'claude-3-5-haiku-latest'],
   [ProviderTypeEnum.DeepSeek]: ['deepseek-chat', 'deepseek-reasoner'],
   [ProviderTypeEnum.Gemini]: [
-    'gemini-2.0-flash',
-    'gemini-2.0-flash-lite',
+    'gemini-2.5-flash-preview-04-17',
     'gemini-2.5-pro-exp-03-25',
+    'gemini-2.5-pro-preview-03-25',
+    'gemini-2.0-flash',
     // 'gemini-2.0-flash-thinking-exp-01-21', // TODO: not support function calling for now
   ],
-  [ProviderTypeEnum.Grok]: ['grok-2', 'grok-2-vision'],
+  [ProviderTypeEnum.Grok]: ['grok-3', 'grok-3-fast', 'grok-3-mini', 'grok-3-mini-fast'],
   [ProviderTypeEnum.Ollama]: [],
-  [ProviderTypeEnum.AzureOpenAI]: ['gpt-4o', 'gpt-4o-mini', 'o3-mini', 'gpt-4.1', 'gpt-4.1-mini', 'gpt-4.1-nano'],
+  [ProviderTypeEnum.AzureOpenAI]: ['gpt-4.1', 'gpt-4.1-mini', 'gpt-4.1-nano', 'gpt-4o', 'gpt-4o-mini', 'o4-mini', 'o3'],
   [ProviderTypeEnum.OpenRouter]: [
-    'openai/gpt-4o-2024-11-20',
     'openai/gpt-4.1',
     'openai/gpt-4.1-mini',
-    'openai/gpt-4.1-nano',
     'openai/o4-mini',
-    'anthropic/claude-3.5-sonnet',
-    'anthropic/claude-3.7-sonnet',
-    'google/gemini-2.0-flash-001',
-    'deepseek/deepseek-chat-v3-0324:free',
+    'openai/gpt-4o-2024-11-20',
+    'google/gemini-2.5-flash-preview',
   ],
   // Custom OpenAI providers don't have predefined models as they are user-defined
 };
@@ -52,44 +49,44 @@ export const llmProviderModelNames = {
 export const llmProviderParameters = {
   [ProviderTypeEnum.OpenAI]: {
     [AgentNameEnum.Planner]: {
-      temperature: 0.01,
-      topP: 0.001,
+      temperature: 0.7,
+      topP: 0.9,
     },
     [AgentNameEnum.Navigator]: {
-      temperature: 0,
-      topP: 0.001,
+      temperature: 0.3,
+      topP: 0.85,
     },
     [AgentNameEnum.Validator]: {
-      temperature: 0,
-      topP: 0.001,
+      temperature: 0.1,
+      topP: 0.8,
     },
   },
   [ProviderTypeEnum.Anthropic]: {
     [AgentNameEnum.Planner]: {
-      temperature: 0.1,
-      topP: 0.1,
+      temperature: 0.3,
+      topP: 0.6,
     },
     [AgentNameEnum.Navigator]: {
-      temperature: 0.1,
-      topP: 0.1,
+      temperature: 0.2,
+      topP: 0.5,
     },
     [AgentNameEnum.Validator]: {
-      temperature: 0.05,
-      topP: 0.1,
+      temperature: 0.1,
+      topP: 0.5,
     },
   },
   [ProviderTypeEnum.Gemini]: {
     [AgentNameEnum.Planner]: {
-      temperature: 0.01,
-      topP: 0.1,
+      temperature: 0.7,
+      topP: 0.9,
     },
     [AgentNameEnum.Navigator]: {
-      temperature: 0.01,
-      topP: 0.1,
+      temperature: 0.3,
+      topP: 0.85,
     },
     [AgentNameEnum.Validator]: {
       temperature: 0.1,
-      topP: 0.1,
+      topP: 0.8,
     },
   },
   [ProviderTypeEnum.Grok]: {
@@ -98,54 +95,54 @@ export const llmProviderParameters = {
       topP: 0.9,
     },
     [AgentNameEnum.Navigator]: {
-      temperature: 0.7,
-      topP: 0.9,
+      temperature: 0.3,
+      topP: 0.85,
     },
     [AgentNameEnum.Validator]: {
-      temperature: 0.7,
-      topP: 0.9,
+      temperature: 0.1,
+      topP: 0.8,
     },
   },
   [ProviderTypeEnum.Ollama]: {
     [AgentNameEnum.Planner]: {
-      temperature: 0,
-      topP: 0.001,
+      temperature: 0.3,
+      topP: 0.9,
     },
     [AgentNameEnum.Navigator]: {
-      temperature: 0.01,
-      topP: 0.001,
+      temperature: 0.1,
+      topP: 0.85,
     },
     [AgentNameEnum.Validator]: {
-      temperature: 0,
-      topP: 0.001,
+      temperature: 0.1,
+      topP: 0.8,
     },
   },
   [ProviderTypeEnum.AzureOpenAI]: {
     [AgentNameEnum.Planner]: {
-      temperature: 0.01,
-      topP: 0.001,
+      temperature: 0.7,
+      topP: 0.9,
     },
     [AgentNameEnum.Navigator]: {
-      temperature: 0,
-      topP: 0.001,
+      temperature: 0.3,
+      topP: 0.85,
     },
     [AgentNameEnum.Validator]: {
-      temperature: 0,
-      topP: 0.001,
+      temperature: 0.1,
+      topP: 0.8,
     },
   },
   [ProviderTypeEnum.OpenRouter]: {
     [AgentNameEnum.Planner]: {
-      temperature: 0.01,
-      topP: 0.001,
+      temperature: 0.7,
+      topP: 0.9,
     },
     [AgentNameEnum.Navigator]: {
-      temperature: 0,
-      topP: 0.001,
+      temperature: 0.3,
+      topP: 0.85,
     },
     [AgentNameEnum.Validator]: {
-      temperature: 0,
-      topP: 0.001,
+      temperature: 0.1,
+      topP: 0.8,
     },
   },
 };