@chaoslabs/ai-sdk 0.0.1

package/dist/request.d.ts

@@ -0,0 +1,48 @@
+import type { CreateResponseParams } from './types.js';
+/** Model ID for wallet mode (v1 API) */
+export declare const WALLET_MODEL = "WALLET_MODEL";
+/** Model ID for ask mode (v1 API) */
+export declare const ASK_MODEL = "ASK_MODEL";
+/**
+ * V1 API request body for wallet mode.
+ */
+export interface V1WalletRequest {
+    model: string;
+    query: string;
+    user_id: string;
+    session_id: string;
+    wallet_id: string;
+    prior_messages?: {
+        role: string;
+        content: string;
+    }[];
+}
+/**
+ * V1 API request body for ask mode.
+ */
+export interface V1AskRequest {
+    model: string;
+    query: string;
+    user_id: string;
+    session_id: string;
+    prior_messages?: {
+        role: string;
+        content: string;
+    }[];
+}
+/**
+ * Transform SDK request to v1 wallet format.
+ */
+export declare function toV1WalletRequest(params: CreateResponseParams, sessionId: string): V1WalletRequest;
+/**
+ * Transform SDK request to v1 ask format.
+ */
+export declare function toV1AskRequest(params: CreateResponseParams, sessionId: string): V1AskRequest;
+/**
+ * Build HTTP headers for v1 API requests (Bearer auth).
+ */
+export declare function buildV1Headers(apiKey: string): Record<string, string>;
+/**
+ * Build the v1 endpoint URL.
+ */
+export declare function buildV1Endpoint(baseUrl: string): string;

package/dist/request.js

@@ -0,0 +1,93 @@
+// V1 API Request Adapter for /v1/chat/stream
+// ============================================================================
+// Model Constants
+// ============================================================================
+/** Model ID for wallet mode (v1 API) */
+export const WALLET_MODEL = 'WALLET_MODEL';
+/** Model ID for ask mode (v1 API) */
+export const ASK_MODEL = 'ASK_MODEL';
+// ============================================================================
+// Request Builders
+// ============================================================================
+/**
+ * Transform SDK request to v1 wallet format.
+ */
+export function toV1WalletRequest(params, sessionId) {
+    const query = extractQuery(params.input);
+    const priorMessages = extractPriorMessages(params.input);
+    return {
+        model: params.model,
+        query,
+        user_id: params.metadata.user_id,
+        session_id: sessionId,
+        wallet_id: params.metadata.wallet_id || '',
+        prior_messages: priorMessages.length > 0 ? priorMessages : undefined,
+    };
+}
+/**
+ * Transform SDK request to v1 ask format.
+ */
+export function toV1AskRequest(params, sessionId) {
+    const query = extractQuery(params.input);
+    const priorMessages = extractPriorMessages(params.input);
+    return {
+        model: params.model,
+        query,
+        user_id: params.metadata.user_id,
+        session_id: sessionId,
+        prior_messages: priorMessages.length > 0 ? priorMessages : undefined,
+    };
+}
+// ============================================================================
+// Helper Functions
+// ============================================================================
+/**
+ * Extract the main query from input items.
+ */
+function extractQuery(input) {
+    for (let i = input.length - 1; i >= 0; i--) {
+        if (input[i].role === 'user') {
+            return input[i].content;
+        }
+    }
+    return '';
+}
+/**
+ * Extract prior messages from input items (excluding the last user message).
+ */
+function extractPriorMessages(input) {
+    const messages = [];
+    let lastUserIndex = -1;
+    for (let i = input.length - 1; i >= 0; i--) {
+        if (input[i].role === 'user') {
+            lastUserIndex = i;
+            break;
+        }
+    }
+    for (let i = 0; i < lastUserIndex; i++) {
+        const item = input[i];
+        if (item.type === 'message' && (item.role === 'user' || item.role === 'system')) {
+            messages.push({
+                role: item.role === 'system' ? 'system' : item.role,
+                content: item.content,
+            });
+        }
+    }
+    return messages;
+}
+/**
+ * Build HTTP headers for v1 API requests (Bearer auth).
+ */
+export function buildV1Headers(apiKey) {
+    return {
+        'Content-Type': 'application/json',
+        'Authorization': `Bearer ${apiKey}`,
+    };
+}
+/**
+ * Build the v1 endpoint URL.
+ */
+export function buildV1Endpoint(baseUrl) {
+    const base = baseUrl.replace(/\/$/, '');
+    return `${base}/v1/chat/stream`;
+}

