mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-15 04:19:08 +00:00
0734e35024
* test: delete 77 legacy tests covered by e2e, unable to fail, or dead in CI * test: keep router helper tests and coverage ignore list unchanged * test: keep cohere error handling tests --------- Co-authored-by: yuneng <yuneng@berri.ai>
84 lines
2.2 KiB
Python
84 lines
2.2 KiB
Python
import os
|
|
|
|
# What this tests?
|
|
## Tests /health + /routes endpoints.
|
|
|
|
import pytest
|
|
import asyncio
|
|
import aiohttp
|
|
|
|
|
|
async def health(session, call_key):
|
|
url = "http://0.0.0.0:4000/health"
|
|
headers = {
|
|
"Authorization": f"Bearer {call_key}",
|
|
"Content-Type": "application/json",
|
|
}
|
|
|
|
async with session.get(url, headers=headers) as response:
|
|
status = response.status
|
|
response_text = await response.text()
|
|
|
|
print(f"Response (Status code: {status}):")
|
|
print(response_text)
|
|
print()
|
|
|
|
if status != 200:
|
|
raise Exception(f"Request did not return a 200 status code: {status}")
|
|
|
|
return await response.json()
|
|
|
|
|
|
async def generate_key(session):
|
|
url = "http://0.0.0.0:4000/key/generate"
|
|
headers = {
|
|
"Authorization": "Bearer " + os.environ["LITELLM_MASTER_KEY"],
|
|
"Content-Type": "application/json",
|
|
}
|
|
data = {
|
|
"models": ["gpt-4", "text-embedding-ada-002", "gpt-image-1"],
|
|
"duration": None,
|
|
}
|
|
|
|
async with session.post(url, headers=headers, json=data) as response:
|
|
status = response.status
|
|
response_text = await response.text()
|
|
|
|
print(response_text)
|
|
print()
|
|
|
|
if status != 200:
|
|
raise Exception(f"Request did not return a 200 status code: {status}")
|
|
return await response.json()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_health():
|
|
"""
|
|
- Call /health
|
|
"""
|
|
async with aiohttp.ClientSession() as session:
|
|
# as admin #
|
|
all_healthy_models = await health(session=session, call_key=os.environ["LITELLM_MASTER_KEY"])
|
|
total_model_count = (
|
|
all_healthy_models["healthy_count"] + all_healthy_models["unhealthy_count"]
|
|
)
|
|
assert total_model_count > 0
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_routes():
|
|
"""
|
|
Check if 200
|
|
"""
|
|
async with aiohttp.ClientSession() as session:
|
|
url = "http://0.0.0.0:4000/routes"
|
|
async with session.get(url) as response:
|
|
status = response.status
|
|
response_text = await response.text()
|
|
|
|
print(response_text)
|
|
print()
|
|
|
|
if status != 200:
|
|
raise Exception(f"Request did not return a 200 status code: {status}")
|