phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-28 04:04:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

support release and debug params for langfuse client

Browse source
This commit is contained in:
Liam Moore 2023-10-25 16:12:14 +02:00
parent b4906b71a0
commit ff8a47aed3
1 changed files with 8 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

12
litellm/integrations/langfuse.py
View file

@ -19,11 +19,15 @@ class LangFuseLogger:
self.secret_key = os.getenv("LANGFUSE_SECRET_KEY") self.secret_key = os.getenv("LANGFUSE_SECRET_KEY")
self.public_key = os.getenv("LANGFUSE_PUBLIC_KEY") self.public_key = os.getenv("LANGFUSE_PUBLIC_KEY")
self.langfuse_host = os.getenv("LANGFUSE_HOST", "https://cloud.langfuse.com") 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( self.Langfuse = Langfuse(
public_key=self.public_key, public_key=self.public_key,
secret_key=self.secret_key, secret_key=self.secret_key,
host=self.langfuse_host, 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): def log_event(self, kwargs, response_obj, start_time, end_time, print_verbose):
# Method definition # Method definition