From fd6bafe37de417185b0962974d0ae37948cb801d Mon Sep 17 00:00:00 2001 From: drbh Date: Wed, 30 Apr 2025 18:02:01 +0000 Subject: [PATCH] fix: adjust test payload --- .../models/test_grammar_response_format_llama.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/integration-tests/models/test_grammar_response_format_llama.py b/integration-tests/models/test_grammar_response_format_llama.py index d3ae735a..b2e69d61 100644 --- a/integration-tests/models/test_grammar_response_format_llama.py +++ b/integration-tests/models/test_grammar_response_format_llama.py @@ -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 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( f"{llama_grammar.base_url}/v1/chat/completions", headers=llama_grammar.headers,