fix max_iterations_override (#1733)

2025-02-06 12:08:31 +08:00
parent 0d5d726785
commit b0d9242532
1 changed files with 1 additions and 1 deletions
--- a/skyvern/forge/sdk/routes/agent_protocol.py
+++ b/skyvern/forge/sdk/routes/agent_protocol.py
@@ -1193,7 +1193,7 @@ async def observer_task(
    background_tasks: BackgroundTasks,
    data: ObserverTaskRequest,
    organization: Organization = Depends(org_auth_service.get_current_org),
-    x_max_iterations_override: Annotated[int | None, Header()] = None,
+    x_max_iterations_override: Annotated[int | str | None, Header()] = None,
 ) -> dict[str, Any]:
    if x_max_iterations_override:
        LOG.info("Overriding max iterations for observer", max_iterations_override=x_max_iterations_override)