package/dist/response.d.ts

@@ -0,0 +1,24 @@
+import type { Response } from './types.js';
+/**
+ * V1 streaming event from NDJSON stream.
+ */
+export interface V1StreamEvent {
+    type?: string;
+    content?: Record<string, unknown>;
+    [key: string]: unknown;
+}
+/**
+ * Final state collected from v1 stream.
+ */
+export interface V1FinalState {
+    blocks: unknown[];
+}
+/**
+ * Parse v1 NDJSON stream and extract blocks.
+ * Handles type: "report" events with content.data.
+ */
+export declare function parseV1Stream(body: ReadableStream<Uint8Array>, onEvent?: (event: V1StreamEvent) => void): Promise<V1FinalState>;
+/**
+ * Convert v1 final state to SDK Response format.
+ */
+export declare function toV1Response(responseId: string, model: string, finalState: V1FinalState): Response;

package/dist/response.js

@@ -0,0 +1,111 @@
+// V1 API Response Parser
+//
+// Parses responses from /v1/chat/stream endpoint.
+// Handles type: "report" events with content.data blocks.
+import { convertBlock } from './converter.js';
+// ============================================================================
+// Response Parser
+// ============================================================================
+/**
+ * Parse v1 NDJSON stream and extract blocks.
+ * Handles type: "report" events with content.data.
+ */
+export async function parseV1Stream(body, onEvent) {
+    const reader = body.getReader();
+    const decoder = new TextDecoder();
+    let buffer = '';
+    const blocks = [];
+    try {
+        while (true) {
+            const { done, value } = await reader.read();
+            if (done)
+                break;
+            buffer += decoder.decode(value, { stream: true });
+            const lines = buffer.split('\n');
+            buffer = lines.pop() || '';
+            for (const line of lines) {
+                if (!line.trim())
+                    continue;
+                try {
+                    const event = JSON.parse(line);
+                    // Handle report events with blocks
+                    if (event.type === 'report' && event.content) {
+                        const content = event.content;
+                        // Extract the data from the report
+                        if (content.data) {
+                            blocks.push(content.data);
+                        }
+                    }
+                    // Call event callback
+                    if (onEvent) {
+                        onEvent(event);
+                    }
+                }
+                catch {
+                    // Skip malformed JSON
+                }
+            }
+        }
+        // Process remaining buffer
+        if (buffer.trim()) {
+            try {
+                const event = JSON.parse(buffer);
+                if (event.type === 'report' && event.content) {
+                    const content = event.content;
+                    if (content.data) {
+                        blocks.push(content.data);
+                    }
+                }
+            }
+            catch {
+                // Skip malformed JSON
+            }
+        }
+    }
+    finally {
+        reader.releaseLock();
+    }
+    return { blocks };
+}
+/**
+ * Convert v1 final state to SDK Response format.
+ */
+export function toV1Response(responseId, model, finalState) {
+    const content = [];
+    // Process blocks from final state
+    if (finalState.blocks && Array.isArray(finalState.blocks)) {
+        for (const rawBlock of finalState.blocks) {
+            if (typeof rawBlock === 'string') {
+                // Text content
+                content.push({
+                    type: 'output_text',
+                    text: rawBlock,
+                });
+            }
+            else if (rawBlock && typeof rawBlock === 'object') {
+                // Block content - use existing converter
+                const block = convertBlock(rawBlock);
+                if (block) {
+                    content.push({
+                        type: 'chaos.block',
+                        block,
+                    });
+                }
+            }
+        }
+    }
+    const output = content.length > 0
+        ? [{
+                type: 'message',
+                role: 'assistant',
+                content,
+            }]
+        : [];
+    return {
+        id: responseId,
+        object: 'response',
+        model,
+        status: 'completed',
+        output,
+    };
+}