From 4f94ec7659aaff8aad7393071b3c2a2674cbf612 Mon Sep 17 00:00:00 2001
From: Marc Klingen <git@marcklingen.com>
Date: Wed, 10 Jan 2024 01:46:32 +0100
Subject: [PATCH] feat(langfuse): add sdk_integration for analytics and
 debugging

---
 litellm/integrations/langfuse.py | 20 +++++++++++++-------
 1 file changed, 13 insertions(+), 7 deletions(-)

diff --git a/litellm/integrations/langfuse.py b/litellm/integrations/langfuse.py
index db410d986f5..c275e00a183 100644
--- a/litellm/integrations/langfuse.py
+++ b/litellm/integrations/langfuse.py
@@ -25,13 +25,19 @@ class LangFuseLogger:
         self.langfuse_host = os.getenv("LANGFUSE_HOST", "https://cloud.langfuse.com")
         self.langfuse_release = os.getenv("LANGFUSE_RELEASE")
         self.langfuse_debug = os.getenv("LANGFUSE_DEBUG")
-        self.Langfuse = Langfuse(
-            public_key=self.public_key,
-            secret_key=self.secret_key,
-            host=self.langfuse_host,
-            release=self.langfuse_release,
-            debug=self.langfuse_debug,
-        )
+
+        parameters = {
+            "public_key": self.public_key,
+            "secret_key": self.secret_key,
+            "host": self.langfuse_host,
+            "release": self.langfuse_release,
+            "debug": self.langfuse_debug,
+        }
+
+        if Version(langfuse.version.__version__) >= Version("2.6.0"):
+            parameters["sdk_integration"] = "litellm"
+        
+        self.Langfuse = Langfuse(**parameters)
 
     def log_event(
         self, kwargs, response_obj, start_time, end_time, user_id, print_verbose