From 789ce6b7476cd5e92da81ed240005aca205c8d20 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Fri, 27 Sep 2024 17:42:15 -0700 Subject: [PATCH] allow setting LANGFUSE_FLUSH_INTERVAL (#5944) --- litellm/integrations/langfuse.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/litellm/integrations/langfuse.py b/litellm/integrations/langfuse.py index 0819cc703a0..d2343d42995 100644 --- a/litellm/integrations/langfuse.py +++ b/litellm/integrations/langfuse.py @@ -44,6 +44,9 @@ class LangFuseLogger: self.langfuse_host = "http://" + self.langfuse_host self.langfuse_release = os.getenv("LANGFUSE_RELEASE") self.langfuse_debug = os.getenv("LANGFUSE_DEBUG") + self.langfuse_flush_interval = ( + os.getenv("LANGFUSE_FLUSH_INTERVAL") or flush_interval + ) parameters = { "public_key": self.public_key, @@ -51,7 +54,7 @@ class LangFuseLogger: "host": self.langfuse_host, "release": self.langfuse_release, "debug": self.langfuse_debug, - "flush_interval": flush_interval, # flush interval in seconds + "flush_interval": self.langfuse_flush_interval, # flush interval in seconds } if Version(langfuse.version.__version__) >= Version("2.6.0"):