From 2dd1988baf69108b6f86ed48092bd1887f4cde60 Mon Sep 17 00:00:00 2001 From: Mason Daugherty Date: Tue, 5 Aug 2025 20:52:29 -0400 Subject: [PATCH] ollama unit test fixes --- .../ollama/tests/unit_tests/test_chat_models_v1.py | 11 +++-------- 1 file changed, 3 insertions(+), 8 deletions(-) diff --git a/libs/partners/ollama/tests/unit_tests/test_chat_models_v1.py b/libs/partners/ollama/tests/unit_tests/test_chat_models_v1.py index f65f962350c..493e1c996de 100644 --- a/libs/partners/ollama/tests/unit_tests/test_chat_models_v1.py +++ b/libs/partners/ollama/tests/unit_tests/test_chat_models_v1.py @@ -222,19 +222,14 @@ class TestChatOllama(ChatModelV1UnitTests): """`ChatOllama` supports structured output via `with_structured_output`.""" return True - @property - def supports_image_content_blocks(self) -> bool: - """`ChatOllama` supports image content blocks (e.g., `gemma3`).""" - return True - @property def supports_reasoning_content_blocks(self) -> bool: """`ChatOllama` supports reasoning/thinking content blocks (e.g., `qwen3`).""" return True @property - def returns_usage_metadata(self) -> bool: - """`ChatOllama` returns usage metadata with token counts.""" + def supports_image_content_blocks(self) -> bool: + """`ChatOllama` supports image content blocks (e.g., `gemma3`).""" return True @property @@ -403,7 +398,7 @@ class TestChatOllama(ChatModelV1UnitTests): def test_validate_model_on_init( mock_client_class: Any, mock_validate_model: Any ) -> 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_class.return_value = mock_client