fix: file

2024-03-19 17:07:44 +08:00 · 2024-03-19 17:07:44 +08:00 · 8386abaed1
parent 74408c4ced
commit 8386abaed1
1 changed files with 1 additions and 7 deletions
--- 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)