@chaoslabs/ai-sdk 0.0.2 → 0.0.4

package/dist/primitives.d.ts

@@ -0,0 +1,157 @@
+/**
+ * Swap tokens on a decentralized exchange.
+ * @example "Swap 1 ETH for USDC"
+ */
+export declare const PRIMITIVE_SWAP = "swap";
+/**
+ * Supply assets to a lending protocol.
+ * @example "Supply 1000 USDC to Aave"
+ */
+export declare const PRIMITIVE_SUPPLY = "supply";
+/**
+ * Withdraw supplied assets from a lending protocol.
+ * @example "Withdraw my USDC from Compound"
+ */
+export declare const PRIMITIVE_WITHDRAW = "withdraw";
+/**
+ * Borrow assets from a lending protocol.
+ * @example "Borrow 500 DAI from Aave"
+ */
+export declare const PRIMITIVE_BORROW = "borrow";
+/**
+ * Repay borrowed assets to a lending protocol.
+ * @example "Repay my DAI debt on Compound"
+ */
+export declare const PRIMITIVE_REPAY = "repay";
+/**
+ * Stake tokens for rewards or governance.
+ * @example "Stake 10 ETH with Lido"
+ */
+export declare const PRIMITIVE_STAKE = "stake";
+/**
+ * Unstake previously staked tokens.
+ * @example "Unstake my stETH"
+ */
+export declare const PRIMITIVE_UNSTAKE = "unstake";
+/**
+ * Claim accrued rewards from protocols.
+ * @example "Claim my Aave rewards"
+ */
+export declare const PRIMITIVE_CLAIM = "claim";
+/**
+ * Bridge assets between chains.
+ * @example "Bridge 100 USDC to Arbitrum"
+ */
+export declare const PRIMITIVE_BRIDGE = "bridge";
+/**
+ * Provide liquidity to a pool.
+ * @example "Add liquidity to ETH-USDC pool"
+ */
+export declare const PRIMITIVE_ADD_LIQUIDITY = "add_liquidity";
+/**
+ * Remove liquidity from a pool.
+ * @example "Remove liquidity from my ETH-USDC position"
+ */
+export declare const PRIMITIVE_REMOVE_LIQUIDITY = "remove_liquidity";
+/**
+ * Open a perpetual futures position.
+ * @example "Open 2x long ETH position"
+ */
+export declare const PRIMITIVE_OPEN_POSITION = "open_position";
+/**
+ * Close a perpetual futures position.
+ * @example "Close my ETH long"
+ */
+export declare const PRIMITIVE_CLOSE_POSITION = "close_position";
+/**
+ * Deposit into a vault strategy.
+ * @example "Deposit 1000 USDC into Yearn"
+ */
+export declare const PRIMITIVE_DEPOSIT = "deposit";
+/**
+ * Mint stablecoins or synthetic assets.
+ * @example "Mint 1000 DAI from my ETH collateral"
+ */
+export declare const PRIMITIVE_MINT = "mint";
+/**
+ * Burn or repay minted assets.
+ * @example "Burn 500 DAI to reduce my debt"
+ */
+export declare const PRIMITIVE_BURN = "burn";
+/**
+ * Restake assets for additional yield.
+ * @example "Restake my stETH on EigenLayer"
+ */
+export declare const PRIMITIVE_RESTAKE = "restake";
+/**
+ * Approve token spending.
+ * @example "Approve USDC for Uniswap"
+ */
+export declare const PRIMITIVE_APPROVE = "approve";
+/**
+ * Transfer tokens to another address.
+ * @example "Send 100 USDC to 0x..."
+ */
+export declare const PRIMITIVE_TRANSFER = "transfer";
+/**
+ * Wrap native tokens (e.g., ETH to WETH).
+ * @example "Wrap 1 ETH to WETH"
+ */
+export declare const PRIMITIVE_WRAP = "wrap";
+/**
+ * Unwrap wrapped tokens (e.g., WETH to ETH).
+ * @example "Unwrap 1 WETH to ETH"
+ */
+export declare const PRIMITIVE_UNWRAP = "unwrap";
+/**
+ * All DeFi primitive types as a union type.
+ */
+export type PrimitiveType = typeof PRIMITIVE_SWAP | typeof PRIMITIVE_SUPPLY | typeof PRIMITIVE_WITHDRAW | typeof PRIMITIVE_BORROW | typeof PRIMITIVE_REPAY | typeof PRIMITIVE_STAKE | typeof PRIMITIVE_UNSTAKE | typeof PRIMITIVE_CLAIM | typeof PRIMITIVE_BRIDGE | typeof PRIMITIVE_ADD_LIQUIDITY | typeof PRIMITIVE_REMOVE_LIQUIDITY | typeof PRIMITIVE_OPEN_POSITION | typeof PRIMITIVE_CLOSE_POSITION | typeof PRIMITIVE_DEPOSIT | typeof PRIMITIVE_MINT | typeof PRIMITIVE_BURN | typeof PRIMITIVE_RESTAKE | typeof PRIMITIVE_APPROVE | typeof PRIMITIVE_TRANSFER | typeof PRIMITIVE_WRAP | typeof PRIMITIVE_UNWRAP;
+/**
+ * All primitive type constants for iteration.
+ */
+export declare const ALL_PRIMITIVES: PrimitiveType[];
+/**
+ * Lending-related primitives.
+ */
+export declare const LENDING_PRIMITIVES: PrimitiveType[];
+/**
+ * Trading-related primitives.
+ */
+export declare const TRADING_PRIMITIVES: PrimitiveType[];
+/**
+ * Staking-related primitives.
+ */
+export declare const STAKING_PRIMITIVES: PrimitiveType[];
+/**
+ * Liquidity-related primitives.
+ */
+export declare const LIQUIDITY_PRIMITIVES: PrimitiveType[];
+/**
+ * Transfer-related primitives.
+ */
+export declare const TRANSFER_PRIMITIVES: PrimitiveType[];
+/**
+ * Check if a primitive type is valid.
+ */
+export declare function isValidPrimitive(primitive: string): primitive is PrimitiveType;
+/**
+ * Check if a primitive is a lending operation.
+ */
+export declare function isLendingPrimitive(primitive: string): boolean;
+/**
+ * Check if a primitive is a trading operation.
+ */
+export declare function isTradingPrimitive(primitive: string): boolean;
+/**
+ * Check if a primitive is a staking operation.
+ */
+export declare function isStakingPrimitive(primitive: string): boolean;
+/**
+ * Check if a primitive is a liquidity operation.
+ */
+export declare function isLiquidityPrimitive(primitive: string): boolean;
+/**
+ * Check if a primitive is a transfer operation.
+ */
+export declare function isTransferPrimitive(primitive: string): boolean;

