chore(api/core): Improve FileVar's type hint and imports. (#7290)

2024-08-15 12:43:18 +08:00
parent 6ff7fd80a1
commit 8f16165f92
7 changed files with 68 additions and 59 deletions
--- a/api/core/file/message_file_parser.py
+++ b/api/core/file/message_file_parser.py
@@ -99,7 +99,7 @@ class MessageFileParser:
        # return all file objs
        return new_files

-    def transform_message_files(self, files: list[MessageFile], file_extra_config: FileExtraConfig) -> list[FileVar]:
+    def transform_message_files(self, files: list[MessageFile], file_extra_config: FileExtraConfig):
        """
        transform message files

@@ -144,7 +144,7 @@ class MessageFileParser:

        return type_file_objs

-    def _to_file_obj(self, file: Union[dict, MessageFile], file_extra_config: FileExtraConfig) -> FileVar:
+    def _to_file_obj(self, file: Union[dict, MessageFile], file_extra_config: FileExtraConfig):
        """
        transform file to file obj