From 43140e695b08a6d0ec725b21c65d5214c168dd0f Mon Sep 17 00:00:00 2001
From: RickyZZQ <39479607+RickyZZQ@users.noreply.github.com>
Date: Fri, 3 Jan 2025 09:05:28 +0800
Subject: [PATCH] [fix]Fix user feedback for query memory bug (#1992)

---
 .../supersonic/chat/server/service/impl/MemoryServiceImpl.java | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/chat/server/src/main/java/com/tencent/supersonic/chat/server/service/impl/MemoryServiceImpl.java b/chat/server/src/main/java/com/tencent/supersonic/chat/server/service/impl/MemoryServiceImpl.java
index e7f828683..abb45c438 100644
--- a/chat/server/src/main/java/com/tencent/supersonic/chat/server/service/impl/MemoryServiceImpl.java
+++ b/chat/server/src/main/java/com/tencent/supersonic/chat/server/service/impl/MemoryServiceImpl.java
@@ -113,6 +113,9 @@ public class MemoryServiceImpl implements MemoryService {
         if (chatMemoryFilter.getAgentId() != null) {
             queryWrapper.lambda().eq(ChatMemoryDO::getAgentId, chatMemoryFilter.getAgentId());
         }
+        if (chatMemoryFilter.getQueryId() != null) {
+            queryWrapper.lambda().eq(ChatMemoryDO::getQueryId, chatMemoryFilter.getQueryId());
+        }
         if (StringUtils.isNotBlank(chatMemoryFilter.getQuestion())) {
             queryWrapper.lambda().like(ChatMemoryDO::getQuestion, chatMemoryFilter.getQuestion());
         }