From aaacb91b7ceefcb558e2ccebc799f8924429bc51 Mon Sep 17 00:00:00 2001
From: Ryuichi Leo Takashige <leo@exolabs.net>
Date: Thu, 29 Jan 2026 21:26:05 +0000
Subject: [PATCH] format

---
 src/exo/worker/engines/mlx/generator/generate.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/exo/worker/engines/mlx/generator/generate.py b/src/exo/worker/engines/mlx/generator/generate.py
index 2afe033e..c392f6a8 100644
--- a/src/exo/worker/engines/mlx/generator/generate.py
+++ b/src/exo/worker/engines/mlx/generator/generate.py
@@ -271,8 +271,12 @@ def mlx_generate(
                 prompt_tokens=int(out.prompt_tokens),
                 completion_tokens=completion_tokens,
                 total_tokens=int(out.prompt_tokens) + completion_tokens,
-                prompt_tokens_details=PromptTokensDetails(cached_tokens=prefix_hit_length),
-                completion_tokens_details=CompletionTokensDetails(reasoning_tokens=reasoning_tokens),
+                prompt_tokens_details=PromptTokensDetails(
+                    cached_tokens=prefix_hit_length
+                ),
+                completion_tokens_details=CompletionTokensDetails(
+                    reasoning_tokens=reasoning_tokens
+                ),
             )
 
         yield GenerationResponse(