negotium 0.2.15 → 0.2.17

package/dist/storage.js

@@ -3361,6 +3361,7 @@ var exports_token_stats = {};
 __export(exports_token_stats, {
   tokenStatsFileId: () => tokenStatsFileId,
   recordUsage: () => recordUsage,
+  getTopicStats: () => getTopicStats,
   getStats: () => getStats,
   calcCost: () => calcCost
 });
@@ -3373,7 +3374,8 @@ function emptyBucket() {
     outputTokens: 0,
     cacheCreationInputTokens: 0,
     cacheReadInputTokens: 0,
-    queries: 0
+    queries: 0,
+    estimatedCostUsd: 0
   };
 }
 function tokenStatsFileId(userId) {
@@ -3400,16 +3402,53 @@ function loadRecords(userId) {
   }
 }
 function calcCost(b) {
-  return b.inputTokens / 1e6 * 3 + b.outputTokens / 1e6 * 15 + b.cacheCreationInputTokens / 1e6 * 3.75 + b.cacheReadInputTokens / 1e6 * 0.3;
+  return b.estimatedCostUsd;
+}
+var TOKEN_PRICES = {
+  "codex:gpt-5.6-sol": { input: 5, cacheRead: 0.5, output: 30 },
+  "codex:gpt-5.6-terra": { input: 2.5, cacheRead: 0.25, output: 15 },
+  "codex:gpt-5.6-luna": { input: 1, cacheRead: 0.1, output: 6 },
+  "claude:fable": { input: 10, cacheWrite: 12.5, cacheRead: 1, output: 50 },
+  "claude:opus": { input: 5, cacheWrite: 6.25, cacheRead: 0.5, output: 25 },
+  "claude:sonnet": { input: 2, cacheWrite: 2.5, cacheRead: 0.2, output: 10 },
+  "maestro:kimi-k3": { input: 3, cacheRead: 0.3, output: 15 },
+  "maestro:kimi-k2.7-code": { input: 0.95, cacheRead: 0.19, output: 4 },
+  "maestro:deepseek-pro": { input: 0.435, cacheRead: 0.003625, output: 0.87 },
+  "maestro:deepseek-flash": { input: 0.14, cacheRead: 0.0028, output: 0.28 }
+};
+function estimateUsageCost(agent, model, usage) {
+  const prices = TOKEN_PRICES[`${agent}:${model}`];
+  if (!prices)
+    return 0;
+  return (usage.inputTokens * prices.input + usage.outputTokens * prices.output + (agent === "claude" ? usage.cacheCreationInputTokens * (prices.cacheWrite ?? prices.input) : 0) + usage.cacheReadInputTokens * prices.cacheRead) / 1e6;
 }
