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 86f85502ee5..0b0f78e74c0 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 @@ -258,7 +258,7 @@ class TestGptOssModelCompatibility: llm_with_tools = llm.bind_tools([get_weather]) # Verify tools are in Harmony format - tools = llm_with_tools.kwargs["tools"] + tools = llm_with_tools.kwargs["tools"] # type: ignore[attr-defined] assert len(tools) == 1 tool = tools[0] assert tool["type"] == "function" @@ -358,3 +358,4 @@ class TestGptOssErrorHandling: # Should have no tool calls assert not response.tool_calls or len(response.tool_calls) == 0 +