Create search_engine.py

Author: frdel

python/tools/search_engine.py

@@ -0,0 +1,38 @@
+import os
+import asyncio
+from python.helpers import dotenv, memory, perplexity_search, duckduckgo_search
+from python.helpers.tool import Tool, Response
+from python.helpers.print_style import PrintStyle
+from python.helpers.errors import handle_error
+from python.helpers.searxng import search as searxng
+
+SEARCH_ENGINE_RESULTS = 10
+
+
+class SearchEngine(Tool):
+    async def execute(self, query="", **kwargs):
+
+
+        searxng_result = await self.searxng_search(query)
+
+        await self.agent.handle_intervention(
+            searxng_result
+        )  # wait for intervention and handle it, if paused
+
+        return Response(message=searxng_result, break_loop=False)
+
+
+    async def searxng_search(self, question):
+        results = await searxng(question)
+        return self.format_result_searxng(results, "Search Engine")
+
+    def format_result_searxng(self, result, source):
+        if isinstance(result, Exception):
+            handle_error(result)
+            return f"{source} search failed: {str(result)}"
+
+        outputs = []
+        for item in result["results"]:
+            outputs.append(f"{item['title']}\n{item['url']}\n{item['content']}")
+
+        return "\n\n".join(outputs[:SEARCH_ENGINE_RESULTS]).strip()