diff --git a/model_server/app/tests/test_function_calling.py b/model_server/app/tests/test_function_calling.py index c10e4b6f..251007d3 100644 --- a/model_server/app/tests/test_function_calling.py +++ b/model_server/app/tests/test_function_calling.py @@ -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