add raise not implemented

Author: asheshvidyut

src/mcp/client/transport_session.py

@@ -1,5 +1,7 @@
-from abc import ABC, abstractmethod
+from abc import ABC
+from abc import abstractmethod
 from datetime import timedelta
+
 from typing import Any
 
 from pydantic import AnyUrl
@@ -14,10 +16,11 @@ class TransportSession(ABC):
     @abstractmethod
     async def initialize(self) -> types.InitializeResult:
         """Send an initialize request."""
-        ...
+        raise NotImplementedError
 
     @abstractmethod
-    async def send_ping(self): ...
+    async def send_ping(self):
+        raise NotImplementedError
 
     @abstractmethod
     async def send_progress_notification(
@@ -26,46 +29,47 @@ async def send_progress_notification(
         progress: float,
         total: float | None = None,
         message: str | None = None,
-    ) -> None: ...
+    ) -> None:
+        raise NotImplementedError
 
     @abstractmethod
     async def set_logging_level(
         self,
         level: types.LoggingLevel,
     ) -> types.EmptyResult:
         """Send a logging/setLevel request."""
-        ...
+        raise NotImplementedError
 
     @abstractmethod
     async def list_resources(
         self,
         cursor: str | None = None,
     ) -> types.ListResourcesResult:
         """Send a resources/list request."""
-        ...
+        raise NotImplementedError
 
     @abstractmethod
     async def list_resource_templates(
         self,
         cursor: str | None = None,
     ) -> types.ListResourceTemplatesResult:
         """Send a resources/templates/list request."""
-        ...
+        raise NotImplementedError
 
     @abstractmethod
     async def read_resource(self, uri: AnyUrl) -> types.ReadResourceResult:
         """Send a resources/read request."""
-        ...
+        raise NotImplementedError
 
     @abstractmethod
     async def subscribe_resource(self, uri: AnyUrl) -> types.EmptyResult:
         """Send a resources/subscribe request."""
-        ...
+        raise NotImplementedError
 
     @abstractmethod
     async def unsubscribe_resource(self, uri: AnyUrl) -> types.EmptyResult:
         """Send a resources/unsubscribe request."""
-        ...
+        raise NotImplementedError
 
     @abstractmethod
     async def call_tool(
@@ -76,7 +80,7 @@ async def call_tool(
         progress_callback: ProgressFnT | None = None,
     ) -> types.CallToolResult:
         """Send a tools/call request with optional progress callback support."""
-        ...
+        raise NotImplementedError
 
     @abstractmethod
     async def _validate_tool_result(
@@ -86,15 +90,15 @@ async def _validate_tool_result(
     ) -> None:
         """Validate the structured content of a tool result against its output
         schema."""
-        ...
+        raise NotImplementedError
 
     @abstractmethod
     async def list_prompts(
         self,
         cursor: str | None = None,
     ) -> types.ListPromptsResult:
         """Send a prompts/list request."""
-        ...
+        raise NotImplementedError
 
     @abstractmethod
     async def get_prompt(
@@ -103,7 +107,7 @@ async def get_prompt(
         arguments: dict[str, str] | None = None,
     ) -> types.GetPromptResult:
         """Send a prompts/get request."""
-        ...
+        raise NotImplementedError
 
     @abstractmethod
     async def complete(
@@ -113,17 +117,17 @@ async def complete(
         context_arguments: dict[str, str] | None = None,
     ) -> types.CompleteResult:
         """Send a completion/complete request."""
-        ...
+        raise NotImplementedError
 
     @abstractmethod
     async def list_tools(
         self,
         cursor: str | None = None,
     ) -> types.ListToolsResult:
         """Send a tools/list request."""
-        ...
+        raise NotImplementedError
 
     @abstractmethod
     async def send_roots_list_changed(self) -> None:
         """Send a roots/list_changed notification."""
-        ...
+        raise NotImplementedError

src/mcp/server/transport_session.py

@@ -6,8 +6,8 @@
 from anyio.streams.memory import MemoryObjectReceiveStream
 from pydantic import AnyUrl
 
-import mcp_grpc.types as types
-from mcp_grpc.server.session import ServerRequestResponder
+import mcp.types as types
+from mcp.server.session import ServerRequestResponder
 
 
 class TransportSession(abc.ABC):