diff --git a/unraid_mcp/tools/virtualization.py b/unraid_mcp/tools/virtualization.py index 4026f89..f54baf7 100644 --- a/unraid_mcp/tools/virtualization.py +++ b/unraid_mcp/tools/virtualization.py @@ -6,21 +6,11 @@ including start, stop, pause, resume, force stop, reboot, and reset. from typing import Any, Literal, get_args -from fastmcp import Context as _Context from fastmcp import FastMCP from ..config.logging import logger from ..core.client import make_graphql_request -from ..core.exceptions import CredentialsNotConfiguredError as _CredErr from ..core.exceptions import ToolError, tool_error_handler -from ..core.setup import elicit_and_configure as _elicit - - -# Re-export at module scope so tests can patch "unraid_mcp.tools.virtualization.elicit_and_configure" -# and "unraid_mcp.tools.virtualization.CredentialsNotConfiguredError" -elicit_and_configure = _elicit -CredentialsNotConfiguredError = _CredErr -Context = _Context QUERIES: dict[str, str] = { @@ -100,7 +90,6 @@ def register_vm_tool(mcp: FastMCP) -> None: action: VM_ACTIONS, vm_id: str | None = None, confirm: bool = False, - ctx: Context | None = None, ) -> dict[str, Any]: """Manage Unraid virtual machines. @@ -128,15 +117,7 @@ def register_vm_tool(mcp: FastMCP) -> None: logger.info(f"Executing unraid_vm action={action}") if action == "list": - try: - data = await make_graphql_request(QUERIES["list"]) - except CredentialsNotConfiguredError: - configured = await elicit_and_configure(ctx) - if not configured: - raise ToolError( - "Credentials required. Run `unraid_health action=setup` to configure." - ) - data = await make_graphql_request(QUERIES["list"]) + data = await make_graphql_request(QUERIES["list"]) if data.get("vms"): vms = data["vms"].get("domains") or data["vms"].get("domain") or [] if isinstance(vms, dict):