Skip to content

Commit fb6314c

Browse files
committed
feat: implement file download functionality in document handler
1 parent faeee1b commit fb6314c

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

apps/knowledge/serializers/document.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,7 @@
4646
TaskType, File, FileSourceType
4747
from knowledge.serializers.common import ProblemParagraphManage, BatchSerializer, \
4848
get_embedding_model_id_by_knowledge_id, MetaSerializer, write_image, zip_dir
49+
from knowledge.serializers.file import FileSerializer
4950
from knowledge.serializers.paragraph import ParagraphSerializers, ParagraphInstanceSerializer, \
5051
delete_problems_and_mappings
5152
from knowledge.task.embedding import embedding_by_document, delete_embedding_by_document_list, \
@@ -548,7 +549,8 @@ def export_zip(self, with_valid=True):
548549
return response
549550

550551
def download_source_file(self):
551-
pass
552+
self.is_valid(raise_exception=True)
553+
return FileSerializer.Operate(id=self.data.get('knowledge_id')).get(with_valid=True)
552554

553555
def one(self, with_valid=False):
554556
if with_valid:

0 commit comments

Comments
 (0)