package/dist/primitives.js

@@ -0,0 +1,220 @@
+// Chaos AI SDK - Primitive Constants
+//
+// Constants for all DeFi primitive types supported by the SDK.
+// Use these when checking primitive types in TransactionActionBlocks.
+// ============================================================================
+// Primitive Type Constants
+// ============================================================================
+/**
+ * Swap tokens on a decentralized exchange.
+ * @example "Swap 1 ETH for USDC"
+ */
+export const PRIMITIVE_SWAP = 'swap';
+/**
+ * Supply assets to a lending protocol.
+ * @example "Supply 1000 USDC to Aave"
+ */
+export const PRIMITIVE_SUPPLY = 'supply';
+/**
+ * Withdraw supplied assets from a lending protocol.
+ * @example "Withdraw my USDC from Compound"
+ */
+export const PRIMITIVE_WITHDRAW = 'withdraw';
+/**
+ * Borrow assets from a lending protocol.
+ * @example "Borrow 500 DAI from Aave"
+ */
+export const PRIMITIVE_BORROW = 'borrow';
+/**
+ * Repay borrowed assets to a lending protocol.
+ * @example "Repay my DAI debt on Compound"
+ */
+export const PRIMITIVE_REPAY = 'repay';
+/**
+ * Stake tokens for rewards or governance.
+ * @example "Stake 10 ETH with Lido"
+ */
+export const PRIMITIVE_STAKE = 'stake';
+/**
+ * Unstake previously staked tokens.
+ * @example "Unstake my stETH"
+ */
+export const PRIMITIVE_UNSTAKE = 'unstake';
+/**
+ * Claim accrued rewards from protocols.
+ * @example "Claim my Aave rewards"
+ */
+export const PRIMITIVE_CLAIM = 'claim';
+/**
+ * Bridge assets between chains.
+ * @example "Bridge 100 USDC to Arbitrum"
+ */
+export const PRIMITIVE_BRIDGE = 'bridge';
+/**
+ * Provide liquidity to a pool.
+ * @example "Add liquidity to ETH-USDC pool"
+ */
+export const PRIMITIVE_ADD_LIQUIDITY = 'add_liquidity';
+/**
+ * Remove liquidity from a pool.
+ * @example "Remove liquidity from my ETH-USDC position"
+ */
+export const PRIMITIVE_REMOVE_LIQUIDITY = 'remove_liquidity';
+/**
+ * Open a perpetual futures position.
+ * @example "Open 2x long ETH position"
+ */
+export const PRIMITIVE_OPEN_POSITION = 'open_position';
+/**
+ * Close a perpetual futures position.
+ * @example "Close my ETH long"
+ */
+export const PRIMITIVE_CLOSE_POSITION = 'close_position';
+/**
+ * Deposit into a vault strategy.
+ * @example "Deposit 1000 USDC into Yearn"
+ */
+export const PRIMITIVE_DEPOSIT = 'deposit';
+/**
+ * Mint stablecoins or synthetic assets.
+ * @example "Mint 1000 DAI from my ETH collateral"
+ */
+export const PRIMITIVE_MINT = 'mint';
+/**
+ * Burn or repay minted assets.
+ * @example "Burn 500 DAI to reduce my debt"
+ */
+export const PRIMITIVE_BURN = 'burn';
+/**
+ * Restake assets for additional yield.
+ * @example "Restake my stETH on EigenLayer"
+ */
+export const PRIMITIVE_RESTAKE = 'restake';
+/**
+ * Approve token spending.
+ * @example "Approve USDC for Uniswap"
+ */
+export const PRIMITIVE_APPROVE = 'approve';
+/**
+ * Transfer tokens to another address.
+ * @example "Send 100 USDC to 0x..."
+ */
+export const PRIMITIVE_TRANSFER = 'transfer';
+/**
+ * Wrap native tokens (e.g., ETH to WETH).
+ * @example "Wrap 1 ETH to WETH"
+ */
+export const PRIMITIVE_WRAP = 'wrap';
+/**
+ * Unwrap wrapped tokens (e.g., WETH to ETH).
+ * @example "Unwrap 1 WETH to ETH"
+ */
+export const PRIMITIVE_UNWRAP = 'unwrap';
+/**
+ * All primitive type constants for iteration.
+ */
+export const ALL_PRIMITIVES = [
+    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,
+];
+/**
+ * Lending-related primitives.
+ */
+export const LENDING_PRIMITIVES = [
+    PRIMITIVE_SUPPLY,
+    PRIMITIVE_WITHDRAW,
+    PRIMITIVE_BORROW,
+    PRIMITIVE_REPAY,
+];
+/**
+ * Trading-related primitives.
+ */
+export const TRADING_PRIMITIVES = [
+    PRIMITIVE_SWAP,
+    PRIMITIVE_OPEN_POSITION,
+    PRIMITIVE_CLOSE_POSITION,
+];
+/**
+ * Staking-related primitives.
+ */
+export const STAKING_PRIMITIVES = [
+    PRIMITIVE_STAKE,
+    PRIMITIVE_UNSTAKE,
+    PRIMITIVE_RESTAKE,
+    PRIMITIVE_CLAIM,
+];
+/**
+ * Liquidity-related primitives.
+ */
+export const LIQUIDITY_PRIMITIVES = [
+    PRIMITIVE_ADD_LIQUIDITY,
+    PRIMITIVE_REMOVE_LIQUIDITY,
+    PRIMITIVE_DEPOSIT,
+];
+/**
+ * Transfer-related primitives.
+ */
+export const TRANSFER_PRIMITIVES = [
+    PRIMITIVE_BRIDGE,
+    PRIMITIVE_TRANSFER,
+    PRIMITIVE_WRAP,
+    PRIMITIVE_UNWRAP,
+];
+// ============================================================================
+// Primitive Helpers
+// ============================================================================
+/**
+ * Check if a primitive type is valid.
+ */
+export function isValidPrimitive(primitive) {
+    return ALL_PRIMITIVES.includes(primitive);
+}
+/**
+ * Check if a primitive is a lending operation.
+ */
+export function isLendingPrimitive(primitive) {
+    return LENDING_PRIMITIVES.includes(primitive);
+}
+/**
+ * Check if a primitive is a trading operation.
+ */
+export function isTradingPrimitive(primitive) {
+    return TRADING_PRIMITIVES.includes(primitive);
+}
+/**
+ * Check if a primitive is a staking operation.
+ */
+export function isStakingPrimitive(primitive) {
+    return STAKING_PRIMITIVES.includes(primitive);
+}
+/**
+ * Check if a primitive is a liquidity operation.
+ */
+export function isLiquidityPrimitive(primitive) {
+    return LIQUIDITY_PRIMITIVES.includes(primitive);
+}
+/**
+ * Check if a primitive is a transfer operation.
+ */
+export function isTransferPrimitive(primitive) {
+    return TRANSFER_PRIMITIVES.includes(primitive);
+}

package/dist/schemas.d.ts

@@ -237,9 +237,9 @@ export declare const TransactionGroupSchema: z.ZodObject<{
 export declare const RiskInfoItemSchema: z.ZodObject<{
     id: z.ZodOptional<z.ZodString>;
     severity: z.ZodOptional<z.ZodEnum<{
+        error: "error";
         info: "info";
         warning: "warning";
-        error: "error";
         critical: "critical";
     }>>;
     title: z.ZodOptional<z.ZodString>;
@@ -252,9 +252,9 @@ export declare const RisksSchema: z.ZodObject<{
     info: z.ZodOptional<z.ZodArray<z.ZodObject<{
         id: z.ZodOptional<z.ZodString>;
         severity: z.ZodOptional<z.ZodEnum<{
+            error: "error";
             info: "info";
             warning: "warning";
-            error: "error";
             critical: "critical";
         }>>;
         title: z.ZodOptional<z.ZodString>;
@@ -313,9 +313,9 @@ export declare const TransactionActionBlockSchema: z.ZodObject<{
         info: z.ZodOptional<z.ZodArray<z.ZodObject<{
             id: z.ZodOptional<z.ZodString>;
             severity: z.ZodOptional<z.ZodEnum<{
+                error: "error";
                 info: "info";
                 warning: "warning";
-                error: "error";
                 critical: "critical";
             }>>;
             title: z.ZodOptional<z.ZodString>;
@@ -625,9 +625,9 @@ export declare const BlockSchema: z.ZodUnion<readonly [z.ZodPipe<z.ZodObject<{
         info: z.ZodOptional<z.ZodArray<z.ZodObject<{
             id: z.ZodOptional<z.ZodString>;
             severity: z.ZodOptional<z.ZodEnum<{
+                error: "error";
                 info: "info";
                 warning: "warning";
-                error: "error";
                 critical: "critical";
             }>>;
             title: z.ZodOptional<z.ZodString>;
@@ -688,7 +688,7 @@ export declare const BlockSchema: z.ZodUnion<readonly [z.ZodPipe<z.ZodObject<{
         warnings?: string[] | undefined;
         info?: {
             id?: string | undefined;
-            severity?: "info" | "warning" | "error" | "critical" | undefined;
+            severity?: "error" | "info" | "warning" | "critical" | undefined;
             title?: string | undefined;
             message?: string | undefined;
         }[] | undefined;
@@ -748,7 +748,7 @@ export declare const BlockSchema: z.ZodUnion<readonly [z.ZodPipe<z.ZodObject<{
         warnings?: string[] | undefined;
         info?: {
             id?: string | undefined;
-            severity?: "info" | "warning" | "error" | "critical" | undefined;
+            severity?: "error" | "info" | "warning" | "critical" | undefined;
             title?: string | undefined;
             message?: string | undefined;
         }[] | undefined;

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.2",
+  "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",