Refactor chat model references and update LangChain4j version to 1.7.1

Author: arey

pom.xml

@@ -38,9 +38,21 @@
     <maven-checkstyle.version>3.6.0</maven-checkstyle.version>
     <nohttp-checkstyle.version>0.0.11</nohttp-checkstyle.version>
     <spring-format.version>0.0.43</spring-format.version>
-    <langchain4j.version>1.0.0-beta3</langchain4j.version>
+    <langchain4j.version>1.7.1</langchain4j.version>
   </properties>
 
+  <dependencyManagement>
+    <dependencies>
+      <dependency>
+        <groupId>dev.langchain4j</groupId>
+        <artifactId>langchain4j-bom</artifactId>
+        <version>${langchain4j.version}</version>
+        <type>pom</type>
+        <scope>import</scope>
+      </dependency>
+    </dependencies>
+  </dependencyManagement>
+
   <dependencies>
     <!-- Spring and Spring Boot dependencies -->
     <dependency>
@@ -77,12 +89,10 @@
     <dependency>
       <groupId>dev.langchain4j</groupId>
       <artifactId>langchain4j-spring-boot-starter</artifactId>
-      <version>${langchain4j.version}</version>
     </dependency>
     <dependency>
       <groupId>dev.langchain4j</groupId>
       <artifactId>langchain4j-embeddings-all-minilm-l6-v2</artifactId>
-      <version>${langchain4j.version}</version>
     </dependency>
 
     <dependency>
@@ -486,7 +496,6 @@
         <dependency>
           <groupId>dev.langchain4j</groupId>
           <artifactId>langchain4j-open-ai-spring-boot-starter</artifactId>
-          <version>${langchain4j.version}</version>
         </dependency>
       </dependencies>
     </profile>
@@ -496,7 +505,6 @@
         <dependency>
           <groupId>dev.langchain4j</groupId>
           <artifactId>langchain4j-azure-open-ai-spring-boot-starter</artifactId>
-          <version>${langchain4j.version}</version>
         </dependency>
       </dependencies>
     </profile>
@@ -506,7 +514,6 @@
         <dependency>
           <groupId>dev.langchain4j</groupId>
           <artifactId>langchain4j-ollama-spring-boot-starter</artifactId>
-          <version>${langchain4j.version}</version>
         </dependency>
       </dependencies>
     </profile>

src/main/java/org/springframework/samples/petclinic/chat/AssistantConfiguration.java

@@ -3,7 +3,7 @@
 import dev.langchain4j.data.segment.TextSegment;
 import dev.langchain4j.memory.chat.ChatMemoryProvider;
 import dev.langchain4j.memory.chat.MessageWindowChatMemory;
-import dev.langchain4j.model.chat.ChatLanguageModel;
+import dev.langchain4j.model.chat.ChatModel;
 import dev.langchain4j.model.embedding.EmbeddingModel;
 import dev.langchain4j.model.embedding.onnx.allminilml6v2.AllMiniLmL6V2EmbeddingModel;
 import dev.langchain4j.rag.DefaultRetrievalAugmentor;
@@ -42,9 +42,9 @@ EmbeddingStoreContentRetriever contentRetriever(InMemoryEmbeddingStore<TextSegme
 	}
 
 	@Bean
-	RetrievalAugmentor retrievalAugmentor(ChatLanguageModel chatLanguageModel, ContentRetriever vetContentRetriever) {
+	RetrievalAugmentor retrievalAugmentor(ChatModel chatModel, ContentRetriever vetContentRetriever) {
 		return DefaultRetrievalAugmentor.builder()
-			.queryRouter(new VetQueryRouter(chatLanguageModel, vetContentRetriever))
+			.queryRouter(new VetQueryRouter(chatModel, vetContentRetriever))
 			.build();
 	}
 

src/main/java/org/springframework/samples/petclinic/chat/VetQueryRouter.java

@@ -1,7 +1,7 @@
 package org.springframework.samples.petclinic.chat;
 
 import dev.langchain4j.data.message.AiMessage;
-import dev.langchain4j.model.chat.ChatLanguageModel;
+import dev.langchain4j.model.chat.ChatModel;
 import dev.langchain4j.model.input.Prompt;
 import dev.langchain4j.model.input.PromptTemplate;
 import dev.langchain4j.rag.content.retriever.ContentRetriever;
@@ -43,18 +43,18 @@ class VetQueryRouter implements QueryRouter {
 
 	private final ContentRetriever vetContentRetriever;
 
-	private final ChatLanguageModel chatLanguageModel;
+	private final ChatModel chatModel;
 
-	public VetQueryRouter(ChatLanguageModel chatLanguageModel, ContentRetriever vetContentRetriever) {
-		this.chatLanguageModel = chatLanguageModel;
+	public VetQueryRouter(ChatModel chatModel, ContentRetriever vetContentRetriever) {
+		this.chatModel = chatModel;
 		this.vetContentRetriever = vetContentRetriever;
 	}
 
 	@Override
 	public Collection<ContentRetriever> route(Query query) {
 		Prompt prompt = PROMPT_TEMPLATE.apply(query.text());
 
-		AiMessage aiMessage = chatLanguageModel.chat(prompt.toUserMessage()).aiMessage();
+		AiMessage aiMessage = chatModel.chat(prompt.toUserMessage()).aiMessage();
 		LOGGER.debug("LLM decided: {}", aiMessage.text());
 
 		if (aiMessage.text().toLowerCase().contains("yes")) {