ceos/fastmcp
1
0
Fork 0
mirror of https://github.com/PrefectHQ/fastmcp.git synced 2026-08-22 13:34:17 +02:00
Code Issues Projects Releases Packages Wiki Activity

Fix typing issue

Browse source
This commit is contained in:
Jeremiah Lowin 2025-04-13 21:45:17 -04:00
commit 29b31573bc
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/prompts/test_base.py
View file

@ -57,7 +57,7 @@ class TestRenderPrompt:
@pytest.mark.anyio
async def test_fn_returns_message(self):
async def fn() -> UserMessage:
async def fn() -> Message:
return UserMessage(content="Hello, world!")
prompt = Prompt.from_function(fn)
@ -67,7 +67,7 @@ class TestRenderPrompt:
@pytest.mark.anyio
async def test_fn_returns_assistant_message(self):
async def fn() -> AssistantMessage:
async def fn() -> Message:
return AssistantMessage(
content=TextContent(type="text", text="Hello, world!")
)
@ -108,7 +108,7 @@ class TestRenderPrompt:
async def test_fn_returns_resource_content(self):
"""Test returning a message with resource content."""
async def fn() -> UserMessage:
async def fn() -> Message:
return UserMessage(
content=EmbeddedResource(
type="resource",