From 2b4fb953f8ecfa284253f12a31493967d1451b82 Mon Sep 17 00:00:00 2001 From: Rajat Ahuja Date: Mon, 26 Jan 2026 14:36:00 -0500 Subject: [PATCH] fix: RepresentationCompletedEvent -> input tokens to messages tokens (#349) --- src/deriver/deriver.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/deriver/deriver.py b/src/deriver/deriver.py index f6a4e523..bf786761 100644 --- a/src/deriver/deriver.py +++ b/src/deriver/deriver.py @@ -243,7 +243,7 @@ async def process_representation_tasks_batch( context_preparation_ms=context_prep_duration, llm_call_ms=llm_duration, total_duration_ms=overall_duration, - input_tokens=response.input_tokens, + input_tokens=messages_tokens, output_tokens=response.output_tokens, ) )