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

fix(langfuse.py): cleanup

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-01 21:24:41 -07:00
parent d8778380d8
commit d096c6c0ee
1 changed files with 3 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

9
litellm/integrations/langfuse.py
View file

@ -5,7 +5,6 @@ import os
import traceback import traceback
from packaging.version import Version from packaging.version import Version
from pydantic import BaseModel
import litellm import litellm
from litellm._logging import verbose_logger from litellm._logging import verbose_logger
@ -44,8 +43,8 @@ class LangFuseLogger:
self.langfuse_debug = os.getenv("LANGFUSE_DEBUG") self.langfuse_debug = os.getenv("LANGFUSE_DEBUG")
parameters = { parameters = {
"public_key": "pk-lf-b3db7e8e-c2f6-4fc7-825c-a541a8fbe003", "public_key": self.public_key,
"secret_key": "sk-lf-b11ef3a8-361c-4445-9652-12318b8596e4", "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,
@ -332,7 +331,7 @@ class LangFuseLogger:
metadata = copy.deepcopy( metadata = copy.deepcopy(
metadata metadata
) # Avoid modifying the original metadata ) # Avoid modifying the original metadata
except Exception: except:
new_metadata = {} new_metadata = {}
for key, value in metadata.items(): for key, value in metadata.items():
if ( if (
@ -343,8 +342,6 @@ class LangFuseLogger:
or isinstance(value, float) or isinstance(value, float)
): ):
new_metadata[key] = copy.deepcopy(value) new_metadata[key] = copy.deepcopy(value)
elif isinstance(value, BaseModel):
new_metadata[key] = value.model_dump()
metadata = new_metadata metadata = new_metadata
supports_tags = Version(langfuse.version.__version__) >= Version("2.6.3") supports_tags = Version(langfuse.version.__version__) >= Version("2.6.3")