From d992f2de95a2e70caf254c87ac0555bcee581154 Mon Sep 17 00:00:00 2001
From: Jonathan Dobson <jon.m.dobson@gmail.com>
Date: Fri, 29 Aug 2025 16:18:22 -0400
Subject: [PATCH] allow ai_fallback to save (#3321)

---
 skyvern/forge/sdk/workflow/service.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/skyvern/forge/sdk/workflow/service.py b/skyvern/forge/sdk/workflow/service.py
index 6fdfb7be..193d1e3d 100644
--- a/skyvern/forge/sdk/workflow/service.py
+++ b/skyvern/forge/sdk/workflow/service.py
@@ -674,6 +674,7 @@ class WorkflowService:
         extra_http_headers: dict[str, str] | None = None,
         generate_script: bool = False,
         cache_key: str | None = None,
+        ai_fallback: bool | None = None,
     ) -> Workflow:
         return await app.DATABASE.create_workflow(
             title=title,
@@ -694,6 +695,7 @@ class WorkflowService:
             extra_http_headers=extra_http_headers,
             generate_script=generate_script,
             cache_key=cache_key,
+            ai_fallback=False if ai_fallback is None else ai_fallback,
         )
 
     async def get_workflow(self, workflow_id: str, organization_id: str | None = None) -> Workflow:
@@ -1647,6 +1649,7 @@ class WorkflowService:
                     status=request.status,
                     generate_script=request.generate_script,
                     cache_key=request.cache_key,
+                    ai_fallback=request.ai_fallback,
                 )
             else:
                 workflow = await self.create_workflow(