Skip to content

Commit 609667d

Browse files
committed
chore: optimize knowledge ID retrieval in base_search_knowledge_node
--bug=1062936 --user=刘瑞斌 【应用】应用访问限制开启登录认证,登录后对话,知识库引用文档标签检索节点的输出参数报错 https://www.tapd.cn/62980211/s/1789493
1 parent 4398500 commit 609667d

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

apps/application/flow/step_node/search_knowledge_node/impl/base_search_knowledge_node.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)