@chaoslabs/ai-sdk 0.0.10 → 1.0.0

package/CHANGELOG.md

@@ -0,0 +1,156 @@
+# Changelog
+
+All notable changes to this project will be documented in this file.
+
+The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.1.0/),
+and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
+
+## [1.0.0] - 2026-01-30
+
+### ⚠️ Breaking Changes
+
+- **Wallet configuration**: Removed `wallet_id` from `RequestMetadata`. Use the `wallets` array instead for single or multi-chain wallet support.
+
+- **Conversation history format**: Changed `prior_messages` to `previous_messages`.
+
+- **Removed `useLegacyFormat` config option**: The SDK now exclusively uses the new API format.
+
+- **Block type discriminators aligned with backend**: All block types now use field names matching the Python backend API:
+  - `CodeBlock.code` renamed to `CodeBlock.content`
+  - `InfoBlock.message` renamed to `InfoBlock.content`
+  - `InfoBlock` now includes `error` field for error states
+  - `InteractiveBlock.prompt` renamed to `InteractiveBlock.title`
+  - `InteractiveBlock` now includes `body`, `context`, and `style` fields
+
+- **`ActionBlock` discriminator updated**: Now uses `type: "action"` in addition to `blockType: "transaction_action"` to match backend schema
+
+- **`InteractiveBlock` structure revised**:
+  - Removed `allowCustom` and `multiSelect` fields
+  - Added `body`, `context`, and `style` fields to match backend schema
+
+- **`InfoBlock` structure revised**:
+  - Removed `title` and `severity` fields
+  - Added `error` field for error messaging
+
+- **Chart block types refined**: `PieChartBlock` and `TimeseriesChartBlock` are now distinct types with specific `chartType` discriminators (`"pie"` and `"timeseries"` respectively)
+
+### Added
+
+- **Multi-wallet support**: The `RequestMetadata` interface now supports multiple wallet addresses across different chains via the new `wallets` field:
+  ```typescript
+  const metadata: RequestMetadata = {
+    user_id: 'user-123',
+    session_id: 'session-456',
+    wallets: [
+      { address: '0x...', chain: 'ethereum' },
+      { address: '0x...', chain: 'base' },
+      { address: 'So1...', chain: 'solana' },
+    ],
+  };
+  ```
+
+- **Assistant role in conversation history**: Users can now use `role: 'assistant'` directly in `InputItem` instead of the `[Assistant]:` prefix workaround. Both formats remain supported for backward compatibility.
+  ```typescript
+  const input: InputItem[] = [
+    { type: 'message', role: 'user', content: 'What is ETH price?' },
+    { type: 'message', role: 'assistant', content: 'ETH is currently $3,500.' },
+    { type: 'message', role: 'user', content: 'And BTC?' },
+  ];
+  ```
+
+- **Typed primitives**: New `Primitive` discriminated union with strongly-typed params for each primitive type:
+  - `SwapParams`, `BridgeParams`, `LendingParams`, `StakingParams`
+  - `PerpsParams`, `VaultParams`, `LiquidityParams`, `RestakingParams`
+  - Primitive type constants and category helpers
+
+- **New `WalletInfo` interface**: Represents a wallet with `address` and `chain` fields
+
+- **`DoneMessage` stream type**: Added support for `done_message` message type signaling workflow completion
+
+### Fixed
+
+- **Defensive stream helpers**: `parseAgentStatus()`, `extractAgentMessageText()`, `extractSuggestions()`, and `extractReportBlock()` now gracefully handle malformed messages instead of throwing errors
+
+### Removed
+
+- **CMS tooling**: Sanity CMS import tooling has been migrated to the docs repository
+
+### Migration Guide
+
+**Wallet configuration:**
+```typescript
+// Before (0.x)
+const metadata = { user_id: '...', session_id: '...', wallet_id: '0x...' };
+
+// After (1.0.0)
+const metadata = {
+  user_id: '...',
+  session_id: '...',
+  wallets: [{ address: '0x...', chain: 'ethereum' }],
+};
+```
+
+**Block types:**
+```typescript
+// Before (0.x)
+const codeBlock: CodeBlock = { type: 'code', code: '...', language: 'typescript' };
+const infoBlock: InfoBlock = { type: 'info', message: '...', severity: 'warning' };
+const interactive: InteractiveBlock = { type: 'interactive', prompt: '...' };
+
+// After (1.0.0)
+const codeBlock: CodeBlock = { type: 'code', content: '...', language: 'typescript' };
+const infoBlock: InfoBlock = { type: 'info', content: '...', error: null };
+const interactive: InteractiveBlock = { type: 'interactive', title: '...', style: 'options' };
+```
+
+## [0.0.11] - 2026-01-27
+
+### Added
+
+- **Typed StreamMessage**: `StreamMessage` is now a discriminated union with properly typed `content` for each message type:
+  - `agent_status_change`: `AgentStatusContent` with typed `status` field
+  - `agent_message`: `AgentMessageContent` with typed `data.message` field
+  - `report`: `ReportContent` with typed `data` (Block), `captions` (Caption[]), `references` (Reference[])
+  - `follow_up_suggestions`: `FollowUpSuggestionsContent` with typed `data.followUpQueries`
+  - `user_input`: `UserInputContent` with typed `candidates`
+
+- **New types exported**:
+  - `Caption` - Block caption with label, value, group, collapsable
+  - `Reference` - Source reference with title, url, content, hiddenUrl, textSpans, engineId
+  - `AgentStatusMessage`, `AgentTextMessage`, `ReportMessage`, `FollowUpSuggestionsMessage`, `UserInputMessage` - Individual message types for type narrowing
+
+- **Type-safe content access**: TypeScript now narrows content type automatically when checking `msg.type`:
+  ```typescript
+  if (msg.type === 'agent_status_change') {
+    console.log(msg.content.status); // TypeScript knows this is AgentStatus
+  }
+  ```
+
+### Changed
+
+- `ReportContent.data` changed from `unknown` to `Block`
+- `ReportContent.captions` changed from `unknown[]` to `Caption[]`
+- `ReportContent.references` changed from `unknown` to `Reference[]`
+- `extractReportBlock()` return type changed from `unknown | null` to `Block | null`
+- `UserInputCandidate.coingecko_coin_id` renamed to `token_id`
+
+### Removed
+
+- Staging environment support removed - SDK now only supports production (`https://ai.chaoslabs.co`)
+- `CHAOS_STAGING_URL` and `CHAOS_STAGING_API_KEY` environment variables no longer used in examples
+
+## [0.0.10] - 2026-01-27
+
+### Changed
+
+- Default `baseUrl` changed from staging to production (`https://ai.chaoslabs.co`)
+
+## [0.0.9] - 2026-01-26
+
+### Added
+
+- Initial public release
+- Core client with streaming support
+- Block types: Markdown, Table, Chart, TransactionAction, Interactive
+- Primitive constants and helpers
+- Conversation class for multi-turn interactions

