Default Ollama to use llama3.2 (#889)

* Default Ollama to use llama3.2 * Update tests for llama3.2 * DRY it up a bit * Add VCR cassettes
patterns-ai-core · Dec 2, 2024 · 454f6ba · 454f6ba
1 parent 4b447e3
commit 454f6ba
Show file tree

Hide file tree

Showing 11 changed files with 228 additions and 343 deletions.
diff --git a/lib/langchain/llm/ollama.rb b/lib/langchain/llm/ollama.rb
@@ -12,9 +12,9 @@ class Ollama < Base
 
     DEFAULTS = {
       temperature: 0.0,
-      completion_model: "llama3.1",
-      embedding_model: "llama3.1",
-      chat_model: "llama3.1",
+      completion_model: "llama3.2",
+      embedding_model: "llama3.2",
+      chat_model: "llama3.2",
       options: {}
     }.freeze
 
@@ -24,6 +24,7 @@ class Ollama < Base
       llama2: 4_096,
       llama3: 4_096,
       "llama3.1": 4_096,
+      "llama3.2": 4_096,
       llava: 4_096,
       mistral: 4_096,
       "mistral-openorca": 4_096,

diff --git a/...sions_when_the_dimensions_size_is_generated_by_the_model_sets_the_dimensions_to_1_024.yml b/...sions_when_the_dimensions_size_is_generated_by_the_model_sets_the_dimensions_to_1_024.yml
diff --git a/spec/fixtures/vcr_cassettes/Langchain_LLM_Ollama_chat_returns_a_chat_completion.yml b/spec/fixtures/vcr_cassettes/Langchain_LLM_Ollama_chat_returns_a_chat_completion.yml
diff --git a/...ixtures/vcr_cassettes/Langchain_LLM_Ollama_chat_returns_a_chat_completion_format_json.yml b/...ixtures/vcr_cassettes/Langchain_LLM_Ollama_chat_returns_a_chat_completion_format_json.yml