TaskRun pydantic model gets renamed to Run (#2042)

This commit is contained in:
Shuchang Zheng
2025-03-30 18:41:24 -07:00
committed by GitHub
parent 05e28931bc
commit 12ef2100b5
3 changed files with 13 additions and 13 deletions

View File

@@ -1,9 +1,9 @@
from skyvern.forge import app
from skyvern.forge.sdk.schemas.runs import TaskRun
from skyvern.forge.sdk.schemas.runs import Run
from skyvern.schemas.runs import RunEngine, RunResponse, RunType
async def get_task_run(run_id: str, organization_id: str | None = None) -> TaskRun | None:
async def get_task_run(run_id: str, organization_id: str | None = None) -> Run | None:
return await app.DATABASE.get_task_run(run_id, organization_id=organization_id)