@@ -43,7 +43,7 @@ async def _handle_agent_call(
4343 run_args : Arguments ,
4444 inputs : dict [str , str | None ],
4545 agent_name : str ,
46- mcp_server : Server , # type: ignore
46+ mcp_server : Server ,
4747 arguments : dict [str , t .Any ],
4848) -> Output :
4949 input_state = _get_input_state_from_request (inputs , arguments )
@@ -86,7 +86,7 @@ def _create_call_handler(
8686 run_args : Arguments ,
8787 inputs : dict [str , str | None ],
8888 agent_name : str ,
89- mcp_server : Server , # type: ignore
89+ mcp_server : Server ,
9090 runtime : Runtime | None ,
9191) -> t .Callable [[mcp_types .CallToolRequest ], t .Coroutine [t .Any , t .Any , mcp_types .ServerResult ]]:
9292 tools_dict = {tool .__name__ : tool for tool in runtime .tools } if runtime else {}
@@ -127,8 +127,8 @@ def create_mcp_server(
127127 runtime : Runtime | None ,
128128 serve_tools : bool = False ,
129129 tools_only : bool = False ,
130- ) -> Server : # type: ignore
131- server = Server (agent_name ) # type: ignore
130+ ) -> Server :
131+ server = Server (agent_name )
132132 tools : list [mcp_types .Tool ] = []
133133
134134 # create a tool for the agent itself
@@ -180,7 +180,7 @@ def create_mcp_server(
180180 return server
181181
182182
183- def create_sse_app (debug : bool , server : Server ) -> Starlette : # type: ignore
183+ def create_sse_app (debug : bool , server : Server ) -> Starlette :
184184 sse = SseServerTransport ("/messages/" )
185185
186186 async def handle_sse (request : t .Any ) -> t .Any :
@@ -196,7 +196,7 @@ async def handle_sse(request: t.Any) -> t.Any:
196196 )
197197
198198
199- async def serve_stdio_app (server : Server , agent_name : str ) -> None : # type: ignore
199+ async def serve_stdio_app (server : Server , agent_name : str ) -> None :
200200 async with stdio_server () as streams :
201201 logger .info (f"🌐 serving { agent_name } on stdout ..." )
202202 await server .run (streams [0 ], streams [1 ], server .create_initialization_options ())
0 commit comments