Cloud OS

┌──────────────────────────────────────────────────────────┐
│                   FastAPI REST API                        │
│                      (Port 8001)                          │
└──────────────────────────────────────────────────────────┘
                           ↓
┌──────────────────────────────────────────────────────────┐
│                   VM Service (Layer 4)                    │
│                  High-level business logic                │
└──────────────────────────────────────────────────────────┘
                           ↓
┌──────────────────────────────────────────────────────────┐
│                Container gRPC Client                      │
│              Connects to container-grpc                   │
└──────────────────────────────────────────────────────────┘
                           ↓
┌──────────────────────────────────────────────────────────┐
│              Container gRPC Service (Go)                  │
│            Manages Firecracker VMs via Ignite            │
└──────────────────────────────────────────────────────────┘

POST /api/v1/vms

{
  "user_id": "user123",
  "image": "ubuntu:22.04",
  "limits": {
    "cpu_count": 2,
    "memory_mb": 2048,
    "disk_size_gb": 10
  }
}

POST /api/v1/vms/{vm_id}/execute

{
  "command": ["python3", "--version"],
  "env": {"MY_VAR": "value"},
  "working_dir": "/home/user",
  "tty": false
}

POST /api/v1/vms/{vm_id}/files/read

{
  "path": "/home/user/myfile.txt"
}

POST /api/v1/vms/{vm_id}/files/write

{
  "path": "/home/user/myfile.txt",
  "content": "Hello, World!",
  "mode": "0644"
}

GET /api/v1/vms?user_id=user123

GET /api/v1/vms/{vm_id}

DELETE /api/v1/vms/{vm_id}

import httpx
 
class CloudOSClient:
    def __init__(self, base_url: str = "http://localhost:8001"):
        self.base_url = base_url
 
    async def create_vm(self, user_id: str, image: str = "ubuntu:22.04"):
        async with httpx.AsyncClient() as client:
            response = await client.post(
                f"{self.base_url}/api/v1/vms",
                json={
                    "user_id": user_id,
                    "image": image,
                    "limits": {
                        "cpu_count": 2,
                        "memory_mb": 2048,
                        "disk_size_gb": 10
                    }
                }
            )
            return response.json()
 
    async def execute(self, vm_id: str, command: list):
        async with httpx.AsyncClient() as client:
            response = await client.post(
                f"{self.base_url}/api/v1/vms/{vm_id}/execute",
                json={"command": command}
            )
            return response.json()
 
    async def read_file(self, vm_id: str, path: str):
        async with httpx.AsyncClient() as client:
            response = await client.post(
                f"{self.base_url}/api/v1/vms/{vm_id}/files/read",
                json={"path": path}
            )
            return response.json()
 
    async def write_file(self, vm_id: str, path: str, content: str):
        async with httpx.AsyncClient() as client:
            response = await client.post(
                f"{self.base_url}/api/v1/vms/{vm_id}/files/write",
                json={"path": path, "content": content}
            )
            return response.json()

from mcp.server import Server
 
@server.tool()
async def create_vm(user_id: str, image: str = "ubuntu:22.04"):
    """Create a VM for the user"""
    client = CloudOSClient()
    return await client.create_vm(user_id, image)
 
@server.tool()
async def run_in_vm(vm_id: str, command: str):
    """Execute command in VM"""
    client = CloudOSClient()
    return await client.execute(vm_id, command.split())