phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add trace_name in langfuse logging v2

Browse source
This commit is contained in:
Andrea Ponti 2024-03-27 16:58:08 +01:00
parent 90b859ebcb
commit 25bacfacdc
1 changed files with 10 additions and 5 deletions
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),