@outkit-dev/react 0.0.5 → 0.0.7

package/dist/index.cjs

@@ -22,11 +22,12 @@ var index_exports = {};
 __export(index_exports, {
   AIRenderer: () => AIRenderer,
   OUTKIT_TOKENS: () => import_renderer2.OUTKIT_TOKENS,
-  completePartialJson: () => completePartialJson,
-  expandWireBlock: () => expandWireBlock,
-  extractCompleteBlocks: () => extractCompleteBlocks,
-  isRenderableBlock: () => isRenderableBlock,
-  parseStreamingBlocks: () => parseStreamingBlocks,
+  OutkitStream: () => import_core2.OutkitStream,
+  completePartialJson: () => import_core3.completePartialJson,
+  expandWireBlock: () => import_core3.expandWireBlock,
+  extractCompleteBlocks: () => import_core3.extractCompleteBlocks,
+  isRenderableBlock: () => import_core3.isRenderableBlock,
+  parseStreamingBlocks: () => import_core3.parseStreamingBlocks,
   useBlockStream: () => useBlockStream
 });
 module.exports = __toCommonJS(index_exports);
@@ -227,150 +228,8 @@ var SingleComponent = (0, import_react.memo)(function SingleComponent2({
 });
 
 // src/use-block-stream.ts
+var import_core = require("@outkit-dev/core");
 var import_react2 = require("react");
-
-// src/streaming/complete-partial-json.ts
-function completePartialJson(partial) {
-  let inString = false;
-  let escaped = false;
-  const stack = [];
-  for (let i = 0; i < partial.length; i++) {
-    const ch = partial[i];
-    if (escaped) {
-      escaped = false;
-      continue;
-    }
-    if (ch === "\\" && inString) {
-      escaped = true;
-      continue;
-    }
-    if (ch === '"') {
-      inString = !inString;
-      continue;
-    }
-    if (inString) continue;
-    if (ch === "{") stack.push("}");
-    else if (ch === "[") stack.push("]");
-    else if ((ch === "}" || ch === "]") && stack.length > 0 && stack[stack.length - 1] === ch) {
-      stack.pop();
-    }
-  }
-  let result = partial;
-  if (inString) {
-    if (escaped) result = result.slice(0, -1);
-    const trailingUnicode = result.match(/\\u[0-9a-fA-F]{0,3}$/);
-    if (trailingUnicode) {
-      result = result.slice(0, -trailingUnicode[0].length);
-    }
-    result += '"';
-  }
-  while (stack.length > 0) {
-    result += stack.pop();
-  }
-  return result;
-}
-
-// src/streaming/parse-streaming-blocks.ts
-function expandWireBlock(block) {
-  if ("type" in block && "component" in block) return block;
-  if ("type" in block && block.type === "text") return block;
-  if (typeof block.c === "string" && typeof block.p === "object" && block.p !== null) {
-    return {
-      type: "component",
-      component: block.c,
-      version: block.v ?? "1.0",
-      props: block.p,
-      ...block.f !== void 0 ? { fallback: block.f } : {},
-      ...block.m ? { meta: block.m } : {}
-    };
-  }
-  return block;
-}
-function isRenderableBlock(block) {
-  if (!block || typeof block !== "object") return false;
-  const b = block;
-  if (b.type === "text" && typeof b.content === "string" && b.content.length > 0)
-    return true;
-  if (b.type === "component" && typeof b.component === "string" && b.component.length >= 2 && typeof b.props === "object" && b.props !== null)
-    return true;
-  return false;
-}
-function extractCompleteBlocks(accumulated) {
-  const blocksIdx = accumulated.indexOf('"blocks"');
-  let arrayStart = -1;
-  if (blocksIdx !== -1) {
-    arrayStart = accumulated.indexOf("[", blocksIdx);
-  } else {
-    arrayStart = accumulated.indexOf("[");
-  }
-  if (arrayStart === -1) return [];
-  const content = accumulated.slice(arrayStart + 1);
-  const blocks = [];
-  let depth = 0;
-  let blockStart = -1;
-  let inString = false;
-  let escaped = false;
-  for (let i = 0; i < content.length; i++) {
-    const ch = content[i];
-    if (escaped) {
-      escaped = false;
-      continue;
-    }
-    if (ch === "\\" && inString) {
-      escaped = true;
-      continue;
-    }
-    if (ch === '"') {
-      inString = !inString;
-      continue;
-    }
-    if (inString) continue;
-    if (ch === "{") {
-      if (depth === 0) blockStart = i;
-      depth++;
-    } else if (ch === "}") {
-      depth--;
-      if (depth === 0 && blockStart !== -1) {
-        try {
-          const parsed = JSON.parse(content.slice(blockStart, i + 1));
-          const expanded = expandWireBlock(parsed);
-          if (isRenderableBlock(expanded)) blocks.push(expanded);
-        } catch {
-        }
-        blockStart = -1;
-      }
-    }
-  }
-  return blocks;
-}
-function parseStreamingBlocks(accumulated) {
-  if (!accumulated.trim()) return [];
-  let target = accumulated;
-  const blocksIdx = accumulated.indexOf('"blocks"');
-  if (blocksIdx > 0) {
-    let braceIdx = blocksIdx - 1;
-    while (braceIdx >= 0 && accumulated[braceIdx] !== "{") braceIdx--;
-    if (braceIdx > 0) target = accumulated.slice(braceIdx);
-  }
-  const patched = completePartialJson(target);
-  try {
-    const parsed = JSON.parse(patched);
-    const rawBlocks = parsed?.blocks ?? (Array.isArray(parsed) ? parsed : []);
-    return rawBlocks.map((b) => {
-      if (!b || typeof b !== "object") return b;
-      return expandWireBlock(b);
-    }).filter(isRenderableBlock).map((block) => {
-      if (!("version" in block)) {
-        return Object.assign({}, block, { version: "1.0" });
-      }
-      return block;
-    });
-  } catch {
-    return extractCompleteBlocks(accumulated);
-  }
-}
-
-// src/use-block-stream.ts
 function useBlockStream() {
   const [blocks, setBlocks] = (0, import_react2.useState)([]);
   const [isStreaming, setIsStreaming] = (0, import_react2.useState)(false);
@@ -378,9 +237,6 @@ function useBlockStream() {
   const pendingRef = (0, import_react2.useRef)(null);
   const rafRef = (0, import_react2.useRef)(0);
   const streamingRef = (0, import_react2.useRef)(false);
-  const accumulatedRef = (0, import_react2.useRef)("");
-  const designReceivedRef = (0, import_react2.useRef)(false);
-  const bufferedBlocksRef = (0, import_react2.useRef)(null);
   const flush = (0, import_react2.useCallback)(() => {
     rafRef.current = 0;
     if (pendingRef.current !== null) {
@@ -388,7 +244,7 @@ function useBlockStream() {
       pendingRef.current = null;
     }
   }, []);
-  const push = (0, import_react2.useCallback)(
+  const batchBlocks = (0, import_react2.useCallback)(
     (newBlocks) => {
       if (!streamingRef.current) {
         streamingRef.current = true;
@@ -401,10 +257,7 @@ function useBlockStream() {
     },
     [flush]
   );
-  const setDesign = (0, import_react2.useCallback)((tokens) => {
-    setDesignState(tokens);
-  }, []);
-  const complete = (0, import_react2.useCallback)(() => {
+  const handleComplete = (0, import_react2.useCallback)(() => {
     if (rafRef.current) {
       cancelAnimationFrame(rafRef.current);
       rafRef.current = 0;
@@ -416,72 +269,96 @@ function useBlockStream() {
     streamingRef.current = false;
     setIsStreaming(false);
   }, []);
+  const streamRef = (0, import_react2.useRef)(null);
+  const getStream = (0, import_react2.useCallback)(() => {
+    if (streamRef.current === null || streamRef.current.streamState === "destroyed") {
+      streamRef.current = new import_core.OutkitStream({
+        onBlocks: (b) => batchBlocks(b),
+        onDesign: (d) => setDesignState(d),
+        onComplete: () => handleComplete()
+      });
+    }
+    return streamRef.current;
+  }, [batchBlocks, handleComplete]);
+  const stream = getStream();
+  (0, import_react2.useEffect)(() => {
+    const s = getStream();
+    return () => {
+      if (rafRef.current) cancelAnimationFrame(rafRef.current);
+      s.destroy();
+    };
+  }, [getStream]);
   const reset = (0, import_react2.useCallback)(() => {
     if (rafRef.current) {
       cancelAnimationFrame(rafRef.current);
       rafRef.current = 0;
     }
     pendingRef.current = null;
-    accumulatedRef.current = "";
-    bufferedBlocksRef.current = null;
-    designReceivedRef.current = false;
     streamingRef.current = false;
+    getStream().reset();
     setBlocks([]);
     setIsStreaming(false);
     setDesignState(void 0);
-  }, []);
+  }, [getStream]);
+  const push = (0, import_react2.useCallback)(
+    (newBlocks) => batchBlocks(newBlocks),
+    [batchBlocks]
+  );
+  const setDesign = (0, import_react2.useCallback)(
+    (tokens) => getStream().feedMeta(tokens),
+    [getStream]
+  );
+  const complete = (0, import_react2.useCallback)(() => getStream().feedDone(), [getStream]);
+  const feedResponse = (0, import_react2.useCallback)(
+    (response) => getStream().feedResponse(response),
+    [getStream]
+  );
+  const feedSSE = (0, import_react2.useCallback)(
+    (rawText) => getStream().feedSSE(rawText),
+    [getStream]
+  );
+  const feedEvent = (0, import_react2.useCallback)(
+    (data) => getStream().feedEvent(data),
+    [getStream]
+  );
   const feedChunk = (0, import_react2.useCallback)(
-    (data) => {
-      accumulatedRef.current += data;
-      const parsed = parseStreamingBlocks(accumulatedRef.current);
-      if (parsed.length > 0) {
-        if (designReceivedRef.current) {
-          push(parsed);
-        } else {
-          bufferedBlocksRef.current = parsed;
-        }
-      }
-    },
-    [push]
+    (data) => getStream().feedChunk(data),
+    [getStream]
   );
   const feedMeta = (0, import_react2.useCallback)(
-    (tokens) => {
-      designReceivedRef.current = true;
-      setDesignState(tokens);
-      if (bufferedBlocksRef.current) {
-        push(bufferedBlocksRef.current);
-        bufferedBlocksRef.current = null;
-      }
-    },
-    [push]
+    (tokens) => getStream().feedMeta(tokens),
+    [getStream]
   );
-  const feedDone = (0, import_react2.useCallback)(() => {
-    if (accumulatedRef.current) {
-      const parsed = parseStreamingBlocks(accumulatedRef.current);
-      if (parsed.length > 0) {
-        pendingRef.current = parsed;
-      }
-    }
-    accumulatedRef.current = "";
-    bufferedBlocksRef.current = null;
-    complete();
-  }, [complete]);
-  (0, import_react2.useEffect)(() => {
-    return () => {
-      if (rafRef.current) {
-        cancelAnimationFrame(rafRef.current);
-      }
-    };
-  }, []);
-  return { blocks, isStreaming, design, feedChunk, feedMeta, feedDone, push, setDesign, complete, reset };
+  const feedDone = (0, import_react2.useCallback)(
+    () => getStream().feedDone(),
+    [getStream]
+  );
+  return {
+    blocks,
+    isStreaming,
+    design,
+    feedResponse,
+    feedSSE,
+    feedEvent,
+    feedChunk,
+    feedMeta,
+    feedDone,
+    push,
+    setDesign,
+    complete,
+    reset
+  };
 }
 
 // src/index.ts
+var import_core2 = require("@outkit-dev/core");
+var import_core3 = require("@outkit-dev/core");
 var import_renderer2 = require("@outkit-dev/renderer");
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   AIRenderer,
   OUTKIT_TOKENS,
+  OutkitStream,
   completePartialJson,
   expandWireBlock,
   extractCompleteBlocks,

package/dist/index.d.cts

@@ -2,6 +2,7 @@ import * as react_jsx_runtime from 'react/jsx-runtime';
 import { ContentBlock, ComponentBlock, OutkitInteractionEvent } from '@outkit-dev/renderer';
 export { ComponentBlock, ContentBlock, InlineAnnotation, OUTKIT_TOKENS, OutkitEnhanceResult, OutkitInteractionEvent } from '@outkit-dev/renderer';
 import React from 'react';
+export { OutkitStream, OutkitStreamCallbacks, OutkitStreamState, completePartialJson, expandWireBlock, extractCompleteBlocks, isRenderableBlock, parseStreamingBlocks } from '@outkit-dev/core';
 
 interface AIRendererProps {
     /** Array of content blocks from outkit_enhance */
@@ -43,69 +44,36 @@ interface UseBlockStreamReturn {
     isStreaming: boolean;
     /** Design tokens received from the stream's meta event. Pass to <AIRenderer design={design} />. */
     design: Record<string, string> | undefined;
-    /** Append a raw SSE data chunk. Parses accumulated buffer and updates blocks. */
+    /** Feed a raw fetch Response. Handles SSE parsing, meta detection, and completion. */
+    feedResponse: (response: Response) => Promise<void>;
+    /** Feed raw SSE-formatted text. Handles splitting, data: stripping, and routing. */
+    feedSSE: (rawText: string) => void;
+    /** Feed one unwrapped event payload. Handles [DONE], meta detection, and chunks. */
+    feedEvent: (data: string) => void;
+    /** Append a raw LLM data chunk. Parses accumulated buffer and updates blocks. */
     feedChunk: (data: string) => void;
-    /** Set design tokens from a meta SSE event. Alias for setDesign. */
+    /** Set design tokens from a meta SSE event. */
     feedMeta: (tokens: Record<string, string>) => void;
     /** Signal end of stream. Final parse + flush + mark complete. */
     feedDone: () => void;
-    /** Push a new full blocks array (the latest snapshot). Batched at 60fps. */
+    /** @deprecated Use feedChunk/feedResponse instead. Direct block snapshot setter. */
     push: (blocks: ContentBlock[]) => void;
-    /** Set design tokens (typically from a meta SSE event). */
+    /** @deprecated Use feedMeta instead. */
     setDesign: (tokens: Record<string, string>) => void;
-    /** Signal that streaming is complete. Flushes any pending update. */
+    /** @deprecated Use feedDone instead. */
     complete: () => void;
-    /** Reset to empty state. */
+    /** Reset to empty state. Aborts any in-flight feedResponse. */
     reset: () => void;
 }
 /**
- * Hook that batches rapid block updates to 60fps via requestAnimationFrame.
+ * React hook for streaming Outkit blocks with rAF-batched updates at 60fps.
  *
- * Usage:
- * ```tsx
- * const { blocks, isStreaming, push, complete } = useBlockStream();
- *
- * useEffect(() => {
- *   const es = new EventSource("/stream");
- *   es.onmessage = (e) => push(JSON.parse(e.data).blocks);
- *   es.addEventListener("done", () => complete());
- *   return () => es.close();
- * }, [push, complete]);
- *
- * return <AIRenderer blocks={blocks} streaming={isStreaming} />;
- * ```
+ * Four-tier API — pick the tier that matches your transport:
+ * - `feedResponse(res)` — fetch proxy (most common, handles everything)
+ * - `feedSSE(text)` — raw SSE bytes (manual reader)
+ * - `feedEvent(data)` — one unwrapped payload (WebSocket, Socket.IO)
+ * - `feedChunk/feedMeta/feedDone` — full manual control
  */
 declare function useBlockStream(): UseBlockStreamReturn;
 
-/**
- * Close unclosed strings, arrays, and objects so partial JSON can be parsed.
- *
- * As the LLM streams a JSON structure token by token, this function patches
- * the accumulated buffer into valid JSON at every chunk boundary. This lets
- * content stream character-by-character while remaining parseable.
- */
-declare function completePartialJson(partial: string): string;
-
-/** Expand a compact wire-format block (c/v/p/f/m) into the full ContentBlock shape. */
-declare function expandWireBlock(block: Record<string, unknown>): Record<string, unknown>;
-/** Check if a parsed object is a renderable block. */
-declare function isRenderableBlock(block: unknown): block is ContentBlock;
-/** Fallback: extract only fully brace-matched block objects from accumulated data. */
-declare function extractCompleteBlocks(accumulated: string): ContentBlock[];
-/**
- * Parse streaming LLM JSON into renderable blocks.
- *
- * Call this with the full accumulated buffer on every chunk. It closes
- * unclosed strings/arrays/objects via `completePartialJson`, parses the
- * result, and returns all currently renderable blocks.
- *
- * Handles both wire format (`{ c, v, p }`) and expanded format
- * (`{ type, component, version, props }`). The renderer's `normalizeBlock()`
- * also handles wire format at render time as a further safety net.
- *
- * @param accumulated - The full accumulated raw LLM output so far
- * @returns Array of renderable ContentBlock objects
- */
-declare function parseStreamingBlocks(accumulated: string): ContentBlock[];
-
-export { AIRenderer, type AIRendererProps, type UseBlockStreamReturn, completePartialJson, expandWireBlock, extractCompleteBlocks, isRenderableBlock, parseStreamingBlocks, useBlockStream };
+export { AIRenderer, type AIRendererProps, type UseBlockStreamReturn, useBlockStream };

package/dist/index.d.ts

@@ -2,6 +2,7 @@ import * as react_jsx_runtime from 'react/jsx-runtime';
 import { ContentBlock, ComponentBlock, OutkitInteractionEvent } from '@outkit-dev/renderer';
 export { ComponentBlock, ContentBlock, InlineAnnotation, OUTKIT_TOKENS, OutkitEnhanceResult, OutkitInteractionEvent } from '@outkit-dev/renderer';
 import React from 'react';
+export { OutkitStream, OutkitStreamCallbacks, OutkitStreamState, completePartialJson, expandWireBlock, extractCompleteBlocks, isRenderableBlock, parseStreamingBlocks } from '@outkit-dev/core';
 
 interface AIRendererProps {
     /** Array of content blocks from outkit_enhance */
@@ -43,69 +44,36 @@ interface UseBlockStreamReturn {
     isStreaming: boolean;
     /** Design tokens received from the stream's meta event. Pass to <AIRenderer design={design} />. */
     design: Record<string, string> | undefined;
-    /** Append a raw SSE data chunk. Parses accumulated buffer and updates blocks. */
+    /** Feed a raw fetch Response. Handles SSE parsing, meta detection, and completion. */
+    feedResponse: (response: Response) => Promise<void>;
+    /** Feed raw SSE-formatted text. Handles splitting, data: stripping, and routing. */
+    feedSSE: (rawText: string) => void;
+    /** Feed one unwrapped event payload. Handles [DONE], meta detection, and chunks. */
+    feedEvent: (data: string) => void;
+    /** Append a raw LLM data chunk. Parses accumulated buffer and updates blocks. */
     feedChunk: (data: string) => void;
-    /** Set design tokens from a meta SSE event. Alias for setDesign. */
+    /** Set design tokens from a meta SSE event. */
     feedMeta: (tokens: Record<string, string>) => void;
     /** Signal end of stream. Final parse + flush + mark complete. */
     feedDone: () => void;
-    /** Push a new full blocks array (the latest snapshot). Batched at 60fps. */
+    /** @deprecated Use feedChunk/feedResponse instead. Direct block snapshot setter. */
     push: (blocks: ContentBlock[]) => void;
-    /** Set design tokens (typically from a meta SSE event). */
+    /** @deprecated Use feedMeta instead. */
     setDesign: (tokens: Record<string, string>) => void;
-    /** Signal that streaming is complete. Flushes any pending update. */
+    /** @deprecated Use feedDone instead. */
     complete: () => void;
-    /** Reset to empty state. */
+    /** Reset to empty state. Aborts any in-flight feedResponse. */
     reset: () => void;
 }
 /**
- * Hook that batches rapid block updates to 60fps via requestAnimationFrame.
+ * React hook for streaming Outkit blocks with rAF-batched updates at 60fps.
  *
- * Usage:
- * ```tsx
- * const { blocks, isStreaming, push, complete } = useBlockStream();
- *
- * useEffect(() => {
- *   const es = new EventSource("/stream");
- *   es.onmessage = (e) => push(JSON.parse(e.data).blocks);
- *   es.addEventListener("done", () => complete());
- *   return () => es.close();
- * }, [push, complete]);
- *
- * return <AIRenderer blocks={blocks} streaming={isStreaming} />;
- * ```
+ * Four-tier API — pick the tier that matches your transport:
+ * - `feedResponse(res)` — fetch proxy (most common, handles everything)
+ * - `feedSSE(text)` — raw SSE bytes (manual reader)
+ * - `feedEvent(data)` — one unwrapped payload (WebSocket, Socket.IO)
+ * - `feedChunk/feedMeta/feedDone` — full manual control
  */
 declare function useBlockStream(): UseBlockStreamReturn;
 
-/**
- * Close unclosed strings, arrays, and objects so partial JSON can be parsed.
- *
- * As the LLM streams a JSON structure token by token, this function patches
- * the accumulated buffer into valid JSON at every chunk boundary. This lets
- * content stream character-by-character while remaining parseable.
- */
-declare function completePartialJson(partial: string): string;
-
-/** Expand a compact wire-format block (c/v/p/f/m) into the full ContentBlock shape. */
-declare function expandWireBlock(block: Record<string, unknown>): Record<string, unknown>;
-/** Check if a parsed object is a renderable block. */
-declare function isRenderableBlock(block: unknown): block is ContentBlock;
-/** Fallback: extract only fully brace-matched block objects from accumulated data. */
-declare function extractCompleteBlocks(accumulated: string): ContentBlock[];
-/**
- * Parse streaming LLM JSON into renderable blocks.
- *
- * Call this with the full accumulated buffer on every chunk. It closes
- * unclosed strings/arrays/objects via `completePartialJson`, parses the
- * result, and returns all currently renderable blocks.
- *
- * Handles both wire format (`{ c, v, p }`) and expanded format
- * (`{ type, component, version, props }`). The renderer's `normalizeBlock()`
- * also handles wire format at render time as a further safety net.
- *
- * @param accumulated - The full accumulated raw LLM output so far
- * @returns Array of renderable ContentBlock objects
- */
-declare function parseStreamingBlocks(accumulated: string): ContentBlock[];
-
-export { AIRenderer, type AIRendererProps, type UseBlockStreamReturn, completePartialJson, expandWireBlock, extractCompleteBlocks, isRenderableBlock, parseStreamingBlocks, useBlockStream };
+export { AIRenderer, type AIRendererProps, type UseBlockStreamReturn, useBlockStream };

package/dist/index.js

@@ -194,150 +194,8 @@ var SingleComponent = memo(function SingleComponent2({
 });
 
 // src/use-block-stream.ts
+import { OutkitStream } from "@outkit-dev/core";
 import { useCallback, useEffect as useEffect2, useRef as useRef2, useState } from "react";
-
-// src/streaming/complete-partial-json.ts
-function completePartialJson(partial) {
-  let inString = false;
-  let escaped = false;
-  const stack = [];
-  for (let i = 0; i < partial.length; i++) {
-    const ch = partial[i];
-    if (escaped) {
-      escaped = false;
-      continue;
-    }
-    if (ch === "\\" && inString) {
-      escaped = true;
-      continue;
-    }
-    if (ch === '"') {
-      inString = !inString;
-      continue;
-    }
-    if (inString) continue;
-    if (ch === "{") stack.push("}");
-    else if (ch === "[") stack.push("]");
-    else if ((ch === "}" || ch === "]") && stack.length > 0 && stack[stack.length - 1] === ch) {
-      stack.pop();
-    }
-  }
-  let result = partial;
-  if (inString) {
-    if (escaped) result = result.slice(0, -1);
-    const trailingUnicode = result.match(/\\u[0-9a-fA-F]{0,3}$/);
-    if (trailingUnicode) {
-      result = result.slice(0, -trailingUnicode[0].length);
-    }
-    result += '"';
-  }
-  while (stack.length > 0) {
-    result += stack.pop();
-  }
-  return result;
-}
-
-// src/streaming/parse-streaming-blocks.ts
-function expandWireBlock(block) {
-  if ("type" in block && "component" in block) return block;
-  if ("type" in block && block.type === "text") return block;
-  if (typeof block.c === "string" && typeof block.p === "object" && block.p !== null) {
-    return {
-      type: "component",
-      component: block.c,
-      version: block.v ?? "1.0",
-      props: block.p,
-      ...block.f !== void 0 ? { fallback: block.f } : {},
-      ...block.m ? { meta: block.m } : {}
-    };
-  }
-  return block;
-}
-function isRenderableBlock(block) {
-  if (!block || typeof block !== "object") return false;
-  const b = block;
-  if (b.type === "text" && typeof b.content === "string" && b.content.length > 0)
-    return true;
-  if (b.type === "component" && typeof b.component === "string" && b.component.length >= 2 && typeof b.props === "object" && b.props !== null)
-    return true;
-  return false;
-}
-function extractCompleteBlocks(accumulated) {
-  const blocksIdx = accumulated.indexOf('"blocks"');
-  let arrayStart = -1;
-  if (blocksIdx !== -1) {
-    arrayStart = accumulated.indexOf("[", blocksIdx);
-  } else {
-    arrayStart = accumulated.indexOf("[");
-  }
-  if (arrayStart === -1) return [];
-  const content = accumulated.slice(arrayStart + 1);
-  const blocks = [];
-  let depth = 0;
-  let blockStart = -1;
-  let inString = false;
-  let escaped = false;
-  for (let i = 0; i < content.length; i++) {
-    const ch = content[i];
-    if (escaped) {
-      escaped = false;
-      continue;
-    }
-    if (ch === "\\" && inString) {
-      escaped = true;
-      continue;
-    }
-    if (ch === '"') {
-      inString = !inString;
-      continue;
-    }
-    if (inString) continue;
-    if (ch === "{") {
-      if (depth === 0) blockStart = i;
-      depth++;
-    } else if (ch === "}") {
-      depth--;
-      if (depth === 0 && blockStart !== -1) {
-        try {
-          const parsed = JSON.parse(content.slice(blockStart, i + 1));
-          const expanded = expandWireBlock(parsed);
-          if (isRenderableBlock(expanded)) blocks.push(expanded);
-        } catch {
-        }
-        blockStart = -1;
-      }
-    }
-  }
-  return blocks;
-}
-function parseStreamingBlocks(accumulated) {
-  if (!accumulated.trim()) return [];
-  let target = accumulated;
-  const blocksIdx = accumulated.indexOf('"blocks"');
-  if (blocksIdx > 0) {
-    let braceIdx = blocksIdx - 1;
-    while (braceIdx >= 0 && accumulated[braceIdx] !== "{") braceIdx--;
-    if (braceIdx > 0) target = accumulated.slice(braceIdx);
-  }
-  const patched = completePartialJson(target);
-  try {
-    const parsed = JSON.parse(patched);
-    const rawBlocks = parsed?.blocks ?? (Array.isArray(parsed) ? parsed : []);
-    return rawBlocks.map((b) => {
-      if (!b || typeof b !== "object") return b;
-      return expandWireBlock(b);
-    }).filter(isRenderableBlock).map((block) => {
-      if (!("version" in block)) {
-        return Object.assign({}, block, { version: "1.0" });
-      }
-      return block;
-    });
-  } catch {
-    return extractCompleteBlocks(accumulated);
-  }
-}
-
-// src/use-block-stream.ts
 function useBlockStream() {
   const [blocks, setBlocks] = useState([]);
   const [isStreaming, setIsStreaming] = useState(false);
@@ -345,9 +203,6 @@ function useBlockStream() {
   const pendingRef = useRef2(null);
   const rafRef = useRef2(0);
   const streamingRef = useRef2(false);
-  const accumulatedRef = useRef2("");
-  const designReceivedRef = useRef2(false);
-  const bufferedBlocksRef = useRef2(null);
   const flush = useCallback(() => {
     rafRef.current = 0;
     if (pendingRef.current !== null) {
@@ -355,7 +210,7 @@ function useBlockStream() {
       pendingRef.current = null;
     }
   }, []);
-  const push = useCallback(
+  const batchBlocks = useCallback(
     (newBlocks) => {
       if (!streamingRef.current) {
         streamingRef.current = true;
@@ -368,10 +223,7 @@ function useBlockStream() {
     },
     [flush]
   );
-  const setDesign = useCallback((tokens) => {
-    setDesignState(tokens);
-  }, []);
-  const complete = useCallback(() => {
+  const handleComplete = useCallback(() => {
     if (rafRef.current) {
       cancelAnimationFrame(rafRef.current);
       rafRef.current = 0;
@@ -383,71 +235,101 @@ function useBlockStream() {
     streamingRef.current = false;
     setIsStreaming(false);
   }, []);
+  const streamRef = useRef2(null);
+  const getStream = useCallback(() => {
+    if (streamRef.current === null || streamRef.current.streamState === "destroyed") {
+      streamRef.current = new OutkitStream({
+        onBlocks: (b) => batchBlocks(b),
+        onDesign: (d) => setDesignState(d),
+        onComplete: () => handleComplete()
+      });
+    }
+    return streamRef.current;
+  }, [batchBlocks, handleComplete]);
+  const stream = getStream();
+  useEffect2(() => {
+    const s = getStream();
+    return () => {
+      if (rafRef.current) cancelAnimationFrame(rafRef.current);
+      s.destroy();
+    };
+  }, [getStream]);
   const reset = useCallback(() => {
     if (rafRef.current) {
       cancelAnimationFrame(rafRef.current);
       rafRef.current = 0;
     }
     pendingRef.current = null;
-    accumulatedRef.current = "";
-    bufferedBlocksRef.current = null;
-    designReceivedRef.current = false;
     streamingRef.current = false;
+    getStream().reset();
     setBlocks([]);
     setIsStreaming(false);
     setDesignState(void 0);
-  }, []);
+  }, [getStream]);
+  const push = useCallback(
+    (newBlocks) => batchBlocks(newBlocks),
+    [batchBlocks]
+  );
+  const setDesign = useCallback(
+    (tokens) => getStream().feedMeta(tokens),
+    [getStream]
+  );
+  const complete = useCallback(() => getStream().feedDone(), [getStream]);
+  const feedResponse = useCallback(
+    (response) => getStream().feedResponse(response),
+    [getStream]
+  );
+  const feedSSE = useCallback(
+    (rawText) => getStream().feedSSE(rawText),
+    [getStream]
+  );
+  const feedEvent = useCallback(
+    (data) => getStream().feedEvent(data),
+    [getStream]
+  );
   const feedChunk = useCallback(
-    (data) => {
-      accumulatedRef.current += data;
-      const parsed = parseStreamingBlocks(accumulatedRef.current);
-      if (parsed.length > 0) {
-        if (designReceivedRef.current) {
-          push(parsed);
-        } else {
-          bufferedBlocksRef.current = parsed;
-        }
-      }
-    },
-    [push]
+    (data) => getStream().feedChunk(data),
+    [getStream]
   );
   const feedMeta = useCallback(
-    (tokens) => {
-      designReceivedRef.current = true;
-      setDesignState(tokens);
-      if (bufferedBlocksRef.current) {
-        push(bufferedBlocksRef.current);
-        bufferedBlocksRef.current = null;
-      }
-    },
-    [push]
+    (tokens) => getStream().feedMeta(tokens),
+    [getStream]
   );
-  const feedDone = useCallback(() => {
-    if (accumulatedRef.current) {
-      const parsed = parseStreamingBlocks(accumulatedRef.current);
-      if (parsed.length > 0) {
-        pendingRef.current = parsed;
-      }
-    }
-    accumulatedRef.current = "";
-    bufferedBlocksRef.current = null;
-    complete();
-  }, [complete]);
-  useEffect2(() => {
-    return () => {
-      if (rafRef.current) {
-        cancelAnimationFrame(rafRef.current);
-      }
-    };
-  }, []);
-  return { blocks, isStreaming, design, feedChunk, feedMeta, feedDone, push, setDesign, complete, reset };
+  const feedDone = useCallback(
+    () => getStream().feedDone(),
+    [getStream]
+  );
+  return {
+    blocks,
+    isStreaming,
+    design,
+    feedResponse,
+    feedSSE,
+    feedEvent,
+    feedChunk,
+    feedMeta,
+    feedDone,
+    push,
+    setDesign,
+    complete,
+    reset
+  };
 }
 
 // src/index.ts
+import { OutkitStream as OutkitStream2 } from "@outkit-dev/core";
+import {
+  parseStreamingBlocks,
+  completePartialJson,
+  expandWireBlock,
+  isRenderableBlock,
+  extractCompleteBlocks
+} from "@outkit-dev/core";
 import { OUTKIT_TOKENS } from "@outkit-dev/renderer";
 export {
   AIRenderer,
   OUTKIT_TOKENS,
+  OutkitStream2 as OutkitStream,
   completePartialJson,
   expandWireBlock,
   extractCompleteBlocks,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@outkit-dev/react",
-  "version": "0.0.5",
+  "version": "0.0.7",
   "type": "module",
   "main": "./dist/index.cjs",
   "module": "./dist/index.js",
@@ -16,7 +16,8 @@
     "dist"
   ],
   "dependencies": {
-    "@outkit-dev/renderer": "0.0.5"
+    "@outkit-dev/renderer": "0.0.5",
+    "@outkit-dev/core": "0.0.7"
   },
   "peerDependencies": {
     "react": "^18.0.0 || ^19.0.0"