kimiflare 0.36.2 → 0.37.0

package/dist/index.js

@@ -6878,9 +6878,9 @@ function buildRightParts(usage, contextLimit, sessionUsage, gatewayMeta, cloudMo
     parts.push(`out ${sessionUsage.completionTokens}`);
     parts.push(`ctx ${pct}%`);
     if (cloudMode) {
-      parts.push(`\x1B[9m${sessionUsage.cost.toFixed(5)}\x1B[29m`);
+      parts.push(`\x1B[9m$${sessionUsage.cost.toFixed(5)}\x1B[29m`);
     } else {
-      parts.push(`${sessionUsage.cost.toFixed(5)}`);
+      parts.push(`$${sessionUsage.cost.toFixed(5)}`);
     }
   } else {
     const cached = usage.prompt_tokens_details?.cached_tokens ?? 0;
@@ -6889,9 +6889,9 @@ function buildRightParts(usage, contextLimit, sessionUsage, gatewayMeta, cloudMo
     parts.push(`out ${usage.completion_tokens}`);
     parts.push(`ctx ${pct}%`);
     if (cloudMode) {
-      parts.push(`\x1B[9m${cost.total.toFixed(5)}\x1B[29m`);
+      parts.push(`\x1B[9m$${cost.total.toFixed(5)}\x1B[29m`);
     } else {
-      parts.push(`${cost.total.toFixed(5)}`);
+      parts.push(`$${cost.total.toFixed(5)}`);
     }
   }
   if (cloudMode && cloudBudget) {
@@ -6902,6 +6902,7 @@ function buildRightParts(usage, contextLimit, sessionUsage, gatewayMeta, cloudMo
   return parts;
 }
 function formatTokens(n) {
+  if (n >= 1e9) return `${(n / 1e9).toFixed(1)}B`;
   if (n >= 1e6) return `${(n / 1e6).toFixed(1)}M`;
   if (n >= 1e3) return `${(n / 1e3).toFixed(1)}K`;
   return String(n);
@@ -14047,6 +14048,17 @@ function App({
             const sid = ensureSessionId();
             void recordUsage(sid, u, gatewayUsageLookupFromConfig(cfg, meta ?? gatewayMetaRef.current));
             void getCostReport(sid).then((report) => setSessionUsage(report.session));
+            if (cfg?.cloudMode && (cloudToken ?? initialCloudToken)) {
+              const token = cloudToken ?? initialCloudToken;
+              const did = cloudDeviceId ?? initialCloudDeviceId;
+              void (async () => {
+                const { fetchCloudUsage: fetchCloudUsage2 } = await Promise.resolve().then(() => (init_auth(), auth_exports));
+                const usage2 = await fetchCloudUsage2(token, did);
+                if (usage2) {
+                  setCloudBudget({ remaining: usage2.remaining, limit: usage2.input_token_limit });
+                }
+              })();
+            }
           },
           onGatewayMeta: updateGatewayMeta,
           askPermission: (req) => new Promise((resolve2) => {
@@ -15126,6 +15138,17 @@ ${lines.join("\n")}` }]);
           const sid = ensureSessionId();
           void recordUsage(sid, u, gatewayUsageLookupFromConfig(cfg, meta ?? gatewayMetaRef.current));
           void getCostReport(sid).then((report) => setSessionUsage(report.session));
+          if (cfg?.cloudMode && (cloudToken ?? initialCloudToken)) {
+            const token = cloudToken ?? initialCloudToken;
+            const did = cloudDeviceId ?? initialCloudDeviceId;
+            void (async () => {
+              const { fetchCloudUsage: fetchCloudUsage2 } = await Promise.resolve().then(() => (init_auth(), auth_exports));
+              const usage2 = await fetchCloudUsage2(token, did);
+              if (usage2) {
+                setCloudBudget({ remaining: usage2.remaining, limit: usage2.input_token_limit });
+              }
+            })();
+          }
         },
         onGatewayMeta: updateGatewayMeta,
         onTasks: (nextTasks) => {