From 1743c97fe95ec325bd2aa9600bc8a88e0194ae53 Mon Sep 17 00:00:00 2001 From: Hassieb Pakzad <68423100+hassiebp@users.noreply.github.com> Date: Thu, 15 May 2025 14:05:59 +0200 Subject: [PATCH] fix(openai): usage parsing for streamed responses --- langfuse/openai.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/langfuse/openai.py b/langfuse/openai.py index 552f8829b..f27fedcff 100644 --- a/langfuse/openai.py +++ b/langfuse/openai.py @@ -476,7 +476,10 @@ def _create_langfuse_update( update["metadata"] = metadata if usage is not None: - update["usage"] = _parse_usage(usage) + parsed_usage = _parse_usage(usage) + + update["usage"] = parsed_usage + update["usage_details"] = parsed_usage generation.update(**update)