From 1d4fe1fbb435970104d69e7deb9517a6ed0e1d1f Mon Sep 17 00:00:00 2001
From: fit2cloud-chenyw <yawen.chen@fit2cloud.com>
Date: Wed, 30 Jul 2025 15:56:57 +0800
Subject: [PATCH] perf: Chat validate ds

---
 backend/apps/chat/task/llm.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/backend/apps/chat/task/llm.py b/backend/apps/chat/task/llm.py
index 895aea5e..c240a04b 100644
--- a/backend/apps/chat/task/llm.py
+++ b/backend/apps/chat/task/llm.py
@@ -1030,7 +1030,12 @@ class LLMService:
     def validate_history_ds(self):
         _invalid_ds = False
         _ds = self.ds
-        _assistant = self.current_assistant
+        if not self.current_assistant:
+            current_ds = self.session.exec(CoreDatasource, _ds.id)
+            _invalid_ds = not current_ds
+        else:
+            _ds_list: list[dict] = get_assistant_ds(session=self.session, llm_service=self)
+            _invalid_ds = any(item.get("id") == _ds.id for item in _ds_list)
         if _invalid_ds:
             yield orjson.dumps({'content': 'ds is invalid', 'type': 'error'}).decode() + '\n\n'