浏览代码

suhh-上传模版写定文件url

‘suhuihui’ 3 月之前
父节点
当前提交
7d4eaa9aa9
共有 1 个文件被更改,包括 4 次插入2 次删除
  1. 4 2
      api/services/dataset_service.py

+ 4 - 2
api/services/dataset_service.py

@@ -581,9 +581,8 @@ class TemplateService:
             raise FileNotExistsError()
 
         file_name = upload_file.name
-        print("文件名称" + file_name)
         file_id = upload_file.id
-        print("文件id" + file_id)
+        file_url=upload_file.key
         data_source_info = {"upload_file_id": upload_file.id}
         template = TemplateService.build_template(
             dataset,
@@ -596,6 +595,7 @@ class TemplateService:
             file_id,
             file_name,
             batch,
+            file_url
         )
         db.session.add(template)
         db.session.flush()
@@ -625,6 +625,7 @@ class TemplateService:
         file_id: str,
         file_name: str,
         batch: str,
+        file_url: str
     ):
         template = Template(
             tenant_id=dataset.tenant_id,
@@ -638,6 +639,7 @@ class TemplateService:
             created_from=created_from,
             created_by=account.id,
             file_id=file_id,
+            file_url="/data/dify/docker/volumes/app/api/storage/"+file_url
         )
         return template