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

fix(langfuse.py): cleanup

Browse source
This commit is contained in:
Krrish Dholakia 2024-06-24 21:43:40 -07:00
parent 93dbdf6d92
commit f8b390d421
1 changed files with 19 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

38
litellm/integrations/langfuse.py
View file

@ -36,9 +36,9 @@ class LangFuseLogger:
self.langfuse_debug = os.getenv("LANGFUSE_DEBUG") self.langfuse_debug = os.getenv("LANGFUSE_DEBUG")
parameters = { parameters = {
"public_key": "pk-lf-a65841e9-5192-4397-a679-cfff029fd5b0", "public_key": self.public_key,
"secret_key": "sk-lf-d58c2891-3717-4f98-89dd-df44826215fd", "secret_key": self.secret_key,
"host": "https://us.cloud.langfuse.com", "host": self.langfuse_host,
"release": self.langfuse_release, "release": self.langfuse_release,
"debug": self.langfuse_debug, "debug": self.langfuse_debug,
"flush_interval": flush_interval, # flush interval in seconds "flush_interval": flush_interval, # flush interval in seconds
@ -311,22 +311,22 @@ class LangFuseLogger:
try: try:
tags = [] tags = []
# try: try:
# metadata = copy.deepcopy( metadata = copy.deepcopy(
# metadata metadata
# ) # Avoid modifying the original metadata ) # Avoid modifying the original metadata
# except: except:
new_metadata = {} new_metadata = {}
for key, value in metadata.items(): for key, value in metadata.items():
if ( if (
isinstance(value, list) isinstance(value, list)
or isinstance(value, dict) or isinstance(value, dict)
or isinstance(value, str) or isinstance(value, str)
or isinstance(value, int) or isinstance(value, int)
or isinstance(value, float) or isinstance(value, float)
): ):
new_metadata[key] = copy.deepcopy(value) new_metadata[key] = copy.deepcopy(value)
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")
supports_prompt = Version(langfuse.version.__version__) >= Version("2.7.3") supports_prompt = Version(langfuse.version.__version__) >= Version("2.7.3")