@erdoai/ui 0.1.69 → 0.1.71

package/dist/index.d.cts

@@ -1,9 +1,10 @@
 import * as react_jsx_runtime from 'react/jsx-runtime';
 import * as React$1 from 'react';
-import React__default from 'react';
-import { ErdoClient, ContentItem, InvokeResult, LogEntry, ToolGroup, InvocationStatus as InvocationStatus$1, StepStreamState, SSEEvent, Thread } from '@erdoai/types';
+import React__default, { ReactNode } from 'react';
+import { ErdoClient, ContentItem, InvokeResult, LogEntry, ToolGroup, InvocationStatus as InvocationStatus$1, AgentStreamState, SSEEvent, Thread } from '@erdoai/types';
 export { ContentItem } from '@erdoai/types';
 import * as RechartsPrimitive from 'recharts';
+import { StoreApi } from 'zustand';
 import { ClassValue } from 'clsx';
 
 /**
@@ -131,9 +132,10 @@ interface ErdoProviderConfig {
      */
     onSuggestionClick?: (suggestion: string) => void;
     /**
-     * Enable the new agent event system.
-     * When true, uses simple flat events (agent_started, text_delta, tool_started, etc.)
-     * When false (default), uses legacy path-based event routing.
+     * Enable the agent event system (flat content model).
+     * When true, routes step_started/step_finished and message content events
+     * through handleAgentEvent into AgentStreamState.
+     * When false (default), uses legacy path-based event routing (bot service).
      */
     useAgentEvents?: boolean;
 }
@@ -842,6 +844,7 @@ interface ContentProps {
     components?: Record<string, React__default.ComponentType<any>>;
 }
 declare function Content({ content, className, components }: ContentProps): react_jsx_runtime.JSX.Element | null;
