feat : using orca-mini for demo purpose

rajadilipkolli · Mar 28, 2024 · 3d148e6 · 3d148e6
1 parent 2524683
commit 3d148e6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/rag-springai-ollama-llm/src/main/resources/application.properties b/rag-springai-ollama-llm/src/main/resources/application.properties
@@ -3,10 +3,10 @@ spring.application.name=rag-springai-ollama-llm
 spring.threads.virtual.enabled=true
 spring.mvc.problemdetails.enabled=true
 
-spring.ai.ollama.chat.options.model=llama2
+spring.ai.ollama.chat.options.model=orca-mini
 spring.ai.ollama.chat.options.responseFormat=json_object
 
-spring.ai.ollama.embedding.options.model=llama2
+spring.ai.ollama.embedding.options.model=orca-mini
 
 spring.ai.vectorstore.redis.index=vector_store
 spring.ai.vectorstore.redis.prefix=ai
diff --git a/...m/src/test/java/com/learning/ai/llmragwithspringai/TestLlmRagWithSpringAiApplication.java b/...m/src/test/java/com/learning/ai/llmragwithspringai/TestLlmRagWithSpringAiApplication.java
@@ -16,8 +16,8 @@ public class TestLlmRagWithSpringAiApplication {
 
     @Bean
     OllamaContainer ollama(DynamicPropertyRegistry properties) {
-        OllamaContainer ollama = new OllamaContainer(
-                DockerImageName.parse("ghcr.io/thomasvitale/ollama-llama2").asCompatibleSubstituteFor("ollama/ollama"));
+        OllamaContainer ollama = new OllamaContainer(DockerImageName.parse("langchain4j/ollama-orca-mini:latest")
+                .asCompatibleSubstituteFor("ollama/ollama"));
         properties.add("spring.ai.ollama.base-url", ollama::getEndpoint);
         return ollama;
     }