Skip to content

Commit 8e5ba29

Browse files
committed
refactor: override document extract node save_content
1 parent aa184aa commit 8e5ba29

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

apps/application/flow/step_node/document_extract_node/impl/base_document_extract_node.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,11 @@ def bytes_to_uploaded_file(file_bytes, file_name="file.txt"):
3838
splitter = '\n`-----------------------------------`\n'
3939

4040
class BaseDocumentExtractNode(IDocumentExtractNode):
41+
def save_context(self, details, workflow_manage):
42+
self.context['content'] = details.get('content')
43+
self.answer_text = details.get('content')
44+
45+
4146
def execute(self, document, chat_id, **kwargs):
4247
get_buffer = FileBufferHandle().get_buffer
4348

0 commit comments

Comments
 (0)