openenv 0.2.1__tar.gz → 0.2.3__tar.gz

openenv-0.2.3/MANIFEST.in

@@ -0,0 +1,2 @@
+global-exclude *.py[cod]
+global-exclude __pycache__/*

{openenv-0.2.1 → openenv-0.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openenv
-Version: 0.2.1
+Version: 0.2.3
 Summary: A unified framework for reinforcement learning environments
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
@@ -17,7 +17,9 @@ Requires-Dist: openai>=2.7.2
 Requires-Dist: tomli>=2.3.0
 Requires-Dist: tomli-w>=1.2.0
 Requires-Dist: websockets>=15.0.1
-Requires-Dist: fastmcp>=2.0.0
+Requires-Dist: fastmcp>=3.0.0
+Requires-Dist: gradio>=4.0.0
+Requires-Dist: httpx>=0.28.1
 Provides-Extra: core
 Requires-Dist: fastapi>=0.104.0; extra == "core"
 Requires-Dist: pydantic>=2.0.0; extra == "core"
@@ -32,9 +34,28 @@ Requires-Dist: huggingface_hub>=0.20.0; extra == "cli"
 Requires-Dist: openai>=2.7.2; extra == "cli"
 Requires-Dist: tomli>=2.3.0; extra == "cli"
 Requires-Dist: tomli-w>=1.2.0; extra == "cli"
+Provides-Extra: docs
+Requires-Dist: sphinx==7.2.6; extra == "docs"
+Requires-Dist: pytorch-sphinx-theme2; extra == "docs"
+Requires-Dist: sphinxcontrib.katex==0.9.10; extra == "docs"
+Requires-Dist: docutils<0.21,>=0.18.1; extra == "docs"
+Requires-Dist: sphinx-design==0.6.1; extra == "docs"
+Requires-Dist: sphinxcontrib-mermaid==1.0.0; extra == "docs"
+Requires-Dist: myst-parser; extra == "docs"
+Requires-Dist: sphinxext-opengraph; extra == "docs"
+Requires-Dist: sphinx-sitemap==2.7.1; extra == "docs"
+Requires-Dist: sphinx-gallery>=0.14.0; extra == "docs"
+Requires-Dist: matplotlib; extra == "docs"
+Requires-Dist: nest-asyncio; extra == "docs"
+Requires-Dist: smolagents; extra == "docs"
 Provides-Extra: all
 Requires-Dist: openenv[core]; extra == "all"
 Requires-Dist: openenv[cli]; extra == "all"
+Provides-Extra: daytona
+Requires-Dist: daytona>=0.136.0; extra == "daytona"
+Requires-Dist: pyyaml>=6.0; extra == "daytona"
+Provides-Extra: inspect
+Requires-Dist: inspect-ai>=0.3.0; extra == "inspect"
 Dynamic: license-file
 
 # <img width="35" height="35" alt="image" src="https://github.com/user-attachments/assets/2700a971-e5d6-4036-b03f-2f89c9791609" /> OpenEnv: Agentic Execution Environments
@@ -50,7 +71,7 @@ An e2e framework for creating, deploying and using isolated execution environmen
 
 **🚀 Featured Example:** Train LLMs to play BlackJack using [torchforge](https://github.com/meta-pytorch/torchforge) (PyTorch's agentic RL framework): [`examples/grpo_blackjack/`](examples/grpo_blackjack/)
 
-**🔥 GPU Mode Tutorial:** End to end tutorial from [GPU Mode](gpu-mode-tutorial/README.md) blog post.
+**🔥 Zero to Hero Tutorial:** End to end tutorial from our [GPU Mode](tutorial/README.md) lecture and other hackathons.
 
 ## Quick Start
 
@@ -69,22 +90,34 @@ pip install git+https://huggingface.co/spaces/openenv/echo_env
 Then use the environment:
 
 ```python
+import asyncio
 from echo_env import EchoAction, EchoEnv
 
-# Connect to a running Space
-client = EchoEnv(base_url="https://openenv-echo-env.hf.space")
+async def main():
+    # Connect to a running Space (async context manager)
+    async with EchoEnv(base_url="https://openenv-echo-env.hf.space") as client:
+        # Reset the environment
+        result = await client.reset()
+        print(result.observation.echoed_message)  # "Echo environment ready!"
 
-# Reset the environment
-result = client.reset()
-print(result.observation.echoed_message)  # "Echo environment ready!"
+        # Send messages
+        result = await client.step(EchoAction(message="Hello, World!"))
+        print(result.observation.echoed_message)  # "Hello, World!"
+        print(result.reward)  # 1.3 (based on message length)
 
-# Send messages
-result = client.step(EchoAction(message="Hello, World!"))
-print(result.observation.echoed_message)  # "Hello, World!"
-print(result.reward)  # 1.3 (based on message length)
+asyncio.run(main())
+```
+
+**Synchronous usage** is also supported via the `.sync()` wrapper:
 
-# Cleanup
-client.close()
+```python
+from echo_env import EchoAction, EchoEnv
+
+# Use .sync() for synchronous context manager
+with EchoEnv(base_url="https://openenv-echo-env.hf.space").sync() as client:
+    result = client.reset()
+    result = client.step(EchoAction(message="Hello, World!"))
+    print(result.observation.echoed_message)
 ```
 
 For a detailed quick start, check out the [docs page](https://meta-pytorch.org/OpenEnv/quickstart/).
@@ -117,6 +150,10 @@ The OpenEnv CLI (`openenv`) provides commands to initialize new environments and
 Below is a list of active and historical RFCs for OpenEnv. RFCs are proposals for major changes or features. Please review and contribute!
 
 - [RFC 001: Baseline API and Interface Specifications](https://github.com/meta-pytorch/OpenEnv/pull/26)
+- [RFC 002: Discoverability of environment tools by agents](https://github.com/meta-pytorch/OpenEnv/pull/32)
+- [RFC 003: Add MCP (Model Context Protocol) support](https://github.com/meta-pytorch/OpenEnv/pull/224)
+- [RFC 004: Add delayed rewards support for trajectory-based scoring](https://github.com/meta-pytorch/OpenEnv/pull/337)
+- [RFC 005: Agentic Harness Integration](https://github.com/meta-pytorch/OpenEnv/pull/387)
 
 ## Architecture
 
@@ -179,6 +216,8 @@ Base class for implementing environment logic:
 
 #### 3. EnvClient (Client-Side)
 Base class for environment communication:
+- **Async by default**: Use `async with` and `await` for all operations
+- **Sync wrapper**: Call `.sync()` to get a `SyncEnvClient` for synchronous usage
 - Handles WebSocket connections to environment server
 - Contains a utility to spin up a docker container locally for the corresponding environment
 - Type-safe action/observation parsing
@@ -261,9 +300,21 @@ See [`envs/README.md`](envs/README.md) for a complete guide on building environm
 To use an environment:
 1. Install the client: `pip install git+https://huggingface.co/spaces/openenv/echo-env`
 2. Import: `from echo_env import EchoAction, EchoEnv`
-3. Create client: `client = EchoEnv(base_url="https://openenv-echo-env.hf.space")`
-4. Interact: `client.reset()`, `client.step(action)`, `client.state()`
-5. Cleanup: `client.close()`
+3. Use async (recommended) or sync API:
+
+**Async (recommended):**
+```python
+async with EchoEnv(base_url="...") as client:
+    result = await client.reset()
+    result = await client.step(action)
+```
+
+**Sync (via `.sync()` wrapper):**
+```python
+with EchoEnv(base_url="...").sync() as client:
+    result = client.reset()
+    result = client.step(action)
+```
 
 See example scripts in `examples/` directory.
 
@@ -368,22 +419,15 @@ See the [Oumi example](https://github.com/oumi-ai/oumi/blob/main/notebooks/Oumi%
 
 ## Example Environments
 
-### Echo Environment
-A simple environment that echoes back messages with metadata. Perfect for:
-- Testing the HTTP server infrastructure
-- Learning the framework basics
-- Verifying container deployment
-
-See: [`envs/echo_env/README.md`](envs/echo_env/README.md)
-
-### Coding Environment
-Executes arbitrary Python code in a sandboxed environment. Features:
-- Safe code execution using smolagents
-- Capture stdout, stderr, and exit codes
-- Persistent execution context within episodes
-- Error handling with detailed messages
+| Environment | Description |
+|---|---|
+| [Echo Environment](envs/echo_env/README.md) | Echoes back messages with metadata. Ideal for testing HTTP server infrastructure, learning framework basics, and verifying container deployment. |
+| [Coding Environment](envs/coding_env/README.md) | Sandboxed Python code execution via smolagents. Captures stdout/stderr/exit codes, supports persistent episode context, and provides detailed error handling. |
+| [Chess Environment](envs/chess_env/README.md) | Chess RL environment with configurable opponents and full rules support. |
+| [Atari Environment](envs/atari_env/README.md) | Classic Arcade Learning Environment tasks for RL benchmarking. |
+| [FinRL Environment](envs/finrl_env/README.md) | Financial market simulations for algorithmic trading experiments. |
 
-See: [`envs/coding_env/README.md`](envs/coding_env/README.md)
+> Browse the full catalog of community environments at [meta-pytorch.org/OpenEnv/environments](https://meta-pytorch.org/OpenEnv/environments/).
 
 ## Community Support & Acknowledgments
 This is an open and community-centric project. If you would like to add your name here, please put up a pull request and tag @jspisak for review. Ty!!

openenv-0.2.1/src/openenv.egg-info/PKG-INFO → openenv-0.2.3/README.md

@@ -1,42 +1,3 @@
-Metadata-Version: 2.4
-Name: openenv
-Version: 0.2.1
-Summary: A unified framework for reinforcement learning environments
-Requires-Python: >=3.10
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: fastapi>=0.104.0
-Requires-Dist: pydantic>=2.0.0
-Requires-Dist: uvicorn>=0.24.0
-Requires-Dist: requests>=2.25.0
-Requires-Dist: typer>=0.9.0
-Requires-Dist: rich>=13.0.0
-Requires-Dist: pyyaml>=6.0
-Requires-Dist: huggingface_hub>=0.20.0
-Requires-Dist: openai>=2.7.2
-Requires-Dist: tomli>=2.3.0
-Requires-Dist: tomli-w>=1.2.0
-Requires-Dist: websockets>=15.0.1
-Requires-Dist: fastmcp>=2.0.0
-Provides-Extra: core
-Requires-Dist: fastapi>=0.104.0; extra == "core"
-Requires-Dist: pydantic>=2.0.0; extra == "core"
-Requires-Dist: uvicorn>=0.24.0; extra == "core"
-Requires-Dist: requests>=2.25.0; extra == "core"
-Requires-Dist: websockets>=15.0.1; extra == "core"
-Provides-Extra: cli
-Requires-Dist: typer>=0.9.0; extra == "cli"
-Requires-Dist: rich>=13.0.0; extra == "cli"
-Requires-Dist: pyyaml>=6.0; extra == "cli"
-Requires-Dist: huggingface_hub>=0.20.0; extra == "cli"
-Requires-Dist: openai>=2.7.2; extra == "cli"
-Requires-Dist: tomli>=2.3.0; extra == "cli"
-Requires-Dist: tomli-w>=1.2.0; extra == "cli"
-Provides-Extra: all
-Requires-Dist: openenv[core]; extra == "all"
-Requires-Dist: openenv[cli]; extra == "all"
-Dynamic: license-file
-
 # <img width="35" height="35" alt="image" src="https://github.com/user-attachments/assets/2700a971-e5d6-4036-b03f-2f89c9791609" /> OpenEnv: Agentic Execution Environments
 
 An e2e framework for creating, deploying and using isolated execution environments for agentic RL training, built using Gymnasium style simple APIs.
@@ -50,7 +11,7 @@ An e2e framework for creating, deploying and using isolated execution environmen
 
 **🚀 Featured Example:** Train LLMs to play BlackJack using [torchforge](https://github.com/meta-pytorch/torchforge) (PyTorch's agentic RL framework): [`examples/grpo_blackjack/`](examples/grpo_blackjack/)
 
-**🔥 GPU Mode Tutorial:** End to end tutorial from [GPU Mode](gpu-mode-tutorial/README.md) blog post.
+**🔥 Zero to Hero Tutorial:** End to end tutorial from our [GPU Mode](tutorial/README.md) lecture and other hackathons.
 
 ## Quick Start
 
@@ -69,22 +30,34 @@ pip install git+https://huggingface.co/spaces/openenv/echo_env
 Then use the environment:
 
 ```python
+import asyncio
 from echo_env import EchoAction, EchoEnv
 
-# Connect to a running Space
-client = EchoEnv(base_url="https://openenv-echo-env.hf.space")
+async def main():
+    # Connect to a running Space (async context manager)
+    async with EchoEnv(base_url="https://openenv-echo-env.hf.space") as client:
+        # Reset the environment
+        result = await client.reset()
+        print(result.observation.echoed_message)  # "Echo environment ready!"
+
+        # Send messages
+        result = await client.step(EchoAction(message="Hello, World!"))
+        print(result.observation.echoed_message)  # "Hello, World!"
+        print(result.reward)  # 1.3 (based on message length)
 
-# Reset the environment
-result = client.reset()
-print(result.observation.echoed_message)  # "Echo environment ready!"
+asyncio.run(main())
+```
 
-# Send messages
-result = client.step(EchoAction(message="Hello, World!"))
-print(result.observation.echoed_message)  # "Hello, World!"
-print(result.reward)  # 1.3 (based on message length)
+**Synchronous usage** is also supported via the `.sync()` wrapper:
 
-# Cleanup
-client.close()
+```python
+from echo_env import EchoAction, EchoEnv
+
+# Use .sync() for synchronous context manager
+with EchoEnv(base_url="https://openenv-echo-env.hf.space").sync() as client:
+    result = client.reset()
+    result = client.step(EchoAction(message="Hello, World!"))
+    print(result.observation.echoed_message)
 ```
 
 For a detailed quick start, check out the [docs page](https://meta-pytorch.org/OpenEnv/quickstart/).
@@ -117,6 +90,10 @@ The OpenEnv CLI (`openenv`) provides commands to initialize new environments and
 Below is a list of active and historical RFCs for OpenEnv. RFCs are proposals for major changes or features. Please review and contribute!
 
 - [RFC 001: Baseline API and Interface Specifications](https://github.com/meta-pytorch/OpenEnv/pull/26)
+- [RFC 002: Discoverability of environment tools by agents](https://github.com/meta-pytorch/OpenEnv/pull/32)
+- [RFC 003: Add MCP (Model Context Protocol) support](https://github.com/meta-pytorch/OpenEnv/pull/224)
+- [RFC 004: Add delayed rewards support for trajectory-based scoring](https://github.com/meta-pytorch/OpenEnv/pull/337)
+- [RFC 005: Agentic Harness Integration](https://github.com/meta-pytorch/OpenEnv/pull/387)
 
 ## Architecture
 
@@ -179,6 +156,8 @@ Base class for implementing environment logic:
 
 #### 3. EnvClient (Client-Side)
 Base class for environment communication:
+- **Async by default**: Use `async with` and `await` for all operations
+- **Sync wrapper**: Call `.sync()` to get a `SyncEnvClient` for synchronous usage
 - Handles WebSocket connections to environment server
 - Contains a utility to spin up a docker container locally for the corresponding environment
 - Type-safe action/observation parsing
@@ -261,9 +240,21 @@ See [`envs/README.md`](envs/README.md) for a complete guide on building environm
 To use an environment:
 1. Install the client: `pip install git+https://huggingface.co/spaces/openenv/echo-env`
 2. Import: `from echo_env import EchoAction, EchoEnv`
-3. Create client: `client = EchoEnv(base_url="https://openenv-echo-env.hf.space")`
-4. Interact: `client.reset()`, `client.step(action)`, `client.state()`
-5. Cleanup: `client.close()`
+3. Use async (recommended) or sync API:
+
+**Async (recommended):**
+```python
+async with EchoEnv(base_url="...") as client:
+    result = await client.reset()
+    result = await client.step(action)
+```
+
+**Sync (via `.sync()` wrapper):**
+```python
+with EchoEnv(base_url="...").sync() as client:
+    result = client.reset()
+    result = client.step(action)
+```
 
 See example scripts in `examples/` directory.
 
@@ -368,22 +359,15 @@ See the [Oumi example](https://github.com/oumi-ai/oumi/blob/main/notebooks/Oumi%
 
 ## Example Environments
 
-### Echo Environment
-A simple environment that echoes back messages with metadata. Perfect for:
-- Testing the HTTP server infrastructure
-- Learning the framework basics
-- Verifying container deployment
-
-See: [`envs/echo_env/README.md`](envs/echo_env/README.md)
-
-### Coding Environment
-Executes arbitrary Python code in a sandboxed environment. Features:
-- Safe code execution using smolagents
-- Capture stdout, stderr, and exit codes
-- Persistent execution context within episodes
-- Error handling with detailed messages
+| Environment | Description |
+|---|---|
+| [Echo Environment](envs/echo_env/README.md) | Echoes back messages with metadata. Ideal for testing HTTP server infrastructure, learning framework basics, and verifying container deployment. |
+| [Coding Environment](envs/coding_env/README.md) | Sandboxed Python code execution via smolagents. Captures stdout/stderr/exit codes, supports persistent episode context, and provides detailed error handling. |
+| [Chess Environment](envs/chess_env/README.md) | Chess RL environment with configurable opponents and full rules support. |
+| [Atari Environment](envs/atari_env/README.md) | Classic Arcade Learning Environment tasks for RL benchmarking. |
+| [FinRL Environment](envs/finrl_env/README.md) | Financial market simulations for algorithmic trading experiments. |
 
-See: [`envs/coding_env/README.md`](envs/coding_env/README.md)
+> Browse the full catalog of community environments at [meta-pytorch.org/OpenEnv/environments](https://meta-pytorch.org/OpenEnv/environments/).
 
 ## Community Support & Acknowledgments
 This is an open and community-centric project. If you would like to add your name here, please put up a pull request and tag @jspisak for review. Ty!!

{openenv-0.2.1 → openenv-0.2.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "openenv"
-version = "0.2.1"
+version = "0.2.3"
 description = "A unified framework for reinforcement learning environments"
 readme = "README.md"
 requires-python = ">=3.10"
@@ -26,7 +26,10 @@ dependencies = [
     "tomli-w>=1.2.0",
     "websockets>=15.0.1",
     # MCP support
-    "fastmcp>=2.0.0",
+    "fastmcp>=3.0.0",
+    # Web UI dependencies
+    "gradio>=4.0.0",
+    "httpx>=0.28.1",
 ]
 
 [project.optional-dependencies]
@@ -46,10 +49,32 @@ cli = [
     "tomli>=2.3.0",
     "tomli-w>=1.2.0",
 ]
+docs = [
+    "sphinx==7.2.6",
+    "pytorch-sphinx-theme2",
+    "sphinxcontrib.katex==0.9.10",
+    "docutils>=0.18.1,<0.21",
+    "sphinx-design==0.6.1",
+    "sphinxcontrib-mermaid==1.0.0",
+    "myst-parser",
+    "sphinxext-opengraph",
+    "sphinx-sitemap==2.7.1",
+    "sphinx-gallery>=0.14.0",
+    "matplotlib",
+    "nest-asyncio",
+    "smolagents",
+]
 all = [
     "openenv[core]",
     "openenv[cli]",
 ]
+daytona = [
+    "daytona>=0.136.0",
+    "pyyaml>=6.0",
+]
+inspect = [
+    "inspect-ai>=0.3.0",
+]
 
 [project.scripts]
 openenv = "openenv.cli.__main__:main"
@@ -61,8 +86,12 @@ include-package-data = true
 [tool.setuptools.package-data]
 "openenv.cli" = ["templates/**/*"]
 
