fix(cli): preserve child task attribution

packages/kilo-gateway/src/api/constants.ts

@@ -65,6 +65,7 @@ export const MODELS_FETCH_TIMEOUT_MS = 10 * 1000
  */
 export const HEADER_ORGANIZATIONID = "X-KILOCODE-ORGANIZATIONID"
 export const HEADER_TASKID = "X-KILOCODE-TASKID"
+export const HEADER_PARENT_TASKID = "X-KILOCODE-PARENT-TASKID"
 export const HEADER_PROJECTID = "X-KILOCODE-PROJECTID"
 export const HEADER_TESTER = "X-KILOCODE-TESTER"
 export const HEADER_EDITORNAME = "X-KILOCODE-EDITORNAME"

packages/kilo-gateway/src/headers.ts

@@ -1,6 +1,7 @@
 import {
   HEADER_ORGANIZATIONID,
   HEADER_TASKID,
+  HEADER_PARENT_TASKID,
   HEADER_PROJECTID,
   HEADER_TESTER,
   HEADER_EDITORNAME,
@@ -21,6 +22,7 @@ import {
  */
 export const X_KILOCODE_ORGANIZATIONID = HEADER_ORGANIZATIONID
 export const X_KILOCODE_TASKID = HEADER_TASKID
+export const X_KILOCODE_PARENT_TASKID = HEADER_PARENT_TASKID
 export const X_KILOCODE_PROJECTID = HEADER_PROJECTID
 export const X_KILOCODE_TESTER = HEADER_TESTER
 export const X_KILOCODE_EDITORNAME = HEADER_EDITORNAME

packages/kilo-gateway/src/index.ts

@@ -100,6 +100,7 @@ export {
   MODELS_FETCH_TIMEOUT_MS,
   HEADER_ORGANIZATIONID,
   HEADER_TASKID,
+  HEADER_PARENT_TASKID,
   HEADER_PROJECTID,
   HEADER_TESTER,
   HEADER_EDITORNAME,

packages/opencode/src/session/llm.ts

@@ -22,7 +22,13 @@ import { Auth } from "@/auth"
 // kilocode_change start
 import { DEFAULT_HEADERS } from "@/kilocode/const"
 import { getKiloProjectId } from "@/kilocode/project-id"
-import { HEADER_FEATURE, HEADER_PROJECTID, HEADER_MACHINEID, HEADER_TASKID } from "@kilocode/kilo-gateway"
+import {
+  HEADER_FEATURE,
+  HEADER_PARENT_TASKID,
+  HEADER_PROJECTID,
+  HEADER_MACHINEID,
+  HEADER_TASKID,
+} from "@kilocode/kilo-gateway"
 import { Identity } from "@kilocode/kilo-telemetry"
 import { makeRuntime } from "@/effect/run-service"
 import { KiloSession } from "@/kilocode/session"
@@ -421,7 +427,8 @@ const live: Layer.Layer<
           ...(isKilo && input.agent.name ? { "x-kilocode-mode": input.agent.name.toLowerCase() } : {}),
           ...(isKilo && kiloProjectId ? { [HEADER_PROJECTID]: kiloProjectId } : {}),
           ...(isKilo && machineId ? { [HEADER_MACHINEID]: machineId } : {}),
-          ...(isKilo ? { [HEADER_TASKID]: attr.rootID } : {}),
+          ...(isKilo ? { [HEADER_TASKID]: input.sessionID } : {}),
+          ...(isKilo && input.parentSessionID ? { [HEADER_PARENT_TASKID]: input.parentSessionID } : {}),
           ...(isKilo && attr.feature ? { [HEADER_FEATURE]: attr.feature } : {}),
           // kilocode_change end
           ...input.model.headers,