opencode-kilocode-auth 1.0.6 → 1.0.8

package/package.json

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

@@ -103,18 +103,22 @@ export async function KilocodeAuthPlugin(input: PluginInput): Promise<Hooks> {
               }
             }
 
-            // Set Kilo Code authorization and required headers
+            // EXACT headers from Kilo Code VSCode extension (DEFAULT_HEADERS + customRequestOptions)
+            // Authorization is set by OpenAI SDK internally
             headers.set("Authorization", `Bearer ${currentToken}`)
-            headers.set("x-api-key", currentToken) // Kilo Code uses both
+
+            // DEFAULT_HEADERS from kilocode/src/api/providers/constants.ts
             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")
+
+            // customRequestOptions from KilocodeOpenrouterHandler
             headers.set("X-KiloCode-EditorName", "Visual Studio Code 1.96.0")
 
-            // Add organization header if present
+            // Organization header (exact casing from headers.ts)
             if (currentKilocodeAuth.organizationId) {
-              headers.set("X-KILOCODE-ORGANIZATIONID", currentKilocodeAuth.organizationId)
+              headers.set("X-KiloCode-OrganizationId", currentKilocodeAuth.organizationId)
             }
 
             // Rewrite URL to Kilo Code endpoint
@@ -126,8 +130,33 @@ export async function KilocodeAuthPlugin(input: PluginInput): Promise<Hooks> {
             // 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
+              if (body && typeof body === "string") {
+                try {
+                  const payload = JSON.parse(body)
+                  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")
+                    const kiloCodeRole = "You are Kilo Code, a highly skilled software engineer with extensive knowledge in many programming languages, frameworks, design patterns, and best practices."
+
+                    if (systemIdx >= 0) {
+                      // Prepend Kilo Code role to existing system message
+                      payload.messages[systemIdx].content = kiloCodeRole + "\n\n" + payload.messages[systemIdx].content
+                    } else {
+                      // Add system message at the beginning
+                      payload.messages.unshift({ role: "system", content: kiloCodeRole })
+                    }
+                    body = JSON.stringify(payload)
+                  }
+                } catch {
+                  // Keep original body if parsing fails
+                }
+              }
             } else if (parsed.pathname.includes("/models")) {
               url = new URL(getApiUrl("/api/openrouter/models", currentToken))
             } else {
@@ -137,6 +166,7 @@ export async function KilocodeAuthPlugin(input: PluginInput): Promise<Hooks> {
             return fetch(url, {
               ...init,
               headers,
+              body,
             })
           },
         }
@@ -216,13 +246,6 @@ export async function KilocodeAuthPlugin(input: PluginInput): Promise<Hooks> {
       ],
     },
 
-    // Add custom headers for Kilo Code requests
-    "chat.headers": async (input, output) => {
-      if (input.model.providerID !== KILOCODE_PROVIDER_ID) return
-
-      output.headers["X-KILOCODE-EDITORNAME"] = "opencode"
-      output.headers["X-KILOCODE-SESSIONID"] = input.sessionID
-    },
   }
 }