From 2c26f77a25528d116d4c647fde9118a86491fdea Mon Sep 17 00:00:00 2001
From: -LAN- <laipz8200@outlook.com>
Date: Thu, 24 Oct 2024 11:43:57 +0800
Subject: [PATCH] fix(api): handle missing upload_file_id for tool_file
 messages (#9756)

---
 api/models/model.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/api/models/model.py b/api/models/model.py
index e289423d16..3bd5886d75 100644
--- a/api/models/model.py
+++ b/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