package/README.md

@@ -29,7 +29,7 @@ import { Chaos, WALLET_MODEL, extractText, extractBlocks } from '@chaoslabs/ai-s
 
 const chaos = new Chaos({
   apiKey: process.env.CHAOS_API_KEY,
-  baseUrl: 'https://ai-staging.chaoslabs.co', // Optional, this is the default
+  baseUrl: 'https://ai.chaoslabs.co', // Optional, this is the default
 });
 
 const response = await chaos.chat.responses.create({
@@ -40,7 +40,10 @@ const response = await chaos.chat.responses.create({
   metadata: {
     user_id: 'user-123',
     session_id: 'session-abc',
-    wallet_id: '0x...', // Your wallet address
+    wallets: [
+      { address: '0x...', chain: 'ethereum' },
+      { address: '0x...', chain: 'base' },
+    ],
   },
 });
 
@@ -67,7 +70,11 @@ const response = await chaos.chat.responses.create({
   input: [
     { type: 'message', role: 'user', content: 'What is in my portfolio?' }
   ],
-  metadata: { user_id: 'user-123', session_id: 'session-abc', wallet_id: '0x...' },
+  metadata: {
+    user_id: 'user-123',
+    session_id: 'session-abc',
+    wallets: [{ address: '0x...', chain: 'ethereum' }],
+  },
   onStreamEvent: (msg: StreamMessage) => {
     // Called immediately as each message arrives from the server
     console.log(`[${msg.type}]`, msg.content);
@@ -98,7 +105,7 @@ new Chaos(config: ChaosConfig)
 | Option | Type | Default | Description |
 |--------|------|---------|-------------|
 | `apiKey` | `string` | Required | Your Chaos API key |
-| `baseUrl` | `string` | `https://ai-staging.chaoslabs.co` | API base URL |
+| `baseUrl` | `string` | `https://ai.chaoslabs.co` | API base URL |
 | `timeout` | `number` | `120000` | Request timeout in milliseconds |
 
 ### Request Options
@@ -110,15 +117,6 @@ new Chaos(config: ChaosConfig)
 | `metadata` | `RequestMetadata` | User, session, and wallet info |
 | `onStreamEvent` | `(msg: StreamMessage) => void` | Real-time streaming callback |
 
-> **Important:** API keys are environment-specific. A staging API key will not work with the production API, and vice versa. Make sure your `apiKey` and `baseUrl` are from the same environment.
-
-**Available Environments:**
-
-| Environment | Base URL |
-|-------------|----------|
-| Staging | `https://ai-staging.chaoslabs.co` |
-| Production | Coming soon |
-
 ### Models
 
 | Model | Description |
@@ -128,14 +126,33 @@ new Chaos(config: ChaosConfig)
 ### Request Metadata
 
 ```typescript
+interface WalletInfo {
+  address: string;    // Wallet address
+  chain: string;      // Chain name (e.g., 'ethereum', 'base', 'solana')
+}
+
 interface RequestMetadata {
-  user_id: string;      // Unique user identifier
-  session_id: string;   // Client-generated session ID for conversation continuity
-  wallet_id?: string;   // Wallet address (required for WALLET_MODEL)
+  user_id: string;        // Unique user identifier
+  session_id: string;     // Client-generated session ID for conversation continuity
+  wallets?: WalletInfo[]; // Array of wallet addresses across multiple chains
 }
 ```
 
-> **⚠️ Upcoming Change:** The `wallet_id` field currently accepts a wallet address string. In a future release, this will be replaced with a `wallet` object that includes additional wallet information (chain, connection status, etc.). We will provide migration guidance when this change is released.
+#### Multi-Chain Wallet Support
+
+The SDK supports multiple wallets across different chains. Provide an array of wallet/chain pairs:
+
+```typescript
+metadata: {
+  user_id: 'user-123',
+  session_id: 'session-abc',
+  wallets: [
+    { address: '0x742d35Cc6634C0532925a3b844Bc9e7595f2bD21', chain: 'ethereum' },
+    { address: '0x8ba1f109551bD432803012645Hac136c22C501a', chain: 'base' },
+    { address: '7xKXtg2CW87d97TXJSDpbD5jBkheTqA83TZRuJosgAsU', chain: 'solana' },
+  ],
+}
+```
 
 ### Rendering Responses
 
@@ -182,11 +199,13 @@ The SDK returns structured blocks for rich content:
 | Block Type | Description |
 |------------|-------------|
 | `table` | Tabular data (portfolio holdings, positions) |
-| `pie_chart` | Distribution charts (allocation breakdown) |
-| `timeseries` | Historical data charts |
-| `transaction_action` | DeFi operations with primitives and risks |
-| `interactive_card` | User prompts and confirmations |
+| `chart` (pie) | Distribution charts (allocation breakdown) |
+| `chart` (timeseries) | Historical data charts |
+| `action` | DeFi operations with primitives and risks |
+| `interactive` | User prompts and confirmations |
 | `markdown` | Rich text content |
+| `info` | Informational messages |
+| `code` | Code blocks with syntax highlighting |
 
 ### Helper Functions
 
@@ -203,10 +222,11 @@ hasBlockers(response: Response): boolean
 
 // Type guards
 isTableBlock(block: Block): block is TableBlock
-isPieChartBlock(block: Block): block is PieChartBlock
-isTimeseriesBlock(block: Block): block is TimeseriesBlock
+isChartBlock(block: Block): block is ChartBlock
+isPieChartBlock(block: ChartBlock): block is PieChartBlock
+isTimeseriesChartBlock(block: ChartBlock): block is TimeseriesChartBlock
 isTransactionActionBlock(block: Block): block is TransactionActionBlock
-isInteractiveCardBlock(block: Block): block is InteractiveCardBlock
+isInteractiveBlock(block: Block): block is InteractiveBlock
 isMarkdownBlock(block: Block): block is MarkdownBlock
 ```
 

package/dist/blocks.d.ts

@@ -1,4 +1,4 @@
-import type { Block, TableBlock, ChartBlock, TransactionActionBlock, MarkdownBlock, InteractiveBlock, Primitive, Response } from './types.js';
+import type { Block, TableBlock, ChartBlock, TransactionActionBlock, MarkdownBlock, InteractiveBlock, InfoBlock, CodeBlock, Primitive, Response, RiskInfoItem } from './types.js';
 /**
  * Extract all table blocks from a response.
  */
@@ -19,6 +19,14 @@ export declare function extractMarkdownBlocks(response: Response): MarkdownBlock
  * Extract all interactive blocks from a response.
  */
 export declare function extractInteractiveBlocks(response: Response): InteractiveBlock[];
+/**
+ * Extract all info blocks from a response.
+ */
+export declare function extractInfoBlocks(response: Response): InfoBlock[];
+/**
+ * Extract all code blocks from a response.
+ */
+export declare function extractCodeBlocks(response: Response): CodeBlock[];
 /**
  * Find the first table block with a matching title.
  */
@@ -55,7 +63,7 @@ export declare function getTableColumn(table: TableBlock, columnName: string): u
 /**
  * Find a row in a table by matching a column value.
  */
-export declare function findTableRow(table: TableBlock, columnName: string, value: unknown): (string | number | boolean | null)[] | undefined;
+export declare function findTableRow(table: TableBlock, columnName: string, value: unknown): unknown[] | undefined;
 /**
  * Get table dimensions.
  */
@@ -85,11 +93,11 @@ export declare function getChartPercentages(chart: ChartBlock): Array<{
 /**
  * Get all warnings from transaction blocks.
  */
-export declare function getAllWarnings(response: Response): string[];
+export declare function getAllWarnings(response: Response): RiskInfoItem[];
 /**
  * Get all blockers from transaction blocks.
  */
-export declare function getAllBlockers(response: Response): string[];
+export declare function getAllBlockers(response: Response): RiskInfoItem[];
 /**
  * Get the highest risk level from all transaction blocks.
  */

package/dist/blocks.js

@@ -2,6 +2,7 @@
 //
 // Helper functions for working with blocks extracted from responses.
 import { extractBlocks, isTableBlock, isChartBlock, isTransactionActionBlock, isMarkdownBlock, isInteractiveBlock, } from './types.js';
+import { isInfoBlock, isCodeBlock } from './type-guards.js';
 // ============================================================================
 // Block Extraction Utilities
 // ============================================================================
@@ -35,6 +36,18 @@ export function extractMarkdownBlocks(response) {
 export function extractInteractiveBlocks(response) {
     return extractBlocks(response).filter(isInteractiveBlock);
 }
+/**
+ * Extract all info blocks from a response.
+ */
+export function extractInfoBlocks(response) {
+    return extractBlocks(response).filter(isInfoBlock);
+}
+/**
+ * Extract all code blocks from a response.
+ */
+export function extractCodeBlocks(response) {
+    return extractBlocks(response).filter(isCodeBlock);
+}
 // ============================================================================
 // Block Search Utilities
 // ============================================================================
@@ -42,13 +55,13 @@ export function extractInteractiveBlocks(response) {
  * Find the first table block with a matching title.
  */
 export function findTableByTitle(response, title) {
-    return extractTableBlocks(response).find((block) => block.title.toLowerCase().includes(title.toLowerCase()));
+    return extractTableBlocks(response).find((block) => block.title?.toLowerCase().includes(title.toLowerCase()));
 }
 /**
  * Find the first chart block with a matching title.
  */
 export function findChartByTitle(response, title) {
-    return extractChartBlocks(response).find((block) => block.title.toLowerCase().includes(title.toLowerCase()));
+    return extractChartBlocks(response).find((block) => block.title?.toLowerCase().includes(title.toLowerCase()));
 }
 /**
  * Find transaction blocks containing a specific primitive type.
@@ -139,10 +152,7 @@ export function getTableDimensions(table) {
  * Works with both segments and legacy data formats.
  */
 export function getChartData(chart) {
-    if (chart.segments) {
-        return chart.segments.map((s) => ({ label: s.label, value: s.value }));
-    }
-    if (chart.data) {
+    if (chart.chartType === 'pie') {
         return chart.data.map(([label, value]) => ({ label, value }));
     }
     return [];
@@ -224,6 +234,8 @@ export function countBlocksByType(response) {
         transaction_action: 0,
         markdown: 0,
         interactive: 0,
+        info: 0,
+        code: 0,
     };
     for (const block of extractBlocks(response)) {
         if (block.type in counts) {

package/dist/conversation.d.ts

@@ -10,8 +10,11 @@ export interface ConversationOptions {
     maxHistoryLength?: number;
     /** User identifier */
     userId: string;
-    /** Wallet identifier */
-    walletId: string;
+    /** Array of wallet addresses across multiple chains */
+    wallets?: Array<{
+        address: string;
+        chain: string;
+    }>;
     /** Optional initial session ID (auto-generated if not provided) */
     sessionId?: string;
 }

package/dist/conversation.js

@@ -44,8 +44,8 @@ export class Conversation {
         this.startedAt = new Date();
         this.metadata = {
             user_id: options.userId,
-            wallet_id: options.walletId,
             session_id: options.sessionId || `session-${Date.now()}-${Math.random().toString(36).slice(2, 8)}`,
+            ...(options.wallets && { wallets: options.wallets }),
         };
     }
     /**
@@ -114,12 +114,10 @@ export class Conversation {
      * Useful for restoring conversation state.
      */
     addAssistantMessage(content) {
-        // Store assistant responses as user messages with prefix for context
-        // This matches the pattern used in the multi-turn example
         this.history.push({
             type: 'message',
-            role: 'user',
-            content: `[Assistant]: ${content}`,
+            role: 'assistant',
+            content,
         });
         this.assistantTurns++;
         this.trimHistory();
@@ -164,22 +162,16 @@ export class Conversation {
             model: this.model,
             maxHistoryLength: this.maxHistoryLength,
             userId: this.metadata.user_id,
-            walletId: this.metadata.wallet_id,
+            wallets: this.metadata.wallets,
         });
         // Copy history
         for (const msg of this.history) {
-            if (msg.role === 'user') {
-                if (msg.content.startsWith('[Assistant]: ')) {
-                    forked.history.push(msg);
-                    forked.assistantTurns++;
-                }
-                else {
-                    forked.history.push(msg);
-                    forked.userTurns++;
-                }
+            forked.history.push(msg);
+            if (msg.role === 'assistant' || (msg.role === 'user' && msg.content.startsWith('[Assistant]: '))) {
+                forked.assistantTurns++;
             }
-            else {
-                forked.history.push(msg);
+            else if (msg.role === 'user') {
+                forked.userTurns++;
             }
         }
         return forked;
@@ -201,19 +193,17 @@ export class Conversation {
     static fromJSON(client, data) {
         const conversation = new Conversation(client, {
             userId: data.metadata.user_id,
-            walletId: data.metadata.wallet_id,
+            wallets: data.metadata.wallets,
             sessionId: data.sessionId,
         });
         // Restore history
         for (const msg of data.history) {
             conversation.history.push(msg);
-            if (msg.role === 'user') {
-                if (msg.content.startsWith('[Assistant]: ')) {
-                    conversation.assistantTurns++;
-                }
-                else {
-                    conversation.userTurns++;
-                }
+            if (msg.role === 'assistant' || (msg.role === 'user' && msg.content.startsWith('[Assistant]: '))) {
+                conversation.assistantTurns++;
+            }
+            else if (msg.role === 'user') {
+                conversation.userTurns++;
             }
         }
         return conversation;

package/dist/index.d.ts

@@ -1,17 +1,20 @@
 export { Chaos as default, Chaos, WALLET_MODEL, ASK_MODEL } from './client.js';
 export { httpStreamRequest, StreamingHttpClient } from './http-streaming.js';
-export type { HttpStreamOptions, StreamingHttpClientOptions, StreamRequestOptions } from './http-streaming.js';
-export type { V1WalletRequest, V1AskRequest } from './request.js';
+export type { HttpStreamOptions, StreamingHttpClientOptions, StreamRequestOptions, } from './http-streaming.js';
+export type { V1WalletRequest, V1AskRequest, PreviousMessage } from './request.js';
+export type { WalletInfo } from './types.js';
 export type { V1StreamEvent, V1FinalState } from './response.js';
-export type { ChaosConfig, CreateResponseParams, RequestMetadata, InputItem, Response, OutputItem, ContentPart, Block, OutputText, ChaosBlock, MarkdownBlock, TableBlock, TableColumnType, TableSortDirection, TableColumnConfig, ChartBlock, ChartSeries, ChartSegment, ChartAxis, ChartDataPoint, TransactionActionBlock, Primitive, PrimitiveIcon, PrimitiveLineItem, PrimitiveDisplay, RawTransaction, TransactionGroup, Risks, RiskInfoItem, InteractiveBlock, InteractiveOption, ResponseError, } from './types.js';
+export type { ChaosConfig, CreateResponseParams, RequestMetadata, InputItem, Response, OutputItem, ContentPart, Block, OutputText, ChaosBlock, MarkdownBlock, TableBlock, ChartBlock, PieChartBlock, TimeseriesChartBlock, ChartSeries, ChartSegment, ChartAxis, ChartDataPoint, TransactionActionBlock, Primitive, RawTransaction, TransactionGroup, Risks, RiskInfoItem, InteractiveBlock, InteractiveOption, InfoBlock, CodeBlock, ResponseError, } from './types.js';
 export { ChaosError, ChaosTimeoutError } from './types.js';
 export { extractText, extractBlocks, hasRisks, hasBlockers, isTableBlock, isChartBlock, isTransactionActionBlock, isInteractiveBlock, isMarkdownBlock, isOutputText, isChaosBlock, } from './types.js';
-export { BlockSchema, parseRawBlock, detectBlockType } from './schemas.js';
+export { BlockSchema, parseRawBlock, detectBlockType, ChaosConfigSchema, InputItemSchema, WalletInfoSchema, RequestMetadataSchema, ResponseSchema, ResponseErrorSchema, OutputItemSchema, OutputTextSchema, ChaosBlockSchema, ContentPartSchema, MarkdownBlockSchema, TableBlockSchema, ChartBlockSchema, PieChartBlockSchema, TimeseriesChartBlockSchema, TransactionActionBlockSchema, InteractiveBlockSchema, InteractiveOptionSchema, InfoBlockSchema, CodeBlockSchema, ChartDataPointSchema, ChartSeriesSchema, ChartSegmentSchema, ChartAxisSchema, PrimitiveSchema, PrimitiveIconSchema, PrimitiveLineItemSchema, PrimitiveDisplaySchema, RawTransactionSchema, TransactionGroupSchema, RisksSchema, RiskCheckSchema, RiskInfoItemSchema, PreviousMessageSchema, V1WalletRequestSchema, V1AskRequestSchema, V1StreamEventSchema, V1FinalStateSchema, MessageTypeSchema, AgentStatusSchema, StreamMessageContextSchema, StreamMessageSchema, AgentStatusMessageSchema, AgentTextMessageSchema, ReportMessageSchema, FollowUpSuggestionsMessageSchema, UserInputMessageSchema, AgentStatusContentSchema, AgentMessageContentSchema, AgentTextDataSchema, ReportContentSchema, CaptionSchema, ReferenceSchema, FollowUpSuggestionsContentSchema, FollowUpDataSchema, UserInputContentSchema, UserInputCandidateSchema, ConversationOptionsSchema, ConversationStatsSchema, PrimitiveTypeSchema, SwapParamsSchema, BridgeParamsSchema, LendingParamsSchema, StakingParamsSchema, ClaimParamsSchema, LiquidityParamsSchema, PositionParamsSchema, DepositParamsSchema, MintParamsSchema, ApproveParamsSchema, TransferParamsSchema, WrapParamsSchema, TypedPrimitiveSchema, SwapPrimitiveSchema, BridgePrimitiveSchema, SupplyPrimitiveSchema, WithdrawPrimitiveSchema, BorrowPrimitiveSchema, RepayPrimitiveSchema, StakePrimitiveSchema, UnstakePrimitiveSchema, ClaimPrimitiveSchema, AddLiquidityPrimitiveSchema, RemoveLiquidityPrimitiveSchema, OpenPositionPrimitiveSchema, ClosePositionPrimitiveSchema, DepositPrimitiveSchema, MintPrimitiveSchema, BurnPrimitiveSchema, RestakePrimitiveSchema, ApprovePrimitiveSchema, TransferPrimitiveSchema, WrapPrimitiveSchema, UnwrapPrimitiveSchema, UnknownPrimitiveSchema, } from './schemas.js';
 export type { BlockParsed } from './schemas.js';
 export { Conversation } from './conversation.js';
 export type { ConversationOptions, ConversationStats } from './conversation.js';
 export { PRIMITIVE_SWAP, PRIMITIVE_SUPPLY, PRIMITIVE_WITHDRAW, PRIMITIVE_BORROW, PRIMITIVE_REPAY, PRIMITIVE_STAKE, PRIMITIVE_UNSTAKE, PRIMITIVE_CLAIM, PRIMITIVE_BRIDGE, PRIMITIVE_ADD_LIQUIDITY, PRIMITIVE_REMOVE_LIQUIDITY, PRIMITIVE_OPEN_POSITION, PRIMITIVE_CLOSE_POSITION, PRIMITIVE_DEPOSIT, PRIMITIVE_MINT, PRIMITIVE_BURN, PRIMITIVE_RESTAKE, PRIMITIVE_APPROVE, PRIMITIVE_TRANSFER, PRIMITIVE_WRAP, PRIMITIVE_UNWRAP, ALL_PRIMITIVES, LENDING_PRIMITIVES, TRADING_PRIMITIVES, STAKING_PRIMITIVES, LIQUIDITY_PRIMITIVES, TRANSFER_PRIMITIVES, isValidPrimitive, isLendingPrimitive, isTradingPrimitive, isStakingPrimitive, isLiquidityPrimitive, isTransferPrimitive, } from './primitives.js';
-export type { PrimitiveType } from './primitives.js';
-export { extractTableBlocks, extractChartBlocks, extractTransactionBlocks, extractMarkdownBlocks, extractInteractiveBlocks, findTableByTitle, findChartByTitle, findTransactionsByPrimitive, extractPrimitives, extractPrimitivesByType, getPrimitiveTypes, tableToObjects, getTableColumn, findTableRow, getTableDimensions, getChartData, getChartTotal, getChartPercentages, getAllWarnings, getAllBlockers, getHighestRiskLevel, countBlocksByType, hasBlocks, hasBlockType, } from './blocks.js';
+export type { PrimitiveType, SwapParams, BridgeParams, LendingParams, StakingParams, ClaimParams, LiquidityParams, PositionParams, DepositParams, MintParams, ApproveParams, TransferParams, WrapParams, Primitive as TypedPrimitive, SwapPrimitive, BridgePrimitive, SupplyPrimitive, WithdrawPrimitive, BorrowPrimitive, RepayPrimitive, StakePrimitive, UnstakePrimitive, ClaimPrimitive, AddLiquidityPrimitive, RemoveLiquidityPrimitive, OpenPositionPrimitive, ClosePositionPrimitive, DepositPrimitive, MintPrimitive, BurnPrimitive, RestakePrimitive, ApprovePrimitive, TransferPrimitive, WrapPrimitive, UnwrapPrimitive, UnknownPrimitive, } from './primitives.js';
+export { extractTableBlocks, extractChartBlocks, extractTransactionBlocks, extractMarkdownBlocks, extractInteractiveBlocks, extractInfoBlocks, extractCodeBlocks, findTableByTitle, findChartByTitle, findTransactionsByPrimitive, extractPrimitives, extractPrimitivesByType, getPrimitiveTypes, tableToObjects, getTableColumn, findTableRow, getTableDimensions, getChartData, getChartTotal, getChartPercentages, getAllWarnings, getAllBlockers, getHighestRiskLevel, countBlocksByType, hasBlocks, hasBlockType, } from './blocks.js';
 export { isAgentStatusMessage, isAgentMessage, isReportMessage, isFollowUpSuggestions, isUserInputMessage, parseAgentStatus, isTerminalStatus, extractAgentMessageText, extractSuggestions, extractReportBlock, parseStreamLine, parseStreamLines, } from './stream.js';
-export type { MessageType, AgentStatus, StreamMessage, StreamMessageContext, } from './stream.js';
+export type { MessageType, AgentStatus, StreamMessage, StreamMessageContext, AgentStatusContent, AgentMessageContent, AgentTextData, ReportContent, Caption, Reference, FollowUpSuggestionsContent, FollowUpData, UserInputContent, UserInputCandidate, AgentStatusMessage, AgentTextMessage, ReportMessage, FollowUpSuggestionsMessage, UserInputMessage, } from './stream.js';
+export { isSwapPrimitive, isBridgePrimitive, isSupplyPrimitive, isWithdrawPrimitive, isBorrowPrimitive, isRepayPrimitive, isStakePrimitive, isUnstakePrimitive, isClaimPrimitive, isAddLiquidityPrimitive, isRemoveLiquidityPrimitive, isOpenPositionPrimitive, isClosePositionPrimitive, isDepositPrimitive, isMintPrimitive, isBurnPrimitive, isRestakePrimitive, isApprovePrimitive, isTransferPrimitive as isTransferTypedPrimitive, // Renamed to avoid conflict with primitives.ts
+isWrapPrimitive, isUnwrapPrimitive, isPieChartBlock, isTimeseriesChartBlock, isInfoBlock, isCodeBlock, } from './type-guards.js';

package/dist/index.js

@@ -6,7 +6,29 @@ export { ChaosError, ChaosTimeoutError } from './types.js';
 // Export helper functions
 export { extractText, extractBlocks, hasRisks, hasBlockers, isTableBlock, isChartBlock, isTransactionActionBlock, isInteractiveBlock, isMarkdownBlock, isOutputText, isChaosBlock, } from './types.js';
 // Export schemas for validation
-export { BlockSchema, parseRawBlock, detectBlockType } from './schemas.js';
+export { 
+// Block detection and parsing
+BlockSchema, parseRawBlock, detectBlockType, 
+// Config and request schemas
+ChaosConfigSchema, InputItemSchema, WalletInfoSchema, RequestMetadataSchema, 
+// Response schemas
+ResponseSchema, ResponseErrorSchema, OutputItemSchema, OutputTextSchema, ChaosBlockSchema, ContentPartSchema, 
+// Block schemas
+MarkdownBlockSchema, TableBlockSchema, ChartBlockSchema, PieChartBlockSchema, TimeseriesChartBlockSchema, TransactionActionBlockSchema, InteractiveBlockSchema, InteractiveOptionSchema, InfoBlockSchema, CodeBlockSchema, 
+// Chart component schemas
+ChartDataPointSchema, ChartSeriesSchema, ChartSegmentSchema, ChartAxisSchema, 
+// Transaction component schemas
+PrimitiveSchema, PrimitiveIconSchema, PrimitiveLineItemSchema, PrimitiveDisplaySchema, RawTransactionSchema, TransactionGroupSchema, RisksSchema, RiskCheckSchema, RiskInfoItemSchema, 
+// V1 API schemas
+PreviousMessageSchema, V1WalletRequestSchema, V1AskRequestSchema, V1StreamEventSchema, V1FinalStateSchema, 
+// Stream message schemas
+MessageTypeSchema, AgentStatusSchema, StreamMessageContextSchema, StreamMessageSchema, AgentStatusMessageSchema, AgentTextMessageSchema, ReportMessageSchema, FollowUpSuggestionsMessageSchema, UserInputMessageSchema, AgentStatusContentSchema, AgentMessageContentSchema, AgentTextDataSchema, ReportContentSchema, CaptionSchema, ReferenceSchema, FollowUpSuggestionsContentSchema, FollowUpDataSchema, UserInputContentSchema, UserInputCandidateSchema, 
+// Conversation schemas
+ConversationOptionsSchema, ConversationStatsSchema, 
+// Primitive type and param schemas
+PrimitiveTypeSchema, SwapParamsSchema, BridgeParamsSchema, LendingParamsSchema, StakingParamsSchema, ClaimParamsSchema, LiquidityParamsSchema, PositionParamsSchema, DepositParamsSchema, MintParamsSchema, ApproveParamsSchema, TransferParamsSchema, WrapParamsSchema, 
+// Typed primitive schemas
+TypedPrimitiveSchema, SwapPrimitiveSchema, BridgePrimitiveSchema, SupplyPrimitiveSchema, WithdrawPrimitiveSchema, BorrowPrimitiveSchema, RepayPrimitiveSchema, StakePrimitiveSchema, UnstakePrimitiveSchema, ClaimPrimitiveSchema, AddLiquidityPrimitiveSchema, RemoveLiquidityPrimitiveSchema, OpenPositionPrimitiveSchema, ClosePositionPrimitiveSchema, DepositPrimitiveSchema, MintPrimitiveSchema, BurnPrimitiveSchema, RestakePrimitiveSchema, ApprovePrimitiveSchema, TransferPrimitiveSchema, WrapPrimitiveSchema, UnwrapPrimitiveSchema, UnknownPrimitiveSchema, } from './schemas.js';
 // Export Conversation class
 export { Conversation } from './conversation.js';
 // Export primitive constants and helpers
@@ -20,7 +42,7 @@ isValidPrimitive, isLendingPrimitive, isTradingPrimitive, isStakingPrimitive, is
 // Export block utilities
 export { 
 // Block extraction
-extractTableBlocks, extractChartBlocks, extractTransactionBlocks, extractMarkdownBlocks, extractInteractiveBlocks, 
+extractTableBlocks, extractChartBlocks, extractTransactionBlocks, extractMarkdownBlocks, extractInteractiveBlocks, extractInfoBlocks, extractCodeBlocks, 
 // Block search
 findTableByTitle, findChartByTitle, findTransactionsByPrimitive, 
 // Primitive extraction
@@ -35,3 +57,12 @@ getAllWarnings, getAllBlockers, getHighestRiskLevel,
 countBlocksByType, hasBlocks, hasBlockType, } from './blocks.js';
 // Export stream message utilities
 export { isAgentStatusMessage, isAgentMessage, isReportMessage, isFollowUpSuggestions, isUserInputMessage, parseAgentStatus, isTerminalStatus, extractAgentMessageText, extractSuggestions, extractReportBlock, parseStreamLine, parseStreamLines, } from './stream.js';
+// Export typed primitive type guards
+export { 
+// Primitive type guards
+isSwapPrimitive, isBridgePrimitive, isSupplyPrimitive, isWithdrawPrimitive, isBorrowPrimitive, isRepayPrimitive, isStakePrimitive, isUnstakePrimitive, isClaimPrimitive, isAddLiquidityPrimitive, isRemoveLiquidityPrimitive, isOpenPositionPrimitive, isClosePositionPrimitive, isDepositPrimitive, isMintPrimitive, isBurnPrimitive, isRestakePrimitive, isApprovePrimitive, isTransferPrimitive as isTransferTypedPrimitive, // Renamed to avoid conflict with primitives.ts
+isWrapPrimitive, isUnwrapPrimitive, 
+// Chart subtype guards
+isPieChartBlock, isTimeseriesChartBlock, 
+// Block type guards
+isInfoBlock, isCodeBlock, } from './type-guards.js';