@chaoslabs/ai-sdk 0.0.1 → 0.0.2

package/dist/index.d.ts

@@ -1,6 +1,8 @@
 export { Chaos as default, Chaos, WALLET_MODEL, ASK_MODEL } from './client.js';
 export type { V1WalletRequest, V1AskRequest } from './request.js';
 export type { V1StreamEvent, V1FinalState } from './response.js';
-export type { ChaosConfig, CreateResponseParams, RequestMetadata, InputItem, Response, OutputItem, ContentPart, Block, OutputText, ChaosBlock, TableBlock, MarkdownBlock, TransactionActionBlock, Primitive, InteractiveCardBlock, InteractiveOption, TimeseriesBlock, PieChartBlock, Risks, ResponseError, } from './types.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 { ChaosError, ChaosTimeoutError } from './types.js';
-export { extractText, extractBlocks, hasRisks, hasBlockers, isTableBlock, isPieChartBlock, isTimeseriesBlock, isTransactionActionBlock, isInteractiveCardBlock, isMarkdownBlock, isOutputText, isChaosBlock, } 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 type { BlockParsed } from './schemas.js';

package/dist/index.js

@@ -3,4 +3,6 @@ export { Chaos as default, Chaos, WALLET_MODEL, ASK_MODEL } from './client.js';
 // Export error classes
 export { ChaosError, ChaosTimeoutError } from './types.js';
 // Export helper functions
-export { extractText, extractBlocks, hasRisks, hasBlockers, isTableBlock, isPieChartBlock, isTimeseriesBlock, isTransactionActionBlock, isInteractiveCardBlock, isMarkdownBlock, isOutputText, isChaosBlock, } from './types.js';
+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';

package/dist/response.js

@@ -2,7 +2,7 @@
 //
 // Parses responses from /v1/chat/stream endpoint.
 // Handles type: "report" events with content.data blocks.
-import { convertBlock } from './converter.js';
+import { parseRawBlock } from './schemas.js';
 // ============================================================================
 // Response Parser
 // ============================================================================
@@ -67,6 +67,48 @@ export async function parseV1Stream(body, onEvent) {
     }
     return { blocks };
 }
+/**
+ * Parse a raw block using schema detection.
+ * Returns the parsed block or null if parsing fails.
+ * Transforms legacy data formats to normalized structure.
+ */
+function parseBlock(rawBlock) {
+    const result = parseRawBlock(rawBlock);
+    if (result.success) {
+        const block = result.data;
+        // Transform legacy chart data format to segments
+        if (block.type === 'chart') {
+            const chartBlock = block;
+            // If server sent data as [[label, value], ...] but not segments, convert it
+            if (chartBlock.data && (!chartBlock.segments || chartBlock.segments.length === 0)) {
+                chartBlock.segments = chartBlock.data.map(([label, value]) => ({
+                    label,
+                    value,
+                }));
+            }
+            // Clean up null values for cleaner API
+            if (chartBlock.segments === null)
+                chartBlock.segments = undefined;
+            if (chartBlock.series === null)
+                chartBlock.series = undefined;
+            if (chartBlock.categories === null)
+                chartBlock.categories = undefined;
+            if (chartBlock.xAxis === null)
+                chartBlock.xAxis = undefined;
+            if (chartBlock.yAxis === null)
+                chartBlock.yAxis = undefined;
+            if (chartBlock.sourceName === null)
+                chartBlock.sourceName = undefined;
+            if (chartBlock.timeframe === null)
+                chartBlock.timeframe = undefined;
+            if (chartBlock.isCurrency === null)
+                chartBlock.isCurrency = undefined;
+        }
+        return block;
+    }
+    console.error('Block parse error:', result.error);
+    return null;
+}
 /**
  * Convert v1 final state to SDK Response format.
  */
@@ -83,8 +125,8 @@ export function toV1Response(responseId, model, finalState) {
                 });
             }
             else if (rawBlock && typeof rawBlock === 'object') {
-                // Block content - use existing converter
-                const block = convertBlock(rawBlock);
+                // Block content - parse with schema
+                const block = parseBlock(rawBlock);
                 if (block) {
                     content.push({
                         type: 'chaos.block',
@@ -95,11 +137,13 @@ export function toV1Response(responseId, model, finalState) {
         }
     }
     const output = content.length > 0
-        ? [{
+        ? [
+            {
                 type: 'message',
                 role: 'assistant',
                 content,
-            }]
+            },
+        ]
         : [];
     return {
         id: responseId,