Skip to content

Commit 4824eeb

Browse files
committed
linting
1 parent 54869d6 commit 4824eeb

File tree

2 files changed

+3
-7
lines changed

2 files changed

+3
-7
lines changed

tests/models/test_bedrock.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1520,10 +1520,7 @@ async def test_cache_point_filtering():
15201520
from pydantic_ai.models.bedrock import BedrockConverseModel
15211521

15221522
# Test the static method directly
1523-
messages = await BedrockConverseModel._map_user_prompt(
1524-
UserPromptPart(content=['text', CachePoint()]),
1525-
count()
1526-
)
1523+
messages = await BedrockConverseModel._map_user_prompt(UserPromptPart(content=['text', CachePoint()]), count())
15271524
# CachePoint should be filtered out, message should still be valid
15281525
assert len(messages) == 1
15291526
assert messages[0]['role'] == 'user'

tests/models/test_huggingface.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1018,15 +1018,14 @@ async def test_hf_model_thinking_part_iter(allow_model_requests: None, huggingfa
10181018
)
10191019

10201020

1021-
10221021
async def test_cache_point_filtering():
10231022
"""Test that CachePoint is filtered out in HuggingFace message mapping."""
10241023
from pydantic_ai import CachePoint, UserPromptPart
10251024
from pydantic_ai.models.huggingface import HuggingFaceModel
1026-
1025+
10271026
# Test the static method directly
10281027
msg = await HuggingFaceModel._map_user_prompt(UserPromptPart(content=['text', CachePoint()]))
1029-
1028+
10301029
# CachePoint should be filtered out
10311030
assert msg['role'] == 'user'
10321031
assert len(msg['content']) == 1

0 commit comments

Comments
 (0)