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

24
litellm/integrations/langfuse.py
View file

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