From d47ad69a1b2d006fbc15acea12bc47cc9d330bc1 Mon Sep 17 00:00:00 2001
From: Fabricio Ceschin <fceschin@todyl.com>
Date: Wed, 21 May 2025 10:43:08 -0400
Subject: [PATCH] Added typed dict for LangfuseUsageDetails

---
 litellm/integrations/langfuse/langfuse.py | 2 +-
 litellm/types/integrations/langfuse.py    | 7 +++++++
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/litellm/integrations/langfuse/langfuse.py b/litellm/integrations/langfuse/langfuse.py
index f4d1a1f79c..17e15e0b03 100644
--- a/litellm/integrations/langfuse/langfuse.py
+++ b/litellm/integrations/langfuse/langfuse.py
@@ -659,7 +659,7 @@ class LangFuseLogger:
                         "completion_tokens": _usage_obj.completion_tokens,
                         "total_cost": cost if self._supports_costs() else None,
                     }
-                    usage_details = {
+                    usage_details : Optional[LangfuseUsageDetails] = {
                         "input": _usage_obj.prompt_tokens,
                         "output": _usage_obj.completion_tokens,
                         "cache_creation_input_tokens": _usage_obj.get('cache_creation_input_tokens', 0),
diff --git a/litellm/types/integrations/langfuse.py b/litellm/types/integrations/langfuse.py
index ecf42d8cd4..c19cd98d16 100644
--- a/litellm/types/integrations/langfuse.py
+++ b/litellm/types/integrations/langfuse.py
@@ -5,3 +5,10 @@ class LangfuseLoggingConfig(TypedDict):
     langfuse_secret: Optional[str]
     langfuse_public_key: Optional[str]
     langfuse_host: Optional[str]
+
+
+class LangfuseUsageDetails(TypedDict):
+    input: Optional[int]
+    output: Optional[int]
+    cache_creation_input_tokens: Optional[int]
+    cache_read_input_tokens: Optional[int]