Fix web search

bioimage-io · Mar 12, 2024 · 73eb811 · 73eb811
1 parent 6a059f0
commit 73eb811
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/bioimageio_chatbot/chatbot_extensions/web_search_extension/llm_web_search.py b/bioimageio_chatbot/chatbot_extensions/web_search_extension/llm_web_search.py
@@ -15,7 +15,7 @@ async def search_duckduckgo(query: str, langchain_compressor: LangchainCompresso
     async with AsyncDDGS() as ddgs:
         if instant_answers:
             answer_list = []
-            async for answer in ddgs.answers(query):
+            for answer in await ddgs.answers(query):
                 answer_list.append(answer)
             if answer_list:
                 max_results -= 1  # We already have 1 result now
@@ -26,7 +26,7 @@ async def search_duckduckgo(query: str, langchain_compressor: LangchainCompresso
 
         results = []
         result_urls = []
-        async for result in ddgs.text(query, region='wt-wt', safesearch='moderate', timelimit=None,
+        for result in await ddgs.text(query, region='wt-wt', safesearch='moderate', timelimit=None,
                                 max_results=num_results_to_process):
             results.append(result)
             result_urls.append(result["href"])

diff --git a/pyproject.toml b/pyproject.toml
@@ -3,7 +3,7 @@ requires = ["setuptools", "wheel"]
 
 [project]
 name = "bioimageio-chatbot"
-version = "0.1.76"
+version = "0.1.77"
 readme = "README.md"
 description = "Your Personal Assistant in BioImage Analysis."
 dependencies = [