From c643472f3c2f8a01f02520719eacca9ac5ee042b Mon Sep 17 00:00:00 2001
From: Shuchang Zheng <wintonzheng0325@gmail.com>
Date: Wed, 20 Nov 2024 07:48:07 -0800
Subject: [PATCH] Fetch less in workflows page (#1226)

---
 skyvern-frontend/src/routes/workflows/WorkflowActions.tsx | 8 +++-----
 skyvern-frontend/src/routes/workflows/Workflows.tsx       | 5 +++--
 2 files changed, 6 insertions(+), 7 deletions(-)

diff --git a/skyvern-frontend/src/routes/workflows/WorkflowActions.tsx b/skyvern-frontend/src/routes/workflows/WorkflowActions.tsx
index ffeb5aca..45e3b4ff 100644
--- a/skyvern-frontend/src/routes/workflows/WorkflowActions.tsx
+++ b/skyvern-frontend/src/routes/workflows/WorkflowActions.tsx
@@ -34,12 +34,11 @@ import { AxiosError } from "axios";
 import { useNavigate } from "react-router-dom";
 import { stringify as convertToYAML } from "yaml";
 import { convert } from "./editor/workflowEditorUtils";
-import { useWorkflowQuery } from "./hooks/useWorkflowQuery";
 import { WorkflowApiResponse } from "./types/workflowTypes";
 import { WorkflowCreateYAMLRequest } from "./types/workflowYamlTypes";
 
 type Props = {
-  id: string;
+  workflow: WorkflowApiResponse;
 };
 
 function downloadFile(fileName: string, contents: string) {
@@ -58,10 +57,9 @@ function downloadFile(fileName: string, contents: string) {
   document.body.removeChild(element);
 }
 
-function WorkflowActions({ id }: Props) {
+function WorkflowActions({ workflow }: Props) {
   const credentialGetter = useCredentialGetter();
   const queryClient = useQueryClient();
-  const { data: workflow } = useWorkflowQuery({ workflowPermanentId: id });
   const navigate = useNavigate();
 
   function handleExport(type: "json" | "yaml") {
@@ -187,7 +185,7 @@ function WorkflowActions({ id }: Props) {
           <Button
             variant="destructive"
             onClick={() => {
-              deleteWorkflowMutation.mutate(id);
+              deleteWorkflowMutation.mutate(workflow.workflow_permanent_id);
             }}
             disabled={deleteWorkflowMutation.isPending}
           >
diff --git a/skyvern-frontend/src/routes/workflows/Workflows.tsx b/skyvern-frontend/src/routes/workflows/Workflows.tsx
index 6e10d43f..c39f7106 100644
--- a/skyvern-frontend/src/routes/workflows/Workflows.tsx
+++ b/skyvern-frontend/src/routes/workflows/Workflows.tsx
@@ -1,5 +1,4 @@
 import { getClient } from "@/api/AxiosClient";
-import { WorkflowApiResponse, WorkflowRunApiResponse } from "@/api/types";
 import { StatusBadge } from "@/components/StatusBadge";
 import { Alert, AlertDescription, AlertTitle } from "@/components/ui/alert";
 import { Button } from "@/components/ui/button";
@@ -42,6 +41,8 @@ import { ImportWorkflowButton } from "./ImportWorkflowButton";
 import { WorkflowCreateYAMLRequest } from "./types/workflowYamlTypes";
 import { WorkflowActions } from "./WorkflowActions";
 import { WorkflowTitle } from "./WorkflowTitle";
+import { WorkflowApiResponse } from "./types/workflowTypes";
+import { WorkflowRunApiResponse } from "@/api/types";
 
 const emptyWorkflowRequest: WorkflowCreateYAMLRequest = {
   title: "New Workflow",
@@ -285,7 +286,7 @@ function Workflows() {
                             <TooltipContent>Create New Run</TooltipContent>
                           </Tooltip>
                         </TooltipProvider>
-                        <WorkflowActions id={workflow.workflow_permanent_id} />
+                        <WorkflowActions workflow={workflow} />
                       </div>
                     </TableCell>
                   </TableRow>