From d12b8a8d1ae356c9eb9013cd09cfff6d0f69f046 Mon Sep 17 00:00:00 2001 From: Pavel Tisnovsky Date: Tue, 11 Aug 2026 09:59:04 +0200 Subject: [PATCH] LCORE-1449: Proper attribute check --- .../capabilities/redaction/test_capability.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/unit/pydantic_ai_lightspeed/capabilities/redaction/test_capability.py b/tests/unit/pydantic_ai_lightspeed/capabilities/redaction/test_capability.py index cc7f6fe68..0307312df 100644 --- a/tests/unit/pydantic_ai_lightspeed/capabilities/redaction/test_capability.py +++ b/tests/unit/pydantic_ai_lightspeed/capabilities/redaction/test_capability.py @@ -268,7 +268,9 @@ async def test_before_model_request_no_match( ) result = await capability.before_model_request(mocker.Mock(), request_context) assert result is request_context - assert req.parts[0].content == "safe text" + part = req.parts[0] + assert isinstance(part, UserPromptPart) + assert part.content == "safe text" @pytest.mark.asyncio() async def test_after_model_request_redacts_response(