From ebb10c08522af185c82b4c30532698d22292be2c Mon Sep 17 00:00:00 2001
From: Vincent Koc <vincentkoc@ieee.org>
Date: Wed, 18 Mar 2026 08:46:58 -0700
Subject: [PATCH] Contracts: fix codex catalog hint assertion

---
 src/plugins/contracts/catalog.contract.test.ts | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/src/plugins/contracts/catalog.contract.test.ts b/src/plugins/contracts/catalog.contract.test.ts
index 146c8b99b78..b564cbf8664 100644
--- a/src/plugins/contracts/catalog.contract.test.ts
+++ b/src/plugins/contracts/catalog.contract.test.ts
@@ -4,6 +4,7 @@ import {
   expectCodexBuiltInSuppression,
   expectCodexMissingAuthHint,
 } from "../provider-runtime.test-support.js";
+import { requireProviderContractProvider } from "./registry.js";
 
 type ResolvePluginProviders = typeof import("../providers.js").resolvePluginProviders;
 type ResolveOwningPluginIdsForProvider =
@@ -30,7 +31,6 @@ vi.mock("../providers.js", () => ({
 }));
 
 let augmentModelCatalogWithProviderPlugins: typeof import("../provider-runtime.js").augmentModelCatalogWithProviderPlugins;
-let buildProviderMissingAuthMessageWithPlugin: typeof import("../provider-runtime.js").buildProviderMissingAuthMessageWithPlugin;
 let resetProviderRuntimeHookCacheForTest: typeof import("../provider-runtime.js").resetProviderRuntimeHookCacheForTest;
 let resolveProviderBuiltInModelSuppression: typeof import("../provider-runtime.js").resolveProviderBuiltInModelSuppression;
 let resolveProviderContractPluginIdsForProvider: typeof import("./registry.js").resolveProviderContractPluginIdsForProvider;
@@ -46,7 +46,6 @@ describe("provider catalog contract", () => {
     } = await import("./registry.js"));
     ({
       augmentModelCatalogWithProviderPlugins,
-      buildProviderMissingAuthMessageWithPlugin,
       resetProviderRuntimeHookCacheForTest,
       resolveProviderBuiltInModelSuppression,
     } = await import("../provider-runtime.js"));
@@ -74,7 +73,10 @@ describe("provider catalog contract", () => {
   });
 
   it("keeps codex-only missing-auth hints wired through the provider runtime", () => {
-    expectCodexMissingAuthHint(buildProviderMissingAuthMessageWithPlugin);
+    const openaiProvider = requireProviderContractProvider("openai");
+    expectCodexMissingAuthHint((params) =>
+      openaiProvider.buildMissingAuthMessage?.(params.context),
+    );
   });
 
   it("keeps built-in model suppression wired through the provider runtime", () => {