github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-16 08:06:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

ollama unit test fixes

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

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

@ -222,19 +222,14 @@ class TestChatOllama(ChatModelV1UnitTests):
"""`ChatOllama` supports structured output via `with_structured_output`.""" """`ChatOllama` supports structured output via `with_structured_output`."""
return True return True
@property
def supports_image_content_blocks(self) -> bool:
"""`ChatOllama` supports image content blocks (e.g., `gemma3`)."""
return True
@property @property
def supports_reasoning_content_blocks(self) -> bool: def supports_reasoning_content_blocks(self) -> bool:
"""`ChatOllama` supports reasoning/thinking content blocks (e.g., `qwen3`).""" """`ChatOllama` supports reasoning/thinking content blocks (e.g., `qwen3`)."""
return True return True
@property @property
def returns_usage_metadata(self) -> bool: def supports_image_content_blocks(self) -> bool:
"""`ChatOllama` returns usage metadata with token counts.""" """`ChatOllama` supports image content blocks (e.g., `gemma3`)."""
return True return True
@property @property
@ -403,7 +398,7 @@ class TestChatOllama(ChatModelV1UnitTests):
def test_validate_model_on_init( def test_validate_model_on_init(
mock_client_class: Any, mock_validate_model: Any mock_client_class: Any, mock_validate_model: Any
) -> None: ) -> None:
"""Test that the model is validated on initialization when requested.""" """Test that local model presence is validated on initialization when requested."""
mock_client = MagicMock() mock_client = MagicMock()
mock_client_class.return_value = mock_client mock_client_class.return_value = mock_client