Skip to content

Commit 0b4e81a

Browse files
committed
Fix mcp_server.py
1 parent 32d0300 commit 0b4e81a

File tree

2 files changed

+121
-96
lines changed

2 files changed

+121
-96
lines changed

tests/mcp_server.py

Lines changed: 15 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,6 @@ async def get_image_resource() -> EmbeddedResource:
5959

6060

6161
@mcp.tool()
62-
@mcp.resource(Path(__file__).parent.joinpath('assets/kiwi.png').absolute().as_uri())
6362
async def get_image_resource_link() -> ResourceLink:
6463
return ResourceLink(
6564
type='resource_link',
@@ -68,6 +67,11 @@ async def get_image_resource_link() -> ResourceLink:
6867
)
6968

7069

70+
@mcp.resource('resource://kiwi.png', mime_type='image/png')
71+
async def kiwi_resource() -> bytes:
72+
return Path(__file__).parent.joinpath('assets/kiwi.png').read_bytes()
73+
74+
7175
@mcp.tool()
7276
async def get_audio_resource() -> EmbeddedResource:
7377
data = Path(__file__).parent.joinpath('assets/marcelo.mp3').read_bytes()
@@ -82,7 +86,6 @@ async def get_audio_resource() -> EmbeddedResource:
8286

8387

8488
@mcp.tool()
85-
@mcp.resource(Path(__file__).parent.joinpath('assets/marcelo.mp3').absolute().as_uri())
8689
async def get_audio_resource_link() -> ResourceLink:
8790
return ResourceLink(
8891
type='resource_link',
@@ -91,6 +94,11 @@ async def get_audio_resource_link() -> ResourceLink:
9194
)
9295

9396

97+
@mcp.resource('resource://marcelo.mp3', mime_type='audio/mpeg')
98+
async def marcelo_resource() -> bytes:
99+
return Path(__file__).parent.joinpath('assets/marcelo.mp3').read_bytes()
100+
101+
94102
@mcp.tool()
95103
async def get_product_name() -> EmbeddedResource:
96104
return EmbeddedResource(
@@ -103,7 +111,6 @@ async def get_product_name() -> EmbeddedResource:
103111

104112

105113
@mcp.tool()
106-
@mcp.resource(Path(__file__).parent.joinpath('assets/product_name.txt').absolute().as_uri())
107114
async def get_product_name_link() -> ResourceLink:
108115
return ResourceLink(
109116
type='resource_link',
@@ -112,6 +119,11 @@ async def get_product_name_link() -> ResourceLink:
112119
)
113120

114121

122+
@mcp.resource('resource://product_name.txt', mime_type='text/plain')
123+
async def product_name_resource() -> str:
124+
return Path(__file__).parent.joinpath('assets/product_name.txt').read_text()
125+
126+
115127
@mcp.tool()
116128
async def get_image() -> Image:
117129
data = Path(__file__).parent.joinpath('assets/kiwi.png').read_bytes()

0 commit comments

Comments
 (0)