@chaoslabs/ai-sdk 0.0.1 → 0.0.3

package/dist/blocks.d.ts

@@ -0,0 +1,108 @@
+import type { Block, TableBlock, ChartBlock, TransactionActionBlock, MarkdownBlock, InteractiveBlock, Primitive, Response } from './types.js';
+/**
+ * Extract all table blocks from a response.
+ */
+export declare function extractTableBlocks(response: Response): TableBlock[];
+/**
+ * Extract all chart blocks from a response.
+ */
+export declare function extractChartBlocks(response: Response): ChartBlock[];
+/**
+ * Extract all transaction action blocks from a response.
+ */
+export declare function extractTransactionBlocks(response: Response): TransactionActionBlock[];
+/**
+ * Extract all markdown blocks from a response.
+ */
+export declare function extractMarkdownBlocks(response: Response): MarkdownBlock[];
+/**
+ * Extract all interactive blocks from a response.
+ */
+export declare function extractInteractiveBlocks(response: Response): InteractiveBlock[];
+/**
+ * Find the first table block with a matching title.
+ */
+export declare function findTableByTitle(response: Response, title: string): TableBlock | undefined;
+/**
+ * Find the first chart block with a matching title.
+ */
+export declare function findChartByTitle(response: Response, title: string): ChartBlock | undefined;
+/**
+ * Find transaction blocks containing a specific primitive type.
+ */
+export declare function findTransactionsByPrimitive(response: Response, primitiveType: string): TransactionActionBlock[];
+/**
+ * Extract all primitives from all transaction blocks.
+ */
+export declare function extractPrimitives(response: Response): Primitive[];
+/**
+ * Extract primitives of a specific type.
+ */
+export declare function extractPrimitivesByType(response: Response, primitiveType: string): Primitive[];
+/**
+ * Get unique primitive types from a response.
+ */
+export declare function getPrimitiveTypes(response: Response): string[];
+/**
+ * Convert a table block to an array of objects.
+ * Each object has keys from the table headers.
+ */
+export declare function tableToObjects<T extends Record<string, unknown>>(table: TableBlock): T[];
+/**
+ * Get a specific column from a table block.
+ */
+export declare function getTableColumn(table: TableBlock, columnName: string): unknown[];
+/**
+ * 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;
+/**
+ * Get table dimensions.
+ */
+export declare function getTableDimensions(table: TableBlock): {
+    columns: number;
+    rows: number;
+};
+/**
+ * Get chart data as label-value pairs.
+ * Works with both segments and legacy data formats.
+ */
+export declare function getChartData(chart: ChartBlock): Array<{
+    label: string;
+    value: number;
+}>;
+/**
+ * Get the total value of all chart segments.
+ */
+export declare function getChartTotal(chart: ChartBlock): number;
+/**
+ * Get chart segment percentages.
+ */
+export declare function getChartPercentages(chart: ChartBlock): Array<{
+    label: string;
+    percentage: number;
+}>;
+/**
+ * Get all warnings from transaction blocks.
+ */
+export declare function getAllWarnings(response: Response): string[];
+/**
+ * Get all blockers from transaction blocks.
+ */
+export declare function getAllBlockers(response: Response): string[];
+/**
+ * Get the highest risk level from all transaction blocks.
+ */
+export declare function getHighestRiskLevel(response: Response): string | undefined;
+/**
+ * Count blocks by type in a response.
+ */
+export declare function countBlocksByType(response: Response): Record<string, number>;
+/**
+ * Check if a response contains any blocks.
+ */
+export declare function hasBlocks(response: Response): boolean;
+/**
+ * Check if a response contains a specific block type.
+ */
+export declare function hasBlockType(response: Response, type: Block['type']): boolean;

package/dist/blocks.js

