huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-09 11:24:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: adjust test payload

Browse Source
This commit is contained in:
drbh 2025-04-30 18:02:01 +00:00
parent 1d63285cf9
commit fd6bafe37d
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
integration-tests/models/test_grammar_response_format_llama.py
View File

@ -73,7 +73,10 @@ async def test_grammar_response_format_llama_json(llama_grammar, response_snapsh
assert called == '{ "unit": "fahrenheit", "temperature": [ 72, 79, 88 ] }' assert called == '{ "unit": "fahrenheit", "temperature": [ 72, 79, 88 ] }'
assert chat_completion == response_snapshot assert chat_completion == response_snapshot
json_payload["response_format"]["type"] = "json_schema" json_payload["response_format"] = {
"type": "json_schema",
"value": {"name": "weather", "strict": True, "schema": Weather.schema()},
}
response = requests.post( response = requests.post(
f"{llama_grammar.base_url}/v1/chat/completions", f"{llama_grammar.base_url}/v1/chat/completions",
headers=llama_grammar.headers, headers=llama_grammar.headers,