Skip to content

Commit 4b76833

Browse files
committed
Add @mcp.resource to ResourceLink tests tools
1 parent 0f0c9de commit 4b76833

File tree

2 files changed

+7
-5
lines changed

2 files changed

+7
-5
lines changed

tests/mcp_server.py

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

6060

6161
@mcp.tool()
62+
@mcp.resource(Path(__file__).parent.joinpath('assets/kiwi.png').absolute().as_uri())
6263
async def get_image_resource_link() -> ResourceLink:
6364
return ResourceLink(
6465
type='resource_link',
65-
uri=AnyUrl(Path(__file__).parent.joinpath('assets/kiwi.png').absolute().as_uri()),
66+
uri=AnyUrl('resource://kiwi.png'),
6667
name='kiwi.png',
6768
)
6869

@@ -81,10 +82,11 @@ async def get_audio_resource() -> EmbeddedResource:
8182

8283

8384
@mcp.tool()
85+
@mcp.resource(Path(__file__).parent.joinpath('assets/marcelo.mp3').absolute().as_uri())
8486
async def get_audio_resource_link() -> ResourceLink:
8587
return ResourceLink(
8688
type='resource_link',
87-
uri=AnyUrl(Path(__file__).parent.joinpath('assets/marcelo.mp3').absolute().as_uri()),
89+
uri=AnyUrl('resource://marcelo.mp3'),
8890
name='marcelo.mp3',
8991
)
9092

@@ -101,10 +103,11 @@ async def get_product_name() -> EmbeddedResource:
101103

102104

103105
@mcp.tool()
106+
@mcp.resource(Path(__file__).parent.joinpath('assets/product_name.txt').absolute().as_uri())
104107
async def get_product_name_link() -> ResourceLink:
105108
return ResourceLink(
106109
type='resource_link',
107-
uri=AnyUrl(Path(__file__).parent.joinpath('assets/product_name.txt').absolute().as_uri()),
110+
uri=AnyUrl('resource://product_name.txt'),
108111
name='product_name.txt',
109112
)
110113

tests/models/cassettes/test_model_names/test_known_model_names.yaml

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ interactions:
1515
response:
1616
headers:
1717
content-length:
18-
- '550'
18+
- '545'
1919
content-security-policy:
2020
- default-src 'none'; frame-ancestors 'none'
2121
content-type:
@@ -46,7 +46,6 @@ interactions:
4646
- text-to-text
4747
- model_id: claude-4-sonnet
4848
regions:
49-
- eu
5049
- us
5150
type:
5251
- text-to-text

0 commit comments

Comments
 (0)