From 3d148e6440e189b983bb899e4d185da7d32f4221 Mon Sep 17 00:00:00 2001 From: Raja Kolli Date: Thu, 28 Mar 2024 09:22:02 +0000 Subject: [PATCH] feat : using orca-mini for demo purpose --- .../src/main/resources/application.properties | 4 ++-- .../llmragwithspringai/TestLlmRagWithSpringAiApplication.java | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/rag-springai-ollama-llm/src/main/resources/application.properties b/rag-springai-ollama-llm/src/main/resources/application.properties index 1af2760..ab8f62d 100644 --- 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/rag-springai-ollama-llm/src/test/java/com/learning/ai/llmragwithspringai/TestLlmRagWithSpringAiApplication.java b/rag-springai-ollama-llm/src/test/java/com/learning/ai/llmragwithspringai/TestLlmRagWithSpringAiApplication.java index 597baad..5f45f51 100644 --- a/rag-springai-ollama-llm/src/test/java/com/learning/ai/llmragwithspringai/TestLlmRagWithSpringAiApplication.java +++ b/rag-springai-ollama-llm/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; }