fix: DuckDuckGo image search tool error (#5606)

This commit is contained in:
非法操作 2024-06-26 13:21:40 +08:00 committed by GitHub
commit b8926ea267
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
8 changed files with 26 additions and 16 deletions

View file

@ -19,7 +19,11 @@ class DuckDuckGoImageSearchTool(BuiltinTool):
"max_results": tool_parameters.get('max_results'),
}
response = DDGS().images(**query_dict)
results = []
result = []
for res in response:
results.append(self.create_image_message(image=res.get("image")))
return results
msg = ToolInvokeMessage(type=ToolInvokeMessage.MessageType.IMAGE_LINK,
message=res.get('image'),
save_as='',
meta=res)
result.append(msg)
return result

View file

@ -2,7 +2,7 @@ import json
from copy import deepcopy
from datetime import datetime, timezone
from mimetypes import guess_type
from typing import Union
from typing import Any, Union
from yarl import URL
@ -250,7 +250,7 @@ class ToolEngine:
agent_message: Message,
invoke_from: InvokeFrom,
user_id: str
) -> list[tuple[MessageFile, bool]]:
) -> list[tuple[Any, str]]:
"""
Create message file
@ -291,7 +291,7 @@ class ToolEngine:
db.session.refresh(message_file)
result.append((
message_file,
message_file.id,
message.save_as
))