apunkt/plano
1
0
Fork 0
mirror of https://github.com/katanemo/plano.git synced 2026-06-17 15:25:17 +02:00
Code Issues Projects Releases Packages Wiki Activity

Update cli and related utils

Browse source
This commit is contained in:
Shuguang Chen 2024-12-08 11:16:34 -08:00
parent b4f4695f16
commit 320f4612b8
6 changed files with 267 additions and 215 deletions
Download patch file Download diff file Expand all files Collapse all files

8
model_server/tests/test_app.py
View file

@ -8,7 +8,7 @@ from src.main import app
client = TestClient(app)
# [TODO] Review: check the following code
# [TODO] Review: check the following code. Seems something wrong with asyncio package❗
# Unit tests for the health check endpoint
@pytest.mark.asyncio
async def test_healthz():
@ -17,7 +17,7 @@ async def test_healthz():
assert response.json() == {"status": "ok"}
# [TODO] Review: check the following code
# [TODO] Review: check the following code. Seems something wrong with asyncio package❗
# Unit test for the models endpoint
@pytest.mark.asyncio
async def test_models():
@ -27,7 +27,7 @@ async def test_models():
assert len(response.json()["data"]) > 0
# [TODO] Review: check the following code
# [TODO] Review: check the following code. Seems something wrong with asyncio package❗
# Unit test for the guardrail endpoint
@pytest.mark.asyncio
async def test_guardrail_endpoint():
@ -37,7 +37,7 @@ async def test_guardrail_endpoint():
assert "jailbreak_verdict" in response.json()
# [TODO] Review: check the following code
# [TODO] Review: check the following code. Seems something wrong with asyncio package❗
# Unit test for the function calling endpoint
@pytest.mark.asyncio
async def test_function_calling_endpoint():