add improve prompt button to task block and action block (#4020)

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;

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;
 }
 

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 });

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 });