From 6dca9b41a000f0ec5d2ef2f032dc5f5e874abbd8 Mon Sep 17 00:00:00 2001 From: CaptainB Date: Thu, 23 Jan 2025 18:56:56 +0800 Subject: [PATCH] refactor: override document extract node save_content --- .../document_extract_node/impl/base_document_extract_node.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/apps/application/flow/step_node/document_extract_node/impl/base_document_extract_node.py b/apps/application/flow/step_node/document_extract_node/impl/base_document_extract_node.py index 8bec59636..34508b7bd 100644 --- a/apps/application/flow/step_node/document_extract_node/impl/base_document_extract_node.py +++ b/apps/application/flow/step_node/document_extract_node/impl/base_document_extract_node.py @@ -38,6 +38,11 @@ def bytes_to_uploaded_file(file_bytes, file_name="file.txt"): splitter = '\n`-----------------------------------`\n' class BaseDocumentExtractNode(IDocumentExtractNode): + def save_context(self, details, workflow_manage): + self.context['content'] = details.get('content') + self.answer_text = details.get('content') + + def execute(self, document, chat_id, **kwargs): get_buffer = FileBufferHandle().get_buffer