From b2e8c1e0cf2ddddeecaf6b3707c40a369a8057e5 Mon Sep 17 00:00:00 2001
From: Kerem Yilmaz <kerem@skyvern.com>
Date: Tue, 1 Oct 2024 07:16:46 -0700
Subject: [PATCH] Put edit workflow button in run page (#893)

---
 .../src/routes/workflows/WorkflowRun.tsx      | 37 +++++++++++++------
 1 file changed, 26 insertions(+), 11 deletions(-)

diff --git a/skyvern-frontend/src/routes/workflows/WorkflowRun.tsx b/skyvern-frontend/src/routes/workflows/WorkflowRun.tsx
index e9a854fd..e630dd9a 100644
--- a/skyvern-frontend/src/routes/workflows/WorkflowRun.tsx
+++ b/skyvern-frontend/src/routes/workflows/WorkflowRun.tsx
@@ -29,13 +29,19 @@ import { useCredentialGetter } from "@/hooks/useCredentialGetter";
 import { basicTimeFormat } from "@/util/timeFormat";
 import { cn } from "@/util/utils";
 import { keepPreviousData, useQuery } from "@tanstack/react-query";
-import { useNavigate, useParams, useSearchParams } from "react-router-dom";
+import {
+  Link,
+  useNavigate,
+  useParams,
+  useSearchParams,
+} from "react-router-dom";
 import { TaskActions } from "../tasks/list/TaskActions";
 import { TaskListSkeletonRows } from "../tasks/list/TaskListSkeletonRows";
 import { useEffect, useState } from "react";
 import { statusIsNotFinalized, statusIsRunningOrQueued } from "../tasks/types";
 import { envCredential } from "@/util/env";
 import { toast } from "@/components/ui/use-toast";
+import { Pencil2Icon, PlayIcon } from "@radix-ui/react-icons";
 
 type StreamMessage = {
   task_id: string;
@@ -230,16 +236,25 @@ function WorkflowRun() {
             <StatusBadge status={workflowRun?.status} />
           ) : null}
         </div>
-        <Button
-          onClick={() => {
-            navigate(`/workflows/${workflowPermanentId}/run`, {
-              state: { data: parameters },
-            });
-          }}
-          variant="secondary"
-        >
-          Rerun Workflow
-        </Button>
+        <div className="flex gap-2">
+          <Button asChild variant="secondary">
+            <Link to={`/workflows/${workflowPermanentId}/edit`}>
+              <Pencil2Icon className="mr-2 h-4 w-4" />
+              Edit Workflow
+            </Link>
+          </Button>
+          <Button asChild>
+            <Link
+              to={`/workflows/${workflowPermanentId}/run`}
+              state={{
+                data: parameters,
+              }}
+            >
+              <PlayIcon className="mr-2 h-4 w-4" />
+              Rerun Workflow
+            </Link>
+          </Button>
+        </div>
       </header>
       {getStream()}
       <div className="space-y-4">