fix startup test

This commit is contained in:
Ishaan Jaff 2024-11-20 17:26:42 -08:00
parent 9e040e12fc
commit 8334cdc3cf

View file

@ -16,13 +16,13 @@ async def test_health_and_chat_completion():
async with session.get("http://0.0.0.0:4000/health/readiness") as response:
assert response.status == 200
readiness_response = await response.json()
assert readiness_response["status"] == "OK"
assert readiness_response["status"] == "connected"
# Test liveness endpoint
async with session.get("http://0.0.0.0:4000/health/liveness") as response:
assert response.status == 200
liveness_response = await response.json()
assert liveness_response["status"] == "OK"
print("liveness_response", liveness_response)
# Make a chat completion call
url = "http://0.0.0.0:4000/chat/completions"