diff --git a/langfuse/openai.py b/langfuse/openai.py index 7dea644d9..c14f7f1fb 100644 --- a/langfuse/openai.py +++ b/langfuse/openai.py @@ -791,6 +791,9 @@ def _wrap( model=model, output=completion, usage_details=usage, + cost_details=_parse_cost(openai_response.usage) + if hasattr(openai_response, "usage") + else None, ).end() return openai_response @@ -855,6 +858,9 @@ async def _wrap_async( output=completion, usage=usage, # backward compat for all V2 self hosters usage_details=usage, + cost_details=_parse_cost(openai_response.usage) + if hasattr(openai_response, "usage") + else None, ).end() return openai_response