address cmt

This commit is contained in:
cotran 2024-11-07 11:46:08 -08:00
parent 1f383eafc4
commit e73e20a91d

View file

@ -87,4 +87,4 @@ async def test_chat_completion(mock_hanlder, mock_client):
second_call_args = mock_client.chat.completions.create.call_args_list[1][1]
assert second_call_args["stream"] == False
assert "model" in second_call_args
assert second_call_args["messages"][-1]["content"] in const.prefill_list
assert second_call_args["messages"][-1]["content"] in const.PREFILL_LIST