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

fix - allow users to opt into langfuse default tags

Browse source
This commit is contained in:
Ishaan Jaff 2024-04-19 16:05:31 -07:00
parent c5b077a02e
commit 57fd49d1e8
1 changed files with 15 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

26
litellm/integrations/langfuse.py
View file

@ -280,13 +280,13 @@ class LangFuseLogger:
clean_metadata = {} clean_metadata = {}
if isinstance(metadata, dict): if isinstance(metadata, dict):
for key, value in metadata.items(): for key, value in metadata.items():
# generate langfuse tags
if key in [ # generate langfuse tags - Default Tags sent to Langfuse from LiteLLM Proxy
"user_api_key_alias", if (
"user_api_key_user_id", litellm._langfuse_default_tags is not None
"user_api_key_team_alias", and isinstance(litellm._langfuse_default_tags, list)
"semantic-similarity", and key in litellm._langfuse_default_tags
]: ):
tags.append(f"{key}:{value}") tags.append(f"{key}:{value}")
# clean litellm metadata before logging # clean litellm metadata before logging
@ -300,10 +300,14 @@ class LangFuseLogger:
else: else:
clean_metadata[key] = value clean_metadata[key] = value
# if user has set proxy_base_url in env -> log to langfuse as a tag if (
proxy_base_url = os.getenv("PROXY_BASE_URL", None) litellm._langfuse_default_tags is not None
if proxy_base_url is not None: and isinstance(litellm._langfuse_default_tags, list)
tags.append(f"proxy_base_url:{proxy_base_url}") and "proxy_base_url" in litellm._langfuse_default_tags
):
proxy_base_url = os.environ.get("PROXY_BASE_URL", None)
if proxy_base_url:
tags.append(f"proxy_base_url:{proxy_base_url}")
api_base = litellm_params.get("api_base", None) api_base = litellm_params.get("api_base", None)
if api_base: if api_base: