added variable for invitation link

This commit is contained in:
shivam 2026-02-26 04:04:20 -08:00
parent a7f5163976
commit f4f834489c
2 changed files with 4 additions and 2 deletions

View file

@ -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,

View file

@ -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": {},