MK-LLM-Mistral / tests /test_api.py
ainow-mk's picture
Upload 65 files
f29d474 verified
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", []))