+[tool.setuptools.exclude-package-data]
+"*" = ["*.pyc", "*.pyo", "__pycache__/*"]
+
 [tool.setuptools.packages.find]
 where = ["src"]
+exclude = ["*.__pycache__", "*.__pycache__.*"]
 
 [tool.coverage.run]
 omit = [
@@ -82,8 +111,42 @@ exclude_lines = [
 ]
 
 [tool.pytest.ini_options]
+asyncio_mode = "auto"
+asyncio_default_fixture_loop_scope = "function"
 markers = [
     "docker: Tests that require Docker to be running",
     "network: Tests that require network access (HuggingFace, etc.)",
     "integration: Integration tests with external resources",
 ]
+
+[dependency-groups]
+dev = [
+    "ruff>=0.14.0",
+    "usort>=1.1.0",
+    "pytest>=7.0",
+    "pytest-asyncio>=0.21",
+]
+
+[tool.usort]
+# Disable first_party auto-detection so all non-stdlib imports land in
+# the same "third_party" bucket (the default_category). This matches
+# pyfmt's usort behavior inside arc f, which groups openenv.* and env
+# package imports together without blank-line separators.
+first_party_detection = false
+
+[tool.ruff]
+line-length = 88
+
+[tool.ruff.lint]
+select = ["E", "F", "W"]
+ignore = [
+    "E402",  # Module level import not at top of file (needed for pytest.importorskip patterns)
+    "E501",  # Line too long (not enforced previously, would require large refactor)
+]
+
+[tool.ruff.lint.per-file-ignores]
+# Context manager variables that are intentionally unused
+"tests/envs/test_websockets.py" = ["F841"]
+"tests/test_cli/test_push.py" = ["F841"]
+# Compatibility shim module
+"src/openenv_core/__init__.py" = ["F401"]

openenv-0.2.3/src/openenv/__init__.py

@@ -0,0 +1,62 @@
+"""Unified OpenEnv package bundling the CLI and core runtime."""
+
+from __future__ import annotations
+
+from importlib import import_module, metadata
+
+__all__ = [
+    "core",
+    "cli",
+    "AutoEnv",
+    "AutoAction",
+    "GenericEnvClient",
+    "GenericAction",
+    "SyncEnvClient",
+]
+
+
+def _load_package_version() -> str:
+    """Resolve the installed distribution version for the OpenEnv package."""
+    for distribution_name in ("openenv", "openenv"):
+        try:
+            return metadata.version(distribution_name)
+        except metadata.PackageNotFoundError:
+            continue
+    return "0.0.0"
+
+
+__version__ = _load_package_version()
+
+
+_LAZY_MODULES = {
+    "core": ".core",
+    "cli": ".cli",
+}
+
+_LAZY_ATTRS = {
+    "AutoEnv": (".auto", "AutoEnv"),
+    "AutoAction": (".auto", "AutoAction"),
+    "GenericEnvClient": (".core", "GenericEnvClient"),
+    "GenericAction": (".core", "GenericAction"),
+    "SyncEnvClient": (".core", "SyncEnvClient"),
+}
+
+
+def __getattr__(name: str):
+    if name in _LAZY_MODULES:
+        module = import_module(_LAZY_MODULES[name], __name__)
+        globals()[name] = module
+        return module
+
+    if name in _LAZY_ATTRS:
+        module_path, attr_name = _LAZY_ATTRS[name]
+        module = import_module(module_path, __name__)
+        value = getattr(module, attr_name)
+        globals()[name] = value
+        return value
+
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
+
+
+def __dir__() -> list[str]:
+    return sorted(set(globals().keys()) | set(__all__))

{openenv-0.2.1 → openenv-0.2.3}/src/openenv/auto/_discovery.py

@@ -24,9 +24,9 @@ import json
 import logging
 import re
 import tempfile
-from dataclasses import dataclass, asdict
+from dataclasses import asdict, dataclass
 from pathlib import Path
-from typing import Dict, List, Optional, Type, Any
+from typing import Any, Dict, Optional, Type
 
 import yaml
 

{openenv-0.2.1 → openenv-0.2.3}/src/openenv/auto/auto_action.py

@@ -33,9 +33,9 @@ Example:
 from __future__ import annotations
 
 import logging
-from typing import Type, Dict, Any
+from typing import Any, Dict, Type
 
-from ._discovery import get_discovery, _is_hub_url
+from ._discovery import _is_hub_url, get_discovery
 from .auto_env import AutoEnv
 
 logger = logging.getLogger(__name__)
@@ -153,9 +153,9 @@ class AutoAction:
 
             if not available_envs:
                 raise ValueError(
-                    f"No OpenEnv environments found.\n"
-                    f"Install an environment with: pip install openenv-<env-name>\n"
-                    f"Or specify a HuggingFace Hub repository: AutoAction.from_env('openenv/echo_env')"
+                    "No OpenEnv environments found.\n"
+                    "Install an environment with: pip install openenv-<env-name>\n"
+                    "Or specify a HuggingFace Hub repository: AutoAction.from_env('openenv/echo_env')"
                 )
 
             # Try to suggest similar environment names

{openenv-0.2.1 → openenv-0.2.3}/src/openenv/auto/auto_env.py

@@ -36,10 +36,13 @@ import os
 import shutil
 import subprocess
 import sys
+from typing import Any, Dict, Optional, TYPE_CHECKING
+
 import requests
-from typing import Any, Optional, TYPE_CHECKING, Dict
+from openenv.core.utils import run_async_safely
+
+from ._discovery import _is_hub_url, get_discovery
 
-from ._discovery import get_discovery, _is_hub_url
 
 if TYPE_CHECKING:
     from openenv.core.containers.runtime import ContainerProvider
@@ -590,12 +593,14 @@ class AutoEnv:
                         f"Space not running at {space_url}, "
                         f"using GenericEnvClient with HF Docker registry"
                     )
