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

Add pyright strict mode on the whole project (#1254)

Browse source
This commit is contained in:
Marcelo Trylesinski 2025-08-11 19:56:37 +02:00 • committed by GitHub
commit c7671e470c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 652 additions and 674 deletions
Download patch file Download diff file Expand all files Collapse all files

42
examples/servers/simple-prompt/mcp_simple_prompt/server.py
View file

@ -2,22 +2,19 @@ import anyio
import click
import mcp.types as types
from mcp.server.lowlevel import Server
from starlette.requests import Request
def create_messages(
context: str | None = None, topic: str | None = None
) -> list[types.PromptMessage]:
def create_messages(context: str | None = None, topic: str | None = None) -> list[types.PromptMessage]:
"""Create the messages for the prompt."""
messages = []
messages: list[types.PromptMessage] = []
# Add context if provided
if context:
messages.append(
types.PromptMessage(
role="user",
content=types.TextContent(
type="text", text=f"Here is some relevant context: {context}"
),
content=types.TextContent(type="text", text=f"Here is some relevant context: {context}"),
)
)
@ -28,11 +25,7 @@ def create_messages(
else:
prompt += "whatever questions I may have."
messages.append(
types.PromptMessage(
role="user", content=types.TextContent(type="text", text=prompt)
)
)
messages.append(types.PromptMessage(role="user", content=types.TextContent(type="text", text=prompt)))
return messages
@ -54,8 +47,7 @@ def main(port: int, transport: str) -> int:
types.Prompt(
name="simple",
title="Simple Assistant Prompt",
description="A simple prompt that can take optional context and topic "
"arguments",
description="A simple prompt that can take optional context and topic arguments",
arguments=[
types.PromptArgument(
name="context",
@ -72,9 +64,7 @@ def main(port: int, transport: str) -> int:
]
@app.get_prompt()
async def get_prompt(
name: str, arguments: dict[str, str] | None = None
) -> types.GetPromptResult:
async def get_prompt(name: str, arguments: dict[str, str] | None = None) -> types.GetPromptResult:
if name != "simple":
raise ValueError(f"Unknown prompt: {name}")
@ -82,9 +72,7 @@ def main(port: int, transport: str) -> int:
arguments = {}
return types.GetPromptResult(
messages=create_messages(
context=arguments.get("context"), topic=arguments.get("topic")
),
messages=create_messages(context=arguments.get("context"), topic=arguments.get("topic")),
description="A simple prompt with optional context and topic arguments",
)
@ -96,13 +84,9 @@ def main(port: int, transport: str) -> int:
sse = SseServerTransport("/messages/")
async def handle_sse(request):
async with sse.connect_sse(
request.scope, request.receive, request._send
) as streams:
await app.run(
streams[0], streams[1], app.create_initialization_options()
)
async def handle_sse(request: Request):
async with sse.connect_sse(request.scope, request.receive, request._send) as streams: # type: ignore[reportPrivateUsage]
await app.run(streams[0], streams[1], app.create_initialization_options())
return Response()
starlette_app = Starlette(
@ -121,9 +105,7 @@ def main(port: int, transport: str) -> int:
async def arun():
async with stdio_server() as streams:
await app.run(
streams[0], streams[1], app.create_initialization_options()
)
await app.run(streams[0], streams[1], app.create_initialization_options())
anyio.run(arun)

2
examples/servers/simple-prompt/pyproject.toml
View file

@ -40,7 +40,7 @@ select = ["E", "F", "I"]
ignore = []
[tool.ruff]
line-length = 88
line-length = 120
target-version = "py310"
[tool.uv]