|
7 | 7 |
|
8 | 8 | from application.flow.i_step_node import NodeResult |
9 | 9 | from application.flow.step_node.document_extract_node.i_document_extract_node import IDocumentExtractNode |
10 | | -from application.models import Chat |
11 | 10 | from dataset.models import File |
12 | 11 | from dataset.serializers.document_serializers import split_handles, parse_table_handle_list, FileBufferHandle |
13 | 12 | from dataset.serializers.file_serializers import FileSerializer |
@@ -36,15 +35,16 @@ def bytes_to_uploaded_file(file_bytes, file_name="file.txt"): |
36 | 35 | return uploaded_file |
37 | 36 |
|
38 | 37 |
|
| 38 | +splitter = '\n`-----------------------------------`\n' |
| 39 | + |
39 | 40 | class BaseDocumentExtractNode(IDocumentExtractNode): |
40 | 41 | def execute(self, document, chat_id, **kwargs): |
41 | 42 | get_buffer = FileBufferHandle().get_buffer |
42 | 43 |
|
43 | 44 | self.context['document_list'] = document |
44 | 45 | content = [] |
45 | | - splitter = '\n`-----------------------------------`\n' |
46 | 46 | if document is None or not isinstance(document, list): |
47 | | - return NodeResult({'content': content}, {}) |
| 47 | + return NodeResult({'content': ''}, {}) |
48 | 48 |
|
49 | 49 | application = self.workflow_manage.work_flow_post_handler.chat_info.application |
50 | 50 |
|
@@ -76,14 +76,14 @@ def save_image(image_list): |
76 | 76 | return NodeResult({'content': splitter.join(content)}, {}) |
77 | 77 |
|
78 | 78 | def get_details(self, index: int, **kwargs): |
| 79 | + content = self.context.get('content', '').split(splitter) |
79 | 80 | # 不保存content全部内容,因为content内容可能会很大 |
80 | | - content = (self.context.get('content')[:500] + '...') if len(self.context.get('content')) > 0 else '' |
81 | 81 | return { |
82 | 82 | 'name': self.node.properties.get('stepName'), |
83 | 83 | "index": index, |
84 | 84 | 'run_time': self.context.get('run_time'), |
85 | 85 | 'type': self.node.type, |
86 | | - 'content': content, |
| 86 | + 'content': [file_content[:500] for file_content in content], |
87 | 87 | 'status': self.status, |
88 | 88 | 'err_message': self.err_message, |
89 | 89 | 'document_list': self.context.get('document_list') |
|
0 commit comments