-                    return GenericEnvClient.from_env(
-                        name,
-                        use_docker=True,
-                        provider=container_provider,
-                        env_vars=env_vars or {},
-                        **kwargs,
+                    return run_async_safely(
+                        GenericEnvClient.from_env(
+                            name,
+                            use_docker=True,
+                            provider=container_provider,
+                            env_vars=env_vars or {},
+                            **kwargs,
+                        )
                     )
 
             # For local environments with skip_install, we need docker_image
@@ -604,12 +609,14 @@ class AutoEnv:
                     f"Using GenericEnvClient with Docker image {docker_image} "
                     f"(skip_install=True)"
                 )
-                return GenericEnvClient.from_docker_image(
-                    image=docker_image,
-                    provider=container_provider,
-                    wait_timeout=wait_timeout,
-                    env_vars=env_vars or {},
-                    **kwargs,
+                return run_async_safely(
+                    GenericEnvClient.from_docker_image(
+                        image=docker_image,
+                        provider=container_provider,
+                        wait_timeout=wait_timeout,
+                        env_vars=env_vars or {},
+                        **kwargs,
+                    )
                 )
             else:
                 raise ValueError(
@@ -715,12 +722,14 @@ class AutoEnv:
                     # Local server not running, auto-start Docker container
                     logger.info(f"❌ Server not available at {base_url}")
                     logger.info(f"🐳 Auto-starting Docker container: {docker_image}")
-                    return client_class.from_docker_image(
-                        image=docker_image,
-                        provider=container_provider,
-                        wait_timeout=wait_timeout,
-                        env_vars=env_vars or {},
-                        **kwargs,
+                    return run_async_safely(
+                        client_class.from_docker_image(
+                            image=docker_image,
+                            provider=container_provider,
+                            wait_timeout=wait_timeout,
+                            env_vars=env_vars or {},
+                            **kwargs,
+                        )
                     )
                 else:
                     # Remote server not available, cannot auto-start
@@ -730,12 +739,14 @@ class AutoEnv:
                     )
             else:
                 # No base_url provided, start new Docker container
-                return client_class.from_docker_image(
-                    image=docker_image,
-                    provider=container_provider,
-                    wait_timeout=wait_timeout,
-                    env_vars=env_vars or {},
-                    **kwargs,
+                return run_async_safely(
+                    client_class.from_docker_image(
+                        image=docker_image,
+                        provider=container_provider,
+                        wait_timeout=wait_timeout,
+                        env_vars=env_vars or {},
+                        **kwargs,
+                    )
                 )
         except Exception as e:
             raise ValueError(