From c8adab51d0233c33c2b704aa1cd29ec5396fe6d2 Mon Sep 17 00:00:00 2001
From: "open-swe[bot]" <open-swe@users.noreply.github.com>
Date: Wed, 6 Aug 2025 16:48:26 +0000
Subject: [PATCH] Apply patch [skip ci]

---
 .../ollama/tests/unit_tests/test_chat_models_gpt_oss.py        | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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 446e52ff1d7..329b8c52340 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
@@ -329,7 +329,7 @@ class TestGptOssIntegration:
         assert result.tool_calls[0]["name"] == "get_weather"
         assert result.tool_calls[0]["args"] == {"location": "Berlin", "unit": "celsius"}
     
-    @patch("ollama.Client")
+    @patch("langchain_ollama.chat_models.Client")
     def test_stream_with_tools(self, mock_client_class: MagicMock) -> None:
         """Test streaming gpt-oss model with tools."""
         # Create mock client instance
@@ -411,3 +411,4 @@ class TestChatParamsWithGptOss:
 
 
 
+