diff --git a/libs/partners/ollama/tests/integration_tests/test_gpt_oss_tools.py b/libs/partners/ollama/tests/integration_tests/test_gpt_oss_tools.py index e2a50165aab..040643da130 100644 --- a/libs/partners/ollama/tests/integration_tests/test_gpt_oss_tools.py +++ b/libs/partners/ollama/tests/integration_tests/test_gpt_oss_tools.py @@ -192,9 +192,7 @@ class TestGptOssToolCallingIntegration: # Stream a response chunks = list( - llm_with_tools.stream( - "What's the weather in Tokyo? Use the weather tool." - ) + llm_with_tools.stream("What's the weather in Tokyo? Use the weather tool.") ) # Should have received chunks @@ -359,4 +357,3 @@ class TestGptOssErrorHandling: # Should have no tool calls assert not response.tool_calls or len(response.tool_calls) == 0 -