diff --git a/skyvern/forge/sdk/routes/agent_protocol.py b/skyvern/forge/sdk/routes/agent_protocol.py index d2684227..dcc7887c 100644 --- a/skyvern/forge/sdk/routes/agent_protocol.py +++ b/skyvern/forge/sdk/routes/agent_protocol.py @@ -567,6 +567,7 @@ async def create_workflow_from_prompt( extra_http_headers=data.extra_http_headers, max_iterations=x_max_iterations_override, max_steps=x_max_steps_override, + status=WorkflowStatus.published if data.publish_workflow else WorkflowStatus.auto_generated, run_with=data.run_with, ai_fallback=data.ai_fallback, ) diff --git a/skyvern/forge/sdk/workflow/service.py b/skyvern/forge/sdk/workflow/service.py index dd0ee214..c9180976 100644 --- a/skyvern/forge/sdk/workflow/service.py +++ b/skyvern/forge/sdk/workflow/service.py @@ -736,6 +736,7 @@ class WorkflowService: extra_http_headers: dict[str, str] | None = None, max_iterations: int | None = None, max_steps: int | None = None, + status: WorkflowStatus = WorkflowStatus.auto_generated, run_with: str | None = None, ai_fallback: bool = True, ) -> Workflow: @@ -779,6 +780,7 @@ class WorkflowService: totp_identifier=totp_identifier, max_screenshot_scrolling_times=max_screenshot_scrolling_times, extra_http_headers=extra_http_headers, + status=status, run_with=run_with, ai_fallback=ai_fallback, )