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