feat : using llama3 model

rajadilipkolli · May 10, 2024 · a950cbd · a950cbd
1 parent d9a1d41
commit a950cbd
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/rag/rag-springai-ollama-llm/docker/docker-compose.yml b/rag/rag-springai-ollama-llm/docker/docker-compose.yml
@@ -1,7 +1,7 @@
 version: '3.7'
 services:
     ollama:
-        image: langchain4j/ollama-llama2:latest
+        image: langchain4j/ollama-llama3:latest
         ports:
           - '11434:11434'
     redis-stack:

diff --git a/rag/rag-springai-ollama-llm/src/main/resources/application.properties b/rag/rag-springai-ollama-llm/src/main/resources/application.properties
@@ -3,12 +3,12 @@ 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=llama3
 spring.ai.ollama.chat.options.temperature=0.3
 spring.ai.ollama.chat.options.top-k=2
 spring.ai.ollama.chat.options.top-p=0.2
 
-spring.ai.ollama.embedding.options.model=llama2
+spring.ai.ollama.embedding.options.model=llama3
 
 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
@@ -19,7 +19,7 @@ OllamaContainer ollama(DynamicPropertyRegistry properties) {
         // The model name to use (e.g., "orca-mini", "mistral", "llama2", "codellama", "phi", or
         // "tinyllama")
         OllamaContainer ollama = new OllamaContainer(
-                DockerImageName.parse("langchain4j/ollama-llama2:latest").asCompatibleSubstituteFor("ollama/ollama"));
+                DockerImageName.parse("langchain4j/ollama-llama3:latest").asCompatibleSubstituteFor("ollama/ollama"));
         properties.add("spring.ai.ollama.base-url", ollama::getEndpoint);
         return ollama;
     }