add list of tools to README.md; remove unused params from ydb_status tool

Author: blinkov

README.md

@@ -28,6 +28,29 @@ To get started with YDB MCP, you'll need to configure your MCP client to communi
 }
 ```
 
+## Available tools
+
+YDB MCP provides the following tools for interacting with YDB databases:
+
+- `ydb_query`: Run a SQL query against a YDB database
+  - Parameters:
+    - `sql`: SQL query string to execute
+
+- `ydb_query_with_params`: Run a parameterized SQL query with JSON parameters
+  - Parameters:
+    - `sql`: SQL query string with parameter placeholders
+    - `params`: JSON string containing parameter values
+
+- `ydb_list_directory`: List directory contents in YDB
+  - Parameters:
+    - `path`: YDB directory path to list
+
+- `ydb_describe_path`: Get detailed information about a YDB path (table, directory, etc.)
+  - Parameters:
+    - `path`: YDB path to describe
+
+- `ydb_status`: Get the current status of the YDB connection
+
 ## Development
 
 The project uses [Make](https://www.gnu.org/software/make/) as its primary development tool, providing a consistent interface for common development tasks.

tests/integration/test_mcp_server_integration.py

@@ -853,3 +853,34 @@ async def test_describe_nonexistent_path_integration(mcp_server):
 
     # Should contain an error message
     assert "Error" in item["text"], f"Expected error message, got: {item['text']}"
+
+
+async def test_ydb_status_integration(mcp_server):
+    """Test getting YDB connection status."""
+    result = await call_mcp_tool(mcp_server, "ydb_status")
+
+    # Parse the JSON result
+    assert isinstance(result, list), f"Expected list result, got {type(result)}"
+    assert len(result) > 0, "Expected non-empty result list"
+
+    item = result[0]
+    assert isinstance(item, dict), f"Expected dict item, got {type(item)}"
+    assert "type" in item, f"Missing 'type' field in item: {item}"
+    assert "text" in item, f"Missing 'text' field in item: {item}"
+
+    # Parse the JSON string within the text field
+    status_data = json.loads(item["text"])
+
+    # Verify the structure
+    assert "status" in status_data, f"Missing 'status' field in status_data: {status_data}"
+    assert "ydb_endpoint" in status_data, f"Missing 'ydb_endpoint' field in status_data: {status_data}"
+    assert "ydb_database" in status_data, f"Missing 'ydb_database' field in status_data: {status_data}"
+    assert "auth_mode" in status_data, f"Missing 'auth_mode' field in status_data: {status_data}"
+    assert "ydb_connection" in status_data, f"Missing 'ydb_connection' field in status_data: {status_data}"
+
+    # For a successful test run, we expect to be connected
+    assert status_data["status"] == "running", f"Expected status to be 'running', got {status_data['status']}"
+    assert status_data["ydb_connection"] == "connected", f"Expected ydb_connection to be 'connected', got {status_data['ydb_connection']}"
+    assert status_data["error"] is None, f"Expected no error, got: {status_data.get('error')}"
+
+    logger.info(f"YDB status check successful: {status_data}")

ydb_mcp/server.py

@@ -575,9 +575,9 @@ def register_tools(self):
                 "description": "Get the current status of the YDB connection",
                 "handler": self.get_connection_status,  # Use real handler
                 "parameters": {
-                    "properties": {"params": {"type": "string", "title": "params"}},
-                    "required": ["params"],
                     "type": "object",
+                    "properties": {},
+                    "required": []
                 },
             },
             {
@@ -614,7 +614,7 @@ def register_tools(self):
                 parameters=spec.get("parameters"),
             )
 
-    async def get_connection_status(self, params) -> List[TextContent]:
+    async def get_connection_status(self) -> List[TextContent]:
         """Get the current status of the YDB connection.
 
         Returns:
@@ -1039,8 +1039,8 @@ async def call_tool(self, tool_name: str, params: Dict[str, Any]) -> List[TextCo
                 result = await self.query(sql=params["sql"])
             elif tool_name == "ydb_query_with_params" and "sql" in params and "params" in params:
                 result = await self.query_with_params(sql=params["sql"], params=params["params"])
-            elif tool_name == "ydb_status" and "params" in params:
-                result = await self.get_connection_status(params=params["params"])
+            elif tool_name == "ydb_status":
+                result = await self.get_connection_status()
             elif tool_name == "ydb_list_directory" and "path" in params:
                 result = await self.list_directory(path=params["path"])
             elif tool_name == "ydb_describe_path" and "path" in params: