aisnitch 0.2.21 → 0.2.23

package/dist/index.cjs

@@ -818,6 +818,9 @@ var EventDataSchema = import_zod.z.strictObject({
   activeFile: import_zod.z.string().min(1).max(4096).optional(),
   model: import_zod.z.string().min(1).max(200).optional(),
   tokensUsed: import_zod.z.number().int().min(0).optional(),
+  inputTokens: import_zod.z.number().int().min(0).optional(),
+  outputTokens: import_zod.z.number().int().min(0).optional(),
+  cachedTokens: import_zod.z.number().int().min(0).optional(),
   errorMessage: import_zod.z.string().min(1).max(1e4).optional(),
   errorType: ErrorTypeSchema.optional(),
   raw: import_zod.z.record(import_zod.z.string(), import_zod.z.unknown()).optional(),
@@ -2215,7 +2218,7 @@ function extractClaudeTranscriptObservations(payload, transcriptPath) {
   });
   const contentParts = extractClaudeContentParts(payload);
   const model = getString(payload, "model") ?? getString(getRecord(payload.message), "model");
-  const tokensUsed = extractTokenUsage(payload);
+  const tokens = extractTokenUsageDetailed(payload);
   const rawPayload = payload;
   const sharedContext = {
     // 📖 Pass process.env so terminal detection works from transcript path too
@@ -2228,7 +2231,10 @@ function extractClaudeTranscriptObservations(payload, transcriptPath) {
   const sharedData = {
     model,
     raw: rawPayload,
-    tokensUsed
+    tokensUsed: tokens.total,
+    inputTokens: tokens.input,
+    outputTokens: tokens.output,
+    cachedTokens: tokens.cached
   };
   const observations = [];
   if (contentParts.some((part) => part.type === "thinking")) {
@@ -2286,23 +2292,28 @@ function extractClaudeContentParts(payload) {
   }
   return content.filter(isRecord);
 }
-function extractTokenUsage(payload) {
+function extractTokenUsageDetailed(payload) {
   const tokens = getNumber(payload, "tokens");
   if (tokens !== void 0) {
-    return tokens;
+    return { total: tokens };
   }
   const usage = getRecord(payload.usage);
   if (!usage) {
-    return void 0;
+    return {};
   }
   const totalTokens = getNumber(usage, "total_tokens");
   if (totalTokens !== void 0) {
-    return totalTokens;
+    return {
+      total: totalTokens,
+      input: getNumber(usage, "input_tokens"),
+      output: getNumber(usage, "output_tokens"),
+      cached: getNumber(usage, "cached_tokens")
+    };
   }
   const inputTokens = getNumber(usage, "input_tokens") ?? 0;
   const outputTokens = getNumber(usage, "output_tokens") ?? 0;
   const usageSum = inputTokens + outputTokens;
-  return usageSum > 0 ? usageSum : void 0;
+  return usageSum > 0 ? { total: usageSum, input: inputTokens, output: outputTokens } : {};
 }
 function extractClaudeToolInput(payload) {
   const toolInput = getRecord(payload.tool_input) ?? getRecord(payload.toolInput);
@@ -7406,7 +7417,17 @@ var OpenCodeAdapter = class extends BaseAdapter {
       project: extractOpenCodeProject(payload),
       raw: payload,
       toolInput: extractOpenCodeToolInput(payload),
-      toolName: extractOpenCodeToolName(payload)
+      toolName: extractOpenCodeToolName(payload),
+      // 📖 Extract token usage from payload (with input/output/cached breakdown)
+      ...(() => {
+        const tokens = extractOpenCodeTokens(payload);
+        return {
+          tokensUsed: tokens.total,
+          inputTokens: tokens.inputTokens,
+          outputTokens: tokens.outputTokens,
+          cachedTokens: tokens.cachedTokens
+        };
+      })()
     };
     switch (eventType) {
       case "session.created": {
@@ -7435,9 +7456,21 @@ var OpenCodeAdapter = class extends BaseAdapter {
         await this.emitStateChange("agent.compact", sharedData, context);
         return;
       }
+      case "thinking": {
+        const thinkingContent = extractOpenCodeThinkingContent(payload);
+        await this.emitStateChange("agent.thinking", {
+          ...sharedData,
+          thinkingContent
+        }, context);
+        return;
+      }
       case "message.updated":
       case "message.part.updated": {
-        await this.emitStateChange("agent.streaming", sharedData, context);
+        const messageContent = extractOpenCodeMessageContent(payload);
+        await this.emitStateChange("agent.streaming", {
+          ...sharedData,
+          messageContent
+        }, context);
         return;
       }
       case "permission.asked": {
@@ -7656,6 +7689,67 @@ function extractOpenCodeToolResult(payload) {
   }
   return void 0;
 }
+function extractOpenCodeMessageContent(payload) {
+  const directMessage = getString10(payload, "text") ?? getString10(payload, "message") ?? getString10(payload, "content") ?? getString10(payload, "output");
+  if (directMessage) {
+    return directMessage;
+  }
+  const part = getRecord9(payload.part) ?? getRecord9(getRecord9(payload.properties)?.part);
+  if (part) {
+    return getString10(part, "text") ?? getString10(part, "content");
+  }
+  const props = getRecord9(payload.properties);
+  if (props) {
+    const message = getRecord9(props.message) ?? getRecord9(props.info);
+    if (message) {
+      return getString10(message, "text") ?? getString10(message, "content");
+    }
+    return getString10(props, "text") ?? getString10(props, "message");
+  }
+  return void 0;
+}
+function extractOpenCodeThinkingContent(payload) {
+  const directThinking = getString10(payload, "thinking") ?? getString10(payload, "thinkingContent") ?? getString10(payload, "reasoning") ?? getString10(payload, "thought");
+  if (directThinking) {
+    return directThinking;
+  }
+  const props = getRecord9(payload.properties);
+  if (props) {
+    return getString10(props, "thinking") ?? getString10(props, "reasoning") ?? getString10(props, "thought");
+  }
+  return void 0;
+}
+function extractOpenCodeTokens(payload) {
+  const directTokens = getNumber8(payload, "tokensUsed") ?? getNumber8(payload, "tokens");
+  if (directTokens !== void 0) {
+    return { total: directTokens };
+  }
+  const props = getRecord9(payload.properties);
+  if (props) {
+    const info = getRecord9(props.info);
+    if (info) {
+      const tokens = getRecord9(info.tokens);
+      if (tokens) {
+        const inputTokens = getNumber8(tokens, "input") ?? 0;
+        const outputTokens = getNumber8(tokens, "output") ?? 0;
+        const reasoningTokens = getNumber8(tokens, "reasoning") ?? 0;
+        const total = inputTokens + outputTokens + reasoningTokens;
+        return total > 0 ? {
+          total,
+          inputTokens,
+          outputTokens,
+          // reasoning tokens are often billed as cached on some providers
+          cachedTokens: reasoningTokens
+        } : {};
+      }
+    }
+    const fallbackTokens = getNumber8(props, "tokensUsed") ?? getNumber8(props, "tokens");
+    if (fallbackTokens !== void 0) {
+      return { total: fallbackTokens };
+    }
+  }
+  return {};
+}
 
 // src/adapters/pi.ts
 var import_node_child_process12 = require("child_process");
@@ -9352,7 +9446,7 @@ var import_ws = require("ws");
 
 // src/package-info.ts
 var AISNITCH_PACKAGE_NAME = "aisnitch";
-var AISNITCH_VERSION = "0.2.21";
+var AISNITCH_VERSION = "0.2.23";
 var AISNITCH_DESCRIPTION = "Universal bridge for AI coding tool activity \u2014 capture, normalize, stream.";
 function getPackageScaffoldInfo() {
   return {