opencode-kilocode-auth 1.0.9 → 1.0.11

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "opencode-kilocode-auth",
   "module": "index.ts",
-  "version": "1.0.9",
+  "version": "1.0.11",
   "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,22 +209,68 @@ 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
+                    // Find system message and prepend Kilo Code system prompt
                     const systemIdx = payload.messages.findIndex((m: any) => m.role === "system")
-                    const kiloCodeRole = "You are Kilo Code, a highly skilled software engineer with extensive knowledge in many programming languages, frameworks, design patterns, and best practices."
+
+                    // EXACT Kilo Code system prompt structure (without tool definitions - OpenCode handles those)
+                    const kiloCodeSystemPrompt = `You are Kilo Code, a highly skilled software engineer with extensive knowledge in many programming languages, frameworks, design patterns, and best practices.
+
+====
+
+MARKDOWN RULES
+
+ALL responses MUST show ANY \`language construct\` OR filename reference as clickable, exactly as [\`filename OR language.declaration()\`](relative/file/path.ext:line); line is required for \`syntax\` and optional for filename links. This applies to ALL markdown responses.
+
+====
+
+SYSTEM INFORMATION
+
+Operating System: ${process.platform === 'win32' ? 'Windows' : process.platform === 'darwin' ? 'MacOS' : 'Linux'}
+Default Shell: ${process.env.SHELL || process.env.COMSPEC || '/bin/bash'}
+Home Directory: ${process.env.HOME || process.env.USERPROFILE || '~'}
+
+====
+
+OBJECTIVE
+
+You accomplish a given task iteratively, breaking it down into clear steps and working through them methodically.
+
+1. Analyze the user's task and set clear, achievable goals to accomplish it. Prioritize these goals in a logical order.
+2. Work through these goals sequentially, utilizing available tools one at a time as necessary. Each goal should correspond to a distinct step in your problem-solving process. You will be informed on the work completed and what's remaining as you go.
+3. Once you've completed the user's task, present the result of the task to the user.
+4. The user may provide feedback, which you can use to make improvements and try again. But DO NOT continue in pointless back and forth conversations, i.e. don't end your responses with questions or offers for further assistance.
+
+====
+
+RULES
+
+- Be concise, direct, and to the point.
+- Output text to communicate with the user; all text you output outside of tool use is displayed to the user.
+- Only use emojis if the user explicitly requests it.
+- When making changes to files, first understand the file's code conventions. Mimic code style, use existing libraries and utilities, and follow existing patterns.
+- NEVER assume that a given library is available, even if it is well known. Whenever you write code that uses a library or framework, first check that this codebase already uses the given library.
+- Always follow security best practices. Never introduce code that exposes or logs secrets and keys.
+- NEVER commit changes unless the user explicitly asks you to.`
 
                     if (systemIdx >= 0) {
-                      // Prepend Kilo Code role to existing system message
-                      payload.messages[systemIdx].content = kiloCodeRole + "\n\n" + payload.messages[systemIdx].content
+                      // Prepend Kilo Code prompt to existing system message
+                      payload.messages[systemIdx].content = kiloCodeSystemPrompt + "\n\n====\n\nADDITIONAL INSTRUCTIONS\n\n" + payload.messages[systemIdx].content
                     } else {
                       // Add system message at the beginning
-                      payload.messages.unshift({ role: "system", content: kiloCodeRole })
+                      payload.messages.unshift({ role: "system", content: kiloCodeSystemPrompt })
                     }
                   }