From 51bbad1afd9ab82551d7e75ea49648df57a06ac3 Mon Sep 17 00:00:00 2001
From: Hassieb Pakzad <68423100+hassiebp@users.noreply.github.com>
Date: Tue, 2 Sep 2025 16:14:57 +0200
Subject: [PATCH] fix(openai): openrouter non-streamed cost tracking

---
 langfuse/openai.py | 6 ++++++
 1 file changed, 6 insertions(+)

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