Types and exports

import { Agent, AnthropicModel, type AgentState } from "stateweave";

type AgentState = {
  version: 1;
  nodes: Array<{
    id: string;
    kind:
      | "system" | "goal" | "inference" | "semantic"
      | "tool_call" | "tool_result" | "resource"
      | "verification" | "answer" | "protocol_error";
    parents: string[];
    payload: unknown;
    createdAt: string;
    sequence: number;
    resourceKey?: string;
  }>;
  frontier: string[];
};

type SemanticNodeType = { name: string; description: string };

// payload shape when a causal node has kind: "semantic"
type SemanticPayload = {
  type: string;
  key: string;
  content: unknown;
};

type ModelInput = {
  prompt: string;
  frame?: GraphFrame;
  mode?: "graph_ops" | "text";
  parameters?: ModelParameters;
  system?: string;
  signal?: AbortSignal;
};

type Model = {
  complete(input: ModelInput): Promise<ModelOutput>;
  stream(input: ModelInput): AsyncIterable<ModelToken>;
};

type Tool = {
  name: string;
  description: string;
  schema: z.ZodSchema;
  execute(args: unknown): Promise<unknown>;
};

type AgentProgress = {
  iteration: number;
  phase: "context" | "model" | "tool" | "final" | "retrying";
  modelCalls: number;
  toolCalls: number;
  totalInputTokens: number;
  outputTokens: number;
  detail: string;
  contextTokens?: number;
  prompt?: string;
  state?: AgentState;
  graph?: StateGraph;
  rawModelOutput?: string;
  action?: "tool" | "final" | "invalid";
  tool?: string;
  error?: string;
};

type AgentTraceStep = {
  step: number;
  nodeIds: string[];
  prompt: string;
  contextTokens: number;
  rawModelOutput: string;
  action: "tool" | "final" | "invalid";
  tool?: string;
  error?: string;
};

type AgentStreamEvent =
  | { type: "metadata"; metadata: AgentStartMetadata }
  | { type: "model_token"; iteration: number; token: string }
  | { type: "model_metadata"; iteration: number; metadata: Record<string, unknown> }
  | { type: "progress"; progress: AgentProgress }
  | { type: "final"; result: AgentRunResult };

type StateGraph = { nodes: GraphNode[]; edges: GraphEdge[] };