youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-04-27 14:08:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: support other file types for Tool

Browse Source
This commit is contained in:
Yeuoly
2024-12-04 19:26:01 +08:00
parent 0af9c4fd9d
commit 01efdee1dd
4 changed files with 12 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
api/core/tools/entities/tool_entities.py
View File

@ -157,6 +157,7 @@ class ToolInvokeMessage(BaseModel):
BLOB = "blob"
JSON = "json"
IMAGE_LINK = "image_link"
BINARY_LINK = "binary_link"
VARIABLE = "variable"
FILE = "file"

2
api/core/tools/utils/message_transformer.py
View File

@ -85,7 +85,7 @@ class ToolFileMessageTransformer:
)
else:
yield ToolInvokeMessage(
type=ToolInvokeMessage.MessageType.LINK,
type=ToolInvokeMessage.MessageType.BINARY_LINK,
message=ToolInvokeMessage.TextMessage(text=url),
meta=message.meta.copy() if message.meta is not None else {},
)