opencode-kilocode-auth 1.0.9 → 1.0.10

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "opencode-kilocode-auth",
   "module": "index.ts",
-  "version": "1.0.9",
+  "version": "1.0.10",
   "author": "ported from Kilo Code",
   "description": "OpenCode plugin for Kilo Code authentication with support for various models including Giga Potato",
   "files": [

package/src/plugin.ts

@@ -3,6 +3,7 @@ import {
   KILOCODE_PROVIDER_ID,
   DEFAULT_MODEL_ID,
   DEVICE_AUTH_POLL_INTERVAL_MS,
+  KILOCODE_VERSION,
 } from "./constants"
 import {
   initiateDeviceAuth,
@@ -24,6 +25,59 @@ interface KilocodeAuth {
   key?: string // for api type (OpenCode uses "key" field)
 }
 
+/**
+ * OpenAI SDK version used by Kilo Code VSCode extension
+ * From: kilocode/src/package.json -> "openai": "^5.12.2"
+ */
+const OPENAI_SDK_VERSION = "0.51.0"
+
+/**
+ * Get X-Stainless-* headers (EXACT match from OpenAI SDK)
+ * These are automatically added by the OpenAI SDK and MUST be included
+ */
+function getStainlessHeaders(): Record<string, string> {
+  const platform = process.platform
+  const arch = process.arch
+  const nodeVersion = process.version
+
+  // Map platform to OS name (EXACT from OpenAI SDK)
+  const osMap: Record<string, string> = {
+    darwin: "MacOS",
+    win32: "Windows",
+    linux: "Linux",
+    freebsd: "FreeBSD",
+    openbsd: "OpenBSD",
+  }
+  const os = osMap[platform] || `Other:${platform}`
+
+  // Map arch (EXACT from OpenAI SDK)
+  const archMap: Record<string, string> = {
+    x32: "x32",
+    x64: "x64",
+    arm: "arm",
+    arm64: "arm64",
+  }
+  const archStr = archMap[arch] || `other:${arch}`
+
+  return {
+    "X-Stainless-Lang": "js",
+    "X-Stainless-Package-Version": OPENAI_SDK_VERSION,
+    "X-Stainless-OS": os,
+    "X-Stainless-Arch": archStr,
+    "X-Stainless-Runtime": "node",
+    "X-Stainless-Runtime-Version": nodeVersion,
+  }
+}
+
+/**
+ * Get editor name header (EXACT match from Kilo Code VSCode extension)
+ * From: kilocode/src/core/kilocode/wrapper.ts -> getEditorNameHeader()
+ */
+function getEditorNameHeader(): string {
+  // Exact format: [appName, version].join(" ")
+  return "Visual Studio Code 1.96.0"
+}
+
 /**
  * Kilo Code Authentication Plugin for OpenCode
  *
@@ -87,8 +141,10 @@ export async function KilocodeAuthPlugin(input: PluginInput): Promise<Hooks> {
               return fetch(requestInput, init)
             }
 
-            // Build headers
+            // Build headers - EXACT match from Kilo Code VSCode extension
             const headers = new Headers()
+
+            // First, copy any existing headers from the request
             if (init?.headers) {
               if (init.headers instanceof Headers) {
                 init.headers.forEach((value, key) => headers.set(key, value))
@@ -103,24 +159,38 @@ export async function KilocodeAuthPlugin(input: PluginInput): Promise<Hooks> {
               }
             }
 
-            // EXACT headers from Kilo Code VSCode extension (DEFAULT_HEADERS + customRequestOptions)
-            // Authorization is set by OpenAI SDK internally
+            // ============================================================
+            // EXACT HEADERS FROM KILO CODE VSCODE EXTENSION (decompiled)
+            // ============================================================
+
+            // 1. OpenAI SDK standard headers
             headers.set("Authorization", `Bearer ${currentToken}`)
+            headers.set("Content-Type", "application/json")
 
-            // DEFAULT_HEADERS from kilocode/src/api/providers/constants.ts
+            // 2. DEFAULT_HEADERS from kilocode/src/api/providers/constants.ts
+            // EXACT: vl = {"HTTP-Referer":"https://kilocode.ai","X-Title":"Kilo Code",[sFa]:Ca.version,"User-Agent":`Kilo-Code/${Ca.version}`}
             headers.set("HTTP-Referer", "https://kilocode.ai")
             headers.set("X-Title", "Kilo Code")
-            headers.set("X-KiloCode-Version", "4.151.0")
-            headers.set("User-Agent", "Kilo-Code/4.151.0")
+            headers.set("X-KiloCode-Version", KILOCODE_VERSION)
+            headers.set("User-Agent", `Kilo-Code/${KILOCODE_VERSION}`)
 
-            // customRequestOptions from KilocodeOpenrouterHandler
-            headers.set("X-KiloCode-EditorName", "Visual Studio Code 1.96.0")
+            // 3. customRequestOptions headers from KilocodeOpenrouterHandler
+            // EXACT: lFa = "X-KiloCode-EditorName"
+            headers.set("X-KiloCode-EditorName", getEditorNameHeader())
 
-            // Organization header (exact casing from headers.ts)
+            // 4. Organization header (if present)
+            // EXACT: dAe = "X-KiloCode-OrganizationId"
             if (currentKilocodeAuth.organizationId) {
               headers.set("X-KiloCode-OrganizationId", currentKilocodeAuth.organizationId)
             }
 
+            // 5. X-Stainless-* headers (AUTOMATICALLY added by OpenAI SDK)
+            // These are CRITICAL - the server may check for these
+            const stainlessHeaders = getStainlessHeaders()
+            for (const [key, value] of Object.entries(stainlessHeaders)) {
+              headers.set(key, value)
+            }
+
             // Rewrite URL to Kilo Code endpoint
             const parsed =
               requestInput instanceof URL
@@ -128,7 +198,6 @@ export async function KilocodeAuthPlugin(input: PluginInput): Promise<Hooks> {
                 : new URL(typeof requestInput === "string" ? requestInput : (requestInput as Request).url)
 
             // Map to Kilo Code OpenRouter-compatible endpoint
-            // Kilo Code uses: https://api.kilo.ai/api/openrouter/chat/completions
             let url: URL
             let body = init?.body
 
@@ -140,11 +209,18 @@ export async function KilocodeAuthPlugin(input: PluginInput): Promise<Hooks> {
                 try {
                   const payload = JSON.parse(body)
 
-                  // Add stream_options like Kilo Code does (ALWAYS included)
+                  // Add stream_options like Kilo Code does
+                  // EXACT: stream_options:{include_usage:!0}
                   if (payload.stream !== false) {
                     payload.stream_options = { include_usage: true }
                   }
 
+                  // 6. For Anthropic models, add x-anthropic-beta header
+                  // EXACT: o.startsWith("anthropic/")&&(v.headers["x-anthropic-beta"]="fine-grained-tool-streaming-2025-05-14")
+                  if (payload.model && payload.model.startsWith("anthropic/")) {
+                    headers.set("x-anthropic-beta", "fine-grained-tool-streaming-2025-05-14")
+                  }
+
                   if (payload.messages && Array.isArray(payload.messages)) {
                     // Find system message and prepend Kilo Code role
                     const systemIdx = payload.messages.findIndex((m: any) => m.role === "system")