show-server-info - fixes

Author: barnuri

src/mcp/types.py

@@ -5,7 +5,6 @@
     Generic,
     Literal,
     TypeAlias,
-    TypedDict,
     TypeVar,
 )
 
@@ -74,7 +73,9 @@ class Meta(BaseModel):
 
 
 RequestParamsT = TypeVar("RequestParamsT", bound=RequestParams | dict[str, Any] | None)
-NotificationParamsT = TypeVar("NotificationParamsT", bound=NotificationParams | dict[str, Any] | None)
+NotificationParamsT = TypeVar(
+    "NotificationParamsT", bound=NotificationParams | dict[str, Any] | None
+)
 MethodT = TypeVar("MethodT", bound=str)
 
 
@@ -190,7 +191,9 @@ class JSONRPCError(BaseModel):
     model_config = ConfigDict(extra="allow")
 
 
-class JSONRPCMessage(RootModel[JSONRPCRequest | JSONRPCNotification | JSONRPCResponse | JSONRPCError]):
+class JSONRPCMessage(
+    RootModel[JSONRPCRequest | JSONRPCNotification | JSONRPCResponse | JSONRPCError]
+):
     pass
 
 
@@ -311,7 +314,9 @@ class InitializeResult(Result):
     """Instructions describing how to use the server and its features."""
 
 
-class InitializedNotification(Notification[NotificationParams | None, Literal["notifications/initialized"]]):
+class InitializedNotification(
+    Notification[NotificationParams | None, Literal["notifications/initialized"]]
+):
     """
     This notification is sent from the client to the server after initialization has
     finished.
@@ -354,7 +359,9 @@ class ProgressNotificationParams(NotificationParams):
     model_config = ConfigDict(extra="allow")
 
 
-class ProgressNotification(Notification[ProgressNotificationParams, Literal["notifications/progress"]]):
+class ProgressNotification(
+    Notification[ProgressNotificationParams, Literal["notifications/progress"]]
+):
     """
     An out-of-band notification used to inform the receiver of a progress update for a
     long-running request.
@@ -450,7 +457,9 @@ class ReadResourceRequestParams(RequestParams):
     model_config = ConfigDict(extra="allow")
 
 
-class ReadResourceRequest(Request[ReadResourceRequestParams, Literal["resources/read"]]):
+class ReadResourceRequest(
+    Request[ReadResourceRequestParams, Literal["resources/read"]]
+):
     """Sent from the client to the server, to read a specific resource URI."""
 
     method: Literal["resources/read"]
@@ -491,7 +500,9 @@ class ReadResourceResult(Result):
 
 
 class ResourceListChangedNotification(
-    Notification[NotificationParams | None, Literal["notifications/resources/list_changed"]]
+    Notification[
+        NotificationParams | None, Literal["notifications/resources/list_changed"]
+    ]
 ):
     """
     An optional notification from the server to the client, informing it that the list
@@ -531,7 +542,9 @@ class UnsubscribeRequestParams(RequestParams):
     model_config = ConfigDict(extra="allow")
 
 
-class UnsubscribeRequest(Request[UnsubscribeRequestParams, Literal["resources/unsubscribe"]]):
+class UnsubscribeRequest(
+    Request[UnsubscribeRequestParams, Literal["resources/unsubscribe"]]
+):
     """
     Sent from the client to request cancellation of resources/updated notifications from
     the server.
@@ -553,7 +566,9 @@ class ResourceUpdatedNotificationParams(NotificationParams):
 
 
 class ResourceUpdatedNotification(
-    Notification[ResourceUpdatedNotificationParams, Literal["notifications/resources/updated"]]
+    Notification[
+        ResourceUpdatedNotificationParams, Literal["notifications/resources/updated"]
+    ]
 ):
     """
     A notification from the server to the client, informing it that a resource has
@@ -681,7 +696,9 @@ class GetPromptResult(Result):
 
 
 class PromptListChangedNotification(
-    Notification[NotificationParams | None, Literal["notifications/prompts/list_changed"]]
+    Notification[
+        NotificationParams | None, Literal["notifications/prompts/list_changed"]
+    ]
 ):
     """
     An optional notification from the server to the client, informing it that the list
@@ -788,7 +805,9 @@ class CallToolResult(Result):
     isError: bool = False
 
 
-class ToolListChangedNotification(Notification[NotificationParams | None, Literal["notifications/tools/list_changed"]]):
+class ToolListChangedNotification(
+    Notification[NotificationParams | None, Literal["notifications/tools/list_changed"]]
+):
     """
     An optional notification from the server to the client, informing it that the list
     of tools it offers has changed.
@@ -798,7 +817,9 @@ class ToolListChangedNotification(Notification[NotificationParams | None, Litera
     params: NotificationParams | None = None
 
 
-LoggingLevel = Literal["debug", "info", "notice", "warning", "error", "critical", "alert", "emergency"]
+LoggingLevel = Literal[
+    "debug", "info", "notice", "warning", "error", "critical", "alert", "emergency"
+]
 
 
 class SetLevelRequestParams(RequestParams):
@@ -831,7 +852,9 @@ class LoggingMessageNotificationParams(NotificationParams):
     model_config = ConfigDict(extra="allow")
 
 
-class LoggingMessageNotification(Notification[LoggingMessageNotificationParams, Literal["notifications/message"]]):
+class LoggingMessageNotification(
+    Notification[LoggingMessageNotificationParams, Literal["notifications/message"]]
+):
     """Notification of a log message passed from server to client."""
 
     method: Literal["notifications/message"]
@@ -926,7 +949,9 @@ class CreateMessageRequestParams(RequestParams):
     model_config = ConfigDict(extra="allow")
 
 
-class CreateMessageRequest(Request[CreateMessageRequestParams, Literal["sampling/createMessage"]]):
+class CreateMessageRequest(
+    Request[CreateMessageRequestParams, Literal["sampling/createMessage"]]
+):
     """A request from the server to sample an LLM via the client."""
 
     method: Literal["sampling/createMessage"]
@@ -1083,7 +1108,9 @@ class CancelledNotificationParams(NotificationParams):
     model_config = ConfigDict(extra="allow")
 
 
-class CancelledNotification(Notification[CancelledNotificationParams, Literal["notifications/cancelled"]]):
+class CancelledNotification(
+    Notification[CancelledNotificationParams, Literal["notifications/cancelled"]]
+):
     """
     This notification can be sent by either side to indicate that it is canceling a
     previously-issued request.
@@ -1114,7 +1141,12 @@ class ClientRequest(
 
 
 class ClientNotification(
-    RootModel[CancelledNotification | ProgressNotification | InitializedNotification | RootsListChangedNotification]
+    RootModel[
+        CancelledNotification
+        | ProgressNotification
+        | InitializedNotification
+        | RootsListChangedNotification
+    ]
 ):
     pass