From bf09a0916392274fee451f1b2ffa8080a49c7884 Mon Sep 17 00:00:00 2001 From: Jeremiah Lowin <153965+jlowin@users.noreply.github.com> Date: Tue, 6 May 2025 13:29:00 -0400 Subject: [PATCH] Fix prompt test --- tests/test_examples.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/tests/test_examples.py b/tests/test_examples.py index a6d20cbc1..1f68390d8 100644 --- a/tests/test_examples.py +++ b/tests/test_examples.py @@ -85,8 +85,8 @@ async def test_echo(): async with Client(mcp) as client: result = await client.get_prompt("echo", {"text": "hello"}) - assert len(result) == 1 - assert isinstance(result[0], PromptMessage) - assert isinstance(result[0].content, TextContent) - assert isinstance(result[0].content.text, str) - assert result[0].content.text == "hello" + assert len(result.messages) == 1 + assert isinstance(result.messages[0], PromptMessage) + assert isinstance(result.messages[0].content, TextContent) + assert isinstance(result.messages[0].content.text, str) + assert result.messages[0].content.text == "hello"