@assistant-ui/react-generative-ui 0.0.3 → 0.0.4

package/dist/renderGenerativeUI.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"renderGenerativeUI.d.ts","names":[],"sources":["../src/renderGenerativeUI.tsx"],"mappings":";;;;;;AAoBA;;;;;;;iBAAgB,kBAAA,CACd,IAAA,WACA,OAAA,EAAS,mBAAA,EACT,OAAA,GAAS,yBAAA,GACR,SAAA"}
+{"version":3,"file":"renderGenerativeUI.d.ts","names":[],"sources":["../src/renderGenerativeUI.tsx"],"mappings":";;;;;;AAqBA;;;;;;;iBAAgB,kBAAA,CACd,IAAA,WACA,OAAA,EAAS,mBAAA,EACT,OAAA,GAAS,yBAAA,GACR,SAAA"}

package/dist/renderGenerativeUI.js

@@ -1,4 +1,5 @@
 import { TYPE_KEY } from "./constants.js";
+import { getPartialJsonObjectMeta } from "assistant-stream/utils";
 import { Fragment } from "react";
 import { jsx } from "react/jsx-runtime";
 //#region src/renderGenerativeUI.tsx
@@ -13,7 +14,8 @@ const DEFAULT_CONTEXT = { status: "done" };
 * recursively so components can nest.
 */
 function renderGenerativeUI(node, library, context = DEFAULT_CONTEXT) {
-	return renderNode(normalizeNode(node), library, context);
+	const meta = getPartialJsonObjectMeta(node);
+	return renderNode(normalizeNode(node, meta?.state === "partial" ? meta.partialPath : void 0), library, context);
 }
 /**
 * The deepest tree we normalize. The input comes from the model, so a runaway
@@ -22,21 +24,32 @@ function renderGenerativeUI(node, library, context = DEFAULT_CONTEXT) {
 * bounds rendering too, since it only walks the normalized tree.
 */
 const MAX_DEPTH = 64;
