@kubuild/ai 0.3.0 → 0.4.0

package/dist/core/document-outline.d.ts

@@ -0,0 +1,60 @@
+import type { Node, PageDocument } from '@kubuild/schema';
+/**
+ * Best-effort human label for a node, derived from its text-carrying props. Returns an
+ * empty string when the node has none (layout containers, mostly).
+ */
+export declare function getNodeLabel(node: Node, max?: number): string;
+export interface DocumentOutlineOptions {
+    /** How deep to walk before collapsing into a `(+N nested)` marker. Default: 3. */
+    maxDepth?: number;
+    /**
+     * When set, the branch containing this node is always expanded in full regardless of
+     * `maxDepth`, and the node itself is marked with `← SELECTED`. This is what makes
+     * "user selected a component on the canvas" legible to the model.
+     */
+    focusNodeId?: string;
+    /** Hard ceiling on emitted rows, so a huge page can't blow up the prompt. Default: 200. */
+    maxNodes?: number;
+}
+/**
+ * Renders a compact, indented outline of a node tree (STORA-530).
+ *
+ * This is the agent's *default* view of the document: one line per node with id, type,
+ * label and child count — never full props/styles. Detail is fetched on demand via the
+ * `read_node` tool, which keeps per-step prompt cost roughly constant no matter how big
+ * the page is. `engine.buildChatPrompt`'s depth-1 section summary can't do this: nested
+ * nodes were invisible to the model, so it could never target them.
+ */
+export declare function summarizeNodeTree(root: Node, options?: DocumentOutlineOptions): string;
+/** Total number of descendants below a node (not counting the node itself). */
+export declare function countDescendants(node: Node): number;
+/** Outline of a whole `PageDocument`, prefixed with its metadata. */
+export declare function summarizeDocument(document: PageDocument, options?: DocumentOutlineOptions): string;
+export interface SelectionContext {
+    node: Node;
+    /** Root-first chain of ancestors, excluding the node itself. */
+    ancestors: Node[];
+    /** The node's siblings, in document order, including the node itself. */
+    siblings: Node[];
+    index: number;
+    parentId: string | null;
+}
+/**
+ * Resolves everything the agent needs to know about the canvas selection (STORA-530).
+ *
+ * Sending only `selectedNodeId` — what `engine.buildChatPrompt` does today — tells the
+ * model an id but not what the node *is*, what it lives inside, or what sits next to it,
+ * so it can't judge whether an instruction like "make this wider" belongs on the node or
+ * on its parent container. Returns `null` when the id isn't in the tree.
+ */
+export declare function buildSelectionContext(root: Node, nodeId: string): SelectionContext | null;
+/**
+ * Depth-limits a node before it is serialized into a prompt. A selected section can carry
+ * hundreds of descendants; dumping all of them would defeat the point of the outline.
+ * Children below `maxDepth` are replaced by a marker node telling the model to call
+ * `read_node` if it needs them.
+ */
+export declare function pruneNodeForPrompt(node: Node, maxDepth?: number, depth?: number): Node;
+/** Renders a `SelectionContext` as the prompt block describing the current selection. */
+export declare function formatSelectionContext(context: SelectionContext): string;
+//# sourceMappingURL=document-outline.d.ts.map

package/dist/core/document-outline.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"document-outline.d.ts","sourceRoot":"","sources":["../../src/core/document-outline.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAmB1D;;;GAGG;AACH,wBAAgB,YAAY,CAAC,IAAI,EAAE,IAAI,EAAE,GAAG,SAAoB,GAAG,MAAM,CASxE;AAED,MAAM,WAAW,sBAAsB;IACrC,kFAAkF;IAClF,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB;;;;OAIG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,2FAA2F;IAC3F,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAOD;;;;;;;;GAQG;AACH,wBAAgB,iBAAiB,CAAC,IAAI,EAAE,IAAI,EAAE,OAAO,GAAE,sBAA2B,GAAG,MAAM,CA4C1F;AAED,+EAA+E;AAC/E,wBAAgB,gBAAgB,CAAC,IAAI,EAAE,IAAI,GAAG,MAAM,CAEnD;AAED,qEAAqE;AACrE,wBAAgB,iBAAiB,CAC/B,QAAQ,EAAE,YAAY,EACtB,OAAO,GAAE,sBAA2B,GACnC,MAAM,CAMR;AAED,MAAM,WAAW,gBAAgB;IAC/B,IAAI,EAAE,IAAI,CAAC;IACX,gEAAgE;IAChE,SAAS,EAAE,IAAI,EAAE,CAAC;IAClB,yEAAyE;IACzE,QAAQ,EAAE,IAAI,EAAE,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;CACzB;AAED;;;;;;;GAOG;AACH,wBAAgB,qBAAqB,CAAC,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,GAAG,gBAAgB,GAAG,IAAI,CAwBzF;AAED;;;;;GAKG;AACH,wBAAgB,kBAAkB,CAAC,IAAI,EAAE,IAAI,EAAE,QAAQ,SAAI,EAAE,KAAK,SAAI,GAAG,IAAI,CAuB5E;AAED,yFAAyF;AACzF,wBAAgB,sBAAsB,CAAC,OAAO,EAAE,gBAAgB,GAAG,MAAM,CAyBxE"}

package/dist/core/index.d.ts

@@ -1,3 +1,5 @@
 export * from './prompt-compiler';
 export * from './normalizer';
+export * from './messages';
+export * from './document-outline';
 //# sourceMappingURL=index.d.ts.map

package/dist/core/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/core/index.ts"],"names":[],"mappings":"AAAA,cAAc,mBAAmB,CAAC;AAClC,cAAc,cAAc,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/core/index.ts"],"names":[],"mappings":"AAAA,cAAc,mBAAmB,CAAC;AAClC,cAAc,cAAc,CAAC;AAC7B,cAAc,YAAY,CAAC;AAC3B,cAAc,oBAAoB,CAAC"}

package/dist/core/messages.d.ts

@@ -0,0 +1,18 @@
+import type { AiChatMessage, AiToolUseBlock, AiToolResultBlock } from '../types';
+/**
+ * Flattens a message's content to plain text (STORA-530).
+ *
+ * `AiChatMessage.content` became `string | AiContentBlock[]` when tool calling landed;
+ * every place that previously assumed a bare string (adapter message mapping, the chat
+ * prompt's "last user message" lookup, chat UIs) goes through this helper instead of
+ * re-implementing the narrowing. Tool-use/tool-result blocks contribute nothing to the
+ * text rendering — they're transported separately by the adapters.
+ */
+export declare function getMessageText(message: Pick<AiChatMessage, 'content'>): string;
+/** All `tool_use` blocks carried by a message, in order. Empty for plain-text messages. */
+export declare function getToolUseBlocks(message: Pick<AiChatMessage, 'content'>): AiToolUseBlock[];
+/** All `tool_result` blocks carried by a message, in order. */
+export declare function getToolResultBlocks(message: Pick<AiChatMessage, 'content'>): AiToolResultBlock[];
+/** True when the message carries any non-text block — i.e. adapters must map it structurally. */
+export declare function hasStructuredContent(message: Pick<AiChatMessage, 'content'>): boolean;
+//# sourceMappingURL=messages.d.ts.map

package/dist/core/messages.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"messages.d.ts","sourceRoot":"","sources":["../../src/core/messages.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAkB,cAAc,EAAE,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAEjG;;;;;;;;GAQG;AACH,wBAAgB,cAAc,CAAC,OAAO,EAAE,IAAI,CAAC,aAAa,EAAE,SAAS,CAAC,GAAG,MAAM,CAM9E;AAED,2FAA2F;AAC3F,wBAAgB,gBAAgB,CAAC,OAAO,EAAE,IAAI,CAAC,aAAa,EAAE,SAAS,CAAC,GAAG,cAAc,EAAE,CAG1F;AAED,+DAA+D;AAC/D,wBAAgB,mBAAmB,CAAC,OAAO,EAAE,IAAI,CAAC,aAAa,EAAE,SAAS,CAAC,GAAG,iBAAiB,EAAE,CAGhG;AAED,iGAAiG;AACjG,wBAAgB,oBAAoB,CAAC,OAAO,EAAE,IAAI,CAAC,aAAa,EAAE,SAAS,CAAC,GAAG,OAAO,CAErF"}

package/dist/core/prompt-compiler.d.ts

