opensource/kilocode
3
0
Fork 0
mirror of https://github.com/Kilo-Org/kilocode.git synced 2026-09-01 15:32:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(cli): sync processor cost after subagent propagation

Browse Source 
finish-step and cleanup previously re-wrote the parent assistant message with a stale in-memory cost, clobbering the subagent cost written by task.ts during tool execution. Refresh from DB before each write so the propagated cost is preserved across steps. Verified live via side-by-side `kilo serve` vs dev `bun serve` with 2 parallel subagents; parent now shows own LLM + children, matching real spend.
This commit is contained in:
Alex Alecu
2026-04-24 18:05:45 +03:00
committed by Imanol Maiztegui
parent f57479c657
commit d9bed85e78
1 changed files with 16 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/opencode/src/session/processor.ts
+16
View File
@@ -410,6 +410,14 @@ export const layer: Layer.Layer<
})
// kilocode_change end
ctx.assistantMessage.finish = value.finishReason
// kilocode_change start - capture any subagent cost propagated by tool calls during this step (#6321)
const fresh = yield* Effect.sync(() =>
MessageV2.get({ sessionID: ctx.assistantMessage.sessionID, messageID: ctx.assistantMessage.id }),
)
if (fresh.info.role === "assistant" && fresh.info.cost > ctx.assistantMessage.cost) {
ctx.assistantMessage.cost = fresh.info.cost
}
// kilocode_change end
ctx.assistantMessage.cost += usage.cost
ctx.assistantMessage.tokens = usage.tokens
yield* session.updatePart({
@@ -567,6 +575,14 @@ export const layer: Layer.Layer<
ctx.toolcalls = {}
KiloSessionProcessor.guardEmptyToolCalls(ctx.assistantMessage, MessageV2.parts(ctx.assistantMessage.id)) // kilocode_change
ctx.assistantMessage.time.completed = Date.now()
// kilocode_change start - reconcile cost with any subagent propagation written during tool calls (#6321)
const fresh = yield* Effect.sync(() =>
MessageV2.get({ sessionID: ctx.assistantMessage.sessionID, messageID: ctx.assistantMessage.id }),
)
if (fresh.info.role === "assistant" && fresh.info.cost > ctx.assistantMessage.cost) {
ctx.assistantMessage.cost = fresh.info.cost
}
// kilocode_change end
yield* session.updateMessage(ctx.assistantMessage)
})