From 639c3a2948abd658629e724dc436d6f7b2762262 Mon Sep 17 00:00:00 2001
From: Jonathan Dobson <jon.m.dobson@gmail.com>
Date: Tue, 18 Nov 2025 08:43:17 -0800
Subject: [PATCH] add improve prompt button to task block and action block
 (#4020)

---
 skyvern-frontend/src/components/ImprovePrompt.tsx             | 2 +-
 .../src/components/WorkflowBlockInputTextarea.tsx             | 2 +-
 .../routes/workflows/editor/nodes/ActionNode/ActionNode.tsx   | 4 ++++
 .../src/routes/workflows/editor/nodes/TaskNode/TaskNode.tsx   | 4 ++++
 4 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/skyvern-frontend/src/components/ImprovePrompt.tsx b/skyvern-frontend/src/components/ImprovePrompt.tsx
index 8e005c72..dc99cd6f 100644
--- a/skyvern-frontend/src/components/ImprovePrompt.tsx
+++ b/skyvern-frontend/src/components/ImprovePrompt.tsx
@@ -25,7 +25,7 @@ import { useCredentialGetter } from "@/hooks/useCredentialGetter";
 import { ImprovePromptForWorkflowResponse } from "@/routes/workflows/types/workflowTypes";
 
 interface Props {
-  context?: string;
+  context?: Record<string, unknown>;
   isVisible?: boolean;
   onBegin?: () => void;
   onEnd?: () => void;
diff --git a/skyvern-frontend/src/components/WorkflowBlockInputTextarea.tsx b/skyvern-frontend/src/components/WorkflowBlockInputTextarea.tsx
index 492230e5..3cb41524 100644
--- a/skyvern-frontend/src/components/WorkflowBlockInputTextarea.tsx
+++ b/skyvern-frontend/src/components/WorkflowBlockInputTextarea.tsx
@@ -10,7 +10,7 @@ import { useDebouncedCallback } from "use-debounce";
 import { ImprovePrompt } from "./ImprovePrompt";
 
 interface AiImprove {
-  context?: string;
+  context?: Record<string, unknown>;
   useCase: string;
 }
 
diff --git a/skyvern-frontend/src/routes/workflows/editor/nodes/ActionNode/ActionNode.tsx b/skyvern-frontend/src/routes/workflows/editor/nodes/ActionNode/ActionNode.tsx
index 31428bc1..64f83f5f 100644
--- a/skyvern-frontend/src/routes/workflows/editor/nodes/ActionNode/ActionNode.tsx
+++ b/skyvern-frontend/src/routes/workflows/editor/nodes/ActionNode/ActionNode.tsx
@@ -139,6 +139,10 @@ function ActionNode({ id, data, type }: NodeProps<ActionNode>) {
                 <HelpTooltip content={navigationGoalTooltip} />
               </div>
               <WorkflowBlockInputTextarea
+                aiImprove={{
+                  context: { block_type: "Action Block" },
+                  useCase: "task_v1_prompt",
+                }}
                 nodeId={id}
                 onChange={(value) => {
                   update({ navigationGoal: value });
diff --git a/skyvern-frontend/src/routes/workflows/editor/nodes/TaskNode/TaskNode.tsx b/skyvern-frontend/src/routes/workflows/editor/nodes/TaskNode/TaskNode.tsx
index e0be0401..b4855a9c 100644
--- a/skyvern-frontend/src/routes/workflows/editor/nodes/TaskNode/TaskNode.tsx
+++ b/skyvern-frontend/src/routes/workflows/editor/nodes/TaskNode/TaskNode.tsx
@@ -137,6 +137,10 @@ function TaskNode({ id, data, type }: NodeProps<TaskNode>) {
                       />
                     </div>
                     <WorkflowBlockInputTextarea
+                      aiImprove={{
+                        context: { block_type: "Task Block" },
+                        useCase: "task_v1_prompt",
+                      }}
                       nodeId={id}
                       onChange={(value) => {
                         update({ navigationGoal: value });