@@ -1,3 +1,4 @@
+import type { PageDocument } from '@kubuild/schema';
 import type { AiCompiledComponentSpec, ComponentRegistryLike, AiGenerationMode } from '../types';
 export declare function compileComponentCatalog(registry?: ComponentRegistryLike): AiCompiledComponentSpec[];
 export declare function buildSystemPrompt(options: {
@@ -7,4 +8,32 @@ export declare function buildSystemPrompt(options: {
     stylePreference?: string;
 }): string;
 export declare function buildJsonSchemaForMode(mode: AiGenerationMode): Record<string, unknown>;
+/**
+ * Renders the component catalog as the compact reference block shared by every prompt.
+ * Extracted from `buildSystemPrompt` so the agent prompt (STORA-530) reuses the exact
+ * same catalog rendering instead of drifting into a second format.
+ */
+export declare function formatComponentCatalog(catalog: AiCompiledComponentSpec[]): string;
+export interface BuildAgentSystemPromptOptions {
+    catalog: AiCompiledComponentSpec[];
+    document: PageDocument;
+    selectedNodeId?: string;
+    prefix?: string;
+    stylePreference?: string;
+    /** Extra host/product context appended verbatim. */
+    additionalContext?: string;
+}
+/**
+ * System prompt for agent mode (STORA-530).
+ *
+ * Deliberately different from `buildSystemPrompt`: the agent does NOT emit document JSON.
+ * It reads the page through tools and edits through tools, so the prompt's job is to
+ * establish (a) the outline as the shared map, (b) the selection as the default target,
+ * and (c) the surgical-edit discipline — the whole point of the feature is that asking to
+ * recolor one button must not regenerate its section.
+ *
+ * The catalog goes before the volatile per-turn context so the stable prefix stays
+ * prompt-cacheable across the steps of a run.
+ */
+export declare function buildAgentSystemPrompt(options: BuildAgentSystemPromptOptions): string;
 //# sourceMappingURL=prompt-compiler.d.ts.map

package/dist/core/prompt-compiler.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"prompt-compiler.d.ts","sourceRoot":"","sources":["../../src/core/prompt-compiler.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,uBAAuB,EAEvB,qBAAqB,EAErB,gBAAgB,EACjB,MAAM,UAAU,CAAC;AAElB,wBAAgB,uBAAuB,CACrC,QAAQ,CAAC,EAAE,qBAAqB,GAC/B,uBAAuB,EAAE,CA+B3B;AAED,wBAAgB,iBAAiB,CAAC,OAAO,EAAE;IACzC,OAAO,EAAE,uBAAuB,EAAE,CAAC;IACnC,IAAI,EAAE,gBAAgB,CAAC;IACvB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,eAAe,CAAC,EAAE,MAAM,CAAC;CAC1B,GAAG,MAAM,CAkHT;AAED,wBAAgB,sBAAsB,CAAC,IAAI,EAAE,gBAAgB,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CA8CtF"}
+{"version":3,"file":"prompt-compiler.d.ts","sourceRoot":"","sources":["../../src/core/prompt-compiler.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AACpD,OAAO,KAAK,EACV,uBAAuB,EAEvB,qBAAqB,EAErB,gBAAgB,EACjB,MAAM,UAAU,CAAC;AAOlB,wBAAgB,uBAAuB,CACrC,QAAQ,CAAC,EAAE,qBAAqB,GAC/B,uBAAuB,EAAE,CA+B3B;AAED,wBAAgB,iBAAiB,CAAC,OAAO,EAAE;IACzC,OAAO,EAAE,uBAAuB,EAAE,CAAC;IACnC,IAAI,EAAE,gBAAgB,CAAC;IACvB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,eAAe,CAAC,EAAE,MAAM,CAAC;CAC1B,GAAG,MAAM,CAkHT;AAED,wBAAgB,sBAAsB,CAAC,IAAI,EAAE,gBAAgB,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CA8CtF;AAGD;;;;GAIG;AACH,wBAAgB,sBAAsB,CAAC,OAAO,EAAE,uBAAuB,EAAE,GAAG,MAAM,CAsBjF;AAED,MAAM,WAAW,6BAA6B;IAC5C,OAAO,EAAE,uBAAuB,EAAE,CAAC;IACnC,QAAQ,EAAE,YAAY,CAAC;IACvB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,oDAAoD;IACpD,iBAAiB,CAAC,EAAE,MAAM,CAAC;CAC5B;AA4BD;;;;;;;;;;;GAWG;AACH,wBAAgB,sBAAsB,CAAC,OAAO,EAAE,6BAA6B,GAAG,MAAM,CA6CrF"}

package/dist/index.cjs

@@ -20,18 +20,167 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // src/index.ts
 var src_exports = {};
 __export(src_exports, {
+  buildAgentSystemPrompt: () => buildAgentSystemPrompt,
   buildJsonSchemaForMode: () => buildJsonSchemaForMode,
+  buildSelectionContext: () => buildSelectionContext,
   buildSystemPrompt: () => buildSystemPrompt,
   compileComponentCatalog: () => compileComponentCatalog,
+  countDescendants: () => countDescendants,
   extractJsonFromResponse: () => extractJsonFromResponse,
+  formatComponentCatalog: () => formatComponentCatalog,
+  formatSelectionContext: () => formatSelectionContext,
+  getMessageText: () => getMessageText,
+  getNodeLabel: () => getNodeLabel,
+  getToolResultBlocks: () => getToolResultBlocks,
+  getToolUseBlocks: () => getToolUseBlocks,
+  hasStructuredContent: () => hasStructuredContent,
   normalizeAndValidatePageDocument: () => normalizeAndValidatePageDocument,
   normalizeAndValidateRefactoredNode: () => normalizeAndValidateRefactoredNode,
   normalizeAndValidateSectionNode: () => normalizeAndValidateSectionNode,
   normalizeNodeTree: () => normalizeNodeTree,
-  normalizeStyles: () => normalizeStyles
+  normalizeStyles: () => normalizeStyles,
+  pruneNodeForPrompt: () => pruneNodeForPrompt,
+  summarizeDocument: () => summarizeDocument,
+  summarizeNodeTree: () => summarizeNodeTree
 });
 module.exports = __toCommonJS(src_exports);
 
+// src/core/document-outline.ts
+var LABEL_PROPS = ["text", "label", "title", "heading", "content", "placeholder", "alt", "src"];
+var DEFAULT_LABEL_MAX = 60;
+function truncate(value, max = DEFAULT_LABEL_MAX) {
+  const oneLine = value.replace(/\s+/g, " ").trim();
+  return oneLine.length > max ? `${oneLine.slice(0, max - 1)}\u2026` : oneLine;
+}
+function getNodeLabel(node, max = DEFAULT_LABEL_MAX) {
+  const props = node.props ?? {};
+  for (const key of LABEL_PROPS) {
+    const value = props[key];
+    if (typeof value === "string" && value.trim()) {
+      return truncate(value, max);
+    }
+  }
+  return "";
+}
+function isOnPathTo(node, targetId) {
+  if (node.id === targetId) return true;
+  return (node.children ?? []).some((child) => isOnPathTo(child, targetId));
+}
+function summarizeNodeTree(root, options = {}) {
+  const { maxDepth = 3, focusNodeId, maxNodes = 200 } = options;
+  const lines = [];
+  let emitted = 0;
+  let truncated = false;
+  const walk = (node, depth) => {
+    if (emitted >= maxNodes) {
+      truncated = true;
+      return;
+    }
+    const indent = "  ".repeat(depth);
+    const label = getNodeLabel(node);
+    const children = node.children ?? [];
+    const parts = [`${indent}- #${node.id} (${node.type})`];
+    if (label) parts.push(`"${label}"`);
+    if (children.length > 0) parts.push(`[${children.length} children]`);
+    if (focusNodeId && node.id === focusNodeId) parts.push("\u2190 SELECTED");
+    lines.push(parts.join(" "));
+    emitted++;
+    if (children.length === 0) return;
+    const onFocusPath = Boolean(focusNodeId) && isOnPathTo(node, focusNodeId);
+    if (depth + 1 > maxDepth && !onFocusPath) {
+      lines.push(`${indent}  (+${countDescendants(node)} nested nodes \u2014 use read_node to expand)`);
+      return;
+    }
+    for (const child of children) {
+      walk(child, depth + 1);
+    }
+  };
+  walk(root, 0);
+  if (truncated) {
+    lines.push(`  (outline truncated at ${maxNodes} nodes \u2014 use find_nodes to locate specifics)`);
+  }
+  return lines.join("\n");
+}
+function countDescendants(node) {
+  return (node.children ?? []).reduce((total, child) => total + 1 + countDescendants(child), 0);
+}
+function summarizeDocument(document, options = {}) {
+  const header = [
+    `Title: ${document.metadata?.title || "Untitled Page"}`,
+    `Description: ${document.metadata?.description || "N/A"}`
+  ].join("\n");
+  return `${header}
+
+Page outline:
+${summarizeNodeTree(document.document, options)}`;
+}
+function buildSelectionContext(root, nodeId) {
+  const ancestors = [];
+  const walk = (node, parent, index) => {
+    if (node.id === nodeId) {
+      return {
+        node,
+        ancestors: [...ancestors],
+        siblings: parent ? parent.children ?? [] : [node],
+        index,
+        parentId: parent?.id ?? null
+      };
+    }
+    ancestors.push(node);
+    const children = node.children ?? [];
+    for (let i = 0; i < children.length; i++) {
+      const found = walk(children[i], node, i);
+      if (found) return found;
+    }
+    ancestors.pop();
+    return null;
+  };
+  return walk(root, null, 0);
+}
+function pruneNodeForPrompt(node, maxDepth = 2, depth = 0) {
+  const children = node.children ?? [];
+  if (children.length === 0) return node;
+  if (depth >= maxDepth) {
+    return {
+      ...node,
+      children: [
+        {
+          id: `${node.id}__collapsed`,
+          type: "__collapsed__",
+          props: {
+            note: `${countDescendants(node)} nested nodes omitted \u2014 call read_node on #${node.id} with includeChildren to expand`
+          }
+        }
+      ]
+    };
+  }
+  return {
+    ...node,
+    children: children.map((child) => pruneNodeForPrompt(child, maxDepth, depth + 1))
+  };
+}
+function formatSelectionContext(context) {
+  const path = context.ancestors.map((a) => `#${a.id}(${a.type})`).join(" > ");
+  const siblings = context.siblings.map((sibling, i) => {
+    const marker = sibling.id === context.node.id ? " \u2190 selected" : "";
+    const label = getNodeLabel(sibling);
+    return `  [${i}] #${sibling.id} (${sibling.type})${label ? ` "${label}"` : ""}${marker}`;
+  }).join("\n");
+  return [
+    "### Currently Selected Component",
+    `Node: #${context.node.id} (${context.node.type})`,
+    `Parent: ${context.parentId ? `#${context.parentId}` : "(root)"} \u2014 position ${context.index}`,
+    path ? `Path: ${path}` : "",
+    "Full node JSON:",
+    "```json",
+    JSON.stringify(pruneNodeForPrompt(context.node), null, 2),
+    "```",
+    "Siblings:",
+    siblings,
+    'When the user says "this", "ini", "komponen ini" or similar, they mean the selected node above.'
+  ].filter(Boolean).join("\n");
+}
+
 // src/core/prompt-compiler.ts
 function compileComponentCatalog(registry) {
   if (!registry) return [];
@@ -213,6 +362,86 @@ function buildJsonSchemaForMode(mode) {
   }
   return nodeSchema;
 }
+function formatComponentCatalog(catalog) {
+  return catalog.map((c) => {
+    let desc = `- **${c.type}** (${c.category}): ${c.label}`;
+    if (c.description) desc += ` - ${c.description}`;
+    desc += ` | acceptsChildren: ${c.acceptsChildren}`;
+    if (c.allowedChildren && c.allowedChildren.length > 0) {
+      desc += ` | allowedChildren: [${c.allowedChildren.join(", ")}]`;
+    }
+    if (c.props && c.props.length > 0) {
+      const propList = c.props.map((p) => {
+        let pStr = `${p.name} (${p.type})`;
+        if (p.options) pStr += `[${p.options.map(String).join("|")}]`;
+        return pStr;
+      }).join(", ");
+      desc += ` | props: { ${propList} }`;
+    }
+    return desc;
+  }).join("\n");
+}
+var AGENT_INSTRUCTIONS = `You are the KUBUILD page-editing agent. You edit an existing page by calling tools \u2014 you never output page JSON directly.
+
+### HOW YOU WORK
+1. The page outline below is your map. It shows every node's id, type and label, but not its props or styles.
+2. To see a node's actual props/styles, call read_node. To locate nodes by type or text, call find_nodes. Never guess a node id \u2014 every id you pass must come from the outline or from a tool result.
+3. Make the change with the smallest tool that expresses it, then stop and summarize what you changed, in the user's language.
+
+### SURGICAL EDIT RULE (the most important rule)
+Change only what the user asked for.
+- Changing text/label/href/src -> update_node_props on that one node.
+- Changing color/spacing/size/alignment -> update_node_styles on that one node.
+- Adding something new -> insert_component (a single element) or insert_section (a whole new section).
+- replace_node is a last resort, only when a node's children genuinely must be restructured. Never use it to change a prop or a style.
+- Never rebuild a section, and never touch nodes the user did not ask about. If an instruction is ambiguous about scope, edit the narrowest node that satisfies it and say so in your summary.
+
+### DESTRUCTIVE ACTIONS
+delete_node and replace_node destroy user content. Only call them when the user asked unambiguously (e.g. "hapus section harga"). If the request is broad or vague ("bersihkan halaman ini", "hapus yang tidak perlu"), do not call them \u2014 ask the user which nodes to remove instead.
+
+### STYLING RULES
+- styles values are CSS primitives (strings/numbers). Use breakpoint ('base' | 'desktop' | 'tablet' | 'mobile') for normal styling and state (':hover', ':focus', ':active') for interactive states. Never nest a pseudo-class inside a breakpoint layer.
+- Only use component types that exist in the catalog below, and respect acceptsChildren/allowedChildren.
+- Never emit scripts, javascript: URIs, or event-handler attributes.
+
+### WHEN A TOOL FAILS
+A tool result with "ok": false means your input was wrong (unknown node id, invalid type, disallowed nesting). Read the error, fix your arguments, and retry \u2014 do not repeat the identical call, and do not give up silently.`;
+function buildAgentSystemPrompt(options) {
+  const { catalog, document, selectedNodeId, prefix, stylePreference, additionalContext } = options;
+  const selection = selectedNodeId ? buildSelectionContext(document.document, selectedNodeId) : null;
+  const sections = [];
+  if (prefix) sections.push(prefix);
+  sections.push(AGENT_INSTRUCTIONS);
+  sections.push(
+    `### REGISTERED COMPONENT CATALOG
+${formatComponentCatalog(catalog) || "(Standard core components: page, section, container, columns, heading, paragraph, text, button, input, textarea, select, image, video, icon, badge, code-block)"}`
+  );
+  sections.push(`### CURRENT PAGE
+${summarizeDocument(document, { focusNodeId: selectedNodeId })}`);
+  if (selection) {
+    sections.push(formatSelectionContext(selection));
+  } else if (selectedNodeId) {
+    sections.push(
+      `### Selection
+The editor reported node "${selectedNodeId}" as selected, but it is not present in the current page. Ignore it and ask the user what to target.`
+    );
+  } else {
+    sections.push(
+      '### Selection\nNo component is selected on the canvas. If the user says "this", ask which component they mean instead of guessing.'
+    );
+  }
+  if (stylePreference) {
+    sections.push(
+      `### Style Preference
+"${stylePreference}" \u2014 follow this aesthetic for anything you create or restyle.`
+    );
+  }
+  if (additionalContext) {
+    sections.push(`### Additional Context
+${additionalContext}`);
+  }
+  return sections.join("\n\n");
+}
 
 // src/core/normalizer.ts
 var import_schema = require("@kubuild/schema");
@@ -453,16 +682,46 @@ function normalizeAndValidateRefactoredNode(rawJson, originalNode, securityLimit
   }
   return parsed.data;
 }
+
+// src/core/messages.ts
+function getMessageText(message) {
+  if (typeof message.content === "string") return message.content;
+  return message.content.filter((block) => block.type === "text").map((block) => block.text).join("\n");
+}
+function getToolUseBlocks(message) {
+  if (typeof message.content === "string") return [];
+  return message.content.filter((b) => b.type === "tool_use");
+}
+function getToolResultBlocks(message) {
+  if (typeof message.content === "string") return [];
+  return message.content.filter((b) => b.type === "tool_result");
+}
+function hasStructuredContent(message) {
+  return typeof message.content !== "string" && message.content.some((b) => b.type !== "text");
+}
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  buildAgentSystemPrompt,
   buildJsonSchemaForMode,
+  buildSelectionContext,
   buildSystemPrompt,
   compileComponentCatalog,
+  countDescendants,
   extractJsonFromResponse,
+  formatComponentCatalog,
+  formatSelectionContext,
+  getMessageText,
+  getNodeLabel,
+  getToolResultBlocks,
+  getToolUseBlocks,
+  hasStructuredContent,
   normalizeAndValidatePageDocument,
   normalizeAndValidateRefactoredNode,
   normalizeAndValidateSectionNode,
   normalizeNodeTree,
-  normalizeStyles
+  normalizeStyles,
+  pruneNodeForPrompt,
+  summarizeDocument,
+  summarizeNodeTree
 });
 //# sourceMappingURL=index.cjs.map