Skip to content

Commit 639d60b

Browse files
committed
chore: fix coverage failure
Add some no cover pragmas in existing code to avoid coverage check failure. Correct raise to avoid resetting traceback in `AgentGraph`.
1 parent 58cbcb3 commit 639d60b

File tree

3 files changed

+4
-4
lines changed

3 files changed

+4
-4
lines changed

pydantic_ai_slim/pydantic_ai/_agent_graph.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -621,7 +621,7 @@ async def process_function_tools( # noqa: C901
621621
result_data = await toolset.call_tool(call, run_context)
622622
except exceptions.UnexpectedModelBehavior as e:
623623
ctx.state.increment_retries(ctx.deps.max_result_retries, e)
624-
raise e
624+
raise # pragma: no cover
625625
except ToolRetryError as e:
626626
ctx.state.increment_retries(ctx.deps.max_result_retries, e)
627627
yield _messages.FunctionToolCallEvent(call)

tests/test_agent.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3603,7 +3603,7 @@ async def only_if_plan_presented(
36033603
async def test_context_manager():
36043604
try:
36053605
from pydantic_ai.mcp import MCPServerStdio
3606-
except ImportError:
3606+
except ImportError: # pragma: no cover
36073607
return
36083608

36093609
server1 = MCPServerStdio('python', ['-m', 'tests.mcp_server'])
@@ -3623,7 +3623,7 @@ async def test_context_manager():
36233623
def test_set_mcp_sampling_model():
36243624
try:
36253625
from pydantic_ai.mcp import MCPServerStdio
3626-
except ImportError:
3626+
except ImportError: # pragma: no cover
36273627
return
36283628

36293629
test_model = TestModel()

tests/test_toolsets.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -483,7 +483,7 @@ async def prepare_add_context(ctx: RunContext[TestDeps], tool_defs: list[ToolDef
483483
async def test_context_manager():
484484
try:
485485
from pydantic_ai.mcp import MCPServerStdio
486-
except ImportError:
486+
except ImportError: # pragma: no cover
487487
return
488488

489489
server1 = MCPServerStdio('python', ['-m', 'tests.mcp_server'])

0 commit comments

Comments
 (0)