Skip to content

Commit 1ee97f6

Browse files
committed
linting
1 parent 91dd236 commit 1ee97f6

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
@@ -1454,10 +1454,7 @@ async def test_cache_point_filtering():
14541454
from pydantic_ai.models.bedrock import BedrockConverseModel
14551455

14561456
# Test the static method directly
1457-
messages = await BedrockConverseModel._map_user_prompt(
1458-
UserPromptPart(content=['text', CachePoint()]),
1459-
count()
1460-
)
1457+
messages = await BedrockConverseModel._map_user_prompt(UserPromptPart(content=['text', CachePoint()]), count())
14611458
# CachePoint should be filtered out, message should still be valid
14621459
assert len(messages) == 1
14631460
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)