ai-streamui 0.1.0

package/dist/index.d.ts

@@ -0,0 +1,136 @@
+import { UIMessage, ChatTransport, ChatRequestOptions, UIMessageChunk, ModelMessage } from 'ai';
+import * as _json_render_core from '@json-render/core';
+import { Spec, SchemaDefinition, InferCatalogInput, Schema, Catalog, UserPromptOptions } from '@json-render/core';
+export { Catalog, Schema, Spec, defineSchema } from '@json-render/core';
+
+interface UITextData {
+    segmentId: string;
+    text: string;
+}
+/**
+ * UIDataTypes for useChat generic parameter.
+ * Enables type-safe access to UI data in message.parts.
+ */
+interface UIDataTypes {
+    ui: Spec;
+    "ui-text": UITextData;
+    [key: string]: unknown;
+}
+
+/**
+ * Options for UITransport.
+ */
+interface UITransportOptions {
+    /** API endpoint URL. Defaults to '/api/chat'. */
+    api?: string;
+    /** Additional headers for requests. */
+    headers?: Record<string, string> | (() => Record<string, string> | Promise<Record<string, string>>);
+    /** Additional body properties for requests. */
+    body?: Record<string, unknown>;
+    /** Request credentials mode. */
+    credentials?: RequestCredentials;
+    /** Custom fetch implementation. */
+    fetch?: typeof fetch;
+}
+/**
+ * UITransport - Custom ChatTransport for streaming UI with SpecStream patches.
+ *
+ * Extends DefaultChatTransport behavior:
+ * - Parses UIMessageStream (JSON event stream)
+ * - Extracts SpecStream patches from text-delta chunks
+ * - Builds Spec incrementally and emits `data-ui` chunks
+ * - Passes through all other chunk types (tool, reasoning, etc.)
+ */
+declare class UITransport<UI_MESSAGE extends UIMessage = UIMessage> implements ChatTransport<UI_MESSAGE> {
+    private api;
+    private headers?;
+    private body?;
+    private credentials?;
+    private customFetch?;
+    constructor(options?: UITransportOptions);
+    sendMessages(options: {
+        trigger: "submit-message" | "regenerate-message";
+        chatId: string;
+        messageId: string | undefined;
+        messages: UI_MESSAGE[];
+        abortSignal: AbortSignal | undefined;
+    } & ChatRequestOptions): Promise<ReadableStream<UIMessageChunk<unknown, UIDataTypes>>>;
+    /**
+     * Reconnect to an existing stream is not supported.
+     */
+    reconnectToStream(_options: {
+        chatId: string;
+    } & ChatRequestOptions): Promise<ReadableStream<UIMessageChunk<unknown, UIDataTypes>> | null>;
+    /**
+     * Process the UIMessageStream response:
+     * - Parse JSON event stream (same format as DefaultChatTransport)
+     * - Extract SpecStream patches from text-delta chunks
+     * - Build Spec incrementally and emit data-ui chunks
+     * - Pass through all other chunk types unchanged
+     */
+    private processResponseStream;
+}
+
+/**
+ * Convert UI messages to model messages.
+ * Automatically converts data-ui parts to text descriptions for LLM context.
+ *
+ * @example
+ * ```ts
+ * const modelMessages = await convertUIMessagesToModelMessages(messages);
+ * const result = streamText({ model, messages: modelMessages });
+ * ```
+ */
+declare function convertUIMessagesToModelMessages(messages: UIMessage<unknown, UIDataTypes>[]): Promise<ModelMessage[]>;
+
+/**
+ * Default schema for streamui - Element Tree format.
+ *
+ * Defines the Spec shape (what AI generates) and the Catalog shape
+ * (what components provide). Compatible with @json-render/react's Renderer.
+ *
+ * Spec structure:
+ * ```json
+ * {
+ *   "root": "card-1",
+ *   "state": { "count": 0 },
+ *   "elements": {
+ *     "card-1": { "type": "Card", "props": {...}, "children": [...], "on": {...}, "repeat": {...} }
+ *   }
+ * }
+ * ```
+ */
+declare const defaultSchema: Schema<{
+    spec: _json_render_core.SchemaType<"object", {
+        root: _json_render_core.SchemaType<"string", unknown>;
+        elements: _json_render_core.SchemaType<"record", _json_render_core.SchemaType<"object", {
+            type: _json_render_core.SchemaType<"ref", string>;
+            props: _json_render_core.SchemaType<"propsOf", string>;
+            children: _json_render_core.SchemaType<"array", _json_render_core.SchemaType<"string", unknown>>;
+            visible: _json_render_core.SchemaType<"any", unknown>;
+            on: _json_render_core.SchemaType<"any", unknown>;
+            repeat: _json_render_core.SchemaType<"any", unknown>;
+        }>>;
+        state: _json_render_core.SchemaType<"any", unknown>;
+    }>;
+    catalog: _json_render_core.SchemaType<"object", {
+        components: _json_render_core.SchemaType<"map", {
+            props: _json_render_core.SchemaType<"zod", unknown>;
+            slots: _json_render_core.SchemaType<"array", _json_render_core.SchemaType<"string", unknown>>;
+            description: _json_render_core.SchemaType<"string", unknown>;
+        }>;
+        actions: _json_render_core.SchemaType<"map", {
+            params: _json_render_core.SchemaType<"zod", unknown>;
+            description: _json_render_core.SchemaType<"string", unknown>;
+        }>;
+    }>;
+}>;
+declare function createCatalog<TDef extends SchemaDefinition, TCatalog extends InferCatalogInput<TDef["catalog"]>>(schema: Schema<TDef>, catalog: TCatalog): Catalog<TDef, TCatalog>;
+/**
+ * Build a user prompt with streamui-specific rules appended.
+ * Wraps `@json-render/core`'s `buildUserPrompt` and adds transport
+ * instructions (no code fences, raw JSONL output).
+ */
+declare function createUserPrompt(options: UserPromptOptions): string;
+
+export { type UIDataTypes, UITransport, type UITransportOptions, convertUIMessagesToModelMessages, createCatalog, createUserPrompt, defaultSchema };

