Skip to content

Commit 0854815

Browse files
authored
fix(openai): openrouter non-streamed cost tracking (#1319)
1 parent a23a34f commit 0854815

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

langfuse/openai.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -791,6 +791,9 @@ def _wrap(
791791
model=model,
792792
output=completion,
793793
usage_details=usage,
794+
cost_details=_parse_cost(openai_response.usage)
795+
if hasattr(openai_response, "usage")
796+
else None,
794797
).end()
795798

796799
return openai_response
@@ -855,6 +858,9 @@ async def _wrap_async(
855858
output=completion,
856859
usage=usage, # backward compat for all V2 self hosters
857860
usage_details=usage,
861+
cost_details=_parse_cost(openai_response.usage)
862+
if hasattr(openai_response, "usage")
863+
else None,
858864
).end()
859865

860866
return openai_response

0 commit comments

Comments
 (0)