feat: add more popular models to OllamaChatModelId

sgomez · Jun 28, 2024 · f9797ad · f9797ad
1 parent 0eacade
commit f9797ad
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/packages/ollama/src/ollama-chat-settings.ts b/packages/ollama/src/ollama-chat-settings.ts
@@ -11,6 +11,9 @@ export type OllamaChatModelId =
   | 'gemma'
   | 'gemma:2b'
   | 'gemma:7b'
+  | 'gemma2'
+  | 'gemma:9b'
+  | 'gemma:27b'
   | 'llama2'
   | 'llama2:7b'
   | 'llama2:13b'
@@ -41,6 +44,17 @@ export type OllamaChatModelId =
   | 'moondream:1.8b'
   | 'openhermes'
   | 'openhermes:v2.5'
+  | 'qwen'
+  | 'qwen:7b'
+  | 'qwen:14b'
+  | 'qwen:32b'
+  | 'qwen:72b'
+  | 'qwen:110b'
+  | 'qwen2'
+  | 'qwen2:0.5b'
+  | 'qwen2:1.5b'
+  | 'qwen2:7b'
+  | 'qwen2:72b'
   | 'phi3'
   | 'phi3:3.8b'
   | (string & NonNullable<unknown>)