package/dist/index.js

@@ -0,0 +1,311 @@
+// src/spec-stream.ts
+import {
+  applySpecStreamPatch
+} from "@json-render/core";
+function parseSpecLine(line) {
+  const trimmed = line.trim();
+  if (!trimmed) return null;
+  try {
+    const parsed = JSON.parse(trimmed);
+    if (parsed.op && parsed.path) {
+      return { type: "patch", data: parsed };
+    }
+    return { type: "text", content: line };
+  } catch {
+    return { type: "text", content: line };
+  }
+}
+var SpecStreamBuffer = class {
+  buffer = "";
+  /**
+   * Process a chunk of text and return parsed lines.
+   * Incomplete lines are buffered for the next chunk.
+   */
+  processChunk(chunk) {
+    this.buffer += chunk;
+    const lines = this.buffer.split("\n");
+    this.buffer = lines.pop() || "";
+    const results = [];
+    for (const line of lines) {
+      const parsed = parseSpecLine(line);
+      if (parsed) {
+        results.push(parsed);
+      }
+    }
+    return results;
+  }
+  /**
+   * Flush any remaining buffered content.
+   * Call this when the stream ends.
+   */
+  flush() {
+    if (!this.buffer.trim()) return null;
+    const result = parseSpecLine(this.buffer);
+    this.buffer = "";
+    return result;
+  }
+};
+function applySpecPatch(spec, patch) {
+  const nextSpec = JSON.parse(JSON.stringify(spec));
+  applySpecStreamPatch(
+    nextSpec,
+    patch
+  );
+  return nextSpec;
+}
+
+// src/transport.ts
+var UITransport = class {
+  api;
+  headers;
+  body;
+  credentials;
+  customFetch;
+  constructor(options = {}) {
+    this.api = options.api ?? "/api/chat";
+    this.headers = options.headers;
+    this.body = options.body;
+    this.credentials = options.credentials;
+    this.customFetch = options.fetch;
+  }
+  async sendMessages(options) {
+    const resolvedHeaders = typeof this.headers === "function" ? await this.headers() : this.headers;
+    const optionHeaders = options.headers instanceof Headers ? Object.fromEntries(options.headers.entries()) : options.headers ?? {};
+    const fetchFn = this.customFetch ?? globalThis.fetch;
+    const response = await fetchFn(this.api, {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+        ...resolvedHeaders,
+        ...optionHeaders
+      },
+      body: JSON.stringify({
+        ...this.body,
+        ...options.body,
+        id: options.chatId,
+        messages: options.messages,
+        trigger: options.trigger,
+        messageId: options.messageId
+      }),
+      credentials: this.credentials,
+      signal: options.abortSignal
+    });
+    if (!response.ok) {
+      throw new Error(
+        await response.text() || "Failed to fetch the chat response."
+      );
+    }
+    if (!response.body) {
+      throw new Error("The response body is empty.");
+    }
+    return this.processResponseStream(response.body);
+  }
+  /**
+   * Reconnect to an existing stream is not supported.
+   */
+  async reconnectToStream(_options) {
+    return null;
+  }
+  /**
+   * Process the UIMessageStream response:
+   * - Parse JSON event stream (same format as DefaultChatTransport)
+   * - Extract SpecStream patches from text-delta chunks
+   * - Build Spec incrementally and emit data-ui chunks
+   * - Pass through all other chunk types unchanged
+   */
+  processResponseStream(stream) {
+    const specBuffer = new SpecStreamBuffer();
+    let currentSpec = { root: "", elements: {} };
+    let segmentIndex = 0;
+    let currentSegmentText = "";
+    const textDecoder = new TextDecoder();
+    return new ReadableStream({
+      async start(controller) {
+        const reader = stream.getReader();
+        let buffer = "";
+        const emitUIText = (text) => {
+          currentSegmentText += text;
+          controller.enqueue({
+            type: "data-ui-text",
+            id: `ui-text-${segmentIndex}`,
+            data: {
+              segmentId: `seg-${segmentIndex}`,
+              text: currentSegmentText
+            }
+          });
+        };
+        const splitTextSegment = () => {
+          if (!currentSegmentText) return;
+          segmentIndex += 1;
+          currentSegmentText = "";
+        };
+        const processLine = (line) => {
+          if (!line.trim()) return;
+          let jsonLine = line.trim();
+          if (jsonLine.startsWith("data:")) {
+            jsonLine = jsonLine.slice(5).trimStart();
+          }
+          if (jsonLine === "[DONE]") return;
+          try {
+            const chunk = JSON.parse(jsonLine);
+            if (chunk.type === "text-delta") {
+              const parsed = specBuffer.processChunk(chunk.delta);
+              for (const item of parsed) {
+                if (item.type === "patch") {
+                  splitTextSegment();
+                  currentSpec = applySpecPatch(
+                    currentSpec,
+                    item.data
+                  );
+                  controller.enqueue({
+                    type: "data-ui",
+                    id: "ui",
+                    data: currentSpec
+                  });
+                } else {
+                  emitUIText(item.content + "\n");
+                }
+              }
+            } else if (chunk.type === "text-end") {
+              const remaining = specBuffer.flush();
+              if (remaining) {
+                if (remaining.type === "patch") {
+                  splitTextSegment();
+                  currentSpec = applySpecPatch(
+                    currentSpec,
+                    remaining.data
+                  );
+                  controller.enqueue({
+                    type: "data-ui",
+                    id: "ui",
+                    data: currentSpec
+                  });
+                } else {
+                  emitUIText(remaining.content);
+                }
+              }
+              splitTextSegment();
+            } else if (chunk.type === "text-start") {
+              return;
+            } else {
+              controller.enqueue(chunk);
+            }
+          } catch {
+          }
+        };
+        try {
+          while (true) {
+            const { done, value } = await reader.read();
+            if (done) {
+              buffer += textDecoder.decode();
+              break;
+            }
+            buffer += textDecoder.decode(value, { stream: true });
+            const lines = buffer.split("\n");
+            buffer = lines.pop() || "";
+            for (const line of lines) {
+              processLine(line);
+            }
+          }
+          if (buffer.trim()) {
+            processLine(buffer);
+          }
+          controller.close();
+        } catch (error) {
+          controller.error(error);
+        } finally {
+          reader.releaseLock();
+        }
+      }
+    });
+  }
+};
+
+// src/utils.ts
+import { convertToModelMessages } from "ai";
+function serializeSpec(spec) {
+  return JSON.stringify(spec);
+}
+async function convertUIMessagesToModelMessages(messages) {
+  return convertToModelMessages(messages, {
+    convertDataPart: (part) => {
+      if (part.type === "data-ui-text" && part.data) {
+        const text = part.data.text;
+        if (typeof text === "string" && text.length > 0) {
+          return { type: "text", text };
+        }
+      }
+      if (part.id === "ui" && part.data) {
+        const specJson = serializeSpec(part.data);
+        return { type: "text", text: `[Generated UI Spec JSON]
+${specJson}` };
+      }
+      return void 0;
+    }
+  });
+}
+
+// src/schema.ts
+import {
+  defineSchema,
+  defineCatalog,
+  buildUserPrompt
+} from "@json-render/core";
+var defaultSchema = defineSchema((s) => ({
+  spec: s.object({
+    root: s.string(),
+    elements: s.record(
+      s.object({
+        type: s.ref("catalog.components"),
+        props: s.propsOf("catalog.components"),
+        children: s.array(s.string()),
+        visible: s.any(),
+        on: s.any(),
+        repeat: s.any()
+      })
+    ),
+    state: s.any()
+  }),
+  catalog: s.object({
+    components: s.map({
+      props: s.zod(),
+      slots: s.array(s.string()),
+      description: s.string()
+    }),
+    actions: s.map({
+      params: s.zod(),
+      description: s.string()
+    })
+  })
+}));
+var STREAMUI_RULES = [
+  "Never wrap output in code fences (``` or ```json). Output raw JSONL lines directly.",
+  "You may include plain text lines before, after, or between JSONL patch lines."
+];
+function createCatalog(schema, catalog) {
+  const base = defineCatalog(schema, catalog);
+  const originalPrompt = base.prompt.bind(base);
+  base.prompt = (options) => {
+    return originalPrompt({
+      ...options,
+      customRules: [...STREAMUI_RULES, ...options?.customRules ?? []]
+    });
+  };
+  return base;
+}
+function createUserPrompt(options) {
+  const base = buildUserPrompt(options);
+  return base + "\n\n" + STREAMUI_RULES.join("\n");
+}
+
+// src/index.ts
+import { defineSchema as defineSchema2 } from "@json-render/core";
+export {
+  UITransport,
+  convertUIMessagesToModelMessages,
+  createCatalog,
+  createUserPrompt,
+  defaultSchema,
+  defineSchema2 as defineSchema
+};
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/spec-stream.ts","../src/transport.ts","../src/utils.ts","../src/schema.ts","../src/index.ts"],"sourcesContent":["import {\n  applySpecStreamPatch,\n  type Spec,\n  type SpecStreamLine,\n} from \"@json-render/core\";\nimport type { ParsedLine, SpecPatch } from \"./types\";\n\n// ---------------------------------------------------------------------------\n// SpecStream Buffer - Parse streaming text into patches and text lines\n// ---------------------------------------------------------------------------\n\n/**\n * Parse a single line as either a spec patch or text.\n * Preserves original whitespace for text content.\n */\nfunction parseSpecLine(line: string): ParsedLine | null {\n  const trimmed = line.trim();\n  if (!trimmed) return null;\n\n  try {\n    const parsed = JSON.parse(trimmed);\n    if (parsed.op && parsed.path) {\n      return { type: \"patch\", data: parsed as SpecPatch };\n    }\n    // Valid JSON but not a patch - return original line content\n    return { type: \"text\", content: line };\n  } catch {\n    // Not JSON - return original line content (preserving whitespace)\n    return { type: \"text\", content: line };\n  }\n}\n\n/**\n * Buffer for processing SpecStream content.\n * Handles incomplete lines across chunk boundaries.\n */\nexport class SpecStreamBuffer {\n  private buffer = \"\";\n\n  /**\n   * Process a chunk of text and return parsed lines.\n   * Incomplete lines are buffered for the next chunk.\n   */\n  processChunk(chunk: string): ParsedLine[] {\n    this.buffer += chunk;\n\n    const lines = this.buffer.split(\"\\n\");\n    this.buffer = lines.pop() || \"\";\n\n    const results: ParsedLine[] = [];\n    for (const line of lines) {\n      const parsed = parseSpecLine(line);\n      if (parsed) {\n        results.push(parsed);\n      }\n    }\n\n    return results;\n  }\n\n  /**\n   * Flush any remaining buffered content.\n   * Call this when the stream ends.\n   */\n  flush(): ParsedLine | null {\n    if (!this.buffer.trim()) return null;\n    const result = parseSpecLine(this.buffer);\n    this.buffer = \"\";\n    return result;\n  }\n}\n\n// ---------------------------------------------------------------------------\n// Spec Patch Application\n// ---------------------------------------------------------------------------\n\n/**\n * Apply a single patch to a Spec. Returns a new Spec (immutable).\n */\nexport function applySpecPatch(spec: Spec, patch: SpecPatch): Spec {\n  // json-render's applySpecStreamPatch mutates the target object.\n  // Clone first to preserve streamui's immutable update behavior.\n  const nextSpec = JSON.parse(JSON.stringify(spec)) as Spec;\n  applySpecStreamPatch(\n    nextSpec as unknown as Record<string, unknown>,\n    patch as unknown as SpecStreamLine,\n  );\n  return nextSpec;\n}\n\n/**\n * Build a Spec from an array of patches.\n */\nexport function buildSpec(patches: SpecPatch[]): Spec {\n  let spec: Spec = { root: \"\", elements: {} };\n  let firstElementKey: string | null = null;\n\n  for (const patch of patches) {\n    const directElementPath = /^\\/elements\\/[^/]+$/.exec(patch.path);\n    if (\n      directElementPath &&\n      (patch.op === \"add\" || patch.op === \"replace\") &&\n      !firstElementKey\n    ) {\n      firstElementKey = directElementPath[0].slice(\"/elements/\".length);\n    }\n    spec = applySpecPatch(spec, patch);\n  }\n\n  // Auto-infer root from first element if not set\n  if (!spec.root && firstElementKey) {\n    spec.root = firstElementKey;\n  }\n\n  return spec;\n}\n","import type {\n  ChatTransport,\n  UIMessage,\n  ChatRequestOptions,\n  UIMessageChunk,\n} from \"ai\";\nimport type { Spec } from \"@json-render/core\";\nimport { SpecStreamBuffer, applySpecPatch } from \"./spec-stream\";\nimport type { SpecPatch, UIDataTypes } from \"./types\";\n\n/**\n * Options for UITransport.\n */\nexport interface UITransportOptions {\n  /** API endpoint URL. Defaults to '/api/chat'. */\n  api?: string;\n  /** Additional headers for requests. */\n  headers?:\n    | Record<string, string>\n    | (() => Record<string, string> | Promise<Record<string, string>>);\n  /** Additional body properties for requests. */\n  body?: Record<string, unknown>;\n  /** Request credentials mode. */\n  credentials?: RequestCredentials;\n  /** Custom fetch implementation. */\n  fetch?: typeof fetch;\n}\n\n/**\n * UITransport - Custom ChatTransport for streaming UI with SpecStream patches.\n *\n * Extends DefaultChatTransport behavior:\n * - Parses UIMessageStream (JSON event stream)\n * - Extracts SpecStream patches from text-delta chunks\n * - Builds Spec incrementally and emits `data-ui` chunks\n * - Passes through all other chunk types (tool, reasoning, etc.)\n */\nexport class UITransport<\n  UI_MESSAGE extends UIMessage = UIMessage,\n> implements ChatTransport<UI_MESSAGE> {\n  private api: string;\n  private headers?: UITransportOptions[\"headers\"];\n  private body?: Record<string, unknown>;\n  private credentials?: RequestCredentials;\n  private customFetch?: typeof fetch;\n\n  constructor(options: UITransportOptions = {}) {\n    this.api = options.api ?? \"/api/chat\";\n    this.headers = options.headers;\n    this.body = options.body;\n    this.credentials = options.credentials;\n    this.customFetch = options.fetch;\n  }\n\n  async sendMessages(\n    options: {\n      trigger: \"submit-message\" | \"regenerate-message\";\n      chatId: string;\n      messageId: string | undefined;\n      messages: UI_MESSAGE[];\n      abortSignal: AbortSignal | undefined;\n    } & ChatRequestOptions,\n  ): Promise<ReadableStream<UIMessageChunk<unknown, UIDataTypes>>> {\n    const resolvedHeaders =\n      typeof this.headers === \"function\" ? await this.headers() : this.headers;\n\n    const optionHeaders =\n      options.headers instanceof Headers\n        ? Object.fromEntries(options.headers.entries())\n        : (options.headers ?? {});\n\n    const fetchFn = this.customFetch ?? globalThis.fetch;\n\n    const response = await fetchFn(this.api, {\n      method: \"POST\",\n      headers: {\n        \"Content-Type\": \"application/json\",\n        ...resolvedHeaders,\n        ...optionHeaders,\n      },\n      body: JSON.stringify({\n        ...this.body,\n        ...options.body,\n        id: options.chatId,\n        messages: options.messages,\n        trigger: options.trigger,\n        messageId: options.messageId,\n      }),\n      credentials: this.credentials,\n      signal: options.abortSignal,\n    });\n\n    if (!response.ok) {\n      throw new Error(\n        (await response.text()) || \"Failed to fetch the chat response.\",\n      );\n    }\n\n    if (!response.body) {\n      throw new Error(\"The response body is empty.\");\n    }\n\n    return this.processResponseStream(response.body);\n  }\n\n  /**\n   * Reconnect to an existing stream is not supported.\n   */\n  async reconnectToStream(\n    _options: { chatId: string } & ChatRequestOptions,\n  ): Promise<ReadableStream<UIMessageChunk<unknown, UIDataTypes>> | null> {\n    return null;\n  }\n\n  /**\n   * Process the UIMessageStream response:\n   * - Parse JSON event stream (same format as DefaultChatTransport)\n   * - Extract SpecStream patches from text-delta chunks\n   * - Build Spec incrementally and emit data-ui chunks\n   * - Pass through all other chunk types unchanged\n   */\n  private processResponseStream(\n    stream: ReadableStream<Uint8Array>,\n  ): ReadableStream<UIMessageChunk<unknown, UIDataTypes>> {\n    const specBuffer = new SpecStreamBuffer();\n    let currentSpec: Spec = { root: \"\", elements: {} };\n    let segmentIndex = 0;\n    let currentSegmentText = \"\";\n\n    const textDecoder = new TextDecoder();\n\n    return new ReadableStream<UIMessageChunk<unknown, UIDataTypes>>({\n      async start(controller) {\n        const reader = stream.getReader();\n        let buffer = \"\";\n        const emitUIText = (text: string) => {\n          currentSegmentText += text;\n          controller.enqueue({\n            type: \"data-ui-text\",\n            id: `ui-text-${segmentIndex}`,\n            data: {\n              segmentId: `seg-${segmentIndex}`,\n              text: currentSegmentText,\n            },\n          });\n        };\n\n        const splitTextSegment = () => {\n          if (!currentSegmentText) return;\n          segmentIndex += 1;\n          currentSegmentText = \"\";\n        };\n\n        const processLine = (line: string) => {\n          if (!line.trim()) return;\n\n          // Handle SSE format (data:{...} or data: {...}) or raw JSON\n          let jsonLine = line.trim();\n          if (jsonLine.startsWith(\"data:\")) {\n            jsonLine = jsonLine.slice(5).trimStart();\n          }\n          if (jsonLine === \"[DONE]\") return;\n\n          try {\n            const chunk = JSON.parse(jsonLine) as UIMessageChunk<\n              unknown,\n              UIDataTypes\n            >;\n\n            // Handle text-delta: extract patches and convert text to data-ui-text\n            if (chunk.type === \"text-delta\") {\n              const parsed = specBuffer.processChunk(chunk.delta);\n\n              for (const item of parsed) {\n                if (item.type === \"patch\") {\n                  splitTextSegment();\n                  currentSpec = applySpecPatch(\n                    currentSpec,\n                    item.data as SpecPatch,\n                  );\n                  controller.enqueue({\n                    type: \"data-ui\",\n                    id: \"ui\",\n                    data: currentSpec,\n                  });\n                } else {\n                  // Keep newline to preserve streamed text formatting.\n                  emitUIText(item.content + \"\\n\");\n                }\n              }\n            }\n            // Handle text-end: flush remaining buffer content\n            else if (chunk.type === \"text-end\") {\n              const remaining = specBuffer.flush();\n              if (remaining) {\n                if (remaining.type === \"patch\") {\n                  splitTextSegment();\n                  currentSpec = applySpecPatch(\n                    currentSpec,\n                    remaining.data as SpecPatch,\n                  );\n                  controller.enqueue({\n                    type: \"data-ui\",\n                    id: \"ui\",\n                    data: currentSpec,\n                  });\n                } else {\n                  emitUIText(remaining.content);\n                }\n              }\n\n              splitTextSegment();\n            }\n            // text-start is transport-internal for us; suppress it.\n            else if (chunk.type === \"text-start\") {\n              return;\n            }\n            // Pass through all other chunk types unchanged\n            else {\n              controller.enqueue(chunk);\n            }\n          } catch {\n            // Skip invalid JSON lines\n          }\n        };\n\n        try {\n          while (true) {\n            const { done, value } = await reader.read();\n            if (done) {\n              buffer += textDecoder.decode();\n              break;\n            }\n\n            buffer += textDecoder.decode(value, { stream: true });\n\n            // Parse JSON events (newline-delimited JSON)\n            const lines = buffer.split(\"\\n\");\n            buffer = lines.pop() || \"\";\n\n            for (const line of lines) {\n              processLine(line);\n            }\n          }\n\n          if (buffer.trim()) {\n            processLine(buffer);\n          }\n\n          controller.close();\n        } catch (error) {\n          controller.error(error);\n        } finally {\n          reader.releaseLock();\n        }\n      },\n    });\n  }\n}\n","import type { UIMessage, ModelMessage } from \"ai\";\nimport { convertToModelMessages } from \"ai\";\nimport type { Spec } from \"@json-render/core\";\nimport type { UIDataTypes } from \"./types\";\n\n/**\n * Serialize a Spec as compact JSON for model context.\n * Avoids format conversion to keep model output aligned with JSON/patch generation.\n */\nfunction serializeSpec(spec: Spec): string {\n  return JSON.stringify(spec);\n}\n\n/**\n * Convert UI messages to model messages.\n * Automatically converts data-ui parts to text descriptions for LLM context.\n *\n * @example\n * ```ts\n * const modelMessages = await convertUIMessagesToModelMessages(messages);\n * const result = streamText({ model, messages: modelMessages });\n * ```\n */\nexport async function convertUIMessagesToModelMessages(\n  messages: UIMessage<unknown, UIDataTypes>[],\n): Promise<ModelMessage[]> {\n  return convertToModelMessages(messages, {\n    convertDataPart: (part) => {\n      if (part.type === \"data-ui-text\" && part.data) {\n        const text = (part.data as { text?: string }).text;\n        if (typeof text === \"string\" && text.length > 0) {\n          return { type: \"text\", text };\n        }\n      }\n\n      if (part.id === \"ui\" && part.data) {\n        const specJson = serializeSpec(part.data as Spec);\n        return { type: \"text\", text: `[Generated UI Spec JSON]\\n${specJson}` };\n      }\n      return undefined;\n    },\n  });\n}\n","import {\n  defineSchema,\n  defineCatalog,\n  buildUserPrompt,\n  type Schema,\n  type SchemaDefinition,\n  type Catalog,\n  type InferCatalogInput,\n  type PromptOptions,\n  type UserPromptOptions,\n} from \"@json-render/core\";\n\n/**\n * Default schema for streamui - Element Tree format.\n *\n * Defines the Spec shape (what AI generates) and the Catalog shape\n * (what components provide). Compatible with @json-render/react's Renderer.\n *\n * Spec structure:\n * ```json\n * {\n *   \"root\": \"card-1\",\n *   \"state\": { \"count\": 0 },\n *   \"elements\": {\n *     \"card-1\": { \"type\": \"Card\", \"props\": {...}, \"children\": [...], \"on\": {...}, \"repeat\": {...} }\n *   }\n * }\n * ```\n */\nexport const defaultSchema = defineSchema((s) => ({\n  spec: s.object({\n    root: s.string(),\n    elements: s.record(\n      s.object({\n        type: s.ref(\"catalog.components\"),\n        props: s.propsOf(\"catalog.components\"),\n        children: s.array(s.string()),\n        visible: s.any(),\n        on: s.any(),\n        repeat: s.any(),\n      }),\n    ),\n    state: s.any(),\n  }),\n  catalog: s.object({\n    components: s.map({\n      props: s.zod(),\n      slots: s.array(s.string()),\n      description: s.string(),\n    }),\n    actions: s.map({\n      params: s.zod(),\n      description: s.string(),\n    }),\n  }),\n}));\n\n/**\n * Create a catalog from a schema and component/action definitions.\n *\n * @example\n * ```typescript\n * import { createCatalog, defaultSchema } from 'streamui';\n * import { z } from 'zod';\n *\n * const catalog = createCatalog(defaultSchema, {\n *   components: {\n *     Card: {\n *       props: z.object({ title: z.string().optional() }),\n *       slots: ['default'],\n *       description: 'Container with optional title',\n *     },\n *     Button: {\n *       props: z.object({ label: z.string() }),\n *       slots: [],\n *       description: 'Clickable button',\n *     },\n *   },\n * });\n *\n * // Generate system prompt for AI\n * const prompt = catalog.prompt();\n *\n * // Validate AI output\n * const result = catalog.validate(spec);\n * ```\n */\n/**\n * Default rules injected into prompt() for streamui.\n * These ensure the LLM outputs raw JSONL compatible with UITransport.\n */\nconst STREAMUI_RULES: string[] = [\n  \"Never wrap output in code fences (``` or ```json). Output raw JSONL lines directly.\",\n  \"You may include plain text lines before, after, or between JSONL patch lines.\",\n];\n\nexport function createCatalog<\n  TDef extends SchemaDefinition,\n  TCatalog extends InferCatalogInput<TDef[\"catalog\"]>,\n>(schema: Schema<TDef>, catalog: TCatalog): Catalog<TDef, TCatalog> {\n  const base = defineCatalog(schema, catalog);\n  const originalPrompt = base.prompt.bind(base);\n\n  base.prompt = (options?: PromptOptions): string => {\n    return originalPrompt({\n      ...options,\n      customRules: [...STREAMUI_RULES, ...(options?.customRules ?? [])],\n    });\n  };\n\n  return base;\n}\n\n/**\n * Build a user prompt with streamui-specific rules appended.\n * Wraps `@json-render/core`'s `buildUserPrompt` and adds transport\n * instructions (no code fences, raw JSONL output).\n */\nexport function createUserPrompt(options: UserPromptOptions): string {\n  const base = buildUserPrompt(options);\n  return base + \"\\n\\n\" + STREAMUI_RULES.join(\"\\n\");\n}\n","// Transport\nexport { UITransport } from \"./transport\";\nexport type { UITransportOptions } from \"./transport\";\n\n// Types for useChat generic\nexport type { UIDataTypes } from \"./types\";\n\n// Utilities\nexport { convertUIMessagesToModelMessages } from \"./utils\";\n\n// Schema / Catalog\nexport { defaultSchema, createCatalog, createUserPrompt } from \"./schema\";\nexport { defineSchema } from \"@json-render/core\";\nexport type { Catalog, Schema } from \"@json-render/core\";\n\n// Re-export Spec for convenience\nexport type { Spec } from \"@json-render/core\";\n"],"mappings":";AAAA;AAAA,EACE;AAAA,OAGK;AAWP,SAAS,cAAc,MAAiC;AACtD,QAAM,UAAU,KAAK,KAAK;AAC1B,MAAI,CAAC,QAAS,QAAO;AAErB,MAAI;AACF,UAAM,SAAS,KAAK,MAAM,OAAO;AACjC,QAAI,OAAO,MAAM,OAAO,MAAM;AAC5B,aAAO,EAAE,MAAM,SAAS,MAAM,OAAoB;AAAA,IACpD;AAEA,WAAO,EAAE,MAAM,QAAQ,SAAS,KAAK;AAAA,EACvC,QAAQ;AAEN,WAAO,EAAE,MAAM,QAAQ,SAAS,KAAK;AAAA,EACvC;AACF;AAMO,IAAM,mBAAN,MAAuB;AAAA,EACpB,SAAS;AAAA;AAAA;AAAA;AAAA;AAAA,EAMjB,aAAa,OAA6B;AACxC,SAAK,UAAU;AAEf,UAAM,QAAQ,KAAK,OAAO,MAAM,IAAI;AACpC,SAAK,SAAS,MAAM,IAAI,KAAK;AAE7B,UAAM,UAAwB,CAAC;AAC/B,eAAW,QAAQ,OAAO;AACxB,YAAM,SAAS,cAAc,IAAI;AACjC,UAAI,QAAQ;AACV,gBAAQ,KAAK,MAAM;AAAA,MACrB;AAAA,IACF;AAEA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,QAA2B;AACzB,QAAI,CAAC,KAAK,OAAO,KAAK,EAAG,QAAO;AAChC,UAAM,SAAS,cAAc,KAAK,MAAM;AACxC,SAAK,SAAS;AACd,WAAO;AAAA,EACT;AACF;AASO,SAAS,eAAe,MAAY,OAAwB;AAGjE,QAAM,WAAW,KAAK,MAAM,KAAK,UAAU,IAAI,CAAC;AAChD;AAAA,IACE;AAAA,IACA;AAAA,EACF;AACA,SAAO;AACT;;;ACnDO,IAAM,cAAN,MAEgC;AAAA,EAC7B;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAER,YAAY,UAA8B,CAAC,GAAG;AAC5C,SAAK,MAAM,QAAQ,OAAO;AAC1B,SAAK,UAAU,QAAQ;AACvB,SAAK,OAAO,QAAQ;AACpB,SAAK,cAAc,QAAQ;AAC3B,SAAK,cAAc,QAAQ;AAAA,EAC7B;AAAA,EAEA,MAAM,aACJ,SAO+D;AAC/D,UAAM,kBACJ,OAAO,KAAK,YAAY,aAAa,MAAM,KAAK,QAAQ,IAAI,KAAK;AAEnE,UAAM,gBACJ,QAAQ,mBAAmB,UACvB,OAAO,YAAY,QAAQ,QAAQ,QAAQ,CAAC,IAC3C,QAAQ,WAAW,CAAC;AAE3B,UAAM,UAAU,KAAK,eAAe,WAAW;AAE/C,UAAM,WAAW,MAAM,QAAQ,KAAK,KAAK;AAAA,MACvC,QAAQ;AAAA,MACR,SAAS;AAAA,QACP,gBAAgB;AAAA,QAChB,GAAG;AAAA,QACH,GAAG;AAAA,MACL;AAAA,MACA,MAAM,KAAK,UAAU;AAAA,QACnB,GAAG,KAAK;AAAA,QACR,GAAG,QAAQ;AAAA,QACX,IAAI,QAAQ;AAAA,QACZ,UAAU,QAAQ;AAAA,QAClB,SAAS,QAAQ;AAAA,QACjB,WAAW,QAAQ;AAAA,MACrB,CAAC;AAAA,MACD,aAAa,KAAK;AAAA,MAClB,QAAQ,QAAQ;AAAA,IAClB,CAAC;AAED,QAAI,CAAC,SAAS,IAAI;AAChB,YAAM,IAAI;AAAA,QACP,MAAM,SAAS,KAAK,KAAM;AAAA,MAC7B;AAAA,IACF;AAEA,QAAI,CAAC,SAAS,MAAM;AAClB,YAAM,IAAI,MAAM,6BAA6B;AAAA,IAC/C;AAEA,WAAO,KAAK,sBAAsB,SAAS,IAAI;AAAA,EACjD;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,kBACJ,UACsE;AACtE,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASQ,sBACN,QACsD;AACtD,UAAM,aAAa,IAAI,iBAAiB;AACxC,QAAI,cAAoB,EAAE,MAAM,IAAI,UAAU,CAAC,EAAE;AACjD,QAAI,eAAe;AACnB,QAAI,qBAAqB;AAEzB,UAAM,cAAc,IAAI,YAAY;AAEpC,WAAO,IAAI,eAAqD;AAAA,MAC9D,MAAM,MAAM,YAAY;AACtB,cAAM,SAAS,OAAO,UAAU;AAChC,YAAI,SAAS;AACb,cAAM,aAAa,CAAC,SAAiB;AACnC,gCAAsB;AACtB,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN,IAAI,WAAW,YAAY;AAAA,YAC3B,MAAM;AAAA,cACJ,WAAW,OAAO,YAAY;AAAA,cAC9B,MAAM;AAAA,YACR;AAAA,UACF,CAAC;AAAA,QACH;AAEA,cAAM,mBAAmB,MAAM;AAC7B,cAAI,CAAC,mBAAoB;AACzB,0BAAgB;AAChB,+BAAqB;AAAA,QACvB;AAEA,cAAM,cAAc,CAAC,SAAiB;AACpC,cAAI,CAAC,KAAK,KAAK,EAAG;AAGlB,cAAI,WAAW,KAAK,KAAK;AACzB,cAAI,SAAS,WAAW,OAAO,GAAG;AAChC,uBAAW,SAAS,MAAM,CAAC,EAAE,UAAU;AAAA,UACzC;AACA,cAAI,aAAa,SAAU;AAE3B,cAAI;AACF,kBAAM,QAAQ,KAAK,MAAM,QAAQ;AAMjC,gBAAI,MAAM,SAAS,cAAc;AAC/B,oBAAM,SAAS,WAAW,aAAa,MAAM,KAAK;AAElD,yBAAW,QAAQ,QAAQ;AACzB,oBAAI,KAAK,SAAS,SAAS;AACzB,mCAAiB;AACjB,gCAAc;AAAA,oBACZ;AAAA,oBACA,KAAK;AAAA,kBACP;AACA,6BAAW,QAAQ;AAAA,oBACjB,MAAM;AAAA,oBACN,IAAI;AAAA,oBACJ,MAAM;AAAA,kBACR,CAAC;AAAA,gBACH,OAAO;AAEL,6BAAW,KAAK,UAAU,IAAI;AAAA,gBAChC;AAAA,cACF;AAAA,YACF,WAES,MAAM,SAAS,YAAY;AAClC,oBAAM,YAAY,WAAW,MAAM;AACnC,kBAAI,WAAW;AACb,oBAAI,UAAU,SAAS,SAAS;AAC9B,mCAAiB;AACjB,gCAAc;AAAA,oBACZ;AAAA,oBACA,UAAU;AAAA,kBACZ;AACA,6BAAW,QAAQ;AAAA,oBACjB,MAAM;AAAA,oBACN,IAAI;AAAA,oBACJ,MAAM;AAAA,kBACR,CAAC;AAAA,gBACH,OAAO;AACL,6BAAW,UAAU,OAAO;AAAA,gBAC9B;AAAA,cACF;AAEA,+BAAiB;AAAA,YACnB,WAES,MAAM,SAAS,cAAc;AACpC;AAAA,YACF,OAEK;AACH,yBAAW,QAAQ,KAAK;AAAA,YAC1B;AAAA,UACF,QAAQ;AAAA,UAER;AAAA,QACF;AAEA,YAAI;AACF,iBAAO,MAAM;AACX,kBAAM,EAAE,MAAM,MAAM,IAAI,MAAM,OAAO,KAAK;AAC1C,gBAAI,MAAM;AACR,wBAAU,YAAY,OAAO;AAC7B;AAAA,YACF;AAEA,sBAAU,YAAY,OAAO,OAAO,EAAE,QAAQ,KAAK,CAAC;AAGpD,kBAAM,QAAQ,OAAO,MAAM,IAAI;AAC/B,qBAAS,MAAM,IAAI,KAAK;AAExB,uBAAW,QAAQ,OAAO;AACxB,0BAAY,IAAI;AAAA,YAClB;AAAA,UACF;AAEA,cAAI,OAAO,KAAK,GAAG;AACjB,wBAAY,MAAM;AAAA,UACpB;AAEA,qBAAW,MAAM;AAAA,QACnB,SAAS,OAAO;AACd,qBAAW,MAAM,KAAK;AAAA,QACxB,UAAE;AACA,iBAAO,YAAY;AAAA,QACrB;AAAA,MACF;AAAA,IACF,CAAC;AAAA,EACH;AACF;;;ACjQA,SAAS,8BAA8B;AAQvC,SAAS,cAAc,MAAoB;AACzC,SAAO,KAAK,UAAU,IAAI;AAC5B;AAYA,eAAsB,iCACpB,UACyB;AACzB,SAAO,uBAAuB,UAAU;AAAA,IACtC,iBAAiB,CAAC,SAAS;AACzB,UAAI,KAAK,SAAS,kBAAkB,KAAK,MAAM;AAC7C,cAAM,OAAQ,KAAK,KAA2B;AAC9C,YAAI,OAAO,SAAS,YAAY,KAAK,SAAS,GAAG;AAC/C,iBAAO,EAAE,MAAM,QAAQ,KAAK;AAAA,QAC9B;AAAA,MACF;AAEA,UAAI,KAAK,OAAO,QAAQ,KAAK,MAAM;AACjC,cAAM,WAAW,cAAc,KAAK,IAAY;AAChD,eAAO,EAAE,MAAM,QAAQ,MAAM;AAAA,EAA6B,QAAQ,GAAG;AAAA,MACvE;AACA,aAAO;AAAA,IACT;AAAA,EACF,CAAC;AACH;;;AC1CA;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,OAOK;AAmBA,IAAM,gBAAgB,aAAa,CAAC,OAAO;AAAA,EAChD,MAAM,EAAE,OAAO;AAAA,IACb,MAAM,EAAE,OAAO;AAAA,IACf,UAAU,EAAE;AAAA,MACV,EAAE,OAAO;AAAA,QACP,MAAM,EAAE,IAAI,oBAAoB;AAAA,QAChC,OAAO,EAAE,QAAQ,oBAAoB;AAAA,QACrC,UAAU,EAAE,MAAM,EAAE,OAAO,CAAC;AAAA,QAC5B,SAAS,EAAE,IAAI;AAAA,QACf,IAAI,EAAE,IAAI;AAAA,QACV,QAAQ,EAAE,IAAI;AAAA,MAChB,CAAC;AAAA,IACH;AAAA,IACA,OAAO,EAAE,IAAI;AAAA,EACf,CAAC;AAAA,EACD,SAAS,EAAE,OAAO;AAAA,IAChB,YAAY,EAAE,IAAI;AAAA,MAChB,OAAO,EAAE,IAAI;AAAA,MACb,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC;AAAA,MACzB,aAAa,EAAE,OAAO;AAAA,IACxB,CAAC;AAAA,IACD,SAAS,EAAE,IAAI;AAAA,MACb,QAAQ,EAAE,IAAI;AAAA,MACd,aAAa,EAAE,OAAO;AAAA,IACxB,CAAC;AAAA,EACH,CAAC;AACH,EAAE;AAoCF,IAAM,iBAA2B;AAAA,EAC/B;AAAA,EACA;AACF;AAEO,SAAS,cAGd,QAAsB,SAA4C;AAClE,QAAM,OAAO,cAAc,QAAQ,OAAO;AAC1C,QAAM,iBAAiB,KAAK,OAAO,KAAK,IAAI;AAE5C,OAAK,SAAS,CAAC,YAAoC;AACjD,WAAO,eAAe;AAAA,MACpB,GAAG;AAAA,MACH,aAAa,CAAC,GAAG,gBAAgB,GAAI,SAAS,eAAe,CAAC,CAAE;AAAA,IAClE,CAAC;AAAA,EACH;AAEA,SAAO;AACT;AAOO,SAAS,iBAAiB,SAAoC;AACnE,QAAM,OAAO,gBAAgB,OAAO;AACpC,SAAO,OAAO,SAAS,eAAe,KAAK,IAAI;AACjD;;;AC7GA,SAAS,gBAAAA,qBAAoB;","names":["defineSchema"]}

