fix: Handle prompt too big error (#401)

agents-api/agents_api/web.py

@@ -11,6 +11,7 @@
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.exceptions import RequestValidationError
 from agents_api.common.exceptions import BaseCommonException
+from agents_api.exceptions import PromptTooBigError
 from pycozo.client import QueryException
 from temporalio.service import RPCError
 
@@ -96,7 +97,7 @@ def register_exceptions(app: FastAPI):
 @app.exception_handler(RPCError)
 async def validation_error_handler(request: Request, exc: RPCError):
  return JSONResponse(
- status_code=400,
+ status_code=status.HTTP_400_BAD_REQUEST,
  content={
  "error": {"message": "job not found or invalid", "code": exc.status.name}
  },
@@ -111,6 +112,14 @@ async def session_not_found_error_handler(request: Request, exc: BaseCommonExcep
  )
 
 
+@app.exception_handler(PromptTooBigError)
+async def prompt_too_big_error(request: Request, exc: PromptTooBigError):
+ return JSONResponse(
+ status_code=status.HTTP_400_BAD_REQUEST,
+ content={"error": {"message": str(exc)}},
+ )
+
+
 def main(
  host="127.0.0.1",
  port=8000,