From 8386abaed1172987d2e5a367e9c6028f225adbfa Mon Sep 17 00:00:00 2001 From: Yeuoly Date: Tue, 19 Mar 2024 17:07:44 +0800 Subject: [PATCH] fix: file --- api/core/file/file_obj.py | 8 +------- 1 file changed, 1 insertion(+), 7 deletions(-) diff --git a/api/core/file/file_obj.py b/api/core/file/file_obj.py index 48f4fbb191..2e31b4bac1 100644 --- a/api/core/file/file_obj.py +++ b/api/core/file/file_obj.py @@ -128,13 +128,7 @@ class FileVar(BaseModel): force_url=force_url ) elif self.transfer_method == FileTransferMethod.TOOL_FILE: - # get extension - if '.' in self.url: - extension = f'.{self.url.split(".")[-1]}' - if len(extension) > 10: - extension = '.bin' - else: - extension = '.bin' + extension = self.extension # add sign url return ToolFileParser.get_tool_file_manager().sign_file(tool_file_id=self.related_id, extension=extension)