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

fix test

Browse source
This commit is contained in:
cotran 2024-12-10 19:31:02 -08:00
parent 2405fb36e3
commit b2ef3f7266
4 changed files with 32 additions and 39 deletions
Download patch file Download diff file Expand all files Collapse all files

4
model_server/tests/test_app.py
View file

@ -34,7 +34,7 @@ async def test_guardrail_endpoint():
request_data = {"input": "Test for jailbreak and toxicity", "task": "jailbreak"}
response = client.post("/guardrails", json=request_data)
assert response.status_code == 200
assert "jailbreak_verdict" in response.json()
assert "response" in response.json()
# [TODO] Review: check the following code. Seems something wrong with asyncio package❗
@ -50,4 +50,4 @@ async def test_function_calling_endpoint():
}
response = await client.post("/function_calling", json=request_data)
assert response.status_code == 200
assert "choices" in response.json()
assert "result" in response.json()