kimiflare 0.67.0 → 0.68.0

package/dist/index.js

@@ -6322,6 +6322,12 @@ var init_renderer = __esm({
 });
 
 // src/cost-attribution/reconcile.ts
+var reconcile_exports = {};
+__export(reconcile_exports, {
+  aggregateByFeature: () => aggregateByFeature,
+  fetchGatewayLogs: () => fetchGatewayLogs,
+  reconcileWithCloudflare: () => reconcileWithCloudflare
+});
 function cacheKey(opts2) {
   return `${opts2.gatewayId ?? "none"}:${opts2.startDate}:${opts2.endDate}`;
 }
@@ -6360,6 +6366,28 @@ async function fetchGatewayLogs(accountId, apiToken, gatewayId, startDate, endDa
   }
   return out;
 }
+function aggregateByFeature(logs) {
+  const map = /* @__PURE__ */ new Map();
+  for (const log2 of logs) {
+    let feature = "unknown";
+    const m = log2.metadata;
+    if (m && typeof m === "object" && !Array.isArray(m)) {
+      const f = m.feature;
+      if (typeof f === "string") feature = f;
+    } else if (typeof m === "string") {
+      try {
+        const parsed = JSON.parse(m);
+        if (typeof parsed.feature === "string") feature = parsed.feature;
+      } catch {
+      }
+    }
+    const entry = map.get(feature) ?? { feature, cost: 0, requests: 0 };
+    entry.cost += typeof log2.cost === "number" ? log2.cost : 0;
+    entry.requests += 1;
+    map.set(feature, entry);
+  }
+  return Array.from(map.values()).sort((a, b) => b.cost - a.cost);
+}
 async function reconcileWithCloudflare(opts2) {
   if (!opts2.accountId || !opts2.apiToken) {
     return {
@@ -6399,7 +6427,8 @@ async function reconcileWithCloudflare(opts2) {
       localCost: opts2.localCost,
       cloudflareCost,
       driftPct: Math.round(driftPct * 1e3) / 10,
-      message: `Reconciled ${logs.length} Gateway log entries`
+      message: `Reconciled ${logs.length} Gateway log entries`,
+      featureBreakdown: aggregateByFeature(logs)
     };
     cache.set(key, { result, expires: Date.now() + 60 * 60 * 1e3 });
     return result;
@@ -9459,7 +9488,8 @@ async function getCostReport(sessionId) {
     gatewayRequests: rawSession.gatewayRequests,
     gatewayCachedRequests: rawSession.gatewayCachedRequests,
     gatewayCost: rawSession.gatewayCost,
-    reconcilePending: hasPendingReconcile(rawSession)
+    reconcilePending: hasPendingReconcile(rawSession),
+    lastTurnMs: latestConfirmedDurationMs(rawSession)
   } : { date, promptTokens: 0, completionTokens: 0, cachedTokens: 0, cost: 0 };
   const todayUsage = log2.days.find((d) => d.date === date) ?? { date, promptTokens: 0, completionTokens: 0, cachedTokens: 0, cost: 0 };
   const monthUsage = {
@@ -9504,6 +9534,14 @@ function hasPendingReconcile(session) {
     (t) => t.logId && t.confirmedCost === void 0 && !t.reconcileFailed
   );
 }
+function latestConfirmedDurationMs(session) {
+  if (!session.turns) return void 0;
+  for (let i = session.turns.length - 1; i >= 0; i--) {
+    const ms = session.turns[i]?.durationMs;
+    if (typeof ms === "number") return ms;
+  }
+  return void 0;
+}
 async function getSessionGatewayLogs(sessionId) {
   const log2 = await loadLog2();
   const session = log2.sessions.find((s) => s.id === sessionId);
@@ -9549,6 +9587,17 @@ function formatGatewaySection(report, accountId, gatewayId, recentLogs2 = []) {
   );
   return lines.join("\n");
 }
+function formatFeatureBreakdown(breakdown) {
+  if (!breakdown || breakdown.length === 0) return "";
+  if (breakdown.length === 1 && breakdown[0].feature === "unknown") return "";
+  const lines = ["\u2500\u2500\u2500 By feature (Gateway-confirmed) \u2500\u2500\u2500"];
+  for (const row of breakdown) {
+    lines.push(
+      `  ${row.feature.padEnd(20)} $${row.cost.toFixed(4)}  (${row.requests} req)`
+    );
+  }
+  return lines.join("\n");
+}
 function formatCostReport(report) {
   const lines = [];
   const add = (label, u) => {
@@ -11752,6 +11801,9 @@ function buildRightParts(usage, contextLimit, sessionUsage, gatewayMeta, cloudMo
     } else {
       parts.push(`${prefix}${sessionUsage.cost.toFixed(2)}`);
     }
+    if (typeof sessionUsage.lastTurnMs === "number") {
+      parts.push(formatDuration(sessionUsage.lastTurnMs));
+    }
   } else {
     const cached = usage.prompt_tokens_details?.cached_tokens ?? 0;
     const cost = calculateCost(usage.prompt_tokens, usage.completion_tokens, cached);
@@ -11780,6 +11832,10 @@ function formatGatewayCacheStatus(gatewayMeta) {
   const status = gatewayMeta?.cacheStatus?.trim();
   return status ? `AI Gateway \xB7 cache ${status.toLowerCase()}` : null;
 }
+function formatDuration(ms) {
+  if (ms < 1e3) return `${Math.round(ms)}ms`;
+  return `${(ms / 1e3).toFixed(1)}s`;
+}
 function formatElapsed2(ms) {
   const total = Math.floor(ms / 1e3);
   const m = Math.floor(total / 60);
@@ -20079,6 +20135,22 @@ ${wcagWarnings.join("\n")}` }
             const logs = sid ? await getSessionGatewayLogs(sid).catch(() => []) : [];
             const gwSection = formatGatewaySection(report, cfg.accountId, cfg.aiGatewayId, logs);
             if (gwSection) lines.push("", gwSection);
+            try {
+              const { reconcileWithCloudflare: reconcileWithCloudflare2 } = await Promise.resolve().then(() => (init_reconcile(), reconcile_exports));
+              const today4 = (/* @__PURE__ */ new Date()).toISOString().slice(0, 10);
+              const sevenDaysAgo = new Date(Date.now() - 7 * 24 * 60 * 60 * 1e3).toISOString().slice(0, 10);
+              const recon = await reconcileWithCloudflare2({
+                localCost: report.month.cost,
+                accountId: cfg.accountId,
+                apiToken: cfg.apiToken,
+                gatewayId: cfg.aiGatewayId,
+                startDate: sevenDaysAgo,
+                endDate: today4
+              });
+              const breakdown = formatFeatureBreakdown(recon.featureBreakdown);
+              if (breakdown) lines.push("", breakdown);
+            } catch {
+            }
           }
           if (cfg?.costAttribution) {
             const { getCategoryReportText: getCategoryReportText2 } = await Promise.resolve().then(() => (init_tui_report(), tui_report_exports));
@@ -20156,6 +20228,24 @@ ${wcagWarnings.join("\n")}` }
             lines.push(`collect-logs: ${cfg.aiGatewayCollectLogPayload ?? false}`);
             const meta = cfg.aiGatewayMetadata;
             lines.push(`metadata: ${meta && Object.keys(meta).length > 0 ? JSON.stringify(meta) : "none"}`);
+            const sid = sessionIdRef.current;
+            if (sid) {
+              void getCostReport(sid).then((report) => {
+                const req = report.session.gatewayRequests ?? 0;
+                if (req === 0) return;
+                const cached = report.session.gatewayCachedRequests ?? 0;
+                const pct = (cached / req * 100).toFixed(1);
+                setEvents((e) => [
+                  ...e,
+                  {
+                    kind: "info",
+                    key: mkKey(),
+                    text: `cache hits (session): ${cached}/${req} (${pct}%)`
+                  }
+                ]);
+              }).catch(() => {
+              });
+            }
           } else {
             lines.push("gateway: off (direct Workers AI)");
           }