Typing fixes

Strict pyright mode results in a lot of issues regarding non fully
determined types, due to Generics. These are some issues I came across
today. We are still far from being clean on pyright.
This commit is contained in:
David Soria Parra
2024-10-11 10:55:55 +01:00
parent 34937b72f2
commit 04ad96e6cd
2 changed files with 4 additions and 5 deletions

View File

@@ -20,6 +20,8 @@ from mcp_python.types import (
CompleteRequest,
ErrorData,
JSONRPCMessage,
ListPromptsRequest,
ListPromptsResult,
ListResourcesRequest,
ListResourcesResult,
LoggingLevel,
@@ -57,8 +59,6 @@ class Server:
return request_ctx.get()
def list_prompts(self):
from mcp_python.types import ListPromptsRequest, ListPromptsResult
def decorator(func: Callable[[], Awaitable[list[Prompt]]]):
logger.debug(f"Registering handler for PromptListRequest")
@@ -90,7 +90,7 @@ class Server:
async def handler(req: GetPromptRequest):
prompt_get = await func(req.params.name, req.params.arguments)
messages = []
messages: list[SamplingMessage] = []
for message in prompt_get.messages:
match message.content:
case str() as text_content: