Fix/assistant none type (#2145)

2024-01-24 00:13:04 +08:00
parent 9525ca08b9
commit 27828f44b9
1 changed files with 3 additions and 2 deletions
--- a/api/core/file/message_file_parser.py
+++ b/api/core/file/message_file_parser.py
@@ -128,8 +128,9 @@ class MessageFileParser:

        # group by file type and convert file args or message files to FileObj
        for file in files:
-            if file.belongs_to == FileBelongsTo.ASSISTANT.value:
-                continue
+            if isinstance(file, MessageFile):
+                if file.belongs_to == FileBelongsTo.ASSISTANT.value:
+                    continue

            file_obj = self._to_file_obj(file, file_upload_config)
            if file_obj.type not in type_file_objs: