fix: Raise agent not found error on assertion failure (#191)

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

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"])

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