opencode-kilocode-auth 1.0.8 → 1.0.9

package/package.json

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

@@ -135,10 +135,16 @@ export async function KilocodeAuthPlugin(input: PluginInput): Promise<Hooks> {
             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 (ALWAYS included)
+                  if (payload.stream !== false) {
+                    payload.stream_options = { include_usage: true }
+                  }
+
                   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 +157,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
                 }