Refactor OpenAI and OpenRouter models to use Iterable

Replaced Generator with Iterable in several methods to improve
compatibility and maintainability.

Author: ajac-zero

pydantic_ai_slim/pydantic_ai/models/openai.py

@@ -4,7 +4,7 @@
 import itertools
 import json
 import warnings
-from collections.abc import AsyncIterable, AsyncIterator, Generator, Iterable, Sequence
+from collections.abc import AsyncIterable, AsyncIterator, Iterable, Sequence
 from contextlib import asynccontextmanager
 from dataclasses import dataclass, field, replace
 from datetime import datetime
@@ -611,7 +611,7 @@ def _process_response(self, response: chat.ChatCompletion | str) -> ModelRespons
             finish_reason=self._map_finish_reason(choice.finish_reason),
         )
 
-    def _process_thinking(self, message: chat.ChatCompletionMessage) -> Generator[ThinkingPart]:
+    def _process_thinking(self, message: chat.ChatCompletionMessage) -> Iterable[ThinkingPart]:
         """Hook that maps reasoning tokens to thinking parts.
 
         This method may be overridden by subclasses of `OpenAIChatModel` to apply custom mappings.
@@ -627,7 +627,7 @@ def _process_thinking(self, message: chat.ChatCompletionMessage) -> Generator[Th
         if reasoning := getattr(message, 'reasoning', None):
             yield ThinkingPart(id='reasoning', content=reasoning, provider_name=self.system)
 
-    def _process_content(self, message: chat.ChatCompletionMessage) -> Generator[TextPart | ThinkingPart]:
+    def _process_content(self, message: chat.ChatCompletionMessage) -> Iterable[TextPart | ThinkingPart]:
         """Hook that maps the message content to thinking or text parts.
 
         This method may be overridden by subclasses of `OpenAIChatModel` to apply custom mappings.
@@ -636,7 +636,7 @@ def _process_content(self, message: chat.ChatCompletionMessage) -> Generator[Tex
             for part in split_content_into_text_and_thinking(message.content, self.profile.thinking_tags):
                 yield replace(part, id='content', provider_name=self.system) if isinstance(part, ThinkingPart) else part
 
-    def _process_tool_calls(self, message: chat.ChatCompletionMessage) -> Generator[ToolCallPart]:
+    def _process_tool_calls(self, message: chat.ChatCompletionMessage) -> Iterable[ToolCallPart]:
         """Hook that maps tool calls to tool call parts.
 
         This method may be overridden by subclasses of `OpenAIChatModel` to apply custom mappings.

pydantic_ai_slim/pydantic_ai/models/openrouter.py

@@ -1,7 +1,7 @@
 from __future__ import annotations as _annotations
 
 import itertools
-from collections.abc import Generator, Iterable
+from collections.abc import Iterable
 from dataclasses import dataclass, field
 from typing import Annotated, Any, Literal, TypeAlias, cast
 
@@ -572,7 +572,7 @@ def _validate_completion(self, response: chat.ChatCompletion) -> _OpenRouterChat
         return response
 
     @override
-    def _process_thinking(self, message: chat.ChatCompletionMessage) -> Generator[ThinkingPart]:
+    def _process_thinking(self, message: chat.ChatCompletionMessage) -> Iterable[ThinkingPart]:
         assert isinstance(message, _OpenRouterCompletionMessage)
 
         if reasoning_details := message.reasoning_details:
@@ -581,7 +581,7 @@ def _process_thinking(self, message: chat.ChatCompletionMessage) -> Generator[Th
         else:
             yield from super()._process_thinking(message)
 
-    def _process_image(self, message: chat.ChatCompletionMessage) -> Generator[FilePart]:
+    def _process_image(self, message: chat.ChatCompletionMessage) -> Iterable[FilePart]:
         assert isinstance(message, _OpenRouterCompletionMessage)
 
         if images := message.images: