@chaoslabs/ai-sdk 0.0.6 → 0.0.7

package/dist/__tests__/stream.test.d.ts

@@ -1 +0,0 @@
-export {};

package/dist/__tests__/stream.test.js

@@ -1,345 +0,0 @@
-import { describe, it, expect } from "bun:test";
-import { isAgentStatusMessage, isAgentMessage, isReportMessage, isFollowUpSuggestions, isUserInputMessage, parseAgentStatus, isTerminalStatus, extractAgentMessageText, extractSuggestions, extractReportBlock, parseStreamLine, parseStreamLines, } from "../stream";
-describe("isAgentStatusMessage", () => {
-    it("returns true for agent_status_change message", () => {
-        const msg = {
-            id: "1",
-            type: "agent_status_change",
-            timestamp: Date.now(),
-            content: { status: "processing" },
-            context: {
-                sessionId: "session-1",
-                artifactId: "artifact-1",
-            },
-        };
-        expect(isAgentStatusMessage(msg)).toBe(true);
-    });
-    it("returns false for other message types", () => {
-        const msg = {
-            id: "2",
-            type: "agent_message",
-            timestamp: Date.now(),
-            content: { messageId: "m1", data: { message: "Hello" } },
-            context: {
-                sessionId: "session-1",
-                artifactId: "artifact-1",
-            },
-        };
-        expect(isAgentStatusMessage(msg)).toBe(false);
-    });
-});
-describe("isAgentMessage", () => {
-    it("returns true for agent_message type", () => {
-        const msg = {
-            id: "1",
-            type: "agent_message",
-            timestamp: Date.now(),
-            content: { messageId: "m1", data: { message: "Hello" } },
-            context: { sessionId: "s1", artifactId: "a1" },
-        };
-        expect(isAgentMessage(msg)).toBe(true);
-    });
-    it("returns false for other message types", () => {
-        const msg = {
-            id: "1",
-            type: "report",
-            timestamp: Date.now(),
-            content: { id: "r1", type: "table", data: {} },
-            context: { sessionId: "s1", artifactId: "a1" },
-        };
-        expect(isAgentMessage(msg)).toBe(false);
-    });
-});
-describe("isReportMessage", () => {
-    it("returns true for report type", () => {
-        const msg = {
-            id: "1",
-            type: "report",
-            timestamp: Date.now(),
-            content: { id: "r1", type: "table", data: {} },
-            context: { sessionId: "s1", artifactId: "a1" },
-        };
-        expect(isReportMessage(msg)).toBe(true);
-    });
-    it("returns false for other message types", () => {
-        const msg = {
-            id: "1",
-            type: "agent_message",
-            timestamp: Date.now(),
-            content: { messageId: "m1", data: { message: "Hello" } },
-            context: { sessionId: "s1", artifactId: "a1" },
-        };
-        expect(isReportMessage(msg)).toBe(false);
-    });
-});
-describe("isFollowUpSuggestions", () => {
-    it("returns true for follow_up_suggestions type", () => {
-        const msg = {
-            id: "1",
-            type: "follow_up_suggestions",
-            timestamp: Date.now(),
-            content: { suggestions: ["Question 1?", "Question 2?"] },
-            context: { sessionId: "s1", artifactId: "a1" },
-        };
-        expect(isFollowUpSuggestions(msg)).toBe(true);
-    });
-    it("returns false for other message types", () => {
-        const msg = {
-            id: "1",
-            type: "agent_message",
-            timestamp: Date.now(),
-            content: { messageId: "m1", data: { message: "Hello" } },
-            context: { sessionId: "s1", artifactId: "a1" },
-        };
-        expect(isFollowUpSuggestions(msg)).toBe(false);
-    });
-});
-describe("isUserInputMessage", () => {
-    it("returns true for user_input type", () => {
-        const msg = {
-            id: "1",
-            type: "user_input",
-            timestamp: Date.now(),
-            content: { query: "What is my balance?" },
-            context: { sessionId: "s1", artifactId: "a1" },
-        };
-        expect(isUserInputMessage(msg)).toBe(true);
-    });
-    it("returns false for other message types", () => {
-        const msg = {
-            id: "1",
-            type: "agent_message",
-            timestamp: Date.now(),
-            content: { messageId: "m1", data: { message: "Hello" } },
-            context: { sessionId: "s1", artifactId: "a1" },
-        };
-        expect(isUserInputMessage(msg)).toBe(false);
-    });
-});
-describe("parseAgentStatus", () => {
-    it("extracts status from agent_status_change message", () => {
-        const msg = {
-            id: "1",
-            type: "agent_status_change",
-            timestamp: Date.now(),
-            content: { status: "processing" },
-            context: { sessionId: "s1", artifactId: "a1" },
-        };
-        expect(parseAgentStatus(msg)).toBe("processing");
-    });
-    it("extracts done status", () => {
-        const msg = {
-            id: "1",
-            type: "agent_status_change",
-            timestamp: Date.now(),
-            content: { status: "done" },
-            context: { sessionId: "s1", artifactId: "a1" },
-        };
-        expect(parseAgentStatus(msg)).toBe("done");
-    });
-    it("returns null for non-status messages", () => {
-        const msg = {
-            id: "1",
-            type: "agent_message",
-            timestamp: Date.now(),
-            content: { messageId: "m1", data: { message: "Hello" } },
-            context: { sessionId: "s1", artifactId: "a1" },
-        };
-        expect(parseAgentStatus(msg)).toBeNull();
-    });
-    it("returns null for invalid content", () => {
-        const msg = {
-            id: "1",
-            type: "agent_status_change",
-            timestamp: Date.now(),
-            content: { notStatus: "wrong" },
-            context: { sessionId: "s1", artifactId: "a1" },
-        };
-        expect(parseAgentStatus(msg)).toBeNull();
-    });
-});
-describe("isTerminalStatus", () => {
-    it("returns true for done status", () => {
-        expect(isTerminalStatus("done")).toBe(true);
-    });
-    it("returns true for error status", () => {
-        expect(isTerminalStatus("error")).toBe(true);
-    });
-    it("returns true for cancelled status", () => {
-        expect(isTerminalStatus("cancelled")).toBe(true);
-    });
-    it("returns false for processing status", () => {
-        expect(isTerminalStatus("processing")).toBe(false);
-    });
-});
-describe("extractAgentMessageText", () => {
-    it("extracts message text from agent_message", () => {
-        const msg = {
-            id: "1",
-            type: "agent_message",
-            timestamp: Date.now(),
-            content: { messageId: "m1", data: { message: "Hello world" } },
-            context: { sessionId: "s1", artifactId: "a1" },
-        };
-        expect(extractAgentMessageText(msg)).toBe("Hello world");
-    });
-    it("returns null for non-agent_message types", () => {
-        const msg = {
-            id: "1",
-            type: "report",
-            timestamp: Date.now(),
-            content: { id: "r1", type: "table", data: {} },
-            context: { sessionId: "s1", artifactId: "a1" },
-        };
-        expect(extractAgentMessageText(msg)).toBeNull();
-    });
-    it("returns null for invalid content structure", () => {
-        const msg = {
-            id: "1",
-            type: "agent_message",
-            timestamp: Date.now(),
-            content: { messageId: "m1" }, // missing data.message
-            context: { sessionId: "s1", artifactId: "a1" },
-        };
-        expect(extractAgentMessageText(msg)).toBeNull();
-    });
-});
-describe("extractSuggestions", () => {
-    it("extracts suggestions array from follow_up_suggestions", () => {
-        const msg = {
-            id: "1",
-            type: "follow_up_suggestions",
-            timestamp: Date.now(),
-            content: { suggestions: ["Question 1?", "Question 2?"] },
-            context: { sessionId: "s1", artifactId: "a1" },
-        };
-        expect(extractSuggestions(msg)).toEqual(["Question 1?", "Question 2?"]);
-    });
-    it("returns empty array for non-follow_up_suggestions types", () => {
-        const msg = {
-            id: "1",
-            type: "agent_message",
-            timestamp: Date.now(),
-            content: { messageId: "m1", data: { message: "Hello" } },
-            context: { sessionId: "s1", artifactId: "a1" },
-        };
-        expect(extractSuggestions(msg)).toEqual([]);
-    });
-    it("returns empty array for invalid content structure", () => {
-        const msg = {
-            id: "1",
-            type: "follow_up_suggestions",
-            timestamp: Date.now(),
-            content: { notSuggestions: "wrong" },
-            context: { sessionId: "s1", artifactId: "a1" },
-        };
-        expect(extractSuggestions(msg)).toEqual([]);
-    });
-});
-describe("extractReportBlock", () => {
-    it("extracts data from report message", () => {
-        const reportData = { columns: ["name"], rows: [["Alice"]] };
-        const msg = {
-            id: "1",
-            type: "report",
-            timestamp: Date.now(),
-            content: { id: "r1", type: "table", data: reportData },
-            context: { sessionId: "s1", artifactId: "a1" },
-        };
-        expect(extractReportBlock(msg)).toEqual(reportData);
-    });
-    it("returns null for non-report types", () => {
-        const msg = {
-            id: "1",
-            type: "agent_message",
-            timestamp: Date.now(),
-            content: { messageId: "m1", data: { message: "Hello" } },
-            context: { sessionId: "s1", artifactId: "a1" },
-        };
-        expect(extractReportBlock(msg)).toBeNull();
-    });
-    it("returns null for invalid content structure", () => {
-        const msg = {
-            id: "1",
-            type: "report",
-            timestamp: Date.now(),
-            content: { id: "r1", type: "table" }, // missing data
-            context: { sessionId: "s1", artifactId: "a1" },
-        };
-        expect(extractReportBlock(msg)).toBeNull();
-    });
-});
-describe("parseStreamLine", () => {
-    it("parses valid NDJSON line to StreamMessage", () => {
-        const msg = {
-            id: "1",
-            type: "agent_message",
-            timestamp: 1234567890,
-            content: { messageId: "m1", data: { message: "Hello" } },
-            context: { sessionId: "s1", artifactId: "a1" },
-        };
-        const line = JSON.stringify(msg);
-        const result = parseStreamLine(line);
-        expect(result).toEqual(msg);
-    });
-    it("returns null for empty line", () => {
-        expect(parseStreamLine("")).toBeNull();
-    });
-    it("returns null for whitespace-only line", () => {
-        expect(parseStreamLine("   ")).toBeNull();
-    });
-    it("returns null for invalid JSON", () => {
-        expect(parseStreamLine("{not valid json}")).toBeNull();
-    });
-    it("returns null for missing required fields", () => {
-        expect(parseStreamLine('{"id": "1"}')).toBeNull();
-    });
-});
-describe("parseStreamLines", () => {
-    it("parses multiple NDJSON lines", () => {
-        const msg1 = {
-            id: "1",
-            type: "agent_status_change",
-            timestamp: 1234567890,
-            content: { status: "processing" },
-            context: { sessionId: "s1", artifactId: "a1" },
-        };
-        const msg2 = {
-            id: "2",
-            type: "agent_message",
-            timestamp: 1234567891,
-            content: { messageId: "m1", data: { message: "Hello" } },
-            context: { sessionId: "s1", artifactId: "a1" },
-        };
-        const text = JSON.stringify(msg1) + "\n" + JSON.stringify(msg2);
-        const result = parseStreamLines(text);
-        expect(result).toEqual([msg1, msg2]);
-    });
-    it("returns empty array for empty string", () => {
-        expect(parseStreamLines("")).toEqual([]);
-    });
-    it("skips invalid lines and returns valid ones", () => {
-        const validMsg = {
-            id: "1",
-            type: "agent_message",
-            timestamp: 1234567890,
-            content: { messageId: "m1", data: { message: "Hello" } },
-            context: { sessionId: "s1", artifactId: "a1" },
-        };
-        const text = "invalid json\n" + JSON.stringify(validMsg) + "\n{broken}";
-        const result = parseStreamLines(text);
-        expect(result).toEqual([validMsg]);
-    });
-    it("handles lines with various whitespace", () => {
-        const msg = {
-            id: "1",
-            type: "agent_message",
-            timestamp: 1234567890,
-            content: { messageId: "m1", data: { message: "Hello" } },
-            context: { sessionId: "s1", artifactId: "a1" },
-        };
-        const text = "\n\n" + JSON.stringify(msg) + "\n\n";
-        const result = parseStreamLines(text);
-        expect(result).toEqual([msg]);
-    });
-});

package/dist/__tests__/trivial.test.d.ts

@@ -1 +0,0 @@
-export {};

package/dist/__tests__/trivial.test.js

@@ -1,6 +0,0 @@
-import { describe, it, expect } from 'bun:test';
-describe('test infrastructure', () => {
-    it('works', () => {
-        expect(1 + 1).toBe(2);
-    });
-});

package/dist/blocks.d.ts

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