From f9e45e6edfb7e39bba2b09b6e00c19a7532298fa Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Sat, 5 Apr 2025 19:59:10 -0700 Subject: [PATCH] test inference --- tests/integration/inference/test_text_inference.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) 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() +