diff --git a/src/main/java/com/huangge1199/ai/config/LangChainConfig.java b/src/main/java/com/huangge1199/ai/config/LangChainConfig.java index 68f1141..4f7ffcc 100644 --- a/src/main/java/com/huangge1199/ai/config/LangChainConfig.java +++ b/src/main/java/com/huangge1199/ai/config/LangChainConfig.java @@ -1,6 +1,8 @@ package com.huangge1199.ai.config; import com.huangge1199.ai.service.LangChainService; +import dev.langchain4j.memory.ChatMemory; +import dev.langchain4j.memory.chat.MessageWindowChatMemory; import dev.langchain4j.model.chat.ChatModel; import dev.langchain4j.service.AiServices; import jakarta.annotation.Resource; @@ -21,6 +23,10 @@ public class LangChainConfig { @Bean public LangChainService langChainService() { - return AiServices.create(LangChainService.class, qwenChatModel); + ChatMemory chatMemory = MessageWindowChatMemory.withMaxMessages(10); + return AiServices.builder(LangChainService.class) + .chatModel(qwenChatModel) + .chatMemory(chatMemory) + .build(); } }