Merge branch 'dev' into feat-Improve-GitHub-action

julep-ai · Apr 16, 2024 · 7bb03ae · 7bb03ae
2 parents 3bab284 + 594d186
commit 7bb03ae
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 2 deletions.
diff --git a/agents-api/agents_api/routers/agents/routers.py b/agents-api/agents_api/routers/agents/routers.py
@@ -133,7 +133,7 @@ async def update_agent(
             detail="Agent not found",
         )
     except QueryException as e:
-        if e.code == "transact::assertion_failure":
+        if e.code in ("transact::assertion_failure", "eval::assert_some_failure"):
             raise AgentNotFoundError(x_developer_id, agent_id)
 
         raise

diff --git a/agents-api/agents_api/routers/sessions/routers.py b/agents-api/agents_api/routers/sessions/routers.py
@@ -161,6 +161,12 @@ async def update_session(
             status_code=status.HTTP_404_NOT_FOUND,
             detail="Session not found",
         )
+    except QueryException as e:
+        # the code is not so informative now, but it may be a good solution in the future
+        if e.code in ("transact::assertion_failure", "eval::assert_some_failure"):
+            raise SessionNotFoundError(x_developer_id, session_id)
+
+        raise
 
 
 @router.patch("/sessions/{session_id}", tags=["sessions"])
@@ -186,6 +192,12 @@ async def patch_session(
             status_code=status.HTTP_404_NOT_FOUND,
             detail="Session not found",
         )
+    except QueryException as e:
+        # the code is not so informative now, but it may be a good solution in the future
+        if e.code in ("transact::assertion_failure", "eval::assert_some_failure"):
+            raise SessionNotFoundError(x_developer_id, session_id)
+
+        raise
 
 
 @router.get("/sessions/{session_id}/suggestions", tags=["sessions"])

diff --git a/agents-api/agents_api/routers/users/routers.py b/agents-api/agents_api/routers/users/routers.py
@@ -108,7 +108,7 @@ async def update_user(
         )
     except QueryException as e:
         # the code is not so informative now, but it may be a good solution in the future
-        if e.code == "transact::assertion_failure":
+        if e.code in ("transact::assertion_failure", "eval::assert_some_failure"):
             raise UserNotFoundError(x_developer_id, user_id)
 
         raise