From ce94f406f440b7fd840a446c50d7734b1bc1d86c Mon Sep 17 00:00:00 2001
From: redshift <213178690+1ftredsh@users.noreply.github.com>
Date: Sat, 14 Mar 2026 15:22:57 +0000
Subject: [PATCH] data usage alternative format

---
 src/daemon.ts | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

diff --git a/src/daemon.ts b/src/daemon.ts
index 96305cb..1fb9740 100644
--- a/src/daemon.ts
+++ b/src/daemon.ts
@@ -337,12 +337,19 @@ function createSSEParserTransform(
         onResponseId?.(responseId.trim());
       }
 
-      if (data.usage && !data.choices?.length) {
-        const cost = data.usage.cost?.total_usd ?? data.usage.cost ?? 0;
+      if (data.usage) {
+        const usageCost = data.usage.cost;
+        const cost =
+          typeof usageCost === "number"
+            ? usageCost
+            : usageCost?.total_usd ??
+              data.metadata?.routstr?.cost?.total_usd ??
+              0;
         const msats =
-          data.usage.cost?.total_msats ??
           data.metadata?.routstr?.cost?.total_msats ??
-          0;
+          (typeof data.usage.cost_sats === "number"
+            ? data.usage.cost_sats * 1000
+            : 0);
         onUsage({
           promptTokens: data.usage.prompt_tokens ?? 0,
           completionTokens: data.usage.completion_tokens ?? 0,