From d1b26d70688491ae5ae1bce47a663437eaa6a019 Mon Sep 17 00:00:00 2001
From: Shuchang Zheng <wintonzheng0325@gmail.com>
Date: Mon, 24 Feb 2025 09:37:02 -0800
Subject: [PATCH] Add max iterations override header (#1825)

---
 .../src/routes/tasks/create/PromptBox.tsx     | 20 +++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/skyvern-frontend/src/routes/tasks/create/PromptBox.tsx b/skyvern-frontend/src/routes/tasks/create/PromptBox.tsx
index 9d9f088e..0a8a018e 100644
--- a/skyvern-frontend/src/routes/tasks/create/PromptBox.tsx
+++ b/skyvern-frontend/src/routes/tasks/create/PromptBox.tsx
@@ -150,6 +150,7 @@ function PromptBox() {
   );
   const [publishWorkflow, setPublishWorkflow] = useState(false);
   const [totpIdentifier, setTotpIdentifier] = useState("");
+  const [maxStepsOverride, setMaxStepsOverride] = useState<string | null>(null);
   const [showAdvancedSettings, setShowAdvancedSettings] = useState(false);
 
   const startObserverCruiseMutation = useMutation({
@@ -164,6 +165,11 @@ function PromptBox() {
           totp_identifier: totpIdentifier,
           publish_workflow: publishWorkflow,
         },
+        {
+          headers: {
+            "x-max-iterations-override": maxStepsOverride,
+          },
+        },
       );
     },
     onSuccess: (response) => {
@@ -384,6 +390,20 @@ function PromptBox() {
                       }}
                     />
                   </div>
+                  <div className="flex gap-16">
+                    <div className="w-48 shrink-0">
+                      <div className="text-sm">Max Steps Override</div>
+                      <div className="text-xs text-slate-400">
+                        The maximum number of steps to take for this task.
+                      </div>
+                    </div>
+                    <Input
+                      value={maxStepsOverride ?? ""}
+                      onChange={(event) => {
+                        setMaxStepsOverride(event.target.value);
+                      }}
+                    />
+                  </div>
                 </div>
               </div>
             ) : null}