Jeremiah Lowin commited on
Commit
26e35ea
·
1 Parent(s): 57edc1b

Fix prompt test

Browse files
Files changed (1) hide show
  1. tests/test_examples.py +5 -5
tests/test_examples.py CHANGED
@@ -85,8 +85,8 @@ async def test_echo():
85
 
86
  async with Client(mcp) as client:
87
  result = await client.get_prompt("echo", {"text": "hello"})
88
- assert len(result) == 1
89
- assert isinstance(result[0], PromptMessage)
90
- assert isinstance(result[0].content, TextContent)
91
- assert isinstance(result[0].content.text, str)
92
- assert result[0].content.text == "hello"
 
85
 
86
  async with Client(mcp) as client:
87
  result = await client.get_prompt("echo", {"text": "hello"})
88
+ assert len(result.messages) == 1
89
+ assert isinstance(result.messages[0], PromptMessage)
90
+ assert isinstance(result.messages[0].content, TextContent)
91
+ assert isinstance(result.messages[0].content.text, str)
92
+ assert result.messages[0].content.text == "hello"