hotfix(agents-api): Fix resource busy 429

Signed-off-by: Diwank Singh Tomer <[email protected]>
julep-ai · Oct 26, 2024 · 73e9977 · 73e9977
1 parent 67fe9d5
commit 73e9977
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/agents-api/agents_api/models/utils.py b/agents-api/agents_api/models/utils.py
@@ -238,10 +238,9 @@ def wrapper(*args: P.args, client=None, **kwargs: P.kwargs) -> pd.DataFrame:
                     print(query)
                     pprint(variables)
 
-                e = getattr(e, "__cause__", None) or e
                 debug and print(repr(e))
 
-                if "busy" in str(e).lower():
+                if "busy" in str(getattr(e, "resp", e)).lower():
                     raise HTTPException(
                         status_code=429, detail="Resource busy. Please try again later."
                     ) from e