@stablekernel/opencode-bifrost 0.1.4 → 0.2.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@stablekernel/opencode-bifrost",
-  "version": "0.1.4",
+  "version": "0.2.1",
   "description": "OpenCode plugin that routes models through the Bifrost gateway, resolving per-project virtual keys via gateway-cli.",
   "type": "module",
   "main": "src/index.ts",

package/src/index.test.ts

@@ -153,6 +153,27 @@ describe("config hook", () => {
     ).toEqual({ input: 1, output: 5, cache_read: 0.1, cache_write: 1.25 });
   });
 
+  test("omits headers when resolve reports none", async () => {
+    const noHeaders: Record<string, unknown> = { ...RESOLUTION };
+    delete noHeaders.headers;
+    const cfg = await applyConfig({
+      ...OK_SHELL,
+      "resolve --json": JSON.stringify(noHeaders),
+    });
+    expect(cfg.provider.bifrost.options.headers).toBeUndefined();
+  });
+
+  test("treats an empty header map the same as no headers", async () => {
+    // Go-side mergeHeaders returns nil for an empty merge and the JSON tag
+    // omits it, so `{}` cannot reach the plugin in practice. This pins the
+    // behavior anyway so a future CLI change cannot quietly flip it.
+    const cfg = await applyConfig({
+      ...OK_SHELL,
+      "resolve --json": JSON.stringify({ ...RESOLUTION, headers: {} }),
+    });
+    expect(cfg.provider.bifrost.options.headers).toBeUndefined();
+  });
+
   test("does not read the key: discovery goes through the CLI", async () => {
     // The CLI holds the credential and returns only non-secret metadata.
     const calls: string[] = [];
@@ -319,6 +340,17 @@ describe("auth.loader", () => {
     expect(opts.baseURL).toBe("https://gw.example/openai/v1");
   });
 
+  test("forwards the resolved headers alongside the key", async () => {
+    // The config hook sets headers at startup, but auth.loader also returns
+    // provider options at request time; both must carry the gateway headers.
+    const hooks = await load(OK_SHELL);
+    const opts = await hooks.auth!.loader!(
+      async () => ({ type: "api", key: "stored" }) as never,
+      {} as never,
+    );
+    expect(opts.headers).toEqual({ "X-Org-Route": "prod" });
+  });
+
   test("falls back to the stored credential without gateway-cli", async () => {
     const hooks = await load({});
     const opts = await hooks.auth!.loader!(

package/src/index.ts

@@ -150,7 +150,9 @@ export const BifrostGateway: Plugin = async ({ directory, $ }) => {
         ...existing,
         options: {
           baseURL: info.openai_base_url,
-          ...(info.headers ? { headers: info.headers } : {}),
+          ...(info.headers && Object.keys(info.headers).length > 0
+            ? { headers: info.headers }
+            : {}),
           ...(existing.options ?? {}),
         },
         models: { ...models, ...(existing.models ?? {}) },
@@ -176,7 +178,9 @@ export const BifrostGateway: Plugin = async ({ directory, $ }) => {
         return {
           apiKey,
           ...(info ? { baseURL: info.openai_base_url } : {}),
-          ...(info?.headers ? { headers: info.headers } : {}),
+          ...(info?.headers && Object.keys(info.headers).length > 0
+            ? { headers: info.headers }
+            : {}),
         };
       },