Skip to content

Commit 5053cf3

Browse files
committed
fix: Workflow knowledge base retrieval node recall error
1 parent 3450ef7 commit 5053cf3

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ def reset_title(title):
4949
class BaseSearchKnowledgeNode(ISearchKnowledgeStepNode):
5050
def save_context(self, details, workflow_manage):
5151
result = details.get('paragraph_list', [])
52-
dataset_setting = self.node_params_serializer.data.get('dataset_setting')
52+
knowledge_setting = self.node_params_serializer.data.get('knowledge_setting')
5353
directly_return = '\n'.join(
5454
[f"{paragraph.get('title', '')}:{paragraph.get('content')}" for paragraph in result if
5555
paragraph.get('is_hit_handling_method')])
@@ -59,7 +59,7 @@ def save_context(self, details, workflow_manage):
5959
self.context['is_hit_handling_method_list'] = [row for row in result if row.get('is_hit_handling_method')]
6060
self.context['data'] = '\n'.join(
6161
[f"{paragraph.get('title', '')}:{paragraph.get('content')}" for paragraph in
62-
result])[0:dataset_setting.get('max_paragraph_char_number', 5000)]
62+
result])[0:knowledge_setting.get('max_paragraph_char_number', 5000)]
6363
self.context['directly_return'] = directly_return
6464

6565
def execute(self, knowledge_id_list, knowledge_setting, question, show_knowledge,

0 commit comments

Comments
 (0)