fix: change Resource URI fields from AnyUrl to str (#1863)
This commit is contained in:
parent
f2abefffaf
commit
cfb2909631
1 changed files with 343 additions and 179 deletions
|
|
@ -3,7 +3,6 @@ from pathlib import Path
|
|||
from tempfile import NamedTemporaryFile
|
||||
|
||||
import pytest
|
||||
from pydantic import AnyUrl, FileUrl
|
||||
|
||||
import mcp.types as types
|
||||
from mcp.server.lowlevel.server import ReadResourceContents, Server
|
||||
|
|
@ -27,7 +26,7 @@ async def test_read_resource_text(temp_file: Path):
|
|||
server = Server("test")
|
||||
|
||||
@server.read_resource()
|
||||
async def read_resource(uri: AnyUrl) -> Iterable[ReadResourceContents]:
|
||||
async def read_resource(uri: str) -> Iterable[ReadResourceContents]:
|
||||
return [ReadResourceContents(content="Hello World", mime_type="text/plain")]
|
||||
|
||||
# Get the handler directly from the server
|
||||
|
|
@ -35,7 +34,7 @@ async def test_read_resource_text(temp_file: Path):
|
|||
|
||||
# Create a request
|
||||
request = types.ReadResourceRequest(
|
||||
params=types.ReadResourceRequestParams(uri=FileUrl(temp_file.as_uri())),
|
||||
params=types.ReadResourceRequestParams(uri=temp_file.as_uri()),
|
||||
)
|
||||
|
||||
# Call the handler
|
||||
|
|
@ -54,7 +53,7 @@ async def test_read_resource_binary(temp_file: Path):
|
|||
server = Server("test")
|
||||
|
||||
@server.read_resource()
|
||||
async def read_resource(uri: AnyUrl) -> Iterable[ReadResourceContents]:
|
||||
async def read_resource(uri: str) -> Iterable[ReadResourceContents]:
|
||||
return [ReadResourceContents(content=b"Hello World", mime_type="application/octet-stream")]
|
||||
|
||||
# Get the handler directly from the server
|
||||
|
|
@ -62,7 +61,7 @@ async def test_read_resource_binary(temp_file: Path):
|
|||
|
||||
# Create a request
|
||||
request = types.ReadResourceRequest(
|
||||
params=types.ReadResourceRequestParams(uri=FileUrl(temp_file.as_uri())),
|
||||
params=types.ReadResourceRequestParams(uri=temp_file.as_uri()),
|
||||
)
|
||||
|
||||
# Call the handler
|
||||
|
|
@ -80,7 +79,7 @@ async def test_read_resource_default_mime(temp_file: Path):
|
|||
server = Server("test")
|
||||
|
||||
@server.read_resource()
|
||||
async def read_resource(uri: AnyUrl) -> Iterable[ReadResourceContents]:
|
||||
async def read_resource(uri: str) -> Iterable[ReadResourceContents]:
|
||||
return [
|
||||
ReadResourceContents(
|
||||
content="Hello World",
|
||||
|
|
@ -93,7 +92,7 @@ async def test_read_resource_default_mime(temp_file: Path):
|
|||
|
||||
# Create a request
|
||||
request = types.ReadResourceRequest(
|
||||
params=types.ReadResourceRequestParams(uri=FileUrl(temp_file.as_uri())),
|
||||
params=types.ReadResourceRequestParams(uri=temp_file.as_uri()),
|
||||
)
|
||||
|
||||
# Call the handler
|
||||
|
|
|
|||
Loading…
Reference in a new issue