Skip to content

Commit 2b9b830

Browse files
committed
fix snapshots
1 parent 5544992 commit 2b9b830

File tree

3 files changed

+14
-1
lines changed

3 files changed

+14
-1
lines changed

tests/models/test_anthropic.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3649,6 +3649,7 @@ async def test_anthropic_mcp_servers_stream(allow_model_requests: None, anthropi
36493649
tool_call_id='mcptoolu_01FZmJ5UspaX5BB9uU339UT1',
36503650
provider_name='anthropic',
36513651
),
3652+
previous_part_kind='thinking',
36523653
),
36533654
PartDeltaEvent(
36543655
index=1,
@@ -3811,6 +3812,7 @@ async def test_anthropic_mcp_servers_stream(allow_model_requests: None, anthropi
38113812
timestamp=IsDatetime(),
38123813
provider_name='anthropic',
38133814
),
3815+
previous_part_kind='builtin-tool-call',
38143816
),
38153817
]
38163818
)

tests/models/test_bedrock.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1363,8 +1363,16 @@ async def test_bedrock_model_stream_empty_text_delta(allow_model_requests: None,
13631363
content='The user just says "Hi". We need to respond appropriately, friendly greeting. No special instructions. Should be short.'
13641364
),
13651365
),
1366-
PartStartEvent(index=1, part=TextPart(content='Hello! How can I help')),
1366+
PartEndEvent(
1367+
index=0,
1368+
part=ThinkingPart(
1369+
content='The user just says "Hi". We need to respond appropriately, friendly greeting. No special instructions. Should be short.'
1370+
),
1371+
next_part_kind='text',
1372+
),
1373+
PartStartEvent(index=1, part=TextPart(content='Hello! How can I help'), previous_part_kind='thinking'),
13671374
FinalResultEvent(tool_name=None, tool_call_id=None),
13681375
PartDeltaEvent(index=1, delta=TextPartDelta(content_delta=' you today?')),
1376+
PartEndEvent(index=1, part=TextPart(content='Hello! How can I help you today?')),
13691377
]
13701378
)

tests/models/test_openai_responses.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6960,6 +6960,7 @@ async def test_openai_responses_model_mcp_server_tool_stream(allow_model_request
69606960
timestamp=IsDatetime(),
69616961
provider_name='openai',
69626962
),
6963+
previous_part_kind='builtin-tool-call',
69636964
),
69646965
PartStartEvent(
69656966
index=3,
@@ -6968,6 +6969,7 @@ async def test_openai_responses_model_mcp_server_tool_stream(allow_model_request
69686969
tool_call_id='mcp_00b9cc7a23d047270068faa0e67fb0819fa9e21302c398e9ac',
69696970
provider_name='openai',
69706971
),
6972+
previous_part_kind='thinking',
69716973
),
69726974
PartDeltaEvent(
69736975
index=3,
@@ -7075,6 +7077,7 @@ async def test_openai_responses_model_mcp_server_tool_stream(allow_model_request
70757077
timestamp=IsDatetime(),
70767078
provider_name='openai',
70777079
),
7080+
previous_part_kind='builtin-tool-call',
70787081
),
70797082
]
70807083
)

0 commit comments

Comments
 (0)