This commit is contained in:
cotran 2024-11-04 10:21:11 -08:00
parent 13fac83381
commit 0910fcdcfa
2 changed files with 13 additions and 23 deletions

View file

@ -73,7 +73,7 @@ async def test_chat_completion(mock_hanlder, mock_client):
mock_hanlder._format_system.return_value = "<formatted_tools>"
response = Response()
chat_response = await chat_completion(request, response)
chat_response = await chat_completion(request, response, prefill_enabled=True)
assert isinstance(chat_response, ChatCompletionResponse)
assert chat_response.choices[0].message.content is not None