diff --git a/main.py b/main.py index 3b1f84e..edefe63 100644 --- a/main.py +++ b/main.py @@ -42,9 +42,8 @@ def scrape_text_from_url(url): print(f"Error: {e}") async def search_results(keywords): - async with AsyncDDGS() as ddgs: - results = [r async for r in ddgs.text(keywords, region='wt-wt', safesearch='off', max_results=3)] - return results + results = await AsyncDDGS().text(keywords, region='wt-wt', safesearch='off', max_results=3) + return results def summarize(text_array): """ @@ -220,8 +219,7 @@ async def handle(command, update, context): ]) print(keywords) - tasks = [] - tasks.append(search_results(keywords)) + tasks = [search_results(keywords)] results = await asyncio.gather(*tasks) print(results) diff --git a/requirements.txt b/requirements.txt index 230ca7e..9244b6d 100644 --- a/requirements.txt +++ b/requirements.txt @@ -11,7 +11,7 @@ litellm==1.21.7 trafilatura==1.8.1 # duckduckgo -duckduckgo_search==4.4.3 +duckduckgo_search==5.3.0 # PDFs PyPDF2==3.0.1