Jeremiah Lowin commited on
Commit
7a96ca2
·
1 Parent(s): f1f97d2

Fix tests

Browse files
tests/client/test_streamable_http.py CHANGED
@@ -33,7 +33,7 @@ def fastmcp_server():
33
  """Elicit a response from the user."""
34
  result = await ctx.elicit("What is your name?", response_type=str)
35
  if result.action == "accept":
36
- return f"Hello, {result.data}!"
37
  else:
38
  return "No name provided"
39
 
@@ -190,8 +190,8 @@ async def test_elicitation_tool(streamable_http_server: str):
190
  transport=StreamableHttpTransport(streamable_http_server),
191
  elicitation_handler=elicitation_handler,
192
  ) as client:
193
- result = await client.call_tool("greet_with_progress", {"name": "Alice"})
194
- assert result.data == "Hello, Alice!"
195
 
196
 
197
  async def test_nested_streamable_http_server_resolves_correctly():
 
33
  """Elicit a response from the user."""
34
  result = await ctx.elicit("What is your name?", response_type=str)
35
  if result.action == "accept":
36
+ return f"You said your name was: {result.data}!"
37
  else:
38
  return "No name provided"
39
 
 
190
  transport=StreamableHttpTransport(streamable_http_server),
191
  elicitation_handler=elicitation_handler,
192
  ) as client:
193
+ result = await client.call_tool("elicit")
194
+ assert result.data == "You said your name was: Alice!"
195
 
196
 
197
  async def test_nested_streamable_http_server_resolves_correctly():