scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-07 08:26:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 36

test(proxy): type the env var regression test params and keep locals Final

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
Devin AI 2026-09-03 01:17:12 +00:00
parent f703b19f56
commit b3d8a79bd4
1 changed files with 7 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
tests/test_litellm/proxy/test_litellm_pre_call_utils.py
View file

@ -8,6 +8,7 @@ import textwrap
import time
from datetime import datetime, timezone
from types import SimpleNamespace
from typing import Final
from unittest.mock import AsyncMock, MagicMock, patch
import pytest
@ -4958,12 +4959,13 @@ def test_apply_overrides_feature_flag_disabled_by_default():
"env_value, expected",
[("true", True), ("True", True), ("false", False), (None, False)],
)
def test_credential_overrides_flag_reads_env_var_at_import(env_value, expected):
def test_credential_overrides_flag_reads_env_var_at_import(env_value: str | None, expected: bool):
"""LITELLM_ENABLE_MODEL_CONFIG_CREDENTIAL_OVERRIDES is read at import, so check it in a fresh interpreter."""
env = {k: v for k, v in os.environ.items() if k != "LITELLM_ENABLE_MODEL_CONFIG_CREDENTIAL_OVERRIDES"}
if env_value is not None:
env["LITELLM_ENABLE_MODEL_CONFIG_CREDENTIAL_OVERRIDES"] = env_value
result = subprocess.run(
base_env: Final = {k: v for k, v in os.environ.items() if k != "LITELLM_ENABLE_MODEL_CONFIG_CREDENTIAL_OVERRIDES"}
env: Final = (
base_env if env_value is None else {**base_env, "LITELLM_ENABLE_MODEL_CONFIG_CREDENTIAL_OVERRIDES": env_value}
)
result: Final = subprocess.run(
[
sys.executable,
"-c",