Skip to content

Commit bfa3219

Browse files
authored
server: add "verbose" field to schema (#24864)
1 parent d6d8995 commit bfa3219

2 files changed

Lines changed: 23 additions & 0 deletions

File tree

tools/server/server-schema.cpp

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,9 @@ std::vector<std::unique_ptr<field>> make_llama_cmpl_schema(const common_params &
1414
fields.emplace_back(f);
1515
};
1616

17+
add((new field_bool("verbose", params.verbose))
18+
->set_desc("Include __verbose field in the response with additional debug information"));
19+
1720
add((new field_bool("timings_per_token", params.timings_per_token))
1821
->set_desc("Include prompt processing and text generation speed information in each response"));
1922

tools/server/tests/unit/test_chat_completion.py

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -603,3 +603,23 @@ def test_chat_completions_token_count():
603603
})
604604
assert res.status_code == 200
605605
assert res.body["input_tokens"] > 5
606+
607+
608+
def test_verbose_debug():
609+
global server
610+
server.start()
611+
for verbose in [True, False]:
612+
res = server.make_request("POST", "/chat/completions", data={
613+
"max_tokens": 2,
614+
"messages": [
615+
{"role": "system", "content": "Book"},
616+
{"role": "user", "content": "What is the best book"},
617+
],
618+
"verbose": verbose,
619+
})
620+
assert res.status_code == 200
621+
if verbose:
622+
assert "__verbose" in res.body
623+
assert "Book" in res.body["__verbose"]["prompt"]
624+
else:
625+
assert "__verbose" not in res.body

0 commit comments

Comments
 (0)