From 6fd342c2148bffbdcb73c6e23dfae844cfc71434 Mon Sep 17 00:00:00 2001 From: Ignasi Date: Mon, 28 Oct 2024 18:11:30 +0100 Subject: [PATCH] Using llama3.2:1b to reduce the size of the docker image to download (#5) --- .../java/com/example/demogenai/ContainersConfiguration.java | 2 +- src/test/resources/application.properties | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/test/java/com/example/demogenai/ContainersConfiguration.java b/src/test/java/com/example/demogenai/ContainersConfiguration.java index 2b25314..5f6eb5e 100644 --- a/src/test/java/com/example/demogenai/ContainersConfiguration.java +++ b/src/test/java/com/example/demogenai/ContainersConfiguration.java @@ -15,7 +15,7 @@ public class ContainersConfiguration { @ServiceConnection @RestartScope OllamaContainer ollama() { - return new OllamaContainer(DockerImageName.parse("ilopezluna/evaluator-llama3.1:0.3.11-8b") + return new OllamaContainer(DockerImageName.parse("ilopezluna/spring-ai-demo-models:llama3.2-1b") .asCompatibleSubstituteFor("ollama/ollama")) { }; } diff --git a/src/test/resources/application.properties b/src/test/resources/application.properties index 4140dc5..fc0dc4c 100644 --- a/src/test/resources/application.properties +++ b/src/test/resources/application.properties @@ -4,7 +4,7 @@ spring.ai.ollama.embedding.options.top-k=2 spring.ai.ollama.embedding.options.seed=42 spring.ai.ollama.embedding.options.temperature=0.0 -spring.ai.ollama.chat.options.model=llama3.1:8b +spring.ai.ollama.chat.options.model=llama3.2:1b spring.ai.ollama.chat.options.top-k=1 spring.ai.ollama.chat.options.seed=42 spring.ai.ollama.chat.options.temperature=0.0