Skip to content

Commit 1d4fe1f

Browse files
perf: Chat validate ds
1 parent 0e2ed07 commit 1d4fe1f

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

backend/apps/chat/task/llm.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1030,7 +1030,12 @@ def run_analysis_or_predict_task(self, action_type: str):
10301030
def validate_history_ds(self):
10311031
_invalid_ds = False
10321032
_ds = self.ds
1033-
_assistant = self.current_assistant
1033+
if not self.current_assistant:
1034+
current_ds = self.session.exec(CoreDatasource, _ds.id)
1035+
_invalid_ds = not current_ds
1036+
else:
1037+
_ds_list: list[dict] = get_assistant_ds(session=self.session, llm_service=self)
1038+
_invalid_ds = any(item.get("id") == _ds.id for item in _ds_list)
10341039
if _invalid_ds:
10351040
yield orjson.dumps({'content': 'ds is invalid', 'type': 'error'}).decode() + '\n\n'
10361041

0 commit comments

Comments
 (0)