From fd56276c5f911f0dce05b806a974997cfd62ae52 Mon Sep 17 00:00:00 2001
From: z30057876 <zhouweitong@huawei.com>
Date: Thu, 23 Jan 2025 16:57:06 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8Dkb=5Fid=E8=AE=BE=E7=BD=AE?=
 =?UTF-8?q?=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/manager/knowledge.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/apps/manager/knowledge.py b/apps/manager/knowledge.py
index dc2250444..d56c3b6a6 100644
--- a/apps/manager/knowledge.py
+++ b/apps/manager/knowledge.py
@@ -16,10 +16,11 @@ class KnowledgeBaseManager:
         """修改当前用户的知识库ID"""
         user_collection = MongoDB.get_collection("user")
         try:
-            result = await user_collection.update_one({"_id": user_sub}, {"$set": {"kb_id": kb_id}})
-            if result.modified_count == 0:
+            user = await user_collection.find_one({"_id": user_sub}, {"kb_id": 1})
+            if user is None:
                 LOGGER.error("[KnowledgeBaseManager] change kb_id error: user_sub not found")
                 return False
+            await user_collection.update_one({"_id": user_sub}, {"$set": {"kb_id": kb_id}})
             return True
         except Exception as e:
             LOGGER.error(f"[KnowledgeBaseManager] change kb_id error: {e}")
-- 
Gitee