diff --git a/apps/application/flow/step_node/document_split_node/impl/base_document_split_node.py b/apps/application/flow/step_node/document_split_node/impl/base_document_split_node.py index 0703ff6e5..ad757aabf 100644 --- a/apps/application/flow/step_node/document_split_node/impl/base_document_split_node.py +++ b/apps/application/flow/step_node/document_split_node/impl/base_document_split_node.py @@ -101,6 +101,7 @@ class BaseDocumentSplitNode(IDocumentSplitNode): 'source_url': file_name, } item['name'] = file_name + item['source_file_id'] = source_file_id item['paragraphs'] = item.pop('content', []) for paragraph in item['paragraphs']: diff --git a/apps/application/flow/step_node/knowledge_write_node/impl/base_knowledge_write_node.py b/apps/application/flow/step_node/knowledge_write_node/impl/base_knowledge_write_node.py index 7b56b34d7..db2080374 100644 --- a/apps/application/flow/step_node/knowledge_write_node/impl/base_knowledge_write_node.py +++ b/apps/application/flow/step_node/knowledge_write_node/impl/base_knowledge_write_node.py @@ -38,6 +38,7 @@ class KnowledgeWriteParamSerializer(serializers.Serializer): source=_('document name')) meta = serializers.DictField(required=False) paragraphs = ParagraphInstanceSerializer(required=False, many=True, allow_null=True) + source_file_id = serializers.UUIDField(required=False, allow_null=True) def convert_uuid_to_str(obj):