adjust test

This commit is contained in:
lorenzejay
2026-01-29 17:17:21 -08:00
parent f63d088115
commit 563512b5e2

View File

@@ -220,7 +220,7 @@ def test_get_conversion_instructions_gpt() -> None:
supports_function_calling.return_value = True
instructions = get_conversion_instructions(SimpleModel, llm)
# Now using OpenAPI schema format for all models
assert "Ensure your final answer strictly adheres to the following OpenAPI schema:" in instructions
assert "Format your final answer according to the following OpenAPI schema:" in instructions
assert '"type": "json_schema"' in instructions
assert '"name": "SimpleModel"' in instructions
assert "Do not include the OpenAPI schema in the final output" in instructions
@@ -231,7 +231,7 @@ def test_get_conversion_instructions_non_gpt() -> None:
with patch.object(LLM, "supports_function_calling", return_value=False):
instructions = get_conversion_instructions(SimpleModel, llm)
# Now using OpenAPI schema format for all models
assert "Ensure your final answer strictly adheres to the following OpenAPI schema:" in instructions
assert "Format your final answer according to the following OpenAPI schema:" in instructions
assert '"type": "json_schema"' in instructions
assert '"name": "SimpleModel"' in instructions
assert "Do not include the OpenAPI schema in the final output" in instructions