fix assert

This commit is contained in:
Dinesh Yeduguru 2024-10-24 14:46:21 -07:00
parent 9bf1388429
commit f7658f9c3a

View file

@ -206,7 +206,7 @@ async def test_completions_structured_output(inference_settings):
assert isinstance(response.content, str) assert isinstance(response.content, str)
answer = Output.parse_raw(response.content) answer = Output.parse_raw(response.content)
assert "Michael Jordan" in answer.name assert answer.name == "Michael Jordan"
assert answer.year_born == "1963" assert answer.year_born == "1963"
assert answer.year_retired == "2003" assert answer.year_retired == "2003"