diff --git a/libs/partners/ollama/tests/unit_tests/test_chat_models_gpt_oss.py b/libs/partners/ollama/tests/unit_tests/test_chat_models_gpt_oss.py index 114e24af0ab..5761f1f7cb5 100644 --- a/libs/partners/ollama/tests/unit_tests/test_chat_models_gpt_oss.py +++ b/libs/partners/ollama/tests/unit_tests/test_chat_models_gpt_oss.py @@ -315,7 +315,7 @@ class TestGptOssIntegration: calc_tool = tool(calculate) # Bind multiple tools - llm_with_tools = llm.bind_tools([get_weather, search_web, calculate]) + llm_with_tools = llm.bind_tools([get_weather, search_tool, calc_tool]) # Check that all tools are bound correctly assert hasattr(llm_with_tools, "kwargs") @@ -403,3 +403,4 @@ class TestChatParamsWithGptOss: +