kimiflare 0.37.0 → 0.38.0

package/dist/index.js

@@ -486,6 +486,7 @@ async function* runKimi(opts2) {
   if (opts2.cloudMode && !opts2.cloudToken) {
     throw new KimiApiError("kimiflare: cloud mode requires a cloud token. Run `kimiflare auth cloud` to authenticate.", void 0, 401);
   }
+  const requestId = opts2.requestId ?? crypto.randomUUID();
   const { url, headers: gatewayHeaders } = buildKimiRequestTarget(opts2);
   const body = {
     messages: sanitizeMessagesForApi(opts2.messages),
@@ -510,6 +511,7 @@ async function* runKimi(opts2) {
         headers["X-Session-ID"] = opts2.sessionId;
         headers["x-session-affinity"] = opts2.sessionId;
       }
+      headers["X-Request-ID"] = requestId;
       res = await fetch(url, {
         method: "POST",
         headers,
@@ -547,7 +549,31 @@ async function* runKimi(opts2) {
     if (!res.body) throw new KimiApiError("kimiflare: empty response body", void 0, res.status);
     const meta = readGatewayMeta(res.headers);
     if (meta) yield { type: "gateway_meta", meta };
-    yield* parseStream(res.body, opts2.signal);
+    let lastUsage = null;
+    for await (const ev of parseStream(res.body, opts2.signal)) {
+      if (ev.type === "usage") lastUsage = ev.usage;
+      yield ev;
+    }
+    if (opts2.cloudMode && lastUsage && opts2.cloudToken) {
+      const reportUrl = "https://api.kimiflare.com/v1/usage/report";
+      const reportHeaders = {
+        Authorization: `Bearer ${opts2.cloudToken}`,
+        "Content-Type": "application/json"
+      };
+      if (opts2.cloudDeviceId) reportHeaders["X-Device-ID"] = opts2.cloudDeviceId;
+      if (opts2.sessionId) reportHeaders["X-Session-ID"] = opts2.sessionId;
+      fetch(reportUrl, {
+        method: "POST",
+        headers: reportHeaders,
+        body: JSON.stringify({
+          request_id: requestId,
+          prompt_tokens: lastUsage.prompt_tokens,
+          completion_tokens: lastUsage.completion_tokens,
+          cached_tokens: lastUsage.prompt_tokens_details?.cached_tokens ?? 0
+        })
+      }).catch(() => {
+      });
+    }
     return;
   }
 }