From 92c034436585ae8a33047b52cf141124d2732f7d Mon Sep 17 00:00:00 2001
From: huangge1199 <huangge1199@hotmail.com>
Date: Sat, 12 Jul 2025 10:51:11 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=9A=E8=AF=9D=E8=AE=B0=E5=BF=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../java/com/huangge1199/ai/config/LangChainConfig.java   | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

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();
     }
 }