refactor: override document extract node save_content

1Panel-dev · Jan 23, 2025 · 6dca9b4 · 6dca9b4
1 parent aa184aa
commit 6dca9b4
Showing 1 changed file with 5 additions and 0 deletions.
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
@@ -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