Bundle firecracker runtime and switch ollama demo to live logs

src/pyro_mcp/server.py

@@ -1,26 +1,68 @@
-"""MCP server definition for the v0.0.1 static tool demo."""
+"""MCP server exposing ephemeral VM lifecycle tools."""
 
 from __future__ import annotations
 
-from typing import Final
+from typing import Any
 
 from mcp.server.fastmcp import FastMCP
 
-HELLO_STATIC_PAYLOAD: Final[dict[str, str]] = {
-    "message": "hello from pyro_mcp",
-    "status": "ok",
-    "version": "0.0.1",
-}
+from pyro_mcp.vm_manager import VmManager
 
 
-def create_server() -> FastMCP:
+def create_server(manager: VmManager | None = None) -> FastMCP:
     """Create and return a configured MCP server instance."""
+    vm_manager = manager or VmManager()
     server = FastMCP(name="pyro_mcp")
 
     @server.tool()
-    async def hello_static() -> dict[str, str]:
-        """Return a deterministic static payload."""
-        return HELLO_STATIC_PAYLOAD.copy()
+    async def vm_list_profiles() -> list[dict[str, object]]:
+        """List standard environment profiles and package highlights."""
+        return vm_manager.list_profiles()
+
+    @server.tool()
+    async def vm_create(
+        profile: str,
+        vcpu_count: int,
+        mem_mib: int,
+        ttl_seconds: int = 600,
+    ) -> dict[str, Any]:
+        """Create an ephemeral VM record with profile and resource sizing."""
+        return vm_manager.create_vm(
+            profile=profile,
+            vcpu_count=vcpu_count,
+            mem_mib=mem_mib,
+            ttl_seconds=ttl_seconds,
+        )
+
+    @server.tool()
+    async def vm_start(vm_id: str) -> dict[str, Any]:
+        """Start a created VM and transition it into a command-ready state."""
+        return vm_manager.start_vm(vm_id)
+
+    @server.tool()
+    async def vm_exec(vm_id: str, command: str, timeout_seconds: int = 30) -> dict[str, Any]:
+        """Run one non-interactive command and auto-clean the VM."""
+        return vm_manager.exec_vm(vm_id, command=command, timeout_seconds=timeout_seconds)
+
+    @server.tool()
+    async def vm_stop(vm_id: str) -> dict[str, Any]:
+        """Stop a running VM."""
+        return vm_manager.stop_vm(vm_id)
+
+    @server.tool()
+    async def vm_delete(vm_id: str) -> dict[str, Any]:
+        """Delete a VM and its runtime artifacts."""
+        return vm_manager.delete_vm(vm_id)
+
+    @server.tool()
+    async def vm_status(vm_id: str) -> dict[str, Any]:
+        """Get the current state and metadata for a VM."""
+        return vm_manager.status_vm(vm_id)
+
+    @server.tool()
+    async def vm_reap_expired() -> dict[str, Any]:
+        """Delete VMs whose TTL has expired."""
+        return vm_manager.reap_expired()
 
     return server