phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

Merge pull request #695 from SlapDrone/set-langfuse-release

Browse source
This commit is contained in:
Krish Dholakia 2023-10-25 08:18:51 -07:00 • committed by GitHub
commit 4e3ca95a72
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
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