From b188da8595280363d2407ae8512b9937acb83196 Mon Sep 17 00:00:00 2001
From: beat4ocean <105700600+beat4ocean@users.noreply.github.com>
Date: Mon, 7 Apr 2025 08:57:06 +0800
Subject: [PATCH] [Fix] cannot save schema-mapping and semantic-S2SQL
 modification. (#2210)

---
 .../chat/server/service/impl/MemoryServiceImpl.java         | 6 ++++++
 1 file changed, 6 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 f6e895549..b28e6336a 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
@@ -91,6 +91,12 @@ public class MemoryServiceImpl implements MemoryService, CommandLineRunner {
             updateWrapper.set(ChatMemoryDO::getHumanReviewCmt,
                     chatMemoryUpdateReq.getHumanReviewCmt());
         }
+        if (Objects.nonNull(chatMemoryUpdateReq.getDbSchema())) {
+            updateWrapper.set(ChatMemoryDO::getDbSchema, chatMemoryUpdateReq.getDbSchema());
+        }
+        if (Objects.nonNull(chatMemoryUpdateReq.getS2sql())) {
+            updateWrapper.set(ChatMemoryDO::getS2sql, chatMemoryUpdateReq.getS2sql());
+        }
         updateWrapper.set(ChatMemoryDO::getUpdatedAt, new Date());
         updateWrapper.set(ChatMemoryDO::getUpdatedBy, user.getName());