github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2026-05-19 05:58:58 +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:45:31 +00:00
parent 2ef6901d57
commit 34a8fc0cd1
1 changed files with 15 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -288,11 +288,15 @@ class TestGptOssResponseParsing:
class TestGptOssIntegration:
"""Integration tests for gpt-oss model with ChatOllama."""
@patch("langchain_ollama.chat_models.ChatOllama._create_chat")
def test_invoke_with_tools(self, mock_create_chat: MagicMock) -> None:
@patch("ollama.Client")
def test_invoke_with_tools(self, mock_client_class: MagicMock) -> None:
"""Test invoking gpt-oss model with tools."""
# Create mock client instance
mock_client = MagicMock()
mock_client_class.return_value = mock_client
# Mock response with tool call
mock_create_chat.return_value = {
mock_client.chat.return_value = {
"message": {
"role": "assistant",
"content": "",
@@ -325,11 +329,15 @@ class TestGptOssIntegration:
assert result.tool_calls[0]["name"] == "get_weather"
assert result.tool_calls[0]["args"] == {"location": "Berlin", "unit": "celsius"}
@patch("langchain_ollama.chat_models.ChatOllama._create_chat_stream")
def test_stream_with_tools(self, mock_create_stream: MagicMock) -> None:
@patch("ollama.Client")
def test_stream_with_tools(self, mock_client_class: MagicMock) -> None:
"""Test streaming gpt-oss model with tools."""
# Create mock client instance
mock_client = MagicMock()
mock_client_class.return_value = mock_client
# Mock streaming response
mock_create_stream.return_value = iter([
mock_client.chat.return_value = iter([
{
"message": {"content": "Let me check the weather for you."},
"done": False,
@@ -401,3 +409,4 @@ class TestChatParamsWithGptOss:
if "type" in prop:
assert isinstance(prop["type"], str)