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

catch and log db error in generate_task api (#734)

Browse Source
This commit is contained in:
Kerem Yilmaz
2024-08-26 21:42:45 +03:00
committed by GitHub
parent 3502093200
commit 672a2e22e4
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
skyvern/forge/sdk/routes/agent_protocol.py
View File

@@ -18,6 +18,7 @@ from fastapi import (
) )
from fastapi.responses import ORJSONResponse from fastapi.responses import ORJSONResponse
from pydantic import BaseModel from pydantic import BaseModel
from sqlalchemy.exc import OperationalError
from skyvern import analytics from skyvern import analytics
from skyvern.exceptions import StepNotFound from skyvern.exceptions import StepNotFound
@@ -788,6 +789,9 @@ async def generate_task(
except LLMProviderError: except LLMProviderError:
LOG.error("Failed to generate task", exc_info=True) LOG.error("Failed to generate task", exc_info=True)
raise HTTPException(status_code=400, detail="Failed to generate task. Please try again later.") 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) @base_router.put("/organizations/", include_in_schema=False)