feat: support select terminology in chat by using assistant

Author: ulleo

backend/apps/chat/task/llm.py

@@ -220,7 +220,8 @@ def generate_analysis(self):
         self.chat_question.data = orjson.dumps(data.get('data')).decode()
         analysis_msg: List[Union[BaseMessage, dict[str, Any]]] = []
 
-        self.chat_question.terminologies = get_terminology_template(self.session, self.chat_question.question, self.current_user.oid)
+        self.chat_question.terminologies = get_terminology_template(self.session, self.chat_question.question,
+                                                                    self.current_user.oid)
 
         analysis_msg.append(SystemMessage(content=self.chat_question.analysis_sys_question()))
         analysis_msg.append(HumanMessage(content=self.chat_question.analysis_user_question()))
@@ -498,6 +499,10 @@ def select_datasource(self):
                                                         answer=orjson.dumps({'content': full_text}).decode(),
                                                         datasource=_datasource,
                                                         engine_type=_engine_type)
+
+        self.chat_question.terminologies = get_terminology_template(self.session, self.chat_question.question,
+                                                                    self.ds.oid if isinstance(self.ds,
+                                                                                              CoreDatasource) else 1)
         self.init_messages()
 
         if _error:
@@ -894,7 +899,10 @@ def run_task_cache(self, in_chat: bool = True):
 
     def run_task(self, in_chat: bool = True):
         try:
-            self.chat_question.terminologies = get_terminology_template(self.session, self.chat_question.question, self.current_user.oid)
+            if self.ds:
+                self.chat_question.terminologies = get_terminology_template(self.session, self.chat_question.question,
+                                                                            self.ds.oid if isinstance(self.ds,
+                                                                                                      CoreDatasource) else 1)
             self.init_messages()
 
             # return id

backend/apps/terminology/curd/terminology.py

@@ -418,7 +418,9 @@ def to_xml_string(_dict: list[dict] | dict, root: str = 'terminologies') -> str:
     return pretty_xml
 
 
-def get_terminology_template(session: SessionDep, question: str, oid: int) -> str:
+def get_terminology_template(session: SessionDep, question: str, oid: Optional[int] = 1) -> str:
+    if not oid:
+        oid = 1
     _results = select_terminology_by_word(session, question, oid)
     if _results and len(_results) > 0:
         terminology = to_xml_string(_results)