-/** Converts the flat wire form into a normalized {@link GenerativeUINode}. */
-function normalizeNode(node, depth = 0) {
+/**
+* Converts the flat wire form into a normalized {@link GenerativeUINode}.
+*
+* `partialPath` is the remaining segment of the parse meta's partial path
+* relative to `node` (`undefined` once the walk leaves the partial frontier,
+* i.e. everything below is complete).
+*/
+function normalizeNode(node, partialPath, depth = 0) {
 	if (depth > MAX_DEPTH) return null;
 	if (node == null || typeof node === "boolean") return null;
 	if (typeof node === "string" || typeof node === "number") return node;
-	if (Array.isArray(node)) return node.map((child) => normalizeNode(child, depth + 1));
+	if (Array.isArray(node)) return node.map((child, index) => normalizeNode(child, descend(partialPath, String(index)), depth + 1));
 	if (typeof node !== "object") return null;
 	const { [TYPE_KEY]: type, ...props } = node;
 	if (typeof type !== "string") return null;
-	if ("children" in props) props["children"] = normalizeNode(props["children"], depth + 1);
+	if (partialPath?.length === 1 && partialPath[0] === "$type") return null;
+	if ("children" in props) props["children"] = normalizeNode(props["children"], descend(partialPath, "children"), depth + 1);
 	return {
 		type,
 		props
 	};
 }
+/** Steps the partial path down into `key`; siblings of the path are complete. */
+function descend(partialPath, key) {
+	return partialPath?.[0] === key ? partialPath.slice(1) : void 0;
+}
 function renderNode(node, library, context) {
 	if (node == null || typeof node === "boolean") return null;
 	if (typeof node === "string" || typeof node === "number") return node;

package/dist/renderGenerativeUI.js.map

@@ -1 +1 @@
-{"version":3,"file":"renderGenerativeUI.js","names":[],"sources":["../src/renderGenerativeUI.tsx"],"sourcesContent":["import { Fragment, type ReactNode } from \"react\";\nimport { TYPE_KEY } from \"./constants\";\nimport type {\n  GenerativeUIElement,\n  GenerativeUILibrary,\n  GenerativeUINode,\n  GenerativeUIRenderContext,\n} from \"./types\";\n\nconst DEFAULT_CONTEXT: GenerativeUIRenderContext = { status: \"done\" };\n\n/**\n * Renders a generative-ui tree against a {@link GenerativeUILibrary}.\n *\n * The model emits each node as a flat object `{ $type, ...props }`. We first\n * normalize that wire form into React-shaped elements (`{ type, props }`), then\n * render: each `type` is looked up in the library and its `props` are passed\n * to the component's `render(props, context)`, with `children` rendered\n * recursively so components can nest.\n */\nexport function renderGenerativeUI(\n  node: unknown,\n  library: GenerativeUILibrary,\n  context: GenerativeUIRenderContext = DEFAULT_CONTEXT,\n): ReactNode {\n  return renderNode(normalizeNode(node), library, context);\n}\n\n/**\n * The deepest tree we normalize. The input comes from the model, so a runaway\n * or adversarial response could nest arbitrarily deep and overflow the stack;\n * past this depth we stop (far beyond any real UI). Bounding normalization\n * bounds rendering too, since it only walks the normalized tree.\n */\nconst MAX_DEPTH = 64;\n\n/** Converts the flat wire form into a normalized {@link GenerativeUINode}. */\nfunction normalizeNode(node: unknown, depth = 0): GenerativeUINode {\n  if (depth > MAX_DEPTH) return null;\n  if (node == null || typeof node === \"boolean\") return null;\n  if (typeof node === \"string\" || typeof node === \"number\") return node;\n  if (Array.isArray(node))\n    return node.map((child) => normalizeNode(child, depth + 1));\n  if (typeof node !== \"object\") return null;\n\n  const { [TYPE_KEY]: type, ...props } = node as Record<string, unknown>;\n  // Args stream in incrementally; a node whose `$type` has not arrived yet is\n  // not an error, it just isn't renderable.\n  if (typeof type !== \"string\") return null;\n\n  if (\"children\" in props) {\n    props[\"children\"] = normalizeNode(props[\"children\"], depth + 1);\n  }\n  return { type, props } as GenerativeUIElement;\n}\n\nfunction renderNode(\n  node: GenerativeUINode,\n  library: GenerativeUILibrary,\n  context: GenerativeUIRenderContext,\n): ReactNode {\n  if (node == null || typeof node === \"boolean\") return null;\n  if (typeof node === \"string\" || typeof node === \"number\") return node;\n  if (Array.isArray(node)) {\n    // The wire format has no per-node id, so the key is positional. Pairing the\n    // index with the node's kind means that when the model splices or reorders\n    // `children` and the kind at an index changes, the key changes and React\n    // remounts instead of handing a streaming node's hook state to a different\n    // component.\n    return node.map((child, index) => (\n      <Fragment key={`${index}:${nodeKind(child)}`}>\n        {renderNode(child, library, context)}\n      </Fragment>\n    ));\n  }\n  return renderElement(node, library, context);\n}\n\nfunction renderElement(\n  element: GenerativeUIElement,\n  library: GenerativeUILibrary,\n  context: GenerativeUIRenderContext,\n): ReactNode {\n  const entry = library[element.type];\n  if (!entry) {\n    reportUnknownComponent(element.type, Object.keys(library));\n    return null;\n  }\n\n  // Components that opt out of prop streaming wait until their props are\n  // complete rather than rendering from a partial parse.\n  if (!entry.streamProperties && context.status === \"streaming\") return null;\n\n  // Inject the framework props last so the model can never override them.\n  const { children, ...rest } = element.props;\n  const props: Record<string, unknown> = { ...rest, $status: context.status };\n  if (children !== undefined) {\n    props[\"children\"] = renderNode(children, library, context);\n  }\n\n  return <GenerativeUIComponentRenderer render={entry.render} props={props} />;\n}\n\n/**\n * Mounts a single node's `render` on its own fiber so the function may use\n * hooks and hold state independently of its siblings and parent.\n */\nfunction GenerativeUIComponentRenderer({\n  render,\n  props,\n}: {\n  render: (props: any) => ReactNode;\n  props: Record<string, unknown>;\n}): ReactNode {\n  return render(props);\n}\n\n/** A coarse kind tag for a child, used in its list key so a node changing kind\n * at a given index forces a remount rather than a wrong-fiber reuse. */\nfunction nodeKind(node: GenerativeUINode): string {\n  if (node == null || typeof node === \"boolean\") return \"\";\n  if (typeof node === \"string\" || typeof node === \"number\") return \"#text\";\n  if (Array.isArray(node)) return \"#array\";\n  return node.type;\n}\n\nfunction reportUnknownComponent(type: string, available: string[]): void {\n  if (process.env[\"NODE_ENV\"] !== \"production\") {\n    // eslint-disable-next-line no-console\n    console.error(\n      `[@assistant-ui/react-generative-ui] Unknown component \"${type}\". ` +\n        `Available components: ${available.join(\", \") || \"(none)\"}.`,\n    );\n  }\n}\n"],"mappings":";;;;AASA,MAAM,kBAA6C,EAAE,QAAQ,OAAO;;;;;;;;;;AAWpE,SAAgB,mBACd,MACA,SACA,UAAqC,iBAC1B;CACX,OAAO,WAAW,cAAc,IAAI,GAAG,SAAS,OAAO;AACzD;;;;;;;AAQA,MAAM,YAAY;;AAGlB,SAAS,cAAc,MAAe,QAAQ,GAAqB;CACjE,IAAI,QAAQ,WAAW,OAAO;CAC9B,IAAI,QAAQ,QAAQ,OAAO,SAAS,WAAW,OAAO;CACtD,IAAI,OAAO,SAAS,YAAY,OAAO,SAAS,UAAU,OAAO;CACjE,IAAI,MAAM,QAAQ,IAAI,GACpB,OAAO,KAAK,KAAK,UAAU,cAAc,OAAO,QAAQ,CAAC,CAAC;CAC5D,IAAI,OAAO,SAAS,UAAU,OAAO;CAErC,MAAM,GAAG,WAAW,MAAM,GAAG,UAAU;CAGvC,IAAI,OAAO,SAAS,UAAU,OAAO;CAErC,IAAI,cAAc,OAChB,MAAM,cAAc,cAAc,MAAM,aAAa,QAAQ,CAAC;CAEhE,OAAO;EAAE;EAAM;CAAM;AACvB;AAEA,SAAS,WACP,MACA,SACA,SACW;CACX,IAAI,QAAQ,QAAQ,OAAO,SAAS,WAAW,OAAO;CACtD,IAAI,OAAO,SAAS,YAAY,OAAO,SAAS,UAAU,OAAO;CACjE,IAAI,MAAM,QAAQ,IAAI,GAMpB,OAAO,KAAK,KAAK,OAAO,UACtB,oBAAC,UAAD,EAAA,UACG,WAAW,OAAO,SAAS,OAAO,EAC3B,GAFK,GAAG,MAAM,GAAG,SAAS,KAAK,GAE/B,CACX;CAEH,OAAO,cAAc,MAAM,SAAS,OAAO;AAC7C;AAEA,SAAS,cACP,SACA,SACA,SACW;CACX,MAAM,QAAQ,QAAQ,QAAQ;CAC9B,IAAI,CAAC,OAAO;EACV,uBAAuB,QAAQ,MAAM,OAAO,KAAK,OAAO,CAAC;EACzD,OAAO;CACT;CAIA,IAAI,CAAC,MAAM,oBAAoB,QAAQ,WAAW,aAAa,OAAO;CAGtE,MAAM,EAAE,UAAU,GAAG,SAAS,QAAQ;CACtC,MAAM,QAAiC;EAAE,GAAG;EAAM,SAAS,QAAQ;CAAO;CAC1E,IAAI,aAAa,KAAA,GACf,MAAM,cAAc,WAAW,UAAU,SAAS,OAAO;CAG3D,OAAO,oBAAC,+BAAD;EAA+B,QAAQ,MAAM;EAAe;CAAQ,CAAA;AAC7E;;;;;AAMA,SAAS,8BAA8B,EACrC,QACA,SAIY;CACZ,OAAO,OAAO,KAAK;AACrB;;;AAIA,SAAS,SAAS,MAAgC;CAChD,IAAI,QAAQ,QAAQ,OAAO,SAAS,WAAW,OAAO;CACtD,IAAI,OAAO,SAAS,YAAY,OAAO,SAAS,UAAU,OAAO;CACjE,IAAI,MAAM,QAAQ,IAAI,GAAG,OAAO;CAChC,OAAO,KAAK;AACd;AAEA,SAAS,uBAAuB,MAAc,WAA2B;CACvE,IAAI,QAAQ,IAAI,gBAAgB,cAE9B,QAAQ,MACN,0DAA0D,KAAK,2BACpC,UAAU,KAAK,IAAI,KAAK,SAAS,EAC9D;AAEJ"}
+{"version":3,"file":"renderGenerativeUI.js","names":[],"sources":["../src/renderGenerativeUI.tsx"],"sourcesContent":["import { getPartialJsonObjectMeta } from \"assistant-stream/utils\";\nimport { Fragment, type ReactNode } from \"react\";\nimport { TYPE_KEY } from \"./constants\";\nimport type {\n  GenerativeUIElement,\n  GenerativeUILibrary,\n  GenerativeUINode,\n  GenerativeUIRenderContext,\n} from \"./types\";\n\nconst DEFAULT_CONTEXT: GenerativeUIRenderContext = { status: \"done\" };\n\n/**\n * Renders a generative-ui tree against a {@link GenerativeUILibrary}.\n *\n * The model emits each node as a flat object `{ $type, ...props }`. We first\n * normalize that wire form into React-shaped elements (`{ type, props }`), then\n * render: each `type` is looked up in the library and its `props` are passed\n * to the component's `render(props, context)`, with `children` rendered\n * recursively so components can nest.\n */\nexport function renderGenerativeUI(\n  node: unknown,\n  library: GenerativeUILibrary,\n  context: GenerativeUIRenderContext = DEFAULT_CONTEXT,\n): ReactNode {\n  // Tool args are parsed incrementally, and the parse meta records which path\n  // is still mid-arrival, so normalization can hold back a node whose `$type`\n  // string has not finished streaming.\n  const meta = getPartialJsonObjectMeta(node as Record<symbol, unknown>);\n  const partialPath = meta?.state === \"partial\" ? meta.partialPath : undefined;\n  return renderNode(normalizeNode(node, partialPath), library, context);\n}\n\n/**\n * The deepest tree we normalize. The input comes from the model, so a runaway\n * or adversarial response could nest arbitrarily deep and overflow the stack;\n * past this depth we stop (far beyond any real UI). Bounding normalization\n * bounds rendering too, since it only walks the normalized tree.\n */\nconst MAX_DEPTH = 64;\n\n/**\n * Converts the flat wire form into a normalized {@link GenerativeUINode}.\n *\n * `partialPath` is the remaining segment of the parse meta's partial path\n * relative to `node` (`undefined` once the walk leaves the partial frontier,\n * i.e. everything below is complete).\n */\nfunction normalizeNode(\n  node: unknown,\n  partialPath: readonly string[] | undefined,\n  depth = 0,\n): GenerativeUINode {\n  if (depth > MAX_DEPTH) return null;\n  if (node == null || typeof node === \"boolean\") return null;\n  if (typeof node === \"string\" || typeof node === \"number\") return node;\n  if (Array.isArray(node))\n    return node.map((child, index) =>\n      normalizeNode(child, descend(partialPath, String(index)), depth + 1),\n    );\n  if (typeof node !== \"object\") return null;\n\n  const { [TYPE_KEY]: type, ...props } = node as Record<string, unknown>;\n  // Args stream in incrementally; a node whose `$type` has not arrived yet\n  // (or whose `$type` string is still mid-arrival) is not an error, it just\n  // isn't renderable.\n  if (typeof type !== \"string\") return null;\n  if (partialPath?.length === 1 && partialPath[0] === TYPE_KEY) return null;\n\n  if (\"children\" in props) {\n    props[\"children\"] = normalizeNode(\n      props[\"children\"],\n      descend(partialPath, \"children\"),\n      depth + 1,\n    );\n  }\n  return { type, props } as GenerativeUIElement;\n}\n\n/** Steps the partial path down into `key`; siblings of the path are complete. */\nfunction descend(\n  partialPath: readonly string[] | undefined,\n  key: string,\n): readonly string[] | undefined {\n  return partialPath?.[0] === key ? partialPath.slice(1) : undefined;\n}\n\nfunction renderNode(\n  node: GenerativeUINode,\n  library: GenerativeUILibrary,\n  context: GenerativeUIRenderContext,\n): ReactNode {\n  if (node == null || typeof node === \"boolean\") return null;\n  if (typeof node === \"string\" || typeof node === \"number\") return node;\n  if (Array.isArray(node)) {\n    // The wire format has no per-node id, so the key is positional. Pairing the\n    // index with the node's kind means that when the model splices or reorders\n    // `children` and the kind at an index changes, the key changes and React\n    // remounts instead of handing a streaming node's hook state to a different\n    // component.\n    return node.map((child, index) => (\n      <Fragment key={`${index}:${nodeKind(child)}`}>\n        {renderNode(child, library, context)}\n      </Fragment>\n    ));\n  }\n  return renderElement(node, library, context);\n}\n\nfunction renderElement(\n  element: GenerativeUIElement,\n  library: GenerativeUILibrary,\n  context: GenerativeUIRenderContext,\n): ReactNode {\n  const entry = library[element.type];\n  if (!entry) {\n    reportUnknownComponent(element.type, Object.keys(library));\n    return null;\n  }\n\n  // Components that opt out of prop streaming wait until their props are\n  // complete rather than rendering from a partial parse.\n  if (!entry.streamProperties && context.status === \"streaming\") return null;\n\n  // Inject the framework props last so the model can never override them.\n  const { children, ...rest } = element.props;\n  const props: Record<string, unknown> = { ...rest, $status: context.status };\n  if (children !== undefined) {\n    props[\"children\"] = renderNode(children, library, context);\n  }\n\n  return <GenerativeUIComponentRenderer render={entry.render} props={props} />;\n}\n\n/**\n * Mounts a single node's `render` on its own fiber so the function may use\n * hooks and hold state independently of its siblings and parent.\n */\nfunction GenerativeUIComponentRenderer({\n  render,\n  props,\n}: {\n  render: (props: any) => ReactNode;\n  props: Record<string, unknown>;\n}): ReactNode {\n  return render(props);\n}\n\n/** A coarse kind tag for a child, used in its list key so a node changing kind\n * at a given index forces a remount rather than a wrong-fiber reuse. */\nfunction nodeKind(node: GenerativeUINode): string {\n  if (node == null || typeof node === \"boolean\") return \"\";\n  if (typeof node === \"string\" || typeof node === \"number\") return \"#text\";\n  if (Array.isArray(node)) return \"#array\";\n  return node.type;\n}\n\nfunction reportUnknownComponent(type: string, available: string[]): void {\n  if (process.env[\"NODE_ENV\"] !== \"production\") {\n    // eslint-disable-next-line no-console\n    console.error(\n      `[@assistant-ui/react-generative-ui] Unknown component \"${type}\". ` +\n        `Available components: ${available.join(\", \") || \"(none)\"}.`,\n    );\n  }\n}\n"],"mappings":";;;;;AAUA,MAAM,kBAA6C,EAAE,QAAQ,OAAO;;;;;;;;;;AAWpE,SAAgB,mBACd,MACA,SACA,UAAqC,iBAC1B;CAIX,MAAM,OAAO,yBAAyB,IAA+B;CAErE,OAAO,WAAW,cAAc,MADZ,MAAM,UAAU,YAAY,KAAK,cAAc,KAAA,CAClB,GAAG,SAAS,OAAO;AACtE;;;;;;;AAQA,MAAM,YAAY;;;;;;;;AASlB,SAAS,cACP,MACA,aACA,QAAQ,GACU;CAClB,IAAI,QAAQ,WAAW,OAAO;CAC9B,IAAI,QAAQ,QAAQ,OAAO,SAAS,WAAW,OAAO;CACtD,IAAI,OAAO,SAAS,YAAY,OAAO,SAAS,UAAU,OAAO;CACjE,IAAI,MAAM,QAAQ,IAAI,GACpB,OAAO,KAAK,KAAK,OAAO,UACtB,cAAc,OAAO,QAAQ,aAAa,OAAO,KAAK,CAAC,GAAG,QAAQ,CAAC,CACrE;CACF,IAAI,OAAO,SAAS,UAAU,OAAO;CAErC,MAAM,GAAG,WAAW,MAAM,GAAG,UAAU;CAIvC,IAAI,OAAO,SAAS,UAAU,OAAO;CACrC,IAAI,aAAa,WAAW,KAAK,YAAY,OAAA,SAAiB,OAAO;CAErE,IAAI,cAAc,OAChB,MAAM,cAAc,cAClB,MAAM,aACN,QAAQ,aAAa,UAAU,GAC/B,QAAQ,CACV;CAEF,OAAO;EAAE;EAAM;CAAM;AACvB;;AAGA,SAAS,QACP,aACA,KAC+B;CAC/B,OAAO,cAAc,OAAO,MAAM,YAAY,MAAM,CAAC,IAAI,KAAA;AAC3D;AAEA,SAAS,WACP,MACA,SACA,SACW;CACX,IAAI,QAAQ,QAAQ,OAAO,SAAS,WAAW,OAAO;CACtD,IAAI,OAAO,SAAS,YAAY,OAAO,SAAS,UAAU,OAAO;CACjE,IAAI,MAAM,QAAQ,IAAI,GAMpB,OAAO,KAAK,KAAK,OAAO,UACtB,oBAAC,UAAD,EAAA,UACG,WAAW,OAAO,SAAS,OAAO,EAC3B,GAFK,GAAG,MAAM,GAAG,SAAS,KAAK,GAE/B,CACX;CAEH,OAAO,cAAc,MAAM,SAAS,OAAO;AAC7C;AAEA,SAAS,cACP,SACA,SACA,SACW;CACX,MAAM,QAAQ,QAAQ,QAAQ;CAC9B,IAAI,CAAC,OAAO;EACV,uBAAuB,QAAQ,MAAM,OAAO,KAAK,OAAO,CAAC;EACzD,OAAO;CACT;CAIA,IAAI,CAAC,MAAM,oBAAoB,QAAQ,WAAW,aAAa,OAAO;CAGtE,MAAM,EAAE,UAAU,GAAG,SAAS,QAAQ;CACtC,MAAM,QAAiC;EAAE,GAAG;EAAM,SAAS,QAAQ;CAAO;CAC1E,IAAI,aAAa,KAAA,GACf,MAAM,cAAc,WAAW,UAAU,SAAS,OAAO;CAG3D,OAAO,oBAAC,+BAAD;EAA+B,QAAQ,MAAM;EAAe;CAAQ,CAAA;AAC7E;;;;;AAMA,SAAS,8BAA8B,EACrC,QACA,SAIY;CACZ,OAAO,OAAO,KAAK;AACrB;;;AAIA,SAAS,SAAS,MAAgC;CAChD,IAAI,QAAQ,QAAQ,OAAO,SAAS,WAAW,OAAO;CACtD,IAAI,OAAO,SAAS,YAAY,OAAO,SAAS,UAAU,OAAO;CACjE,IAAI,MAAM,QAAQ,IAAI,GAAG,OAAO;CAChC,OAAO,KAAK;AACd;AAEA,SAAS,uBAAuB,MAAc,WAA2B;CACvE,IAAI,QAAQ,IAAI,gBAAgB,cAE9B,QAAQ,MACN,0DAA0D,KAAK,2BACpC,UAAU,KAAK,IAAI,KAAK,SAAS,EAC9D;AAEJ"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/react-generative-ui",
-  "version": "0.0.3",
+  "version": "0.0.4",
   "description": "Generative UI components for assistant-ui",
   "keywords": [
     "generative-ui",
@@ -58,7 +58,7 @@
     "react-dom": "^19.2.7",
     "vitest": "^4.1.8",
     "zod": "^4.4.3",
-    "@assistant-ui/react": "0.14.15",
+    "@assistant-ui/react": "0.14.18",
     "@assistant-ui/x-buildutils": "0.0.12"
   },
   "publishConfig": {

package/src/renderGenerativeUI.test.tsx

@@ -1,5 +1,6 @@
-import { describe, it, expect } from "vitest";
+import { describe, it, expect, vi } from "vitest";
 import { renderToStaticMarkup } from "react-dom/server";
+import { parsePartialJsonObject } from "assistant-stream/utils";
 import { z } from "zod";
 import { renderGenerativeUI } from "./renderGenerativeUI";
 import { buildPresentParameters } from "./buildPresentParameters";
@@ -90,6 +91,41 @@ describe("renderGenerativeUI", () => {
     expect(html).toBe("");
   });
 
+  it("holds back a node whose `$type` is still streaming", () => {
+    const error = vi.spyOn(console, "error").mockImplementation(() => {});
+    try {
+      for (const argsText of ['{"$type": "', '{"$type": "Tex']) {
+        const html = renderToStaticMarkup(
+          <>
+            {renderGenerativeUI(parsePartialJsonObject(argsText), library, {
+              status: "streaming",
+            })}
+          </>,
+        );
+        expect(html).toBe("");
+      }
+      expect(error).not.toHaveBeenCalled();
+    } finally {
+      error.mockRestore();
+    }
+  });
+
+  it("renders completed siblings while a nested `$type` streams", () => {
+    const error = vi.spyOn(console, "error").mockImplementation(() => {});
+    try {
+      const args = parsePartialJsonObject(
+        '{"$type": "Live", "label": "hi", "children": [{"$type": "Liv',
+      );
+      const html = renderToStaticMarkup(
+        <>{renderGenerativeUI(args, library, { status: "streaming" })}</>,
+      );
+      expect(html).toBe('<span data-status="streaming">hi</span>');
+      expect(error).not.toHaveBeenCalled();
+    } finally {
+      error.mockRestore();
+    }
+  });
+
   it("renders nothing for a node without a resolved type", () => {
     const html = renderToStaticMarkup(<>{renderGenerativeUI({}, library)}</>);
     expect(html).toBe("");

package/src/renderGenerativeUI.tsx

@@ -1,3 +1,4 @@
+import { getPartialJsonObjectMeta } from "assistant-stream/utils";
 import { Fragment, type ReactNode } from "react";
 import { TYPE_KEY } from "./constants";
 import type {
@@ -23,7 +24,12 @@ export function renderGenerativeUI(
   library: GenerativeUILibrary,
   context: GenerativeUIRenderContext = DEFAULT_CONTEXT,
 ): ReactNode {
-  return renderNode(normalizeNode(node), library, context);
+  // Tool args are parsed incrementally, and the parse meta records which path
+  // is still mid-arrival, so normalization can hold back a node whose `$type`
+  // string has not finished streaming.
+  const meta = getPartialJsonObjectMeta(node as Record<symbol, unknown>);
+  const partialPath = meta?.state === "partial" ? meta.partialPath : undefined;
+  return renderNode(normalizeNode(node, partialPath), library, context);
 }
 
 /**
@@ -34,26 +40,52 @@ export function renderGenerativeUI(
  */
 const MAX_DEPTH = 64;
 
-/** Converts the flat wire form into a normalized {@link GenerativeUINode}. */
-function normalizeNode(node: unknown, depth = 0): GenerativeUINode {
+/**
+ * Converts the flat wire form into a normalized {@link GenerativeUINode}.
+ *
+ * `partialPath` is the remaining segment of the parse meta's partial path
+ * relative to `node` (`undefined` once the walk leaves the partial frontier,
+ * i.e. everything below is complete).
+ */
+function normalizeNode(
+  node: unknown,
+  partialPath: readonly string[] | undefined,
+  depth = 0,
+): GenerativeUINode {
   if (depth > MAX_DEPTH) return null;
   if (node == null || typeof node === "boolean") return null;
   if (typeof node === "string" || typeof node === "number") return node;
   if (Array.isArray(node))
-    return node.map((child) => normalizeNode(child, depth + 1));
+    return node.map((child, index) =>
+      normalizeNode(child, descend(partialPath, String(index)), depth + 1),
+    );
   if (typeof node !== "object") return null;
 
   const { [TYPE_KEY]: type, ...props } = node as Record<string, unknown>;
-  // Args stream in incrementally; a node whose `$type` has not arrived yet is
-  // not an error, it just isn't renderable.
+  // Args stream in incrementally; a node whose `$type` has not arrived yet
+  // (or whose `$type` string is still mid-arrival) is not an error, it just
+  // isn't renderable.
   if (typeof type !== "string") return null;
+  if (partialPath?.length === 1 && partialPath[0] === TYPE_KEY) return null;
 
   if ("children" in props) {
-    props["children"] = normalizeNode(props["children"], depth + 1);
+    props["children"] = normalizeNode(
+      props["children"],
+      descend(partialPath, "children"),
+      depth + 1,
+    );
   }
   return { type, props } as GenerativeUIElement;
 }
 
+/** Steps the partial path down into `key`; siblings of the path are complete. */
+function descend(
+  partialPath: readonly string[] | undefined,
+  key: string,
+): readonly string[] | undefined {
+  return partialPath?.[0] === key ? partialPath.slice(1) : undefined;
+}
+
 function renderNode(
   node: GenerativeUINode,
   library: GenerativeUILibrary,