Fix: event loop for DuckDuckGoSearch (#220)

InternLM · Aug 2, 2024 · d37ff17 · d37ff17
1 parent 34284d7
commit d37ff17
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/lagent/actions/bing_browser.py b/lagent/actions/bing_browser.py
@@ -1,3 +1,4 @@
+import asyncio
 import json
 import logging
 import random
@@ -70,6 +71,8 @@ def search(self, query: str, max_retry: int = 3) -> dict:
             'Failed to get search results from DuckDuckGo after retries.')
 
     def _call_ddgs(self, query: str, **kwargs) -> dict:
+        loop = asyncio.new_event_loop()
+        asyncio.set_event_loop(loop)
         ddgs = DDGS(**kwargs)
         response = ddgs.text(query.strip("'"), max_results=10)
         return response