From bc688d46625bc9abbb07005c41eed63380320fcf Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Fri, 2 Feb 2024 14:03:18 -0800 Subject: [PATCH] fix(utils.py): dynamic switch langfuse client based on keys --- litellm/utils.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/litellm/utils.py b/litellm/utils.py index d27c658c23..eb124f9ba8 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -1297,7 +1297,10 @@ class Logging: else: print_verbose("reaches langfuse for streaming logging!") result = kwargs["complete_streaming_response"] - if langFuseLogger is None: + if langFuseLogger is None or ( + self.langfuse_public_key != langFuseLogger.public_key + and self.langfuse_secret != langFuseLogger.secret_key + ): langFuseLogger = LangFuseLogger( langfuse_public_key=self.langfuse_public_key, langfuse_secret=self.langfuse_secret,