package/package.json

@@ -0,0 +1,61 @@
+{
+  "name": "ai-streamui",
+  "version": "0.1.0",
+  "description": "Streaming UI from AI in chat — a Transport for AI SDK with json-render",
+  "type": "module",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/miurla/streamui.git",
+    "directory": "packages/streamui"
+  },
+  "homepage": "https://github.com/miurla/streamui#readme",
+  "bugs": {
+    "url": "https://github.com/miurla/streamui/issues"
+  },
+  "keywords": [
+    "ai",
+    "streaming",
+    "ui",
+    "chat",
+    "ai-sdk",
+    "json-render",
+    "transport",
+    "llm"
+  ],
+  "files": [
+    "dist"
+  ],
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js"
+    }
+  },
+  "scripts": {
+    "build": "tsup",
+    "dev": "tsup --watch",
+    "lint": "eslint src",
+    "check-types": "tsc --noEmit",
+    "test": "vitest run"
+  },
+  "dependencies": {
+    "@json-render/core": "^0.5.0"
+  },
+  "peerDependencies": {
+    "ai": "^6.0.0"
+  },
+  "devDependencies": {
+    "@ai-sdk/react": "^3.0.0",
+    "@repo/eslint-config": "workspace:*",
+    "@repo/typescript-config": "workspace:*",
+    "ai": "^6.0.0",
+    "eslint": "^9.0.0",
+    "tsup": "^8.3.0",
+    "typescript": "^5.9.0",
+    "vitest": "^3.0.0",
+    "zod": "^4.0.0"
+  }
+}