Skip to content

Commit 07adc29

Browse files
author
Tapan Chugh
committed
replace deprecated with legacy
1 parent 2854f32 commit 07adc29

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

tests/server/fastmcp/test_elicitation.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -340,26 +340,26 @@ async def select_favorite_colors(ctx: Context[ServerSession, None]) -> str:
340340
return f"User: {result.data.user_name}, Colors: {', '.join(result.data.favorite_colors)}"
341341
return f"User {result.action}"
342342

343-
# Test deprecated enumNames format
344-
class DeprecatedColorSchema(BaseModel):
343+
# Test legacy enumNames format
344+
class LegacyColorSchema(BaseModel):
345345
user_name: str = Field(description="Your name")
346346
color: str = Field(
347347
description="Select a color",
348348
json_schema_extra={"enum": ["red", "green", "blue"], "enumNames": ["Red", "Green", "Blue"]},
349349
)
350350

351-
@mcp.tool(description="Deprecated enum format")
352-
async def select_color_deprecated(ctx: Context[ServerSession, None]) -> str:
353-
result = await ctx.elicit(message="Select a color (deprecated format)", schema=DeprecatedColorSchema)
351+
@mcp.tool(description="Legacy enum format")
352+
async def select_color_legacy(ctx: Context[ServerSession, None]) -> str:
353+
result = await ctx.elicit(message="Select a color (legacy format)", schema=LegacyColorSchema)
354354
if result.action == "accept" and result.data:
355355
return f"User: {result.data.user_name}, Color: {result.data.color}"
356356
return f"User {result.action}"
357357

358358
async def enum_callback(context: RequestContext[ClientSession, Any], params: ElicitRequestParams):
359-
if "colors" in params.message and "deprecated" not in params.message:
359+
if "colors" in params.message and "legacy" not in params.message:
360360
return ElicitResult(action="accept", content={"user_name": "Bob", "favorite_colors": ["red", "green"]})
361361
elif "color" in params.message:
362-
if "deprecated" in params.message:
362+
if "legacy" in params.message:
363363
return ElicitResult(action="accept", content={"user_name": "Charlie", "color": "green"})
364364
else:
365365
return ElicitResult(action="accept", content={"user_name": "Alice", "favorite_color": "blue"})
@@ -371,5 +371,5 @@ async def enum_callback(context: RequestContext[ClientSession, Any], params: Eli
371371
# Test multi-select with titles
372372
await call_tool_and_assert(mcp, enum_callback, "select_favorite_colors", {}, "User: Bob, Colors: red, green")
373373

374-
# Test deprecated enumNames format
375-
await call_tool_and_assert(mcp, enum_callback, "select_color_deprecated", {}, "User: Charlie, Color: green")
374+
# Test legacy enumNames format
375+
await call_tool_and_assert(mcp, enum_callback, "select_color_legacy", {}, "User: Charlie, Color: green")

0 commit comments

Comments
 (0)