@builder.io/ai-utils 0.95.0 → 0.97.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@builder.io/ai-utils",
-  "version": "0.95.0",
+  "version": "0.97.0",
   "description": "Builder.io AI utils",
   "files": [
     "src"

package/src/codegen/investigation-context.d.ts

@@ -12,11 +12,34 @@ export interface InvestigationCompletionJSON {
     messages?: MessageParam[];
 }
 export declare function serializeContentToBlocks(message: string | ContentMessage | undefined): string;
+export declare function serializeMessage(message: MessageParam): string;
 export declare function messagesToText(messages: MessageParam[]): string;
 export declare function truncateToolResultsInTranscript(text: string): string;
+export interface InvestigationTimelineEntry {
+    id: string;
+    timestampMs?: number;
+    role?: string;
+    message?: string;
+    hasError?: boolean;
+}
 export interface BuildInvestigationSystemPromptInput {
     event: InvestigationEvent;
     completionJson: InvestigationCompletionJSON | undefined;
+    timeline?: InvestigationTimelineEntry[];
+}
+export declare function buildInvestigationSystemPrompt({ event, completionJson, timeline, }: BuildInvestigationSystemPromptInput): string;
+export interface InvestigationSessionTurn {
+    id: string;
+    timestampMs?: number;
+    hasError?: boolean;
+    completionJson?: InvestigationCompletionJSON;
+}
+export interface BuildSessionInvestigationSystemPromptInput {
+    focusEvent: InvestigationEvent;
+    focusCompletionJson: InvestigationCompletionJSON | undefined;
+    turns: InvestigationSessionTurn[];
+    timeline?: InvestigationTimelineEntry[];
 }
-export declare function buildInvestigationSystemPrompt({ event, completionJson, }: BuildInvestigationSystemPromptInput): string;
+export declare function buildSessionInvestigationSystemPrompt({ focusEvent, focusCompletionJson, turns, timeline, }: BuildSessionInvestigationSystemPromptInput): string;
 export declare const SUGGESTED_QUESTIONS: readonly string[];
+export declare const SESSION_SUGGESTED_QUESTIONS: readonly string[];

package/src/codegen/investigation-context.js

@@ -48,13 +48,12 @@ export function serializeContentToBlocks(message) {
     })
         .join("\n\n");
 }
+export function serializeMessage(message) {
+    const content = serializeContentToBlocks(message.content);
+    return `<___llm_message___ role=${JSON.stringify(message.role)}>\n${content}\n</___llm_message___>`;
+}
 export function messagesToText(messages) {
-    return messages
-        .map((message) => {
-        const content = serializeContentToBlocks(message.content);
-        return `<___llm_message___ role=${JSON.stringify(message.role)}>\n${content}\n</___llm_message___>`;
-    })
-        .join("\n\n");
+    return messages.map(serializeMessage).join("\n\n");
 }
 export function truncateToolResultsInTranscript(text) {
     // Truncate any oversized tool_result block content so a single huge tool
@@ -110,33 +109,139 @@ export function truncateToolResultsInTranscript(text) {
     }
     return result;
 }
-export function buildInvestigationSystemPrompt({ event, completionJson, }) {
+// Oldest-first; entries without a timestamp keep their relative order at the end.
+function byTimestampAsc(entries) {
+    return [...entries].sort((a, b) => { var _a, _b; return ((_a = a.timestampMs) !== null && _a !== void 0 ? _a : Infinity) - ((_b = b.timestampMs) !== null && _b !== void 0 ? _b : Infinity); });
+}
+function buildTimelineBlock(timeline, currentEventId) {
+    if (!(timeline === null || timeline === void 0 ? void 0 : timeline.length))
+        return "";
+    const MESSAGE_PREVIEW_CHARS = 200;
+    const lines = byTimestampAsc(timeline).map((entry, index) => {
+        var _a;
+        const when = entry.timestampMs != null && Number.isFinite(entry.timestampMs)
+            ? new Date(entry.timestampMs).toISOString()
+            : "(no timestamp)";
+        const role = entry.role ? `[${entry.role}]` : "[?]";
+        const marker = entry.id === currentEventId ? " ▶" : "";
+        const error = entry.hasError ? " (error)" : "";
+        const flat = ((_a = entry.message) !== null && _a !== void 0 ? _a : "").replace(/\s+/g, " ").trim();
+        const preview = flat.length > MESSAGE_PREVIEW_CHARS
+            ? `${flat.slice(0, MESSAGE_PREVIEW_CHARS)}…`
+            : flat;
+        return `${index + 1}. ${when} ${role}${marker}${error} ${JSON.stringify(preview)}`;
+    });
+    return `## Session timeline (turn timestamps)\n\nEach line is one turn in this session — the same list shown on the left of the UI — oldest first, timestamps in UTC (ISO 8601). The event under investigation is marked ▶.\n\n${lines.join("\n")}`;
+}
+export function buildInvestigationSystemPrompt({ event, completionJson, timeline, }) {
     var _a, _b;
     const role = `You are a senior AI engineer helping debug a single codegen agent call.
 You have exactly the same context the engineer sees on the LLM tab for this
 event: the system prompt and the message transcript (with tool_use,
-tool_result, and thinking blocks). Be concise, specific, and critical.
+tool_result, and thinking blocks), plus a session timeline giving the
+timestamp of every turn. Be concise, specific, and critical.
 
 Ground every answer in the concrete data below: cite specific tool calls
-(by name + id), specific messages (by role + index), or system prompt
-sections. When you suggest improvements, propose concrete changes (e.g.
-"remove tool X", "add this instruction", "split this prompt section"). If
-the data does not support a claim, say so.`;
+(by name + id), specific messages (by role + index), timeline entries (by
+timestamp), or system prompt sections. When you suggest improvements, propose
+concrete changes (e.g. "remove tool X", "add this instruction", "split this
+prompt section"). If the data does not support a claim, say so.`;
     const header = `## Event\n\nid: ${event.id}\nmodel: ${(_a = completionJson === null || completionJson === void 0 ? void 0 : completionJson.model) !== null && _a !== void 0 ? _a : "(unknown)"}${event.hasError ? "\nstatus: ERROR" : ""}`;
+    const timelineBlock = buildTimelineBlock(timeline, event.id);
     // Use XML-style delimiters instead of triple-backtick fences: codegen
     // transcripts often contain markdown code fences themselves which would
     // otherwise close the outer block and corrupt the prompt.
     const systemPromptBlock = `## System prompt (verbatim)\n\n<___event_system_prompt___>\n${truncate((_b = completionJson === null || completionJson === void 0 ? void 0 : completionJson.systemPrompt) !== null && _b !== void 0 ? _b : "(no system prompt)", MAX_SYSTEM_PROMPT_CHARS)}\n</___event_system_prompt___>`;
+    // Debugging a long session needs the whole transcript verbatim, so this is
+    // sent in full — no per-tool-result or overall transcript truncation.
     const transcriptText = (completionJson === null || completionJson === void 0 ? void 0 : completionJson.messages)
-        ? truncateToolResultsInTranscript(messagesToText(completionJson.messages))
+        ? messagesToText(completionJson.messages)
         : "(no messages)";
-    const transcriptBlock = `## Message transcript (tool_use / tool_result / thinking)\n\n<___event_transcript___>\n${truncate(transcriptText, MAX_TRANSCRIPT_CHARS)}\n</___event_transcript___>`;
+    const transcriptBlock = `## Message transcript (tool_use / tool_result / thinking)\n\n<___event_transcript___>\n${transcriptText}\n</___event_transcript___>`;
     const guidance = `## How to answer
 
 - Start with a 1-2 sentence direct answer.
 - Back it up with specific evidence (tool names, message indices, prompt phrases).
 - End with concrete improvement suggestions when applicable.`;
-    return [role, header, systemPromptBlock, transcriptBlock, guidance].join("\n\n");
+    return [
+        role,
+        header,
+        timelineBlock,
+        systemPromptBlock,
+        transcriptBlock,
+        guidance,
+    ]
+        .filter(Boolean)
+        .join("\n\n");
+}
+export function buildSessionInvestigationSystemPrompt({ focusEvent, focusCompletionJson, turns, timeline, }) {
+    var _a;
+    const role = `You are a senior AI engineer helping debug an entire codegen session:
+a sequence of agent calls (turns). You have the same context the engineer sees
+in the UI — a timeline of every turn with timestamps, and the message
+transcript of the session (with tool_use, tool_result, and thinking blocks).
+Because each turn re-sends the whole prior conversation, each turn below shows
+only the messages it added; earlier messages it carried over are counted, not
+repeated. Be concise, specific, and critical.
+
+Ground every answer in the concrete data below: cite specific turns (by event
+id or timestamp), tool calls (by name + id), or messages. Reason about how the
+session evolved across turns — what changed, what regressed, and where. If the
+data does not support a claim, say so.`;
+    const header = `## Session\n\nturns: ${turns.length}\nfocus turn (currently selected in UI): ${focusEvent.id}`;
+    const timelineBlock = buildTimelineBlock(timeline, focusEvent.id);
+    // Per-turn system prompts are largely shared within a session; include the
+    // focus turn's verbatim and point elsewhere for exact per-turn prompts.
+    const systemPromptBlock = `## System prompt (verbatim — from focus turn ${focusEvent.id}; other turns may differ)\n\n<___event_system_prompt___>\n${truncate((_a = focusCompletionJson === null || focusCompletionJson === void 0 ? void 0 : focusCompletionJson.systemPrompt) !== null && _a !== void 0 ? _a : "(no system prompt)", MAX_SYSTEM_PROMPT_CHARS)}\n</___event_system_prompt___>`;
+    // Each turn re-sends the whole prior conversation, so a turn's messages are
+    // a growing prefix of the next. Emit only the messages a turn adds beyond the
+    // previous one, so every message appears once instead of O(turns) times.
+    let prevSerialized = [];
+    const turnBlocks = byTimestampAsc(turns)
+        .map((turn, index) => {
+        var _a, _b;
+        var _c;
+        const when = turn.timestampMs != null && Number.isFinite(turn.timestampMs)
+            ? new Date(turn.timestampMs).toISOString()
+            : "(no timestamp)";
+        const model = (_c = (_a = turn.completionJson) === null || _a === void 0 ? void 0 : _a.model) !== null && _c !== void 0 ? _c : "(unknown)";
+        const status = turn.hasError ? ' status="ERROR"' : "";
+        const focus = turn.id === focusEvent.id ? ' focus="true"' : "";
+        if (!((_b = turn.completionJson) === null || _b === void 0 ? void 0 : _b.messages)) {
+            return `<___session_turn___ index="${index + 1}" event_id="${turn.id}" ts="${when}" model="${model}"${status}${focus}>\n(transcript not loaded)\n</___session_turn___>`;
+        }
+        const serialized = turn.completionJson.messages.map(serializeMessage);
+        let common = 0;
+        while (common < serialized.length &&
+            common < prevSerialized.length &&
+            serialized[common] === prevSerialized[common]) {
+            common++;
+        }
+        prevSerialized = serialized;
+        const added = serialized.slice(common);
+        const body = added.length
+            ? added.join("\n\n")
+            : "(no new messages beyond the previous turn)";
+        const carried = common > 0 ? ` carried_over_messages="${common}"` : "";
+        return `<___session_turn___ index="${index + 1}" event_id="${turn.id}" ts="${when}" model="${model}"${status}${focus}${carried}>\n${body}\n</___session_turn___>`;
+    })
+        .join("\n\n");
+    const transcriptBlock = `## Message transcripts (one block per turn, oldest first)\n\nEach turn re-sends the full prior conversation, so only the messages a turn adds are shown; \`carried_over_messages\` counts identical earlier messages omitted from that turn.\n\n${turnBlocks}`;
+    const guidance = `## How to answer
+
+- Start with a 1-2 sentence direct answer.
+- Back it up with specific evidence (turn ids/timestamps, tool names, message indices).
+- End with concrete improvement suggestions when applicable.`;
+    return [
+        role,
+        header,
+        timelineBlock,
+        systemPromptBlock,
+        transcriptBlock,
+        guidance,
+    ]
+        .filter(Boolean)
+        .join("\n\n");
 }
 export const SUGGESTED_QUESTIONS = [
     "Why did this call fail or behave suboptimally?",
@@ -145,3 +250,10 @@ export const SUGGESTED_QUESTIONS = [
     "How could the system prompt be improved for this case?",
     "Walk me through the agent's reasoning step by step.",
 ];
+export const SESSION_SUGGESTED_QUESTIONS = [
+    "Summarize what happened across this whole session.",
+    "Where did the session go wrong, and at which turn?",
+    "What did the user keep asking for that wasn't delivered?",
+    "Which turns were redundant or repeated work?",
+    "How did the agent's approach change over time?",
+];

package/src/codegen.d.ts

@@ -384,6 +384,18 @@ export declare const GetScreenshotToolInputSchema: z.ZodObject<{
     height: z.ZodOptional<z.ZodNumber>;
 }, z.core.$strip>;
 export type GetScreenshotToolInput = z.infer<typeof GetScreenshotToolInputSchema>;
+export declare const JudgeCommandSafetyInputSchema: z.ZodObject<{
+    command: z.ZodString;
+}, z.core.$strip>;
+export type JudgeCommandSafetyInput = z.infer<typeof JudgeCommandSafetyInputSchema>;
+export declare const JudgeCommandSafetyResultSchema: z.ZodObject<{
+    verdict: z.ZodEnum<{
+        destructive: "destructive";
+        safe: "safe";
+    }>;
+    reason: z.ZodOptional<z.ZodString>;
+}, z.core.$strip>;
+export type JudgeCommandSafetyResult = z.infer<typeof JudgeCommandSafetyResultSchema>;
 export declare const NavigatePreviewToolInputSchema: z.ZodObject<{
     href: z.ZodString;
 }, z.core.$strip>;
@@ -2107,7 +2119,7 @@ export declare const DEFAULT_QUEUE_BEHAVIOR: QueueBehavior;
 /** True for any schedule that aborts the in-flight run on enqueue. */
 export declare function isInterruptSchedule(schedule: QueueSchedule): boolean;
 export declare function normalizeQueueMode(mode: QueueMode | undefined): QueueBehavior;
-export declare const BASE_CODEGEN_POSITIONS: readonly ["fusion", "editor-ai", "repo-indexing", "cli", "create-app-firebase", "create-app-lovable", "builder-code-panel", "setup-project", "code-review-orchestrator", "project-configuration", "org-agent", "org-worker", "browser-testing", "projects-scheduler-memory-extraction", "builder-code", "unknown", "dsi-mcp", "design-system-indexer", "builder-publish-integration"];
+export declare const BASE_CODEGEN_POSITIONS: readonly ["fusion", "editor-ai", "repo-indexing", "cli", "create-app-firebase", "create-app-lovable", "builder-code-panel", "setup-project", "code-review-orchestrator", "project-configuration", "org-agent", "org-worker", "browser-testing", "projects-scheduler-memory-extraction", "builder-code", "unknown", "dsi-mcp", "design-system-indexer", "builder-publish-integration", "publish-agent"];
 export type BaseCodeGenPosition = (typeof BASE_CODEGEN_POSITIONS)[number];
 export declare const BaseCodeGenPositionSchema: z.ZodEnum<{
     "browser-testing": "browser-testing";
@@ -2126,6 +2138,7 @@ export declare const BaseCodeGenPositionSchema: z.ZodEnum<{
     "org-worker": "org-worker";
     "project-configuration": "project-configuration";
     "projects-scheduler-memory-extraction": "projects-scheduler-memory-extraction";
+    "publish-agent": "publish-agent";
     "repo-indexing": "repo-indexing";
     "setup-project": "setup-project";
     unknown: "unknown";
@@ -2147,11 +2160,97 @@ export declare const CodeGenPositionSchema: z.ZodUnion<readonly [z.ZodEnum<{
     "org-worker": "org-worker";
     "project-configuration": "project-configuration";
     "projects-scheduler-memory-extraction": "projects-scheduler-memory-extraction";
+    "publish-agent": "publish-agent";
     "repo-indexing": "repo-indexing";
     "setup-project": "setup-project";
     unknown: "unknown";
-}>, z.ZodTemplateLiteral<"browser-testing-agent" | "builder-code-agent" | "builder-code-panel-agent" | "builder-publish-integration-agent" | "cli-agent" | "code-review-orchestrator-agent" | "create-app-firebase-agent" | "create-app-lovable-agent" | "design-system-indexer-agent" | "dsi-mcp-agent" | "editor-ai-agent" | "fusion-agent" | "org-agent-agent" | "org-worker-agent" | "project-configuration-agent" | "projects-scheduler-memory-extraction-agent" | "repo-indexing-agent" | "setup-project-agent" | "unknown-agent">]>;
+}>, z.ZodTemplateLiteral<"browser-testing-agent" | "builder-code-agent" | "builder-code-panel-agent" | "builder-publish-integration-agent" | "cli-agent" | "code-review-orchestrator-agent" | "create-app-firebase-agent" | "create-app-lovable-agent" | "design-system-indexer-agent" | "dsi-mcp-agent" | "editor-ai-agent" | "fusion-agent" | "org-agent-agent" | "org-worker-agent" | "project-configuration-agent" | "projects-scheduler-memory-extraction-agent" | "publish-agent-agent" | "repo-indexing-agent" | "setup-project-agent" | "unknown-agent">]>;
 export type CodeGenPosition = z.infer<typeof CodeGenPositionSchema>;
+export declare const PublishAgentMessageRequestSchema: z.ZodObject<{
+    message: z.ZodOptional<z.ZodString>;
+    sessionId: z.ZodOptional<z.ZodString>;
+    toolResults: z.ZodOptional<z.ZodArray<z.ZodObject<{
+        type: z.ZodLiteral<"tool_result">;
+        tool_use_id: z.ZodString;
+        tool_name: z.ZodOptional<z.ZodString>;
+        tool_input: z.ZodOptional<z.ZodString>;
+        title: z.ZodOptional<z.ZodString>;
+        content: z.ZodUnion<readonly [z.ZodString, z.ZodArray<z.ZodDiscriminatedUnion<[z.ZodObject<{
+            type: z.ZodLiteral<"text">;
+            text: z.ZodString;
+            cache: z.ZodOptional<z.ZodBoolean>;
+            citations: z.ZodOptional<z.ZodNullable<z.ZodArray<z.ZodDiscriminatedUnion<[z.ZodObject<{
+                cited_text: z.ZodString;
+                document_index: z.ZodNumber;
+                document_title: z.ZodNullable<z.ZodString>;
+                end_char_index: z.ZodNumber;
+                start_char_index: z.ZodNumber;
+                type: z.ZodLiteral<"char_location">;
+            }, z.core.$strip>, z.ZodObject<{
+                cited_text: z.ZodString;
+                document_index: z.ZodNumber;
+                document_title: z.ZodNullable<z.ZodString>;
+                end_page_number: z.ZodNumber;
+                start_page_number: z.ZodNumber;
+                type: z.ZodLiteral<"page_location">;
+            }, z.core.$strip>, z.ZodObject<{
+                cited_text: z.ZodString;
+                document_index: z.ZodNumber;
+                document_title: z.ZodNullable<z.ZodString>;
+                end_block_index: z.ZodNumber;
+                start_block_index: z.ZodNumber;
+                type: z.ZodLiteral<"content_block_location">;
+            }, z.core.$strip>, z.ZodObject<{
+                cited_text: z.ZodString;
+                encrypted_index: z.ZodString;
+                title: z.ZodNullable<z.ZodString>;
+                type: z.ZodLiteral<"web_search_result_location">;
+                url: z.ZodString;
+            }, z.core.$strip>], "type">>>>;
+            ephemeral: z.ZodOptional<z.ZodBoolean>;
+            thoughtSignature: z.ZodOptional<z.ZodString>;
+            tag: z.ZodOptional<z.ZodString>;
+        }, z.core.$strip>, z.ZodObject<{
+            type: z.ZodLiteral<"image">;
+            source: z.ZodDiscriminatedUnion<[z.ZodObject<{
+                type: z.ZodLiteral<"base64">;
+                media_type: z.ZodEnum<{
+                    "image/gif": "image/gif";
+                    "image/jpeg": "image/jpeg";
+                    "image/png": "image/png";
+                    "image/webp": "image/webp";
+                }>;
+                data: z.ZodString;
+                original_url: z.ZodOptional<z.ZodString>;
+            }, z.core.$strip>, z.ZodObject<{
+                type: z.ZodLiteral<"url">;
+                url: z.ZodString;
+            }, z.core.$strip>], "type">;
+            cache: z.ZodOptional<z.ZodBoolean>;
+            ephemeral: z.ZodOptional<z.ZodBoolean>;
+        }, z.core.$strip>, z.ZodObject<{
+            type: z.ZodLiteral<"resource">;
+            resource: z.ZodObject<{
+                uri: z.ZodString;
+                mimeType: z.ZodOptional<z.ZodString>;
+                text: z.ZodOptional<z.ZodString>;
+                blob: z.ZodOptional<z.ZodString>;
+            }, z.core.$strip>;
+        }, z.core.$strip>], "type">>]>;
+        is_error: z.ZodOptional<z.ZodBoolean>;
+        cache: z.ZodOptional<z.ZodBoolean>;
+        ephemeral: z.ZodOptional<z.ZodBoolean>;
+        structured_result: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodAny>>;
+    }, z.core.$strip>>>;
+    confirmation: z.ZodOptional<z.ZodObject<{
+        actionId: z.ZodString;
+        decision: z.ZodEnum<{
+            approve: "approve";
+            reject: "reject";
+        }>;
+    }, z.core.$strip>>;
+}, z.core.$strip>;
+export type PublishAgentMessageRequest = z.infer<typeof PublishAgentMessageRequestSchema>;
 export declare const RepoIndexingConfigSchema: z.ZodObject<{
     designSystems: z.ZodArray<z.ZodString>;
 }, z.core.$strip>;
@@ -2422,10 +2521,11 @@ export declare const CodeGenInputOptionsSchema: z.ZodObject<{
         "org-worker": "org-worker";
         "project-configuration": "project-configuration";
         "projects-scheduler-memory-extraction": "projects-scheduler-memory-extraction";
+        "publish-agent": "publish-agent";
         "repo-indexing": "repo-indexing";
         "setup-project": "setup-project";
         unknown: "unknown";
-    }>, z.ZodTemplateLiteral<"browser-testing-agent" | "builder-code-agent" | "builder-code-panel-agent" | "builder-publish-integration-agent" | "cli-agent" | "code-review-orchestrator-agent" | "create-app-firebase-agent" | "create-app-lovable-agent" | "design-system-indexer-agent" | "dsi-mcp-agent" | "editor-ai-agent" | "fusion-agent" | "org-agent-agent" | "org-worker-agent" | "project-configuration-agent" | "projects-scheduler-memory-extraction-agent" | "repo-indexing-agent" | "setup-project-agent" | "unknown-agent">]>;
+    }>, z.ZodTemplateLiteral<"browser-testing-agent" | "builder-code-agent" | "builder-code-panel-agent" | "builder-publish-integration-agent" | "cli-agent" | "code-review-orchestrator-agent" | "create-app-firebase-agent" | "create-app-lovable-agent" | "design-system-indexer-agent" | "dsi-mcp-agent" | "editor-ai-agent" | "fusion-agent" | "org-agent-agent" | "org-worker-agent" | "project-configuration-agent" | "projects-scheduler-memory-extraction-agent" | "publish-agent-agent" | "repo-indexing-agent" | "setup-project-agent" | "unknown-agent">]>;
     eventName: z.ZodOptional<z.ZodString>;
     sessionId: z.ZodString;
     codeGenMode: z.ZodOptional<z.ZodEnum<{
@@ -2606,6 +2706,7 @@ export declare const CodeGenInputOptionsSchema: z.ZodObject<{
         v1: "v1";
         v2: "v2";
         v3: "v3";
+        v4: "v4";
     }>>;
     reasoning: z.ZodOptional<z.ZodEnum<{
         auto: "auto";
@@ -3472,10 +3573,11 @@ export declare const CustomAgentDefinitionSchema: z.ZodObject<{
         "org-worker": "org-worker";
         "project-configuration": "project-configuration";
         "projects-scheduler-memory-extraction": "projects-scheduler-memory-extraction";
+        "publish-agent": "publish-agent";
         "repo-indexing": "repo-indexing";
         "setup-project": "setup-project";
         unknown: "unknown";
-    }>, z.ZodTemplateLiteral<"browser-testing-agent" | "builder-code-agent" | "builder-code-panel-agent" | "builder-publish-integration-agent" | "cli-agent" | "code-review-orchestrator-agent" | "create-app-firebase-agent" | "create-app-lovable-agent" | "design-system-indexer-agent" | "dsi-mcp-agent" | "editor-ai-agent" | "fusion-agent" | "org-agent-agent" | "org-worker-agent" | "project-configuration-agent" | "projects-scheduler-memory-extraction-agent" | "repo-indexing-agent" | "setup-project-agent" | "unknown-agent">]>>;
+    }>, z.ZodTemplateLiteral<"browser-testing-agent" | "builder-code-agent" | "builder-code-panel-agent" | "builder-publish-integration-agent" | "cli-agent" | "code-review-orchestrator-agent" | "create-app-firebase-agent" | "create-app-lovable-agent" | "design-system-indexer-agent" | "dsi-mcp-agent" | "editor-ai-agent" | "fusion-agent" | "org-agent-agent" | "org-worker-agent" | "project-configuration-agent" | "projects-scheduler-memory-extraction-agent" | "publish-agent-agent" | "repo-indexing-agent" | "setup-project-agent" | "unknown-agent">]>>;
     needDevServer: z.ZodOptional<z.ZodBoolean>;
     needValidation: z.ZodOptional<z.ZodBoolean>;
     includeMemories: z.ZodOptional<z.ZodBoolean>;
