@agent-native/core 0.84.18 → 0.84.21

package/corpus/core/src/client/extensions/ExtensionViewer.tsx

@@ -2,6 +2,7 @@ import {
   IconArrowLeft,
   IconArrowBackUp,
   IconChevronRight,
+  IconCode,
   IconDotsVertical,
   IconHistory,
   IconLoader2,
@@ -20,6 +21,7 @@ import { getThemeVars } from "../../extensions/theme.js";
 import { sendToAgentChat } from "../agent-chat.js";
 import { AgentToggleButton } from "../AgentPanel.js";
 import { agentNativePath, appPath } from "../api-path.js";
+import { Dialog, DialogContent, DialogTitle } from "../components/ui/dialog.js";
 import {
   Popover,
   PopoverContent,
@@ -324,6 +326,131 @@ function applyCanonicalLink(path: string): () => void {
   };
 }
 
+function SourceCodeDialog({
+  extension,
+  onSaved,
+}: {
+  extension: Extension;
+  onSaved?: () => void;
+}) {
+  const queryClient = useQueryClient();
+  const [open, setOpen] = useState(false);
+  const [code, setCode] = useState(extension.content ?? "");
+  const [saving, setSaving] = useState(false);
+  const [error, setError] = useState<string | null>(null);
+
+  // Sync in when the dialog opens, or when the viewed extension changes
+  // while the dialog stays mounted (e.g. re-parented to a different id).
+  useEffect(() => {
+    if (open) setCode(extension.content ?? "");
+  }, [open, extension.id]);
+
+  const isDirty = code !== (extension.content ?? "");
+
+  // Block Escape / outside-click from closing while there are unsaved edits.
+  const handleOpenChange = (next: boolean) => {
+    if (!next && isDirty) return;
+    setOpen(next);
+    if (!next) setError(null);
+  };
+
+  const handleCancel = () => {
+    setCode(extension.content ?? "");
+    setOpen(false);
+    setError(null);
+  };
+
+  const handleSave = async () => {
+    setSaving(true);
+    setError(null);
+    try {
+      const res = await fetch(
+        agentNativePath(`/_agent-native/extensions/${extension.id}`),
+        {
+          method: "PUT",
+          headers: { "Content-Type": "application/json" },
+          body: JSON.stringify({ content: code }),
+        },
+      );
+      if (!res.ok) {
+        const body = await res.json().catch(() => ({}));
+        throw new Error(body?.error ?? `Save failed (${res.status})`);
+      }
+      setOpen(false);
+      queryClient.setQueryData<Extension>(["extension", extension.id], (old) =>
+        old ? { ...old, content: code } : old,
+      );
+      queryClient.invalidateQueries({
+        queryKey: ["extension", extension.id],
+      });
+      queryClient.invalidateQueries({ queryKey: ["extensions"] });
+      onSaved?.();
+    } catch (err: any) {
+      setError(err?.message ?? "Save failed");
+    } finally {
+      setSaving(false);
+    }
+  };
+
+  return (
+    <>
+      <Tooltip>
+        <TooltipTrigger asChild>
+          <button
+            type="button"
+            onClick={() => setOpen(true)}
+            className="inline-flex items-center justify-center rounded-md h-8 w-8 text-muted-foreground hover:bg-accent hover:text-accent-foreground cursor-pointer"
+          >
+            <IconCode className="h-4 w-4" />
+          </button>
+        </TooltipTrigger>
+        <TooltipContent>View / edit source</TooltipContent>
+      </Tooltip>
+      <Dialog open={open} onOpenChange={handleOpenChange}>
+        <DialogContent className="flex h-[85vh] w-[90vw] max-w-[900px] flex-col gap-0 overflow-hidden p-0">
+          <div className="flex shrink-0 items-center border-b border-border px-5 py-3 pr-12">
+            <DialogTitle className="truncate text-sm font-medium">
+              {extension.name} — source
+            </DialogTitle>
+          </div>
+          <textarea
+            className="flex-1 resize-none bg-muted/40 px-5 py-4 font-mono text-xs leading-relaxed text-foreground focus:outline-none"
+            value={code}
+            onChange={(e) => setCode(e.target.value)}
+            spellCheck={false}
+          />
+          <div className="flex shrink-0 items-center justify-between border-t border-border px-5 py-3">
+            {error ? (
+              <p className="text-xs text-destructive">{error}</p>
+            ) : (
+              <span className="text-xs text-muted-foreground">
+                Alpine.js / HTML &middot; {code.length.toLocaleString()} chars
+              </span>
+            )}
+            <div className="flex items-center gap-2">
+              <button
+                type="button"
+                onClick={handleCancel}
+                className="inline-flex h-8 cursor-pointer items-center rounded-md border border-input px-3 text-xs hover:bg-accent"
+              >
+                Cancel
+              </button>
+              <button
+                type="button"
+                onClick={handleSave}
+                disabled={saving}
+                className="inline-flex h-8 cursor-pointer items-center rounded-md bg-primary px-4 text-xs font-medium text-primary-foreground hover:opacity-90 disabled:opacity-50"
+              >
+                {saving ? "Saving…" : "Save"}
+              </button>
+            </div>
+          </div>
+        </DialogContent>
+      </Dialog>
+    </>
+  );
+}
+
 function EditToolPopover({
   extension,
   onOpenChange,
@@ -1157,6 +1284,12 @@ export function ExtensionViewer({ extensionId }: ExtensionViewerProps) {
                   onRestored={() => setRefreshKey((k) => k + 1)}
                   onOpenChange={onPopoverOpenChange}
                 />
+                {extension.canEdit && (
+                  <SourceCodeDialog
+                    extension={extension}
+                    onSaved={() => setRefreshKey((k) => k + 1)}
+                  />
+                )}
                 <EditToolPopover
                   extension={extension}
                   onOpenChange={onPopoverOpenChange}

package/corpus/core/src/client/extensions/ExtensionsSidebarSection.tsx

@@ -1,7 +1,7 @@
 import {
   IconChevronDown,
   IconPlus,
-  IconSettings,
+  IconFilter,
   IconStar,
   IconStarFilled,
   IconTrash,
@@ -570,7 +570,7 @@ function ExtensionSortMenu({
               className="inline-flex h-6 w-6 shrink-0 items-center justify-center rounded-md text-muted-foreground/45 opacity-0 transition-all hover:bg-accent hover:text-foreground focus:opacity-100 focus:outline-none focus:ring-2 focus:ring-ring group-hover/extensions-section:opacity-100"
               aria-label={copy.sortOptions}
             >
-              <IconSettings className="h-3.5 w-3.5" />
+              <IconFilter className="h-3.5 w-3.5" />
             </button>
           </DropdownMenuTrigger>
         </TooltipTrigger>

package/corpus/core/src/client/i18n.tsx

@@ -674,7 +674,7 @@ export function LanguagePicker({
   className?: string;
   includeSystem?: boolean;
   label?: string;
-  variant?: "select" | "icon";
+  variant?: "select" | "icon" | "ghost-icon";
 }) {
   const { locale, preference, setPreference } = useLocale();
   const [open, setOpen] = useState(false);
@@ -716,10 +716,15 @@ export function LanguagePicker({
             title={triggerLabel}
             data-language-picker-trigger
             className={cn(
-              "shrink-0 rounded-md border border-border bg-background text-foreground outline-none transition-colors hover:border-foreground/30 hover:bg-accent/40 hover:text-foreground focus-visible:ring-2 focus-visible:ring-ring focus-visible:ring-offset-2 data-[state=open]:border-foreground/30 data-[state=open]:bg-accent/40",
+              "shrink-0 rounded-md outline-none transition-colors focus-visible:ring-2 focus-visible:ring-ring focus-visible:ring-offset-2",
+              variant === "ghost-icon"
+                ? "flex h-9 w-9 items-center justify-center text-muted-foreground hover:bg-accent hover:text-foreground data-[state=open]:bg-accent data-[state=open]:text-foreground"
+                : "border border-border bg-background text-foreground hover:border-foreground/30 hover:bg-accent/40 hover:text-foreground data-[state=open]:border-foreground/30 data-[state=open]:bg-accent/40",
               variant === "icon"
                 ? "flex h-8 w-8 items-center justify-center"
-                : "flex h-9 w-full items-center justify-between gap-2 px-3 text-start text-sm",
+                : variant === "select"
+                  ? "flex h-9 w-full items-center justify-between gap-2 px-3 text-start text-sm"
+                  : null,
             )}
           >
             <span className="flex min-w-0 items-center gap-2">
@@ -740,12 +745,12 @@ export function LanguagePicker({
         </PopoverPrimitive.Trigger>
         <PopoverPrimitive.Portal>
           <PopoverPrimitive.Content
-            align={variant === "icon" ? "end" : "start"}
+            align={variant === "select" ? "start" : "end"}
             sideOffset={6}
             role="menu"
             className={cn(
               "z-[9999] max-h-[min(20rem,var(--radix-popover-content-available-height))] overflow-y-auto rounded-lg border border-border bg-popover p-1 text-popover-foreground shadow-lg outline-none will-change-[transform,opacity] data-[state=open]:animate-in data-[state=closed]:animate-out data-[state=closed]:fade-out-0 data-[state=open]:fade-in-0 data-[state=closed]:zoom-out-95 data-[state=open]:zoom-in-95 data-[state=closed]:duration-100 data-[state=open]:duration-150 data-[state=closed]:ease-in data-[state=open]:ease-out data-[side=bottom]:slide-in-from-top-1 data-[side=left]:slide-in-from-right-1 data-[side=right]:slide-in-from-left-1 data-[side=top]:slide-in-from-bottom-1",
-              variant === "icon"
+              variant === "icon" || variant === "ghost-icon"
                 ? "min-w-56"
                 : "w-[min(20rem,calc(100vw-2rem))] min-w-[var(--radix-popover-trigger-width)]",
             )}

package/corpus/core/src/client/sse-event-processor.ts

@@ -49,6 +49,7 @@ export interface SSEEvent {
   /** Server-assigned call identifier emitted on tool_start / tool_done events. */
   id?: string;
   label?: string;
+  progressBytes?: number;
   input?: Record<string, string>;
   result?: string;
   isError?: boolean;
@@ -144,9 +145,58 @@ export class AgentAutoContinueSignal extends Error {
 }
 
 export const SSE_NO_PROGRESS_TIMEOUT_MS = 75_000;
+export const SSE_ACTION_PREPARATION_STALL_TIMEOUT_MS = 90_000;
 
 type ActivityTrailEntry = AgentActivityTrailEntry;
 
+type PreparingActionState = {
+  tool?: string;
+  startedAt?: number;
+  /**
+   * Timestamp of the last real streaming progress for the in-preparation tool
+   * input. The server emits a throttled `activity` heartbeat per
+   * `tool-input-delta`, so while the model is actively streaming a (possibly
+   * very large) tool argument this keeps advancing. The stall guard measures
+   * silence from HERE — not from `startedAt` — so a legitimately large, still-
+   * streaming input is never aborted; only genuine silence (keepalive-only, no
+   * further deltas) can trip it.
+   */
+  lastProgressAt?: number;
+};
+
+function formatProgressBytes(bytes: number): string {
+  if (bytes < 1024) return `${bytes} B`;
+  if (bytes < 1024 * 1024) return `${(bytes / 1024).toFixed(1)} KB`;
+  return `${(bytes / (1024 * 1024)).toFixed(1)} MB`;
+}
+
+function activityProgressBytes(ev: SSEEvent): number | undefined {
+  return typeof ev.progressBytes === "number" &&
+    Number.isFinite(ev.progressBytes) &&
+    ev.progressBytes >= 0
+    ? Math.floor(ev.progressBytes)
+    : undefined;
+}
+
+function isPreparingActionActivity(ev: SSEEvent): boolean {
+  if (ev.type !== "activity") return false;
+  const label = (ev.label ?? "").trim().toLowerCase();
+  return label.startsWith("preparing ") && label.includes(" action");
+}
+
+function baseActivityLabel(ev: SSEEvent, tool?: string): string {
+  return humanizeToolLabelText(ev.label ?? "Working", tool);
+}
+
+function visibleActivityLabel(ev: SSEEvent, tool?: string): string {
+  const label = baseActivityLabel(ev, tool);
+  const progressBytes = activityProgressBytes(ev);
+  if (progressBytes === undefined || !isPreparingActionActivity(ev)) {
+    return label;
+  }
+  return `${label} (${formatProgressBytes(progressBytes)} streamed)`;
+}
+
 function findPendingToolCallIndex(
   content: ContentPart[],
   toolName: string,
@@ -199,6 +249,53 @@ function appendActivityTrail(
   }
 }
 
+function updatePreparingActionState(
+  state: PreparingActionState,
+  ev: SSEEvent,
+  now: number,
+) {
+  if (ev.type === "activity" && isPreparingActionActivity(ev)) {
+    const tool = ev.tool?.trim() || undefined;
+    if (!tool) return;
+    if (state.tool !== tool || state.startedAt === undefined) {
+      state.tool = tool;
+      state.startedAt = now;
+    }
+    // Every tool-input activity is a throttled proof the model is still
+    // streaming this action's argument. Treat it as progress so large inputs
+    // that take many seconds/minutes to stream are NOT flagged as stalled.
+    state.lastProgressAt = now;
+    return;
+  }
+
+  if (
+    ev.type === "clear" ||
+    ev.type === "text" ||
+    ev.type === "tool_start" ||
+    ev.type === "tool_done" ||
+    ev.type === "done" ||
+    ev.type === "error" ||
+    ev.type === "missing_api_key"
+  ) {
+    state.tool = undefined;
+    state.startedAt = undefined;
+    state.lastProgressAt = undefined;
+  }
+}
+
+function hasStalledPreparingAction(state: PreparingActionState, now: number) {
+  // Fire only when a tool input has gone SILENT — no further streaming deltas
+  // for the whole window — never merely because a large input has been
+  // streaming for a long time. `lastProgressAt` advances on every delta
+  // heartbeat, so an actively-streaming large output keeps resetting this and
+  // survives; a genuinely stuck prep (keepalive-only, no deltas) trips it.
+  return (
+    state.tool !== undefined &&
+    state.lastProgressAt !== undefined &&
+    now - state.lastProgressAt >= SSE_ACTION_PREPARATION_STALL_TIMEOUT_MS
+  );
+}
+
 async function readChunkWithProgressTimeout(
   reader: ReadableStreamDefaultReader<Uint8Array>,
   lastMeaningfulEventAt: number,
@@ -491,7 +588,7 @@ export function processEvent(
 
   if (ev.type === "activity") {
     const tool = ev.tool?.trim() || undefined;
-    const label = humanizeToolLabelText(ev.label ?? "Working", tool);
+    const label = visibleActivityLabel(ev, tool);
     if (typeof window !== "undefined") {
       window.dispatchEvent(
         new CustomEvent("agent-chat:activity", {
@@ -941,6 +1038,7 @@ export async function* readSSEStream(
   let buf = "";
   let lastMeaningfulEventAt = Date.now();
   const activityTrail: ActivityTrailEntry[] = [];
+  const preparingActionState: PreparingActionState = {};
   const processEventState: ProcessEventState = {
     completedToolsAfterLastAssistantText: new Set(),
   };
@@ -1000,7 +1098,9 @@ export async function* readSSEStream(
           continue;
         }
         sawDataEvent = true;
-        lastMeaningfulEventAt = Date.now();
+        const now = Date.now();
+        lastMeaningfulEventAt = now;
+        updatePreparingActionState(preparingActionState, ev, now);
 
         // Track sequence number for reconnection
         if (ev.seq !== undefined && onSeq) {
@@ -1012,7 +1112,7 @@ export async function* readSSEStream(
         } else if (ev.type === "activity") {
           const tool = ev.tool?.trim() || undefined;
           appendActivityTrail(activityTrail, {
-            label: humanizeToolLabelText(ev.label ?? "Working", tool),
+            label: baseActivityLabel(ev, tool),
             ...(tool ? { tool } : {}),
           });
         } else if (ev.type === "tool_start") {
@@ -1039,6 +1139,12 @@ export async function* readSSEStream(
         );
 
         if (result) yield withStreamMetadata(result);
+        if (hasStalledPreparingAction(preparingActionState, Date.now())) {
+          throw new AgentAutoContinueSignal({
+            reason: "no_progress",
+            activityTrail: [...activityTrail],
+          });
+        }
         if (action === "auto_continue") {
           throw new AgentAutoContinueSignal(
             autoContinue
@@ -1099,6 +1205,8 @@ export async function readSSEStreamRaw(
   const decoder = new TextDecoder();
   let buf = "";
   let lastMeaningfulEventAt = Date.now();
+  const activityTrail: ActivityTrailEntry[] = [];
+  const preparingActionState: PreparingActionState = {};
   const processEventState: ProcessEventState = {
     completedToolsAfterLastAssistantText: new Set(),
   };
@@ -1134,12 +1242,37 @@ export async function readSSEStreamRaw(
           continue;
         }
         sawDataEvent = true;
-        lastMeaningfulEventAt = Date.now();
+        const now = Date.now();
+        lastMeaningfulEventAt = now;
+        updatePreparingActionState(preparingActionState, ev, now);
 
         if (ev.seq !== undefined && onSeq) {
           onSeq(ev.seq);
         }
 
+        if (ev.type === "clear") {
+          activityTrail.length = 0;
+        } else if (ev.type === "activity") {
+          const tool = ev.tool?.trim() || undefined;
+          appendActivityTrail(activityTrail, {
+            label: baseActivityLabel(ev, tool),
+            ...(tool ? { tool } : {}),
+          });
+        } else if (ev.type === "tool_start") {
+          const tool = ev.tool ?? "unknown";
+          appendActivityTrail(activityTrail, {
+            label: runningToolLabel(tool),
+            tool,
+          });
+        } else if (ev.type === "tool_done") {
+          const tool = ev.tool ?? "unknown";
+          for (let i = activityTrail.length - 1; i >= 0; i--) {
+            if (activityTrail[i]?.tool === tool) {
+              activityTrail.splice(i, 1);
+            }
+          }
+        }
+
         const { action, autoContinue } = processEvent(
           ev,
           content,
@@ -1159,9 +1292,18 @@ export async function readSSEStreamRaw(
         if (action === "auto_continue") {
           onUpdate([...content]);
           throw new AgentAutoContinueSignal(
-            autoContinue ?? { reason: "stream_ended" },
+            autoContinue
+              ? { ...autoContinue, activityTrail: [...activityTrail] }
+              : { reason: "stream_ended", activityTrail: [...activityTrail] },
           );
         }
+        if (hasStalledPreparingAction(preparingActionState, Date.now())) {
+          onUpdate([...content]);
+          throw new AgentAutoContinueSignal({
+            reason: "no_progress",
+            activityTrail: [...activityTrail],
+          });
+        }
         if (
           action === "done" ||
           action === "error" ||

package/corpus/core/src/extensions/html-shell.ts

@@ -226,11 +226,19 @@ export function buildExtensionHtml(
   </style>
 	  <style>
 	    *, *::before, *::after { border-color: hsl(var(--border)); }
+	    html, body {
+	      /* Transparent so the iframe inherits the host surface (dashboard panel,
+	         sidebar, chat) instead of painting the browser's default white canvas.
+	         The dark class still flips the theme vars; content paints its own
+	         bg-background / bg-card surfaces. */
+	      background: transparent;
+	    }
 	    body {
 	      --agent-native-extension-padding: clamp(16px, 2vw, 24px);
 	      /* Legacy alias for pre-rename extension content (do not remove). */
 	      --agent-native-tool-padding: var(--agent-native-extension-padding);
 	      box-sizing: border-box;
+	      color: hsl(var(--foreground));
 	      font-family: 'Inter', sans-serif;
 	      margin: 0;
 	      min-height: 100vh;
@@ -403,8 +411,14 @@ export function buildExtensionHtml(
 	      }
 
 	      if (!res.ok) {
-	        var err = res.body || { error: res.statusText };
-	        throw new Error(err.error || 'Action failed: ' + res.status);
+	        var err = res.body || {};
+	        var rawError = typeof err === 'string' ? err : err.error;
+	        var message = (typeof rawError === 'string' && rawError.trim())
+	          ? rawError
+	          : (res.status === 404
+	            ? "Action '" + name + "' is not available over HTTP (404). It may be agent-only (http: false); expose it with an HTTP-mounted action to call it from an extension."
+	            : "Action '" + name + "' failed (" + (res.status || 'network error') + ")");
+	        throw new Error(message);
 	      }
 	      return res.body;
 	    }
@@ -730,7 +744,7 @@ export function buildExtensionHtml(
 	    });
 	  </script>
 	</head>
-	<body${extensionId ? ` data-extension-id="${extensionIdAttr}" data-tool-id="${extensionIdAttr}"` : ""} class="bg-background text-foreground">
+	<body${extensionId ? ` data-extension-id="${extensionIdAttr}" data-tool-id="${extensionIdAttr}"` : ""} class="text-foreground">
 	${content}
 	<div id="__extension-error-toast">
 	  <div style="display:flex;align-items:flex-start;gap:8px;">

package/corpus/core/src/extensions/theme.ts

@@ -1,3 +1,42 @@
+/**
+ * Canonical set of semantic theme token names shared by the app and extension
+ * iframes. This is the single source of truth: `getThemeVars` bakes default
+ * values for these into every iframe, and the host reads the *actual* computed
+ * values for the same names to sync its live palette into embedded extensions
+ * (see EmbeddedExtension). Custom properties can't be enumerated portably from
+ * a computed style, so the names must be listed explicitly.
+ */
+export const THEME_VAR_NAMES = [
+  "--background",
+  "--foreground",
+  "--card",
+  "--card-foreground",
+  "--popover",
+  "--popover-foreground",
+  "--primary",
+  "--primary-foreground",
+  "--secondary",
+  "--secondary-foreground",
+  "--muted",
+  "--muted-foreground",
+  "--accent",
+  "--accent-foreground",
+  "--destructive",
+  "--destructive-foreground",
+  "--border",
+  "--input",
+  "--ring",
+  "--radius",
+  "--sidebar-background",
+  "--sidebar-foreground",
+  "--sidebar-primary",
+  "--sidebar-primary-foreground",
+  "--sidebar-accent",
+  "--sidebar-accent-foreground",
+  "--sidebar-border",
+  "--sidebar-ring",
+] as const;
+
 /**
  * CSS variables baked into every extension iframe. Both light and dark are
  * always emitted — the `.dark` class on the iframe's `<html>` toggles between

package/corpus/core/src/file-upload/builder.ts

@@ -27,6 +27,15 @@ function makeBody(bytes: Uint8Array, mimeType: string): BodyInit {
     : (bytes as unknown as BodyInit);
 }
 
+function shouldUseSignedUrlUpload(
+  bytes: Uint8Array,
+  mimeType: string,
+): boolean {
+  return (
+    bytes.byteLength > LARGE_FILE_THRESHOLD_BYTES || /^video\//i.test(mimeType)
+  );
+}
+
 function fetchWithTimeout(url: string, init: RequestInit): Promise<Response> {
   const controller = new AbortController();
   const timer = setTimeout(() => controller.abort(), UPLOAD_TIMEOUT_MS);
@@ -86,6 +95,7 @@ async function uploadLargeFileViaSignedUrl(
     privateKey,
     assetId,
     input.filename,
+    { skipCompressionWait: input.skipCompressionWait },
   );
   console.log(`[builder-upload] done [${assetId}]: ${url}`);
   return { url, id, provider: "builder" };
@@ -139,19 +149,21 @@ async function completeBuilderUpload(
   privateKey: string,
   assetId: string,
   filename: string | undefined,
+  options?: { skipCompressionWait?: boolean },
 ): Promise<{ url: string; id?: string }> {
   const host = builderUploadHost();
-  const res = await fetchWithTimeout(
-    new URL("/api/v1/upload/complete", host).toString(),
-    {
-      method: "POST",
-      headers: {
-        Authorization: `Bearer ${privateKey}`,
-        "Content-Type": "application/json",
-      },
-      body: JSON.stringify({ assetId, name: filename }),
+  const url = new URL("/api/v1/upload/complete", host);
+  if (options?.skipCompressionWait) {
+    url.searchParams.set("skipCompressionWait", "true");
+  }
+  const res = await fetchWithTimeout(url.toString(), {
+    method: "POST",
+    headers: {
+      Authorization: `Bearer ${privateKey}`,
+      "Content-Type": "application/json",
     },
-  );
+    body: JSON.stringify({ assetId, name: filename }),
+  });
   await assertOk(res, "Builder.io upload complete failed");
   const json = (await res.json()) as { url?: string; id?: string };
   if (!json.url) throw new Error("Builder.io upload/complete returned no URL");
@@ -204,7 +216,8 @@ export const builderFileUploadProvider: FileUploadProvider = {
   id: "builder",
   name: "Builder.io",
   isConfigured: () => !!process.env.BUILDER_PRIVATE_KEY,
-  upload: async ({ data, filename, mimeType }: FileUploadInput) => {
+  upload: async (input: FileUploadInput) => {
+    const { data, filename, mimeType } = input;
     const { resolveBuilderPrivateKey } =
       await import("../server/credential-provider.js");
     const privateKey = await resolveBuilderPrivateKey();
@@ -226,9 +239,9 @@ export const builderFileUploadProvider: FileUploadProvider = {
       data instanceof Uint8Array ? data : new Uint8Array(data as any);
     const mb = (bytes.byteLength / (1024 * 1024)).toFixed(1);
 
-    if (bytes.byteLength > LARGE_FILE_THRESHOLD_BYTES) {
+    if (shouldUseSignedUrlUpload(bytes, bareMimeType)) {
       return uploadLargeFileViaSignedUrl(
-        { data, filename, mimeType },
+        input,
         privateKey,
         bareMimeType,
         bytes,
@@ -241,6 +254,9 @@ export const builderFileUploadProvider: FileUploadProvider = {
 
     const url = new URL("/api/v1/upload", builderUploadHost());
     if (filename) url.searchParams.set("name", filename);
+    if (input.skipCompressionWait) {
+      url.searchParams.set("skipCompressionWait", "true");
+    }
 
     const response = await uploadSmallFile(url, {
       method: "POST",
@@ -366,7 +382,7 @@ export const builderFileUploadProvider: FileUploadProvider = {
         : new Error("GCS PUT failed after retries");
     },
 
-    async completeSession(session, filename) {
+    async completeSession(session, filename, options) {
       const { resolveBuilderPrivateKey } =
         await import("../server/credential-provider.js");
       const privateKey = await resolveBuilderPrivateKey();
@@ -378,6 +394,11 @@ export const builderFileUploadProvider: FileUploadProvider = {
         privateKey,
         assetId,
         filename,
+        {
+          skipCompressionWait:
+            options?.skipCompressionWait ||
+            session.meta.skipCompressionWait === true,
+        },
       );
       console.log(`[builder-resumable] upload complete: ${url}`);
       return url;

package/corpus/core/src/file-upload/types.ts

@@ -16,6 +16,8 @@ export interface FileUploadInput {
   mimeType?: string;
   /** Optional owner email for per-user scoping in fallback storage. */
   ownerEmail?: string;
+  /** Builder.io upload hint: return after asset registration instead of waiting for server-side compression. */
+  skipCompressionWait?: boolean;
 }
 
 export interface FileUploadResult {
@@ -77,6 +79,7 @@ export interface FileUploadProvider {
     completeSession(
       session: ResumableUploadSession,
       filename: string,
+      options?: { skipCompressionWait?: boolean },
     ): Promise<string>;
   };
 }