Spaces:
Paused
Paused
File size: 1,226 Bytes
8d1819a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 |
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()
|