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

conditionally set sdk_integration

Browse source
This commit is contained in:
Marc Klingen 2024-03-24 16:46:31 +01:00
parent 2d3eb57e17
commit c28ae655fc
1 changed files with 13 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

22
litellm/integrations/langfuse.py
View file

@ -28,15 +28,19 @@ class LangFuseLogger:
self.langfuse_release = os.getenv("LANGFUSE_RELEASE") self.langfuse_release = os.getenv("LANGFUSE_RELEASE")
self.langfuse_debug = os.getenv("LANGFUSE_DEBUG") self.langfuse_debug = os.getenv("LANGFUSE_DEBUG")
self.Langfuse = Langfuse( parameters = {
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, "release": self.langfuse_release,
debug=self.langfuse_debug, "debug": self.langfuse_debug,
flush_interval=1, # flush interval in seconds "flush_interval": 1,
sdk_integration="litellm", }
)
if Version(langfuse.version.__version__) >= Version("2.6.0"):
parameters["sdk_integration"] = "litellm"
self.Langfuse = Langfuse(**parameters)
if os.getenv("UPSTREAM_LANGFUSE_SECRET_KEY") is not None: if os.getenv("UPSTREAM_LANGFUSE_SECRET_KEY") is not None:
self.upstream_langfuse_secret_key = os.getenv( self.upstream_langfuse_secret_key = os.getenv(