@mindstudio-ai/remy 0.1.178 → 0.1.180

package/dist/index.js

@@ -90,7 +90,7 @@ async function* streamChat(params) {
   const url = `${baseUrl2}/_internal/v2/agent/remy/chat`;
   const startTime = Date.now();
   const subAgentId = rest.subAgentId;
-  const requestBody = { ...rest, ...model && { modelId: model } };
+  const requestBody = { ...rest, modelId: model };
   log.info("API request", {
     requestId,
     ...subAgentId && { subAgentId },
@@ -2024,6 +2024,104 @@ var init_prompt = __esm({
   }
 });
 
+// src/models/surfaces.ts
+function resolveModel(surfaceId, models, fallback) {
+  return models?.[surfaceId] ?? fallback ?? MODEL_SURFACES[surfaceId].default;
+}
+var MODEL_SURFACES, ALLOWED_MODELS_BY_TYPE;
+var init_surfaces = __esm({
+  "src/models/surfaces.ts"() {
+    "use strict";
+    MODEL_SURFACES = {
+      parent: {
+        default: "claude-4-7-opus",
+        label: "Remy",
+        description: "The main Remy agent you chat with about your product. Writes code and manages delegation to other agents.",
+        modelType: "text",
+        userPickable: true
+      },
+      visualDesignExpert: {
+        default: "claude-4-7-opus",
+        label: "Design Agent",
+        description: "Designs your product's interfaces, including components, layouts, typography, color, and visual identity.",
+        modelType: "text",
+        userPickable: true
+      },
+      productVision: {
+        default: "claude-4-6-sonnet",
+        label: "Roadmap Agent",
+        description: "Owns your product's roadmap and pitch deck. Helps decide what to build next and how to frame the big picture.",
+        modelType: "text",
+        userPickable: true
+      },
+      browserAutomation: {
+        default: "claude-4-6-sonnet",
+        label: "QA Agent",
+        description: "Tests features and UI flows in an automated browser to verify they work end to end.",
+        modelType: "text",
+        userPickable: true
+      },
+      codeSanityCheck: {
+        default: "claude-4-6-sonnet",
+        label: "Architecture Agent",
+        description: "Reviews the architecture and structure of code changes to avoid technical debt.",
+        modelType: "text",
+        userPickable: true
+      },
+      imageGeneration: {
+        default: "seedream-4.5",
+        label: "Image Generation",
+        description: "Creates images for your product \u2014 icons, illustrations, photos, and any other visual assets.",
+        modelType: "image_generation",
+        userPickable: true
+      },
+      imageAnalysis: {
+        default: "claude-4-6-sonnet",
+        label: "Image Analysis",
+        description: "Reads screenshots taken by the QA agent during automated browser tests. Other agents use their own built-in image analysis when they need to read images.",
+        modelType: "vision",
+        userPickable: true
+      },
+      conversationSummarizer: {
+        default: "claude-4-6-sonnet",
+        label: "Compaction Utility",
+        description: "Compresses long conversations into summaries to keep things responsive.",
+        modelType: "text",
+        userPickable: true
+      },
+      brandExtractor: {
+        default: "claude-4-6-sonnet",
+        label: "Brand Utility",
+        description: "Extracts your product's name, colors, and fonts from your spec for use in branded documents.",
+        modelType: "text",
+        userPickable: true
+      },
+      // Internal surface — not user-pickable. Remy uses this to rewrite design
+      // briefs into model-optimized image prompts before image generation.
+      imagePromptEnhancer: {
+        default: "claude-4-6-sonnet",
+        label: "Image Prompt Enhancer",
+        description: "Rewrites image briefs into model-optimized prompts before image generation.",
+        modelType: "text",
+        userPickable: false
+      }
+    };
+    ALLOWED_MODELS_BY_TYPE = {
+      text: [
+        "claude-4-7-opus",
+        "claude-4-6-opus",
+        "claude-4-6-sonnet",
+        "gpt-5.5",
+        "gemini-3-pro",
+        "gemini-3.1-pro",
+        "gemini-3-flash"
+      ]
+      // vision: undefined — unconstrained
+      // image_generation: undefined — unconstrained
+    };
+  }
+});
+
 // src/compaction/trigger.ts
 function getPendingSummaries() {
   return pendingSummaries.splice(0);
@@ -2044,7 +2142,7 @@ function triggerCompaction(state, apiConfig, opts = {}) {
     apiConfig,
     system,
     tools2,
-    state.models?.conversationSummarizer ?? model
+    resolveModel("conversationSummarizer", state.models, model)
   ).then((summaries) => {
     pendingSummaries.push(...summaries);
     listener?.({ type: "complete", requestId });
@@ -2067,6 +2165,7 @@ var init_trigger = __esm({
     init_prompt();
     init_tools6();
     init_logger();
+    init_surfaces();
     log3 = createLogger("compaction:trigger");
     pendingSummaries = [];
     inflightCompaction = null;
@@ -3085,7 +3184,7 @@ var init_queryDatabase = __esm({
 
 // src/subagents/common/analyzeImage.ts
 async function analyzeImage(params) {
-  const { prompt, imageUrl, timeout = 2e5, onLog } = params;
+  const { prompt, imageUrl, model, timeout = 2e5, onLog } = params;
   return runMindstudioCli(
     [
       "analyze-image",
@@ -3094,21 +3193,15 @@ async function analyzeImage(params) {
       "--image-url",
       imageUrl,
       "--vision-model-override",
-      JSON.stringify(VISION_MODEL_OVERRIDE)
+      JSON.stringify({ model })
     ],
     { outputKey: "analysis", timeout, onLog }
   );
 }
-var VISION_MODEL, VISION_MODEL_OVERRIDE;
 var init_analyzeImage = __esm({
   "src/subagents/common/analyzeImage.ts"() {
     "use strict";
     init_runMindstudioCli();
-    VISION_MODEL = "claude-4-6-sonnet";
-    VISION_MODEL_OVERRIDE = {
-      model: VISION_MODEL,
-      config: { thinkingBudget: "off" }
-    };
   }
 });
 
@@ -3130,13 +3223,14 @@ ${TEXT_WRAP_DISCLAIMER}`;
   return p;
 }
 async function streamScreenshotAnalysis(opts) {
-  const { url, prompt, styleMap, onLog } = opts;
+  const { url, prompt, styleMap, onLog, model } = opts;
   onLog?.(JSON.stringify({ url, analysis: null }));
   const analysisPrompt = buildScreenshotAnalysisPrompt({ prompt, styleMap });
   let accumulated = "";
   const analysis = await analyzeImage({
     prompt: analysisPrompt,
     imageUrl: url,
+    model,
     onLog: (chunk) => {
       accumulated += chunk;
       onLog?.(JSON.stringify({ url, analysis: accumulated }));
@@ -3148,12 +3242,14 @@ async function captureAndAnalyzeScreenshot(promptOrOptions) {
   let prompt;
   let existingUrl;
   let onLog;
+  let model;
   let path13;
   if (typeof promptOrOptions === "object" && promptOrOptions !== null) {
     prompt = promptOrOptions.prompt;
     existingUrl = promptOrOptions.imageUrl;
     path13 = promptOrOptions.path;
     onLog = promptOrOptions.onLog;
+    model = promptOrOptions.model;
   } else {
     prompt = promptOrOptions;
   }
@@ -3178,11 +3274,17 @@ async function captureAndAnalyzeScreenshot(promptOrOptions) {
   if (prompt === false) {
     return url;
   }
+  if (!model) {
+    throw new Error(
+      "captureAndAnalyzeScreenshot: `model` is required when analysis is enabled"
+    );
+  }
   return streamScreenshotAnalysis({
     url,
     prompt: prompt || void 0,
     styleMap,
-    onLog
+    onLog,
+    model
   });
 }
 var SCREENSHOT_ANALYSIS_PROMPT, TEXT_WRAP_DISCLAIMER;
@@ -4074,6 +4176,121 @@ var init_prompt2 = __esm({
 });
 
 // src/subagents/browserAutomation/index.ts
+async function runBrowserAutomation(task, context) {
+  const release = await acquireBrowserLock();
+  try {
+    const result = await runSubAgent({
+      system: getBrowserAutomationPrompt(),
+      task,
+      tools: BROWSER_TOOLS,
+      externalTools: BROWSER_EXTERNAL_TOOLS,
+      executeTool: async (name, _input, _toolCallId, onLog) => {
+        if (name === "setupBrowser") {
+          try {
+            const result2 = await sidecarRequest(
+              "/setup-browser",
+              {
+                auth: _input.auth,
+                path: _input.path
+              },
+              { timeout: 15e3 }
+            );
+            return JSON.stringify(result2);
+          } catch (err) {
+            return `Error setting up browser: ${err.message}`;
+          }
+        }
+        if (name === "screenshotFullPage") {
+          try {
+            return await captureAndAnalyzeScreenshot({
+              path: _input.path,
+              onLog,
+              model: resolveModel(
+                "imageAnalysis",
+                context.models,
+                context.model
+              )
+            });
+          } catch (err) {
+            return `Error taking screenshot: ${err.message}`;
+          }
+        }
+        return `Error: unknown local tool "${name}"`;
+      },
+      apiConfig: context.apiConfig,
+      model: resolveModel("browserAutomation", context.models, context.model),
+      subAgentId: "browserAutomation",
+      signal: context.signal,
+      parentToolId: context.toolCallId,
+      requestId: context.requestId,
+      onEvent: context.onEvent,
+      resolveExternalTool: async (id, name, input) => {
+        if (!context.resolveExternalTool) {
+          return "Error: no external tool resolver";
+        }
+        const result2 = await context.resolveExternalTool(id, name, input);
+        if (name === "browserCommand") {
+          try {
+            const parsed = JSON.parse(result2);
+            const screenshotSteps = (parsed.steps || []).filter(
+              (s) => s.command === "screenshotViewport" && s.result?.url
+            );
+            if (screenshotSteps.length > 0) {
+              const visionOverride = {
+                model: resolveModel(
+                  "imageAnalysis",
+                  context.models,
+                  context.model
+                )
+              };
+              const batchInput = screenshotSteps.map((s) => ({
+                stepType: "analyzeImage",
+                step: {
+                  imageUrl: s.result.url,
+                  prompt: buildScreenshotAnalysisPrompt({
+                    styleMap: s.result.styleMap
+                  }),
+                  visionModelOverride: visionOverride
+                }
+              }));
+              const batchResult = await runMindstudioCli(
+                ["batch", JSON.stringify(batchInput)],
+                { timeout: 2e5, caller: "browserAutomation" }
+              );
+              try {
+                const analyses = JSON.parse(batchResult);
+                let ai = 0;
+                for (const step of parsed.steps) {
+                  if (step.command === "screenshotViewport" && step.result?.url && ai < analyses.length) {
+                    step.result.analysis = analyses[ai]?.output?.analysis || analyses[ai]?.output || "";
+                    ai++;
+                  }
+                }
+              } catch {
+                log6.debug("Failed to parse batch analysis result", {
+                  batchResult
+                });
+              }
+              return JSON.stringify(parsed);
+            }
+          } catch {
+          }
+        }
+        return result2;
+      },
+      toolRegistry: context.toolRegistry,
+      captureArtifacts: ["screenshotFullPage"]
+    });
+    context.subAgentMessages?.set(context.toolCallId, result.messages);
+    const ss = result.artifacts?.screenshotFullPage;
+    return {
+      text: result.text,
+      ...ss?.url ? { screenshot: { url: ss.url, styleMap: ss.styleMap } } : {}
+    };
+  } finally {
+    release();
+  }
+}
 var log6, browserAutomationTool;
 var init_browserAutomation = __esm({
   "src/subagents/browserAutomation/index.ts"() {
@@ -4085,7 +4302,7 @@ var init_browserAutomation = __esm({
     init_browserLock();
     init_screenshot();
     init_runMindstudioCli();
-    init_analyzeImage();
+    init_surfaces();
     init_logger();
     log6 = createLogger("browser-automation");
     browserAutomationTool = {
@@ -4108,109 +4325,13 @@ var init_browserAutomation = __esm({
         if (!context) {
           return "Error: browser automation requires execution context (only available in headless mode)";
         }
-        const release = await acquireBrowserLock();
-        try {
-          const result = await runSubAgent({
-            system: getBrowserAutomationPrompt(),
-            task: input.task,
-            tools: BROWSER_TOOLS,
-            externalTools: BROWSER_EXTERNAL_TOOLS,
-            executeTool: async (name, _input, _toolCallId, onLog) => {
-              if (name === "setupBrowser") {
-                try {
-                  const result2 = await sidecarRequest(
-                    "/setup-browser",
-                    {
-                      auth: _input.auth,
-                      path: _input.path
-                    },
-                    { timeout: 15e3 }
-                  );
-                  return JSON.stringify(result2);
-                } catch (err) {
-                  return `Error setting up browser: ${err.message}`;
-                }
-              }
-              if (name === "screenshotFullPage") {
-                try {
-                  return await captureAndAnalyzeScreenshot({
-                    path: _input.path,
-                    onLog
-                  });
-                } catch (err) {
-                  return `Error taking screenshot: ${err.message}`;
-                }
-              }
-              return `Error: unknown local tool "${name}"`;
-            },
-            apiConfig: context.apiConfig,
-            model: context.models?.browserAutomation ?? context.model,
-            subAgentId: "browserAutomation",
-            signal: context.signal,
-            parentToolId: context.toolCallId,
-            requestId: context.requestId,
-            onEvent: context.onEvent,
-            resolveExternalTool: async (id, name, input2) => {
-              if (!context.resolveExternalTool) {
-                return "Error: no external tool resolver";
-              }
-              const result2 = await context.resolveExternalTool(id, name, input2);
-              if (name === "browserCommand") {
-                try {
-                  const parsed = JSON.parse(result2);
-                  const screenshotSteps = (parsed.steps || []).filter(
-                    (s) => s.command === "screenshotViewport" && s.result?.url
-                  );
-                  if (screenshotSteps.length > 0) {
-                    const batchInput = screenshotSteps.map((s) => ({
-                      stepType: "analyzeImage",
-                      step: {
-                        imageUrl: s.result.url,
-                        prompt: buildScreenshotAnalysisPrompt({
-                          styleMap: s.result.styleMap
-                        }),
-                        visionModelOverride: VISION_MODEL_OVERRIDE
-                      }
-                    }));
-                    const batchResult = await runMindstudioCli(
-                      ["batch", JSON.stringify(batchInput)],
-                      { timeout: 2e5, caller: "browserAutomation" }
-                    );
-                    try {
-                      const analyses = JSON.parse(batchResult);
-                      let ai = 0;
-                      for (const step of parsed.steps) {
-                        if (step.command === "screenshotViewport" && step.result?.url && ai < analyses.length) {
-                          step.result.analysis = analyses[ai]?.output?.analysis || analyses[ai]?.output || "";
-                          ai++;
-                        }
-                      }
-                    } catch {
-                      log6.debug("Failed to parse batch analysis result", {
-                        batchResult
-                      });
-                    }
-                    return JSON.stringify(parsed);
-                  }
-                } catch {
-                }
-              }
-              return result2;
-            },
-            toolRegistry: context.toolRegistry,
-            captureArtifacts: ["screenshotFullPage"]
-          });
-          context.subAgentMessages?.set(context.toolCallId, result.messages);
-          const ss = result.artifacts?.screenshotFullPage;
-          if (ss?.url) {
-            return `${result.text}
+        const result = await runBrowserAutomation(input.task, context);
+        if (result.screenshot) {
+          return `${result.text}
 
-![Final state](${ss.url})`;
-          }
-          return result.text;
-        } finally {
-          release();
+![Final state](${result.screenshot.url})`;
         }
+        return result.text;
       }
     };
   }
@@ -4224,6 +4345,7 @@ var init_screenshot2 = __esm({
     init_screenshot();
     init_browserLock();
     init_browserAutomation();
+    init_surfaces();
     screenshotTool = {
       clearable: true,
       definition: {
@@ -4257,29 +4379,22 @@ var init_screenshot2 = __esm({
             return await captureAndAnalyzeScreenshot({
               prompt: input.prompt,
               imageUrl: input.imageUrl,
-              onLog: context?.onLog
+              onLog: context?.onLog,
+              model: resolveModel("imageAnalysis", context?.models, context?.model)
             });
           }
           if (input.instructions && context) {
             const task = input.path ? `Navigate to "${input.path}", then: ${input.instructions}. After completing these steps, take a full-page screenshot.` : `${input.instructions}. After completing these steps, take a full-page screenshot.`;
-            const result = await browserAutomationTool.execute({ task }, context);
-            const resultStr = result;
-            let url;
-            let styleMap;
-            try {
-              const parsed = JSON.parse(resultStr);
-              url = parsed.screenshotUrl;
-              styleMap = parsed.styleMap;
-            } catch {
-            }
-            if (!url) {
-              return `Error: browser navigation completed but no screenshot URL was returned. Agent output: ${resultStr}`;
+            const result = await runBrowserAutomation(task, context);
+            if (!result.screenshot) {
+              return result.text;
             }
             return await streamScreenshotAnalysis({
-              url,
+              url: result.screenshot.url,
               prompt: input.prompt,
-              styleMap,
-              onLog: context?.onLog
+              styleMap: result.screenshot.styleMap,
+              onLog: context?.onLog,
+              model: resolveModel("imageAnalysis", context?.models, context?.model)
             });
           }
           const release = await acquireBrowserLock();
@@ -4287,7 +4402,8 @@ var init_screenshot2 = __esm({
             return await captureAndAnalyzeScreenshot({
               prompt: input.prompt,
               path: input.path,
-              onLog: context?.onLog
+              onLog: context?.onLog,
+              model: resolveModel("imageAnalysis", context?.models, context?.model)
             });
           } finally {
             release();
@@ -4478,7 +4594,7 @@ __export(analyzeDesign_exports, {
   definition: () => definition3,
   execute: () => execute3
 });
-async function execute3(input, onLog) {
+async function execute3(input, onLog, context) {
   const url = input.url;
   const analysisPrompt = input.prompt || DESIGN_REFERENCE_PROMPT;
   const isImageUrl = /\.(png|jpe?g|webp|gif|svg|avif)(\?|$)/i.test(url);
@@ -4511,7 +4627,8 @@ async function execute3(input, onLog) {
   const analysis = await analyzeImage({
     prompt: analysisPrompt,
     imageUrl,
-    onLog
+    onLog,
+    model: resolveModel("imageAnalysis", context?.models, context?.model)
   });
   return JSON.stringify({ url: imageUrl, analysis });
 }
@@ -4521,6 +4638,7 @@ var init_analyzeDesign = __esm({
     "use strict";
     init_runMindstudioCli();
     init_analyzeImage();
+    init_surfaces();
     DESIGN_REFERENCE_PROMPT = `
 You are analyzing a screenshot of a real website or app for a designer's personal technique/inspiration reference notes.
 
@@ -4572,7 +4690,7 @@ __export(analyzeImage_exports, {
   definition: () => definition4,
   execute: () => execute4
 });
-async function execute4(input, onLog) {
+async function execute4(input, onLog, context) {
   const imageUrl = input.imageUrl;
   const prompt = buildScreenshotAnalysisPrompt({
     prompt: input.prompt
@@ -4580,7 +4698,8 @@ async function execute4(input, onLog) {
   const analysis = await analyzeImage({
     prompt,
     imageUrl,
-    onLog
+    onLog,
+    model: resolveModel("imageAnalysis", context?.models, context?.model)
   });
   return JSON.stringify({ url: imageUrl, analysis });
 }
@@ -4590,6 +4709,7 @@ var init_analyzeImage2 = __esm({
     "use strict";
     init_analyzeImage();
     init_screenshot();
+    init_surfaces();
     definition4 = {
       clearable: true,
       name: "analyzeImage",
@@ -4622,24 +4742,16 @@ async function execute5(input, onLog, context) {
   if (input.instructions && context) {
     try {
       const task = input.path ? `Navigate to "${input.path}", then: ${input.instructions}. After completing these steps, take a full-page screenshot.` : `${input.instructions}. After completing these steps, take a full-page screenshot.`;
-      const result = await browserAutomationTool.execute({ task }, context);
-      const resultStr = result;
-      let url;
-      let styleMap;
-      try {
-        const parsed = JSON.parse(resultStr);
-        url = parsed.screenshotUrl;
-        styleMap = parsed.styleMap;
-      } catch {
-      }
-      if (!url) {
-        return `Error: browser navigation completed but no screenshot URL was returned. Agent output: ${resultStr}`;
+      const result = await runBrowserAutomation(task, context);
+      if (!result.screenshot) {
+        return result.text;
       }
       return await streamScreenshotAnalysis({
-        url,
+        url: result.screenshot.url,
         prompt: input.prompt,
-        styleMap,
-        onLog
+        styleMap: result.screenshot.styleMap,
+        onLog,
+        model: resolveModel("imageAnalysis", context?.models, context?.model)
       });
     } catch (err) {
       return `Error taking interactive screenshot: ${err.message}`;
@@ -4650,7 +4762,8 @@ async function execute5(input, onLog, context) {
     return await captureAndAnalyzeScreenshot({
       prompt: input.prompt,
       path: input.path,
-      onLog
+      onLog,
+      model: resolveModel("imageAnalysis", context?.models, context?.model)
     });
   } catch (err) {
     return `Error taking screenshot: ${err.message}`;
@@ -4665,6 +4778,7 @@ var init_screenshot3 = __esm({
     init_screenshot();
     init_browserLock();
     init_browserAutomation();
+    init_surfaces();
     definition5 = {
       clearable: true,
       name: "screenshot",
@@ -4692,7 +4806,7 @@ var init_screenshot3 = __esm({
 
 // src/subagents/designExpert/tools/images/enhancePrompt.ts
 async function enhanceImagePrompt(params) {
-  const { brief, width, height, transparentBackground, onLog } = params;
+  const { brief, width, height, transparentBackground, onLog, model } = params;
   const contextParts = [
     `Dimensions: ${width}x${height}${width > height ? " (landscape)" : width < height ? " (portrait)" : " (square)"}`
   ];
@@ -4717,23 +4831,18 @@ ${brief}
       "--message",
       message,
       "--model-override",
-      JSON.stringify(MODEL_OVERRIDE)
+      JSON.stringify({ model, config: { reasoning: "false" } })
     ],
     { outputKey: "content", timeout: 6e4, onLog, caller: "designExpert" }
   );
   return enhanced.trim();
 }
-var ENHANCE_MODEL, MODEL_OVERRIDE, SYSTEM_PROMPT;
+var SYSTEM_PROMPT;
 var init_enhancePrompt = __esm({
   "src/subagents/designExpert/tools/images/enhancePrompt.ts"() {
     "use strict";
     init_runMindstudioCli();
     init_assets();
-    ENHANCE_MODEL = "claude-4-6-sonnet";
-    MODEL_OVERRIDE = {
-      model: ENHANCE_MODEL,
-      config: { reasoning: "false" }
-    };
     SYSTEM_PROMPT = readAsset(
       "subagents/designExpert/tools/images/enhance-image-prompt.md"
     );
@@ -4742,7 +4851,15 @@ var init_enhancePrompt = __esm({
 
 // src/subagents/designExpert/tools/images/imageGenerator.ts
 async function generateImageAssets(opts) {
-  const { prompts, sourceImages, transparentBackground, onLog } = opts;
+  const {
+    prompts,
+    sourceImages,
+    transparentBackground,
+    onLog,
+    imageGenerationModel: genModel,
+    imageAnalysisModel,
+    imagePromptEnhancerModel
+  } = opts;
   const width = opts.width || 2048;
   const height = opts.height || 2048;
   const config = { width, height };
@@ -4757,7 +4874,8 @@ async function generateImageAssets(opts) {
         width,
         height,
         transparentBackground,
-        onLog
+        onLog,
+        model: imagePromptEnhancerModel
       })
     )
   );
@@ -4766,7 +4884,7 @@ async function generateImageAssets(opts) {
     const step = JSON.stringify({
       prompt: enhancedPrompts[0],
       imageModelOverride: {
-        model: "seedream-4.5",
+        model: genModel,
         config
       }
     });
@@ -4785,7 +4903,7 @@ async function generateImageAssets(opts) {
       step: {
         prompt,
         imageModelOverride: {
-          model: "seedream-4.5",
+          model: genModel,
           config
         }
       }
@@ -4837,7 +4955,8 @@ async function generateImageAssets(opts) {
       const analysis = await analyzeImage({
         prompt: ANALYZE_PROMPT,
         imageUrl: url,
-        onLog
+        onLog,
+        model: imageAnalysisModel
       });
       return {
         url,
@@ -4868,13 +4987,28 @@ __export(generateImages_exports, {
   definition: () => definition6,
   execute: () => execute6
 });
-async function execute6(input, onLog) {
+async function execute6(input, onLog, context) {
   return generateImageAssets({
     prompts: input.prompts,
     width: input.width,
     height: input.height,
     transparentBackground: input.transparentBackground,
-    onLog
+    onLog,
+    imageGenerationModel: resolveModel(
+      "imageGeneration",
+      context?.models,
+      context?.model
+    ),
+    imageAnalysisModel: resolveModel(
+      "imageAnalysis",
+      context?.models,
+      context?.model
+    ),
+    imagePromptEnhancerModel: resolveModel(
+      "imagePromptEnhancer",
+      context?.models,
+      context?.model
+    )
   });
 }
 var definition6;
@@ -4882,6 +5016,7 @@ var init_generateImages = __esm({
   "src/subagents/designExpert/tools/images/generateImages.ts"() {
     "use strict";
     init_imageGenerator();
+    init_surfaces();
     definition6 = {
       clearable: false,
       name: "generateImages",
@@ -4921,14 +5056,29 @@ __export(editImages_exports, {
   definition: () => definition7,
   execute: () => execute7
 });
-async function execute7(input, onLog) {
+async function execute7(input, onLog, context) {
   return generateImageAssets({
     prompts: input.prompts,
     sourceImages: input.sourceImages,
     width: input.width,
     height: input.height,
     transparentBackground: input.transparentBackground,
-    onLog
+    onLog,
+    imageGenerationModel: resolveModel(
+      "imageGeneration",
+      context?.models,
+      context?.model
+    ),
+    imageAnalysisModel: resolveModel(
+      "imageAnalysis",
+      context?.models,
+      context?.model
+    ),
+    imagePromptEnhancerModel: resolveModel(
+      "imagePromptEnhancer",
+      context?.models,
+      context?.model
+    )
   });
 }
 var definition7;
@@ -4936,6 +5086,7 @@ var init_editImages = __esm({
   "src/subagents/designExpert/tools/images/editImages.ts"() {
     "use strict";
     init_imageGenerator();
+    init_surfaces();
     definition7 = {
       clearable: false,
       name: "editImages",
@@ -5474,6 +5625,7 @@ var init_designExpert = __esm({
     init_tools2();
     init_prompt3();
     init_history();
+    init_surfaces();
     DESCRIPTION = `
 Visual design expert. Describe the situation and what you need \u2014 the agent decides what to deliver. It reads the spec files automatically. Include relevant user requirements and context it can't get from the spec, but do not list specific deliverables or tell it how to do its job. Do not suggest implementation details or ideas - only relay what is needed.
 `.trim();
@@ -5522,7 +5674,7 @@ Visual design expert. Describe the situation and what you need \u2014 the agent
             );
           },
           apiConfig: context.apiConfig,
-          model: context.models?.visualDesignExpert ?? context.model,
+          model: resolveModel("visualDesignExpert", context.models, context.model),
           subAgentId: "visualDesignExpert",
           signal: context.signal,
           parentToolId: context.toolCallId,
@@ -5740,6 +5892,7 @@ var init_productVision = __esm({
     init_executor();
     init_prompt4();
     init_history();
+    init_surfaces();
     productVisionTool = {
       clearable: false,
       definition: {
@@ -5779,7 +5932,7 @@ var init_productVision = __esm({
             return executeVisionTool(name, input2, childCtx);
           },
           apiConfig: context.apiConfig,
-          model: context.models?.productVision ?? context.model,
+          model: resolveModel("productVision", context.models, context.model),
           subAgentId: "productVision",
           signal: context.signal,
           parentToolId: context.toolCallId,
@@ -5870,6 +6023,7 @@ var init_codeSanityCheck = __esm({
     init_context();
     init_tools6();
     init_tools5();
+    init_surfaces();
     BASE_PROMPT3 = readAsset("subagents/codeSanityCheck", "prompt.md");
     codeSanityCheckTool = {
       clearable: false,
@@ -5905,7 +6059,7 @@ var init_codeSanityCheck = __esm({
           externalTools: /* @__PURE__ */ new Set(),
           executeTool: (name, toolInput) => executeTool(name, toolInput, context),
           apiConfig: context.apiConfig,
-          model: context.models?.codeSanityCheck ?? context.model,
+          model: resolveModel("codeSanityCheck", context.models, context.model),
           subAgentId: "codeSanityCheck",
           signal: context.signal,
           parentToolId: context.toolCallId,
@@ -6869,7 +7023,7 @@ async function runTurn(params) {
         onEvent({ type: "tool_input_delta", id, name, result: content });
       }
     }
-    const parentModel = state.models?.parent ?? model;
+    const parentModel = resolveModel("parent", state.models, model);
     try {
       for await (const event of streamChatWithRetry(
         {
@@ -7213,7 +7367,10 @@ async function runTurn(params) {
           isError: r.isError
         });
         if (!r.isError && BRAND_TRIGGERING_TOOLS.has(tc.name)) {
-          triggerBrandExtraction(apiConfig);
+          triggerBrandExtraction(
+            apiConfig,
+            resolveModel("brandExtractor", state.models, model)
+          );
         }
         return r;
       })
@@ -7270,6 +7427,7 @@ var init_agent = __esm({
     init_tools6();
     init_sentinel();
     init_trigger2();
+    init_surfaces();
     log10 = createLogger("agent");
     BRAND_TRIGGERING_TOOLS = /* @__PURE__ */ new Set(["writeSpec", "editSpec"]);
     EXTERNAL_TOOLS = /* @__PURE__ */ new Set([
@@ -7731,6 +7889,7 @@ var init_headless = __esm({
     init_lsp();
     init_agent();
     init_session();
+    init_surfaces();
     init_toolRegistry();
     init_attachments();
     init_planFile();
@@ -7811,12 +7970,14 @@ var init_headless = __esm({
           this.emit("session_restored", {
             messageCount: this.state.messages.length,
             ...this.state.models && { models: this.state.models },
+            modelSurfaces: MODEL_SURFACES,
+            allowedModelsByType: ALLOWED_MODELS_BY_TYPE,
             ...this.queueFields()
           });
         }
         triggerBrandExtraction(
           this.config,
-          this.state.models?.brandExtractor ?? this.opts.model
+          resolveModel("brandExtractor", this.state.models, this.opts.model)
         );
         this.toolRegistry.onEvent = this.onEvent;
         setCompactionListener((event) => {
@@ -8494,6 +8655,8 @@ var init_headless = __esm({
             running: this.running,
             ...this.running && this.currentRequestId ? { currentRequestId: this.currentRequestId } : {},
             ...this.state.models && { models: this.state.models },
+            modelSurfaces: MODEL_SURFACES,
+            allowedModelsByType: ALLOWED_MODELS_BY_TYPE,
             ...this.queueFields()
           }));
           return;