diff --git a/litellm/integrations/langfuse.py b/litellm/integrations/langfuse.py index ff84d171aef..af371964f1b 100644 --- a/litellm/integrations/langfuse.py +++ b/litellm/integrations/langfuse.py @@ -28,15 +28,19 @@ class LangFuseLogger: 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, - flush_interval=1, # flush interval in seconds - sdk_integration="litellm", - ) + parameters = { + "public_key": self.public_key, + "secret_key": self.secret_key, + "host": self.langfuse_host, + "release": self.langfuse_release, + "debug": self.langfuse_debug, + "flush_interval": 1, + } + + if Version(langfuse.version.__version__) >= Version("2.6.0"): + parameters["sdk_integration"] = "litellm" + + self.Langfuse = Langfuse(**parameters) if os.getenv("UPSTREAM_LANGFUSE_SECRET_KEY") is not None: self.upstream_langfuse_secret_key = os.getenv(