Fix unclosed event loop (#235)

InternLM · Aug 16, 2024 · 17197d0 · 17197d0
1 parent a8ae387
commit 17197d0
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/lagent/actions/bing_browser.py b/lagent/actions/bing_browser.py
@@ -72,10 +72,13 @@ def search(self, query: str, max_retry: int = 3) -> dict:
 
     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
+        try:
+            asyncio.set_event_loop(loop)
+            ddgs = DDGS(**kwargs)
+            response = ddgs.text(query.strip("'"), max_results=10)
+            return response
+        finally:
+            loop.close()
 
     def _parse_response(self, response: dict) -> dict:
         raw_results = []