From 8ae4554a8a5739197f7dd68588cc06ff9b4cf620 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Sat, 30 Dec 2023 20:19:03 +0530 Subject: [PATCH] (feat) langfuse - set custom trace_id, trace_user_id --- litellm/integrations/langfuse.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/litellm/integrations/langfuse.py b/litellm/integrations/langfuse.py index d418aad2a..ac2fc6489 100644 --- a/litellm/integrations/langfuse.py +++ b/litellm/integrations/langfuse.py @@ -170,11 +170,13 @@ class LangFuseLogger: name=metadata.get("generation_name", "litellm-completion"), input=input, output=output, - user_id=user_id, + user_id=metadata.get("trace_user_id", user_id), + id=metadata.get("trace_id", None), ) trace.generation( name=metadata.get("generation_name", "litellm-completion"), + id=metadata.get("generation_id", None), startTime=start_time, endTime=end_time, model=kwargs["model"],