@@ -85,11 +85,11 @@ def execute(self, knowledge_id_list, knowledge_setting, question, show_knowledge
8585 knowledge_id_list = self .get_reference_content (search_scope_reference )
8686 else : # 文档
8787 document_id_list = self .get_reference_content (search_scope_reference )
88- knowledge_id_list = QuerySet (Document ).filter (
88+ knowledge_id_list = list ( QuerySet (Document ).filter (
8989 id__in = document_id_list
9090 ).values_list (
9191 'knowledge_id' , flat = True
92- ).distinct ()
92+ ).distinct ())
9393
9494 get_knowledge_list_of_authorized = DatabaseModelManage .get_model ('get_knowledge_list_of_authorized' )
9595 chat_user_type = self .workflow_manage .get_body ().get ('chat_user_type' )
@@ -107,7 +107,8 @@ def execute(self, knowledge_id_list, knowledge_setting, question, show_knowledge
107107 QuerySet (Document ).filter (
108108 knowledge_id__in = knowledge_id_list ,
109109 is_active = False )]
110- embedding_list = vector .query (question , embedding_value , knowledge_id_list , document_id_list , exclude_document_id_list ,
110+ embedding_list = vector .query (question , embedding_value , knowledge_id_list , document_id_list ,
111+ exclude_document_id_list ,
111112 exclude_paragraph_id_list , True , knowledge_setting .get ('top_n' ),
112113 knowledge_setting .get ('similarity' ),
113114 SearchMode (knowledge_setting .get ('search_mode' )))
0 commit comments