From 2ef6901d57f2157db1739f61d124d8710c3f6720 Mon Sep 17 00:00:00 2001 From: "open-swe[bot]" Date: Wed, 6 Aug 2025 16:44:38 +0000 Subject: [PATCH] Apply patch [skip ci] --- .../tests/unit_tests/test_chat_models_gpt_oss.py | 13 +++---------- 1 file changed, 3 insertions(+), 10 deletions(-) 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 222d55e55b8..165bc5ec9dd 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 @@ -130,11 +130,7 @@ class TestHarmonyToolConversion: class TestGptOssToolBinding: """Test tool binding for gpt-oss models.""" - @patch("langchain_ollama.chat_models.ChatOllama._create_chat_stream") - @patch("langchain_ollama.chat_models.ChatOllama._create_chat") - def test_bind_tools_with_gpt_oss( - self, mock_create_chat: MagicMock, mock_create_stream: MagicMock - ) -> None: + def test_bind_tools_with_gpt_oss(self) -> None: """Test that tools are correctly bound for gpt-oss models.""" # Create ChatOllama instance with gpt-oss model llm = ChatOllama(model="gpt-oss:20b") @@ -160,11 +156,7 @@ class TestGptOssToolBinding: if "type" in prop: assert isinstance(prop["type"], str) - @patch("langchain_ollama.chat_models.ChatOllama._create_chat_stream") - @patch("langchain_ollama.chat_models.ChatOllama._create_chat") - def test_bind_tools_with_non_gpt_oss( - self, mock_create_chat: MagicMock, mock_create_stream: MagicMock - ) -> None: + def test_bind_tools_with_non_gpt_oss(self) -> None: """Test that tools use standard format for non-gpt-oss models.""" # Create ChatOllama instance with non-gpt-oss model llm = ChatOllama(model="llama2") @@ -408,3 +400,4 @@ class TestChatParamsWithGptOss: for prop in props.values(): if "type" in prop: assert isinstance(prop["type"], str) +