From fe2b214b7aef7330c486d9f8b83e50cd00bad622 Mon Sep 17 00:00:00 2001 From: Liam Moore Date: Wed, 25 Oct 2023 16:12:14 +0200 Subject: [PATCH] support release and debug params for langfuse client --- litellm/integrations/langfuse.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/litellm/integrations/langfuse.py b/litellm/integrations/langfuse.py index a7dda442adff..75826ddfb006 100644 --- a/litellm/integrations/langfuse.py +++ b/litellm/integrations/langfuse.py @@ -19,11 +19,15 @@ def __init__(self): self.secret_key = os.getenv("LANGFUSE_SECRET_KEY") self.public_key = os.getenv("LANGFUSE_PUBLIC_KEY") 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, - ) + public_key=self.public_key, + secret_key=self.secret_key, + host=self.langfuse_host, + release=self.langfuse_release, + debug=self.langfuse_debug + ) def log_event(self, kwargs, response_obj, start_time, end_time, print_verbose): # Method definition