@erdoai/ui 0.1.67 → 0.1.69

package/dist/index.d.ts

@@ -1,7 +1,7 @@
 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, SSEEvent, Thread } from '@erdoai/types';
+import { ErdoClient, ContentItem, InvokeResult, LogEntry, ToolGroup, InvocationStatus as InvocationStatus$1, StepStreamState, SSEEvent, Thread } from '@erdoai/types';
 export { ContentItem } from '@erdoai/types';
 import * as RechartsPrimitive from 'recharts';
 import { ClassValue } from 'clsx';
@@ -130,6 +130,12 @@ 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.
+     */
+    useAgentEvents?: boolean;
 }
 interface ErdoProviderProps {
     children: React__default.ReactNode;
@@ -462,7 +468,7 @@ declare function isParsingInProgress(value: any): boolean;
  * They represent the structure of entities that can be built from SSE events.
  */
 
-type ContentType = 'text' | 'json' | 'xml' | 'thinking' | 'redacted_thinking' | 'bot_invocation' | 'bot_invocation_result' | 'tool_invocation' | 'tool_result';
+type ContentType = 'text' | 'json' | 'xml' | 'thinking' | 'redacted_thinking' | 'agent_invocation' | 'bot_invocation' | 'bot_invocation_result' | 'tool_invocation' | 'tool_result';
 declare function isJsonLike(contentType: ContentType, uiContentType: string): boolean;
 type EntityType = 'message' | 'message_content' | 'invocation' | 'output' | 'output_content' | 'step' | 'result_handler' | 'status' | 'log';
 type MessageStreamingState = 'in_progress' | 'completed';
@@ -901,7 +907,14 @@ interface JsonContentProps {
     content: unknown;
     className?: string;
 }
-declare function JsonContent({ content, className }: JsonContentProps): react_jsx_runtime.JSX.Element | null;
+/**
+ * Renders JSON content.
+ *
+ * Content should be normalized upstream (via normalizeContent or SSE handler).
+ * If content is a styled message object (with `message` and optional `style`),
+ * it renders as styled text. Otherwise, returns null.
+ */
+declare function JsonContent({ content }: JsonContentProps): react_jsx_runtime.JSX.Element | null;
 
 interface MarkdownContentProps {
     /** The markdown content to render */
@@ -1007,6 +1020,8 @@ interface LogProps {
 declare function Log({ log }: LogProps): react_jsx_runtime.JSX.Element;
 /**
  * Renders log entries with level-based styling (info, error, requires_info).
+ *
+ * Content should be normalized upstream (via normalizeContent or SSE handler).
  */
 declare function LogContent({ item, className }: LogContentProps): react_jsx_runtime.JSX.Element;
 
@@ -1087,6 +1102,8 @@ 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).
  */
 declare function CodegenContent({ item, className }: CodegenContentProps): react_jsx_runtime.JSX.Element | null;
 
@@ -1113,6 +1130,7 @@ interface ToolGroupContentProps {
 }
 /**
  * Renders tool invocations and their results in a collapsible format.
+ * Uses StatusSpinner for loading/finished/failed states — no hardcoded icons.
  */
 declare function ToolGroupContent({ toolGroup, ContentComponent, className, fallbackClassName, }: ToolGroupContentProps): react_jsx_runtime.JSX.Element;
 
@@ -1190,6 +1208,10 @@ interface BotInvocationContentProps {
  *
  * This component is designed to receive already-processed invocation data.
  * Data fetching should be handled by the consuming application.
+ *
+ * When `content.transparent` is true, the wrapper (status, collapsible trigger) is hidden
+ * and only the events are rendered directly. This is used for root invocations where
+ * the container should be invisible but the step outputs should still be visible.
  */
 declare function BotInvocationContent({ content, botInvocation, isLoading, defaultOpen, adminLinkHref, adminLinkText, showAdminLink, onAdminClick, onExpand, EventsComponent, ContentComponent: _ContentComponent, className, }: BotInvocationContentProps): react_jsx_runtime.JSX.Element | null;
 
@@ -1259,6 +1281,23 @@ interface MemoryContentProps {
  */
 declare function MemoryContent({ content, className }: MemoryContentProps): react_jsx_runtime.JSX.Element;
 
+interface StepTreeContentProps {
+    /** The step stream state containing all steps */
+    stepState: StepStreamState;
+    /** Optional Content component for rendering ui_generation/codegen */
+    ContentComponent?: React__default.ComponentType<{
+        content: ContentItem;
+        className?: string;
+    }>;
+    className?: string;
+}
+/**
+ * Renders a step hierarchy with proper SDK styling.
+ *
+ * Used for both streaming and reload paths to ensure consistent rendering.
+ */
+declare function StepTreeContent({ stepState, ContentComponent, className }: StepTreeContentProps): react_jsx_runtime.JSX.Element | null;
+
 /** Node structure from UI generation content */
 interface UINode {
     _type: string;
@@ -1426,77 +1465,30 @@ interface UseDatasetContentsResult {
     error: Error | null;
     refetch: () => void;
 }
+/**
+ * A request to fetch dataset contents with optional query and resource key.
+ */
+interface DatasetFetchRequest {
+    datasetSlug: string;
+    sqlQuery?: string;
+    resourceKey?: string;
+}
 /**
  * Hook to fetch contents of a single dataset.
- * Uses the DataFetcher from ErdoProvider if available, otherwise falls back to REST API.
- *
- * @param datasetSlug - The slug/identifier of the dataset
- * @param invocationId - The invocation ID context
- * @param sqlQuery - Optional SQL query to run against the dataset
- * @returns Object with data, isLoading, error, and refetch function
- *
- * @example
- * ```tsx
- * const { data, isLoading, error } = useDatasetContents('my-dataset', invocationId);
- * if (isLoading) return <Loading />;
- * if (error) return <Error message={error.message} />;
- * return <Chart data={data} />;
- * ```
- *
- * @example With SQL query
- * ```tsx
- * const { data } = useDatasetContents(
- *   'my-database',
- *   invocationId,
- *   'SELECT * FROM users ORDER BY created_at DESC LIMIT 10'
- * );
- * ```
- *
- * @example With resource key (for Excel files with headers not on row 1)
- * ```tsx
- * const { data } = useDatasetContents(
- *   'my-excel-dataset',
- *   invocationId,
- *   undefined,
- *   'abc123::Sheet1::table_1'
- * );
- * ```
  */
 declare function useDatasetContents(datasetSlug: string, invocationId: string, sqlQuery?: string, resourceKey?: string): UseDatasetContentsResult;
 /**
- * Hook to fetch contents of multiple datasets in parallel.
- * Uses the DataFetcher from ErdoProvider if available, otherwise falls back to REST API.
+ * Hook to fetch contents for multiple dataset requests in parallel.
+ * Deduplicates requests with the same slug+query+resourceKey combo.
  *
- * @param datasetSlugs - Array of dataset slugs/identifiers
+ * @param requests - Array of fetch requests (slug + optional query/resourceKey)
  * @param invocationId - The invocation ID context
- * @param sqlQueries - Optional map of dataset slug to SQL query
- * @returns Array of result objects, one per dataset
- *
- * @example
- * ```tsx
- * const results = useMultipleDatasetContents(['dataset-1', 'dataset-2'], invocationId);
- * const isLoading = results.some(r => r.isLoading);
- * const allData = results.map(r => r.data || []);
- * ```
- *
- * @example With SQL queries
- * ```tsx
- * const results = useMultipleDatasetContents(
- *   ['my-db', 'other-db'],
- *   invocationId,
- *   { 'my-db': 'SELECT * FROM users LIMIT 10' }
- * );
- * ```
- *
- * @example With resource keys
- * ```tsx
- * const results = useMultipleDatasetContents(
- *   ['dataset-1', 'dataset-2'],
- *   invocationId,
- *   undefined,
- *   { 'dataset-1': 'abc123::Sheet1::table_1' }
- * );
- * ```
+ * @returns Array of results in the same order as input requests
+ */
+declare function useMultipleDatasetRequests(requests: DatasetFetchRequest[], invocationId: string): UseDatasetContentsResult[];
+/**
+ * @deprecated Use useMultipleDatasetRequests instead for proper per-series fetching.
+ * This function only supports one query per dataset slug.
  */
 declare function useMultipleDatasetContents(datasetSlugs: string[], invocationId: string, sqlQueries?: Record<string, string>, resourceKeys?: Record<string, string>): UseDatasetContentsResult[];
 
@@ -1521,6 +1513,8 @@ interface UseThreadReturn {
     error: Error | null;
     /** Active messages being streamed (with parsedJson populated) */
     activeMessages: MessageWithContents[];
+    /** Step stream state (for agent system) */
+    stepState: StepStreamState;
     /**
      * Flattened, filtered contents ready for rendering with the Content component.
      * Only includes visible content (user_visibility === 'visible').
@@ -1684,6 +1678,52 @@ declare function toSnakeCase(str: string): string;
  */
 declare function resolveKeyFromData(dataRows: any[], configKey: string): string;
 
+/**
+ * Content Normalization Utilities
+ *
+ * These functions normalize content from API responses so it's ready for rendering.
+ * Parsing happens HERE, not in renderers.
+ *
+ * Usage:
+ * - SSE streaming: content is normalized automatically via handleSSEEventNormalized
+ * - API responses (history): call normalizeContent() before passing to <Content />
+ */
+
+/**
+ * Raw content from API responses (e.g., getThreadMessages).
+ * This is the shape before normalization.
+ */
+interface RawApiContent {
+    id: string;
+    content_type: string;
+    content: unknown;
+    created_by_invocation_id?: string;
+    user_visibility?: string;
+    ui_content_type?: string | {
+        String: string;
+        Valid: boolean;
+    };
+}
+/**
+ * Normalize a single content item from API response.
+ *
+ * For JSON-like content (ui_generation, json), parses the content string
+ * and populates `parsedJson` so renderers don't need to do any parsing.
+ *
+ * @param raw - Raw content from API
+ * @returns Normalized ContentItem ready for <Content />
+ */
+declare function normalizeContent(raw: RawApiContent): ContentItem;
+/**
+ * Normalize an array of content items from API response.
+ *
+ * Filters to only visible content and parses JSON-like content.
+ *
+ * @param rawContents - Array of raw content from API
+ * @returns Array of normalized ContentItems ready for <Content />
+ */
+declare function normalizeContents(rawContents: RawApiContent[]): ContentItem[];
+
 /**
  * Normalizes LLM-generated UI nodes to handle common variations:
  * - camelCase → snake_case keys in node content
@@ -1711,19 +1751,6 @@ declare function normalizeUINodes(nodes: unknown[]): UINodeLike[];
  * message/invocation tree structure for rendering.
  */
 
-/**
- * Handles incremental parsing of mixed JSON/text content as it streams in.
- *
- * This function is called each time new content arrives during streaming.
- * It parses the content to detect JSON blocks interspersed with text,
- * updating the outputContent's parsed_blocks array as complete JSON
- * objects are detected.
- *
- * @param outputContent - The message or output content to update
- * @param content - The full accumulated content string so far
- * @returns The updated content object with parsed_blocks populated
- */
-declare function handleIncrementalMixedJsonParsing(outputContent: MessageContent | OutputContent, content: string): MessageContent | OutputContent;
 /**
  * Handles different types of SSE events and updates the message tree accordingly.
  *
@@ -1763,4 +1790,75 @@ declare function handleIncrementalMixedJsonParsing(outputContent: MessageContent
  */
 declare function handleSSEEvent(eventType: string, event: SSEEventData, path: string[], threadID: string, activeMessagesByID: Record<string, MessageWithContents>, currentEntity?: Entity | null, parents?: Entity[], entityIds?: Record<string, string>): Entity | null;
 
-export { 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, 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 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, 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 UseThreadOptions, type UseThreadReturn, WebParseContent, type WebParseContentProps, WebSearchContent, type WebSearchContentProps, type WrapperType, WritingLoader, cn, extractContentItems, formatValue, handleIncrementalMixedJsonParsing, handleSSEEvent, isJsonLike, isNullString, isParsingComplete, isParsingInProgress, isWhitespaceChar, nodeComponents, normalizeUINodes, parseCompleteJson, parseMixedJson, parseToDate, resolveKeyFromData, toSnakeCase, unwrapNullString, unwrapNullStringOr, useChartZoom, useDatasetContents, useErdoConfig, useErdoConfigOptional, useMultipleDatasetContents, useThread };
+/**
+ * JSON Parsing Utilities for SSE Streaming
+ *
+ * Handles incremental parsing of mixed JSON/text content as it streams in.
+ */
+
+/**
+ * Handles incremental parsing of mixed JSON/text content as it streams in.
+ *
+ * This function is called each time new content arrives during streaming.
+ * It parses the content to detect JSON blocks interspersed with text,
+ * updating the outputContent's parsed_blocks array as complete JSON
+ * objects are detected.
+ *
+ * @param outputContent - The message or output content to update
+ * @param content - The full accumulated content string so far
+ * @returns The updated content object with parsed_blocks populated
+ */
+declare function handleIncrementalMixedJsonParsing(outputContent: MessageContent | OutputContent, content: string): MessageContent | OutputContent;
+
+/**
+ * Agent Event Handler
+ *
+ * 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.
+ */
+
+/**
+ * Check if an SSE event is a step event type.
+ */
+declare function isStepEvent(event: SSEEvent): boolean;
+/**
+ * Creates initial state for step-based streaming.
+ */
+declare function createInitialStepState(): StepStreamState;
+/**
+ * Handles a step SSE event and returns updated state.
+ * Pure function - does not mutate input state.
+ */
+declare function handleStepEvent(state: StepStreamState, event: SSEEvent): StepStreamState;
+/**
+ * Converts step state 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)
+ */
+declare function stepStateToContents(state: StepStreamState): 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;
+};
+
+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 };