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

feat(langfuse): add sdk_integration for analytics and debugging

Browse source
This commit is contained in:
Marc Klingen 2024-01-10 01:46:32 +01:00 • committed by GitHub
parent 0862157905
commit 9d559dc3a3
1 changed files with 13 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

20
litellm/integrations/langfuse.py
View file

@ -25,13 +25,19 @@ class LangFuseLogger:
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,
release=self.langfuse_release,
debug=self.langfuse_debug,
)
parameters = {
"public_key": self.public_key,
"secret_key": self.secret_key,
"host": self.langfuse_host,
"release": self.langfuse_release,
"debug": self.langfuse_debug,
}
if Version(langfuse.version.__version__) >= Version("2.6.0"):
parameters["sdk_integration"] = "litellm"
self.Langfuse = Langfuse(**parameters)
def log_event(
self, kwargs, response_obj, start_time, end_time, user_id, print_verbose