refactor: improve X-Initiator agent detection logic

sachnun · sachnun · commit 9ec54ae18619 · 2025-10-20T05:22:59.000+07:00
Extends agent detection to use pattern matching for better coverage of AI assistant models
diff --git a/src/services/copilot/create-chat-completions.ts b/src/services/copilot/create-chat-completions.ts
@@ -22,13 +22,13 @@ export const createChatCompletions = async (
     ["assistant", "tool"].includes(msg.role),
   )
 
-  // Check if model is claude-sonnet-4.5 (should also be treated as agent)
-  const isClaudeSonnet45 = payload.model === "claude-sonnet-4.5"
+  // Check if model starts with "claude-" (should also be treated as agent)
+  const isClaudeModel = payload.model.startsWith("claude-")
 
   // Build headers and add X-Initiator
   const headers: Record<string, string> = {
     ...copilotHeaders(state, enableVision),
-    "X-Initiator": isAgentCall || isClaudeSonnet45 ? "agent" : "user",
+    "X-Initiator": isAgentCall || isClaudeModel ? "agent" : "user",
   }
 
   const response = await fetch(`${copilotBaseUrl(state)}/chat/completions`, {

Original file line number	Diff line number	Diff line change
`@@ -22,13 +22,13 @@ export const createChatCompletions = async (`
`22`	`22`	`["assistant", "tool"].includes(msg.role),`
`23`	`23`	`)`
`24`	`24`
`25`		`- // Check if model is claude-sonnet-4.5 (should also be treated as agent)`
`26`		`- const isClaudeSonnet45 = payload.model === "claude-sonnet-4.5"`
	`25`	`+ // Check if model starts with "claude-" (should also be treated as agent)`
	`26`	`+ const isClaudeModel = payload.model.startsWith("claude-")`
`27`	`27`
`28`	`28`	`// Build headers and add X-Initiator`
`29`	`29`	`const headers: Record<string, string> = {`
`30`	`30`	`...copilotHeaders(state, enableVision),`
`31`		`- "X-Initiator": isAgentCall \|\| isClaudeSonnet45 ? "agent" : "user",`
	`31`	`+ "X-Initiator": isAgentCall \|\| isClaudeModel ? "agent" : "user",`
`32`	`32`	`}`
`33`	`33`
`34`	`34`	const response = await fetch(`${copilotBaseUrl(state)}/chat/completions`, {