[Init] Initial commit - NetMesh terminal manager

infrastructure/ai/modelDiscoveryHeaders.test.ts

@@ -0,0 +1,59 @@
+import assert from "node:assert/strict";
+import test from "node:test";
+
+import { buildModelDiscoveryHeaders, resolveModelsDiscoveryEndpoint } from "./modelDiscoveryHeaders";
+
+test("buildModelDiscoveryHeaders uses x-api-key+anthropic-version for the anthropic family", () => {
+  assert.deepEqual(buildModelDiscoveryHeaders("anthropic", "sk-test"), {
+    "x-api-key": "sk-test",
+    "anthropic-version": "2023-06-01",
+  });
+});
+
+test("buildModelDiscoveryHeaders uses Bearer auth for the openai-compatible family", () => {
+  assert.deepEqual(buildModelDiscoveryHeaders("openai", "sk-test"), {
+    Authorization: "Bearer sk-test",
+  });
+});
+
+test("buildModelDiscoveryHeaders uses x-goog-api-key for the google family", () => {
+  // Google Generative AI rejects Bearer auth — discovery has to match
+  // the createGoogle runtime client, which uses x-goog-api-key.
+  assert.deepEqual(buildModelDiscoveryHeaders("google", "AIza-test"), {
+    "x-goog-api-key": "AIza-test",
+  });
+});
+
+test("buildModelDiscoveryHeaders returns no headers when the api key is missing", () => {
+  assert.deepEqual(buildModelDiscoveryHeaders("anthropic", undefined), {});
+  assert.deepEqual(buildModelDiscoveryHeaders("openai", ""), {});
+});
+
+test("buildModelDiscoveryHeaders honors the style override on an anthropic providerId pointing at an OpenAI-compatible backend", () => {
+  // Regression: PR #1105 lets users pick `style` independently from
+  // `providerId`. Without this fix the discovery call still sent
+  // `x-api-key` because the old code switched on `providerId === "anthropic"`.
+  assert.deepEqual(buildModelDiscoveryHeaders("openai", "sk-test"), {
+    Authorization: "Bearer sk-test",
+  });
+});
+
+test("resolveModelsDiscoveryEndpoint follows the resolved style by default", () => {
+  assert.equal(resolveModelsDiscoveryEndpoint("openai"), "/models");
+  assert.equal(resolveModelsDiscoveryEndpoint("anthropic"), "/v1/models");
+  assert.equal(resolveModelsDiscoveryEndpoint("google"), undefined);
+});
+
+test("resolveModelsDiscoveryEndpoint overrides the preset path when style flips", () => {
+  // Anthropic providerId preset would otherwise pin /v1/models, but the user
+  // switched style to openai — pick /models instead so the path matches the
+  // protocol family the headers already speak.
+  assert.equal(resolveModelsDiscoveryEndpoint("openai", "/v1/models"), "/models");
+  assert.equal(resolveModelsDiscoveryEndpoint("anthropic", "/models"), "/v1/models");
+});
+
+test("resolveModelsDiscoveryEndpoint falls back to the preset path only when the style has no convention", () => {
+  // google has no STYLE_DEFAULT — preserve whatever the caller passed.
+  assert.equal(resolveModelsDiscoveryEndpoint("google", "/custom/list"), "/custom/list");
+  assert.equal(resolveModelsDiscoveryEndpoint("google", undefined), undefined);
+});