Fix summarizer cache test typing

2026-03-14 13:12:52 -07:00 · 2026-03-14 13:12:52 -07:00 · 482bb6cb16
parent e5841882ad
commit 482bb6cb16
1 changed files with 2 additions and 0 deletions
--- a/tests/utils/test_summarizer.py
+++ b/tests/utils/test_summarizer.py
@ -246,6 +246,7 @@ class TestSummaryPromptCaching:
            )

        assert response is mock_response
+        assert mock_call.await_args is not None
        call_kwargs = mock_call.await_args.kwargs
        assert call_kwargs["prompt"] == ""
        assert len(call_kwargs["messages"]) == 2
@ -277,6 +278,7 @@ class TestSummaryPromptCaching:
            )

        assert response is mock_response
+        assert mock_call.await_args is not None
        call_kwargs = mock_call.await_args.kwargs
        assert call_kwargs["prompt"] == ""
        assert len(call_kwargs["messages"]) == 2