-function recordUsage(userId, session, usage) {
+function isQueryRecord(value) {
+  if (!value || typeof value !== "object")
+    return false;
+  const record = value;
+  return record.schemaVersion === 2 && typeof record.timestamp === "string" && typeof record.session === "string" && typeof record.topicId === "string" && typeof record.agent === "string" && typeof record.model === "string" && typeof record.inputTokens === "number" && typeof record.outputTokens === "number" && typeof record.cacheCreationInputTokens === "number" && typeof record.cacheReadInputTokens === "number" && typeof record.estimatedCostUsd === "number";
+}
+function recordUsage(userId, session, usage, context) {
+  const cacheReadInputTokens = usage.cacheReadInputTokens ?? 0;
+  const inputTokens = context.agent === "claude" ? usage.inputTokens : Math.max(0, usage.inputTokens - cacheReadInputTokens);
+  const normalized = {
+    inputTokens,
+    outputTokens: usage.outputTokens,
+    cacheCreationInputTokens: usage.cacheCreationInputTokens ?? 0,
+    cacheReadInputTokens
+  };
   const record = {
+    schemaVersion: 2,
     timestamp: new Date().toISOString(),
     session,
-    inputTokens: usage.inputTokens,
-    outputTokens: usage.outputTokens,
-    cacheCreationInputTokens: usage.cacheCreationInputTokens ?? 0,
-    cacheReadInputTokens: usage.cacheReadInputTokens ?? 0
+    topicId: context.topicId,
+    ...context.providerSessionId ? { providerSessionId: context.providerSessionId } : {},
+    agent: context.agent,
+    model: context.model,
+    ...normalized,
+    ...usage.contextTokens !== undefined ? { contextTokens: usage.contextTokens } : {},
+    ...usage.contextWindow !== undefined ? { contextWindow: usage.contextWindow } : {},
+    estimatedCostUsd: usage.costUsd ?? estimateUsageCost(context.agent, context.model, normalized)
   };
   try {
     appendJsonlEntry(queriesPath(userId), record);
@@ -3423,12 +3462,26 @@ function getStats(userId, from, to) {
   const toTs = to ? new Date(to).getTime() : Infinity;
   if (from && Number.isNaN(fromTs) || to && Number.isNaN(toTs)) {
     logger.warn({ from, to }, "token-stats: Invalid date range, returning empty");
-    return { total: emptyBucket(), byHour: {}, bySession: {}, estimatedCostUsd: 0 };
+    return {
+      total: emptyBucket(),
+      byHour: {},
+      bySession: {},
+      currentSessions: [],
+      ignoredLegacyRecords: 0,
+      estimatedCostUsd: 0
+    };
   }
   const total = emptyBucket();
   const byHour = {};
   const bySession = {};
-  for (const r of records) {
+  const currentSessions = new Map;
+  let ignoredLegacyRecords = 0;
+  for (const raw of records) {
+    if (!isQueryRecord(raw)) {
+      ignoredLegacyRecords += 1;
+      continue;
+    }
+    const r = raw;
     const ts = new Date(r.timestamp).getTime();
     if (ts < fromTs || ts > toTs)
       continue;
@@ -3443,9 +3496,56 @@ function getStats(userId, from, to) {
       bucket.cacheCreationInputTokens += r.cacheCreationInputTokens;
       bucket.cacheReadInputTokens += r.cacheReadInputTokens;
       bucket.queries += 1;
+      bucket.estimatedCostUsd += r.estimatedCostUsd;
+    }
+    if (r.contextTokens !== undefined && r.contextWindow !== undefined && r.contextWindow > 0) {
+      currentSessions.set(r.topicId, {
+        timestamp: r.timestamp,
+        topicId: r.topicId,
+        topicTitle: r.session,
+        ...r.providerSessionId ? { providerSessionId: r.providerSessionId } : {},
+        agent: r.agent,
+        model: r.model,
+        contextTokens: r.contextTokens,
+        contextWindow: r.contextWindow
+      });
+    }
+  }
+  return {
+    total,
+    byHour,
+    bySession,
+    currentSessions: [...currentSessions.values()].sort((a, b) => b.timestamp.localeCompare(a.timestamp)),
+    ignoredLegacyRecords,
+    estimatedCostUsd: calcCost(total)
+  };
+}
+function getTopicStats(userId, topicId) {
+  const total = emptyBucket();
+  let currentSession;
+  for (const raw of loadRecords(userId)) {
+    if (!isQueryRecord(raw) || raw.topicId !== topicId)
+      continue;
+    total.inputTokens += raw.inputTokens;
+    total.outputTokens += raw.outputTokens;
+    total.cacheCreationInputTokens += raw.cacheCreationInputTokens;
+    total.cacheReadInputTokens += raw.cacheReadInputTokens;
+    total.queries += 1;
+    total.estimatedCostUsd += raw.estimatedCostUsd;
+    if (raw.contextTokens !== undefined && raw.contextWindow !== undefined && raw.contextWindow > 0 && (!currentSession || raw.timestamp > currentSession.timestamp)) {
+      currentSession = {
+        timestamp: raw.timestamp,
+        topicId: raw.topicId,
+        topicTitle: raw.session,
+        ...raw.providerSessionId ? { providerSessionId: raw.providerSessionId } : {},
+        agent: raw.agent,
+        model: raw.model,
+        contextTokens: raw.contextTokens,
+        contextWindow: raw.contextWindow
+      };
     }
   }
-  return { total, byHour, bySession, estimatedCostUsd: calcCost(total) };
+  return { topicId, ...total, ...currentSession ? { currentSession } : {} };
 }
 // ../../packages/core/src/storage/topic-archive.ts
 var exports_topic_archive = {};
@@ -3971,6 +4071,7 @@ export {
   grantSubagentTellTarget,
   getWikiDir,
   getUserConfig,
+  getTopicStats,
   getTopicSessionId,
   getTopicNames,
   getTopicMemoryOrigin,
@@ -4065,4 +4166,4 @@ export {
   ConversationLogDivergedError
 };
 
-//# debugId=CD82519368DB318D64756E2164756E21
+//# debugId=0E9898BA8C097A7864756E2164756E21