opencode-kilocode-auth 1.0.8 → 1.0.10

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "opencode-kilocode-auth",
   "module": "index.ts",
-  "version": "1.0.8",
+  "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/constants.ts

@@ -17,21 +17,31 @@ export const KILOCODE_API_BASE_URL = "https://api.kilo.ai";
  */
 export const KILOCODE_PROVIDER_ID = "kilocode";
 
+/**
+ * Kilo Code extension version (must match official extension)
+ */
+export const KILOCODE_VERSION = "4.151.0";
+
 /**
  * Default headers for Kilo Code API requests
+ * EXACT match from kilocode/src/api/providers/constants.ts DEFAULT_HEADERS
  */
 export const DEFAULT_HEADERS = {
   "Content-Type": "application/json",
-  "User-Agent": "opencode-kilocode-auth/1.0.0",
+  "HTTP-Referer": "https://kilocode.ai",
+  "X-Title": "Kilo Code",
+  "X-KiloCode-Version": KILOCODE_VERSION,
+  "User-Agent": `Kilo-Code/${KILOCODE_VERSION}`,
 } as const;
 
 /**
- * Kilo Code custom headers
+ * Kilo Code custom headers (EXACT casing from kilocode/src/shared/kilocode/headers.ts)
  */
-export const X_KILOCODE_ORGANIZATIONID = "X-KILOCODE-ORGANIZATIONID";
-export const X_KILOCODE_TASKID = "X-KILOCODE-TASKID";
-export const X_KILOCODE_PROJECTID = "X-KILOCODE-PROJECTID";
-export const X_KILOCODE_EDITORNAME = "X-KILOCODE-EDITORNAME";
+export const X_KILOCODE_VERSION = "X-KiloCode-Version";
+export const X_KILOCODE_ORGANIZATIONID = "X-KiloCode-OrganizationId";
+export const X_KILOCODE_TASKID = "X-KiloCode-TaskId";
+export const X_KILOCODE_PROJECTID = "X-KiloCode-ProjectId";
+export const X_KILOCODE_EDITORNAME = "X-KiloCode-EditorName";
 
 /**
  * Default model ID if none is specified

package/src/kilocode/auth.ts

@@ -4,6 +4,7 @@ import {
   DEFAULT_HEADERS,
   DEVICE_AUTH_POLL_INTERVAL_MS,
   DEFAULT_MODEL_ID,
+  X_KILOCODE_EDITORNAME,
 } from "../constants";
 import type {
   DeviceAuthInitiateResponse,
@@ -82,6 +83,13 @@ export async function pollDeviceAuth(code: string): Promise<DeviceAuthPollRespon
   return (await response.json()) as DeviceAuthPollResponse;
 }
 
+/**
+ * Get editor name header value (matches Kilo Code VSCode extension)
+ */
+function getEditorNameHeader(): string {
+  return "Visual Studio Code 1.96.0";
+}
+
 /**
  * Fetch user profile from Kilo Code API
  */
@@ -90,6 +98,7 @@ export async function getKilocodeProfile(token: string): Promise<KilocodeProfile
     headers: {
       ...DEFAULT_HEADERS,
       Authorization: `Bearer ${token}`,
+      [X_KILOCODE_EDITORNAME]: getEditorNameHeader(),
     },
   });
 
@@ -119,6 +128,7 @@ export async function getKilocodeDefaultModel(
       headers: {
         ...DEFAULT_HEADERS,
         Authorization: `Bearer ${token}`,
+        [X_KILOCODE_EDITORNAME]: getEditorNameHeader(),
       },
     });
 
@@ -146,10 +156,11 @@ export async function getKilocodeModels(
     const headers: Record<string, string> = {
       ...DEFAULT_HEADERS,
       Authorization: `Bearer ${token}`,
+      [X_KILOCODE_EDITORNAME]: getEditorNameHeader(),
     };
 
     if (organizationId) {
-      headers["X-KILOCODE-ORGANIZATIONID"] = organizationId;
+      headers["X-KiloCode-OrganizationId"] = organizationId;
     }
 
     const response = await fetch(baseUrl, { headers });

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,17 +198,29 @@ 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
 
             if (parsed.pathname.includes("/chat/completions")) {
               url = new URL(getApiUrl("/api/openrouter/chat/completions", currentToken))
 
-              // Modify request body to include Kilo Code system prompt prefix
+              // Modify request body to match Kilo Code format EXACTLY
               if (body && typeof body === "string") {
                 try {
                   const payload = JSON.parse(body)
+
+                  // 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")
@@ -151,8 +233,9 @@ export async function KilocodeAuthPlugin(input: PluginInput): Promise<Hooks> {
                       // Add system message at the beginning
                       payload.messages.unshift({ role: "system", content: kiloCodeRole })
                     }
-                    body = JSON.stringify(payload)
                   }
+
+                  body = JSON.stringify(payload)
                 } catch {
                   // Keep original body if parsing fails
                 }