|
@@ -178,14 +178,12 @@ class ToolNode(BaseNode[ToolNodeData]):
|
|
|
for message in message_stream:
|
|
|
if message.type in {ToolInvokeMessage.MessageType.IMAGE_LINK, ToolInvokeMessage.MessageType.IMAGE}:
|
|
|
assert isinstance(message.message, ToolInvokeMessage.TextMessage)
|
|
|
- assert message.meta
|
|
|
|
|
|
url = message.message.text
|
|
|
- ext = path.splitext(url)[1]
|
|
|
- tool_file_id = str(url).split("/")[-1].split(".")[0]
|
|
|
- mimetype = message.meta.get("mime_type", "image/jpeg")
|
|
|
- filename = message.save_as or url.split("/")[-1]
|
|
|
- transfer_method = message.meta.get("transfer_method", FileTransferMethod.TOOL_FILE)
|
|
|
+ if message.meta:
|
|
|
+ transfer_method = message.meta.get("transfer_method", FileTransferMethod.TOOL_FILE)
|
|
|
+ else:
|
|
|
+ transfer_method = FileTransferMethod.TOOL_FILE
|
|
|
|
|
|
with Session(db.engine) as session:
|
|
|
stmt = select(ToolFile).where(ToolFile.id == tool_file_id)
|