klinsmaya/SQLBot
1
0
Fork 0
mirror of https://gh-proxy.org/https://github.com/klinsmaya/SQLBot.git synced 2026-11-08 01:06:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: Adjust the organizational attribution logic of dialogues in MCP

Browse Source 
#1221
This commit is contained in:
ulleo
2026-07-09 18:17:03 +08:00
parent b947cf2476
commit aa594a5681
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/apps/chat/task/llm.py
+4 -1
View File
@@ -137,11 +137,14 @@ class LLMService:
raise SingleMessageError(f"Chat with id {chat_id} not found")
self.oid = chat.oid
if self.oid:
if self.oid and not current_assistant:
w_list = user_ws_list(session, self.current_user.id)
oid_list = [item.id for item in w_list]
if int(self.oid) not in oid_list:
raise SingleMessageError("Current user cannot not access this chat")
if self.oid and current_assistant:
if self.oid != self.current_user.oid:
raise SingleMessageError("Current assistant user cannot not access this chat")
self.current_user.oid = chat.oid
ds: CoreDatasource | AssistantOutDsSchema | None = None