mu-core 0.12.0 → 0.15.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "mu-core",
-  "version": "0.12.0",
+  "version": "0.15.0",
   "description": "Agent loop orchestration core: types, plugin SDK, channels, sessions",
   "type": "module",
   "main": "./src/index.ts",

package/src/agent.ts

@@ -87,6 +87,7 @@ async function* streamTurn(
   let reasoning = '';
   let usage = 0;
   let cachedPromptTokens = 0;
+  let promptTokens = 0;
   const toolCalls: ToolCall[] = [];
 
   const hooks = registry.getHooks();
@@ -104,6 +105,7 @@ async function* streamTurn(
     tools: toolDefinitions,
     onUsage: (u) => {
       usage = u.totalTokens;
+      promptTokens = u.promptTokens;
       cachedPromptTokens = u.cachedPromptTokens ?? 0;
     },
   })) {
@@ -121,7 +123,7 @@ async function* streamTurn(
     }
   }
 
-  const result: TurnResult = { content, reasoning, toolCalls, usage, cachedPromptTokens };
+  const result: TurnResult = { content, reasoning, toolCalls, usage, promptTokens, cachedPromptTokens };
   return await runAfterLlmHooks(hooks, result);
 }
 
@@ -210,7 +212,7 @@ export async function* runAgent(
     while (!signal.aborted) {
       // Re-evaluate every turn so per-turn agent switches are honoured.
       const tools = await registry.getFilteredTools();
-      const { content, reasoning, toolCalls, usage, cachedPromptTokens } = yield* streamTurn(
+      const { content, reasoning, toolCalls, usage, promptTokens, cachedPromptTokens } = yield* streamTurn(
         current,
         mergedConfig,
         model,
@@ -220,7 +222,7 @@ export async function* runAgent(
       );
 
       if (usage > 0) {
-        yield { type: 'usage', totalTokens: usage, cachedTokens: cachedPromptTokens };
+        yield { type: 'usage', totalTokens: usage, promptTokens, cachedTokens: cachedPromptTokens };
       }
       if (signal.aborted) {
         break;

package/src/plugin.ts

@@ -232,6 +232,8 @@ export interface TurnResult {
   reasoning: string;
   toolCalls: ToolCall[];
   usage: number;
+  /** Input tokens sent to the model for this turn (prompt size). */
+  promptTokens: number;
   /** Subset of prompt tokens served from the server's prompt cache, when
    *  reported. 0 when unsupported or no cache hit. */
   cachedPromptTokens?: number;
@@ -330,7 +332,7 @@ export interface SlashCommand {
 export type AgentEvent =
   | { type: 'content'; text: string }
   | { type: 'reasoning'; text: string }
-  | { type: 'usage'; totalTokens: number; cachedTokens?: number }
+  | { type: 'usage'; totalTokens: number; promptTokens: number; cachedTokens?: number }
   | { type: 'messages'; messages: ChatMessage[] }
   | { type: 'turn_end' };
 

package/src/session.ts

@@ -18,7 +18,7 @@ export type SessionEvent =
   | { type: 'stream_partial'; text: string; reasoning?: string }
   | { type: 'stream_started' }
   | { type: 'stream_ended' }
-  | { type: 'usage'; totalTokens: number; cachedTokens: number }
+  | { type: 'usage'; totalTokens: number; promptTokens: number; cachedTokens: number }
   | { type: 'error'; message: string };
 
 export interface RunTurnOptions {
@@ -164,7 +164,12 @@ class SessionImpl implements Session {
         final = this.messages.slice();
         this.emit({ type: 'messages_changed', messages: this.messages.slice() });
       } else if (e.type === 'usage') {
-        this.emit({ type: 'usage', totalTokens: e.totalTokens, cachedTokens: e.cachedTokens ?? 0 });
+        this.emit({
+          type: 'usage',
+          totalTokens: e.totalTokens,
+          promptTokens: e.promptTokens,
+          cachedTokens: e.cachedTokens ?? 0,
+        });
       } else if (e.type === 'turn_end') {
         // Clear the locally-tracked partial buffers AND notify subscribers,
         // otherwise the host's `stream` state still holds the previous step's