github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2026-05-18 21:44:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Apply patch [skip ci]

Browse Source
This commit is contained in:
open-swe[bot]
2025-08-06 16:44:38 +00:00
parent 931e4ce80f
commit 2ef6901d57
1 changed files with 3 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
libs/partners/ollama/tests/unit_tests/test_chat_models_gpt_oss.py
View File

@@ -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)