Explorar o código

fix(api): handle missing upload_file_id for tool_file messages (#9756)

-LAN- hai 6 meses
pai
achega
2c26f77a25
Modificáronse 1 ficheiros con 4 adicións e 0 borrados
  1. 4 0
      api/models/model.py

+ 4 - 0
api/models/model.py

@@ -977,6 +977,9 @@ class Message(db.Model):
                     config=FileExtraConfig(),
                 )
             elif message_file.transfer_method == "tool_file":
+                if message_file.upload_file_id is None:
+                    assert message_file.url is not None
+                    message_file.upload_file_id = message_file.url.split("/")[-1].split(".")[0]
                 mapping = {
                     "id": message_file.id,
                     "type": message_file.type,
@@ -1001,6 +1004,7 @@ class Message(db.Model):
             for (file, message_file) in zip(files, message_files)
         ]
 
+        db.session.commit()
         return result
 
     @property