From ce13d2fa7c0957f53ed37576719aa52e3a54e1ec Mon Sep 17 00:00:00 2001
From: David Manouchehri <david.manouchehri@ai.moda>
Date: Thu, 11 Apr 2024 21:29:45 +0000
Subject: [PATCH] (feat) - Improve Langfuse to log request headers.

https://github.com/orgs/langfuse/discussions/1070
---
 litellm/integrations/langfuse.py | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/litellm/integrations/langfuse.py b/litellm/integrations/langfuse.py
index 56a6e894747..cef5900c42d 100644
--- a/litellm/integrations/langfuse.py
+++ b/litellm/integrations/langfuse.py
@@ -320,6 +320,24 @@ class LangFuseLogger:
                     clean_metadata["cache_hit"] = kwargs["cache_hit"]
                 trace_params.update({"tags": tags})
 
+            proxy_server_request = litellm_params.get("proxy_server_request", None)
+            if proxy_server_request:
+                method = proxy_server_request.get("method", None)
+                url = proxy_server_request.get("url", None)
+                headers = proxy_server_request.get("headers", None)
+                clean_headers = {}
+                if headers:
+                    for key, value in headers.items():
+                        # these headers can leak our API keys and/or JWT tokens
+                        if key.lower() not in ["authorization", "cookie", "referer"]:
+                            clean_headers[key] = value
+
+                clean_metadata["request"] = {
+                    "method": method,
+                    "url": url,
+                    "headers": clean_headers,
+                }
+
             print_verbose(f"trace_params: {trace_params}")
 
             trace = self.Langfuse.trace(**trace_params)