phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(feat) langfuse - set custom trace_id, trace_user_id

Browse source
This commit is contained in:
ishaan-jaff 2023-12-30 20:19:03 +05:30
parent cc7b964433
commit 8ae4554a8a
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/integrations/langfuse.py
View file

@ -170,11 +170,13 @@ class LangFuseLogger:
name=metadata.get("generation_name", "litellm-completion"), name=metadata.get("generation_name", "litellm-completion"),
input=input, input=input,
output=output, output=output,
user_id=user_id, user_id=metadata.get("trace_user_id", user_id),
id=metadata.get("trace_id", None),
) )
trace.generation( trace.generation(
name=metadata.get("generation_name", "litellm-completion"), name=metadata.get("generation_name", "litellm-completion"),
id=metadata.get("generation_id", None),
startTime=start_time, startTime=start_time,
endTime=end_time, endTime=end_time,
model=kwargs["model"], model=kwargs["model"],