github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-13 22:59:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

hack fix unit

Browse Source
This commit is contained in:
Mason Daugherty 2025-08-05 21:07:18 -04:00
parent 20a41ec5df
commit ffda528f36
No known key found for this signature in database
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
libs/partners/ollama/tests/unit_tests/test_chat_models_v1.py
View File

@ -2,7 +2,7 @@
import json
import logging
from collections.abc import AsyncIterator, Iterator
from collections.abc import AsyncIterator, Generator, Iterator
from typing import Any
from unittest.mock import MagicMock, patch
@ -242,8 +242,9 @@ class TestChatOllama(ChatModelV1UnitTests):
"""
return False
# TODO: investigate type ignore, this feels hacky
@pytest.fixture
def model(self) -> Iterator[ChatOllama]:
def model(self) -> Generator[ChatOllama, None, None]: # type: ignore[override]
"""Create a ChatOllama instance for testing."""
sync_patcher = patch("langchain_ollama.chat_models_v1.Client")
async_patcher = patch("langchain_ollama.chat_models_v1.AsyncClient")