@chaoslabs/ai-sdk 0.0.9 → 0.0.11

package/CHANGELOG.md

@@ -0,0 +1,58 @@
+# 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).
+
+## [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({
@@ -98,7 +98,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 +110,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 |

package/dist/client.d.ts

@@ -55,7 +55,7 @@ declare class V1Chat {
  *
  * const chaos = new Chaos({
  *   apiKey: 'ck-...',
- *   baseUrl: 'https://ai-staging.chaoslabs.co'
+ *   baseUrl: 'https://ai.chaoslabs.co'
  * });
  *
  * const response = await chaos.chat.responses.create({

package/dist/client.js

@@ -155,7 +155,7 @@ class V1Chat {
  *
  * const chaos = new Chaos({
  *   apiKey: 'ck-...',
- *   baseUrl: 'https://ai-staging.chaoslabs.co'
+ *   baseUrl: 'https://ai.chaoslabs.co'
  * });
  *
  * const response = await chaos.chat.responses.create({
@@ -181,7 +181,7 @@ export class Chaos {
         const useNativeHttp = config.useNativeHttp !== false;
         this.config = {
             apiKey: config.apiKey,
-            baseUrl: config.baseUrl || 'https://ai-staging.chaoslabs.co',
+            baseUrl: config.baseUrl || 'https://ai.chaoslabs.co',
             timeout: config.timeout || DEFAULT_TIMEOUT,
             useNativeHttp,
         };

package/dist/index.d.ts

@@ -14,4 +14,4 @@ export { PRIMITIVE_SWAP, PRIMITIVE_SUPPLY, PRIMITIVE_WITHDRAW, PRIMITIVE_BORROW,
 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 { 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';

package/dist/stream.d.ts

@@ -3,7 +3,12 @@
  *
  * Utilities for parsing and working with WebSocket stream messages
  * from the Chaos AI backend.
+ *
+ * StreamMessage is a discriminated union - the `type` field determines
+ * the shape of `content`. TypeScript will narrow the content type
+ * automatically when you check the type field.
  */
+import type { Block } from "./types.js";
 export type MessageType = "agent_status_change" | "agent_message" | "report" | "follow_up_suggestions" | "user_input";
 export type AgentStatus = "processing" | "done" | "error" | "cancelled";
 export interface StreamMessageContext {
@@ -11,22 +16,212 @@ export interface StreamMessageContext {
     artifactId: string;
     query?: string;
 }
-export interface StreamMessage {
+/**
+ * Content for agent_status_change messages.
+ * Indicates the current status of the agent workflow.
+ */
+export interface AgentStatusContent {
+    status: AgentStatus;
+}
+/**
+ * Content for agent_message messages (thinking/progress messages).
+ * Contains the agent's current thinking or progress update.
+ */
+export interface AgentMessageContent {
+    messageId: string;
+    artifactId: string;
+    type: string;
+    data: AgentTextData;
+    order: number;
+    createdAt: number;
+    updatedAt: number;
+    metadata?: Record<string, unknown>;
+}
+/**
+ * The data payload within an agent message.
+ */
+export interface AgentTextData {
+    message: string;
+}
+/**
+ * A caption for a report block.
+ */
+export interface Caption {
+    label: string;
+    value: unknown;
+    group?: string;
+    collapsable?: boolean;
+}
+/**
+ * A reference used to generate the response.
+ */
+export interface Reference {
+    title: string;
+    url: string;
+    content: string;
+    hiddenUrl?: string;
+    textSpans?: number[][];
+    engineId?: string;
+}
+/**
+ * Content for report messages.
+ * Contains a block of content (markdown, table, chart, etc.)
+ */
+export interface ReportContent {
+    id: string;
+    artifactId: string;
+    type: string;
+    data: Block;
+    order: number;
+    createdAt: number;
+    updatedAt: number;
+    createdBy?: string;
+    metadata?: Record<string, unknown>;
+    captions?: Caption[];
+    references?: Reference[];
+    version?: number;
+}
+/**
+ * Content for follow_up_suggestions messages.
+ * Contains suggested follow-up queries.
+ */
+export interface FollowUpSuggestionsContent {
+    messageId: string;
+    artifactId: string;
+    type: string;
+    data: FollowUpData;
+    order: number;
+    createdAt: number;
+    updatedAt: number;
+    metadata?: Record<string, unknown>;
+}
+/**
+ * The data payload for follow-up suggestions.
+ */
+export interface FollowUpData {
+    followUpQueries: string[];
+}
+/**
+ * Content for user_input messages.
+ * Indicates the agent needs clarification from the user.
+ */
+export interface UserInputContent {
+    type: string;
+    candidates: Record<string, UserInputCandidate[]>;
+}
+/**
+ * A candidate option for user disambiguation.
+ */
+export interface UserInputCandidate {
+    symbol: string;
+    name: string;
+    market_cap: number;
+    token_id: string;
+    score: number;
+}
+interface StreamMessageBase {
     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;
+/**
+ * Agent status change message.
+ * @example
+ * if (msg.type === 'agent_status_change') {
+ *   console.log(msg.content.status); // TypeScript knows this is AgentStatus
+ * }
+ */
+export interface AgentStatusMessage extends StreamMessageBase {
+    type: "agent_status_change";
+    content: AgentStatusContent;
+}
+/**
+ * Agent thinking/progress message.
+ * @example
+ * if (msg.type === 'agent_message') {
+ *   console.log(msg.content.data.message); // TypeScript knows the shape
+ * }
+ */
+export interface AgentTextMessage extends StreamMessageBase {
+    type: "agent_message";
+    content: AgentMessageContent;
+}
+/**
+ * Report block message (markdown, table, chart, etc.)
+ * @example
+ * if (msg.type === 'report') {
+ *   const blockData = msg.content.data; // The actual block
+ * }
+ */
+export interface ReportMessage extends StreamMessageBase {
+    type: "report";
+    content: ReportContent;
+}
+/**
+ * Follow-up suggestions message.
+ * @example
+ * if (msg.type === 'follow_up_suggestions') {
+ *   const queries = msg.content.data.followUpQueries;
+ * }
+ */
+export interface FollowUpSuggestionsMessage extends StreamMessageBase {
+    type: "follow_up_suggestions";
+    content: FollowUpSuggestionsContent;
+}
+/**
+ * User input request message.
+ * @example
+ * if (msg.type === 'user_input') {
+ *   const candidates = msg.content.candidates;
+ * }
+ */
+export interface UserInputMessage extends StreamMessageBase {
+    type: "user_input";
+    content: UserInputContent;
+}
+/**
+ * Union of all stream message types.
+ *
+ * This is a discriminated union - check the `type` field to narrow
+ * the content type automatically:
+ *
+ * @example
+ * function handleMessage(msg: StreamMessage) {
+ *   switch (msg.type) {
+ *     case 'agent_status_change':
+ *       // TypeScript knows: msg.content is AgentStatusContent
+ *       console.log(msg.content.status);
+ *       break;
+ *     case 'agent_message':
+ *       // TypeScript knows: msg.content is AgentMessageContent
+ *       console.log(msg.content.data.message);
+ *       break;
+ *     case 'report':
+ *       // TypeScript knows: msg.content is ReportContent
+ *       console.log(msg.content.data);
+ *       break;
+ *     case 'follow_up_suggestions':
+ *       // TypeScript knows: msg.content is FollowUpSuggestionsContent
+ *       console.log(msg.content.data.followUpQueries);
+ *       break;
+ *     case 'user_input':
+ *       // TypeScript knows: msg.content is UserInputContent
+ *       console.log(msg.content.candidates);
+ *       break;
+ *   }
+ * }
+ */
+export type StreamMessage = AgentStatusMessage | AgentTextMessage | ReportMessage | FollowUpSuggestionsMessage | UserInputMessage;
+export declare function isAgentStatusMessage(msg: StreamMessage): msg is AgentStatusMessage;
+export declare function isAgentMessage(msg: StreamMessage): msg is AgentTextMessage;
+export declare function isReportMessage(msg: StreamMessage): msg is ReportMessage;
+export declare function isFollowUpSuggestions(msg: StreamMessage): msg is FollowUpSuggestionsMessage;
+export declare function isUserInputMessage(msg: StreamMessage): msg is UserInputMessage;
 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 extractReportBlock(msg: StreamMessage): Block | null;
 export declare function parseStreamLine(line: string): StreamMessage | null;
 export declare function parseStreamLines(text: string): StreamMessage[];
+export {};

package/dist/stream.js

@@ -3,6 +3,10 @@
  *
  * Utilities for parsing and working with WebSocket stream messages
  * from the Chaos AI backend.
+ *
+ * StreamMessage is a discriminated union - the `type` field determines
+ * the shape of `content`. TypeScript will narrow the content type
+ * automatically when you check the type field.
  */
 // ============================================================================
 // Type Guards
@@ -30,49 +34,32 @@ 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;
+    return msg.content.status;
 }
 const TERMINAL_STATUSES = ["done", "error", "cancelled"];
 export function isTerminalStatus(status) {
     return TERMINAL_STATUSES.includes(status);
 }
 // ============================================================================
-// Content Extraction
+// Content Extraction (for convenience, type guards provide same functionality)
 // ============================================================================
 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;
+    return msg.content.data.message;
 }
 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 [];
+    return msg.content.data.followUpQueries;
 }
 export function extractReportBlock(msg) {
     if (msg.type !== "report") {
         return null;
     }
-    const content = msg.content;
-    if (content?.data !== undefined) {
-        return content.data;
-    }
-    return null;
+    return msg.content.data;
 }
 // ============================================================================
 // Stream Parser

package/dist/types.d.ts

@@ -4,7 +4,7 @@
 export interface ChaosConfig {
     /** API key for authentication */
     apiKey: string;
-    /** Base URL for the API (defaults to https://ai-staging.chaoslabs.co) */
+    /** Base URL for the API (defaults to https://ai.chaoslabs.co) */
     baseUrl?: string;
     /** Request timeout in milliseconds (defaults to 120000) */
     timeout?: number;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@chaoslabs/ai-sdk",
-  "version": "0.0.9",
+  "version": "0.0.11",
   "description": "Chaos AI SDK - TypeScript SDK for the Chaos AI API",
   "type": "module",
   "main": "./dist/index.js",
@@ -14,6 +14,7 @@
   "files": [
     "dist",
     "README.md",
+    "CHANGELOG.md",
     "LICENSE"
   ],
   "scripts": {