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

test: use local model cost map in import-isolation subprocess

Browse source
This commit is contained in:
mateo-berri 2026-08-30 12:47:15 -07:00
parent f0a2a23127
commit b8e11a75fa
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/test_litellm/proxy/hooks/litellm_skills/test_main.py
View file

@ -1,3 +1,4 @@
import os
import subprocess
import sys
from typing import Final
@ -78,6 +79,9 @@ def test_importing_proxy_hooks_does_not_mutate_litellm_callbacks():
"assert len(litellm.callbacks) == 0, f'mutated: {litellm.callbacks}'"
)
result: Final = subprocess.run(
[sys.executable, "-c", script], capture_output=True, text=True
[sys.executable, "-c", script],
capture_output=True,
text=True,
env={**os.environ, "LITELLM_LOCAL_MODEL_COST_MAP": "True"},
)
assert result.returncode == 0, f"stdout: {result.stdout}\nstderr: {result.stderr}"