diff --git a/examples/filters/langfuse_filter_pipeline.py b/examples/filters/langfuse_filter_pipeline.py index 8c05f467..7789fdd1 100644 --- a/examples/filters/langfuse_filter_pipeline.py +++ b/examples/filters/langfuse_filter_pipeline.py @@ -239,10 +239,10 @@ async def outlet(self, body: dict, user: Optional[dict] = None) -> dict: # Extract usage if available usage = None if assistant_message_obj: - info = assistant_message_obj.get("info", {}) - if isinstance(info, dict): - input_tokens = info.get("prompt_eval_count") or info.get("prompt_tokens") - output_tokens = info.get("eval_count") or info.get("completion_tokens") + message_usage = assistant_message_obj.get("usage", {}) + if isinstance(message_usage, dict): + input_tokens = message_usage.get("prompt_eval_count") or message_usage.get("prompt_tokens") + output_tokens = message_usage.get("eval_count") or message_usage.get("completion_tokens") if input_tokens is not None and output_tokens is not None: usage = { "input": input_tokens,