Spaces:
Running
Running
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"
|
| 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("
|
| 194 |
-
assert result.data == "
|
| 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():
|