test(ci): bound image tool iMessage fixtures

steipete · steipete · commit 550a9b459a35 · 2026-05-27T18:50:49.000-04:00
diff --git a/src/agents/tools/image-tool.test.ts b/src/agents/tools/image-tool.test.ts
@@ -544,6 +544,9 @@ function installImageUnderstandingProviderDeps(
     loadImageWebMediaRuntime?: NonNullable<
       Parameters<typeof testing.setProviderDepsForTest>[0]
     >["loadImageWebMediaRuntime"];
+    resolveImageCompressionPolicy?: NonNullable<
+      Parameters<typeof testing.setProviderDepsForTest>[0]
+    >["resolveImageCompressionPolicy"];
     resolveModelAsync?: NonNullable<
       Parameters<typeof testing.setProviderDepsForTest>[0]
     >["resolveModelAsync"];
@@ -574,6 +577,9 @@ function installImageUnderstandingProviderDeps(
       capability === "image" ? ["openai", "anthropic"] : [],
     resolveDefaultMediaModel: ({ providerId, capability }) =>
       capability === "image" ? defaultImageModels.get(providerId.toLowerCase()) : undefined,
+    ...(options?.resolveImageCompressionPolicy
+      ? { resolveImageCompressionPolicy: options.resolveImageCompressionPolicy }
+      : {}),
     ...(options?.resolveModelAsync ? { resolveModelAsync: options.resolveModelAsync } : {}),
     ...(options?.loadImageWebMediaRuntime
       ? { loadImageWebMediaRuntime: options.loadImageWebMediaRuntime }
@@ -587,6 +593,7 @@ function installImageUnderstandingProviderStubs(...providers: MediaUnderstanding
 
 function installFastLocalImageProviderStubs(...providers: MediaUnderstandingProvider[]) {
   installImageUnderstandingProviderDeps(providers, {
+    resolveImageCompressionPolicy: async ({ imageCount }) => ({ imageCount }),
     resolveModelAsync: async (provider, model) => ({
       model: {
         id: model,
diff --git a/src/agents/tools/image-tool.ts b/src/agents/tools/image-tool.ts
@@ -111,6 +111,7 @@ const imageToolProviderDeps = {
   resolveDefaultMediaModel,
   resolveBundledStaticCatalogModel,
   resolveModelAsync,
+  resolveImageCompressionPolicy,
   loadImageWebMediaRuntime,
 };
 
@@ -173,6 +174,7 @@ export const testing = {
     resolveDefaultMediaModel?: typeof resolveDefaultMediaModel;
     resolveBundledStaticCatalogModel?: typeof resolveBundledStaticCatalogModel;
     resolveModelAsync?: typeof resolveModelAsync;
+    resolveImageCompressionPolicy?: typeof resolveImageCompressionPolicy;
     loadImageWebMediaRuntime?: typeof loadImageWebMediaRuntime;
   }) {
     imageToolProviderDeps.buildProviderRegistry =
@@ -190,6 +192,8 @@ export const testing = {
     imageToolProviderDeps.resolveBundledStaticCatalogModel =
       overrides?.resolveBundledStaticCatalogModel ?? resolveBundledStaticCatalogModel;
     imageToolProviderDeps.resolveModelAsync = overrides?.resolveModelAsync ?? resolveModelAsync;
+    imageToolProviderDeps.resolveImageCompressionPolicy =
+      overrides?.resolveImageCompressionPolicy ?? resolveImageCompressionPolicy;
     imageToolProviderDeps.loadImageWebMediaRuntime =
       overrides?.loadImageWebMediaRuntime ?? loadImageWebMediaRuntime;
   },
@@ -830,7 +834,7 @@ export function createImageTool(options?: {
           "No image model is configured. Set agents.defaults.imageModel or configure an image-capable provider.",
         );
       }
-      const imageCompression = await resolveImageCompressionPolicy({
+      const imageCompression = await imageToolProviderDeps.resolveImageCompressionPolicy({
         cfg: options?.config,
         imageModelConfig,
         modelOverride,
