shu/test discriminated run response schema (#2046)

2025-03-30 21:14:52 -07:00
parent 8253738c7b
commit ea67502161
6 changed files with 158 additions and 112 deletions
--- a/skyvern/forge/sdk/db/client.py
+++ b/skyvern/forge/sdk/db/client.py
@@ -2945,7 +2945,7 @@ class AgentDB:
            task_run = (await session.scalars(query)).first()
            return Run.model_validate(task_run) if task_run else None

-    async def get_task_run(
+    async def get_run(
        self,
        run_id: str,
        organization_id: str | None = None,