From d1613e8ad54c30a9e05c7e7a239d9b893b5ba7db Mon Sep 17 00:00:00 2001 From: Fresent <102426695+fresent@users.noreply.github.com> Date: Thu, 14 May 2026 08:28:18 +0530 Subject: [PATCH] Fix issue with Ollama usage not being tracked and shown in 9router UI (#1102) Co-authored-by: Abhi --- open-sse/utils/usageTracking.js | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/open-sse/utils/usageTracking.js b/open-sse/utils/usageTracking.js index a8d676c..aed4111 100644 --- a/open-sse/utils/usageTracking.js +++ b/open-sse/utils/usageTracking.js @@ -220,6 +220,16 @@ export function extractUsage(chunk) { }); } + // Ollama NDJSON format (raw from provider, before translation) + // Ollama sends: {"model":"...","done":true,"prompt_eval_count":N,"eval_count":M} + if (chunk.done === true && typeof chunk.prompt_eval_count === "number") { + return normalizeUsage({ + prompt_tokens: chunk.prompt_eval_count || 0, + completion_tokens: chunk.eval_count || 0, + total_tokens: (chunk.prompt_eval_count || 0) + (chunk.eval_count || 0) + }); + } + return null; }