Fix lint errors and remove orphaned mocks

- Remove duplicate IsStr imports (already imported from dirty_equals)
- Remove orphaned book_flight/authenticate_with_airline mocks from removed example

Author: cjohnhanson

tests/test_agent.py

@@ -66,7 +66,7 @@
 from pydantic_ai.tools import DeferredToolRequests, DeferredToolResults, ToolDefinition, ToolDenied
 from pydantic_ai.usage import RequestUsage
 
-from .conftest import IsDatetime, IsNow, IsStr, TestEnv
+from .conftest import IsDatetime, IsNow, TestEnv
 
 pytestmark = pytest.mark.anyio
 

tests/test_examples.py

@@ -711,21 +711,6 @@ async def model_logic(  # noqa: C901
                     TextPart(content='The factorial of 15 is **1,307,674,368,000**.'),
                 ]
             )
-        elif m.content == 'Book a flight to Lisbon, Portugal and link my SkyWay Airlines account':
-            return ModelResponse(
-                parts=[
-                    ToolCallPart(
-                        tool_name='book_flight',
-                        args={'destination': 'Lisbon, Portugal'},
-                        tool_call_id='pyd_ai_tool_call_id_1',
-                    ),
-                    ToolCallPart(
-                        tool_name='authenticate_with_airline',
-                        args={'airline': 'SkyWay Airlines'},
-                        tool_call_id='pyd_ai_tool_call_id_2',
-                    ),
-                ]
-            )
 
     elif isinstance(m, ToolReturnPart) and m.tool_name == 'roulette_wheel':
         win = m.content == 'winner'
@@ -889,16 +874,6 @@ async def model_logic(  # noqa: C901
         return ModelResponse(
             parts=[TextPart('The answer to the ultimate question of life, the universe, and everything is 42.')]
         )
-    elif isinstance(m, ToolReturnPart) and m.tool_name in ('book_flight', 'authenticate_with_airline'):
-        # After deferred tools complete, check if we have all results to provide final response
-        tool_names = {part.tool_name for msg in messages for part in msg.parts if isinstance(part, ToolReturnPart)}
-        if 'book_flight' in tool_names and 'authenticate_with_airline' in tool_names:
-            return ModelResponse(
-                parts=[TextPart('Flight to Lisbon booked successfully and your SkyWay Airlines account is now linked.')]
-            )
-        # If we don't have both results yet, just acknowledge the tool result
-        return ModelResponse(parts=[TextPart(f'Received result from {m.tool_name}')])
-
     if isinstance(m, ToolReturnPart):
         sys.stdout.write(str(debug.format(messages, info)))
         raise RuntimeError(f'Unexpected message: {m}')

tests/test_streaming.py

@@ -51,7 +51,7 @@
 from pydantic_ai.usage import RequestUsage
 from pydantic_graph import End
 
-from .conftest import IsDatetime, IsInt, IsNow, IsStr
+from .conftest import IsDatetime, IsInt, IsNow
 
 pytestmark = pytest.mark.anyio
 

tests/test_tools.py

@@ -39,7 +39,7 @@
 from pydantic_ai.tools import DeferredToolRequests, DeferredToolResults, ToolApproved, ToolDefinition, ToolDenied
 from pydantic_ai.usage import RequestUsage
 
-from .conftest import IsDatetime, IsStr
+from .conftest import IsDatetime
 
 
 def test_tool_no_ctx():