Merge pull request #719 from ashakirin/fix/moved-conversation_id-to-chat_memory_service

fix: moved adding conversation id into ChatMemoryService

Author: ybezsonov

samples/spring-ai/assistant/src/main/java/com/example/assistant/service/ChatMemoryService.java

@@ -1,5 +1,7 @@
 package com.example.assistant.service;
 
+import org.springframework.ai.chat.client.ChatClient;
+import org.springframework.ai.chat.memory.ChatMemory;
 import org.springframework.ai.chat.memory.MessageWindowChatMemory;
 import org.springframework.ai.chat.memory.repository.jdbc.JdbcChatMemoryRepository;
 import org.springframework.ai.chat.memory.repository.jdbc.PostgresChatMemoryRepositoryDialect;
@@ -51,9 +53,9 @@ public String getCurrentConversationId() {
     /**
      * Add the conversation ID parameter to the advisor
      */
-    public void addConversationIdToAdvisor(Object advisor) {
+    public void addConversationIdToAdvisor(ChatClient.AdvisorSpec advisor) {
         String conversationId = getCurrentConversationId();
-        advisor.getClass().cast(advisor);
         logger.debug("Adding conversation ID {} to advisor", conversationId);
+        advisor.param(ChatMemory.CONVERSATION_ID, conversationId);
     }
 }

samples/spring-ai/assistant/src/main/java/com/example/assistant/service/ChatService.java

@@ -63,7 +63,7 @@ public String processChat(ChatRequest request) {
                 return RetryUtils.executeWithRetry(() -> {
                     var chatResponse = chatClient
                         .prompt().user(request.prompt())
-                        .advisors(advisor -> advisor.param(ChatMemory.CONVERSATION_ID, chatMemoryService.getCurrentConversationId()))
+                        .advisors(chatMemoryService::addConversationIdToAdvisor)
                         .call().chatResponse();
 
                     return (chatResponse != null) ? chatResponse.getResult().getOutput().getText() : "I don't know - no response received.";