ai_se/SearchManager.py

34 lines
892 B
Python

from ddgs import DDGS
#query = "python programming"
def perform_search(query: str) -> dict:
results = DDGS().text(query, max_results=3)
#print(results)
return results
#def parse_results(raw_results: dict) -> list:
# formatted_results = []
# for result in raw_results:
# for key in result:
# if key == "href":
# formatted_results.append(result.get(key))
# if key == "body":
# text = result.get(key)
# if len(text) > 500:
# text = text[0:500]
# formatted_results.append(text)
return formatted_results
def parse_results(results: list[dict]) -> list[dict]:
return [
{
"url": r.get("href", ""),
"snippet": r.get("body", "")[:500]
}
for r in results
]
#parse_results(perform_search(query))