From c59051895b4be9cef3a9dfe687a2e466ecf114e9 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Fri, 19 Apr 2024 16:05:31 -0700 Subject: [PATCH] fix - allow users to opt into langfuse default tags --- litellm/integrations/langfuse.py | 26 +++++++++++++++----------- 1 file changed, 15 insertions(+), 11 deletions(-) diff --git a/litellm/integrations/langfuse.py b/litellm/integrations/langfuse.py index 38536f55c46..3322fd4c474 100644 --- a/litellm/integrations/langfuse.py +++ b/litellm/integrations/langfuse.py @@ -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,10 +300,14 @@ 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: - tags.append(f"proxy_base_url:{proxy_base_url}") + 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) if api_base: