@chaoslabs/ai-sdk 0.0.6 → 0.0.7

package/dist/blocks.js

@@ -1,246 +0,0 @@
-// 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

@@ -1,136 +0,0 @@
-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;
-}

package/dist/conversation.js

@@ -1,230 +0,0 @@
-// Chaos AI SDK - Conversation Management
-//
-// A class for managing multi-turn conversations with history tracking.
-import { extractText } from './types.js';
-import { WALLET_MODEL } from './request.js';
-/**
- * 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 class Conversation {
-    client;
-    model;
-    maxHistoryLength;
-    history = [];
-    metadata;
-    userTurns = 0;
-    assistantTurns = 0;
-    startedAt;
-    lastMessageAt = null;
-    constructor(client, options) {
-        this.client = client;
-        this.model = options.model || WALLET_MODEL;
-        this.maxHistoryLength = options.maxHistoryLength || 50;
-        this.startedAt = new Date();
-        this.metadata = {
-            user_id: options.userId,
-            wallet_id: options.walletId,
-            session_id: options.sessionId || `session-${Date.now()}-${Math.random().toString(36).slice(2, 8)}`,
-        };
-    }
-    /**
-     * Get the current session ID.
-     */
-    get sessionId() {
-        return this.metadata.session_id;
-    }
-    /**
-     * Get the conversation history.
-     */
-    get messages() {
-        return [...this.history];
-    }
-    /**
-     * Get conversation statistics.
-     */
-    get stats() {
-        return {
-            userTurns: this.userTurns,
-            assistantTurns: this.assistantTurns,
-            totalMessages: this.history.length,
-            sessionId: this.metadata.session_id,
-            startedAt: this.startedAt,
-            lastMessageAt: this.lastMessageAt,
-        };
-    }
-    /**
-     * Send a message and get a response.
-     * Automatically manages conversation history.
-     */
-    async send(message) {
-        // Add user message to history
-        this.addUserMessage(message);
-        // Send request with full history
-        const response = await this.client.chat.responses.create({
-            model: this.model,
-            input: [...this.history],
-            metadata: this.metadata,
-        });
-        // Add assistant response to history if successful
-        if (response.status === 'completed') {
-            const text = extractText(response);
-            if (text) {
-                this.addAssistantMessage(text);
-            }
-        }
-        this.lastMessageAt = new Date();
-        return response;
-    }
-    /**
-     * Add a user message to the history without sending.
-     * Useful for restoring conversation state.
-     */
-    addUserMessage(content) {
-        this.history.push({
-            type: 'message',
-            role: 'user',
-            content,
-        });
-        this.userTurns++;
-        this.trimHistory();
-    }
-    /**
-     * Add an assistant message to the history.
-     * Useful for restoring conversation state.
-     */
-    addAssistantMessage(content) {
-        // Store assistant responses as user messages with prefix for context
-        // This matches the pattern used in the multi-turn example
-        this.history.push({
-            type: 'message',
-            role: 'user',
-            content: `[Assistant]: ${content}`,
-        });
-        this.assistantTurns++;
-        this.trimHistory();
-    }
-    /**
-     * Add a system message to the history.
-     */
-    addSystemMessage(content) {
-        this.history.push({
-            type: 'message',
-            role: 'system',
-            content,
-        });
-        this.trimHistory();
-    }
-    /**
-     * Reset the conversation to start fresh.
-     * Generates a new session ID and clears history.
-     */
-    reset() {
-        this.history = [];
-        this.userTurns = 0;
-        this.assistantTurns = 0;
-        this.startedAt = new Date();
-        this.lastMessageAt = null;
-        this.metadata.session_id = `session-${Date.now()}-${Math.random().toString(36).slice(2, 8)}`;
-    }
-    /**
-     * Clear the history but keep the same session ID.
-     */
-    clearHistory() {
-        this.history = [];
-        this.userTurns = 0;
-        this.assistantTurns = 0;
-    }
-    /**
-     * Fork this conversation to create a branch.
-     * The new conversation has the same history but a new session ID.
-     */
-    fork() {
-        const forked = new Conversation(this.client, {
-            model: this.model,
-            maxHistoryLength: this.maxHistoryLength,
-            userId: this.metadata.user_id,
-            walletId: this.metadata.wallet_id,
-        });
-        // Copy history
-        for (const msg of this.history) {
-            if (msg.role === 'user') {
-                if (msg.content.startsWith('[Assistant]: ')) {
-                    forked.history.push(msg);
-                    forked.assistantTurns++;
-                }
-                else {
-                    forked.history.push(msg);
-                    forked.userTurns++;
-                }
-            }
-            else {
-                forked.history.push(msg);
-            }
-        }
-        return forked;
-    }
-    /**
-     * Serialize the conversation to JSON for persistence.
-     */
-    toJSON() {
-        return {
-            sessionId: this.metadata.session_id,
-            history: [...this.history],
-            metadata: { ...this.metadata },
-            stats: this.stats,
-        };
-    }
-    /**
-     * Restore conversation state from JSON.
-     */
-    static fromJSON(client, data) {
-        const conversation = new Conversation(client, {
-            userId: data.metadata.user_id,
-            walletId: data.metadata.wallet_id,
-            sessionId: data.sessionId,
-        });
-        // Restore history
-        for (const msg of data.history) {
-            conversation.history.push(msg);
-            if (msg.role === 'user') {
-                if (msg.content.startsWith('[Assistant]: ')) {
-                    conversation.assistantTurns++;
-                }
-                else {
-                    conversation.userTurns++;
-                }
-            }
-        }
-        return conversation;
-    }
-    /**
-     * Trim history to stay within maxHistoryLength.
-     */
-    trimHistory() {
-        if (this.history.length > this.maxHistoryLength) {
-            const excess = this.history.length - this.maxHistoryLength;
-            this.history = this.history.slice(excess);
-        }
-    }
-}