AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-12 16:59:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: remote image not display in answer node (#6867)

Browse Source
This commit is contained in:
非法操作 2024-08-01 13:21:49 +08:00 committed by GitHub
parent 14367ddc09
commit 29e3c3061c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/core/tools/provider/builtin/duckduckgo/tools/ddgo_img.py
View File

@ -2,6 +2,7 @@ from typing import Any
from duckduckgo_search import DDGS from duckduckgo_search import DDGS
from core.file.file_obj import FileTransferMethod
from core.tools.entities.tool_entities import ToolInvokeMessage from core.tools.entities.tool_entities import ToolInvokeMessage
from core.tools.tool.builtin_tool import BuiltinTool from core.tools.tool.builtin_tool import BuiltinTool
@ -21,6 +22,7 @@ class DuckDuckGoImageSearchTool(BuiltinTool):
response = DDGS().images(**query_dict) response = DDGS().images(**query_dict)
result = [] result = []
for res in response: for res in response:
res['transfer_method'] = FileTransferMethod.REMOTE_URL
msg = ToolInvokeMessage(type=ToolInvokeMessage.MessageType.IMAGE_LINK, msg = ToolInvokeMessage(type=ToolInvokeMessage.MessageType.IMAGE_LINK,
message=res.get('image'), message=res.get('image'),
save_as='', save_as='',

3
api/core/workflow/nodes/tool/tool_node.py
View File

@ -174,13 +174,14 @@ class ToolNode(BaseNode):
ext = path.splitext(url)[1] ext = path.splitext(url)[1]
mimetype = response.meta.get('mime_type', 'image/jpeg') mimetype = response.meta.get('mime_type', 'image/jpeg')
filename = response.save_as or url.split('/')[-1] filename = response.save_as or url.split('/')[-1]
transfer_method = response.meta.get('transfer_method', FileTransferMethod.TOOL_FILE)
# get tool file id # get tool file id
tool_file_id = url.split('/')[-1].split('.')[0] tool_file_id = url.split('/')[-1].split('.')[0]
result.append(FileVar( result.append(FileVar(
tenant_id=self.tenant_id, tenant_id=self.tenant_id,
type=FileType.IMAGE, type=FileType.IMAGE,
transfer_method=FileTransferMethod.TOOL_FILE, transfer_method=transfer_method,
url=url, url=url,
related_id=tool_file_id, related_id=tool_file_id,
filename=filename, filename=filename,