feat(output_tokens_details): include also output_tokens_details

Signed-off-by: Anna Warno <[email protected]>

Author: AWarno

packages/nemo-evaluator/src/nemo_evaluator/adapters/interceptors/reasoning_interceptor.py

@@ -340,19 +340,30 @@ def _process_reasoning_message(
             else:
                 reasoning_finished = True
             if usage:
+                # First try to get reasoning_tokens directly from usage
                 reasoning_tokens = usage.get("reasoning_tokens", "unknown")
                 updated_content_tokens = usage.get("content_tokens", "unknown")
 
-                # Check if reasoning_tokens is in completion_tokens_details
-                if (
-                    reasoning_tokens == "unknown"
-                    and "completion_tokens_details" in usage
-                ):
-                    completion_details = usage["completion_tokens_details"]
-                    if isinstance(completion_details, dict):
-                        reasoning_tokens = completion_details.get(
-                            "reasoning_tokens", "unknown"
-                        )
+                # If not found, check in completion_tokens_details and output_tokens_details
+                if reasoning_tokens == "unknown":
+                    for key in ["completion_tokens_details", "output_tokens_details"]:
+                        if key in usage:
+                            details = usage[key]
+                            if isinstance(details, dict):
+                                reasoning_tokens = details.get(
+                                    "reasoning_tokens", "unknown"
+                                )
+                                if reasoning_tokens != "unknown":
+                                    self.logger.debug(
+                                        f"Found reasoning_tokens in {key}: {reasoning_tokens}"
+                                    )
+                                    break
+
+                # Log if reasoning tokens were found
+                if reasoning_tokens != "unknown":
+                    self.logger.debug(f"Reasoning tokens extracted: {reasoning_tokens}")
+                else:
+                    self.logger.debug("No reasoning tokens found in usage data")
 
         else:
             reasoning_finished = False

packages/nemo-evaluator/tests/unit_tests/adapters/interceptors/test_reasoning.py

@@ -1004,6 +1004,15 @@ def test_cached_response_reasoning_behavior(
             20,
             40,
         ),
+        # Format 3: reasoning_tokens in output_tokens_details
+        (
+            {
+                "output_tokens_details": {"reasoning_tokens": 25},
+                "content_tokens": 50,
+            },
+            25,
+            50,
+        ),
     ],
 )
 def test_reasoning_tokens_different_formats(