Skip to content

Commit 0253a8b

Browse files
Tapan Chughfelixweinberger
authored andcommitted
replace deprecated with legacy
1 parent bf620e2 commit 0253a8b

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
@@ -346,26 +346,26 @@ async def select_favorite_colors(ctx: Context[ServerSession, None]) -> str:
346346
return f"User: {result.data.user_name}, Colors: {', '.join(result.data.favorite_colors)}"
347347
return f"User {result.action}"
348348

349-
# Test deprecated enumNames format
350-
class DeprecatedColorSchema(BaseModel):
349+
# Test legacy enumNames format
350+
class LegacyColorSchema(BaseModel):
351351
user_name: str = Field(description="Your name")
352352
color: str = Field(
353353
description="Select a color",
354354
json_schema_extra={"enum": ["red", "green", "blue"], "enumNames": ["Red", "Green", "Blue"]},
355355
)
356356

357-
@mcp.tool(description="Deprecated enum format")
358-
async def select_color_deprecated(ctx: Context[ServerSession, None]) -> str:
359-
result = await ctx.elicit(message="Select a color (deprecated format)", schema=DeprecatedColorSchema)
357+
@mcp.tool(description="Legacy enum format")
358+
async def select_color_legacy(ctx: Context[ServerSession, None]) -> str:
359+
result = await ctx.elicit(message="Select a color (legacy format)", schema=LegacyColorSchema)
360360
if result.action == "accept" and result.data:
361361
return f"User: {result.data.user_name}, Color: {result.data.color}"
362362
return f"User {result.action}"
363363

364364
async def enum_callback(context: RequestContext[ClientSession, Any], params: ElicitRequestParams):
365-
if "colors" in params.message and "deprecated" not in params.message:
365+
if "colors" in params.message and "legacy" not in params.message:
366366
return ElicitResult(action="accept", content={"user_name": "Bob", "favorite_colors": ["red", "green"]})
367367
elif "color" in params.message:
368-
if "deprecated" in params.message:
368+
if "legacy" in params.message:
369369
return ElicitResult(action="accept", content={"user_name": "Charlie", "color": "green"})
370370
else:
371371
return ElicitResult(action="accept", content={"user_name": "Alice", "favorite_color": "blue"})
@@ -377,5 +377,5 @@ async def enum_callback(context: RequestContext[ClientSession, Any], params: Eli
377377
# Test multi-select with titles
378378
await call_tool_and_assert(mcp, enum_callback, "select_favorite_colors", {}, "User: Bob, Colors: red, green")
379379

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

0 commit comments

Comments
 (0)