Prechádzať zdrojové kódy

add default models for Ollama

alexchenzl 3 mesiacov pred
rodič
commit
6298338b5c

+ 1 - 1
packages/storage/lib/settings/llmProviders.ts

@@ -117,7 +117,7 @@ export function getDefaultProviderConfig(providerId: string): ProviderConfig {
         apiKey: 'ollama', // Set default API key for Ollama
         name: getDefaultDisplayNameFromProviderId(ProviderTypeEnum.Ollama),
         type: ProviderTypeEnum.Ollama,
-        modelNames: [], // Ollama uses modelNames (user adds them)
+        modelNames: llmProviderModelNames[providerId],
         baseUrl: 'http://localhost:11434',
         createdAt: Date.now(),
       };

+ 1 - 1
packages/storage/lib/settings/types.ts

@@ -33,7 +33,7 @@ export const llmProviderModelNames = {
     // 'gemini-2.0-flash-thinking-exp-01-21', // TODO: not support function calling for now
   ],
   [ProviderTypeEnum.Grok]: ['grok-3', 'grok-3-fast', 'grok-3-mini', 'grok-3-mini-fast'],
-  [ProviderTypeEnum.Ollama]: [],
+  [ProviderTypeEnum.Ollama]: ['falcon3:10b', 'qwen2.5-coder:14b', 'mistral-small:24b'],
   [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-4.1',