From 110a520a404b6878e0aef088261d0cc6e1273fee Mon Sep 17 00:00:00 2001 From: Christian Tzolov Date: Sun, 29 Sep 2024 15:53:24 +0200 Subject: [PATCH] Add Llama 3.2 models to OllamaModel enum - Include LLAMA3_2 (3B) and LLAMA3_2_1B (1B) variants --- .../org/springframework/ai/ollama/api/OllamaModel.java | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/models/spring-ai-ollama/src/main/java/org/springframework/ai/ollama/api/OllamaModel.java b/models/spring-ai-ollama/src/main/java/org/springframework/ai/ollama/api/OllamaModel.java index 32481d810..5687e2aeb 100644 --- a/models/spring-ai-ollama/src/main/java/org/springframework/ai/ollama/api/OllamaModel.java +++ b/models/spring-ai-ollama/src/main/java/org/springframework/ai/ollama/api/OllamaModel.java @@ -41,6 +41,16 @@ public enum OllamaModel implements ChatModelDescription { */ LLAMA3_1("llama3.1"), + /** + * The Llama 3.2 3B language model from Meta. + */ + LLAMA3_2("llama3.2"), + + /** + * The Llama 3.2 1B language model from Meta. + */ + LLAMA3_2_1B("llama3.2:1b"), + /** * The 7B parameters model */