@koda-sl/baker-bridge 0.162.0 → 0.164.0-dev.d9feffe34

package/dist/hono/bridge-job.test.js

@@ -18,7 +18,7 @@ vi.mock("./runtime-files.ts", async (importOriginal) => {
 });
 import { AVATAR_CASTING_TOOL_NAME } from "./avatar-casting.js";
 import { FACE_PICK_QUESTION } from "./avatar-face-question.js";
-import { BridgeJobExecutor, createStallDetector, isContextOverflowRefusal, isControlChannelDeny, parsePlanDecision, remapAnswersToQuestionText, } from "./bridge-job.js";
+import { agentFeature, BridgeJobExecutor, createStallDetector, isContextOverflowRefusal, isControlChannelDeny, parsePlanDecision, remapAnswersToQuestionText, } from "./bridge-job.js";
 import { CONNECTION_INPUT_TOOL_NAME } from "./connection-input.js";
 import { CREDENTIAL_INPUT_TOOL_NAME } from "./credential-input.js";
 import { FLOW_INPUT_TOOL_NAME } from "./flow-input.js";
@@ -1244,6 +1244,9 @@ describe("BridgeJobExecutor", () => {
     });
     it("propagates the inherited process env (OpenRouter routing) to the agent subprocess", async () => {
         vi.stubEnv("ANTHROPIC_AUTH_TOKEN", "sk-or-test-key");
+        vi.stubEnv("BAKER_ENVIRONMENT", "production");
+        vi.stubEnv("BAKER_RUNTIME_COMPANY_ID", "company_env");
+        vi.stubEnv("BAKER_CHAT_ID", "chat_env");
         const queryInputs = [];
         const executor = createExecutor({
             queryInputs,
@@ -1265,8 +1268,25 @@ describe("BridgeJobExecutor", () => {
                 allow_fallbacks: true,
             },
             session_id: "thread_env",
+            user: "agent.turn | company:company_env | chat:chat_env",
+            trace: {
+                trace_name: "agent.turn",
+                generation_name: "agent.turn",
+                feature: "agent.turn",
+                service: "agent",
+                environment: "production",
+                company_id: "company_env",
+                chat_id: "chat_env",
+            },
             plugins: [{ id: "web", enabled: false }],
         });
+        // The same tag as headers, which is what names the app in OpenRouter.
+        expect(env?.ANTHROPIC_CUSTOM_HEADERS).toBe([
+            "HTTP-Referer: https://agent.production.withbaker.com",
+            "X-OpenRouter-Title: Baker agent (production)",
+            "X-Title: Baker agent (production)",
+            "x-session-id: thread_env",
+        ].join("\n"));
     });
     // OpenRouter has no Bedrock endpoint for a per-message effort, so while the CLI
     // sends one every Opus 5.5 request skips Bedrock and bills OpenRouter credits.
@@ -4966,4 +4986,11 @@ describe("BridgeJobExecutor model refusals", () => {
         expect(outcome).toMatchObject({ kind: "completed" });
     });
 });
+describe("agentFeature", () => {
+    it("tells OpenRouter a Session review, a plan and a chat turn apart", () => {
+        expect(agentFeature({ review: {} })).toBe("chatReview.session");
+        expect(agentFeature({ planMode: true })).toBe("agent.plan");
+        expect(agentFeature({})).toBe("agent.turn");
+    });
+});
 //# sourceMappingURL=bridge-job.test.js.map