diff --git a/skyvern/forge/sdk/services/org_auth_service.py b/skyvern/forge/sdk/services/org_auth_service.py index 7a5ecea0..cf4bd28f 100644 --- a/skyvern/forge/sdk/services/org_auth_service.py +++ b/skyvern/forge/sdk/services/org_auth_service.py @@ -24,8 +24,8 @@ ALGORITHM = "HS256" async def get_current_org( - x_api_key: Annotated[str | None, Header()] = None, - authorization: Annotated[str | None, Header()] = None, + x_api_key: Annotated[str | None, Header(description="API key for authentication")] = None, + authorization: Annotated[str | None, Header(include_in_schema=False)] = None, ) -> Organization: if not x_api_key and not authorization: raise HTTPException( diff --git a/skyvern/services/task_v2_service.py b/skyvern/services/task_v2_service.py index fde4b150..953ddd8c 100644 --- a/skyvern/services/task_v2_service.py +++ b/skyvern/services/task_v2_service.py @@ -494,8 +494,8 @@ async def run_task_v2_helper( browser_state = app.BROWSER_MANAGER.get_for_workflow_run( workflow_run_id=workflow_run_id, parent_workflow_run_id=workflow_run.parent_workflow_run_id ) + fallback_occurred = False if browser_state is None: - fallback_occurred = False try: browser_state = await app.BROWSER_MANAGER.get_or_create_for_workflow_run( workflow_run=workflow_run,