Update Icon.sizes to use string array format (#1411)
This commit is contained in:
parent
60f4b2d10a
commit
80c0d23d48
1 changed files with 13 additions and 13 deletions
|
|
@ -15,7 +15,7 @@ async def test_icons_and_website_url():
|
|||
test_icon = Icon(
|
||||
src="data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mNkYPhfDwAChwGA60e6kgAAAABJRU5ErkJggg==",
|
||||
mimeType="image/png",
|
||||
sizes="1x1",
|
||||
sizes=["1x1"],
|
||||
)
|
||||
|
||||
# Create server with website URL and icon
|
||||
|
|
@ -80,9 +80,9 @@ async def test_multiple_icons():
|
|||
"""Test that multiple icons can be added to tools, resources, and prompts."""
|
||||
|
||||
# Create multiple test icons
|
||||
icon1 = Icon(src="data:image/png;base64,icon1", mimeType="image/png", sizes="16x16")
|
||||
icon2 = Icon(src="data:image/png;base64,icon2", mimeType="image/png", sizes="32x32")
|
||||
icon3 = Icon(src="data:image/png;base64,icon3", mimeType="image/png", sizes="64x64")
|
||||
icon1 = Icon(src="data:image/png;base64,icon1", mimeType="image/png", sizes=["16x16"])
|
||||
icon2 = Icon(src="data:image/png;base64,icon2", mimeType="image/png", sizes=["32x32"])
|
||||
icon3 = Icon(src="data:image/png;base64,icon3", mimeType="image/png", sizes=["64x64"])
|
||||
|
||||
mcp = FastMCP("MultiIconServer")
|
||||
|
||||
|
|
@ -98,9 +98,9 @@ async def test_multiple_icons():
|
|||
tool = tools[0]
|
||||
assert tool.icons is not None
|
||||
assert len(tool.icons) == 3
|
||||
assert tool.icons[0].sizes == "16x16"
|
||||
assert tool.icons[1].sizes == "32x32"
|
||||
assert tool.icons[2].sizes == "64x64"
|
||||
assert tool.icons[0].sizes == ["16x16"]
|
||||
assert tool.icons[1].sizes == ["32x32"]
|
||||
assert tool.icons[2].sizes == ["64x64"]
|
||||
|
||||
|
||||
async def test_no_icons_or_website():
|
||||
|
|
|
|||
Loading…
Reference in a new issue