@chaoslabs/ai-sdk 0.0.3 → 0.0.4

package/dist/schemas.js

@@ -52,24 +52,11 @@ export const ChartDataPointSchema = z.union([
     z.object({ x: z.union([z.number(), z.string()]), y: z.number() }),
     z.array(z.unknown()),
 ]);
-// Array format series item (legacy)
 export const ChartSeriesSchema = z.object({
     name: z.string(),
     data: z.array(ChartDataPointSchema),
     color: z.string().optional(),
 });
-// Timeseries format: server sends series as { "24H": [{ label, data }, ...] }
-export const TimeseriesSeriesItemSchema = z.object({
-    label: z.string(),
-    data: z.array(z.array(z.union([z.number(), z.string()]))), // [[timestamp, value], ...]
-});
-// Series can be either:
-// - Array format: [{ name, data, color }, ...]
-// - Object format (timeseries): { "24H": [{ label, data }, ...], "7D": [...] }
-export const ChartSeriesFieldSchema = z.union([
-    z.array(ChartSeriesSchema),
-    z.record(z.string(), z.array(TimeseriesSeriesItemSchema)),
-]);
 export const ChartSegmentSchema = z.object({
     label: z.string(),
     value: z.number(),
@@ -87,7 +74,7 @@ export const ChartBlockSchema = z.object({
     blockType: z.literal('chart'),
     title: z.string(),
     chartType: z.enum(['pie', 'donut', 'line', 'area', 'bar', 'timeseries']).optional().nullable(),
-    series: ChartSeriesFieldSchema.nullish(),
+    series: z.array(ChartSeriesSchema).optional().nullable(),
     segments: z.array(ChartSegmentSchema).optional().nullable(),
     // Legacy data format from server: [[label, value], ...]
     data: ChartLegacyDataSchema.optional().nullable(),
@@ -97,8 +84,6 @@ export const ChartBlockSchema = z.object({
     isCurrency: z.boolean().optional().nullable(),
     sourceName: z.string().optional().nullable(),
     timeframe: z.string().optional().nullable(),
-    captions: z.array(z.object({ label: z.string(), value: z.string() })).nullish(),
-    titleCryptIcons: z.array(z.object({ name: z.string(), entity: z.string().optional() })).nullish(),
     tool_params: z.unknown().optional().nullable(),
     tool_name: z.string().optional().nullable(),
 });
@@ -147,31 +132,19 @@ export const TransactionGroupSchema = z.object({
     requiresApproval: z.boolean().optional(),
     verificationUnavailable: z.boolean().optional(),
 });
-// Risk impact (quantified risk details)
-export const RiskImpactSchema = z.object({
-    metric: z.string(),
-    current: z.number().nullish(),
-    projected: z.number().nullish(),
-    threshold: z.number().nullish(),
-    at_risk_usd: z.number().nullish(),
-    liquidation_price: z.string().nullish(),
-});
-// Risk check item (used in blockers, warnings, info arrays)
-export const RiskCheckSchema = z.object({
-    id: z.string(),
-    severity: z.enum(['block', 'warn', 'info']),
-    title: z.string(),
-    message: z.string(),
-    impact: RiskImpactSchema.nullish(),
+// Risk info item
+export const RiskInfoItemSchema = z.object({
+    id: z.string().optional(),
+    severity: z.enum(['info', 'warning', 'error', 'critical']).optional(),
+    title: z.string().optional(),
+    message: z.string().optional(),
 });
-// Legacy alias for backwards compatibility
-export const RiskInfoItemSchema = RiskCheckSchema;
 // Risks object
 export const RisksSchema = z.object({
-    level: z.enum(['low', 'medium', 'high', 'critical']).nullish(),
-    blockers: z.array(RiskCheckSchema).nullish(),
-    warnings: z.array(RiskCheckSchema).nullish(),
-    info: z.array(RiskCheckSchema).nullish(),
+    level: z.string().optional(),
+    blockers: z.array(z.string()).optional(),
+    warnings: z.array(z.string()).optional(),
+    info: z.array(RiskInfoItemSchema).optional(),
 });
 export const TransactionActionBlockSchema = z.object({
     blockType: z.literal('transaction_action'),
@@ -187,21 +160,19 @@ export const TransactionActionBlockSchema = z.object({
     tool_name: z.string().optional().nullable(),
 });
 // --- Interactive Block ---
-// Server sends: { blockType: "interactive", title, body?, context?, style?, options? }
-// Note: Using .nullish() to accept both null and undefined (Python sends null for None)
+// Server sends: { blockType: "interactive", ... }
 export const InteractiveOptionSchema = z.object({
     id: z.string(),
     label: z.string(),
-    description: z.string().nullish(), // Accept null OR undefined
-    metadata: z.record(z.string(), z.unknown()).nullish(),
+    description: z.string().optional(),
+    value: z.unknown().optional(),
 });
 export const InteractiveBlockSchema = z.object({
     blockType: z.literal('interactive'),
-    title: z.string(), // Required, matches Python
-    body: z.string().nullish(), // Optional, matches Python
-    context: z.string().nullish(), // Optional, matches Python
-    style: z.enum(['options', 'confirm_cancel']).nullish(),
-    options: z.array(InteractiveOptionSchema).nullish(),
+    prompt: z.string().optional(),
+    options: z.array(InteractiveOptionSchema).optional(),
+    allowCustom: z.boolean().optional(),
+    multiSelect: z.boolean().optional(),
 });
 // ============================================================================
 // Block Detection and Parsing

package/dist/stream.d.ts

@@ -0,0 +1,32 @@
+/**
+ * Stream Message Types and Utilities
+ *
+ * Utilities for parsing and working with WebSocket stream messages
+ * from the Chaos AI backend.
+ */
+export type MessageType = "agent_status_change" | "agent_message" | "report" | "follow_up_suggestions" | "user_input";
+export type AgentStatus = "processing" | "done" | "error" | "cancelled";
+export interface StreamMessageContext {
+    sessionId: string;
+    artifactId: string;
+    query?: string;
+}
+export interface StreamMessage {
+    id: string;
+    type: MessageType;
+    timestamp: number;
+    content: unknown;
+    context: StreamMessageContext;
+}
+export declare function isAgentStatusMessage(msg: StreamMessage): boolean;
+export declare function isAgentMessage(msg: StreamMessage): boolean;
+export declare function isReportMessage(msg: StreamMessage): boolean;
+export declare function isFollowUpSuggestions(msg: StreamMessage): boolean;
+export declare function isUserInputMessage(msg: StreamMessage): boolean;
+export declare function parseAgentStatus(msg: StreamMessage): AgentStatus | null;
+export declare function isTerminalStatus(status: AgentStatus): boolean;
+export declare function extractAgentMessageText(msg: StreamMessage): string | null;
+export declare function extractSuggestions(msg: StreamMessage): string[];
+export declare function extractReportBlock(msg: StreamMessage): unknown | null;
+export declare function parseStreamLine(line: string): StreamMessage | null;
+export declare function parseStreamLines(text: string): StreamMessage[];

package/dist/stream.js

@@ -0,0 +1,127 @@
+/**
+ * Stream Message Types and Utilities
+ *
+ * Utilities for parsing and working with WebSocket stream messages
+ * from the Chaos AI backend.
+ */
+// ============================================================================
+// Type Guards
+// ============================================================================
+export function isAgentStatusMessage(msg) {
+    return msg.type === "agent_status_change";
+}
+export function isAgentMessage(msg) {
+    return msg.type === "agent_message";
+}
+export function isReportMessage(msg) {
+    return msg.type === "report";
+}
+export function isFollowUpSuggestions(msg) {
+    return msg.type === "follow_up_suggestions";
+}
+export function isUserInputMessage(msg) {
+    return msg.type === "user_input";
+}
+// ============================================================================
+// Status Utilities
+// ============================================================================
+const VALID_STATUSES = ["processing", "done", "error", "cancelled"];
+export function parseAgentStatus(msg) {
+    if (msg.type !== "agent_status_change") {
+        return null;
+    }
+    const content = msg.content;
+    if (typeof content?.status === "string" &&
+        VALID_STATUSES.includes(content.status)) {
+        return content.status;
+    }
+    return null;
+}
+const TERMINAL_STATUSES = ["done", "error", "cancelled"];
+export function isTerminalStatus(status) {
+    return TERMINAL_STATUSES.includes(status);
+}
+// ============================================================================
+// Content Extraction
+// ============================================================================
+export function extractAgentMessageText(msg) {
+    if (msg.type !== "agent_message") {
+        return null;
+    }
+    const content = msg.content;
+    if (typeof content?.data?.message === "string") {
+        return content.data.message;
+    }
+    return null;
+}
+export function extractSuggestions(msg) {
+    if (msg.type !== "follow_up_suggestions") {
+        return [];
+    }
+    const content = msg.content;
+    if (Array.isArray(content?.suggestions)) {
+        return content.suggestions.filter((s) => typeof s === "string");
+    }
+    return [];
+}
+export function extractReportBlock(msg) {
+    if (msg.type !== "report") {
+        return null;
+    }
+    const content = msg.content;
+    if (content?.data !== undefined) {
+        return content.data;
+    }
+    return null;
+}
+// ============================================================================
+// Stream Parser
+// ============================================================================
+const VALID_MESSAGE_TYPES = [
+    "agent_status_change",
+    "agent_message",
+    "report",
+    "follow_up_suggestions",
+    "user_input",
+];
+function isValidStreamMessage(obj) {
+    if (typeof obj !== "object" || obj === null) {
+        return false;
+    }
+    const msg = obj;
+    return (typeof msg.id === "string" &&
+        typeof msg.type === "string" &&
+        VALID_MESSAGE_TYPES.includes(msg.type) &&
+        typeof msg.timestamp === "number" &&
+        typeof msg.context === "object" &&
+        msg.context !== null &&
+        typeof msg.context.sessionId === "string" &&
+        typeof msg.context.artifactId === "string");
+}
+export function parseStreamLine(line) {
+    const trimmed = line.trim();
+    if (!trimmed) {
+        return null;
+    }
+    try {
+        const parsed = JSON.parse(trimmed);
+        if (isValidStreamMessage(parsed)) {
+            return parsed;
+        }
+        return null;
+    }
+    catch {
+        return null;
+    }
+}
+export function parseStreamLines(text) {
+    const lines = text.split("\n");
+    const messages = [];
+    for (const line of lines) {
+        const msg = parseStreamLine(line);
+        if (msg !== null) {
+            messages.push(msg);
+        }
+    }
+    return messages;
+}

package/dist/types.d.ts

@@ -8,7 +8,13 @@ export interface ChaosConfig {
     baseUrl?: string;
     /** Request timeout in milliseconds (defaults to 120000) */
     timeout?: number;
+    /** Use native http/https modules for streaming instead of fetch (defaults to true) */
+    useNativeHttp?: boolean;
 }
+/**
+ * Parameters for creating a response.
+ */
+import type { StreamMessage } from './stream.js';
 /**
  * Parameters for creating a response.
  */
@@ -25,6 +31,11 @@ export interface CreateResponseParams {
     stream?: boolean;
     /** Metadata for the request */
     metadata: RequestMetadata;
+    /**
+     * Callback for real-time stream events.
+     * Called for each message as it arrives from the server.
+     */
+    onStreamEvent?: (event: StreamMessage) => void;
 }
 /**
  * An input message in the conversation.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@chaoslabs/ai-sdk",
-  "version": "0.0.3",
+  "version": "0.0.4",
   "description": "Chaos AI SDK - TypeScript SDK for the Chaos AI API",
   "type": "module",
   "main": "./dist/index.js",
@@ -17,6 +17,7 @@
     "LICENSE"
   ],
   "scripts": {
+    "test": "bun test",
     "build": "tsc",
     "typecheck": "tsc --noEmit",
     "prepublishOnly": "bun run build",