fix: search knowledge (#3439)

Author: shaohuzhang1

apps/application/chat_pipeline/step/search_dataset_step/i_search_dataset_step.py

@@ -50,7 +50,7 @@ def get_step_serializer(self, manage: PipelineManage) -> Type[InstanceSerializer
         return self.InstanceSerializer
 
     def _run(self, manage: PipelineManage):
-        paragraph_list = self.execute(**self.context['step_args'])
+        paragraph_list = self.execute(**self.context['step_args'], manage=manage)
         manage.context['paragraph_list'] = paragraph_list
         self.context['paragraph_list'] = paragraph_list
 
@@ -59,6 +59,7 @@ def execute(self, problem_text: str, knowledge_id_list: list[str], exclude_docum
                 exclude_paragraph_id_list: list[str], top_n: int, similarity: float, padding_problem_text: str = None,
                 search_mode: str = None,
                 workspace_id=None,
+                manage: PipelineManage = None,
                 **kwargs) -> List[ParagraphPipelineModel]:
         """
         关于 用户和补全问题 说明: 补全问题如果有就使用补全问题去查询 反之就用用户原始问题查询

apps/application/chat_pipeline/step/search_dataset_step/impl/base_search_dataset_step.py

@@ -53,11 +53,12 @@ def execute(self, problem_text: str, knowledge_id_list: list[str], exclude_docum
                 exclude_paragraph_id_list: list[str], top_n: int, similarity: float, padding_problem_text: str = None,
                 search_mode: str = None,
                 workspace_id=None,
+                manage=None,
                 **kwargs) -> List[ParagraphPipelineModel]:
         get_knowledge_list_of_authorized = DatabaseModelManage.get_model('get_knowledge_list_of_authorized')
-        chat_user_type = self.context.get('chat_user_type')
+        chat_user_type = manage.context.get('chat_user_type')
         if get_knowledge_list_of_authorized is not None and RoleConstants.CHAT_USER.value.name == chat_user_type:
-            knowledge_id_list = get_knowledge_list_of_authorized(self.context.get('chat_user_id'),
+            knowledge_id_list = get_knowledge_list_of_authorized(manage.context.get('chat_user_id'),
                                                                  knowledge_id_list)
         if len(knowledge_id_list) == 0:
             return []

apps/chat/serializers/chat.py

@@ -307,6 +307,7 @@ def re_open_chat_work_flow(self, chat_id, application):
             raise ChatException(500, _("The application has not been published. Please use it after publishing."))
 
         chat_info = ChatInfo(chat_id, self.data.get('chat_user_id'), self.data.get('chat_user_type'), [], [],
+                             application.id,
                              application, work_flow_version)
         chat_record_list = list(QuerySet(ChatRecord).filter(chat_id=chat_id).order_by('-create_time')[0:5])
         chat_record_list.sort(key=lambda r: r.create_time)