33from anyio .streams .memory import MemoryObjectReceiveStream , MemoryObjectSendStream
44from pydantic import AnyUrl
55
6- from mcp_python .shared .session import BaseSession
7- from mcp_python .shared .version import SUPPORTED_PROTOCOL_VERSIONS
8- from mcp_python .types import (
6+ from mcp .shared .session import BaseSession
7+ from mcp .shared .version import SUPPORTED_PROTOCOL_VERSIONS
8+ from mcp .types import (
99 LATEST_PROTOCOL_VERSION ,
1010 CallToolResult ,
1111 ClientCapabilities ,
@@ -56,7 +56,7 @@ def __init__(
5656 )
5757
5858 async def initialize (self ) -> InitializeResult :
59- from mcp_python .types import (
59+ from mcp .types import (
6060 InitializeRequest ,
6161 InitializeRequestParams ,
6262 )
@@ -77,7 +77,7 @@ async def initialize(self) -> InitializeResult:
7777 listChanged = True
7878 ),
7979 ),
80- clientInfo = Implementation (name = "mcp_python " , version = "0.1.0" ),
80+ clientInfo = Implementation (name = "mcp " , version = "0.1.0" ),
8181 ),
8282 )
8383 ),
@@ -100,7 +100,7 @@ async def initialize(self) -> InitializeResult:
100100
101101 async def send_ping (self ) -> EmptyResult :
102102 """Send a ping request."""
103- from mcp_python .types import PingRequest
103+ from mcp .types import PingRequest
104104
105105 return await self .send_request (
106106 ClientRequest (
@@ -115,7 +115,7 @@ async def send_progress_notification(
115115 self , progress_token : str | int , progress : float , total : float | None = None
116116 ) -> None :
117117 """Send a progress notification."""
118- from mcp_python .types import (
118+ from mcp .types import (
119119 ProgressNotification ,
120120 ProgressNotificationParams ,
121121 )
@@ -135,7 +135,7 @@ async def send_progress_notification(
135135
136136 async def set_logging_level (self , level : LoggingLevel ) -> EmptyResult :
137137 """Send a logging/setLevel request."""
138- from mcp_python .types import (
138+ from mcp .types import (
139139 SetLevelRequest ,
140140 SetLevelRequestParams ,
141141 )
@@ -152,7 +152,7 @@ async def set_logging_level(self, level: LoggingLevel) -> EmptyResult:
152152
153153 async def list_resources (self ) -> ListResourcesResult :
154154 """Send a resources/list request."""
155- from mcp_python .types import (
155+ from mcp .types import (
156156 ListResourcesRequest ,
157157 )
158158
@@ -167,7 +167,7 @@ async def list_resources(self) -> ListResourcesResult:
167167
168168 async def read_resource (self , uri : AnyUrl ) -> ReadResourceResult :
169169 """Send a resources/read request."""
170- from mcp_python .types import (
170+ from mcp .types import (
171171 ReadResourceRequest ,
172172 ReadResourceRequestParams ,
173173 )
@@ -184,7 +184,7 @@ async def read_resource(self, uri: AnyUrl) -> ReadResourceResult:
184184
185185 async def subscribe_resource (self , uri : AnyUrl ) -> EmptyResult :
186186 """Send a resources/subscribe request."""
187- from mcp_python .types import (
187+ from mcp .types import (
188188 SubscribeRequest ,
189189 SubscribeRequestParams ,
190190 )
@@ -201,7 +201,7 @@ async def subscribe_resource(self, uri: AnyUrl) -> EmptyResult:
201201
202202 async def unsubscribe_resource (self , uri : AnyUrl ) -> EmptyResult :
203203 """Send a resources/unsubscribe request."""
204- from mcp_python .types import (
204+ from mcp .types import (
205205 UnsubscribeRequest ,
206206 UnsubscribeRequestParams ,
207207 )
@@ -220,7 +220,7 @@ async def call_tool(
220220 self , name : str , arguments : dict | None = None
221221 ) -> CallToolResult :
222222 """Send a tools/call request."""
223- from mcp_python .types import (
223+ from mcp .types import (
224224 CallToolRequest ,
225225 CallToolRequestParams ,
226226 )
@@ -237,7 +237,7 @@ async def call_tool(
237237
238238 async def list_prompts (self ) -> ListPromptsResult :
239239 """Send a prompts/list request."""
240- from mcp_python .types import ListPromptsRequest
240+ from mcp .types import ListPromptsRequest
241241
242242 return await self .send_request (
243243 ClientRequest (
@@ -252,7 +252,7 @@ async def get_prompt(
252252 self , name : str , arguments : dict [str , str ] | None = None
253253 ) -> GetPromptResult :
254254 """Send a prompts/get request."""
255- from mcp_python .types import GetPromptRequest , GetPromptRequestParams
255+ from mcp .types import GetPromptRequest , GetPromptRequestParams
256256
257257 return await self .send_request (
258258 ClientRequest (
@@ -268,7 +268,7 @@ async def complete(
268268 self , ref : ResourceReference | PromptReference , argument : dict
269269 ) -> CompleteResult :
270270 """Send a completion/complete request."""
271- from mcp_python .types import (
271+ from mcp .types import (
272272 CompleteRequest ,
273273 CompleteRequestParams ,
274274 CompletionArgument ,
@@ -289,7 +289,7 @@ async def complete(
289289
290290 async def list_tools (self ) -> ListToolsResult :
291291 """Send a tools/list request."""
292- from mcp_python .types import ListToolsRequest
292+ from mcp .types import ListToolsRequest
293293
294294 return await self .send_request (
295295 ClientRequest (
@@ -302,7 +302,7 @@ async def list_tools(self) -> ListToolsResult:
302302
303303 async def send_roots_list_changed (self ) -> None :
304304 """Send a roots/list_changed notification."""
305- from mcp_python .types import RootsListChangedNotification
305+ from mcp .types import RootsListChangedNotification
306306
307307 await self .send_notification (
308308 ClientNotification (
0 commit comments