@@ -0,0 +1,246 @@
+// Chaos AI SDK - Block Utilities
+//
+// Helper functions for working with blocks extracted from responses.
+import { extractBlocks, isTableBlock, isChartBlock, isTransactionActionBlock, isMarkdownBlock, isInteractiveBlock, } from './types.js';
+// ============================================================================
+// Block Extraction Utilities
+// ============================================================================
+/**
+ * Extract all table blocks from a response.
+ */
+export function extractTableBlocks(response) {
+    return extractBlocks(response).filter(isTableBlock);
+}
+/**
+ * Extract all chart blocks from a response.
+ */
+export function extractChartBlocks(response) {
+    return extractBlocks(response).filter(isChartBlock);
+}
+/**
+ * Extract all transaction action blocks from a response.
+ */
+export function extractTransactionBlocks(response) {
+    return extractBlocks(response).filter(isTransactionActionBlock);
+}
+/**
+ * Extract all markdown blocks from a response.
+ */
+export function extractMarkdownBlocks(response) {
+    return extractBlocks(response).filter(isMarkdownBlock);
+}
+/**
+ * Extract all interactive blocks from a response.
+ */
+export function extractInteractiveBlocks(response) {
+    return extractBlocks(response).filter(isInteractiveBlock);
+}
+// ============================================================================
+// Block Search Utilities
+// ============================================================================
+/**
+ * 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()));
+}
+/**
+ * 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()));
+}
+/**
+ * Find transaction blocks containing a specific primitive type.
+ */
+export function findTransactionsByPrimitive(response, primitiveType) {
+    return extractTransactionBlocks(response).filter((block) => block.primitives?.some((p) => p.primitive === primitiveType));
+}
+// ============================================================================
+// Primitive Extraction
+// ============================================================================
+/**
+ * Extract all primitives from all transaction blocks.
+ */
+export function extractPrimitives(response) {
+    const primitives = [];
+    for (const block of extractTransactionBlocks(response)) {
+        if (block.primitives) {
+            primitives.push(...block.primitives);
+        }
+    }
+    return primitives;
+}
+/**
+ * Extract primitives of a specific type.
+ */
+export function extractPrimitivesByType(response, primitiveType) {
+    return extractPrimitives(response).filter((p) => p.primitive === primitiveType);
+}
+/**
+ * Get unique primitive types from a response.
+ */
+export function getPrimitiveTypes(response) {
+    const types = new Set();
+    for (const primitive of extractPrimitives(response)) {
+        types.add(primitive.primitive);
+    }
+    return Array.from(types);
+}
+// ============================================================================
+// Table Utilities
+// ============================================================================
+/**
+ * Convert a table block to an array of objects.
+ * Each object has keys from the table headers.
+ */
+export function tableToObjects(table) {
+    const { tableHeaders, tableRows } = table;
+    return tableRows.map((row) => {
+        const obj = {};
+        for (let i = 0; i < tableHeaders.length; i++) {
+            obj[tableHeaders[i]] = row[i];
+        }
+        return obj;
+    });
+}
+/**
+ * Get a specific column from a table block.
+ */
+export function getTableColumn(table, columnName) {
+    const columnIndex = table.tableHeaders.indexOf(columnName);
+    if (columnIndex === -1)
+        return [];
+    return table.tableRows.map((row) => row[columnIndex]);
+}
+/**
+ * Find a row in a table by matching a column value.
+ */
+export function findTableRow(table, columnName, value) {
+    const columnIndex = table.tableHeaders.indexOf(columnName);
+    if (columnIndex === -1)
+        return undefined;
+    return table.tableRows.find((row) => row[columnIndex] === value);
+}
+/**
+ * Get table dimensions.
+ */
+export function getTableDimensions(table) {
+    return {
+        columns: table.tableHeaders.length,
+        rows: table.tableRows.length,
+    };
+}
+// ============================================================================
+// Chart Utilities
+// ============================================================================
+/**
+ * Get chart data as label-value pairs.
+ * 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) {
+        return chart.data.map(([label, value]) => ({ label, value }));
+    }
+    return [];
+}
+/**
+ * Get the total value of all chart segments.
+ */
+export function getChartTotal(chart) {
+    return getChartData(chart).reduce((sum, item) => sum + item.value, 0);
+}
+/**
+ * Get chart segment percentages.
+ */
+export function getChartPercentages(chart) {
+    const data = getChartData(chart);
+    const total = data.reduce((sum, item) => sum + item.value, 0);
+    if (total === 0)
+        return data.map((d) => ({ label: d.label, percentage: 0 }));
+    return data.map((d) => ({
+        label: d.label,
+        percentage: (d.value / total) * 100,
+    }));
+}
+// ============================================================================
+// Risk Utilities
+// ============================================================================
+/**
+ * Get all warnings from transaction blocks.
+ */
+export function getAllWarnings(response) {
+    const warnings = [];
+    for (const block of extractTransactionBlocks(response)) {
+        if (block.risks?.warnings) {
+            warnings.push(...block.risks.warnings);
+        }
+    }
+    return warnings;
+}
+/**
+ * Get all blockers from transaction blocks.
+ */
+export function getAllBlockers(response) {
+    const blockers = [];
+    for (const block of extractTransactionBlocks(response)) {
+        if (block.risks?.blockers) {
+            blockers.push(...block.risks.blockers);
+        }
+    }
+    return blockers;
+}
+/**
+ * Get the highest risk level from all transaction blocks.
+ */
+export function getHighestRiskLevel(response) {
+    const riskLevels = ['low', 'medium', 'high', 'critical'];
+    let highestIndex = -1;
+    let highestLevel;
+    for (const block of extractTransactionBlocks(response)) {
+        if (block.risks?.level) {
+            const index = riskLevels.indexOf(block.risks.level.toLowerCase());
+            if (index > highestIndex) {
+                highestIndex = index;
+                highestLevel = block.risks.level;
+            }
+        }
+    }
+    return highestLevel;
+}
+// ============================================================================
+// Block Statistics
+// ============================================================================
+/**
+ * Count blocks by type in a response.
+ */
+export function countBlocksByType(response) {
+    const counts = {
+        table: 0,
+        chart: 0,
+        transaction_action: 0,
+        markdown: 0,
+        interactive: 0,
+    };
+    for (const block of extractBlocks(response)) {
+        if (block.type in counts) {
+            counts[block.type]++;
+        }
+    }
+    return counts;
+}
+/**
+ * Check if a response contains any blocks.
+ */
+export function hasBlocks(response) {
+    return extractBlocks(response).length > 0;
+}
+/**
+ * Check if a response contains a specific block type.
+ */
+export function hasBlockType(response, type) {
+    return extractBlocks(response).some((block) => block.type === type);
+}

