From c269c653719f2848c48aa7d03560efbe9b6e4835 Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Sat, 30 Dec 2023 20:25:23 +0530
Subject: [PATCH] (docs) langfuse log trace id, trace user id

---
 .../observability/langfuse_integration.md     | 37 +++++++++++++++++++
 1 file changed, 37 insertions(+)

diff --git a/docs/my-website/docs/observability/langfuse_integration.md b/docs/my-website/docs/observability/langfuse_integration.md
index 97600affc..3de426ec3 100644
--- a/docs/my-website/docs/observability/langfuse_integration.md
+++ b/docs/my-website/docs/observability/langfuse_integration.md
@@ -57,6 +57,8 @@ response = litellm.completion(
 ## Advanced
 ### Set Custom Generation names, pass metadata
 
+Pass `generation_name` in `metadata`
+
 ```python
 import litellm
 from litellm import completion
@@ -91,6 +93,41 @@ print(response)
 
 ```
 
+### Set Custom Trace ID, Trace User ID
+
+Pass `trace_id`, `trace_user_id` in `metadata`
+
+```python
+import litellm
+from litellm import completion
+import os
+
+# from https://cloud.langfuse.com/
+os.environ["LANGFUSE_PUBLIC_KEY"] = ""
+os.environ["LANGFUSE_SECRET_KEY"] = ""
+
+os.environ['OPENAI_API_KEY']=""
+
+# set langfuse as a callback, litellm will send the data to langfuse
+litellm.success_callback = ["langfuse"] 
+
+# set custom langfuse trace params and generation params
+response = completion(
+  model="gpt-3.5-turbo",
+  messages=[
+    {"role": "user", "content": "Hi 👋 - i'm openai"}
+  ],
+  metadata={
+      "generation_name": "ishaan-test-generation",  # set langfuse Generation Name
+      "generation_id": "gen-id22",                  # set langfuse Generation ID 
+      "trace_id": "trace-id22",                     # set langfuse Trace ID
+      "trace_user_id": "user-id2",                  # set langfuse Trace User ID
+  },
+)
+
+print(response)
+
+```
 
 
 ## Troubleshooting & Errors