enhance: add stopping of workflows

obot-platform · Dec 26, 2024 · f4a3aeb · f4a3aeb
1 parent 6ce2501
commit f4a3aeb
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 20 deletions.
diff --git a/ui/admin/app/components/chat/Chat.tsx b/ui/admin/app/components/chat/Chat.tsx
@@ -12,16 +12,7 @@ type ChatProps = React.HTMLAttributes<HTMLDivElement> & {
 };
 
 export function Chat({ className }: ChatProps) {
-    const {
-        id,
-        messages,
-        threadId,
-        mode,
-        invoke,
-        readOnly,
-        isInvoking,
-        isRunning,
-    } = useChat();
+    const { id, messages, threadId, mode, invoke, readOnly } = useChat();
     const [runTriggered, setRunTriggered] = useState(false);
 
     const showMessagePane =
@@ -59,8 +50,6 @@ export function Chat({ className }: ChatProps) {
                         popoverContentProps={{
                             className: cn({ "translate-y-[-50%]": !threadId }),
                         }}
-                        loading={isInvoking || isRunning}
-                        disabled={isInvoking || isRunning}
                     >
                         Run
                     </RunWorkflow>

diff --git a/ui/admin/app/components/chat/RunWorkflow.tsx b/ui/admin/app/components/chat/RunWorkflow.tsx
@@ -1,9 +1,11 @@
+import { SquareIcon } from "lucide-react";
 import { ComponentProps, useState } from "react";
 import useSWR from "swr";
 
 import { WorkflowService } from "~/lib/service/api/workflowService";
 import { cn } from "~/lib/utils";
 
+import { useChat } from "~/components/chat/ChatContext";
 import { RunWorkflowForm } from "~/components/chat/RunWorkflowForm";
 import { ModelProviderTooltip } from "~/components/model-providers/ModelProviderTooltip";
 import { Button, ButtonProps } from "~/components/ui/button";
@@ -32,19 +34,35 @@ export function RunWorkflow({
         WorkflowService.getWorkflowById.key(workflowId),
         ({ workflowId }) => WorkflowService.getWorkflowById(workflowId)
     );
+    const { abortRunningThread, isInvoking, isRunning } = useChat();
 
     const params = workflow?.params;
 
-    if (!params || isLoading || !modelProviderConfigured)
+    const loading = props.loading || isLoading || isInvoking;
+    const disabled = props.disabled || loading || !modelProviderConfigured;
+    if (isRunning) {
+        return (
+            <Button
+                onClick={() => abortRunningThread()}
+                {...props}
+                disabled={disabled}
+                startContent={
+                    <SquareIcon className="fill-primary-foreground text-primary-foreground !w-3 !h-3" />
+                }
+            >
+                Stop Workflow
+            </Button>
+        );
+    }
+
+    if (!params || !modelProviderConfigured)
         return (
             <ModelProviderTooltip enabled={modelProviderConfigured}>
                 <Button
                     onClick={() => onSubmit()}
                     {...props}
-                    disabled={
-                        props.disabled || isLoading || !modelProviderConfigured
-                    }
-                    loading={isLoading || props.loading}
+                    disabled={disabled}
+                    loading={loading}
                 >
                     Run Workflow
                 </Button>
@@ -56,8 +74,8 @@ export function RunWorkflow({
             <PopoverTrigger asChild>
                 <Button
                     {...props}
-                    disabled={props.disabled || open || isLoading}
-                    loading={props.loading || isLoading}
+                    disabled={disabled || open}
+                    loading={loading}
                     onClick={() => setOpen((prev) => !prev)}
                 >
                     Run Workflow

diff --git a/ui/admin/app/components/model-providers/ModelProviderConfigure.tsx b/ui/admin/app/components/model-providers/ModelProviderConfigure.tsx
@@ -148,7 +148,7 @@ export function ModelProviderConfigureContent({
     return (
         <>
             <DialogHeader className="space-y-0">
-                <DialogTitle className="flex items-center gap-2 px-6 py-4">
+                <DialogTitle className="flex items-center gap-2 px-4 py-4">
                     <ModelProviderIcon modelProvider={modelProvider} />{" "}
                     {modelProvider.configured
                         ? `Configure ${modelProvider.name}`