diff --git a/apps/web/src/components/widgets/AgentStatusWidget.tsx b/apps/web/src/components/widgets/AgentStatusWidget.tsx index adb6a03..6dcdb8a 100644 --- a/apps/web/src/components/widgets/AgentStatusWidget.tsx +++ b/apps/web/src/components/widgets/AgentStatusWidget.tsx @@ -16,6 +16,21 @@ interface Agent { error?: string; } +function isWorking(status: string): boolean { + const s = status.toLowerCase(); + return s === "running" || s === "working"; +} + +function isIdle(status: string): boolean { + const s = status.toLowerCase(); + return s === "idle" || s === "spawning" || s === "waiting" || s === "queued"; +} + +function isErrored(status: string): boolean { + const s = status.toLowerCase(); + return s === "failed" || s === "error"; +} + export function AgentStatusWidget({ id: _id, config: _config }: WidgetProps): React.JSX.Element { const [agents, setAgents] = useState([]); const [isLoading, setIsLoading] = useState(true); @@ -74,25 +89,20 @@ export function AgentStatusWidget({ id: _id, config: _config }: WidgetProps): Re }, [fetchAgents]); const getStatusIcon = (status: string): React.JSX.Element => { - const statusLower = status.toLowerCase(); - switch (statusLower) { - case "running": - case "working": - return ; - case "spawning": - case "queued": - return ; - case "completed": - return ; - case "failed": - case "error": - return ; - case "terminated": - case "killed": - return ; - default: - return ; + if (isWorking(status)) { + return ; } + if (isIdle(status)) { + return ; + } + if (isErrored(status)) { + return ; + } + const s = status.toLowerCase(); + if (s === "completed" || s === "terminated" || s === "killed") { + return ; + } + return ; }; const getStatusText = (status: string): string => { @@ -121,9 +131,9 @@ export function AgentStatusWidget({ id: _id, config: _config }: WidgetProps): Re const stats = { total: agents.length, - working: agents.filter((a) => a.status.toLowerCase() === "running").length, - idle: agents.filter((a) => a.status.toLowerCase() === "spawning").length, - error: agents.filter((a) => a.status.toLowerCase() === "failed").length, + working: agents.filter((a) => isWorking(a.status)).length, + idle: agents.filter((a) => isIdle(a.status)).length, + error: agents.filter((a) => isErrored(a.status)).length, }; if (isLoading) { @@ -176,9 +186,9 @@ export function AgentStatusWidget({ id: _id, config: _config }: WidgetProps): Re