Spaces:
Configuration error
Configuration error
File size: 2,002 Bytes
447ebeb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 |
import inspect
import json
import os
import sys
from unittest import mock
import httpx
import pytest
import respx
from fastapi.testclient import TestClient
sys.path.insert(0, os.path.abspath("../..")) #
import importlib
import litellm
from litellm import constants
def test_all_numeric_constants_can_be_overridden():
"""
Test that all integer and float constants in constants.py can be overridden with environment variables.
This ensures that any new constants added in the future will be configurable via environment variables.
"""
# Get all attributes from the constants module
constants_attributes = inspect.getmembers(constants)
# Filter for uppercase constants (by convention) that are integers or floats
numeric_constants = [
(name, value)
for name, value in constants_attributes
if name.isupper() and isinstance(value, (int, float))
]
# Ensure we found some constants to test
assert len(numeric_constants) > 0, "No numeric constants found to test"
print("all numeric constants", json.dumps(numeric_constants, indent=4))
# Verify all numeric constants have environment variable support
for name, value in numeric_constants:
# Skip constants that are not meant to be overridden (if any)
if name.startswith("_"):
continue
# Create a test value that's different from the default
test_value = value + 1 if isinstance(value, int) else value + 0.1
# Set the environment variable
with mock.patch.dict(os.environ, {name: str(test_value)}):
print("overriding", name, "with", test_value)
importlib.reload(constants)
# Get the new value after reload
new_value = getattr(constants, name)
# Verify the value was overridden
assert (
new_value == test_value
), f"Failed to override {name} with environment variable. Expected {test_value}, got {new_value}"
|