From f4f834489c438b3bace30b68306431543d3ef81b Mon Sep 17 00:00:00 2001 From: shivam Date: Thu, 26 Feb 2026 04:04:20 -0800 Subject: [PATCH] added variable for invitation link --- litellm/__init__.py | 3 ++- litellm/proxy/proxy_server.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/litellm/__init__.py b/litellm/__init__.py index f949046e4d3..7277f4e7d5e 100644 --- a/litellm/__init__.py +++ b/litellm/__init__.py @@ -16,7 +16,8 @@ import os # Load .env before any other litellm imports so env vars (e.g. LITELLM_UI_SESSION_DURATION) are available import dotenv as _dotenv -_dotenv.load_dotenv() +if os.getenv("LITELLM_MODE", "DEV") == "DEV": + _dotenv.load_dotenv() from typing import ( Callable, diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 607306f3806..2ed40213fb7 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -50,6 +50,7 @@ from litellm.constants import ( LITELLM_EMBEDDING_PROVIDERS_SUPPORTING_INPUT_ARRAY_OF_TOKENS, LITELLM_SETTINGS_SAFE_DB_OVERRIDES, LITELLM_UI_ALLOW_HEADERS, + LITELLM_UI_SESSION_DURATION, ) from litellm.litellm_core_utils.litellm_logging import ( _init_custom_logger_compatible_class, @@ -10767,7 +10768,7 @@ async def onboarding(invite_link: str, request: Request): request_type="key", **{ "user_role": user_obj.user_role, - "duration": "24hr", + "duration": LITELLM_UI_SESSION_DURATION, "key_max_budget": litellm.max_ui_session_budget, "models": [], "aliases": {},