@blockrun/clawrouter 0.11.10 → 0.11.11

package/dist/cli.js

@@ -5185,6 +5185,7 @@ async function proxyRequest(req, res, apiBase, payFetch, options, routerOpts, de
   let maxTokens = 4096;
   let routingProfile = null;
   let accumulatedContent = "";
+  let responseInputTokens;
   const isChatCompletion = req.url?.includes("/chat/completions");
   const sessionId = getSessionId(req.headers);
   let effectiveSessionId = sessionId;
@@ -6039,6 +6040,10 @@ async function proxyRequest(req, res, apiBase, payFetch, options, routerOpts, de
         const jsonStr = jsonBody.toString();
         try {
           const rsp = JSON.parse(jsonStr);
+          if (rsp.usage && typeof rsp.usage === "object") {
+            const u = rsp.usage;
+            if (typeof u.prompt_tokens === "number") responseInputTokens = u.prompt_tokens;
+          }
           const baseChunk = {
             id: rsp.id ?? `chatcmpl-${Date.now()}`,
             object: "chat.completion.chunk",
@@ -6187,6 +6192,10 @@ async function proxyRequest(req, res, apiBase, payFetch, options, routerOpts, de
         if (rspJson.choices?.[0]?.message?.content) {
           accumulatedContent = rspJson.choices[0].message.content;
         }
+        if (rspJson.usage && typeof rspJson.usage === "object") {
+          if (typeof rspJson.usage.prompt_tokens === "number")
+            responseInputTokens = rspJson.usage.prompt_tokens;
+        }
       } catch {
       }
     }
@@ -6235,7 +6244,8 @@ async function proxyRequest(req, res, apiBase, payFetch, options, routerOpts, de
       cost: costWithBuffer,
       baselineCost: baselineWithBuffer,
       savings: accurateCosts.savings,
-      latencyMs: Date.now() - startTime
+      latencyMs: Date.now() - startTime,
+      ...responseInputTokens !== void 0 && { inputTokens: responseInputTokens }
     };
     logUsage(entry).catch(() => {
     });