AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-07-26 08:44:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
-LAN- 2024-10-24 11:43:57 +08:00 committed by GitHub
parent 95dc90e6b2
commit 2c26f77a25
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/models/model.py
View File

@ -977,6 +977,9 @@ class Message(db.Model):
config=FileExtraConfig(), config=FileExtraConfig(),
) )
elif message_file.transfer_method == "tool_file": 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 = { mapping = {
"id": message_file.id, "id": message_file.id,
"type": message_file.type, "type": message_file.type,
@ -1001,6 +1004,7 @@ class Message(db.Model):
for (file, message_file) in zip(files, message_files) for (file, message_file) in zip(files, message_files)
] ]
db.session.commit()
return result return result
@property @property