From 672a2e22e4850d8da49bc405f5a8c10b80428cf6 Mon Sep 17 00:00:00 2001 From: Kerem Yilmaz Date: Mon, 26 Aug 2024 21:42:45 +0300 Subject: [PATCH] catch and log db error in generate_task api (#734) --- skyvern/forge/sdk/routes/agent_protocol.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/skyvern/forge/sdk/routes/agent_protocol.py b/skyvern/forge/sdk/routes/agent_protocol.py index e45bd001..0dca5dc6 100644 --- a/skyvern/forge/sdk/routes/agent_protocol.py +++ b/skyvern/forge/sdk/routes/agent_protocol.py @@ -18,6 +18,7 @@ from fastapi import ( ) from fastapi.responses import ORJSONResponse from pydantic import BaseModel +from sqlalchemy.exc import OperationalError from skyvern import analytics from skyvern.exceptions import StepNotFound @@ -788,6 +789,9 @@ async def generate_task( except LLMProviderError: LOG.error("Failed to generate task", exc_info=True) raise HTTPException(status_code=400, detail="Failed to generate task. Please try again later.") + except OperationalError: + LOG.error("Database error when generating task", exc_info=True, user_prompt=data.prompt) + raise HTTPException(status_code=500, detail="Failed to generate task. Please try again later.") @base_router.put("/organizations/", include_in_schema=False)