Allow querying multiple statuses (#232)

2024-04-24 22:27:15 +03:00
parent 1d329539e4
commit 95f8ad191b
3 changed files with 10 additions and 7 deletions
--- a/skyvern/forge/sdk/routes/agent_protocol.py
+++ b/skyvern/forge/sdk/routes/agent_protocol.py
@@ -313,7 +313,7 @@ async def get_task_internal(
 async def get_agent_tasks(
    page: int = Query(1, ge=1),
    page_size: int = Query(10, ge=1),
-    task_status: TaskStatus | None = None,
+    task_status: Annotated[list[TaskStatus] | None, Query()] = None,
    current_org: Organization = Depends(org_auth_service.get_current_org),
 ) -> Response:
    """