|
|
|
@ -32,14 +32,14 @@ class ToolFileMessageTransformer:
|
|
|
|
try:
|
|
|
|
try:
|
|
|
|
assert isinstance(message.message, ToolInvokeMessage.TextMessage)
|
|
|
|
assert isinstance(message.message, ToolInvokeMessage.TextMessage)
|
|
|
|
tool_file_manager = ToolFileManager()
|
|
|
|
tool_file_manager = ToolFileManager()
|
|
|
|
file = tool_file_manager.create_file_by_url(
|
|
|
|
tool_file = tool_file_manager.create_file_by_url(
|
|
|
|
user_id=user_id,
|
|
|
|
user_id=user_id,
|
|
|
|
tenant_id=tenant_id,
|
|
|
|
tenant_id=tenant_id,
|
|
|
|
file_url=message.message.text,
|
|
|
|
file_url=message.message.text,
|
|
|
|
conversation_id=conversation_id,
|
|
|
|
conversation_id=conversation_id,
|
|
|
|
)
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
url = f"/files/tools/{file.id}{guess_extension(file.mimetype) or '.png'}"
|
|
|
|
url = f"/files/tools/{tool_file.id}{guess_extension(tool_file.mimetype) or '.png'}"
|
|
|
|
|
|
|
|
|
|
|
|
yield ToolInvokeMessage(
|
|
|
|
yield ToolInvokeMessage(
|
|
|
|
type=ToolInvokeMessage.MessageType.IMAGE_LINK,
|
|
|
|
type=ToolInvokeMessage.MessageType.IMAGE_LINK,
|
|
|
|
@ -68,7 +68,7 @@ class ToolFileMessageTransformer:
|
|
|
|
|
|
|
|
|
|
|
|
assert isinstance(message.message.blob, bytes)
|
|
|
|
assert isinstance(message.message.blob, bytes)
|
|
|
|
tool_file_manager = ToolFileManager()
|
|
|
|
tool_file_manager = ToolFileManager()
|
|
|
|
file = tool_file_manager.create_file_by_raw(
|
|
|
|
tool_file = tool_file_manager.create_file_by_raw(
|
|
|
|
user_id=user_id,
|
|
|
|
user_id=user_id,
|
|
|
|
tenant_id=tenant_id,
|
|
|
|
tenant_id=tenant_id,
|
|
|
|
conversation_id=conversation_id,
|
|
|
|
conversation_id=conversation_id,
|
|
|
|
@ -77,7 +77,7 @@ class ToolFileMessageTransformer:
|
|
|
|
filename=filename,
|
|
|
|
filename=filename,
|
|
|
|
)
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
url = cls.get_tool_file_url(tool_file_id=file.id, extension=guess_extension(file.mimetype))
|
|
|
|
url = cls.get_tool_file_url(tool_file_id=tool_file.id, extension=guess_extension(tool_file.mimetype))
|
|
|
|
|
|
|
|
|
|
|
|
# check if file is image
|
|
|
|
# check if file is image
|
|
|
|
if "image" in mimetype:
|
|
|
|
if "image" in mimetype:
|
|
|
|
|