feat: add new models

sgomez · Dec 6, 2024 · 7de9ec5 · 7de9ec5
1 parent 4a06cdc
commit 7de9ec5
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/packages/ollama/src/ollama-chat-settings.ts b/packages/ollama/src/ollama-chat-settings.ts
@@ -64,6 +64,8 @@ export type OllamaChatModelId =
   | 'llama3.2-vision'
   | 'llama3.2-vision:11b'
   | 'llama3.2-vision:90b'
+  | 'llama3.3'
+  | 'llama3.3:70b'
   | 'llama-guard3'
   | 'llama-guard3:1b'
   | 'llama-guard3:8b'
@@ -75,6 +77,8 @@ export type OllamaChatModelId =
   | 'llava-llama3:8b'
   | 'llava-phi3'
   | 'llava-phi3:3.8b'
+  | 'marco-o1'
+  | 'marco-o1:7b'
   | 'mistral'
   | 'mistral:7b'
   | 'mistral-large'
@@ -128,6 +132,12 @@ export type OllamaChatModelId =
   | 'qwen2.5-coder:7b'
   | 'qwen2.5-coder:14b'
   | 'qwen2.5-coder:32b'
+  | 'qwq'
+  | 'qwq:32b'
+  | 'sailor2'
+  | 'sailor2:1b'
+  | 'sailor2:8b'
+  | 'sailor2:20b'
   | 'shieldgemma'
   | 'shieldgemma:2b'
   | 'shieldgemma:9b'
@@ -136,6 +146,8 @@ export type OllamaChatModelId =
   | 'smollm:135m'
   | 'smollm:360m'
   | 'smollm:1.7b'
+  | 'tinyllama'
+  | 'tinyllama:1.1b'
   | 'tulu3'
   | 'tulu3:8b'
   | 'tulu3:70b'

diff --git a/packages/ollama/src/ollama-embedding-settings.ts b/packages/ollama/src/ollama-embedding-settings.ts
@@ -15,6 +15,8 @@ export type OllamaEmbeddingModelId =
   | 'snowflake-arctic-embed:22m'
   | 'snowflake-arctic-embed:335m'
   | 'snowflake-arctic-embed:33m'
+  | 'snowflake-arctic-embed2'
+  | 'snowflake-arctic-embed2:568m'
   | OllamaChatModelId
   | (string & NonNullable<unknown>)