package/dist/conversation.d.ts

@@ -0,0 +1,136 @@
+import type { Chaos } from './client.js';
+import type { InputItem, Response, RequestMetadata } from './types.js';
+/**
+ * Options for creating a new conversation.
+ */
+export interface ConversationOptions {
+    /** Model to use (defaults to WALLET_MODEL) */
+    model?: string;
+    /** Maximum number of messages to keep in history */
+    maxHistoryLength?: number;
+    /** User identifier */
+    userId: string;
+    /** Wallet identifier */
+    walletId: string;
+    /** Optional initial session ID (auto-generated if not provided) */
+    sessionId?: string;
+}
+/**
+ * Statistics about the conversation.
+ */
+export interface ConversationStats {
+    /** Number of user turns */
+    userTurns: number;
+    /** Number of assistant turns */
+    assistantTurns: number;
+    /** Total messages in history */
+    totalMessages: number;
+    /** Session ID */
+    sessionId: string;
+    /** When the conversation started */
+    startedAt: Date;
+    /** When the last message was sent */
+    lastMessageAt: Date | null;
+}
+/**
+ * Manages a multi-turn conversation with history tracking.
+ *
+ * @example
+ * ```typescript
+ * import { Chaos, Conversation } from '@chaoslabs/ai-sdk';
+ *
+ * const chaos = new Chaos({ apiKey: 'ck-...' });
+ * const conversation = new Conversation(chaos, {
+ *   userId: 'user-123',
+ *   walletId: '0x...',
+ * });
+ *
+ * // Send messages - history is tracked automatically
+ * const response1 = await conversation.send("What's my portfolio value?");
+ * const response2 = await conversation.send("Which asset has the highest allocation?");
+ *
+ * // Get conversation stats
+ * console.log(conversation.stats);
+ *
+ * // Reset for a new conversation
+ * conversation.reset();
+ * ```
+ */
+export declare class Conversation {
+    private client;
+    private model;
+    private maxHistoryLength;
+    private history;
+    private metadata;
+    private userTurns;
+    private assistantTurns;
+    private startedAt;
+    private lastMessageAt;
+    constructor(client: Chaos, options: ConversationOptions);
+    /**
+     * Get the current session ID.
+     */
+    get sessionId(): string;
+    /**
+     * Get the conversation history.
+     */
+    get messages(): readonly InputItem[];
+    /**
+     * Get conversation statistics.
+     */
+    get stats(): ConversationStats;
+    /**
+     * Send a message and get a response.
+     * Automatically manages conversation history.
+     */
+    send(message: string): Promise<Response>;
+    /**
+     * Add a user message to the history without sending.
+     * Useful for restoring conversation state.
+     */
+    addUserMessage(content: string): void;
+    /**
+     * Add an assistant message to the history.
+     * Useful for restoring conversation state.
+     */
+    addAssistantMessage(content: string): void;
+    /**
+     * Add a system message to the history.
+     */
+    addSystemMessage(content: string): void;
+    /**
+     * Reset the conversation to start fresh.
+     * Generates a new session ID and clears history.
+     */
+    reset(): void;
+    /**
+     * Clear the history but keep the same session ID.
+     */
+    clearHistory(): void;
+    /**
+     * Fork this conversation to create a branch.
+     * The new conversation has the same history but a new session ID.
+     */
+    fork(): Conversation;
+    /**
+     * Serialize the conversation to JSON for persistence.
+     */
+    toJSON(): {
+        sessionId: string;
+        history: InputItem[];
+        metadata: RequestMetadata;
+        stats: ConversationStats;
+    };
+    /**
+     * Restore conversation state from JSON.
+     */
+    static fromJSON(client: Chaos, data: {
+        sessionId: string;
+        history: InputItem[];
+        metadata: RequestMetadata;
+    }): Conversation;
+    /**
+     * Trim history to stay within maxHistoryLength.
+     */
+    private trimHistory;
+}