feat: add more models

sgomez · Nov 6, 2024 · 1143b83 · 1143b83
1 parent a535e4d
commit 1143b83
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/packages/ollama/src/ollama-chat-settings.ts b/packages/ollama/src/ollama-chat-settings.ts
@@ -1,4 +1,7 @@
 export type OllamaChatModelId =
+  | 'aya-expanse'
+  | 'aya-expanse:8b'
+  | 'aya-expanse:32b'
   | 'codegemma'
   | 'codegemma:2b'
   | 'codegemma:7b'
@@ -30,6 +33,9 @@ export type OllamaChatModelId =
   | 'granite3-dense'
   | 'granite3-dense:2b'
   | 'granite3-dense:8b'
+  | 'granite3-guardian'
+  | 'granite3-guardian:2b'
+  | 'granite3-guardian:8b'
   | 'granite3-moe'
   | 'granite3-moe:1b'
   | 'granite3-moe:3b'
@@ -53,6 +59,12 @@ export type OllamaChatModelId =
   | 'llama3.2'
   | 'llama3.2:1b'
   | 'llama3.2:3b'
+  | 'llama3.2-vision'
+  | 'llama3.2-vision:11b'
+  | 'llama3.2-vision:90b'
+  | 'llama-guard3'
+  | 'llama-guard3:1b'
+  | 'llama-guard3:8b'
   | 'llava'
   | 'llava:7b'
   | 'llava:13b'
@@ -76,6 +88,8 @@ export type OllamaChatModelId =
   | 'moondream:1.8b'
   | 'openhermes'
   | 'openhermes:v2.5'
+  | 'nemotron'
+  | 'nemotron:70b'
   | 'nemotron-mini'
   | 'nemotron-mini:4b'
   | 'phi3'
@@ -105,6 +119,10 @@ export type OllamaChatModelId =
   | 'qwen2.5-coder'
   | 'qwen2.5-coder:1.5b'
   | 'qwen2.5-coder:7b'
+  | 'shieldgemma'
+  | 'shieldgemma:2b'
+  | 'shieldgemma:9b'
+  | 'shieldgemma:27b'
   | 'smollm'
   | 'smollm:135m'
   | 'smollm:360m'