@@ -22,8 +22,7 @@ async def test_client_session_satisfies_base_client_session_protocol():
2222 server = MCPServer (name = "test" )
2323
2424 @server .tool ()
25- def greet (name : str ) -> str :
26- return f"Hello, { name } !"
25+ def greet (name : str ) -> str : ...
2726
2827 async with Client (server ) as client :
2928 # Verify isinstance works with @runtime_checkable Protocol
@@ -63,8 +62,7 @@ async def test_base_client_session_complete_and_set_logging_level():
6362 server = MCPServer (name = "test" )
6463
6564 @server .prompt ()
66- def greeting (name : str ) -> str :
67- return f"Hello { name } !"
65+ def greeting (name : str ) -> str : ...
6866
6967 async with Client (server ) as client :
7068 session : BaseClientSession = client .session
@@ -90,15 +88,13 @@ async def send_request(
9088 request_read_timeout_seconds : float | None = None ,
9189 metadata : Any = None ,
9290 progress_callback : ProgressFnT | None = None ,
93- ) -> Any :
94- return types .EmptyResult ()
91+ ) -> Any : ...
9592
9693 async def send_notification (
9794 self ,
9895 notification : types .ClientNotification ,
9996 related_request_id : Any = None ,
100- ) -> None :
101- pass
97+ ) -> None : ...
10298
10399 async def send_progress_notification (
104100 self ,
@@ -108,35 +104,25 @@ async def send_progress_notification(
108104 message : str | None = None ,
109105 * ,
110106 meta : RequestParamsMeta | None = None ,
111- ) -> None :
112- pass
107+ ) -> None : ...
113108
114- async def initialize (self ) -> types .InitializeResult :
115- return types .InitializeResult (
116- protocol_version = "2024-11-05" ,
117- capabilities = types .ServerCapabilities (),
118- server_info = types .Implementation (name = "stub" , version = "0" ),
119- )
109+ async def initialize (self ) -> types .InitializeResult : ...
120110
121- async def send_ping (self , * , meta : RequestParamsMeta | None = None ) -> types .EmptyResult :
122- return types .EmptyResult ()
111+ async def send_ping (self , * , meta : RequestParamsMeta | None = None ) -> types .EmptyResult : ...
123112
124- async def list_resources (self , * , params : types .PaginatedRequestParams | None = None ) -> types .ListResourcesResult :
125- return types .ListResourcesResult (resources = [])
113+ async def list_resources (
114+ self , * , params : types .PaginatedRequestParams | None = None
115+ ) -> types .ListResourcesResult : ...
126116
127117 async def list_resource_templates (
128118 self , * , params : types .PaginatedRequestParams | None = None
129- ) -> types .ListResourceTemplatesResult :
130- return types .ListResourceTemplatesResult (resource_templates = [])
119+ ) -> types .ListResourceTemplatesResult : ...
131120
132- async def read_resource (self , uri : str , * , meta : RequestParamsMeta | None = None ) -> types .ReadResourceResult :
133- return types .ReadResourceResult (contents = [])
121+ async def read_resource (self , uri : str , * , meta : RequestParamsMeta | None = None ) -> types .ReadResourceResult : ...
134122
135- async def subscribe_resource (self , uri : str , * , meta : RequestParamsMeta | None = None ) -> types .EmptyResult :
136- return types .EmptyResult ()
123+ async def subscribe_resource (self , uri : str , * , meta : RequestParamsMeta | None = None ) -> types .EmptyResult : ...
137124
138- async def unsubscribe_resource (self , uri : str , * , meta : RequestParamsMeta | None = None ) -> types .EmptyResult :
139- return types .EmptyResult ()
125+ async def unsubscribe_resource (self , uri : str , * , meta : RequestParamsMeta | None = None ) -> types .EmptyResult : ...
140126
141127 async def call_tool (
142128 self ,
@@ -146,42 +132,35 @@ async def call_tool(
146132 progress_callback : ProgressFnT | None = None ,
147133 * ,
148134 meta : RequestParamsMeta | None = None ,
149- ) -> types .CallToolResult :
150- return types .CallToolResult (content = [])
135+ ) -> types .CallToolResult : ...
151136
152- async def list_prompts (self , * , params : types .PaginatedRequestParams | None = None ) -> types .ListPromptsResult :
153- return types .ListPromptsResult (prompts = [])
137+ async def list_prompts (self , * , params : types .PaginatedRequestParams | None = None ) -> types .ListPromptsResult : ...
154138
155139 async def get_prompt (
156140 self ,
157141 name : str ,
158142 arguments : dict [str , str ] | None = None ,
159143 * ,
160144 meta : RequestParamsMeta | None = None ,
161- ) -> types .GetPromptResult :
162- return types .GetPromptResult (messages = [])
145+ ) -> types .GetPromptResult : ...
163146
164- async def list_tools (self , * , params : types .PaginatedRequestParams | None = None ) -> types .ListToolsResult :
165- return types .ListToolsResult (tools = [])
147+ async def list_tools (self , * , params : types .PaginatedRequestParams | None = None ) -> types .ListToolsResult : ...
166148
167149 async def complete (
168150 self ,
169151 ref : types .ResourceTemplateReference | types .PromptReference ,
170152 argument : dict [str , str ],
171153 context_arguments : dict [str , str ] | None = None ,
172- ) -> types .CompleteResult :
173- return types .CompleteResult (completion = types .Completion (values = []))
154+ ) -> types .CompleteResult : ...
174155
175156 async def set_logging_level (
176157 self ,
177158 level : types .LoggingLevel ,
178159 * ,
179160 meta : RequestParamsMeta | None = None ,
180- ) -> types .EmptyResult :
181- return types .EmptyResult ()
161+ ) -> types .EmptyResult : ...
182162
183- async def send_roots_list_changed (self ) -> None :
184- pass
163+ async def send_roots_list_changed (self ) -> None : ...
185164
186165
187166def test_custom_session_satisfies_protocol ():
0 commit comments