feat(sandbox): add CoreWeave Sandbox (cwsandbox) provider - #76
Merged
dbczumar merged 4 commits intoJun 15, 2026
Conversation
brandonrjacobs
force-pushed
the
feat/cwsandbox-provider
branch
from
June 14, 2026 23:56
3000d5f to
8d1d379
Compare
Add CoreWeave Sandbox (aviato) as a sandbox provider alongside Modal and
Daytona. CWSandboxLauncher wraps the official `cwsandbox` Python SDK as an
optional, lazily-imported extra (`omnigent[cwsandbox]`), supporting both
server-managed hosts (`sandbox.provider: cwsandbox`) and the CLI bootstrap.
- omnigent/onboarding/sandboxes/cwsandbox.py: the launcher
- register in the provider table + server managed-host YAML config
- pyproject: `cwsandbox` extra + mypy override; uv.lock pins cwsandbox 0.26.0
(per-package cooldown exemption in uv.toml, since the SDK is first-party)
- tests + deploy/cwsandbox/{README,smoke_test,e2e_managed}
The managed launch-token TTL is derived from OMNIGENT_CWSANDBOX_MAX_LIFETIME_S
so it always outlives the (operator-overridable) sandbox lifetime. The e2e
driver runs a real agent LLM turn inside a managed sandbox; it can target an
existing server (--server) or spin one up in a CW sandbox with a public
service, and only tears down the child sandboxes it created. Validated
end-to-end against api.cwsandbox.com.
brandonrjacobs
force-pushed
the
feat/cwsandbox-provider
branch
from
June 14, 2026 23:58
8d1d379 to
f0830da
Compare
dbczumar
self-requested a review
June 15, 2026 06:07
Signed-off-by: dbczumar <corey.zumar@databricks.com>
dbczumar
previously approved these changes
Jun 15, 2026
dbczumar
left a comment
Contributor
There was a problem hiding this comment.
Approving. Code-reviewed the cwsandbox provider and ran the unit tests, lint (ruff), and type checks (mypy) locally — all green. Pushed a small follow-up commit relocating the e2e/smoke scripts under tests/e2e/integrations/deploy/cwsandbox/ and fixing a ruff-format miss.
Resolve provider-registry conflicts by keeping both the new cwsandbox launcher (this PR) and the islo launcher (landed on main): - omnigent/onboarding/sandboxes/__init__.py: register both launchers - omnigent/server/managed_hosts.py: both in SUPPORTED/MANAGED sets, both elif branches + parse/factory helpers Signed-off-by: dbczumar <corey.zumar@databricks.com>
…e from islo cwsandbox README now covers host image, CLI create/connect, authed-server injection, managed-host/server-auth caveat, LLM + git credentials, security considerations, troubleshooting, and an env-var reference table — matching the modal/daytona/islo guides. Also removes the SSE provisioning-refresh troubleshooting bullet from the islo README. Signed-off-by: dbczumar <corey.zumar@databricks.com>
dbczumar
approved these changes
Jun 15, 2026
dbczumar
left a comment
Contributor
There was a problem hiding this comment.
Re-approving the current head: merged main + provider-registry conflict resolution keeping both cwsandbox and islo, and README parity updates. Unit tests (cwsandbox+islo), ruff, and mypy pass locally on the merged tree.
dbczumar
approved these changes
Jun 15, 2026
dbczumar
left a comment
Contributor
There was a problem hiding this comment.
LGTM! Thanks @brandonrjacobs !
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Summary
Adds CoreWeave Sandbox (https://docs.coreweave.com/products/sandboxes) as a sandbox provider,
alongside Modal and Daytona.
CWSandboxLauncherwraps the officialcwsandboxPython SDK as anoptional, lazily-imported extra (
omnigent[cwsandbox]) — the same posture asthe existing modal/daytona launchers.
Supports both server-managed hosts (
sandbox.provider: cwsandbox) and theCLI bootstrap (
omnigent sandbox create --provider cwsandbox). The onlyunsupported primitive is
forward_local_port(CW Sandbox has no local→sandboxpath), which is irrelevant here — managed hosts authenticate with a
server-minted launch token, not the Databricks App OAuth flow.
Changes
omnigent/onboarding/sandboxes/cwsandbox.py— the launcher (provision / run /put / terminate + streaming exec / foreground).
cwsandboxin the provider table and the server managed-host YAMLconfig (
SUPPORTED_SANDBOX_PROVIDERS,PROVIDERS_WITH_MANAGED_LAUNCH,factory +
_parse_cwsandbox_*validators).pyproject.toml:cwsandboxextra + mypy override.uv.lockregenerated(pins
cwsandbox 0.26.0).sys.modules, no network).deploy/cwsandbox/: README, a zero-dependency API smoke test, and a managede2e driver.
Validation
Exercised end-to-end against production
api.cwsandbox.com:terminate.
ingress), created a managed session, and the server provisioned a child CW
sandbox running
omnigent hostthat dialed back and registered.openai-agentsharness against an OpenAI-compatible CoreWeave inference endpoint) inside the
managed sandbox and returned a correct answer.
Unit tests +
ruff+mypy+uv lock --checkall green.Notes for reviewers
_ensure_sdk()in__init__is a deliberate workaround: the cwsandboxSDK installs signal handlers at import time, which raises when imported in a
worker thread (the managed flow calls
prepare()/provision()viaasyncio.to_thread). Importing at construction (the main thread) sidestepsit. Removable once Import time signal handler install crashes from non-main threads such as marimo coreweave/cwsandbox-client#136 (PR
fix: skip signal handlers outside main thread (#136) coreweave/cwsandbox-client#138) ships.
uv.tomlcooldown exemption. The repo's P7Dexclude-newercooldowncan't yet resolve
cwsandbox 0.26.0(published within the window), so ascoped
exclude-newer-packageexemption for this one first-party SDK lets thelock pin the tested version; the global cooldown is unchanged for every other
dependency. It can be dropped once 0.26.0 ages past P7D.
OMNIGENT_CWSANDBOX_MAX_LIFETIME_Sso it always outlives the(operator-overridable) sandbox lifetime.
Testing