Skip to content

Commit 11c99e0

Browse files
authored
Merge pull request lightspeed-core#2403 from tisnik/lcore-1449-proper-attribute-check
LCORE-1449: Proper attribute check
2 parents 2aaa039 + d12b8a8 commit 11c99e0

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

tests/unit/pydantic_ai_lightspeed/capabilities/redaction/test_capability.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -268,7 +268,9 @@ async def test_before_model_request_no_match(
268268
)
269269
result = await capability.before_model_request(mocker.Mock(), request_context)
270270
assert result is request_context
271-
assert req.parts[0].content == "safe text"
271+
part = req.parts[0]
272+
assert isinstance(part, UserPromptPart)
273+
assert part.content == "safe text"
272274

273275
@pytest.mark.asyncio()
274276
async def test_after_model_request_redacts_response(

0 commit comments

Comments
 (0)