Apply suggestions from code review

Co-authored-by: David Montague <[email protected]>

Author: mattbrandman

pydantic_ai_slim/pydantic_ai/models/cohere.py

@@ -227,11 +227,11 @@ def _map_messages(self, messages: list[ModelMessage]) -> list[ChatMessageV2]:
                         tool_calls.append(self._map_tool_call(item))
                     elif isinstance(item, ServerToolCallPart):  # pragma: no cover
                         # ServerToolCallPart represents a tool call from a remote server
-                        # Never returned from cohere
+                        # This is currently never returned from cohere
                         pass
                     elif isinstance(item, ServerToolReturnPart):  # pragma: no cover
                         # ServerToolReturnPart represents a tool return from a remote server
-                        # Never returned from cohere
+                        # This is currently never returned from cohere
                         pass
                     else:
                         assert_never(item)

pydantic_ai_slim/pydantic_ai/models/gemini.py

@@ -626,11 +626,11 @@ def _content_model_response(m: ModelResponse) -> _GeminiContent:
                 parts.append(_GeminiTextPart(text=item.content))
         elif isinstance(item, ServerToolCallPart):  # pragma: no cover
             # Handle ServerToolCallPart the same as ToolCallPart
-            # Never returned from gemini
+            # This is currently never returned from gemini
             pass
         elif isinstance(item, ServerToolReturnPart):  # pragma: no cover
             # Convert ServerToolReturnPart to a function response part
-            # Never returned from gemini
+            # This is currently never returned from gemini
             pass
         else:
             assert_never(item)

pydantic_ai_slim/pydantic_ai/models/google.py

@@ -510,10 +510,10 @@ def _content_model_response(m: ModelResponse) -> ContentDict:
             # parts.append({'text': item.content, 'thought': True})
             pass
         elif isinstance(item, ServerToolCallPart):  # pragma: no cover
-            # Never returned from google
+            # This is currently never returned from google
             pass
         elif isinstance(item, ServerToolReturnPart):  # pragma: no cover
-            # Never returned from google
+            # This is currently never returned from google
             pass
         else:
             assert_never(item)

pydantic_ai_slim/pydantic_ai/models/groq.py

@@ -319,11 +319,11 @@ def _map_messages(self, messages: list[ModelMessage]) -> list[chat.ChatCompletio
                         continue
                     elif isinstance(item, ServerToolCallPart):  # pragma: no cover
                         # ServerToolCallPart is handled separately in server-side tools
-                        # Never returned from groq
+                        # This is currently never returned from groq
                         pass
                     elif isinstance(item, ServerToolReturnPart):  # pragma: no cover
                         # ServerToolReturnPart is handled separately in server-side tools
-                        # Never returned from groq
+                        # This is currently never returned from groq
                         pass
                     else:
                         assert_never(item)

pydantic_ai_slim/pydantic_ai/models/mistral.py

@@ -505,11 +505,11 @@ def _map_messages(self, messages: list[ModelMessage]) -> list[MistralMessages]:
                         tool_calls.append(self._map_tool_call(part))
                     elif isinstance(part, ServerToolCallPart):  # pragma: no cover
                         # Handle ServerToolCallPart the same as ToolCallPart
-                        # Never returned from mistral
+                        # This is currently never returned from mistral
                         pass
                     elif isinstance(part, ServerToolReturnPart):  # pragma: no cover
                         # For now, we'll add ServerToolReturnPart as text content
-                        # Never returned from mistral
+                        # This is currently never returned from mistral
                         pass
                     else:
                         assert_never(part)

pydantic_ai_slim/pydantic_ai/models/openai.py

@@ -449,7 +449,7 @@ async def _map_messages(self, messages: list[ModelMessage]) -> list[chat.ChatCom
                         pass
                     elif isinstance(item, ToolCallPart):
                         tool_calls.append(self._map_tool_call(item))
-                    # OpenAI doesn't return server tools calls.
+                    # OpenAI doesn't return server tool calls
                     elif isinstance(item, (ServerToolCallPart, ServerToolReturnPart)):  # pragma: no cover
                         continue
                     else:
@@ -897,7 +897,7 @@ async def _map_messages(
                         openai_messages.append(responses.EasyInputMessageParam(role='assistant', content=item.content))
                     elif isinstance(item, ToolCallPart):
                         openai_messages.append(self._map_tool_call(item))
-                    # OpenAI doesn't return server tools calls.
+                    # OpenAI doesn't return server tool calls
                     elif isinstance(item, (ServerToolCallPart, ServerToolReturnPart)):
                         continue
                     elif isinstance(item, ThinkingPart):