awesome-templates

agents-cli/commands.py

"""Codex and Qwen Code on the sandbox Agents API.

The image does not install either CLI. A turn looks up ``codex`` or ``qwen``
on ``PATH`` and runs only that binary. Install the one you are about to call:

    npm install -g @openai/codex
    npm install -g @qwen-code/qwen-code

``E2E_RUN_LLM=0`` (set by the local template runner) prepends ``e2e-bin/``.
Those scripts stand in for the CLIs so the HTTP surface can be tested without
a model key. They are not shipped as the default ``PATH``.
"""

from __future__ import annotations

import os
from pathlib import Path

from agent_sandbox.server import SandboxServer, ServerPolicy

_HERE = Path(__file__).resolve().parent
if os.environ.get("E2E_RUN_LLM") == "0":
    os.environ["PATH"] = str(_HERE / "e2e-bin") + os.pathsep + os.environ.get("PATH", "")

# Endpoint rules are the only switch: ``ServerPolicy.only`` turns every endpoint off
# except the ones named here. An exact rule also opens an endpoint of an opt-in group
# such as ``agents``. Add an endpoint id to expose that one route.
policy = ServerPolicy.only(
    "agents_sessions_create",
    "agents_sessions_get",
    "agents_sessions_list",
    "agents_sessions_events_create",
    "agents_sessions_items_list",
    "agents_runtimes_list",
)

server = SandboxServer(policy=policy)
if __name__ == "__main__":
    server.serve(port=int(os.environ.get("ABX_SERVER_PORT", "9000")))