scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-07 08:26:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 36

Merge pull request #2715 from andreaponti5/langfuse-trace-name

Browse source 
Add `trace_name` in Langfuse logging
This commit is contained in:
Krish Dholakia 2024-03-28 14:59:32 -07:00 • committed by GitHub
commit 90de9d3c10
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 10 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
litellm/integrations/langfuse.py
View file

@ -246,13 +246,13 @@ class LangFuseLogger:
metadata_tags = metadata.get("tags", [])
tags = metadata_tags
generation_name = metadata.get("generation_name", None)
if generation_name is None:
# just log `litellm-{call_type}` as the generation name
generation_name = f"litellm-{kwargs.get('call_type', 'completion')}"
trace_name = metadata.get("trace_name", None)
if trace_name is None:
# just log `litellm-{call_type}` as the trace name
trace_name = f"litellm-{kwargs.get('call_type', 'completion')}"
trace_params = {
"name": generation_name,
"name": trace_name,
"input": input,
"user_id": metadata.get("trace_user_id", user_id),
"id": metadata.get("trace_id", None),
@ -311,6 +311,11 @@ class LangFuseLogger:
"completion_tokens": response_obj["usage"]["completion_tokens"],
"total_cost": cost if supports_costs else None,
}
generation_name = metadata.get("generation_name", None)
if generation_name is None:
# just log `litellm-{call_type}` as the generation name
generation_name = f"litellm-{kwargs.get('call_type', 'completion')}"
generation_params = {
"name": generation_name,
"id": metadata.get("generation_id", generation_id),