+declare function normalizeUIGenerationChunks(value: unknown): ContentChunk[] | null;
 
 interface ErdoUIProps {
     /** Result from an Erdo invocation */
@@ -1005,8 +1008,8 @@ interface ThinkingContentProps {
     className?: string;
 }
 /**
- * Renders Claude's thinking/reasoning content in a collapsible panel.
- * Handles both regular thinking and redacted (encrypted) thinking.
+ * Renders Claude's thinking/reasoning content as subtle inline text.
+ * No container/chrome; supports redacted (encrypted) thinking blocks.
  */
 declare function ThinkingContent({ item, className }: ThinkingContentProps): react_jsx_runtime.JSX.Element;
 
@@ -1085,7 +1088,7 @@ interface CollapsibleCodeBlockProps {
 /**
  * A collapsible code block with syntax highlighting.
  */
-declare function CollapsibleCodeBlock({ language, value, defaultOpen, title, closeTitle, className, theme, }: CollapsibleCodeBlockProps): react_jsx_runtime.JSX.Element;
+declare const CollapsibleCodeBlock: React$1.NamedExoticComponent<CollapsibleCodeBlockProps>;
 
 interface SqlContentProps {
     item: ContentItem;
@@ -1102,8 +1105,7 @@ interface CodegenContentProps {
 }
 /**
  * Renders code generation content with step-by-step explanation, code, and warnings.
- *
- * Expects parsedJson to be populated upstream (via normalizeContent or SSE handler).
+ * Prefers canonical `content` (parsed object); falls back to legacy `parsedJson` (DB string path).
  */
 declare function CodegenContent({ item, className }: CodegenContentProps): react_jsx_runtime.JSX.Element | null;
 
@@ -1113,6 +1115,7 @@ interface CodeexecContentProps {
 }
 /**
  * Renders code execution output (stdout, stderr, warnings).
+ * Prefers canonical `content`; falls back to legacy `parsedJson`.
  */
 declare function CodeexecContent({ item, className }: CodeexecContentProps): react_jsx_runtime.JSX.Element | null;
 
@@ -1282,8 +1285,8 @@ interface MemoryContentProps {
 declare function MemoryContent({ content, className }: MemoryContentProps): react_jsx_runtime.JSX.Element;
 
 interface StepTreeContentProps {
-    /** The step stream state containing all steps */
-    stepState: StepStreamState;
+    /** The agent stream state containing all steps and content */
+    agentState: AgentStreamState;
     /** Optional Content component for rendering ui_generation/codegen */
     ContentComponent?: React__default.ComponentType<{
         content: ContentItem;
@@ -1296,7 +1299,7 @@ interface StepTreeContentProps {
  *
  * Used for both streaming and reload paths to ensure consistent rendering.
  */
-declare function StepTreeContent({ stepState, ContentComponent, className }: StepTreeContentProps): react_jsx_runtime.JSX.Element | null;
+declare function StepTreeContent({ agentState, ContentComponent, className }: StepTreeContentProps): react_jsx_runtime.JSX.Element | null;
 
 /** Node structure from UI generation content */
 interface UINode {
@@ -1492,6 +1495,42 @@ declare function useMultipleDatasetRequests(requests: DatasetFetchRequest[], inv
  */
 declare function useMultipleDatasetContents(datasetSlugs: string[], invocationId: string, sqlQueries?: Record<string, string>, resourceKeys?: Record<string, string>): UseDatasetContentsResult[];
 
+/**
+ * Zustand store for agent streaming state.
+ *
+ * Holds the global AgentStreamState and per-invocation sliced states.
+ * Components subscribe to individual invocation slices via useInvocationState(),
+ * so only the affected component re-renders when an event arrives.
+ *
+ * The store calls handleAgentEvent internally — consumers just push raw SSE events.
+ */
+
+declare function sliceAgentStateToRoot(state: AgentStreamState, rootStepID: string): AgentStreamState;
+interface AgentStreamStore {
+    /** The single merged state from all events */
+    globalState: AgentStreamState;
+    /** Per-invocation sliced states. Key: invocation_id */
+    invocationStates: Record<string, AgentStreamState>;
+    /** Maps invocation_id → root step_id */
+    invocationRootSteps: Record<string, string>;
+    /** Queue for invocations awaiting their root step_started */
+    pendingInvocationQueue: string[];
+    /**
+     * Push a raw SSE event into the store.
+     * Calls handleAgentEvent internally and re-slices affected invocations.
+     * @param event The raw SSE event
+     * @param reconcile Optional post-processing (e.g., reconcileThinkingContentState)
+     */
+    pushEvent: (event: SSEEvent, reconcile?: (state: AgentStreamState) => AgentStreamState) => void;
+    /** Register an invocation_id (from agent_invocation event). Queues it for root mapping. */
+    registerInvocation: (invocationId: string) => void;
+    /** Map an invocation_id to its root step_id (from step_started with type: 'agent'). */
+    mapInvocationRoot: (invocationId: string, rootStepId: string) => void;
+    /** Reset the store (new session). */
+    reset: () => void;
+}
+declare function createAgentStreamStore(): StoreApi<AgentStreamStore>;
+
 interface UseThreadOptions {
     /** Initial thread ID to use (if resuming an existing thread) */
     threadId?: string;
@@ -1513,8 +1552,8 @@ interface UseThreadReturn {
     error: Error | null;
     /** Active messages being streamed (with parsedJson populated) */
     activeMessages: MessageWithContents[];
-    /** Step stream state (for agent system) */
-    stepState: StepStreamState;
+    /** Agent stream state (flat content model) */
+    agentState: AgentStreamState;
     /**
      * Flattened, filtered contents ready for rendering with the Content component.
      * Only includes visible content (user_visibility === 'visible').
@@ -1526,6 +1565,8 @@ interface UseThreadReturn {
     cancel: () => void;
     /** Function to create or set the thread */
     setThread: (thread: Thread) => void;
+    /** The Zustand store — pass to AgentStreamStoreProvider for granular subscriptions */
+    agentStreamStore: StoreApi<AgentStreamStore>;
 }
 /**
  * Hook for managing a thread and sending messages.
@@ -1584,23 +1625,23 @@ interface UseThreadReturn {
  *
  * @example
  * ```tsx
- * import { useThread, Content } from '@erdoai/ui';
+ * import { useThread, Content, AgentStreamStoreProvider } from '@erdoai/ui';
  *
  * function MyChat() {
- *   const { isStreaming, streamingContents, sendMessage } = useThread({
+ *   const { isStreaming, streamingContents, sendMessage, agentStreamStore } = useThread({
  *     botKey: 'org.my-bot',
  *     onFinish: (finalContents) => console.log('Message complete with', finalContents.length, 'items'),
  *   });
  *
  *   return (
- *     <div>
+ *     <AgentStreamStoreProvider store={agentStreamStore}>
  *       {streamingContents.map((content, idx) => (
  *         <Content key={content.id || idx} content={content} />
  *       ))}
  *       <button onClick={() => sendMessage('Hello')} disabled={isStreaming}>
  *         Send
  *       </button>
- *     </div>
+ *     </AgentStreamStoreProvider>
  *   );
  * }
  * ```
@@ -1811,54 +1852,71 @@ declare function handleSSEEvent(eventType: string, event: SSEEventData, path: st
 declare function handleIncrementalMixedJsonParsing(outputContent: MessageContent | OutputContent, content: string): MessageContent | OutputContent;
 
 /**
- * Agent Event Handler
+ * Agent Event Handler (flat content model)
  *
- * Handles the agent system's step SSE events.
- * Each piece of visible work (agent, LLM call, tool call) gets a UUID.
- * The SDK maintains a flat Map<stepID, Step> and produces ContentItems for rendering.
+ * Replaces the nested step content model (handle-step-events.ts) with a flat model:
+ * - Steps are structural nodes (no content on them)
+ * - Content is always message content with optional stepId for location
+ * - No step_content event type
+ * - One tree, one rendering pass
  */
 
+/** Create initial empty agent state. */
+declare function createInitialAgentState(): AgentStreamState;
 /**
- * Check if an SSE event is a step event type.
- */
-declare function isStepEvent(event: SSEEvent): boolean;
-/**
- * Creates initial state for step-based streaming.
+ * Handle an agent SSE event and return updated state.
+ * Pure function — does not mutate input state.
  */
-declare function createInitialStepState(): StepStreamState;
+declare function handleAgentEvent(state: AgentStreamState, event: SSEEvent): AgentStreamState;
+declare function completeStreamingContent(state: AgentStreamState): AgentStreamState;
 /**
- * Handles a step SSE event and returns updated state.
- * Pure function - does not mutate input state.
+ * Returns true when the agent state contains assistant content that will
+ * visibly render (actual text/JSON from the assistant, not structural steps).
+ *
+ * Specifically checks for content nodes with:
+ * - Non-empty text (from message content delta)
+ * - Parsed JSON chunks (incremental JSON streaming)
+ * - A final result (from message content result)
+ *
+ * Excludes tool_invocation / tool_result / agent_invocation content types
+ * which are structural, not user-visible assistant output.
+ *
+ * Does NOT count step tree rows (tool_call, agent steps) — those are structural
+ * indicators, not assistant content. This ensures typing stays visible until the
+ * first actual text/chart/output token arrives.
+ *
+ * Use this to gate typing indicators: keep typing visible until this returns true.
  */
-declare function handleStepEvent(state: StepStreamState, event: SSEEvent): StepStreamState;
+declare function hasRenderableAssistantOutput(state: AgentStreamState): boolean;
 /**
- * Converts step state into ContentItem[] for rendering.
+ * Convert AgentStreamState into ContentItem[] for rendering.
  *
- * Walks the step tree depth-first from the root agent. Renders:
- * - Text from root agent's LLM calls → markdown
- * - Regular tool results → tool_invocation / tool_result
- * - Sub-agent tool_calls → tool_invocation while running, tool_result when done
- *   (sub-agent internals are NOT shown — AgentProgress handles progress labels)
+ * Single forward pass over state.nodes in event-arrival order. Renders
+ * assistant content (text, JSON) while skipping tool artifacts (rendered
+ * exclusively by StepTreeContent's buildStepTree).
  */
-declare function stepStateToContents(state: StepStreamState): ContentItem[];
+declare function agentStateToContents(state: AgentStreamState): ContentItem[];
 
-interface AgentMessageBridgeState {
-    assistantMessageID: string | null;
-}
-declare function createInitialAgentMessageBridgeState(): AgentMessageBridgeState;
-declare function applyStreamingEventToMessageState(params: {
-    useAgentEvents: boolean;
-    eventType: string;
-    eventData: SSEEventData;
-    path: string[];
-    threadID: string;
-    bridgeState: AgentMessageBridgeState;
-    stepState: StepStreamState;
-    activeMessagesByID: Record<string, MessageWithContents>;
-}): {
-    bridgeState: AgentMessageBridgeState;
-    stepState: StepStreamState;
-    stateModified: boolean;
-};
+interface AgentStreamStoreProviderProps {
+    store: StoreApi<AgentStreamStore>;
+    children: ReactNode;
+}
+declare function AgentStreamStoreProvider({ store, children }: AgentStreamStoreProviderProps): react_jsx_runtime.JSX.Element;
+/**
+ * Get the raw store API for imperative use (e.g., in event handlers).
+ * Returns null if no provider is present (backward compat).
+ */
+declare function useAgentStreamStoreOptional(): StoreApi<AgentStreamStore> | null;
+/**
+ * Subscribe to a specific invocation's sliced agent state.
+ * Only re-renders when that invocation's state reference changes.
+ * Returns undefined if the invocation is not yet registered or no provider.
+ */
+declare function useInvocationState(invocationId: string | undefined): AgentStreamState | undefined;
+/**
+ * Subscribe to the full global agent state.
+ * Re-renders on every event — use sparingly.
+ */
+declare function useAgentStreamGlobalState(): AgentStreamState;
 
-export { type AgentMessageBridgeState, BarChart, type BaseChartProps, type BotInvocation, BotInvocationContent, type BotInvocationContentData, type BotInvocationContentProps, type BotInvocationData, type BotInvocationEventInfo, type BotInvocationStatusInfo, Chart, type ChartConfig, ChartContainer, ChartContent, type ChartContentProps, ChartLegend, ChartLegendContent, ChartNode, type ChartNodeContent, ChartStyle, ChartTooltip, ChartTooltipContent, CodeexecContent, type CodeexecContentProps, CodegenContent, type CodegenContentProps, CollapsibleCodeBlock, type CollapsibleCodeBlockProps, Content, type ContentChartConfig, type ContentChunk, type ContentComponentProps, type ContentProps, type ContentType, type DataFetcher, DatasetChart, type DatasetChartProps, type DatasetDetails, DatasetDownload, type DatasetDownloadProps, type DatasetFetchRequest, DatasetTable, type DatasetTableProps, DownloadButtonNode, type DownloadButtonNodeContent, type Entity, type EntityType, ErdoProvider, type ErdoProviderConfig, type ErdoProviderProps, ErdoUI, type ErdoUIProps, ErrorBoundary, ExecutionStatus, ExpandableOutputContent, type ExpandableOutputContentProps, FileDownloadsNode, type FileDownloadsNodeContent, type GeneratedFile, HeatmapChart, InvocationEvent, type InvocationEventProps, InvocationEvents, type InvocationEventsProps, InvocationStatus, JSONStreamParser, type JSONValue, JsonContent, type JsonContentProps, LineChart, Loader, Log, LogContent, type LogContentProps, type LogProps, MarkdownContent, type MarkdownContentProps, type Memory, MemoryContent, type MemoryContentProps, type Message, type MessageContent, type MessageStreamingState, type MessageWithContents, type NodeRendererProps, type NullString, Output, type OutputContent, type OutputProps, type OutputWithContents, PieChart, type RawApiContent, type ResolvedBotInvocation, type ResultHandler, type SSEEventData, ScatterChart, type SpinnerStatus, SqlContent, type SqlContentProps, type Status, type StatusEvent, StatusSpinner, type StatusSpinnerProps, StderrText, StdoutText, StdwarnText, type Step, StepInvocation, type StepInvocationProps, StepInvocationStatus, type StepInvocationStatusProps, StepTreeContent, type StepTreeContentProps, SuggestionsNode, type SuggestionsNodeContent, TableContent, type TableContentProps, TableNode, type TableNodeContent, TextContent, type TextContentProps, TextMarkdownNode, type TextMarkdownNodeContent, ThinkingContent, type ThinkingContentProps, ToolGroupContent, type ToolGroupContentProps, UIGenerationNodes, type UIGenerationNodesProps, type UINode, type UseDatasetContentsResult, type UseThreadOptions, type UseThreadReturn, WebParseContent, type WebParseContentProps, WebSearchContent, type WebSearchContentProps, type WrapperType, WritingLoader, applyStreamingEventToMessageState, cn, createInitialAgentMessageBridgeState, createInitialStepState, extractContentItems, formatValue, handleIncrementalMixedJsonParsing, handleSSEEvent, handleStepEvent, isJsonLike, isNullString, isParsingComplete, isParsingInProgress, isStepEvent, isWhitespaceChar, nodeComponents, normalizeContent, normalizeContents, normalizeUINodes, parseCompleteJson, parseMixedJson, parseToDate, resolveKeyFromData, stepStateToContents, toSnakeCase, unwrapNullString, unwrapNullStringOr, useChartZoom, useDatasetContents, useErdoConfig, useErdoConfigOptional, useMultipleDatasetContents, useMultipleDatasetRequests, useThread };
+export { type AgentStreamStore, AgentStreamStoreProvider, type AgentStreamStoreProviderProps, BarChart, type BaseChartProps, type BotInvocation, BotInvocationContent, type BotInvocationContentData, type BotInvocationContentProps, type BotInvocationData, type BotInvocationEventInfo, type BotInvocationStatusInfo, Chart, type ChartConfig, ChartContainer, ChartContent, type ChartContentProps, ChartLegend, ChartLegendContent, ChartNode, type ChartNodeContent, ChartStyle, ChartTooltip, ChartTooltipContent, CodeexecContent, type CodeexecContentProps, CodegenContent, type CodegenContentProps, CollapsibleCodeBlock, type CollapsibleCodeBlockProps, Content, type ContentChartConfig, type ContentChunk, type ContentComponentProps, type ContentProps, type ContentType, type DataFetcher, DatasetChart, type DatasetChartProps, type DatasetDetails, DatasetDownload, type DatasetDownloadProps, type DatasetFetchRequest, DatasetTable, type DatasetTableProps, DownloadButtonNode, type DownloadButtonNodeContent, type Entity, type EntityType, ErdoProvider, type ErdoProviderConfig, type ErdoProviderProps, ErdoUI, type ErdoUIProps, ErrorBoundary, ExecutionStatus, ExpandableOutputContent, type ExpandableOutputContentProps, FileDownloadsNode, type FileDownloadsNodeContent, type GeneratedFile, HeatmapChart, InvocationEvent, type InvocationEventProps, InvocationEvents, type InvocationEventsProps, InvocationStatus, JSONStreamParser, type JSONValue, JsonContent, type JsonContentProps, LineChart, Loader, Log, LogContent, type LogContentProps, type LogProps, MarkdownContent, type MarkdownContentProps, type Memory, MemoryContent, type MemoryContentProps, type Message, type MessageContent, type MessageStreamingState, type MessageWithContents, type NodeRendererProps, type NullString, Output, type OutputContent, type OutputProps, type OutputWithContents, PieChart, type RawApiContent, type ResolvedBotInvocation, type ResultHandler, type SSEEventData, ScatterChart, type SpinnerStatus, SqlContent, type SqlContentProps, type Status, type StatusEvent, StatusSpinner, type StatusSpinnerProps, StderrText, StdoutText, StdwarnText, type Step, StepInvocation, type StepInvocationProps, StepInvocationStatus, type StepInvocationStatusProps, StepTreeContent, type StepTreeContentProps, SuggestionsNode, type SuggestionsNodeContent, TableContent, type TableContentProps, TableNode, type TableNodeContent, TextContent, type TextContentProps, TextMarkdownNode, type TextMarkdownNodeContent, ThinkingContent, type ThinkingContentProps, ToolGroupContent, type ToolGroupContentProps, UIGenerationNodes, type UIGenerationNodesProps, type UINode, type UseDatasetContentsResult, type UseThreadOptions, type UseThreadReturn, WebParseContent, type WebParseContentProps, WebSearchContent, type WebSearchContentProps, type WrapperType, WritingLoader, agentStateToContents, cn, completeStreamingContent, createAgentStreamStore, createInitialAgentState, extractContentItems, formatValue, handleAgentEvent, handleIncrementalMixedJsonParsing, handleSSEEvent, hasRenderableAssistantOutput, isJsonLike, isNullString, isParsingComplete, isParsingInProgress, isWhitespaceChar, nodeComponents, normalizeContent, normalizeContents, normalizeUIGenerationChunks, normalizeUINodes, parseCompleteJson, parseMixedJson, parseToDate, resolveKeyFromData, sliceAgentStateToRoot, toSnakeCase, unwrapNullString, unwrapNullStringOr, useAgentStreamGlobalState, useAgentStreamStoreOptional, useChartZoom, useDatasetContents, useErdoConfig, useErdoConfigOptional, useInvocationState, useMultipleDatasetContents, useMultipleDatasetRequests, useThread };

package/dist/index.d.ts

@@ -1,9 +1,10 @@
 import * as react_jsx_runtime from 'react/jsx-runtime';
 import * as React$1 from 'react';
-import React__default from 'react';
-import { ErdoClient, ContentItem, InvokeResult, LogEntry, ToolGroup, InvocationStatus as InvocationStatus$1, StepStreamState, SSEEvent, Thread } from '@erdoai/types';
+import React__default, { ReactNode } from 'react';
+import { ErdoClient, ContentItem, InvokeResult, LogEntry, ToolGroup, InvocationStatus as InvocationStatus$1, AgentStreamState, SSEEvent, Thread } from '@erdoai/types';
 export { ContentItem } from '@erdoai/types';
 import * as RechartsPrimitive from 'recharts';
+import { StoreApi } from 'zustand';
 import { ClassValue } from 'clsx';
 
 /**
@@ -131,9 +132,10 @@ interface ErdoProviderConfig {
      */
     onSuggestionClick?: (suggestion: string) => void;
     /**
-     * Enable the new agent event system.
-     * When true, uses simple flat events (agent_started, text_delta, tool_started, etc.)
-     * When false (default), uses legacy path-based event routing.
+     * Enable the agent event system (flat content model).
+     * When true, routes step_started/step_finished and message content events
+     * through handleAgentEvent into AgentStreamState.
+     * When false (default), uses legacy path-based event routing (bot service).
      */
     useAgentEvents?: boolean;
 }
@@ -842,6 +844,7 @@ interface ContentProps {
     components?: Record<string, React__default.ComponentType<any>>;
 }
 declare function Content({ content, className, components }: ContentProps): react_jsx_runtime.JSX.Element | null;
+declare function normalizeUIGenerationChunks(value: unknown): ContentChunk[] | null;
 
 interface ErdoUIProps {
     /** Result from an Erdo invocation */
@@ -1005,8 +1008,8 @@ interface ThinkingContentProps {
     className?: string;
 }
 /**
- * Renders Claude's thinking/reasoning content in a collapsible panel.
- * Handles both regular thinking and redacted (encrypted) thinking.
+ * Renders Claude's thinking/reasoning content as subtle inline text.
+ * No container/chrome; supports redacted (encrypted) thinking blocks.
  */
 declare function ThinkingContent({ item, className }: ThinkingContentProps): react_jsx_runtime.JSX.Element;
 
@@ -1085,7 +1088,7 @@ interface CollapsibleCodeBlockProps {
 /**
  * A collapsible code block with syntax highlighting.
  */
-declare function CollapsibleCodeBlock({ language, value, defaultOpen, title, closeTitle, className, theme, }: CollapsibleCodeBlockProps): react_jsx_runtime.JSX.Element;
+declare const CollapsibleCodeBlock: React$1.NamedExoticComponent<CollapsibleCodeBlockProps>;
 
 interface SqlContentProps {
     item: ContentItem;
@@ -1102,8 +1105,7 @@ interface CodegenContentProps {
 }
 /**
  * Renders code generation content with step-by-step explanation, code, and warnings.
- *
- * Expects parsedJson to be populated upstream (via normalizeContent or SSE handler).
+ * Prefers canonical `content` (parsed object); falls back to legacy `parsedJson` (DB string path).
  */
 declare function CodegenContent({ item, className }: CodegenContentProps): react_jsx_runtime.JSX.Element | null;
 
@@ -1113,6 +1115,7 @@ interface CodeexecContentProps {
 }
 /**
  * Renders code execution output (stdout, stderr, warnings).
+ * Prefers canonical `content`; falls back to legacy `parsedJson`.
  */
 declare function CodeexecContent({ item, className }: CodeexecContentProps): react_jsx_runtime.JSX.Element | null;
 
@@ -1282,8 +1285,8 @@ interface MemoryContentProps {
 declare function MemoryContent({ content, className }: MemoryContentProps): react_jsx_runtime.JSX.Element;
 
 interface StepTreeContentProps {
-    /** The step stream state containing all steps */
-    stepState: StepStreamState;
+    /** The agent stream state containing all steps and content */
+    agentState: AgentStreamState;
     /** Optional Content component for rendering ui_generation/codegen */
     ContentComponent?: React__default.ComponentType<{
         content: ContentItem;
@@ -1296,7 +1299,7 @@ interface StepTreeContentProps {
  *
  * Used for both streaming and reload paths to ensure consistent rendering.
  */
-declare function StepTreeContent({ stepState, ContentComponent, className }: StepTreeContentProps): react_jsx_runtime.JSX.Element | null;
+declare function StepTreeContent({ agentState, ContentComponent, className }: StepTreeContentProps): react_jsx_runtime.JSX.Element | null;
 
 /** Node structure from UI generation content */
 interface UINode {
@@ -1492,6 +1495,42 @@ declare function useMultipleDatasetRequests(requests: DatasetFetchRequest[], inv
  */
 declare function useMultipleDatasetContents(datasetSlugs: string[], invocationId: string, sqlQueries?: Record<string, string>, resourceKeys?: Record<string, string>): UseDatasetContentsResult[];
 
+/**
+ * Zustand store for agent streaming state.
+ *
+ * Holds the global AgentStreamState and per-invocation sliced states.
+ * Components subscribe to individual invocation slices via useInvocationState(),
+ * so only the affected component re-renders when an event arrives.
+ *
+ * The store calls handleAgentEvent internally — consumers just push raw SSE events.
+ */
+
+declare function sliceAgentStateToRoot(state: AgentStreamState, rootStepID: string): AgentStreamState;
+interface AgentStreamStore {
+    /** The single merged state from all events */
+    globalState: AgentStreamState;
+    /** Per-invocation sliced states. Key: invocation_id */
+    invocationStates: Record<string, AgentStreamState>;
+    /** Maps invocation_id → root step_id */
+    invocationRootSteps: Record<string, string>;
+    /** Queue for invocations awaiting their root step_started */
+    pendingInvocationQueue: string[];
+    /**
+     * Push a raw SSE event into the store.
+     * Calls handleAgentEvent internally and re-slices affected invocations.
+     * @param event The raw SSE event
+     * @param reconcile Optional post-processing (e.g., reconcileThinkingContentState)
+     */
+    pushEvent: (event: SSEEvent, reconcile?: (state: AgentStreamState) => AgentStreamState) => void;
+    /** Register an invocation_id (from agent_invocation event). Queues it for root mapping. */
+    registerInvocation: (invocationId: string) => void;
+    /** Map an invocation_id to its root step_id (from step_started with type: 'agent'). */
+    mapInvocationRoot: (invocationId: string, rootStepId: string) => void;
+    /** Reset the store (new session). */
+    reset: () => void;
+}
+declare function createAgentStreamStore(): StoreApi<AgentStreamStore>;
+
 interface UseThreadOptions {
     /** Initial thread ID to use (if resuming an existing thread) */
     threadId?: string;
@@ -1513,8 +1552,8 @@ interface UseThreadReturn {
     error: Error | null;
     /** Active messages being streamed (with parsedJson populated) */
     activeMessages: MessageWithContents[];
-    /** Step stream state (for agent system) */
-    stepState: StepStreamState;
+    /** Agent stream state (flat content model) */
+    agentState: AgentStreamState;
     /**
      * Flattened, filtered contents ready for rendering with the Content component.
      * Only includes visible content (user_visibility === 'visible').
@@ -1526,6 +1565,8 @@ interface UseThreadReturn {
     cancel: () => void;
     /** Function to create or set the thread */
     setThread: (thread: Thread) => void;
+    /** The Zustand store — pass to AgentStreamStoreProvider for granular subscriptions */
+    agentStreamStore: StoreApi<AgentStreamStore>;
 }
 /**
  * Hook for managing a thread and sending messages.
@@ -1584,23 +1625,23 @@ interface UseThreadReturn {
  *
  * @example
  * ```tsx
- * import { useThread, Content } from '@erdoai/ui';
+ * import { useThread, Content, AgentStreamStoreProvider } from '@erdoai/ui';
  *
  * function MyChat() {
- *   const { isStreaming, streamingContents, sendMessage } = useThread({
+ *   const { isStreaming, streamingContents, sendMessage, agentStreamStore } = useThread({
  *     botKey: 'org.my-bot',
  *     onFinish: (finalContents) => console.log('Message complete with', finalContents.length, 'items'),
  *   });
  *
  *   return (
- *     <div>
+ *     <AgentStreamStoreProvider store={agentStreamStore}>
  *       {streamingContents.map((content, idx) => (
  *         <Content key={content.id || idx} content={content} />
  *       ))}
  *       <button onClick={() => sendMessage('Hello')} disabled={isStreaming}>
  *         Send
  *       </button>
- *     </div>
+ *     </AgentStreamStoreProvider>
  *   );
  * }
  * ```
@@ -1811,54 +1852,71 @@ declare function handleSSEEvent(eventType: string, event: SSEEventData, path: st
 declare function handleIncrementalMixedJsonParsing(outputContent: MessageContent | OutputContent, content: string): MessageContent | OutputContent;
 
 /**
- * Agent Event Handler
+ * Agent Event Handler (flat content model)
  *
- * Handles the agent system's step SSE events.
- * Each piece of visible work (agent, LLM call, tool call) gets a UUID.
- * The SDK maintains a flat Map<stepID, Step> and produces ContentItems for rendering.
+ * Replaces the nested step content model (handle-step-events.ts) with a flat model:
+ * - Steps are structural nodes (no content on them)
+ * - Content is always message content with optional stepId for location
+ * - No step_content event type
+ * - One tree, one rendering pass
  */
 
+/** Create initial empty agent state. */
+declare function createInitialAgentState(): AgentStreamState;
 /**
- * Check if an SSE event is a step event type.
- */
-declare function isStepEvent(event: SSEEvent): boolean;
-/**
- * Creates initial state for step-based streaming.
+ * Handle an agent SSE event and return updated state.
+ * Pure function — does not mutate input state.
  */
-declare function createInitialStepState(): StepStreamState;
+declare function handleAgentEvent(state: AgentStreamState, event: SSEEvent): AgentStreamState;
+declare function completeStreamingContent(state: AgentStreamState): AgentStreamState;
 /**
- * Handles a step SSE event and returns updated state.
- * Pure function - does not mutate input state.
+ * Returns true when the agent state contains assistant content that will
+ * visibly render (actual text/JSON from the assistant, not structural steps).
+ *
+ * Specifically checks for content nodes with:
+ * - Non-empty text (from message content delta)
+ * - Parsed JSON chunks (incremental JSON streaming)
+ * - A final result (from message content result)
+ *
+ * Excludes tool_invocation / tool_result / agent_invocation content types
+ * which are structural, not user-visible assistant output.
+ *
+ * Does NOT count step tree rows (tool_call, agent steps) — those are structural
+ * indicators, not assistant content. This ensures typing stays visible until the
+ * first actual text/chart/output token arrives.
+ *
+ * Use this to gate typing indicators: keep typing visible until this returns true.
  */
-declare function handleStepEvent(state: StepStreamState, event: SSEEvent): StepStreamState;
+declare function hasRenderableAssistantOutput(state: AgentStreamState): boolean;
 /**
- * Converts step state into ContentItem[] for rendering.
+ * Convert AgentStreamState into ContentItem[] for rendering.
  *
- * Walks the step tree depth-first from the root agent. Renders:
- * - Text from root agent's LLM calls → markdown
- * - Regular tool results → tool_invocation / tool_result
- * - Sub-agent tool_calls → tool_invocation while running, tool_result when done
- *   (sub-agent internals are NOT shown — AgentProgress handles progress labels)
+ * Single forward pass over state.nodes in event-arrival order. Renders
+ * assistant content (text, JSON) while skipping tool artifacts (rendered
+ * exclusively by StepTreeContent's buildStepTree).
  */
-declare function stepStateToContents(state: StepStreamState): ContentItem[];
+declare function agentStateToContents(state: AgentStreamState): ContentItem[];
 
-interface AgentMessageBridgeState {
-    assistantMessageID: string | null;
-}
-declare function createInitialAgentMessageBridgeState(): AgentMessageBridgeState;
-declare function applyStreamingEventToMessageState(params: {
-    useAgentEvents: boolean;
-    eventType: string;
-    eventData: SSEEventData;
-    path: string[];
-    threadID: string;
-    bridgeState: AgentMessageBridgeState;
-    stepState: StepStreamState;
-    activeMessagesByID: Record<string, MessageWithContents>;
-}): {
-    bridgeState: AgentMessageBridgeState;
-    stepState: StepStreamState;
-    stateModified: boolean;
-};
+interface AgentStreamStoreProviderProps {
+    store: StoreApi<AgentStreamStore>;
+    children: ReactNode;
+}
+declare function AgentStreamStoreProvider({ store, children }: AgentStreamStoreProviderProps): react_jsx_runtime.JSX.Element;
+/**
+ * Get the raw store API for imperative use (e.g., in event handlers).
+ * Returns null if no provider is present (backward compat).
+ */
+declare function useAgentStreamStoreOptional(): StoreApi<AgentStreamStore> | null;
+/**
+ * Subscribe to a specific invocation's sliced agent state.
+ * Only re-renders when that invocation's state reference changes.
+ * Returns undefined if the invocation is not yet registered or no provider.
+ */
+declare function useInvocationState(invocationId: string | undefined): AgentStreamState | undefined;
+/**
+ * Subscribe to the full global agent state.
+ * Re-renders on every event — use sparingly.
+ */
+declare function useAgentStreamGlobalState(): AgentStreamState;
 
-export { type AgentMessageBridgeState, BarChart, type BaseChartProps, type BotInvocation, BotInvocationContent, type BotInvocationContentData, type BotInvocationContentProps, type BotInvocationData, type BotInvocationEventInfo, type BotInvocationStatusInfo, Chart, type ChartConfig, ChartContainer, ChartContent, type ChartContentProps, ChartLegend, ChartLegendContent, ChartNode, type ChartNodeContent, ChartStyle, ChartTooltip, ChartTooltipContent, CodeexecContent, type CodeexecContentProps, CodegenContent, type CodegenContentProps, CollapsibleCodeBlock, type CollapsibleCodeBlockProps, Content, type ContentChartConfig, type ContentChunk, type ContentComponentProps, type ContentProps, type ContentType, type DataFetcher, DatasetChart, type DatasetChartProps, type DatasetDetails, DatasetDownload, type DatasetDownloadProps, type DatasetFetchRequest, DatasetTable, type DatasetTableProps, DownloadButtonNode, type DownloadButtonNodeContent, type Entity, type EntityType, ErdoProvider, type ErdoProviderConfig, type ErdoProviderProps, ErdoUI, type ErdoUIProps, ErrorBoundary, ExecutionStatus, ExpandableOutputContent, type ExpandableOutputContentProps, FileDownloadsNode, type FileDownloadsNodeContent, type GeneratedFile, HeatmapChart, InvocationEvent, type InvocationEventProps, InvocationEvents, type InvocationEventsProps, InvocationStatus, JSONStreamParser, type JSONValue, JsonContent, type JsonContentProps, LineChart, Loader, Log, LogContent, type LogContentProps, type LogProps, MarkdownContent, type MarkdownContentProps, type Memory, MemoryContent, type MemoryContentProps, type Message, type MessageContent, type MessageStreamingState, type MessageWithContents, type NodeRendererProps, type NullString, Output, type OutputContent, type OutputProps, type OutputWithContents, PieChart, type RawApiContent, type ResolvedBotInvocation, type ResultHandler, type SSEEventData, ScatterChart, type SpinnerStatus, SqlContent, type SqlContentProps, type Status, type StatusEvent, StatusSpinner, type StatusSpinnerProps, StderrText, StdoutText, StdwarnText, type Step, StepInvocation, type StepInvocationProps, StepInvocationStatus, type StepInvocationStatusProps, StepTreeContent, type StepTreeContentProps, SuggestionsNode, type SuggestionsNodeContent, TableContent, type TableContentProps, TableNode, type TableNodeContent, TextContent, type TextContentProps, TextMarkdownNode, type TextMarkdownNodeContent, ThinkingContent, type ThinkingContentProps, ToolGroupContent, type ToolGroupContentProps, UIGenerationNodes, type UIGenerationNodesProps, type UINode, type UseDatasetContentsResult, type UseThreadOptions, type UseThreadReturn, WebParseContent, type WebParseContentProps, WebSearchContent, type WebSearchContentProps, type WrapperType, WritingLoader, applyStreamingEventToMessageState, cn, createInitialAgentMessageBridgeState, createInitialStepState, extractContentItems, formatValue, handleIncrementalMixedJsonParsing, handleSSEEvent, handleStepEvent, isJsonLike, isNullString, isParsingComplete, isParsingInProgress, isStepEvent, isWhitespaceChar, nodeComponents, normalizeContent, normalizeContents, normalizeUINodes, parseCompleteJson, parseMixedJson, parseToDate, resolveKeyFromData, stepStateToContents, toSnakeCase, unwrapNullString, unwrapNullStringOr, useChartZoom, useDatasetContents, useErdoConfig, useErdoConfigOptional, useMultipleDatasetContents, useMultipleDatasetRequests, useThread };
+export { type AgentStreamStore, AgentStreamStoreProvider, type AgentStreamStoreProviderProps, BarChart, type BaseChartProps, type BotInvocation, BotInvocationContent, type BotInvocationContentData, type BotInvocationContentProps, type BotInvocationData, type BotInvocationEventInfo, type BotInvocationStatusInfo, Chart, type ChartConfig, ChartContainer, ChartContent, type ChartContentProps, ChartLegend, ChartLegendContent, ChartNode, type ChartNodeContent, ChartStyle, ChartTooltip, ChartTooltipContent, CodeexecContent, type CodeexecContentProps, CodegenContent, type CodegenContentProps, CollapsibleCodeBlock, type CollapsibleCodeBlockProps, Content, type ContentChartConfig, type ContentChunk, type ContentComponentProps, type ContentProps, type ContentType, type DataFetcher, DatasetChart, type DatasetChartProps, type DatasetDetails, DatasetDownload, type DatasetDownloadProps, type DatasetFetchRequest, DatasetTable, type DatasetTableProps, DownloadButtonNode, type DownloadButtonNodeContent, type Entity, type EntityType, ErdoProvider, type ErdoProviderConfig, type ErdoProviderProps, ErdoUI, type ErdoUIProps, ErrorBoundary, ExecutionStatus, ExpandableOutputContent, type ExpandableOutputContentProps, FileDownloadsNode, type FileDownloadsNodeContent, type GeneratedFile, HeatmapChart, InvocationEvent, type InvocationEventProps, InvocationEvents, type InvocationEventsProps, InvocationStatus, JSONStreamParser, type JSONValue, JsonContent, type JsonContentProps, LineChart, Loader, Log, LogContent, type LogContentProps, type LogProps, MarkdownContent, type MarkdownContentProps, type Memory, MemoryContent, type MemoryContentProps, type Message, type MessageContent, type MessageStreamingState, type MessageWithContents, type NodeRendererProps, type NullString, Output, type OutputContent, type OutputProps, type OutputWithContents, PieChart, type RawApiContent, type ResolvedBotInvocation, type ResultHandler, type SSEEventData, ScatterChart, type SpinnerStatus, SqlContent, type SqlContentProps, type Status, type StatusEvent, StatusSpinner, type StatusSpinnerProps, StderrText, StdoutText, StdwarnText, type Step, StepInvocation, type StepInvocationProps, StepInvocationStatus, type StepInvocationStatusProps, StepTreeContent, type StepTreeContentProps, SuggestionsNode, type SuggestionsNodeContent, TableContent, type TableContentProps, TableNode, type TableNodeContent, TextContent, type TextContentProps, TextMarkdownNode, type TextMarkdownNodeContent, ThinkingContent, type ThinkingContentProps, ToolGroupContent, type ToolGroupContentProps, UIGenerationNodes, type UIGenerationNodesProps, type UINode, type UseDatasetContentsResult, type UseThreadOptions, type UseThreadReturn, WebParseContent, type WebParseContentProps, WebSearchContent, type WebSearchContentProps, type WrapperType, WritingLoader, agentStateToContents, cn, completeStreamingContent, createAgentStreamStore, createInitialAgentState, extractContentItems, formatValue, handleAgentEvent, handleIncrementalMixedJsonParsing, handleSSEEvent, hasRenderableAssistantOutput, isJsonLike, isNullString, isParsingComplete, isParsingInProgress, isWhitespaceChar, nodeComponents, normalizeContent, normalizeContents, normalizeUIGenerationChunks, normalizeUINodes, parseCompleteJson, parseMixedJson, parseToDate, resolveKeyFromData, sliceAgentStateToRoot, toSnakeCase, unwrapNullString, unwrapNullStringOr, useAgentStreamGlobalState, useAgentStreamStoreOptional, useChartZoom, useDatasetContents, useErdoConfig, useErdoConfigOptional, useInvocationState, useMultipleDatasetContents, useMultipleDatasetRequests, useThread };