'use client' import { useEffect, useState } from 'react' import { HugeiconsIcon } from '@hugeicons/react' import { AlertCircleIcon, CheckmarkCircle02Icon, Clock01Icon, FlashIcon, Rocket01Icon, SentIcon, Settings01Icon, } from '@hugeicons/core-free-icons' import type { CrewMember } from '@/hooks/use-crew-status' import { cn } from '@/lib/utils' type Mode = 'auto' | 'manual' | 'broadcast' type Assignment = { workerId: string; task: string; rationale: string; expectedOutput?: string; dependsOn?: Array; reviewRequired?: boolean } type ParsedCheckpoint = { stateLabel: string result: string | null blocker: string | null nextAction: string | null filesChanged?: string | null commandsRun?: string | null } type DispatchResult = { workerId: string ok: boolean output: string error: string | null durationMs: number exitCode: number | null delivery?: 'tmux' | 'oneshot' checkpoint?: ParsedCheckpoint | null checkpointStatus?: 'checkpointed' | 'timeout' | 'not-requested' } export type DispatchResponse = { dispatchedAt: number completedAt: number results: Array } type FollowUpResponse = { ok: boolean summary?: { checkpointed: number; stale: number; waiting: number; unavailable: number } continuation?: { results?: Array } | null error?: string } type Props = { members: Array roomIds: Array selectedId: string | null open: boolean onOpen?: () => void onClose: () => void showClosedDock?: boolean embedded?: boolean seedPrompt?: string | null seedMode?: Mode seedKey?: string | number | null onResults: (response: DispatchResponse) => void } function roleForMember(members: Array, id: string): string { return members.find((member) => member.id === id)?.role || 'Worker' } const QUICK_ROUTES = [ 'Research', 'Builder', 'Reviewer', 'Docs', 'Ops', 'Best match', 'Auto', ] export function RouterChat({ members, roomIds, selectedId, open, onOpen, onClose, showClosedDock = false, embedded = false, seedPrompt, seedMode, seedKey, onResults, }: Props) { const [mode, setMode] = useState('auto') const [prompt, setPrompt] = useState('') const [decomposing, setDecomposing] = useState(false) const [decomposeError, setDecomposeError] = useState(null) const [assignments, setAssignments] = useState>([]) const [unassigned, setUnassigned] = useState>([]) const [dispatching, setDispatching] = useState(false) const [dispatchError, setDispatchError] = useState(null) const [results, setResults] = useState(null) const [followUp, setFollowUp] = useState(null) useEffect(() => { if (!seedPrompt?.trim()) return setPrompt(seedPrompt) if (seedMode) setMode(seedMode) setAssignments([]) setUnassigned([]) setResults(null) setFollowUp(null) setDecomposeError(null) setDispatchError(null) }, [seedKey, seedMode, seedPrompt]) useEffect(() => { if ( mode === 'manual' && selectedId && assignments.length === 0 && prompt.trim() ) { setAssignments([ { workerId: selectedId, task: prompt.trim(), rationale: 'Manual target.', }, ]) } }, [mode, selectedId, assignments.length, prompt]) if (!open) { if (!showClosedDock || !onOpen) return null const targetLabel = selectedId ? selectedId : roomIds.length ? `${roomIds.length} in room` : `${members.length} workers` return (
) } const eligibleWorkers = members.map((m) => ({ id: m.id, role: m.role, model: m.model, specialty: m.specialty, mission: m.mission || m.lastSessionTitle || undefined, skills: m.skills ?? [], capabilities: m.capabilities ?? [], notes: [ m.specialty, m.mission || m.lastSessionTitle, m.skills?.length ? `skills=${m.skills.join(',')}` : '', `${m.sessionCount} sess · ${m.totalTokens} tok`, ].filter(Boolean).join(' · '), })) async function autoDecompose(): Promise | null> { if (!prompt.trim()) return null setDecomposing(true) setDecomposeError(null) try { const res = await fetch('/api/swarm-decompose', { method: 'POST', headers: { 'Content-Type': 'application/json' }, body: JSON.stringify({ prompt: prompt.trim(), workers: eligibleWorkers, }), }) if (!res.ok) { const text = await res.text() throw new Error(text || `HTTP ${res.status}`) } const data = (await res.json()) as { ok: boolean assignments?: Array unassigned?: Array error?: string } if (!data.ok) throw new Error(data.error || 'decompose failed') const nextAssignments = data.assignments ?? [] setAssignments(nextAssignments) setUnassigned(data.unassigned ?? []) return nextAssignments } catch (err) { setDecomposeError(err instanceof Error ? err.message : 'decompose failed') return null } finally { setDecomposing(false) } } async function dispatch() { let plan: Array = [] if (mode === 'auto') { if (assignments.length === 0) { const nextAssignments = await autoDecompose() if (!nextAssignments || nextAssignments.length === 0) return plan = nextAssignments } else { plan = assignments } } else if (mode === 'manual') { if (!selectedId) return plan = [ { workerId: selectedId, task: prompt.trim(), rationale: 'Manual single-target.', }, ] } else { const targets = roomIds.length > 0 ? roomIds : members.map((m) => m.id) plan = targets.map((id) => ({ workerId: id, task: prompt.trim(), rationale: 'Broadcast.', })) } if (plan.length === 0) return setDispatching(true) setDispatchError(null) setResults(null) setFollowUp(null) try { const res = await fetch('/api/swarm-dispatch', { method: 'POST', headers: { 'Content-Type': 'application/json' }, body: JSON.stringify({ assignments: plan, timeoutSeconds: 300, waitForCheckpoint: true, checkpointPollSeconds: 90, }), }) if (!res.ok) { const text = await res.text() throw new Error(text || `HTTP ${res.status}`) } const data = (await res.json()) as DispatchResponse setResults(data) onResults(data) if (plan.length > 1 && data.results.some((result) => result.checkpointStatus === 'checkpointed')) { const reviewer = members.find((member) => member.id === 'swarm6') ?? members.find((member) => /review|qa|critic/i.test(`${member.role} ${member.specialty ?? ''}`)) const follow = await fetch('/api/swarm-orchestrator-loop', { method: 'POST', headers: { 'Content-Type': 'application/json' }, body: JSON.stringify({ workerIds: [...new Set([...plan.map((item) => item.workerId), ...(reviewer ? [reviewer.id] : [])])], staleMinutes: 3, autoContinue: true, allowExecution: false, reviewWorkerId: reviewer?.id, }), }) const followData = (await follow.json().catch(() => null)) as FollowUpResponse | null if (followData) setFollowUp(followData) } } catch (err) { setDispatchError(err instanceof Error ? err.message : 'dispatch failed') } finally { setDispatching(false) } } function reset() { setPrompt('') setAssignments([]) setUnassigned([]) setResults(null) setFollowUp(null) setDecomposeError(null) setDispatchError(null) } return (
{!embedded ? ( <>
Agent Router Chat
Type a mission, choose routing, dispatch. Keep workers selected in cards.
) : null}
{!embedded ? : null} {!embedded ? ( ) : null}