Spaces:
Paused
Paused
| from python.helpers.tool import Tool, Response | |
| 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() | |