style: Fix imports and line length formatting

tests/server/fastmcp/prompts/test_base.py

@@ -1,16 +1,18 @@
-from pydantic import FileUrl
 import pytest
+from pydantic import FileUrl
+
 from mcp.server.fastmcp.prompts.base import (
-    Prompt,
-    UserMessage,
-    TextContent,
     AssistantMessage,
     Message,
+    Prompt,
+    TextContent,
+    UserMessage,
 )
 from mcp.types import EmbeddedResource, TextResourceContents
 
 
 class TestRenderPrompt:
+    @pytest.mark.anyio
     async def test_basic_fn(self):
         def fn() -> str:
             return "Hello, world!"
@@ -20,6 +22,7 @@ class TestRenderPrompt:
             UserMessage(content=TextContent(type="text", text="Hello, world!"))
         ]
 
+    @pytest.mark.anyio
     async def test_async_fn(self):
         async def fn() -> str:
             return "Hello, world!"
@@ -29,6 +32,7 @@ class TestRenderPrompt:
             UserMessage(content=TextContent(type="text", text="Hello, world!"))
         ]
 
+    @pytest.mark.anyio
     async def test_fn_with_args(self):
         async def fn(name: str, age: int = 30) -> str:
             return f"Hello, {name}! You're {age} years old."
@@ -42,6 +46,7 @@ class TestRenderPrompt:
             )
         ]
 
+    @pytest.mark.anyio
     async def test_fn_with_invalid_kwargs(self):
         async def fn(name: str, age: int = 30) -> str:
             return f"Hello, {name}! You're {age} years old."
@@ -50,6 +55,7 @@ class TestRenderPrompt:
         with pytest.raises(ValueError):
             await prompt.render(arguments=dict(age=40))
 
+    @pytest.mark.anyio
     async def test_fn_returns_message(self):
         async def fn() -> UserMessage:
             return UserMessage(content="Hello, world!")
@@ -59,6 +65,7 @@ class TestRenderPrompt:
             UserMessage(content=TextContent(type="text", text="Hello, world!"))
         ]
 
+    @pytest.mark.anyio
     async def test_fn_returns_assistant_message(self):
         async def fn() -> AssistantMessage:
             return AssistantMessage(
@@ -70,6 +77,7 @@ class TestRenderPrompt:
             AssistantMessage(content=TextContent(type="text", text="Hello, world!"))
         ]
 
+    @pytest.mark.anyio
     async def test_fn_returns_multiple_messages(self):
         expected = [
             UserMessage("Hello, world!"),
@@ -83,6 +91,7 @@ class TestRenderPrompt:
         prompt = Prompt.from_function(fn)
         assert await prompt.render() == expected
 
+    @pytest.mark.anyio
     async def test_fn_returns_list_of_strings(self):
         expected = [
             "Hello, world!",
@@ -95,6 +104,7 @@ class TestRenderPrompt:
         prompt = Prompt.from_function(fn)
         assert await prompt.render() == [UserMessage(t) for t in expected]
 
+    @pytest.mark.anyio
     async def test_fn_returns_resource_content(self):
         """Test returning a message with resource content."""
 
@@ -124,6 +134,7 @@ class TestRenderPrompt:
             )
         ]
 
+    @pytest.mark.anyio
     async def test_fn_returns_mixed_content(self):
         """Test returning messages with mixed content types."""
 
@@ -163,6 +174,7 @@ class TestRenderPrompt:
             ),
         ]
 
+    @pytest.mark.anyio
     async def test_fn_returns_dict_with_resource(self):
         """Test returning a dict with resource content."""
 

tests/server/fastmcp/prompts/test_manager.py

@@ -1,5 +1,6 @@
 import pytest
-from mcp.server.fastmcp.prompts.base import UserMessage, TextContent, Prompt
+
+from mcp.server.fastmcp.prompts.base import Prompt, TextContent, UserMessage
 from mcp.server.fastmcp.prompts.manager import PromptManager
 
 
@@ -60,6 +61,7 @@ class TestPromptManager:
         assert len(prompts) == 2
         assert prompts == [prompt1, prompt2]
 
+    @pytest.mark.anyio
     async def test_render_prompt(self):
         """Test rendering a prompt."""
 
@@ -74,6 +76,7 @@ class TestPromptManager:
             UserMessage(content=TextContent(type="text", text="Hello, world!"))
         ]
 
+    @pytest.mark.anyio
     async def test_render_prompt_with_args(self):
         """Test rendering a prompt with arguments."""
 
@@ -88,12 +91,14 @@ class TestPromptManager:
             UserMessage(content=TextContent(type="text", text="Hello, World!"))
         ]
 
+    @pytest.mark.anyio
     async def test_render_unknown_prompt(self):
         """Test rendering a non-existent prompt."""
         manager = PromptManager()
         with pytest.raises(ValueError, match="Unknown prompt: unknown"):
             await manager.render_prompt("unknown")
 
+    @pytest.mark.anyio
     async def test_render_prompt_with_missing_args(self):
         """Test rendering a prompt with missing required arguments."""