diff --git a/extensions/browser/src/browser/server.control-server.test-harness.ts b/extensions/browser/src/browser/server.control-server.test-harness.ts index 310dd7476d4..a0b7c29af38 100644 --- a/extensions/browser/src/browser/server.control-server.test-harness.ts +++ b/extensions/browser/src/browser/server.control-server.test-harness.ts @@ -348,7 +348,6 @@ async function loadBrowserServerModule(): Promise { if (browserServerModule) { return browserServerModule; } - vi.resetModules(); browserServerModule = await import("../server.js"); return browserServerModule; } @@ -484,7 +483,6 @@ export async function startBrowserControlServerFromConfig() { export async function stopBrowserControlServer(): Promise { const server = browserServerModule; - browserServerModule = null; if (!server) { return; } diff --git a/src/agents/models-config.providers.normalize-keys.test.ts b/src/agents/models-config.providers.normalize-keys.test.ts index 1e0e6cb15b6..2a8155a28f4 100644 --- a/src/agents/models-config.providers.normalize-keys.test.ts +++ b/src/agents/models-config.providers.normalize-keys.test.ts @@ -8,14 +8,21 @@ import { normalizeProviders } from "./models-config.providers.normalize.js"; import { resolveApiKeyFromProfiles } from "./models-config.providers.secret-helpers.js"; import { enforceSourceManagedProviderSecrets } from "./models-config.providers.source-managed.js"; -vi.mock("./models-config.providers.policy.runtime.js", async () => { - const { normalizeLmstudioProviderConfig } = await vi.importActual< - typeof import("../plugin-sdk/lmstudio-runtime.js") - >("../plugin-sdk/lmstudio-runtime.js"); +function normalizeLmstudioBaseUrl(baseUrl: string): string { + const trimmed = baseUrl.trim().replace(/\/+$/, ""); + return trimmed.replace(/\/api\/v1$/, "").replace(/\/v1$/, "") + "/v1"; +} + +vi.mock("./models-config.providers.policy.runtime.js", () => { return { applyProviderNativeStreamingUsagePolicy: () => undefined, - normalizeProviderConfigPolicy: (providerKey: string, provider: unknown) => - providerKey === "lmstudio" ? normalizeLmstudioProviderConfig(provider as never) : undefined, + normalizeProviderConfigPolicy: ( + providerKey: string, + provider: { baseUrl?: unknown } | undefined, + ) => + providerKey === "lmstudio" && typeof provider?.baseUrl === "string" + ? { ...provider, baseUrl: normalizeLmstudioBaseUrl(provider.baseUrl) } + : undefined, resolveProviderConfigApiKeyPolicy: () => undefined, }; });