scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-10 03:28:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 85

remove test

Browse source
This commit is contained in:
Sameer Kankute 2025-08-21 00:32:16 +05:30
parent 1d1d637ac0
commit 5b5a2d12a8
1 changed files with 0 additions and 29 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
tests/test_litellm/llms/deepinfra/test_deepinfra_rerank_transformation.py
View file

@ -42,35 +42,6 @@ class TestDeepinfraRerankTransform:
with pytest.raises(ValueError, match="Deepinfra API Base is required"):
self.config.get_complete_url(None, model)
def test_validate_environment(self):
"""Test environment validation with API key."""
# Test with API key provided
headers = self.config.validate_environment(
headers={}, model="test", api_key="test_key"
)
assert "Authorization" in headers
assert headers["Authorization"] == "Bearer test_key"
assert headers["accept"] == "application/json"
assert headers["content-type"] == "application/json"
# Test headers override
custom_headers = {"custom": "header", "Authorization": "Bearer custom_key"}
headers = self.config.validate_environment(
headers=custom_headers, model="test", api_key="test_key"
)
assert headers["custom"] == "header"
assert (
headers["Authorization"] == "Bearer custom_key"
) # Custom auth should override
# Test without API key should use environment variable (DEEPINFRA_API_KEY from .env)
headers = self.config.validate_environment(
headers={}, model="test", api_key=None
)
assert "Authorization" in headers
assert headers["Authorization"].startswith("Bearer ")
assert headers["accept"] == "application/json"
assert headers["content-type"] == "application/json"
def test_map_cohere_rerank_params_basic(self):
"""Test basic parameter mapping for DeepInfra rerank."""