diff --git a/packages/web/src/components/AgentOutputViewer.tsx b/packages/web/src/components/AgentOutputViewer.tsx index f47eae8..e9e9532 100644 --- a/packages/web/src/components/AgentOutputViewer.tsx +++ b/packages/web/src/components/AgentOutputViewer.tsx @@ -1,7 +1,7 @@ import { useEffect, useRef, useState } from "react"; import { Button } from "@/components/ui/button"; import { Badge } from "@/components/ui/badge"; -import { ArrowDown, Pause, Play, AlertCircle } from "lucide-react"; +import { ArrowDown, Pause, Play, AlertCircle, Square } from "lucide-react"; import { trpc } from "@/lib/trpc"; import { useSubscriptionWithErrorHandling } from "@/hooks"; import { @@ -13,9 +13,11 @@ import { interface AgentOutputViewerProps { agentId: string; agentName?: string; + status?: string; + onStop?: (id: string) => void; } -export function AgentOutputViewer({ agentId, agentName }: AgentOutputViewerProps) { +export function AgentOutputViewer({ agentId, agentName, status, onStop }: AgentOutputViewerProps) { const [messages, setMessages] = useState([]); const [follow, setFollow] = useState(true); const containerRef = useRef(null); @@ -106,6 +108,17 @@ export function AgentOutputViewer({ agentId, agentName }: AgentOutputViewerProps )}
+ {onStop && (status === "running" || status === "waiting_for_input") && ( + + )}