feat: add agent/user X-Initiator header to chat completions and tests. fixes ericc-ch#68

- Refactor header logic in createChatCompletions to set X-Initiator
based on message roles
- Add Bun test to verify agent/user header logic
- Lint and type fixes for strict TypeScript

🤖 Generated with [opencode](https://opencode.ai)

Co-Authored-By: opencode <noreply@opencode.ai>

Author: ericc-ch

src/services/copilot/create-chat-completions.ts

@@ -16,9 +16,21 @@ export const createChatCompletions = async (
       && x.content?.some((x) => x.type === "image_url"),
   )
 
+  // Agent/user check for X-Initiator header
+  // Determine if any message is from an agent ("assistant" or "tool")
+  const isAgentCall = payload.messages.some((msg) =>
+    ["assistant", "tool"].includes(msg.role),
+  )
+
+  // Build headers and add X-Initiator
+  const headers: Record<string, string> = {
+    ...copilotHeaders(state, enableVision),
+    "X-Initiator": isAgentCall ? "agent" : "user",
+  }
+
   const response = await fetch(`${copilotBaseUrl(state)}/chat/completions`, {
     method: "POST",
-    headers: copilotHeaders(state, enableVision),
+    headers,
     body: JSON.stringify(payload),
   })
 

tests/create-chat-completions.test.ts

@@ -0,0 +1,55 @@
+import { test, expect, mock } from "bun:test"
+
+import type { ChatCompletionsPayload } from "../src/services/copilot/create-chat-completions"
+
+import { state } from "../src/lib/state"
+import { createChatCompletions } from "../src/services/copilot/create-chat-completions"
+
+// Mock state
+state.copilotToken = "test-token"
+state.vsCodeVersion = "1.0.0"
+state.accountType = "individual"
+
+// Helper to mock fetch
+const fetchMock = mock(
+  (_url: string, opts: { headers: Record<string, string> }) => {
+    return {
+      ok: true,
+      json: () => ({ id: "123", object: "chat.completion", choices: [] }),
+      headers: opts.headers,
+    }
+  },
+)
+;(globalThis as unknown as { fetch: typeof fetch }).fetch = fetchMock
+
+test("sets X-Initiator to agent if tool/assistant present", async () => {
+  const payload: ChatCompletionsPayload = {
+    messages: [
+      { role: "user", content: "hi" },
+      { role: "tool", content: "tool call" },
+    ],
+    model: "gpt-test",
+  }
+  await createChatCompletions(payload)
+  expect(fetchMock).toHaveBeenCalled()
+  const headers = (
+    fetchMock.mock.calls[0][1] as { headers: Record<string, string> }
+  ).headers
+  expect(headers["X-Initiator"]).toBe("agent")
+})
+
+test("sets X-Initiator to user if only user present", async () => {
+  const payload: ChatCompletionsPayload = {
+    messages: [
+      { role: "user", content: "hi" },
+      { role: "user", content: "hello again" },
+    ],
+    model: "gpt-test",
+  }
+  await createChatCompletions(payload)
+  expect(fetchMock).toHaveBeenCalled()
+  const headers = (
+    fetchMock.mock.calls[1][1] as { headers: Record<string, string> }
+  ).headers
+  expect(headers["X-Initiator"]).toBe("user")
+})