diff --git a/apps/application/flow/step_node/search_knowledge_node/impl/base_search_knowledge_node.py b/apps/application/flow/step_node/search_knowledge_node/impl/base_search_knowledge_node.py index 4b6e96f17..a3a431337 100644 --- a/apps/application/flow/step_node/search_knowledge_node/impl/base_search_knowledge_node.py +++ b/apps/application/flow/step_node/search_knowledge_node/impl/base_search_knowledge_node.py @@ -85,11 +85,11 @@ class BaseSearchKnowledgeNode(ISearchKnowledgeStepNode): knowledge_id_list = self.get_reference_content(search_scope_reference) else: # 文档 document_id_list = self.get_reference_content(search_scope_reference) - knowledge_id_list = QuerySet(Document).filter( + knowledge_id_list = list(QuerySet(Document).filter( id__in=document_id_list ).values_list( 'knowledge_id', flat=True - ).distinct() + ).distinct()) get_knowledge_list_of_authorized = DatabaseModelManage.get_model('get_knowledge_list_of_authorized') chat_user_type = self.workflow_manage.get_body().get('chat_user_type') @@ -107,7 +107,8 @@ class BaseSearchKnowledgeNode(ISearchKnowledgeStepNode): QuerySet(Document).filter( knowledge_id__in=knowledge_id_list, is_active=False)] - embedding_list = vector.query(question, embedding_value, knowledge_id_list, document_id_list, exclude_document_id_list, + embedding_list = vector.query(question, embedding_value, knowledge_id_list, document_id_list, + exclude_document_id_list, exclude_paragraph_id_list, True, knowledge_setting.get('top_n'), knowledge_setting.get('similarity'), SearchMode(knowledge_setting.get('search_mode')))