diff --git a/tests/pytest/test_chat.py b/tests/pytest/test_chat.py index 540b1cda7..4fb9e167c 100644 --- a/tests/pytest/test_chat.py +++ b/tests/pytest/test_chat.py @@ -476,7 +476,12 @@ def test_as_ollama_message(): import ollama from ollama import Message as OllamaMessage - assert "ollama._types.Message" in str(ollama.chat.__annotations__["messages"]) + # ollama 0.4.2 added Callable to the type hints, but pyright complains about + # missing arguments to the Callable type. We'll ignore this for now. + # https://github.com/ollama/ollama-python/commit/b50a65b + chat = ollama.chat # type: ignore + + assert "ollama._types.Message" in str(chat.__annotations__["messages"]) from shiny.ui._chat_provider_types import as_ollama_message