diff --git a/examples/filters/langfuse_filter_pipeline.py b/examples/filters/langfuse_filter_pipeline.py index 2d73e5de..ab930305 100644 --- a/examples/filters/langfuse_filter_pipeline.py +++ b/examples/filters/langfuse_filter_pipeline.py @@ -266,8 +266,8 @@ async def outlet(self, body: dict, user: Optional[dict] = None) -> dict: if assistant_message_obj: info = assistant_message_obj.get("usage", {}) 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") + input_tokens = info.get("prompt_eval_count") or info.get("prompt_tokens") or info.get("input_tokens") + output_tokens = info.get("eval_count") or info.get("completion_tokens") or info.get("output_tokens") if input_tokens is not None and output_tokens is not None: usage = { "input": input_tokens, diff --git a/examples/filters/langfuse_v3_filter_pipeline.py b/examples/filters/langfuse_v3_filter_pipeline.py index a046eeea..47ba5cdb 100644 --- a/examples/filters/langfuse_v3_filter_pipeline.py +++ b/examples/filters/langfuse_v3_filter_pipeline.py @@ -317,8 +317,8 @@ async def outlet(self, body: dict, user: Optional[dict] = None) -> dict: if assistant_message_obj: info = assistant_message_obj.get("usage", {}) 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") + input_tokens = info.get("prompt_eval_count") or info.get("prompt_tokens") or info.get("input_tokens") + output_tokens = info.get("eval_count") or info.get("completion_tokens") or info.get("output_tokens") if input_tokens is not None and output_tokens is not None: usage = { "input": input_tokens,