vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

test_invalid_input_extra_req

Browse source
This commit is contained in:
Xuan Son Nguyen 2024-12-08 22:58:58 +01:00
parent 055aa9e2ea
commit 3a81c60698
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
examples/server/tests/unit/test_infill.py
View file

@ -48,9 +48,9 @@ def test_invalid_input_extra_req(input_extra):
global server global server
server.start() server.start()
res = server.make_request("POST", "/infill", data={ res = server.make_request("POST", "/infill", data={
"prompt": "Complete this",
"input_extra": [input_extra], "input_extra": [input_extra],
"input_prefix": "#include <cstdio>\n#include \"llama.h\"\n\nint main() {\n int n_threads = llama_", "input_prefix": "#include <cstdio>\n#include \"llama.h\"\n\nint main() {\n",
"prompt": " int n_threads = llama_",
"input_suffix": "}\n", "input_suffix": "}\n",
}) })
assert res.status_code == 400 assert res.status_code == 400