blun-king-cli 9.1.243 → 9.1.244

package/bin/loop-event-record-policy.cjs

@@ -11,6 +11,10 @@ const STEP_END_LIVE_ONLY_FIELDS = [
 ];
 const STEP_END_ROUTINE_FINISH_REASONS = new Set(['tool_use', 'end_turn']);
 
+function usageTotal(usage) {
+	return usage.inputCacheRead + usage.inputCacheCreation + usage.inputOther + usage.output;
+}
+
 function hasOwn(object, key) {
 	return Object.prototype.hasOwnProperty.call(object, key);
 }
@@ -56,11 +60,13 @@ function projectLoopEventForRecord(event) {
 	}
 
 	const hasRoutineFinishReason = STEP_END_ROUTINE_FINISH_REASONS.has(event?.finishReason);
+	const hasUsage = event?.type === 'step.end' && hasOwn(event, 'usage');
 	if (
 		event?.type !== 'step.end' ||
 		(
 			!hasOwn(event, 'turnId') &&
 			!hasOwn(event, 'step') &&
+			!hasUsage &&
 			!hasRoutineFinishReason &&
 			!STEP_END_LIVE_ONLY_FIELDS.some((field) => hasOwn(event, field))
 		)
@@ -78,10 +84,14 @@ function projectLoopEventForRecord(event) {
 		llmServerDecodeMs: _llmServerDecodeMs,
 		llmClientConsumeMs: _llmClientConsumeMs,
 		messageId: _messageId,
+		usage: _usage,
 		...recordedEvent
 	} = event;
-	if (!hasRoutineFinishReason) return recordedEvent;
-	const { finishReason: _finishReason, ...withoutRoutineFinishReason } = recordedEvent;
+	const compactUsageEvent = hasUsage
+		? { ...recordedEvent, contextTokens: usageTotal(event.usage) }
+		: recordedEvent;
+	if (!hasRoutineFinishReason) return compactUsageEvent;
+	const { finishReason: _finishReason, ...withoutRoutineFinishReason } = compactUsageEvent;
 	return withoutRoutineFinishReason;
 }
 

package/blun.mjs

@@ -79132,10 +79132,10 @@ var init_context$2 = __esmMin((() => {
 				case "step.end": {
 					const openStep = this.openSteps.get(event.uuid);
 					this.openSteps.delete(event.uuid);
-					if (event.usage !== void 0) {
+					if (event.usage !== void 0 || event.contextTokens !== void 0) {
 						const openStepIndex = openStep === void 0 ? -1 : this._history.indexOf(openStep);
 						const coveredCount = openStepIndex === -1 ? this._history.length : openStepIndex + 1;
-						const totalUsage = event.usage.inputCacheRead + event.usage.inputCacheCreation + event.usage.inputOther + event.usage.output;
+						const totalUsage = event.contextTokens ?? event.usage.inputCacheRead + event.usage.inputCacheCreation + event.usage.inputOther + event.usage.output;
 						if (totalUsage > 0) this._tokenCount = totalUsage;
 						else {
 							const previousCoveredCount = this.tokenCountCoveredMessageCount;
@@ -398968,8 +398968,8 @@ function projectContext(entries, mode = "model") {
 						});
 					}
 				} else if (ev.type === "step.end") {
-					if ("usage" in ev && ev.usage !== void 0) {
-						const fill = ev.usage.inputCacheRead + ev.usage.inputCacheCreation + ev.usage.inputOther + ev.usage.output;
+					if ("usage" in ev && ev.usage !== void 0 || ev.contextTokens !== void 0) {
+						const fill = ev.contextTokens ?? ev.usage.inputCacheRead + ev.usage.inputCacheCreation + ev.usage.inputOther + ev.usage.output;
 						if (fill > 0) contextTokens = fill;
 					}
 					openSteps.delete(ev.uuid);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "blun-king-cli",
-  "version": "9.1.243",
+  "version": "9.1.244",
   "description": "BLUN CLI - your own AI agent with a Telegram channel. Get it done. With BLUN.",
   "license": "MIT",
   "bin": {