modelcontextprotocol/python-sdk
Watch
1
0
Fork
You've already forked python-sdk
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix streamable http sampling (#693)
Some checks failed
Main branch checks / checks (push) Has been cancelled
Details

Browse source
This commit is contained in:
ihrpr 2025-05-12 18:31:35 +01:00 • committed by GitHub
commit c6fb822c86
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 152 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

10
src/mcp/server/session.py
View file

@ -47,7 +47,7 @@ from pydantic import AnyUrl
import mcp.types as types
from mcp.server.models import InitializationOptions
from mcp.shared.message import SessionMessage
from mcp.shared.message import ServerMessageMetadata, SessionMessage
from mcp.shared.session import (
BaseSession,
RequestResponder,
@ -230,10 +230,11 @@ class ServerSession(
stop_sequences: list[str] | None = None,
metadata: dict[str, Any] | None = None,
model_preferences: types.ModelPreferences | None = None,
related_request_id: types.RequestId | None = None,
) -> types.CreateMessageResult:
"""Send a sampling/create_message request."""
return await self.send_request(
types.ServerRequest(
request=types.ServerRequest(
types.CreateMessageRequest(
method="sampling/createMessage",
params=types.CreateMessageRequestParams(
@ -248,7 +249,10 @@ class ServerSession(
),
)
),
types.CreateMessageResult,
result_type=types.CreateMessageResult,
metadata=ServerMessageMetadata(
related_request_id=related_request_id,
),
)
async def list_roots(self) -> types.ListRootsResult: