@animalabs/membrane 0.5.41 → 0.5.43

package/src/types/yielding-stream.ts

@@ -10,7 +10,7 @@
 
 import type { ContentBlock } from './content.js';
 import type { ToolCall, ToolResult, ToolContext } from './tools.js';
-import type { BasicUsage, NormalizedResponse, StopReason } from './response.js';
+import type { DetailedUsage, NormalizedResponse, StopReason } from './response.js';
 import type { ChunkMeta, BlockEvent } from './streaming.js';
 
 // ============================================================================
@@ -49,7 +49,7 @@ export interface ToolCallsEvent {
  */
 export interface UsageEvent {
   type: 'usage';
-  usage: BasicUsage;
+  usage: DetailedUsage;
 }
 
 /**

package/src/utils/cost.ts

@@ -0,0 +1,29 @@
+import type { CostBreakdown } from '../types/response.js';
+import type { ModelPricing } from '../types/provider.js';
+
+export interface CostableUsage {
+  inputTokens: number;
+  outputTokens: number;
+  cacheCreationTokens?: number;
+  cacheReadTokens?: number;
+}
+
+export function calculateCost(usage: CostableUsage, pricing: ModelPricing): CostBreakdown {
+  const input = usage.inputTokens * pricing.inputPerMillion / 1_000_000;
+  const output = usage.outputTokens * pricing.outputPerMillion / 1_000_000;
+  const cacheWrite = pricing.cacheWritePerMillion != null
+    ? (usage.cacheCreationTokens ?? 0) * pricing.cacheWritePerMillion / 1_000_000
+    : undefined;
+  const cacheRead = pricing.cacheReadPerMillion != null
+    ? (usage.cacheReadTokens ?? 0) * pricing.cacheReadPerMillion / 1_000_000
+    : undefined;
+
+  return {
+    input,
+    output,
+    cacheWrite,
+    cacheRead,
+    total: input + output + (cacheWrite ?? 0) + (cacheRead ?? 0),
+    currency: pricing.currency,
+  };
+}

package/src/utils/index.ts

@@ -13,3 +13,6 @@ export {
   unescapeXml,
   type ToolDefinitionForPrompt,
 } from './tool-parser.js';
+
+export { calculateCost } from './cost.js';
+export type { CostableUsage } from './cost.js';