diff --git a/src/agents/models-config.providers.policy.test.ts b/src/agents/models-config.providers.policy.test.ts index 1d683f4dd7a..f0056432903 100644 --- a/src/agents/models-config.providers.policy.test.ts +++ b/src/agents/models-config.providers.policy.test.ts @@ -1,13 +1,4 @@ -import { beforeEach, describe, expect, it, vi } from "vitest"; - -type NormalizeProviderSpecificConfig = - typeof import("./models-config.providers.policy.js").normalizeProviderSpecificConfig; -type ResolveProviderConfigApiKeyResolver = - typeof import("./models-config.providers.policy.js").resolveProviderConfigApiKeyResolver; - -const GOOGLE_BASE_URL = "https://generativelanguage.googleapis.com"; -let normalizeProviderSpecificConfig: NormalizeProviderSpecificConfig; -let resolveProviderConfigApiKeyResolver: ResolveProviderConfigApiKeyResolver; +import { describe, expect, it, vi } from "vitest"; vi.mock("../plugins/provider-runtime.js", () => ({ applyProviderNativeStreamingUsageCompatWithPlugin: () => undefined, @@ -46,11 +37,12 @@ vi.mock("../plugins/provider-runtime.js", () => ({ }, })); -beforeEach(async () => { - vi.resetModules(); - ({ normalizeProviderSpecificConfig, resolveProviderConfigApiKeyResolver } = - await import("./models-config.providers.policy.js")); -}); +import { + normalizeProviderSpecificConfig, + resolveProviderConfigApiKeyResolver, +} from "./models-config.providers.policy.js"; + +const GOOGLE_BASE_URL = "https://generativelanguage.googleapis.com"; describe("models-config.providers.policy", () => { it("resolves config apiKey markers through provider plugin hooks", () => {