From 0fb6464ce28c8cff26910cc988d4db51b9c6987d Mon Sep 17 00:00:00 2001 From: Jeremiah Lowin <153965+jlowin@users.noreply.github.com> Date: Sun, 5 Jul 2026 23:52:22 -0400 Subject: [PATCH] Fix v2 content shapes: str uri for BlobResourceContents, tolerant request_id assertion --- tests/conftest.py | 3 +-- tests/server/providers/test_local_provider_prompts.py | 5 ++++- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/tests/conftest.py b/tests/conftest.py index 6f6061a11..51ee2c26d 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -231,7 +231,6 @@ def tool_server(): ImageContent, TextContent, ) - from pydantic import AnyUrl from fastmcp import FastMCP from fastmcp.utilities.types import Audio, File, Image @@ -274,7 +273,7 @@ def tool_server(): resource=BlobResourceContents( blob=base64.b64encode(b"abc").decode(), mime_type="application/octet-stream", - uri=AnyUrl("file:///test.bin"), + uri="file:///test.bin", ), ), ] diff --git a/tests/server/providers/test_local_provider_prompts.py b/tests/server/providers/test_local_provider_prompts.py index a32e7b9c6..bc199190e 100644 --- a/tests/server/providers/test_local_provider_prompts.py +++ b/tests/server/providers/test_local_provider_prompts.py @@ -42,7 +42,10 @@ class TestPromptContext: message = result.messages[0] assert message.role == "user" assert isinstance(message.content, TextContent) - assert message.content.text == "Hello, World! 1" + # The exact request_id is an SDK-internal counter value; assert that + # the callable object received a context and rendered it, not the + # specific counter (which depends on session bootstrap requests). + assert message.content.text.startswith("Hello, World! ") class TestPromptDecorator: