@juliantanx/aiusage 1.0.1

package/dist/chunk-C6IKRE2P.js

@@ -0,0 +1,458 @@
+// ../core/dist/index.js
+import { createHash } from "crypto";
+var MODEL_PROVIDER_MAP = [
+  ["claude-", "anthropic"],
+  ["gpt-", "openai"],
+  ["o1-", "openai"],
+  ["o3-", "openai"],
+  ["o4-", "openai"],
+  ["o-", "openai"],
+  ["deepseek-", "deepseek"],
+  ["gemini-", "google"],
+  ["glm-", "zhipu"],
+  ["mimo-", "xiaomi"],
+  ["minimax-", "minimax"],
+  ["kimi-", "moonshot"],
+  ["qianfan-", "baidu"],
+  ["qwen", "alibaba"],
+  ["z-ai/", "zhipu"],
+  ["accounts/fireworks/", "fireworks"],
+  ["frank/", "zhipu"],
+  ["nvidia/", "nvidia"],
+  ["moonshotai/", "moonshot"],
+  ["zai-org/", "zhipu"]
+];
+function inferProvider(model) {
+  for (const [prefix, provider] of MODEL_PROVIDER_MAP) {
+    if (model.startsWith(prefix)) return provider;
+  }
+  return "unknown";
+}
+var DEFAULT_PRICE_TABLE = {
+  // Anthropic — https://platform.claude.com/docs/en/about-claude/pricing
+  "claude-opus-4-7": { input: 5, output: 25, cacheRead: 0.5, cacheWrite: 6.25 },
+  "claude-opus-4-6": { input: 5, output: 25, cacheRead: 0.5, cacheWrite: 6.25 },
+  "claude-sonnet-4-6": { input: 3, output: 15, cacheRead: 0.3, cacheWrite: 3.75 },
+  "claude-sonnet-4-5": { input: 3, output: 15, cacheRead: 0.3, cacheWrite: 3.75 },
+  "claude-haiku-4-5": { input: 1, output: 5, cacheRead: 0.1, cacheWrite: 1.25 },
+  // OpenAI — https://developers.openai.com/api/docs/pricing
+  "gpt-5.5-pro": { input: 30, output: 180 },
+  "gpt-5.5": { input: 5, output: 30 },
+  "gpt-5.4-pro": { input: 30, output: 180 },
+  "gpt-5.4-mini": { input: 0.75, output: 4.5 },
+  "gpt-5.4-nano": { input: 0.2, output: 1.25 },
+  "gpt-5.4": { input: 2.5, output: 15 },
+  "gpt-5.3-codex": { input: 1.75, output: 14 },
+  "gpt-4o-mini": { input: 0.15, output: 0.6 },
+  "gpt-4o": { input: 2.5, output: 10 },
+  "gpt-4.1": { input: 2, output: 8 },
+  "o4-mini": { input: 1.1, output: 4.4 },
+  "o3": { input: 2, output: 8 },
+  // Google Gemini — https://ai.google.dev/gemini-api/docs/pricing
+  "gemini-3.1-pro": { input: 2, output: 12 },
+  "gemini-3.1-flash-lite": { input: 0.25, output: 1.5 },
+  "gemini-3-flash": { input: 0.5, output: 3 },
+  "gemini-2.5-pro": { input: 1.25, output: 10 },
+  "gemini-2.5-flash-lite": { input: 0.1, output: 0.4 },
+  "gemini-2.5-flash": { input: 0.3, output: 2.5 },
+  "gemini-2.0-flash": { input: 0.1, output: 0.4 },
+  // DeepSeek — https://api-docs.deepseek.com/quick_start/pricing
+  "deepseek-v4-pro": { input: 1.74, output: 3.48, cacheRead: 0.0145 },
+  "deepseek-v4-flash": { input: 0.14, output: 0.28, cacheRead: 28e-4 }
+};
+var userOverrides = {};
+var PRICE_TABLE = { ...DEFAULT_PRICE_TABLE };
+function getPriceTable() {
+  return { ...DEFAULT_PRICE_TABLE, ...userOverrides };
+}
+function setPriceOverride(model, entry) {
+  userOverrides[model] = entry;
+  PRICE_TABLE[model] = entry;
+}
+function removePriceOverride(model) {
+  delete userOverrides[model];
+  if (DEFAULT_PRICE_TABLE[model]) {
+    PRICE_TABLE[model] = DEFAULT_PRICE_TABLE[model];
+  } else {
+    delete PRICE_TABLE[model];
+  }
+}
+function getUserOverrides() {
+  return { ...userOverrides };
+}
+var PROVIDER_PREFIXES = [
+  "accounts/fireworks/models/",
+  "moonshotai/",
+  "z-ai/",
+  "zai-org/",
+  "frank/",
+  "nvidia/"
+];
+function resolvePrice(model) {
+  if (PRICE_TABLE[model]) return PRICE_TABLE[model];
+  let stripped = model;
+  for (const prefix of PROVIDER_PREFIXES) {
+    if (stripped.startsWith(prefix)) {
+      stripped = stripped.slice(prefix.length);
+      break;
+    }
+  }
+  if (stripped !== model) {
+    const lc = stripped.toLowerCase();
+    if (PRICE_TABLE[lc]) return PRICE_TABLE[lc];
+    if (PRICE_TABLE[stripped]) return PRICE_TABLE[stripped];
+  }
+  let bestPrefix = "";
+  let bestEntry;
+  const candidates = [model, stripped, stripped.toLowerCase()];
+  for (const c of candidates) {
+    for (const [prefix, entry] of Object.entries(PRICE_TABLE)) {
+      if (c.startsWith(prefix) && prefix.length > bestPrefix.length) {
+        bestPrefix = prefix;
+        bestEntry = entry;
+      }
+    }
+  }
+  return bestEntry;
+}
+function calculateCost(model, tokens) {
+  const price = resolvePrice(model);
+  if (!price) return 0;
+  const inputCost = tokens.inputTokens / 1e6 * price.input;
+  const outputCost = tokens.outputTokens / 1e6 * price.output;
+  const cacheReadCost = tokens.cacheReadTokens / 1e6 * (price.cacheRead ?? 0);
+  const cacheWriteCost = tokens.cacheWriteTokens / 1e6 * (price.cacheWrite ?? 0);
+  const thinkingPrice = price.thinking ?? price.output;
+  const thinkingCost = tokens.thinkingTokens / 1e6 * thinkingPrice;
+  return inputCost + outputCost + cacheReadCost + cacheWriteCost + thinkingCost;
+}
+function generateRecordId(deviceInstanceId, sourceFile, lineOffset) {
+  const hash = createHash("sha256").update(deviceInstanceId + sourceFile + lineOffset).digest("hex");
+  return hash.slice(0, 16);
+}
+function generateSyncRecordId(deviceInstanceId, sourceFile, lineOffset) {
+  return generateRecordId(deviceInstanceId, sourceFile, lineOffset);
+}
+function generateToolCallId(recordId, name, ts, callIndex) {
+  const hash = createHash("sha256").update(recordId + name + ts + callIndex).digest("hex");
+  return hash.slice(0, 16);
+}
+function generateOrphanToolCallId(tool, name, ts, callIndex) {
+  const hash = createHash("sha256").update(tool + ":" + name + ":" + ts + ":" + callIndex).digest("hex");
+  return hash.slice(0, 16);
+}
+function generateSessionKey(device, sessionId) {
+  const hash = createHash("sha256").update(device + sessionId).digest("hex");
+  return hash.slice(0, 24);
+}
+var ClaudeCodeParser = class {
+  tool = "claude-code";
+  parseLine(line, context) {
+    let parsed;
+    try {
+      parsed = JSON.parse(line);
+    } catch {
+      return null;
+    }
+    if (!parsed.message?.usage) return null;
+    const usage = parsed.message.usage;
+    const model = parsed.message.model ?? "unknown";
+    if (model === "<synthetic>") return null;
+    const rawTs = parsed.message.timestamp ?? parsed.timestamp ?? context.now;
+    const ts = typeof rawTs === "number" ? rawTs : new Date(rawTs).getTime();
+    const inputTokens = usage.input_tokens ?? 0;
+    const outputTokens = usage.output_tokens ?? 0;
+    const cacheWriteTokens = usage.cache_creation_input_tokens ?? 0;
+    const cacheReadTokens = usage.cache_read_input_tokens ?? 0;
+    const thinkingTokens = usage.thinking_tokens ?? 0;
+    const cost = model === "unknown" ? 0 : calculateCost(model, {
+      inputTokens,
+      outputTokens,
+      cacheReadTokens,
+      cacheWriteTokens,
+      thinkingTokens
+    });
+    const costSource = model === "unknown" ? "unknown" : "pricing";
+    const provider = inferProvider(model);
+    const record = {
+      id: generateRecordId(context.deviceInstanceId, context.sourceFile, context.lineOffset),
+      ts,
+      ingestedAt: context.now,
+      updatedAt: context.now,
+      lineOffset: context.lineOffset,
+      tool: this.tool,
+      model,
+      provider,
+      inputTokens,
+      outputTokens,
+      cacheReadTokens,
+      cacheWriteTokens,
+      thinkingTokens,
+      cost,
+      costSource,
+      sessionId: context.sessionId,
+      sourceFile: context.sourceFile,
+      device: context.device,
+      deviceInstanceId: context.deviceInstanceId,
+      platform: context.platform
+    };
+    const toolCalls = [];
+    if (Array.isArray(parsed.message.content)) {
+      let callIndex = 0;
+      for (const block of parsed.message.content) {
+        if (block.type === "tool_use") {
+          const rawName = block.name ?? "";
+          const cleanName = rawName.replace(/[=:"'{\[\s].*$/s, "").replace(/[^a-zA-Z0-9_-]/g, "");
+          if (!cleanName) {
+            callIndex++;
+            continue;
+          }
+          toolCalls.push({
+            id: generateToolCallId(record.id, cleanName, ts, callIndex),
+            recordId: record.id,
+            name: cleanName,
+            ts,
+            callIndex
+          });
+          callIndex++;
+        }
+      }
+    }
+    return { record, toolCalls };
+  }
+};
+var CodexParser = class {
+  tool = "codex";
+  pendingToolCalls = [];
+  currentModel = null;
+  parseLine(line, context) {
+    let parsed;
+    try {
+      parsed = JSON.parse(line);
+    } catch {
+      return null;
+    }
+    const payload = parsed.event_msg?.payload ?? (parsed.type === "event_msg" ? parsed.payload : void 0);
+    const turnCtx = parsed.type === "turn_context" ? parsed.payload : void 0;
+    if (turnCtx?.model) {
+      this.currentModel = turnCtx.model;
+    }
+    if (payload?.type === "turn_context" && payload.model) {
+      this.currentModel = payload.model;
+    }
+    if (!payload) return null;
+    if (payload.type !== "token_count" && payload.type !== "function_call") return null;
+    if (payload.type === "function_call") {
+      const rawFcTs = parsed.event_msg?.timestamp ?? parsed.timestamp ?? context.now;
+      this.pendingToolCalls.push({
+        name: payload.function?.name ?? "unknown",
+        ts: typeof rawFcTs === "number" ? rawFcTs : new Date(rawFcTs).getTime()
+      });
+      return null;
+    }
+    const usage = payload.last_token_usage ?? payload.info?.last_token_usage;
+    if (!usage) {
+      return null;
+    }
+    const model = payload.model ?? parsed.model ?? this.currentModel ?? "unknown";
+    const rawTs = parsed.event_msg?.timestamp ?? parsed.timestamp ?? context.now;
+    const ts = typeof rawTs === "number" ? rawTs : new Date(rawTs).getTime();
+    const inputTokens = usage.input_tokens ?? 0;
+    const outputTokens = usage.output_tokens ?? 0;
+    const cacheReadTokens = usage.cached_input_tokens ?? 0;
+    const thinkingTokens = usage.reasoning_output_tokens ?? 0;
+    const cacheWriteTokens = 0;
+    const cost = model === "unknown" ? 0 : calculateCost(model, {
+      inputTokens,
+      outputTokens,
+      cacheReadTokens,
+      cacheWriteTokens,
+      thinkingTokens
+    });
+    const costSource = model === "unknown" ? "unknown" : "pricing";
+    const provider = inferProvider(model);
+    const recordId = generateRecordId(context.deviceInstanceId, context.sourceFile, context.lineOffset);
+    const record = {
+      id: recordId,
+      ts,
+      ingestedAt: context.now,
+      updatedAt: context.now,
+      lineOffset: context.lineOffset,
+      tool: this.tool,
+      model,
+      provider,
+      inputTokens,
+      outputTokens,
+      cacheReadTokens,
+      cacheWriteTokens,
+      thinkingTokens,
+      cost,
+      costSource,
+      sessionId: context.sessionId,
+      sourceFile: context.sourceFile,
+      device: context.device,
+      deviceInstanceId: context.deviceInstanceId,
+      platform: context.platform
+    };
+    const toolCalls = this.pendingToolCalls.map((tc, callIndex) => ({
+      id: generateToolCallId(recordId, tc.name, tc.ts, callIndex),
+      recordId,
+      name: tc.name,
+      ts: tc.ts,
+      callIndex
+    }));
+    this.pendingToolCalls = [];
+    return { record, toolCalls };
+  }
+  finalize() {
+    if (this.pendingToolCalls.length === 0) return [];
+    const toolCalls = this.pendingToolCalls.map((tc, callIndex) => ({
+      id: generateOrphanToolCallId(this.tool, tc.name, tc.ts, callIndex),
+      recordId: null,
+      name: tc.name,
+      ts: tc.ts,
+      callIndex
+    }));
+    this.pendingToolCalls = [];
+    return [{ record: null, toolCalls }];
+  }
+};
+var OpenClawParser = class {
+  tool = "openclaw";
+  parseLine(line, context) {
+    let parsed;
+    try {
+      parsed = JSON.parse(line);
+    } catch {
+      return null;
+    }
+    if (!parsed.message?.usage) return null;
+    const usage = parsed.message.usage;
+    const model = parsed.message.model ?? "unknown";
+    const rawTs = parsed.message.timestamp ?? context.now;
+    const ts = typeof rawTs === "number" ? rawTs : new Date(rawTs).getTime();
+    const inputTokens = usage.input ?? 0;
+    const outputTokens = usage.output ?? 0;
+    const cacheReadTokens = usage.cacheRead ?? 0;
+    const cacheWriteTokens = usage.cacheWrite ?? 0;
+    const thinkingTokens = 0;
+    let cost;
+    let costSource;
+    if (usage.cost != null && typeof usage.cost === "object") {
+      cost = usage.cost.total ?? 0;
+      costSource = "log";
+    } else if (typeof usage.cost === "number") {
+      cost = usage.cost;
+      costSource = "log";
+    } else if (model === "unknown") {
+      cost = 0;
+      costSource = "unknown";
+    } else {
+      cost = calculateCost(model, {
+        inputTokens,
+        outputTokens,
+        cacheReadTokens,
+        cacheWriteTokens,
+        thinkingTokens
+      });
+      costSource = "pricing";
+    }
+    const provider = parsed.message.provider ?? inferProvider(model);
+    const record = {
+      id: generateRecordId(context.deviceInstanceId, context.sourceFile, context.lineOffset),
+      ts,
+      ingestedAt: context.now,
+      updatedAt: context.now,
+      lineOffset: context.lineOffset,
+      tool: this.tool,
+      model,
+      provider,
+      inputTokens,
+      outputTokens,
+      cacheReadTokens,
+      cacheWriteTokens,
+      thinkingTokens,
+      cost,
+      costSource,
+      sessionId: context.sessionId,
+      sourceFile: context.sourceFile,
+      device: context.device,
+      deviceInstanceId: context.deviceInstanceId,
+      platform: context.platform
+    };
+    const toolCalls = [];
+    if (Array.isArray(parsed.message.content)) {
+      let callIndex = 0;
+      for (const block of parsed.message.content) {
+        if (block.type === "tool_use") {
+          toolCalls.push({
+            id: generateToolCallId(record.id, block.name, ts, callIndex),
+            recordId: record.id,
+            name: block.name,
+            ts,
+            callIndex
+          });
+          callIndex++;
+        }
+      }
+    }
+    return { record, toolCalls };
+  }
+};
+var Aggregator = class {
+  parsers;
+  activeParser = null;
+  constructor() {
+    this.parsers = /* @__PURE__ */ new Map([
+      ["claude-code", new ClaudeCodeParser()],
+      ["codex", new CodexParser()],
+      ["openclaw", new OpenClawParser()]
+    ]);
+  }
+  createContext(options) {
+    return {
+      ...options,
+      now: Date.now()
+    };
+  }
+  parseLine(line, context) {
+    if (!this.activeParser || this.activeParser.tool !== context.tool) {
+      this.activeParser = this.parsers.get(context.tool) ?? null;
+    }
+    if (!this.activeParser) {
+      return null;
+    }
+    return this.activeParser.parseLine(line, context);
+  }
+  finalize() {
+    const results = [];
+    for (const parser of this.parsers.values()) {
+      if (parser.finalize) {
+        results.push(...parser.finalize());
+      }
+    }
+    return results;
+  }
+};
+
+export {
+  MODEL_PROVIDER_MAP,
+  inferProvider,
+  DEFAULT_PRICE_TABLE,
+  PRICE_TABLE,
+  getPriceTable,
+  setPriceOverride,
+  removePriceOverride,
+  getUserOverrides,
+  resolvePrice,
+  calculateCost,
+  generateRecordId,
+  generateSyncRecordId,
+  generateToolCallId,
+  generateOrphanToolCallId,
+  generateSessionKey,
+  ClaudeCodeParser,
+  CodexParser,
+  OpenClawParser,
+  Aggregator
+};

package/dist/dist-XHYTXDVT.js

@@ -0,0 +1,42 @@
+import {
+  Aggregator,
+  ClaudeCodeParser,
+  CodexParser,
+  DEFAULT_PRICE_TABLE,
+  MODEL_PROVIDER_MAP,
+  OpenClawParser,
+  PRICE_TABLE,
+  calculateCost,
+  generateOrphanToolCallId,
+  generateRecordId,
+  generateSessionKey,
+  generateSyncRecordId,
+  generateToolCallId,
+  getPriceTable,
+  getUserOverrides,
+  inferProvider,
+  removePriceOverride,
+  resolvePrice,
+  setPriceOverride
+} from "./chunk-C6IKRE2P.js";
+export {
+  Aggregator,
+  ClaudeCodeParser,
+  CodexParser,
+  DEFAULT_PRICE_TABLE,
+  MODEL_PROVIDER_MAP,
+  OpenClawParser,
+  PRICE_TABLE,
+  calculateCost,
+  generateOrphanToolCallId,
+  generateRecordId,
+  generateSessionKey,
+  generateSyncRecordId,
+  generateToolCallId,
+  getPriceTable,
+  getUserOverrides,
+  inferProvider,
+  removePriceOverride,
+  resolvePrice,
+  setPriceOverride
+};

package/dist/index.d.ts

@@ -0,0 +1 @@
+#!/usr/bin/env node