Skip to content

Commit a936f36

Browse files
committed
fix: pyright errors
1 parent f9c17f6 commit a936f36

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/server/fastmcp/prompts/test_manager.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33
import pytest
44

55
from mcp.server.fastmcp.prompts.manager import PromptManager
6+
from mcp.types import TextContent
67

78

89
class TestPromptManager:
@@ -107,6 +108,7 @@ def fn() -> str:
107108
messages = await manager.render_prompt("fn")
108109
assert len(messages) == 1
109110
assert messages[0].role == "user"
111+
assert isinstance(messages[0].content, TextContent)
110112
assert messages[0].content.text == "Hello, world!"
111113

112114
@pytest.mark.anyio
@@ -122,6 +124,7 @@ def fn(name: str) -> str:
122124
messages = await manager.render_prompt("fn", {"name": "Alice"})
123125
assert len(messages) == 1
124126
assert messages[0].role == "user"
127+
assert isinstance(messages[0].content, TextContent)
125128
assert messages[0].content.text == "Hello, Alice!"
126129

127130
@pytest.mark.anyio

0 commit comments

Comments
 (0)