fix: DuckDuckGo image search tool error (#5606)

2024-06-26 13:21:40 +08:00 · 2024-06-26 13:21:40 +08:00 · b8926ea267
commit b8926ea267
parent 43335b5c87
8 changed files with 26 additions and 16 deletions
--- a/api/core/tools/provider/builtin/duckduckgo/tools/ddgo_img.py
+++ b/api/core/tools/provider/builtin/duckduckgo/tools/ddgo_img.py
@ -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