import type { UIMessage } from "ai"; import type { Agent, SubAgentClass } from "./index"; export type AgentToolRunStatus = | "starting" | "running" | "completed" | "error" | "aborted" | "interrupted"; export type AgentToolTerminalStatus = Extract< AgentToolRunStatus, "completed" | "error" | "aborted" | "interrupted" >; export type AgentToolDisplayMetadata = { name?: string; icon?: string; } & Record; export type AgentToolRunInfo = { runId: string; parentToolCallId?: string; agentType: string; inputPreview?: unknown; status: AgentToolRunStatus; display?: AgentToolDisplayMetadata; displayOrder: number; startedAt: number; completedAt?: number; }; export type AgentToolLifecycleResult = { status: AgentToolTerminalStatus; summary?: string; error?: string; }; export type RunAgentToolOptions = { input: Input; runId?: string; parentToolCallId?: string; displayOrder?: number; signal?: AbortSignal; inputPreview?: unknown; display?: AgentToolDisplayMetadata; }; export type RunAgentToolResult = { runId: string; agentType: string; status: AgentToolTerminalStatus; output?: Output; summary?: string; error?: string; }; export type ChatCapableAgentClass = SubAgentClass; export type AgentToolRunInspection = { runId: string; status: Exclude; requestId?: string; streamId?: string; output?: Output; summary?: string; error?: string; startedAt: number; completedAt?: number; }; export type AgentToolStoredChunk = { sequence: number; body: string; }; export type AgentToolChildAdapter = { startAgentToolRun( input: Input, options: { runId: string; signal?: AbortSignal } ): Promise>; cancelAgentToolRun(runId: string, reason?: unknown): Promise; inspectAgentToolRun( runId: string ): Promise | null>; getAgentToolChunks( runId: string, options?: { afterSequence?: number } ): Promise; tailAgentToolRun?( runId: string, options?: { afterSequence?: number; signal?: AbortSignal } ): Promise>; }; export type AgentToolEvent = | { kind: "started"; runId: string; agentType: string; inputPreview?: unknown; order: number; display?: AgentToolDisplayMetadata; } | { kind: "chunk"; runId: string; body: string; } | { kind: "finished"; runId: string; summary: string; } | { kind: "error"; runId: string; error: string; } | { kind: "aborted"; runId: string; reason?: string; } | { kind: "interrupted"; runId: string; error: string; }; export type AgentToolEventMessage = { type: "agent-tool-event"; parentToolCallId?: string; sequence: number; replay?: true; event: AgentToolEvent; }; export type AgentToolRunState = { runId: string; agentType: string; parentToolCallId?: string; inputPreview?: unknown; order: number; display?: AgentToolDisplayMetadata; status: "running" | "completed" | "error" | "aborted" | "interrupted"; parts: UIMessage["parts"]; summary?: string; error?: string; subAgent: { agent: string; name: string }; }; export type AgentToolEventState = { runsById: Record; runsByToolCallId: Record; unboundRuns: AgentToolRunState[]; };