File size: 346 Bytes
f29d474 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
from fastapi.testclient import TestClient
from inference.api import app
def test_models_list():
client = TestClient(app)
resp = client.get("/v1/models")
assert resp.status_code == 200
data = resp.json()
assert data.get("object") == "list"
assert any(m.get("id") == "mk-llm" for m in data.get("data", []))
|