Handle rate limit exception (#84)

definitive-io · Dec 22, 2023 · 2e2324b · 2e2324b
1 parent 9db9256
commit 2e2324b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/packages/openassistants/openassistants/contrib/langchain_ddg_tool.py b/packages/openassistants/openassistants/contrib/langchain_ddg_tool.py
@@ -4,7 +4,7 @@
 from openassistants.data_models.function_output import FunctionOutput, TextOutput
 from openassistants.functions.base import BaseFunction, FunctionExecutionDependency
 from openassistants.functions.utils import AsyncStreamVersion
-
+from duckduckgo_search.exceptions import RateLimitException
 
 def ddgs_text(query: str, max_results: Optional[int] = None) -> List[Dict[str, str]]:
     """Run query through DuckDuckGo text search and return results."""
@@ -47,7 +47,7 @@ async def execute(
 
                 try:
                     results = ddgs_text(query, max_results=4)
-                except RuntimeError:
+                except RateLimitException:
                     yield [TextOutput(text="DuckDuckGo is currently unavailable, please try again later.")]
                     return