vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

add SkyvernHTTPException base exception to automate known error handling in agent service (#317)

Browse Source
This commit is contained in:
Shuchang Zheng
2024-05-15 13:35:45 -07:00
committed by GitHub
parent 164a4da03a
commit c17b4ed287
2 changed files with 27 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
skyvern/forge/api_app.py
View File

@@ -10,6 +10,7 @@ from starlette.requests import HTTPConnection, Request
from starlette_context.middleware import RawContextMiddleware
from starlette_context.plugins.base import Plugin
from skyvern.exceptions import SkyvernHTTPException
from skyvern.forge import app as forge_app
from skyvern.forge.sdk.core import skyvern_context
from skyvern.forge.sdk.core.skyvern_context import SkyvernContext
@@ -73,6 +74,10 @@ def get_agent_app(router: APIRouter = base_router) -> FastAPI:
LOG.info("Server startup complete. Skyvern is now online")
@app.exception_handler(SkyvernHTTPException)
async def handle_skyvern_http_exception(request: Request, exc: SkyvernHTTPException) -> JSONResponse:
return JSONResponse(status_code=exc.status_code, content={"detail": exc.message})
@app.exception_handler(Exception)
async def unexpected_exception(request: Request, exc: Exception) -> JSONResponse:
LOG.exception("Unexpected error in agent server.", exc_info=exc)