@@ -4348,6 +4450,12 @@ export interface FusionConfig {
     /** PR number for fork PRs - used for fetching refs/pulls/x/head */
     prNumber?: number;
     refreshPreview?: boolean;
+    /**
+     * Keep the session's commits when HEAD has drifted off `aiBranch`: move the
+     * AI branch to HEAD rather than checking out onto it, and repair the same
+     * drift at backup time instead of refusing to back up.
+     */
+    preserveWorkOnBranchDrift?: boolean;
     workingDirectory?: string;
     bashWorkingDirectory?: string;
     workspace?: WorkspaceConfiguration;
@@ -4468,6 +4576,14 @@ export interface FusionConfig {
      * This is a dangerous setting that removes safety guardrails - equivalent to full shell access.
      */
     skipCommandSecurity?: boolean;
+    /**
+     * LaunchDarkly `command-safety-judge-enabled` flag, resolved server-side at
+     * session start and threaded down since the sandbox has no LaunchDarkly
+     * access of its own. When true, `executeShellCommand` routes Tier 2
+     * judgment-requiring commands to the `JudgeCommandSafety` endpoint instead
+     * of hard-blocking them.
+     */
+    commandSafetyJudgeEnabled?: boolean;
     /** @deprecated use devCommand */
     command?: string;
 }
@@ -4824,6 +4940,7 @@ export interface SearchFileTreeResult {
 }
 export interface SetupCommandResult {
     code: number | null;
+    signal?: string;
     output: string;
     capturedEnv?: Record<string, string>;
 }
@@ -4867,6 +4984,8 @@ export interface BuildCommandResultFailure {
     code?: number;
     signal?: string;
     output: string;
+    /** Which step produced `output`. Absent means the build command itself. */
+    phase?: "setup" | "build";
 }
 export interface ConfigureDevOrchestratorUpdates {
     devCommand: boolean;
@@ -5014,6 +5133,8 @@ export interface SessionData {
     repoHash: string | undefined;
     repoBranch: string | undefined;
     description?: string;
+    /** Hidden from the session list, but still readable by id. Creator-set. */
+    archived?: boolean;
     hasPlanToApply?: boolean;
     cost: number;
     /** Cost from tools (e.g. WebSearch LLM calls) and sub-agent roll-ups. Total session cost = cost + extraCost. */

package/src/codegen.js

@@ -580,6 +580,26 @@ export const GetScreenshotToolInputSchema = z
     height: z.number().optional(),
 })
     .meta({ title: "GetScreenshotToolInput" });
+export const JudgeCommandSafetyInputSchema = z
+    .object({
+    // Bounded so a caller can't force multi-megabyte payloads through this
+    // directly-callable, AI-credit-exempt endpoint into every completionLLM
+    // call — real shell commands are nowhere near this size.
+    command: z.string().max(20000).meta({
+        description: "The shell command to classify.",
+    }),
+})
+    .meta({ title: "JudgeCommandSafetyInput" });
+export const JudgeCommandSafetyResultSchema = z
+    .object({
+    verdict: z.enum(["safe", "destructive"]).meta({
+        description: "Whether the command is safe to run in this ephemeral, fully-writable sandbox, or destructive and should be denied.",
+    }),
+    reason: z.string().optional().meta({
+        description: "Required when verdict is destructive: a short, non-scary explanation shown to the user. Omit when safe.",
+    }),
+})
+    .meta({ title: "JudgeCommandSafetyResult" });
 export const NavigatePreviewToolInputSchema = z
     .object({
     href: z.string().meta({
@@ -1746,6 +1766,7 @@ export const BASE_CODEGEN_POSITIONS = [
     "dsi-mcp",
     "design-system-indexer",
     "builder-publish-integration",
+    "publish-agent",
 ];
 export const BaseCodeGenPositionSchema = z
     .enum(BASE_CODEGEN_POSITIONS)
@@ -1756,6 +1777,27 @@ export const CodeGenPositionSchema = z
     z.templateLiteral([BaseCodeGenPositionSchema, "-agent"]),
 ])
     .meta({ title: "CodeGenPosition" });
+export const PublishAgentMessageRequestSchema = z
+    .object({
+    message: z.string().max(100000).optional(),
+    sessionId: z.string().min(1).max(256).optional(),
+    toolResults: z.array(ContentMessageItemToolResultSchema).optional(),
+    confirmation: z
+        .object({
+        actionId: z.string().min(1).max(256),
+        decision: z.enum(["approve", "reject"]),
+    })
+        .optional(),
+})
+    .refine(({ message, toolResults }) => Boolean(message) || Boolean(toolResults === null || toolResults === void 0 ? void 0 : toolResults.length), {
+    path: ["message"],
+    message: "A message or at least one tool result is required.",
+})
+    .refine(({ message, sessionId }) => Boolean(message) || Boolean(sessionId), {
+    path: ["sessionId"],
+    message: "A session ID is required when continuing with tool results.",
+})
+    .meta({ title: "PublishAgentMessageRequest" });
 export const RepoIndexingConfigSchema = z
     .object({
     designSystems: z.array(z.string()),
@@ -1940,7 +1982,7 @@ export const CodeGenInputOptionsSchema = z
     modelOverride: z.string().optional(),
     errorIfHadCompaction: z.boolean().optional(),
     softContextWindow: z.number().optional(),
-    promptVersion: z.enum(["v1", "v2", "v3"]).optional(),
+    promptVersion: z.enum(["v1", "v2", "v3", "v4"]).optional(),
     reasoning: ReasoningEffortSchema.optional(),
     redactUserMessages: z.boolean().optional(),
     redactLLMMessages: z.boolean().optional(),

package/src/design-systems.js

@@ -159,7 +159,7 @@ export const listSourceFilesQuerySchema = z.object({
     ref: z.string().optional(),
 });
 export const listPublicSourceFilesQuerySchema = z.object({
-    /** Full repo URL (e.g. https://github.com/owner/repo). GitHub only. */
+    /** Full repo URL (e.g. https://github.com/owner/repo). Supported across all git providers. */
     repoUrl: z.string().url(),
     /** Directory to list, repo-relative. Empty/omitted → repo root. */
     path: z.string().optional(),

package/src/editor-ai.d.ts

@@ -29,6 +29,7 @@ export declare const EditorAiEditRequestSchema: z.ZodObject<{
     components: z.ZodOptional<z.ZodArray<z.ZodObject<{
         name: z.ZodString;
     }, z.core.$loose>>>;
+    allowedComponentNames: z.ZodOptional<z.ZodArray<z.ZodString>>;
 }, z.core.$strip>;
 export type EditorAiEditRequest = z.infer<typeof EditorAiEditRequestSchema>;
 export declare const EditorAiEditResponseSchema: z.ZodObject<{
@@ -44,6 +45,7 @@ export declare const EditorAiWriteRequestSchema: z.ZodObject<{
     components: z.ZodOptional<z.ZodArray<z.ZodObject<{
         name: z.ZodString;
     }, z.core.$loose>>>;
+    allowedComponentNames: z.ZodOptional<z.ZodArray<z.ZodString>>;
 }, z.core.$strip>;
 export type EditorAiWriteRequest = z.infer<typeof EditorAiWriteRequestSchema>;
 export declare const EditorAiWriteResponseSchema: z.ZodObject<{
@@ -51,7 +53,7 @@ export declare const EditorAiWriteResponseSchema: z.ZodObject<{
     modifiedBuilderContent: z.ZodRecord<z.ZodString, z.ZodUnknown>;
 }, z.core.$strip>;
 export type EditorAiWriteResponse = z.infer<typeof EditorAiWriteResponseSchema>;
-export type EditorAiErrorCode = "BAD_REQUEST" | "UNAUTHORIZED" | "FORBIDDEN" | "NOT_FOUND" | "NO_MATCH" | "VALIDATION_FAILED" | "INTERNAL";
+export type EditorAiErrorCode = "BAD_REQUEST" | "UNAUTHORIZED" | "FORBIDDEN" | "NOT_FOUND" | "NO_MATCH" | "VALIDATION_FAILED" | "COMPONENT_NOT_ALLOWED" | "INTERNAL";
 export declare const EditorAiErrorBodySchema: z.ZodObject<{
     error: z.ZodObject<{
         code: z.ZodString;

package/src/editor-ai.js

@@ -13,6 +13,9 @@ export const EditorAiComponentSchema = z
 })
     .meta({ title: "EditorAiComponent" });
 const componentsField = z.array(EditorAiComponentSchema).optional();
+// Omitted is unrestricted; empty forbids every component. Only components-only
+// spaces send it.
+const allowedComponentNamesField = z.array(z.string().min(1)).optional();
 export const EditorAiReadRequestSchema = z
     .object({
     contentId: z.string().min(1),
@@ -45,6 +48,7 @@ export const EditorAiEditRequestSchema = z
     // Locale to operate on; defaults to "Default" when omitted.
     activeLocale: z.string().min(1).optional(),
     components: componentsField,
+    allowedComponentNames: allowedComponentNamesField,
 })
     .meta({ title: "EditorAiEditRequest" });
 export const EditorAiEditResponseSchema = z
@@ -61,6 +65,7 @@ export const EditorAiWriteRequestSchema = z
     // Locale to operate on; defaults to "Default" when omitted.
     activeLocale: z.string().min(1).optional(),
     components: componentsField,
+    allowedComponentNames: allowedComponentNamesField,
 })
     .meta({ title: "EditorAiWriteRequest" });
 export const EditorAiWriteResponseSchema = z

package/src/events.d.ts

@@ -4,6 +4,7 @@ import type { AssistantSettings } from "./settings.js";
 import type { ExitState, UserSource, CodeGenPosition, ReviewSeverity, ReviewVerdict } from "./codegen.js";
 import type { FileUpload } from "./messages.js";
 import type { BuilderConfigRealtime } from "./builder-config.js";
+import type { DeployFailureReport } from "./projects.js";
 export type BuilderAssistantEventHandler = (ev: BuilderAssistantEvent) => void;
 export type BuilderAssistantEvent = AssistantCompletionResultEvent | AssistantErrorEvent | AssistantStreamErrorEvent | AppCloseEvent | AppMessagesClickEvent | AppMessagesGenerationEvent | AppMessageEditCustomInstructionsEvent | AppPromptAbortEvent | AppPromptFocusEvent | AppPromptSubmitEvent | AppReadyEvent | AppSettingsSetEvent | AppThreadNewEvent | AssistantStatsEvent | AssistantThemeEvent | BuilderEditorAuthEvent | BuilderEditorStateEvent | ContentUpdateEvent | ContentApplySnapshotEvent | ModelUndoEvent | ModelRedoEvent | ResultEvent | ThreadCreatedEvent | ThreadMessageCompletedEvent | ThreadMessageCreatedEvent | ThreadMessageDeltaEvent | ThreadMessageFeedbackEvent | ThreadRunStepCreatedEvent | ThreadRunStepDeltaEvent | AppAcceptChangeEvent | AppAcceptRejectEvent | AssistantTrackEvent | AssistantEditorAuthMessage | AppAttachmentTemplateEvent | AppPasteSmartExportEvent | ThreadMessageRetryEvent | AppFigmaImportEvent | AppWebImportEvent | AppMcpServersEvent | AssistantContentInitialEvent | ThreadMessageSummaryEvent | ThreadMessageSummaryCodegenDeltaEvent | ThreadMessageThinkingDeltaEvent | AssistantHeartbeatEvent | ShowUpgradeDialogEvent | AssistantFusionSuggestionEvent | AppNavigateToFusionEvent | ModelPermissionRequiredEvent | ModelPermissionResponseEvent;
 export interface AssistantCompletionResultEvent {
@@ -901,6 +902,13 @@ export type ProjectSnapshotRefreshV1 = FusionEventVariant<"project.snapshot.refr
     projectId: string;
     requestedBy: string;
     reason: string;
+    /**
+     * When set, the resulting VolumeSnapshot is labeled
+     * `fusion-golden-source=<id>` (a baked golden starter snapshot) in addition
+     * to the usual per-project labels. Only set for the golden-starter-snapshots
+     * producer.
+     */
+    goldenTemplateId?: string;
 }, {
     projectId: string;
 }, 1>;
@@ -982,9 +990,12 @@ export type ProjectSnapshotPodWatchV1 = FusionEventVariant<"project.snapshot.pod
     namespace: string;
     podName: string;
     source: "pr.merged" | "project.snapshot.refresh";
+    provider?: GitPrMergedV1["data"]["provider"];
     gitSha?: string;
     startedAtMs: number;
     timeoutMs: number;
+    /** Propagated to the golden starter snapshot's `fusion-golden-source` label. */
+    goldenTemplateId?: string;
 }, {
     projectId: string;
 }, 1>;
@@ -1283,6 +1294,13 @@ export type ClientDevtoolsBuildUploadedV1 = FusionEventVariant<"client.devtools.
      * falls back to reading the repo at the deployed commit.
      */
     realtimeTransport?: BuilderConfigRealtime["transport"];
+    /**
+     * What the pod's build env resolved each `HOSTED_DEPLOY_ENV_DEFAULTS` key to, so
+     * the runtime env can agree with the artifact that was built — a build-scope opt-in
+     * shouldn't emit a scheduled function the runtime then reports as disabled. Older
+     * pods omit it and the service falls back to the defaults.
+     */
+    resolvedHostedEnvDefaults?: Record<string, string>;
 }, {
     projectId: string;
     deployId: string;
@@ -1295,7 +1313,7 @@ export type ClientDevtoolsBuildFailedV1 = FusionEventVariant<"client.devtools.bu
     deployId: string;
     projectId: string;
     error: string;
-}, {
+} & DeployFailureReport, {
     projectId: string;
     deployId: string;
 }, 1>;

package/src/hosted-env-defaults.test.d.ts

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