File size: 971 Bytes
c44dbf3 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 | try:
from openenv.core.env_server.http_server import create_app
except Exception as e:
raise ImportError("openenv is required. Run: uv sync") from e
try:
from ..models import RedTeamAction, RedTeamObservation
from .environment import RedTeamPentestEnvironment
except ImportError:
from models import RedTeamAction, RedTeamObservation
from server.environment import RedTeamPentestEnvironment
app = create_app(
RedTeamPentestEnvironment,
RedTeamAction,
RedTeamObservation,
env_name="redteampentestlab",
max_concurrent_envs=1,
)
@app.get("/")
def root():
"""Lightweight root endpoint for platform probes and manual checks."""
return {
"status": "ok",
"service": "redteampentestlab",
"routes": ["/reset", "/step", "/state", "/health"],
}
def main(host: str = "0.0.0.0", port: int = 8000):
import uvicorn
uvicorn.run(app, host=host, port=port)
if __name__ == "__main__":
main()
|