From a1e2a2e540cdc7bf0a35df28bc41b12d8766a50d Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Fri, 19 Apr 2024 15:48:00 -0700 Subject: [PATCH] feat - log base_url to langfuse as a tag --- litellm/integrations/langfuse.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/litellm/integrations/langfuse.py b/litellm/integrations/langfuse.py index 3b13446a6a..38536f55c4 100644 --- a/litellm/integrations/langfuse.py +++ b/litellm/integrations/langfuse.py @@ -300,6 +300,11 @@ 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}") + api_base = litellm_params.get("api_base", None) if api_base: clean_metadata["api_base"] = api_base