fix: DuckDuckGo image search tool error (#5606)
This commit is contained in:
parent
43335b5c87
commit
b8926ea267
8 changed files with 26 additions and 16 deletions
|
|
@ -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
|
||||
|
|
|
|||
|
|
@ -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
|
||||
))
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue