fix lint and a test

Author: DouweM

pydantic_ai_slim/pydantic_ai/_function_schema.py

@@ -57,10 +57,6 @@ class FunctionSchema:
     async def call(self, args_dict: dict[str, Any], ctx: RunContext[Any]) -> Any:
         args, kwargs = self._call_args(args_dict, ctx)
         if self.is_async_iterator:
-            assert ctx.event_stream is not None, (
-                'RunContext.event_stream needs to be set to use FunctionSchema.call with async iterators'
-            )
-
             return_value: Return | None = None
             async for event_data in self.function(*args, **kwargs):
                 if return_value is not None:

tests/test_agent.py

@@ -4378,7 +4378,7 @@ def analyze_data() -> list[Any]:
 
     with pytest.raises(
         UserError,
-        match="The return value of tool 'analyze_data' contains invalid nested `ToolReturn` objects. `ToolReturn` should be used directly.",
+        match="The return value of tool 'analyze_data' contains invalid nested `Return` objects. `Return` should be used directly.",
     ):
         agent.run_sync('Please analyze the data')