Fix inputSchema attribute usage

Signed-off-by: Anuradha Karuppiah <[email protected]>

Author: AnuradhaKaruppiah

src/aiq/cli/commands/info/list_mcp.py

@@ -150,9 +150,6 @@ def list_mcp(ctx, direct, url, client_type, command, args, env, tool, detail, js
         if not command:
             click.echo("[ERROR] --command is required when using stdio client type", err=True)
             return
-        if url:
-            click.echo("[ERROR] --url is not allowed when using stdio client type", err=True)
-            return
 
     if client_type == 'sse':
         if not url:

src/aiq/tool/mcp/mcp_client.py

@@ -18,7 +18,8 @@
 import logging
 from abc import ABC
 from abc import abstractmethod
-from contextlib import AsyncContextManager
+from collections.abc import Callable
+from contextlib import AbstractAsyncContextManager
 from contextlib import asynccontextmanager
 from enum import Enum
 from typing import Any
@@ -101,7 +102,7 @@ def __init__(self, client_type: str = 'sse'):
 
     @abstractmethod
     @asynccontextmanager
-    async def connect_to_server(self) -> AsyncContextManager[ClientSession]:
+    async def connect_to_server(self):
         """
         Establish a session with an MCP server within an async context
         """
@@ -119,7 +120,7 @@ async def get_tools(self):
                 MCPToolClient(connect_fn=self.connect_to_server,
                               tool_name=tool.name,
                               tool_description=tool.description,
-                              tool_input_schema=tool.input_schema)
+                              tool_input_schema=tool.inputSchema)
             for tool in response.tools
         }
 
@@ -164,7 +165,7 @@ def __init__(self, url: str, client_type: str = 'sse'):
         self._url = url
 
     @asynccontextmanager
-    async def connect_to_server(self) -> AsyncContextManager[ClientSession]:
+    async def connect_to_server(self):
         """
         Establish a session with an MCP SSE server within an async context
         """
@@ -215,7 +216,7 @@ async def stop_persistent_session(self):
             self._session_cm = None
 
     @asynccontextmanager
-    async def connect_to_server(self) -> AsyncContextManager[ClientSession]:
+    async def connect_to_server(self):
         """
         Establish a session with an MCP server via stdio within an async context
         """
@@ -241,7 +242,7 @@ class MCPToolClient:
     """
 
     def __init__(self,
-                 connect_fn: callable[[], AsyncContextManager[ClientSession]],
+                 connect_fn: Callable[[], AbstractAsyncContextManager[ClientSession]],
                  tool_name: str,
                  tool_description: str | None,
                  tool_input_schema: dict | None = None):