From e73e20a91d24fc5e676541a3202081fe9067042e Mon Sep 17 00:00:00 2001 From: cotran Date: Thu, 7 Nov 2024 11:46:08 -0800 Subject: [PATCH] address cmt --- model_server/app/tests/test_function_calling.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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