浏览代码

fix: duckduckgo image search not work (#9821)

tags/0.10.2
非法操作 1年前
父节点
当前提交
ac9f1e9de5
没有帐户链接到提交者的电子邮件
共有 1 个文件被更改,包括 5 次插入8 次删除
  1. 5
    8
      api/core/tools/provider/builtin/duckduckgo/tools/ddgo_img.py

+ 5
- 8
api/core/tools/provider/builtin/duckduckgo/tools/ddgo_img.py 查看文件



from duckduckgo_search import DDGS from duckduckgo_search import DDGS


from core.file.models 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


"max_results": tool_parameters.get("max_results"), "max_results": tool_parameters.get("max_results"),
} }
response = DDGS().images(**query_dict) response = DDGS().images(**query_dict)
result = []
markdown_result = "\n\n"
json_result = []
for res in response: for res in response:
res["transfer_method"] = FileTransferMethod.REMOTE_URL
msg = ToolInvokeMessage(
type=ToolInvokeMessage.MessageType.IMAGE_LINK, message=res.get("image"), save_as="", meta=res
)
result.append(msg)
return result
markdown_result += f"![{res.get('title') or ''}]({res.get('image') or ''})"
json_result.append(self.create_json_message(res))
return [self.create_text_message(markdown_result)] + json_result

正在加载...
取消
保存