ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added typed dict for LangfuseUsageDetails

Browse Source
This commit is contained in:
Fabricio Ceschin 2025-05-21 10:43:08 -04:00
parent 8f785e9a1f
commit d47ad69a1b
2 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
litellm/integrations/langfuse/langfuse.py
View File

@ -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),

7
litellm/types/integrations/langfuse.py
View File

@ -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]