diff --git a/tests/integration/inference/test_text_inference.py b/tests/integration/inference/test_text_inference.py index e4a401e54..7d5d7fee4 100644 --- a/tests/integration/inference/test_text_inference.py +++ b/tests/integration/inference/test_text_inference.py @@ -565,6 +565,5 @@ def test_multi_turn_chat_completion(client_with_models, text_model_id): messages=messages, ) - print(response.completion_message.tool_calls) - print(response) - assert response.completion_message.content == "raining" + assert "rain" in response.completion_message.content.lower() +