Skip to content

Commit ba41897

Browse files
committed
added exc_info declaration
1 parent 8b30f69 commit ba41897

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

python/example_code/bedrock-agent-runtime/test/test_bedrock_agent_runtime_wrapper.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ async def test_invoke_agent(make_stubber, error_code):
4040
if error_code is None:
4141
wrapper.invoke_agent(agent_id, agent_alias_id, session_id, prompt)
4242
else:
43-
with pytest.raises(ClientError):
43+
with pytest.raises(ClientError) as exc_info:
4444
async for _ in wrapper.invoke_agent(
4545
agent_id, agent_alias_id, session_id, prompt
4646
):
@@ -84,7 +84,7 @@ async def test_invoke_flow(make_stubber, error_code):
8484
if error_code is None:
8585
wrapper.invoke_flow(flow_id, flow_alias_id, inputs, execution_id)
8686
else:
87-
with pytest.raises(ClientError):
87+
with pytest.raises(ClientError) as exc_info:
8888
async for _ in wrapper.invoke_flow(
8989
flow_id, flow_alias_id, inputs, execution_id
9090
):

0 commit comments

Comments
 (0)