Fix

Author: medaminezghal

docs/graph.md

@@ -352,7 +352,7 @@ stateDiagram-v2
   Feedback --> [*]
 ```
 
-```python {title="genai_email_feedback.py" py="3.10" test="skip"}
+```python {title="genai_email_feedback.py" py="3.10"}
 from __future__ import annotations as _annotations
 
 from dataclasses import dataclass, field

docs/models/google.md

@@ -58,7 +58,7 @@ To use Vertex AI, you may need to set up [application default credentials](https
 
 If you have the [`gcloud` CLI](https://cloud.google.com/sdk/gcloud) installed and configured, you can use:
 
-```python {test="skip"}
+```python
 from pydantic_ai import Agent
 from pydantic_ai.models.google import GoogleModel
 from pydantic_ai.providers.google import GoogleProvider

pydantic_ai_slim/pydantic_ai/mcp.py

@@ -141,7 +141,7 @@ async def call_tool(
         except McpError as e:
             raise exceptions.ModelRetry(e.error.message)
 
-        content = [self._map_tool_result_part(part) for part in result.content]
+        content = [await self._map_tool_result_part(part) for part in result.content]
 
         if result.isError:
             text = '\n'.join(str(part) for part in content)
@@ -207,7 +207,7 @@ async def _sampling_callback(
             model=self.sampling_model.model_name,
         )
 
-    def _map_tool_result_part(
+    async def _map_tool_result_part(
         self, part: mcp_types.ContentBlock
     ) -> str | messages.BinaryContent | dict[str, Any] | list[Any]:
         # See https://github.com/jlowin/fastmcp/blob/main/docs/servers/tools.mdx#return-values
@@ -230,30 +230,26 @@ def _map_tool_result_part(
             )  # pragma: no cover
         elif isinstance(part, mcp_types.EmbeddedResource):
             resource = part.resource
-            if isinstance(resource, mcp_types.TextResourceContents):
-                return resource.text
-            elif isinstance(resource, mcp_types.BlobResourceContents):
-                return messages.BinaryContent(
-                    data=base64.b64decode(resource.blob),
-                    media_type=resource.mimeType or 'application/octet-stream',
-                )
-            else:
-                assert_never(resource)
+            return self._get_content(resource)
         elif isinstance(part, mcp_types.ResourceLink):
-            resource_result: mcp_types.ReadResourceResult = anyio.run(lambda: self._client.read_resource(part.uri))
-            content = resource_result.contents[0]
-            if isinstance(content, mcp_types.TextResourceContents):
-                return content.text
-            elif isinstance(content, mcp_types.BlobResourceContents):
-                return messages.BinaryContent(
-                    data=base64.b64decode(content.blob), media_type=content.mimeType or 'application/octet-stream'
-                )
-            else:
-                assert_never(content)
+            resource_result: mcp_types.ReadResourceResult = await self._client.read_resource(part.uri)
+            return [self._get_content(resource) for resource in resource_result.contents]
 
         else:
             assert_never(part)
 
+    def _get_content(
+        self, resource: mcp_types.TextResourceContents | mcp_types.BlobResourceContents
+    ) -> str | messages.BinaryContent:
+        if isinstance(resource, mcp_types.TextResourceContents):
+            return resource.text
+        elif isinstance(resource, mcp_types.BlobResourceContents):
+            return messages.BinaryContent(
+                data=base64.b64decode(resource.blob), media_type=resource.mimeType or 'application/octet-stream'
+            )
+        else:
+            assert_never(resource)
+
 
 @dataclass
 class MCPServerStdio(MCPServer):

tests/mcp_server.py

@@ -5,7 +5,13 @@
 from mcp.server.fastmcp import Context, FastMCP, Image
 from mcp.server.session import ServerSessionT
 from mcp.shared.context import LifespanContextT, RequestT
-from mcp.types import BlobResourceContents, EmbeddedResource, SamplingMessage, TextContent, TextResourceContents
+from mcp.types import (
+    BlobResourceContents,
+    EmbeddedResource,
+    SamplingMessage,
+    TextContent,
+    TextResourceContents,
+)
 from pydantic import AnyUrl
 
 mcp = FastMCP('PydanticAI MCP Server')