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

fix(langfuse.py): support 'existing_trace_id' param

Browse source 
allow user to call out a trace as pre-existing, this prevents creating a default trace name, and potentially overwriting past traces
This commit is contained in:
Krrish Dholakia 2024-04-29 16:39:01 -07:00
parent 2cf069befb
commit 853b70aba9
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/integrations/langfuse.py
View file

@ -266,7 +266,8 @@ class LangFuseLogger:
trace_name = metadata.get("trace_name", None) trace_name = metadata.get("trace_name", None)
trace_id = metadata.get("trace_id", None) trace_id = metadata.get("trace_id", None)
if trace_name is None and trace_id is None: existing_trace_id = metadata.get("existing_trace_id", None)
if trace_name is None and existing_trace_id is None:
# just log `litellm-{call_type}` as the trace name # just log `litellm-{call_type}` as the trace name
## DO NOT SET TRACE_NAME if trace-id set. this can lead to overwriting of past traces. ## DO NOT SET TRACE_NAME if trace-id set. this can lead to overwriting of past traces.
trace_name = f"litellm-{kwargs.get('call_type', 'completion')}" trace_name = f"litellm-{kwargs.get('call_type', 'completion')}"
@ -275,7 +276,7 @@ class LangFuseLogger:
"name": trace_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": trace_id, "id": trace_id or existing_trace_id,
"session_id": metadata.get("session_id", None), "session_id": metadata.get("session_id", None),
} }