Address Douwe's feedback

- Add 'optional' to second metadata annotation
- Revert unrelated test_examples.py changes

Author: cjohnhanson

docs/deferred-tools.md

@@ -329,7 +329,7 @@ async def main():
 ```
 
 1. Generate a task ID that can be tracked independently of the tool call ID.
-2. The `metadata` parameter passes the `task_id` so it can be matched with results later, accessible in `DeferredToolRequests.metadata` keyed by `tool_call_id`.
+2. The optional `metadata` parameter passes the `task_id` so it can be matched with results later, accessible in `DeferredToolRequests.metadata` keyed by `tool_call_id`.
 3. In reality, this would typically happen in a separate process that polls for the task status or is notified when all pending tasks are complete.
 
 _(This example is complete, it can be run "as is" — you'll need to add `asyncio.run(main())` to run `main`)_

tests/test_examples.py

@@ -179,7 +179,6 @@ def print(self, *args: Any, **kwargs: Any) -> None:
     env.set('MOONSHOTAI_API_KEY', 'testing')
     env.set('DEEPSEEK_API_KEY', 'testing')
     env.set('OVHCLOUD_API_KEY', 'testing')
-    env.set('PYDANTIC_AI_GATEWAY_API_KEY', 'testing')
 
     prefix_settings = example.prefix_settings()
     opt_test = prefix_settings.get('test', '')
@@ -259,9 +258,7 @@ def print(self, *args: Any, **kwargs: Any) -> None:
 def print_callback(s: str) -> str:
     s = re.sub(r'datetime\.datetime\(.+?\)', 'datetime.datetime(...)', s, flags=re.DOTALL)
     s = re.sub(r'\d\.\d{4,}e-0\d', '0.0...', s)
-    s = re.sub(r'datetime.date\(', 'date(', s)
-    s = re.sub(r"run_id='.+?'", "run_id='...'", s)
-    return s
+    return re.sub(r'datetime.date\(', 'date(', s)
 
 
 def mock_render_duration(seconds: float, force_signed: bool) -> str:
@@ -874,14 +871,10 @@ async def model_logic(  # noqa: C901
         return ModelResponse(
             parts=[TextPart('The answer to the ultimate question of life, the universe, and everything is 42.')]
         )
-    if isinstance(m, ToolReturnPart):
+    else:
         sys.stdout.write(str(debug.format(messages, info)))
         raise RuntimeError(f'Unexpected message: {m}')
 
-    # Fallback for any other message type
-    sys.stdout.write(str(debug.format(messages, info)))
-    raise RuntimeError(f'Unexpected message type: {type(m).__name__}')
-
 
 async def stream_model_logic(  # noqa C901
     messages: list[ModelMessage], info: AgentInfo