Skip to main content

Secure runtime for AI agents, and tools -- free and open-source from Celesto AI 🧡

Project description

SmolVM

Secure runtime for AI agents and tools

License Python 3.10+

DocsExamplesGitHub


SmolVM is a lightning-fast, secure microVM runtime designed for high-density isolation. It provides AI agents and tools with a safe, hardware-virtualized environment to execute untrusted code without risking the host system.

✨ Features

  • 🔒 Secure Isolation: Hardware-level virtualization (utilizing Firecracker) for strong sandbox boundaries.
  • ⚡ Blazing Fast: MicroVMs boot in sub-second time with minimal overhead.
  • 🐍 Python Native: Clean, high-level SDK for managing VM lifecycles and command execution.
  • 🌐 Automatic Networking: Built-in NAT, port forwarding, and SSH tunneling.
  • 🛠️ Custom Images: Build specialized Debian-based rootfs images with your own tools.
  • 🧹 Auto-Cleanup: Integrated resource management to keep your host system clean.

🤔 Why SmolVM?

AI agents often need to execute arbitrary code (Python, JS, shell scripts) generated by LLMs. Running this code directly on your host or in standard containers can be risky.

  • MicroVM-based Security: Unlike containers that share the host kernel, SmolVM uses KVM-backed microVMs. This provides a significantly smaller attack surface and stronger hardware-level isolation.
  • Agent-First Design: SmolVM abstracts away the complexity of microVM networking, storage, and TAP devices into a simple, pythonic API.

🚀 Quickstart

1. Prerequisites

  • Linux + Firecracker backend: KVM support (Ubuntu/Debian/Fedora).
  • macOS + QEMU backend: Homebrew and QEMU (qemu-system-*).

2. Installation

# Install the Python package
pip install smolvm

Linux (Firecracker):

sudo ./scripts/system-setup.sh --configure-runtime

macOS (QEMU):

./scripts/system-setup-macos.sh
# Optional explicit backend override:
# export SMOLVM_BACKEND=qemu

3. Basic Usage

Initialize a VM with no arguments for an auto-configured, SSH-ready environment:

from smolvm import SmolVM

# Start sandboxed runtime
vm = SmolVM()
vm.start()

# Run ANY command like a real system
result = vm.run("echo 'Hello from the sandbox!'")
print(result.output)

# Stop the runtime
vm.stop()

Customize auto-config memory and disk size:

from smolvm import SmolVM

# Use with context manager (auto start and deletes after use)
with SmolVM(mem_size_mib=2048, disk_size_mib=4096) as vm:
    print(vm.run("free -m").output)

4. Reconnect to an existing VM

You can also reconnect to a running VM by its ID:

from smolvm import SmolVM

# Reconnect to an existing VM
vm = SmolVM.from_id("vm-abcdef12")
print(f"Status: {vm.status}")

4.1 Disk isolation defaults

SmolVM now defaults to isolated per-VM disks (disk_mode="isolated"), so each VM gets its own writable rootfs clone (sandbox-by-default).

If you intentionally want shared/persistent image behavior across VMs, set:

from smolvm import VMConfig

config = VMConfig(..., disk_mode="shared")

5. Port Forwarding

Expose a guest application to your local machine securely. expose_local prefers host-local nftables forwarding and automatically falls back to an SSH tunnel when needed.

from smolvm import SmolVM

with SmolVM() as vm:
    # Example: App in VM listening on port 8080, expose to host port 18080
    host_port = vm.expose_local(guest_port=8080, host_port=18080)
    print(f"App available at http://localhost:{host_port}")

6. Environment Variables

Inject environment variables into a running VM. Variables are persisted in /etc/profile.d/smolvm_env.sh and apply to new SSH/login shell sessions.

from smolvm import SmolVM

with SmolVM() as vm:
    vm.set_env_vars({"API_KEY": "sk-...", "DEBUG": "1"})
    print(vm.list_env_vars())
    print(vm.run("echo $API_KEY").output)

CLI:

smolvm env set <vm_id> API_KEY=sk-... DEBUG=1
smolvm env list <vm_id> --show-values
smolvm env unset <vm_id> DEBUG

Diagnostics:

# Auto-detect backend (Darwin -> qemu, Linux -> firecracker)
smolvm doctor

# Force backend checks
smolvm doctor --backend firecracker
smolvm doctor --backend qemu

# CI-friendly mode
smolvm doctor --json --strict

⚡ Performance

SmolVM is optimized for low-latency agent workflows. Latest lifecycle timings (p50) on a standard Linux host:

Phase Time
Create + Start ~572ms
SSH ready ~2.1s
Command execution ~43ms
Stop + Delete ~751ms
Full lifecycle (boot → run → teardown) ~3.5s

Run the benchmark yourself:

python scripts/benchmarks/bench_subprocess.py --vms 10 -v

Measured on AMD Ryzen 7 7800X3D (8C/16T), Ubuntu Linux, KVM/Firecracker backend.

📄 License

Apache 2.0 License - see LICENSE for details.


Built with 🧡 by Celesto AI

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

smolvm-0.0.4.tar.gz (112.4 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

smolvm-0.0.4-py3-none-any.whl (86.4 kB view details)

Uploaded Python 3

File details

Details for the file smolvm-0.0.4.tar.gz.

File metadata

  • Download URL: smolvm-0.0.4.tar.gz
  • Upload date:
  • Size: 112.4 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.7

File hashes

Hashes for smolvm-0.0.4.tar.gz
Algorithm Hash digest
SHA256 6edc5bb0f9e54f1c24b263bf0db0a72727f42721c97688cf137ddc19c3fd0265
MD5 58542ff94f806f99e2e0197c7beb2a5b
BLAKE2b-256 80733a83d893ce4f85ce42f632cb683e8370c2dbf0e2b3a5e2eafaab7373e15f

See more details on using hashes here.

Provenance

The following attestation bundles were made for smolvm-0.0.4.tar.gz:

Publisher: publish.yml on CelestoAI/SmolVM

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file smolvm-0.0.4-py3-none-any.whl.

File metadata

  • Download URL: smolvm-0.0.4-py3-none-any.whl
  • Upload date:
  • Size: 86.4 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.7

File hashes

Hashes for smolvm-0.0.4-py3-none-any.whl
Algorithm Hash digest
SHA256 247d936334fa36d5a3979b5ef8a889b7fd869be5da10f445cbe731d48f211746
MD5 608338d7849a397ef15eff3843e0dae4
BLAKE2b-256 d83eb2cedc2c98e857ea5817af3625d901fd16d858c94f8a469f6f8b1cbb39f6

See more details on using hashes here.

Provenance

The following attestation bundles were made for smolvm-0.0.4-py3-none-any.whl:

Publisher: publish.yml on CelestoAI/SmolVM

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page