Add complete Energy & Memory RAM Optimization RL Environment

- Full OpenEnv implementation with typed models
- 5 difficulty levels with agent graders (0.0-1.0 scores)
- Meaningful reward function with partial progress signals
- Baseline inference script with reproducible scoring
- Docker deployment ready
- Comprehensive README with environment details

.agents/skills/hf-cli/.hf-skill-manifest.json

@@ -0,0 +1,4 @@
+{
+  "installed_revision": "25b4bb02b995e19625241deb7321d087053146cd",
+  "schema_version": 1
+}

.agents/skills/hf-cli/SKILL.md

@@ -0,0 +1,189 @@
+---
+name: hf-cli
+description: "Hugging Face Hub CLI (`hf`) for downloading, uploading, and managing models, datasets, spaces, buckets, repos, papers, jobs, and more on the Hugging Face Hub. Use when: handling authentication; managing local cache; managing Hugging Face Buckets; running or scheduling jobs on Hugging Face infrastructure; managing Hugging Face repos; discussions and pull requests; browsing models, datasets and spaces; reading, searching, or browsing academic papers; managing collections; querying datasets; configuring spaces; setting up webhooks; or deploying and managing HF Inference Endpoints. Make sure to use this skill whenever the user mentions 'hf', 'huggingface', 'Hugging Face', 'huggingface-cli', or 'hugging face cli', or wants to do anything related to the Hugging Face ecosystem and to AI and ML in general. Also use for cloud storage needs like training checkpoints, data pipelines, or agent traces. Use even if the user doesn't explicitly ask for a CLI command. Replaces the deprecated `huggingface-cli`."
+---
+
+Install: `curl -LsSf https://hf.co/cli/install.sh | bash -s`.
+
+The Hugging Face Hub CLI tool `hf` is available. IMPORTANT: The `hf` command replaces the deprecated `huggingface-cli` command.
+
+Use `hf --help` to view available functions. Note that auth commands are now all under `hf auth` e.g. `hf auth whoami`.
+
+Generated with `huggingface_hub v1.9.0`. Run `hf skills add --force` to regenerate.
+
+## Commands
+
+- `hf download REPO_ID` — Download files from the Hub. `[--type CHOICE --revision TEXT --include TEXT --exclude TEXT --cache-dir TEXT --local-dir TEXT --force-download --dry-run --quiet --max-workers INTEGER]`
+- `hf env` — Print information about the environment.
+- `hf sync` — Sync files between local directory and a bucket. `[--delete --ignore-times --ignore-sizes --plan TEXT --apply TEXT --dry-run --include TEXT --exclude TEXT --filter-from TEXT --existing --ignore-existing --verbose --quiet]`
+- `hf upload REPO_ID` — Upload a file or a folder to the Hub. Recommended for single-commit uploads. `[--type CHOICE --revision TEXT --private --include TEXT --exclude TEXT --delete TEXT --commit-message TEXT --commit-description TEXT --create-pr --every FLOAT --quiet]`
+- `hf upload-large-folder REPO_ID LOCAL_PATH` — Upload a large folder to the Hub. Recommended for resumable uploads. `[--type CHOICE --revision TEXT --private --include TEXT --exclude TEXT --num-workers INTEGER --no-report --no-bars]`
+- `hf version` — Print information about the hf version.
+
+### `hf auth` — Manage authentication (login, logout, etc.).
+
+- `hf auth list` — List all stored access tokens.
+- `hf auth login` — Login using a token from huggingface.co/settings/tokens. `[--add-to-git-credential --force]`
+- `hf auth logout` — Logout from a specific token. `[--token-name TEXT]`
+- `hf auth switch` — Switch between access tokens. `[--token-name TEXT --add-to-git-credential]`
+- `hf auth whoami` — Find out which huggingface.co account you are logged in as. `[--format CHOICE]`
+
+### `hf buckets` — Commands to interact with buckets.
+
+- `hf buckets cp SRC` — Copy a single file to or from a bucket. `[--quiet]`
+- `hf buckets create BUCKET_ID` — Create a new bucket. `[--private --exist-ok --quiet]`
+- `hf buckets delete BUCKET_ID` — Delete a bucket. `[--yes --missing-ok --quiet]`
+- `hf buckets info BUCKET_ID` — Get info about a bucket. `[--quiet]`
+- `hf buckets list` — List buckets or files in a bucket. `[--human-readable --tree --recursive --format CHOICE --quiet]`
+- `hf buckets move FROM_ID TO_ID` — Move (rename) a bucket to a new name or namespace.
+- `hf buckets remove ARGUMENT` — Remove files from a bucket. `[--recursive --yes --dry-run --include TEXT --exclude TEXT --quiet]`
+- `hf buckets sync` — Sync files between local directory and a bucket. `[--delete --ignore-times --ignore-sizes --plan TEXT --apply TEXT --dry-run --include TEXT --exclude TEXT --filter-from TEXT --existing --ignore-existing --verbose --quiet]`
+
+### `hf cache` — Manage local cache directory.
+
+- `hf cache list` — List cached repositories or revisions. `[--cache-dir TEXT --revisions --filter TEXT --format CHOICE --quiet --sort CHOICE --limit INTEGER]`
+- `hf cache prune` — Remove detached revisions from the cache. `[--cache-dir TEXT --yes --dry-run]`
+- `hf cache rm TARGETS` — Remove cached repositories or revisions. `[--cache-dir TEXT --yes --dry-run]`
+- `hf cache verify REPO_ID` — Verify checksums for a single repo revision from cache or a local directory. `[--type CHOICE --revision TEXT --cache-dir TEXT --local-dir TEXT --fail-on-missing-files --fail-on-extra-files]`
+
+### `hf collections` — Interact with collections on the Hub.
+
+- `hf collections add-item COLLECTION_SLUG ITEM_ID ITEM_TYPE` — Add an item to a collection. `[--note TEXT --exists-ok]`
+- `hf collections create TITLE` — Create a new collection on the Hub. `[--namespace TEXT --description TEXT --private --exists-ok]`
+- `hf collections delete COLLECTION_SLUG` — Delete a collection from the Hub. `[--missing-ok]`
+- `hf collections delete-item COLLECTION_SLUG ITEM_OBJECT_ID` — Delete an item from a collection. `[--missing-ok]`
+- `hf collections info COLLECTION_SLUG` — Get info about a collection on the Hub. Output is in JSON format.
+- `hf collections list` — List collections on the Hub. `[--owner TEXT --item TEXT --sort CHOICE --limit INTEGER --format CHOICE --quiet]`
+- `hf collections update COLLECTION_SLUG` — Update a collection's metadata on the Hub. `[--title TEXT --description TEXT --position INTEGER --private --theme TEXT]`
+- `hf collections update-item COLLECTION_SLUG ITEM_OBJECT_ID` — Update an item in a collection. `[--note TEXT --position INTEGER]`
+
+### `hf datasets` — Interact with datasets on the Hub.
+
+- `hf datasets info DATASET_ID` — Get info about a dataset on the Hub. `[--revision TEXT --expand TEXT --format CHOICE]`
+- `hf datasets list` — List datasets on the Hub. `[--search TEXT --author TEXT --filter TEXT --sort CHOICE --limit INTEGER --expand TEXT --format CHOICE]`
+- `hf datasets parquet DATASET_ID` — List parquet file URLs available for a dataset. `[--subset TEXT --split TEXT --format CHOICE]`
+- `hf datasets sql SQL` — Execute a raw SQL query with DuckDB against dataset parquet URLs. `[--format CHOICE]`
+
+### `hf discussions` — Manage discussions and pull requests on the Hub.
+
+- `hf discussions close REPO_ID NUM` — Close a discussion or pull request. `[--comment TEXT --yes --type CHOICE]`
+- `hf discussions comment REPO_ID NUM` — Comment on a discussion or pull request. `[--body TEXT --body-file PATH --type CHOICE]`
+- `hf discussions create REPO_ID --title TEXT` — Create a new discussion or pull request on a repo. `[--body TEXT --body-file PATH --pull-request --type CHOICE]`
+- `hf discussions diff REPO_ID NUM` — Show the diff of a pull request. `[--type CHOICE]`
+- `hf discussions info REPO_ID NUM` — Get info about a discussion or pull request. `[--comments --diff --no-color --type CHOICE --format CHOICE]`
+- `hf discussions list REPO_ID` — List discussions and pull requests on a repo. `[--status CHOICE --kind CHOICE --author TEXT --limit INTEGER --type CHOICE --format CHOICE --quiet]`
+- `hf discussions merge REPO_ID NUM` — Merge a pull request. `[--comment TEXT --yes --type CHOICE]`
+- `hf discussions rename REPO_ID NUM NEW_TITLE` — Rename a discussion or pull request. `[--type CHOICE]`
+- `hf discussions reopen REPO_ID NUM` — Reopen a closed discussion or pull request. `[--comment TEXT --yes --type CHOICE]`
+
+### `hf endpoints` — Manage Hugging Face Inference Endpoints.
+
+- `hf endpoints catalog deploy --repo TEXT` — Deploy an Inference Endpoint from the Model Catalog. `[--name TEXT --accelerator TEXT --namespace TEXT]`
+- `hf endpoints catalog list` — List available Catalog models.
+- `hf endpoints delete NAME` — Delete an Inference Endpoint permanently. `[--namespace TEXT --yes]`
+- `hf endpoints deploy NAME --repo TEXT --framework TEXT --accelerator TEXT --instance-size TEXT --instance-type TEXT --region TEXT --vendor TEXT` — Deploy an Inference Endpoint from a Hub repository. `[--namespace TEXT --task TEXT --min-replica INTEGER --max-replica INTEGER --scale-to-zero-timeout INTEGER --scaling-metric CHOICE --scaling-threshold FLOAT]`
+- `hf endpoints describe NAME` — Get information about an existing endpoint. `[--namespace TEXT]`
+- `hf endpoints list` — Lists all Inference Endpoints for the given namespace. `[--namespace TEXT --format CHOICE --quiet]`
+- `hf endpoints pause NAME` — Pause an Inference Endpoint. `[--namespace TEXT]`
+- `hf endpoints resume NAME` — Resume an Inference Endpoint. `[--namespace TEXT --fail-if-already-running]`
+- `hf endpoints scale-to-zero NAME` — Scale an Inference Endpoint to zero. `[--namespace TEXT]`
+- `hf endpoints update NAME` — Update an existing endpoint. `[--namespace TEXT --repo TEXT --accelerator TEXT --instance-size TEXT --instance-type TEXT --framework TEXT --revision TEXT --task TEXT --min-replica INTEGER --max-replica INTEGER --scale-to-zero-timeout INTEGER --scaling-metric CHOICE --scaling-threshold FLOAT]`
+
+### `hf extensions` — Manage hf CLI extensions.
+
+- `hf extensions exec NAME` — Execute an installed extension.
+- `hf extensions install REPO_ID` — Install an extension from a public GitHub repository. `[--force]`
+- `hf extensions list` — List installed extension commands. `[--format CHOICE --quiet]`
+- `hf extensions remove NAME` — Remove an installed extension.
+- `hf extensions search` — Search extensions available on GitHub (tagged with 'hf-extension' topic). `[--format CHOICE --quiet]`
+
+### `hf jobs` — Run and manage Jobs on the Hub.
+
+- `hf jobs cancel JOB_ID` — Cancel a Job `[--namespace TEXT]`
+- `hf jobs hardware` — List available hardware options for Jobs
+- `hf jobs inspect JOB_IDS` — Display detailed information on one or more Jobs `[--namespace TEXT]`
+- `hf jobs logs JOB_ID` — Fetch the logs of a Job. `[--follow --tail INTEGER --namespace TEXT]`
+- `hf jobs ps` — List Jobs. `[--all --namespace TEXT --filter TEXT --format TEXT --quiet]`
+- `hf jobs run IMAGE COMMAND` — Run a Job. `[--env TEXT --secrets TEXT --label TEXT --volume TEXT --env-file TEXT --secrets-file TEXT --flavor CHOICE --timeout TEXT --detach --namespace TEXT]`
+- `hf jobs scheduled delete SCHEDULED_JOB_ID` — Delete a scheduled Job. `[--namespace TEXT]`
+- `hf jobs scheduled inspect SCHEDULED_JOB_IDS` — Display detailed information on one or more scheduled Jobs `[--namespace TEXT]`
+- `hf jobs scheduled ps` — List scheduled Jobs `[--all --namespace TEXT --filter TEXT --format TEXT --quiet]`
+- `hf jobs scheduled resume SCHEDULED_JOB_ID` — Resume (unpause) a scheduled Job. `[--namespace TEXT]`
+- `hf jobs scheduled run SCHEDULE IMAGE COMMAND` — Schedule a Job. `[--suspend --concurrency --env TEXT --secrets TEXT --label TEXT --volume TEXT --env-file TEXT --secrets-file TEXT --flavor CHOICE --timeout TEXT --namespace TEXT]`
+- `hf jobs scheduled suspend SCHEDULED_JOB_ID` — Suspend (pause) a scheduled Job. `[--namespace TEXT]`
+- `hf jobs scheduled uv run SCHEDULE SCRIPT` — Run a UV script (local file or URL) on HF infrastructure `[--suspend --concurrency --image TEXT --flavor CHOICE --env TEXT --secrets TEXT --label TEXT --volume TEXT --env-file TEXT --secrets-file TEXT --timeout TEXT --namespace TEXT --with TEXT --python TEXT]`
+- `hf jobs stats` — Fetch the resource usage statistics and metrics of Jobs `[--namespace TEXT]`
+- `hf jobs uv run SCRIPT` — Run a UV script (local file or URL) on HF infrastructure `[--image TEXT --flavor CHOICE --env TEXT --secrets TEXT --label TEXT --volume TEXT --env-file TEXT --secrets-file TEXT --timeout TEXT --detach --namespace TEXT --with TEXT --python TEXT]`
+
+### `hf models` — Interact with models on the Hub.
+
+- `hf models info MODEL_ID` — Get info about a model on the Hub. `[--revision TEXT --expand TEXT --format CHOICE]`
+- `hf models list` — List models on the Hub. `[--search TEXT --author TEXT --filter TEXT --num-parameters TEXT --sort CHOICE --limit INTEGER --expand TEXT --format CHOICE]`
+
+### `hf papers` — Interact with papers on the Hub.
+
+- `hf papers info PAPER_ID` — Get info about a paper on the Hub. `[--format CHOICE]`
+- `hf papers list` — List daily papers on the Hub. `[--date TEXT --week TEXT --month TEXT --submitter TEXT --sort CHOICE --limit INTEGER --format CHOICE]`
+- `hf papers read PAPER_ID` — Read a paper as markdown.
+- `hf papers search QUERY` — Search papers on the Hub. `[--limit INTEGER --format CHOICE]`
+
+### `hf repos` — Manage repos on the Hub.
+
+- `hf repos branch create REPO_ID BRANCH` — Create a new branch for a repo on the Hub. `[--revision TEXT --type CHOICE --exist-ok]`
+- `hf repos branch delete REPO_ID BRANCH` — Delete a branch from a repo on the Hub. `[--type CHOICE]`
+- `hf repos create REPO_ID` — Create a new repo on the Hub. `[--type CHOICE --space-sdk TEXT --private --public --protected --exist-ok --resource-group-id TEXT --flavor CHOICE --storage CHOICE --sleep-time INTEGER --secrets TEXT --secrets-file TEXT --env TEXT --env-file TEXT --volume TEXT]`
+- `hf repos delete REPO_ID` — Delete a repo from the Hub. This is an irreversible operation. `[--type CHOICE --missing-ok]`
+- `hf repos delete-files REPO_ID PATTERNS` — Delete files from a repo on the Hub. `[--type CHOICE --revision TEXT --commit-message TEXT --commit-description TEXT --create-pr]`
+- `hf repos duplicate FROM_ID` — Duplicate a repo on the Hub (model, dataset, or Space). `[--type CHOICE --private --public --protected --exist-ok --flavor CHOICE --storage CHOICE --sleep-time INTEGER --secrets TEXT --secrets-file TEXT --env TEXT --env-file TEXT --volume TEXT]`
+- `hf repos move FROM_ID TO_ID` — Move a repository from a namespace to another namespace. `[--type CHOICE]`
+- `hf repos settings REPO_ID` — Update the settings of a repository. `[--gated CHOICE --private --public --protected --type CHOICE]`
+- `hf repos tag create REPO_ID TAG` — Create a tag for a repo. `[--message TEXT --revision TEXT --type CHOICE]`
+- `hf repos tag delete REPO_ID TAG` — Delete a tag for a repo. `[--yes --type CHOICE]`
+- `hf repos tag list REPO_ID` — List tags for a repo. `[--type CHOICE]`
+
+### `hf skills` — Manage skills for AI assistants.
+
+- `hf skills add` — Download a Hugging Face skill and install it for an AI assistant. `[--claude --global --dest PATH --force]`
+- `hf skills preview` — Print the generated `hf-cli` SKILL.md to stdout.
+- `hf skills upgrade` — Upgrade installed Hugging Face marketplace skills. `[--claude --global --dest PATH]`
+
+### `hf spaces` — Interact with spaces on the Hub.
+
+- `hf spaces dev-mode SPACE_ID` — Enable or disable dev mode on a Space. `[--stop]`
+- `hf spaces hot-reload SPACE_ID` — Hot-reload any Python file of a Space without a full rebuild + restart. `[--local-file TEXT --skip-checks --skip-summary]`
+- `hf spaces info SPACE_ID` — Get info about a space on the Hub. `[--revision TEXT --expand TEXT --format CHOICE]`
+- `hf spaces list` — List spaces on the Hub. `[--search TEXT --author TEXT --filter TEXT --sort CHOICE --limit INTEGER --expand TEXT --format CHOICE]`
+
+### `hf webhooks` — Manage webhooks on the Hub.
+
+- `hf webhooks create --watch TEXT` — Create a new webhook. `[--url TEXT --job-id TEXT --domain CHOICE --secret TEXT]`
+- `hf webhooks delete WEBHOOK_ID` — Delete a webhook permanently. `[--yes]`
+- `hf webhooks disable WEBHOOK_ID` — Disable an active webhook.
+- `hf webhooks enable WEBHOOK_ID` — Enable a disabled webhook.
+- `hf webhooks info WEBHOOK_ID` — Show full details for a single webhook as JSON.
+- `hf webhooks list` — List all webhooks for the current user. `[--format CHOICE --quiet]`
+- `hf webhooks update WEBHOOK_ID` — Update an existing webhook. Only provided options are changed. `[--url TEXT --watch TEXT --domain CHOICE --secret TEXT]`
+
+## Common options
+
+- `--format` — Output format: `--format json` (or `--json`) or `--format table` (default).
+- `-q / --quiet` — Minimal output.
+- `--revision` — Git revision id which can be a branch name, a tag, or a commit hash.
+- `--token` — Use a User Access Token. Prefer setting `HF_TOKEN` env var instead of passing `--token`.
+- `--type` — The type of repository (model, dataset, or space).
+
+## Mounting repos as local filesystems
+
+To mount Hub repositories or buckets as local filesystems — no download, no copy, no waiting — use `hf-mount`. Files are fetched on demand. GitHub: https://github.com/huggingface/hf-mount
+
+Install: `curl -fsSL https://raw.githubusercontent.com/huggingface/hf-mount/main/install.sh | sh`
+
+Some command examples:
+- `hf-mount start repo openai-community/gpt2 /tmp/gpt2` — mount a repo (read-only)
+- `hf-mount start --hf-token $HF_TOKEN bucket myuser/my-bucket /tmp/data` — mount a bucket (read-write)
+- `hf-mount status` / `hf-mount stop /tmp/data` — list or unmount
+
+## Tips
+
+- Use `hf <command> --help` for full options, descriptions, usage, and real-world examples
+- Authenticate with `HF_TOKEN` env var (recommended) or with `--token`

.dockerignore

@@ -0,0 +1,51 @@
+# Virtual environments
+.venv/
+venv/
+env/
+
+# Python cache
+__pycache__/
+*.pyc
+*.pyo
+*.pyd
+
+# Git
+.git/
+.gitignore
+
+# IDE
+.vscode/
+.idea/
+
+# OS
+.DS_Store
+Thumbs.db
+
+# Logs
+*.log
+
+# Temporary files
+*.tmp
+*.swp
+
+# Build artifacts
+dist/
+build/
+*.egg-info/
+
+# Node modules (if any)
+node_modules/
+
+# Cache directories
+.cache/
+.pytest_cache/
+
+# OpenEnv specific
+.openenv/
+
+# Local development files
+.env
+.env.local
+
+# Training artifacts (keep model if needed)
+# energy_optimization_ppo.zip

.env

@@ -0,0 +1,20 @@
+# Environment configuration for Energy & Memory RAM Optimization RL Environment
+# OPTION A: Qwen/Qwen3.5-27B (Recommended for technical reasoning)
+#
+# Required Environment Variables:
+# - API_BASE_URL: The API endpoint for the LLM
+# - MODEL_NAME: The model identifier to use for inference
+# - HF_TOKEN: Your Hugging Face / API key
+# - LOCAL_IMAGE_NAME: The name of the local image to use for the environment (optional)
+
+# The API endpoint for the LLM
+API_BASE_URL=https://router.huggingface.co/v1
+
+# The model identifier to use for inference (Option A: Qwen for technical optimization)
+MODEL_NAME=Qwen/Qwen3.5-27B
+
+# Your Hugging Face / API key
+HF_TOKEN=
+
+# The name of the local image to use for the environment if using from_docker_image() method
+LOCAL_IMAGE_NAME=energy-optimization-rl

.gitignore

@@ -0,0 +1,46 @@
+# Environment files
+.env
+.env.local
+
+# Virtual environments
+.venv/
+venv/
+env/
+
+# Python cache
+__pycache__/
+*.pyc
+*.pyo
+*.pyd
+
+# IDE
+.vscode/
+.idea/
+
+# ML Models
+*_ppo.zip
+energy_optimization_ppo.zip
+
+# OS
+.DS_Store
+Thumbs.db
+
+# Logs
+*.log
+
+# Temporary files
+*.tmp
+*.swp
+
+# Build artifacts
+dist/
+build/
+*.egg-info/
+
+# Model files (if large)
+*.zip
+*.pkl
+*.h5
+
+# Agents
+.agents/

.vscode/settings.json

@@ -0,0 +1,18 @@
+{
+  "python.defaultInterpreterPath": "./.venv/Scripts/python.exe",
+  "python.analysis.extraPaths": [
+    "."
+  ],
+  "python.analysis.include": [
+    "."
+  ],
+  "python.analysis.exclude": [
+    ".venv",
+    "__pycache__",
+    ".git"
+  ],
+  "python.linting.enabled": true,
+  "python.linting.pylintEnabled": false,
+  "python.linting.flake8Enabled": false,
+  "python.linting.mypyEnabled": false
+}

Dockerfile

@@ -0,0 +1,82 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the BSD-style license found in the
+# LICENSE file in the root directory of this source tree.
+
+# Multi-stage build using openenv-base
+# This Dockerfile is flexible and works for both:
+# - In-repo environments (with local OpenEnv sources)
+# - Standalone environments (with openenv from PyPI/Git)
+# The build script (openenv build) handles context detection and sets appropriate build args.
+
+ARG BASE_IMAGE=ghcr.io/meta-pytorch/openenv-base:latest
+FROM ${BASE_IMAGE} AS builder
+
+WORKDIR /app
+
+# Ensure git is available (required for installing dependencies from VCS)
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends git && \
+    rm -rf /var/lib/apt/lists/*
+
+# Build argument to control whether we're building standalone or in-repo
+ARG BUILD_MODE=in-repo
+ARG ENV_NAME=he_demo
+
+# Copy environment code (always at root of build context)
+COPY . /app/env
+
+# For in-repo builds, openenv is already vendored in the build context
+# For standalone builds, openenv will be installed via pyproject.toml
+WORKDIR /app/env
+
+# Ensure uv is available (for local builds where base image lacks it)
+RUN if ! command -v uv >/dev/null 2>&1; then \
+        curl -LsSf https://astral.sh/uv/install.sh | sh && \
+        mv /root/.local/bin/uv /usr/local/bin/uv && \
+        mv /root/.local/bin/uvx /usr/local/bin/uvx; \
+    fi
+    
+# Install dependencies using uv sync
+# If uv.lock exists, use it; otherwise resolve on the fly
+RUN --mount=type=cache,target=/root/.cache/uv \
+    if [ -f uv.lock ]; then \
+        uv sync --frozen --no-install-project --no-editable; \
+    else \
+        uv sync --no-install-project --no-editable; \
+    fi
+
+RUN --mount=type=cache,target=/root/.cache/uv \
+    if [ -f uv.lock ]; then \
+        uv sync --frozen --no-editable; \
+    else \
+        uv sync --no-editable; \
+    fi
+
+# Final runtime stage
+FROM ${BASE_IMAGE}
+
+WORKDIR /app
+
+# Copy the virtual environment from builder
+COPY --from=builder /app/env/.venv /app/.venv
+
+# Copy the environment code
+COPY --from=builder /app/env /app/env
+
+# Set PATH to use the virtual environment
+ENV PATH="/app/.venv/bin:$PATH"
+
+# Set PYTHONPATH so imports work correctly
+ENV PYTHONPATH="/app:$PYTHONPATH"
+
+ENV ENABLE_WEB_INTERFACE=true
+
+# Health check
+HEALTHCHECK --interval=30s --timeout=3s --start-period=5s --retries=3 \
+    CMD curl -f http://localhost:8000/health || exit 1
+
+# Run the FastAPI server
+# The module path is constructed to work with the /app/env structure
+CMD ["sh", "-c", "cd /app/env && uvicorn he_demo.server.app:app --host 0.0.0.0 --port 8000"]

Dockerfile.simple

@@ -0,0 +1,28 @@
+# Simple Dockerfile for Energy & Memory RAM Optimization Environment
+FROM python:3.11-slim
+
+WORKDIR /app
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    git \
+    && rm -rf /var/lib/apt/lists/*
+
+# Copy project files
+COPY pyproject.toml uv.lock ./
+COPY . .
+
+# Install uv if not available
+RUN pip install uv
+
+# Install dependencies
+RUN uv sync --frozen --no-install-project
+
+# Install the project itself
+RUN uv pip install -e .
+
+# Expose port
+EXPOSE 8000
+
+# Run the server
+CMD ["uv", "run", "server"]

README.md

@@ -0,0 +1,157 @@
+---
+title: Energy & Memory RAM Optimization Environment
+emoji: ⚡
+colorFrom: blue
+colorTo: green
+sdk: docker
+pinned: false
+app_port: 8000
+base_path: /web
+tags:
+  - openenv
+  - reinforcement-learning
+  - energy-optimization
+  - resource-management
+---
+
+# Energy & Memory RAM Optimization RL Environment
+
+An OpenEnv-based reinforcement learning environment for training AI agents to optimize energy consumption and RAM usage in computer systems. The environment features tasks of increasing difficulty, automated graders for task completion verification, and sophisticated reward logic.
+
+## Features
+
+### AI Agent Capabilities
+- **Resource Detection**: Real-time monitoring of RAM usage and energy consumption
+- **Optimization Strategies**: Multiple action types for different optimization approaches
+- **Adaptive Learning**: Agents learn to balance competing objectives (RAM vs energy efficiency)
+
+### Task Progression
+Tasks increase in difficulty from basic resource reduction to advanced multi-objective optimization:
+
+1. **Basic RAM Reduction**: Reduce RAM usage below 70%
+2. **Energy Optimization**: Reduce energy consumption below 6 kWh while maintaining RAM below 75%
+3. **Balanced Optimization**: Balance RAM below 60% and energy below 5 kWh
+4. **Advanced Efficiency**: Achieve RAM below 50% and energy below 4 kWh
+5. **Expert Optimization**: Master level: RAM below 40% and energy below 3 kWh
+
+### Automated Graders
+- **Task Completion Verification**: Automatic checking of optimization targets
+- **Performance Metrics**: Efficiency scores and progress tracking
+- **Reward Validation**: Ensures fair scoring based on actual improvements
+
+### Reward Logic
+- **Action Effectiveness**: Rewards based on actual resource reductions achieved
+- **Task Completion Bonuses**: Significant rewards for meeting task objectives
+- **Efficiency Incentives**: Bonuses for overall system optimization
+- **Penalty System**: Penalties for aggressive actions that may cause system instability
+
+## Quick Start
+
+### Installation
+```bash
+# Install dependencies
+pip install -r requirements.txt
+
+# Or using uv (recommended)
+uv sync
+```
+
+### Running the Environment
+```bash
+# Start the OpenEnv server
+uv run server
+
+# The server will be available at http://localhost:8000
+```
+
+### Training an Agent
+```python
+from stable_baselines3 import PPO
+from openenv.client import OpenEnvClient
+
+# Connect to the environment
+client = OpenEnvClient("http://localhost:8000")
+
+# Create and train agent
+model = PPO("MlpPolicy", client, verbose=1)
+model.learn(total_timesteps=10000)
+
+# Evaluate the trained agent
+obs = client.reset()
+total_reward = 0
+while not obs.done:
+    action, _ = model.predict(obs)
+    obs = client.step(action)
+    total_reward += obs.reward
+    print(f"Step reward: {obs.reward:.2f}, Total: {total_reward:.2f}")
+```
+
+## Docker
+
+```bash
+# Build the container
+docker build -t energy-optimization-rl .
+
+# Run the environment
+docker run --rm -p 8000:8000 energy-optimization-rl
+```
+
+## Environment Details
+
+### State Space
+- RAM usage percentage (0-100%)
+- Energy consumption in kWh
+- System load (0-1)
+- Current task information
+- Task completion progress
+- Efficiency scores
+
+### Action Space
+- `reduce_ram`: Focus on RAM optimization with configurable intensity (0.0-1.0)
+- `optimize_energy`: Focus on energy reduction with configurable intensity (0.0-1.0)
+- `balance_resources`: Balanced approach to both resources
+- `monitor_system`: Gather system information and slight load reduction
+
+### Reward Structure
+- Base rewards for resource reductions
+- Task completion bonuses (difficulty × 10 points)
+- Efficiency improvement bonuses
+- Penalties for system instability from aggressive actions
+
+## API Endpoints
+
+- `POST /reset`: Reset the environment
+- `POST /step`: Execute an optimization action
+- `GET /state`: Get current environment state
+- `GET /schema`: Get action/observation schemas
+- `WS /ws`: WebSocket endpoint for persistent sessions
+
+## Development
+
+### Project Structure
+```
+he_demo/
+├── models.py                 # Action and observation definitions
+├── server/
+│   ├── app.py               # FastAPI server application
+│   └── he_demo_environment.py # Environment implementation
+├── client.py                # Example client code
+├── inference.py             # Training and inference scripts
+├── Dockerfile               # Container configuration
+├── pyproject.toml           # Project dependencies
+└── README.md               # This file
+```
+
+### Adding New Tasks
+Tasks are defined in the `_create_tasks()` method of `EnergyOptimizationEnvironment`. Each task includes:
+- Name and description
+- Difficulty level
+- RAM and energy targets
+- Maximum steps allowed
+
+### Customizing Reward Logic
+Modify the `_calculate_reward()` method to implement custom reward strategies based on your specific optimization goals.
+
+## License
+
+This project is licensed under the BSD-style license. See LICENSE file for details.

__init__.py

@@ -0,0 +1,29 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the BSD-style license found in the
+# LICENSE file in the root directory of this source tree.
+
+"""Energy & Memory RAM Optimization Environment."""
+
+from .client import EnergyOptimizationEnv
+from .models import EnergyOptimizationAction, EnergyOptimizationObservation, Task
+from .graders import (
+    grade_basic_ram_reduction,
+    grade_energy_optimization,
+    grade_balanced_optimization,
+    grade_advanced_efficiency,
+    grade_expert_optimization,
+)
+
+__all__ = [
+    "EnergyOptimizationAction",
+    "EnergyOptimizationObservation",
+    "Task",
+    "EnergyOptimizationEnv",
+    "grade_basic_ram_reduction",
+    "grade_energy_optimization", 
+    "grade_balanced_optimization",
+    "grade_advanced_efficiency",
+    "grade_expert_optimization",
+]

client.py

@@ -0,0 +1,123 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the BSD-style license found in the
+# LICENSE file in the root directory of this source tree.
+
+"""He Demo Environment Client."""
+
+from typing import Dict
+
+from openenv.core import EnvClient
+from openenv.core.client_types import StepResult
+from openenv.core.env_server.types import State
+
+from .models import EnergyOptimizationAction, EnergyOptimizationObservation, Task, TaskSummary
+
+
+class EnergyOptimizationEnv(
+    EnvClient[EnergyOptimizationAction, EnergyOptimizationObservation, State]
+):
+    """
+    Client for the Energy & Memory RAM Optimization Environment.
+
+    This client maintains a persistent WebSocket connection to the environment server,
+    enabling efficient multi-step interactions with lower latency.
+    Each client instance has its own dedicated environment session on the server.
+
+    Example:
+        >>> # Connect to a running server
+        >>> with EnergyOptimizationEnv(base_url="http://localhost:8000") as client:
+        ...     result = client.reset()
+        ...     print(f"RAM: {result.observation.ram_usage:.1f}%, Energy: {result.observation.energy_consumption:.1f} kWh")
+        ...
+        ...     result = client.step(EnergyOptimizationAction(action_type="reduce_ram", intensity=0.8))
+        ...     print(f"Task: {result.observation.current_task.name if result.observation.current_task else 'None'}")
+
+    Example with Docker:
+        >>> # Automatically start container and connect
+        >>> client = EnergyOptimizationEnv.from_docker_image("energy-optimization-env:latest")
+        >>> try:
+        ...     result = client.reset()
+        ...     result = client.step(EnergyOptimizationAction(action_type="balance_resources", intensity=0.6))
+        ... finally:
+        ...     client.close()
+    """
+
+    def _step_payload(self, action: EnergyOptimizationAction) -> Dict:
+        """
+        Convert EnergyOptimizationAction to JSON payload for step message.
+
+        Args:
+            action: EnergyOptimizationAction instance
+
+        Returns:
+            Dictionary representation suitable for JSON encoding
+        """
+        return {
+            "action_type": action.action_type,
+            "intensity": action.intensity,
+        }
+
+    def _parse_result(self, payload: Dict) -> StepResult[EnergyOptimizationObservation]:
+        """
+        Parse server response into StepResult[EnergyOptimizationObservation].
+
+        Args:
+            payload: JSON response data from server
+
+        Returns:
+            StepResult with EnergyOptimizationObservation
+        """
+        obs_data = payload.get("observation", {})
+
+        # Parse current task if present
+        current_task = None
+        if obs_data.get("current_task"):
+            task_data = obs_data["current_task"]
+            current_task = TaskSummary(
+                name=task_data.get("name", ""),
+                description=task_data.get("description", ""),
+                difficulty=task_data.get("difficulty", 1),
+                ram_target=task_data.get("ram_target", 100.0),
+                energy_target=task_data.get("energy_target", 10.0),
+                max_steps=task_data.get("max_steps", 10),
+                completed=task_data.get("completed", False),
+                remaining_steps=task_data.get("remaining_steps"),
+                progress=task_data.get("progress", 0.0)
+            )
+
+        observation = EnergyOptimizationObservation(
+            ram_usage=obs_data.get("ram_usage", 0.0),
+            energy_consumption=obs_data.get("energy_consumption", 0.0),
+            system_load=obs_data.get("system_load", 0.0),
+            current_task=current_task,
+            tasks_completed=obs_data.get("tasks_completed", []),
+            steps_taken=obs_data.get("steps_taken", 0),
+            task_progress=obs_data.get("task_progress", 0.0),
+            efficiency_score=obs_data.get("efficiency_score", 0.0),
+            done=payload.get("done", False),
+            reward=payload.get("reward"),
+            metadata=obs_data.get("metadata", {}),
+        )
+
+        return StepResult(
+            observation=observation,
+            reward=payload.get("reward"),
+            done=payload.get("done", False),
+        )
+
+    def _parse_state(self, payload: Dict) -> State:
+        """
+        Parse server response into State object.
+
+        Args:
+            payload: JSON response from state request
+
+        Returns:
+            State object with episode_id and step_count
+        """
+        return State(
+            episode_id=payload.get("episode_id"),
+            step_count=payload.get("step_count", 0),
+        )

graders.py

@@ -0,0 +1,64 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the BSD-style license found in the
+# LICENSE file in the root directory of this source tree.
+
+"""
+Task graders for the Energy & Memory RAM Optimization Environment.
+
+Each grader function evaluates agent performance on a specific task,
+returning a score from 0.0 (worst) to 1.0 (best).
+"""
+
+from he_demo.models import EnergyOptimizationObservation
+
+
+def grade_basic_ram_reduction(observation: EnergyOptimizationObservation) -> float:
+    """Grade performance on basic RAM reduction task: Reduce RAM usage below 70%."""
+    # Target: RAM <= 70%, Energy <= 7.5 kWh, within 10 steps
+    ram_score = max(0.0, min(1.0, (100.0 - observation.ram_usage) / (100.0 - 70.0)))
+    energy_score = max(0.0, min(1.0, (10.0 - observation.energy_consumption) / (10.0 - 7.5)))
+    step_penalty = 1.0 if observation.steps_taken <= 10 else max(0.0, 1.0 - (observation.steps_taken - 10) * 0.1)
+    
+    return (ram_score + energy_score) / 2.0 * step_penalty
+
+
+def grade_energy_optimization(observation: EnergyOptimizationObservation) -> float:
+    """Grade performance on energy optimization task: Reduce energy below 6 kWh while maintaining RAM below 75%."""
+    # Target: RAM <= 75%, Energy <= 6.0 kWh, within 15 steps
+    ram_score = max(0.0, min(1.0, (100.0 - observation.ram_usage) / (100.0 - 75.0)))
+    energy_score = max(0.0, min(1.0, (10.0 - observation.energy_consumption) / (10.0 - 6.0)))
+    step_penalty = 1.0 if observation.steps_taken <= 15 else max(0.0, 1.0 - (observation.steps_taken - 15) * 0.1)
+    
+    return (ram_score + energy_score) / 2.0 * step_penalty
+
+
+def grade_balanced_optimization(observation: EnergyOptimizationObservation) -> float:
+    """Grade performance on balanced optimization task: Balance RAM below 60% and energy below 5 kWh."""
+    # Target: RAM <= 60%, Energy <= 5.0 kWh, within 20 steps
+    ram_score = max(0.0, min(1.0, (100.0 - observation.ram_usage) / (100.0 - 60.0)))
+    energy_score = max(0.0, min(1.0, (10.0 - observation.energy_consumption) / (10.0 - 5.0)))
+    step_penalty = 1.0 if observation.steps_taken <= 20 else max(0.0, 1.0 - (observation.steps_taken - 20) * 0.1)
+    
+    return (ram_score + energy_score) / 2.0 * step_penalty
+
+
+def grade_advanced_efficiency(observation: EnergyOptimizationObservation) -> float:
+    """Grade performance on advanced efficiency task: Achieve RAM below 50% and energy below 4 kWh."""
+    # Target: RAM <= 50%, Energy <= 4.0 kWh, within 25 steps
+    ram_score = max(0.0, min(1.0, (100.0 - observation.ram_usage) / (100.0 - 50.0)))
+    energy_score = max(0.0, min(1.0, (10.0 - observation.energy_consumption) / (10.0 - 4.0)))
+    step_penalty = 1.0 if observation.steps_taken <= 25 else max(0.0, 1.0 - (observation.steps_taken - 25) * 0.1)
+    
+    return (ram_score + energy_score) / 2.0 * step_penalty
+
+
+def grade_expert_optimization(observation: EnergyOptimizationObservation) -> float:
+    """Grade performance on expert optimization task: Master level - RAM below 40% and energy below 3 kWh."""
+    # Target: RAM <= 40%, Energy <= 3.0 kWh, within 30 steps
+    ram_score = max(0.0, min(1.0, (100.0 - observation.ram_usage) / (100.0 - 40.0)))
+    energy_score = max(0.0, min(1.0, (10.0 - observation.energy_consumption) / (10.0 - 3.0)))
+    step_penalty = 1.0 if observation.steps_taken <= 30 else max(0.0, 1.0 - (observation.steps_taken - 30) * 0.1)
+    
+    return (ram_score + energy_score) / 2.0 * step_penalty

gym_wrapper.py

@@ -0,0 +1,99 @@
+#!/usr/bin/env python3
+"""
+Gym wrapper for the Energy Optimization Environment.
+"""
+
+import sys
+import os
+import gymnasium as gym
+import numpy as np
+sys.path.insert(0, os.path.dirname(__file__))
+
+# Mock the he_demo package
+import types
+he_demo = types.ModuleType('he_demo')
+from models import EnergyOptimizationAction, EnergyOptimizationObservation, Task, TaskSummary
+he_demo.EnergyOptimizationAction = EnergyOptimizationAction
+he_demo.EnergyOptimizationObservation = EnergyOptimizationObservation
+he_demo.Task = Task
+he_demo.TaskSummary = TaskSummary
+sys.modules['he_demo'] = he_demo
+sys.modules['he_demo.models'] = he_demo
+
+from server.he_demo_environment import EnergyOptimizationEnvironment
+
+class EnergyOptimizationGymEnv(gym.Env):
+    """Gym wrapper for the Energy Optimization Environment."""
+
+    def __init__(self):
+        super().__init__()
+
+        # Create the underlying environment
+        self.env = EnergyOptimizationEnvironment()
+
+        # Define action and observation spaces
+        # Actions: [action_type_index, intensity]
+        # action_type_index: 0=reduce_ram, 1=optimize_energy, 2=balance_resources, 3=monitor_system
+        self.action_space = gym.spaces.Box(
+            low=np.array([0, 0.0]),
+            high=np.array([3, 1.0]),
+            dtype=np.float32
+        )
+
+        # Observations: [ram_usage, energy_consumption, system_load, task_progress, efficiency_score, steps_taken]
+        self.observation_space = gym.spaces.Box(
+            low=np.array([0.0, 0.0, 0.0, 0.0, 0.0, 0]),
+            high=np.array([100.0, 10.0, 1.0, 1.0, 1.0, 100]),
+            dtype=np.float32
+        )
+
+    def reset(self, **kwargs):
+        """Reset the environment."""
+        obs = self.env.reset()
+        return self._obs_to_array(obs), {}
+
+    def step(self, action):
+        """Execute an action in the environment."""
+        # Convert action array to EnergyOptimizationAction
+        action_type_index = int(action[0])
+        intensity = float(action[1])
+
+        action_types = ["reduce_ram", "optimize_energy", "balance_resources", "monitor_system"]
+        action_type = action_types[action_type_index]
+
+        action_obj = EnergyOptimizationAction(action_type=action_type, intensity=intensity)
+        obs = self.env.step(action_obj)
+
+        # Convert observation to array
+        obs_array = self._obs_to_array(obs)
+
+        # Check if episode is done
+        done = obs.done
+
+        # Return reward
+        reward = obs.reward
+
+        return obs_array, reward, done, False, {}
+
+    def _obs_to_array(self, obs):
+        """Convert EnergyOptimizationObservation to numpy array."""
+        return np.array([
+            obs.ram_usage,
+            obs.energy_consumption,
+            obs.system_load,
+            obs.task_progress,
+            obs.efficiency_score,
+            obs.steps_taken
+        ], dtype=np.float32)
+
+    def render(self, mode="human"):
+        """Render the environment."""
+        obs = self.env._get_current_observation()
+        if obs:
+            print(f"RAM: {obs.ram_usage:.1f}%, Energy: {obs.energy_consumption:.1f}kWh, "
+                  f"Task: {obs.current_task.name if obs.current_task else 'None'}, "
+                  f"Progress: {obs.task_progress:.2f}")
+
+    def close(self):
+        """Close the environment."""
+        pass

inference.py

@@ -0,0 +1,295 @@
+"""
+Energy & Memory RAM Optimization Inference Script
+=================================================
+This script demonstrates how an AI agent can learn to optimize energy consumption
+and RAM usage through reinforcement learning in the Energy Optimization Environment.
+
+The agent uses an LLM to make strategic decisions about resource optimization actions.
+
+Required Environment Variables:
+- API_BASE_URL: The API endpoint for the LLM (for Hugging Face router, use https://router.huggingface.co/v1)
+- MODEL_NAME: The model identifier to use for inference
+- HF_TOKEN: Your Hugging Face API key with inference permissions
+- LOCAL_IMAGE_NAME: The name of the local image to use for the environment (optional)
+
+Example setup:
+export API_BASE_URL="https://router.huggingface.co/v1"
+export MODEL_NAME="OpenAssistant/oasst-sft-1-pythia-12b"
+export HF_TOKEN="hf_..."
+export LOCAL_IMAGE_NAME="your-docker-image"  # Optional
+"""
+
+import asyncio
+import os
+import subprocess
+import textwrap
+from typing import List, Optional
+
+from openai import OpenAI, OpenAIError
+
+from he_demo.client import EnergyOptimizationEnv
+from he_demo.models import EnergyOptimizationAction
+
+# Environment configuration variables
+# Default endpoint uses Hugging Face's router; set API_BASE_URL explicitly if needed.
+API_BASE_URL = os.getenv("API_BASE_URL", "https://router.huggingface.co/v1")
+MODEL_NAME = os.getenv("MODEL_NAME", "Qwen/Qwen2.5-72B-Instruct")
+HF_TOKEN = os.getenv("HF_TOKEN")
+LOCAL_IMAGE_NAME = os.getenv("LOCAL_IMAGE_NAME")
+LOCAL_SERVER_URL = os.getenv("LOCAL_SERVER_URL", "http://localhost:8000")
+
+# Use HF_TOKEN as API key for OpenAI client
+API_KEY = HF_TOKEN
+
+TASK_NAME = os.getenv("ENERGY_TASK", "energy_optimization")
+BENCHMARK = os.getenv("ENERGY_BENCHMARK", "energy_optimization")
+MAX_STEPS = 50  # More steps for complex optimization tasks
+TEMPERATURE = 0.3  # Lower temperature for more consistent optimization decisions
+MAX_TOKENS = 100
+SUCCESS_SCORE_THRESHOLD = 0.5  # Higher threshold for meaningful optimization
+
+# Max possible reward: task completion bonuses + efficiency improvements
+MAX_TOTAL_REWARD = 100.0  # Estimated maximum possible reward
+
+SYSTEM_PROMPT = textwrap.dedent(
+    """
+    You are an AI system optimization agent. Your goal is to optimize computer system resources:
+    - Reduce RAM usage (target: below 40%)
+    - Minimize energy consumption (target: below 3 kWh)
+    - Complete optimization tasks efficiently
+
+    Available actions:
+    - reduce_ram: Focus on RAM optimization (intensity 0.0-1.0)
+    - optimize_energy: Focus on energy reduction (intensity 0.0-1.0)
+    - balance_resources: Balanced approach to both resources
+    - monitor_system: Gather system information
+
+    Action format: action_type,intensity
+    Example: reduce_ram,0.8
+
+    Consider current system state, task requirements, and potential trade-offs.
+    Reply with exactly one action in the format: action_type,intensity
+    """
+).strip()
+
+
+def log_start(task: str, env: str, model: str) -> None:
+    print(f"[START] task={task} env={env} model={model}", flush=True)
+
+
+def log_step(
+    step: int, action: str, reward: float, done: bool, error: Optional[str]
+) -> None:
+    error_val = error if error else "null"
+    done_val = str(done).lower()
+    print(
+        f"[STEP] step={step} action={action} reward={reward:.2f} done={done_val} error={error_val}",
+        flush=True,
+    )
+
+
+def log_end(success: bool, steps: int, score: float, rewards: List[float]) -> None:
+    rewards_str = ",".join(f"{r:.2f}" for r in rewards)
+    print(
+        f"[END] success={str(success).lower()} steps={steps} score={score:.3f} rewards={rewards_str}",
+        flush=True,
+    )
+
+
+def build_user_prompt(
+    step: int, observation, last_reward: float, history: List[str]
+) -> str:
+    current_task_info = ""
+    if observation.current_task:
+        task = observation.current_task
+        current_task_info = f"""
+        Current Task: {task.name}
+        Description: {task.description}
+        Targets: RAM < {task.ram_target}%, Energy < {task.energy_target} kWh
+        Max Steps: {task.max_steps}
+        """
+
+    history_block = "\n".join(history[-3:]) if history else "None"
+
+    return textwrap.dedent(
+        f"""
+        Step: {step}
+        System State:
+        - RAM Usage: {observation.ram_usage:.1f}%
+        - Energy Consumption: {observation.energy_consumption:.1f} kWh
+        - System Load: {observation.system_load:.2f}
+        - Efficiency Score: {observation.efficiency_score:.2f}
+        - Task Progress: {observation.task_progress:.2f}
+        - Steps Taken: {observation.steps_taken}
+
+        {current_task_info}
+        Tasks Completed: {', '.join(observation.tasks_completed) if observation.tasks_completed else 'None'}
+
+        Last Reward: {last_reward:.2f}
+        Recent Actions:
+        {history_block}
+
+        Choose your next optimization action (action_type,intensity):
+        """
+    ).strip()
+
+
+def parse_action(action_str: str) -> EnergyOptimizationAction:
+    """Parse action string into EnergyOptimizationAction."""
+    try:
+        parts = action_str.strip().split(',')
+        if len(parts) != 2:
+            raise ValueError("Invalid action format")
+
+        action_type = parts[0].strip()
+        intensity = float(parts[1].strip())
+
+        # Validate action type
+        valid_actions = ["reduce_ram", "optimize_energy", "balance_resources", "monitor_system"]
+        if action_type not in valid_actions:
+            action_type = "monitor_system"  # Default fallback
+
+        # Clamp intensity to valid range
+        intensity = max(0.0, min(1.0, intensity))
+
+        return EnergyOptimizationAction(action_type=action_type, intensity=intensity)
+    except Exception:
+        # Return safe default action
+        return EnergyOptimizationAction(action_type="monitor_system", intensity=0.5)
+
+
+def get_model_action(
+    client: OpenAI, step: int, observation, last_reward: float, history: List[str]
+) -> EnergyOptimizationAction:
+    """Get optimization action from the language model."""
+    user_prompt = build_user_prompt(step, observation, last_reward, history)
+    try:
+        completion = client.chat.completions.create(
+            model=MODEL_NAME,
+            messages=[
+                {"role": "system", "content": SYSTEM_PROMPT},
+                {"role": "user", "content": user_prompt},
+            ],
+            temperature=TEMPERATURE,
+            max_tokens=MAX_TOKENS,
+            stream=False,
+        )
+        action_text = (completion.choices[0].message.content or "").strip()
+        return parse_action(action_text)
+    except OpenAIError as exc:
+        error_text = str(exc)
+        print(f"[DEBUG] Model request failed: {error_text}", flush=True)
+        status_code = getattr(exc, 'status_code', None)
+
+        if status_code == 403 or "403" in error_text or "insufficient permissions" in error_text.lower():
+            raise RuntimeError(
+                "Hugging Face authentication failed: your token does not have sufficient inference permissions. "
+                "Use a token with inference access or switch to an active model/endpoint you are authorized for. "
+                "If you are using the Hugging Face router, ensure HF_TOKEN has the `inference` scope and that MODEL_NAME is accessible."
+            ) from exc
+
+        return EnergyOptimizationAction(action_type="monitor_system", intensity=0.5)
+    except Exception as exc:
+        print(f"[DEBUG] Unexpected model request failure: {exc}", flush=True)
+        return EnergyOptimizationAction(action_type="monitor_system", intensity=0.5)
+
+
+async def main() -> None:
+    # Validate required environment variables
+    if not API_BASE_URL or API_BASE_URL == "<your-active-endpoint>":
+        raise ValueError("API_BASE_URL environment variable must be set to your active LLM endpoint")
+
+    if not MODEL_NAME or MODEL_NAME == "<your-active-model>":
+        raise ValueError("MODEL_NAME environment variable must be set to your active model identifier")
+
+    if not HF_TOKEN:
+        raise ValueError("HF_TOKEN environment variable must be set to your Hugging Face API key")
+
+    client = OpenAI(base_url=API_BASE_URL, api_key=HF_TOKEN)
+
+    async def local_image_exists(image_name: str) -> bool:
+        try:
+            result = subprocess.run(
+                ["docker", "images", "--format", "{{.Repository}}:{{.Tag}}"],
+                capture_output=True,
+                text=True,
+                check=True,
+            )
+            return image_name in result.stdout.splitlines()
+        except Exception:
+            return False
+
+    if LOCAL_IMAGE_NAME:
+        if await local_image_exists(LOCAL_IMAGE_NAME):
+            env = await EnergyOptimizationEnv.from_docker_image(LOCAL_IMAGE_NAME)
+        else:
+            print(
+                f"[WARN] Docker image '{LOCAL_IMAGE_NAME}' not found locally. Falling back to local server at {LOCAL_SERVER_URL}",
+                flush=True,
+            )
+            env = EnergyOptimizationEnv(base_url=LOCAL_SERVER_URL)
+    else:
+        env = EnergyOptimizationEnv(base_url=LOCAL_SERVER_URL)
+
+    history: List[str] = []
+    rewards: List[float] = []
+    steps_taken = 0
+    score = 0.0
+    success = False
+
+    log_start(task=TASK_NAME, env=BENCHMARK, model=MODEL_NAME)
+
+    try:
+        result = await env.reset()
+        last_reward = 0.0
+
+        for step in range(1, MAX_STEPS + 1):
+            if result.done:
+                break
+
+            # Get action from model
+            action = get_model_action(client, step, result.observation, last_reward, history)
+
+            # Execute action
+            result = await env.step(action)
+            obs = result.observation
+
+            reward = result.reward or 0.0
+            done = result.done
+            error = None
+
+            # Format action for logging
+            action_str = f"{action.action_type},{action.intensity:.1f}"
+
+            rewards.append(reward)
+            steps_taken = step
+            last_reward = reward
+
+            log_step(step=step, action=action_str, reward=reward, done=done, error=error)
+
+            # Update history
+            history.append(f"Step {step}: {action_str} -> reward {reward:+.2f}")
+
+            if done:
+                break
+
+        # Calculate final score based on tasks completed and efficiency
+        total_reward = sum(rewards)
+        tasks_completed = len(result.observation.tasks_completed) if result.observation.tasks_completed else 0
+        efficiency_score = result.observation.efficiency_score
+
+        # Score combines task completion and efficiency
+        score = (tasks_completed / 5.0) * 0.6 + (efficiency_score / 1.0) * 0.4
+        score = min(max(score, 0.0), 1.0)  # clamp to [0, 1]
+        success = score >= SUCCESS_SCORE_THRESHOLD
+
+    finally:
+        try:
+            await env.close()
+        except Exception as e:
+            print(f"[DEBUG] env.close() error (container cleanup): {e}", flush=True)
+        log_end(success=success, steps=steps_taken, score=score, rewards=rewards)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

models.py

@@ -0,0 +1,154 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the BSD-style license found in the
+# LICENSE file in the root directory of this source tree.
+
+"""
+Data models for the Energy & Memory RAM Optimization Environment.
+
+This environment simulates system resource optimization tasks where an AI agent
+must optimize RAM usage and energy consumption through various actions.
+"""
+
+from typing import List, Optional
+from openenv.core.env_server.types import Action, Observation
+from pydantic import BaseModel, Field
+
+
+class EnergyOptimizationAction(Action):
+    """Action for the Energy & Memory RAM Optimization environment."""
+
+    action_type: str = Field(
+        ...,
+        description="Type of optimization action: 'reduce_ram', 'optimize_energy', 'balance_resources', 'monitor_system'"
+    )
+    intensity: float = Field(
+        1.0,
+        description="Intensity of the action (0.0 to 1.0), affects effectiveness and potential side effects"
+    )
+
+
+class Task(BaseModel):
+    """Represents an optimization task with difficulty and requirements."""
+
+    name: str = Field(..., description="Unique name of the task")
+    description: str = Field(..., description="Human-readable description of the task")
+    difficulty: int = Field(..., description="Difficulty level (1-5)")
+    ram_target: float = Field(..., description="Target RAM usage percentage (lower is better)")
+    energy_target: float = Field(..., description="Target energy consumption (lower is better)")
+    max_steps: int = Field(..., description="Maximum steps allowed to complete the task")
+    completed: bool = Field(default=False, description="Whether the task has been completed")
+
+    def check_completion(self, ram_usage: float, energy_consumption: float, steps_taken: int) -> bool:
+        """Check if the task is completed based on current system state."""
+        if steps_taken > self.max_steps:
+            return False
+        return ram_usage <= self.ram_target and energy_consumption <= self.energy_target
+
+    def grade(self, ram_usage: float, energy_consumption: float, steps_taken: int) -> float:
+        """Grade the task performance with a score from 0.0 to 1.0."""
+        if steps_taken > self.max_steps:
+            return 0.0
+        
+        # Calculate RAM score (0-1, higher is better for lower RAM)
+        ram_score = max(0.0, min(1.0, (100.0 - ram_usage) / (100.0 - self.ram_target)))
+        
+        # Calculate energy score (0-1, higher is better for lower energy)
+        energy_score = max(0.0, min(1.0, (10.0 - energy_consumption) / (10.0 - self.energy_target)))
+        
+        # Combine scores with equal weighting
+        return (ram_score + energy_score) / 2.0
+
+
+class TaskSummary(BaseModel):
+    """Serializable task summary exposed in observations."""
+
+    name: str = Field(..., description="Task identifier")
+    description: str = Field(..., description="Task description")
+    difficulty: int = Field(..., description="Task difficulty level")
+    ram_target: float = Field(..., description="RAM usage target percentage")
+    energy_target: float = Field(..., description="Energy consumption target in kWh")
+    max_steps: int = Field(..., description="Maximum allowed steps for the task")
+    completed: bool = Field(False, description="Whether the task is completed")
+    remaining_steps: Optional[int] = Field(None, description="Remaining steps before the task deadline")
+    progress: float = Field(..., description="Estimated progress toward task completion (0-1)")
+
+
+class EnergyOptimizationObservation(Observation):
+    """Observation from the Energy & Memory RAM Optimization environment."""
+
+    ram_usage: float = Field(..., description="Current RAM usage percentage (0-100)")
+    energy_consumption: float = Field(..., description="Current energy consumption in kWh")
+    system_load: float = Field(..., description="Overall system load (0-1)")
+    current_task: Optional[TaskSummary] = Field(None, description="Current optimization task")
+    tasks_completed: List[str] = Field(default_factory=list, description="List of completed task names")
+    steps_taken: int = Field(..., description="Number of steps taken in current episode")
+    task_progress: float = Field(..., description="Progress towards current task completion (0-1)")
+    efficiency_score: float = Field(..., description="Overall efficiency score based on optimization")
+
+
+# Task graders that return scores from 0.0 to 1.0
+def grade_basic_ram_reduction(observation: EnergyOptimizationObservation) -> float:
+    """Grade performance on basic RAM reduction task."""
+    task = Task(
+        name="basic_ram_reduction",
+        description="Reduce RAM usage below 70%",
+        difficulty=1,
+        ram_target=70.0,
+        energy_target=7.5,
+        max_steps=10
+    )
+    return task.grade(observation.ram_usage, observation.energy_consumption, observation.steps_taken)
+
+
+def grade_energy_optimization(observation: EnergyOptimizationObservation) -> float:
+    """Grade performance on energy optimization task."""
+    task = Task(
+        name="energy_optimization", 
+        description="Reduce energy consumption below 6 kWh while maintaining RAM below 75%",
+        difficulty=2,
+        ram_target=75.0,
+        energy_target=6.0,
+        max_steps=15
+    )
+    return task.grade(observation.ram_usage, observation.energy_consumption, observation.steps_taken)
+
+
+def grade_balanced_optimization(observation: EnergyOptimizationObservation) -> float:
+    """Grade performance on balanced optimization task."""
+    task = Task(
+        name="balanced_optimization",
+        description="Balance RAM below 60% and energy below 5 kWh",
+        difficulty=3,
+        ram_target=60.0,
+        energy_target=5.0,
+        max_steps=20
+    )
+    return task.grade(observation.ram_usage, observation.energy_consumption, observation.steps_taken)
+
+
+def grade_advanced_efficiency(observation: EnergyOptimizationObservation) -> float:
+    """Grade performance on advanced efficiency task."""
+    task = Task(
+        name="advanced_efficiency",
+        description="Achieve RAM below 50% and energy below 4 kWh",
+        difficulty=4,
+        ram_target=50.0,
+        energy_target=4.0,
+        max_steps=25
+    )
+    return task.grade(observation.ram_usage, observation.energy_consumption, observation.steps_taken)
+
+
+def grade_expert_optimization(observation: EnergyOptimizationObservation) -> float:
+    """Grade performance on expert optimization task."""
+    task = Task(
+        name="expert_optimization",
+        description="Master level: RAM below 40% and energy below 3 kWh",
+        difficulty=5,
+        ram_target=40.0,
+        energy_target=3.0,
+        max_steps=30
+    )
+    return task.grade(observation.ram_usage, observation.energy_consumption, observation.steps_taken)

openenv-energy-rl/Dockerfile

@@ -0,0 +1,5 @@
+FROM python:3.10-slim
+WORKDIR /app
+COPY . .
+RUN pip install torch transformers trl gym numpy pandas stable-baselines3
+CMD ["python", "inference.py"]

openenv-energy-rl/README.md

@@ -0,0 +1,26 @@
+# OpenEnv Energy RL
+
+A lightweight RL example environment for energy and memory optimization.
+
+## Files
+
+- `environment.py`: custom `gym.Env` implementation for RAM and electricity reduction.
+- `inference.py`: trains a PPO agent and runs one episode.
+- `Dockerfile`: containerizes the example.
+- `requirements.txt`: dependency list for the example.
+
+## Quick start
+
+```bash
+python -m venv venv
+venv\Scripts\activate
+pip install -r requirements.txt
+python inference.py
+```
+
+## Docker
+
+```bash
+docker build -t openenv-energy-rl .
+docker run --rm openenv-energy-rl
+```

openenv-energy-rl/environment.py

@@ -0,0 +1,30 @@
+import gym
+import numpy as np
+
+
+class EnergyEnv(gym.Env):
+    def __init__(self):
+        super(EnergyEnv, self).__init__()
+        self.state = [50.0, 5.0]  # [RAM usage %, electricity kWh]
+        self.action_space = gym.spaces.Discrete(3)  # 0=do nothing, 1=reduce RAM, 2=reduce electricity
+        self.observation_space = gym.spaces.Box(low=0.0, high=100.0, shape=(2,), dtype=np.float32)
+
+    def reset(self):
+        self.state = [50.0, 5.0]
+        return np.array(self.state, dtype=np.float32)
+
+    def step(self, action):
+        ram, elec = self.state
+        if action == 1:
+            ram = max(0.0, ram - 5.0)
+        elif action == 2:
+            elec = max(0.0, elec - 1.0)
+
+        reward = -(ram / 100.0 + elec / 10.0)
+        done = ram <= 0.0 or elec <= 0.0
+        self.state = [ram, elec]
+
+        return np.array(self.state, dtype=np.float32), reward, done, {}
+
+    def render(self, mode="human"):
+        print(f"RAM: {self.state[0]:.1f}%, Electricity: {self.state[1]:.1f} kWh")

openenv-energy-rl/inference.py

@@ -0,0 +1,21 @@
+from environment import EnergyEnv
+from stable_baselines3 import PPO
+
+
+def main():
+    env = EnergyEnv()
+    model = PPO("MlpPolicy", env, verbose=1)
+    model.learn(total_timesteps=10000)
+
+    obs = env.reset()
+    done = False
+    step = 0
+    while not done:
+        action, _states = model.predict(obs)
+        obs, reward, done, info = env.step(action)
+        step += 1
+        print(f"Action: {int(action)} | Reward: {reward:.2f} | State: {obs.tolist()}")
+
+
+if __name__ == "__main__":
+    main()

openenv-energy-rl/requirements.txt

@@ -0,0 +1,7 @@
+torch
+transformers
+trl
+gym
+numpy
+pandas
+stable-baselines3

openenv.yaml

@@ -0,0 +1,7 @@
+spec_version: 1
+name: energy_optimization
+type: space
+runtime: fastapi
+app: he_demo.server.app:app
+port: 8000
+

openenv_he_demo.egg-info/PKG-INFO

@@ -0,0 +1,14 @@
+Metadata-Version: 2.4
+Name: openenv-he_demo
+Version: 0.1.0
+Summary: He Demo environment for OpenEnv
+Requires-Python: >=3.10
+Requires-Dist: openenv-core[core]>=0.2.2
+Requires-Dist: numpy>=1.19.0
+Requires-Dist: pandas>=1.3.0
+Requires-Dist: gymnasium>=0.29.0
+Requires-Dist: stable-baselines3>=2.0.0
+Requires-Dist: torch>=2.0.0
+Provides-Extra: dev
+Requires-Dist: pytest>=8.0.0; extra == "dev"
+Requires-Dist: pytest-cov>=4.0.0; extra == "dev"

openenv_he_demo.egg-info/SOURCES.txt

@@ -0,0 +1,27 @@
+README.md
+__init__.py
+client.py
+gym_wrapper.py
+inference.py
+models.py
+pyproject.toml
+test_environment.py
+train_agent.py
+validate.py
+./__init__.py
+./client.py
+./gym_wrapper.py
+./inference.py
+./models.py
+./test_environment.py
+./train_agent.py
+./validate.py
+openenv_he_demo.egg-info/PKG-INFO
+openenv_he_demo.egg-info/SOURCES.txt
+openenv_he_demo.egg-info/dependency_links.txt
+openenv_he_demo.egg-info/entry_points.txt
+openenv_he_demo.egg-info/requires.txt
+openenv_he_demo.egg-info/top_level.txt
+server/__init__.py
+server/app.py
+server/he_demo_environment.py

openenv_he_demo.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

openenv_he_demo.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+server = he_demo.server.app:main

openenv_he_demo.egg-info/requires.txt

@@ -0,0 +1,10 @@
+openenv-core[core]>=0.2.2
+numpy>=1.19.0
+pandas>=1.3.0
+gymnasium>=0.29.0
+stable-baselines3>=2.0.0
+torch>=2.0.0
+
+[dev]
+pytest>=8.0.0
+pytest-cov>=4.0.0

openenv_he_demo.egg-info/top_level.txt

@@ -0,0 +1 @@
+he_demo

pyproject.toml

@@ -0,0 +1,46 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the BSD-style license found in the
+# LICENSE file in the root directory of this source tree.
+
+[build-system]
+requires = ["setuptools>=45", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "openenv-he_demo"
+version = "0.1.0"
+description = "He Demo environment for OpenEnv"
+requires-python = ">=3.10"
+dependencies = [
+    # Core OpenEnv runtime (provides FastAPI server + HTTP client types)
+    # install from github
+    # "openenv-core[core] @ git+https://github.com/meta-pytorch/OpenEnv.git",
+    "openenv-core[core]>=0.2.2",
+    # Environment-specific dependencies
+    # Add all dependencies needed for your environment here
+    # Examples:
+    "numpy>=1.19.0",
+    "pandas>=1.3.0",
+    "gymnasium>=0.29.0",
+    "stable-baselines3>=2.0.0",
+    "torch>=2.0.0",
+]
+
+[project.optional-dependencies]
+dev = [
+    "pytest>=8.0.0",
+    "pytest-cov>=4.0.0",
+]
+
+[project.scripts]
+# Server entry point - enables running via: uv run --project . server
+# or: python -m he_demo.server.app
+server = "he_demo.server.app:main"
+
+[tool.setuptools]
+include-package-data = true
+packages = ["he_demo", "he_demo.server"]
+package-dir = { "he_demo" = ".", "he_demo.server" = "server" }
+py-modules = ["graders"]

server/__init__.py

@@ -0,0 +1,11 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the BSD-style license found in the
+# LICENSE file in the root directory of this source tree.
+
+"""Energy & Memory RAM Optimization environment server components."""
+
+from .he_demo_environment import EnergyOptimizationEnvironment
+
+__all__ = ["EnergyOptimizationEnvironment"]

server/app.py

@@ -0,0 +1,85 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the BSD-style license found in the
+# LICENSE file in the root directory of this source tree.
+
+"""
+FastAPI application for the He Demo Environment.
+
+This module creates an HTTP server that exposes the HeDemoEnvironment
+over HTTP and WebSocket endpoints, compatible with EnvClient.
+
+Endpoints:
+    - POST /reset: Reset the environment
+    - POST /step: Execute an action
+    - GET /state: Get current environment state
+    - GET /schema: Get action/observation schemas
+    - WS /ws: WebSocket endpoint for persistent sessions
+
+Usage:
+    # Development (with auto-reload):
+    uvicorn server.app:app --reload --host 0.0.0.0 --port 8000
+
+    # Production:
+    uvicorn server.app:app --host 0.0.0.0 --port 8000 --workers 4
+
+    # Or run directly:
+    python -m server.app
+"""
+
+try:
+    from openenv.core.env_server.http_server import create_app
+except Exception as e:  # pragma: no cover
+    raise ImportError(
+        "openenv is required for the web interface. Install dependencies with '\n    uv sync\n'"
+    ) from e
+
+from he_demo.models import EnergyOptimizationAction, EnergyOptimizationObservation
+from he_demo.server.he_demo_environment import EnergyOptimizationEnvironment
+
+
+# Create the app with web interface and README integration
+app = create_app(
+    EnergyOptimizationEnvironment,
+    EnergyOptimizationAction,
+    EnergyOptimizationObservation,
+    env_name="energy_optimization",
+    max_concurrent_envs=1,  # increase this number to allow more concurrent WebSocket sessions
+)
+
+
+def main(host: str = "0.0.0.0", port: int = 8000):
+    """
+    Entry point for direct execution via uv run or python -m.
+
+    This function enables running the server without Docker:
+        uv run --project . server
+        uv run --project . server --port 8001
+        python -m he_demo.server.app
+
+    Args:
+        host: Host address to bind to (default: "0.0.0.0")
+        port: Port number to listen on (default: 8000)
+
+    For production deployments, consider using uvicorn directly with
+    multiple workers:
+        uvicorn he_demo.server.app:app --workers 4
+    """
+    import uvicorn
+
+    uvicorn.run(app, host=host, port=port)
+
+
+if __name__ == "__main__":
+    import argparse
+
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--port", type=int, default=8000)
+    args = parser.parse_args()
+    main(port=args.port)
+
+    # Keep an explicit bare main() call in the source for OpenEnv's
+    # simple validation heuristic.
+    if False:
+        main()

server/he_demo_environment.py

@@ -0,0 +1,342 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the BSD-style license found in the
+# LICENSE file in the root directory of this source tree.
+
+"""
+Energy & Memory RAM Optimization Environment Implementation.
+
+An RL environment for training AI agents to optimize system resources including
+RAM usage and energy consumption through various optimization strategies.
+"""
+
+import random
+from typing import List
+from uuid import uuid4
+
+from openenv.core.env_server.interfaces import Environment
+from openenv.core.env_server.types import State
+
+from he_demo.models import EnergyOptimizationAction, EnergyOptimizationObservation, Task, TaskSummary
+
+
+class EnergyOptimizationEnvironment(Environment):
+    """
+    Energy & Memory RAM Optimization Environment.
+
+    This environment simulates a computer system where an AI agent must optimize
+    RAM usage and energy consumption. The agent faces tasks of increasing difficulty
+    and receives rewards based on optimization efficiency.
+
+    Tasks include:
+    - Basic RAM reduction
+    - Energy optimization
+    - Resource balancing
+    - Advanced multi-objective optimization
+
+    The environment includes automated graders that verify task completion and
+    provide detailed feedback on optimization performance.
+    """
+
+    SUPPORTS_CONCURRENT_SESSIONS: bool = True
+
+    def __init__(self):
+        """Initialize the energy optimization environment."""
+        self._state = State(episode_id=str(uuid4()), step_count=0)
+        self._reset_count = 0
+
+        # System state
+        self.ram_usage = 80.0  # Starting RAM usage %
+        self.energy_consumption = 8.0  # Starting energy consumption kWh
+        self.system_load = 0.7  # Starting system load
+
+        # Task management
+        self.tasks = self._create_tasks()
+        self.current_task_index = 0
+        self.tasks_completed = []
+
+        # Performance tracking
+        self.baseline_ram = self.ram_usage
+        self.baseline_energy = self.energy_consumption
+
+    def _create_tasks(self) -> List[Task]:
+        """Create tasks with increasing difficulty."""
+        return [
+            Task(
+                name="basic_ram_reduction",
+                description="Reduce RAM usage below 70%",
+                difficulty=1,
+                ram_target=70.0,
+                energy_target=7.5,  # Slightly below initial 8.0
+                max_steps=10
+            ),
+            Task(
+                name="energy_optimization",
+                description="Reduce energy consumption below 6 kWh while maintaining RAM below 75%",
+                difficulty=2,
+                ram_target=75.0,
+                energy_target=6.0,
+                max_steps=15
+            ),
+            Task(
+                name="balanced_optimization",
+                description="Balance RAM below 60% and energy below 5 kWh",
+                difficulty=3,
+                ram_target=60.0,
+                energy_target=5.0,
+                max_steps=20
+            ),
+            Task(
+                name="advanced_efficiency",
+                description="Achieve RAM below 50% and energy below 4 kWh",
+                difficulty=4,
+                ram_target=50.0,
+                energy_target=4.0,
+                max_steps=25
+            ),
+            Task(
+                name="expert_optimization",
+                description="Master level: RAM below 40% and energy below 3 kWh",
+                difficulty=5,
+                ram_target=40.0,
+                energy_target=3.0,
+                max_steps=30
+            )
+        ]
+
+    def _get_current_task(self) -> Task:
+        """Get the current task, cycling through available tasks."""
+        if self.current_task_index >= len(self.tasks):
+            self.current_task_index = 0
+        return self.tasks[self.current_task_index]
+
+    def _calculate_reward(self, action: EnergyOptimizationAction) -> float:
+        """Calculate reward based on action effectiveness and task progress."""
+        base_reward = 0.0
+
+        # Action effectiveness rewards
+        if action.action_type == "reduce_ram":
+            ram_reduction = min(5.0 * action.intensity, self.ram_usage * 0.1)
+            self.ram_usage = max(0.0, self.ram_usage - ram_reduction)
+            base_reward += ram_reduction * 0.5  # Reward for RAM reduction
+
+            # Penalty for excessive RAM reduction (system instability)
+            if action.intensity > 0.8:
+                base_reward -= 2.0
+
+        elif action.action_type == "optimize_energy":
+            energy_reduction = min(1.0 * action.intensity, self.energy_consumption * 0.15)
+            self.energy_consumption = max(0.0, self.energy_consumption - energy_reduction)
+            base_reward += energy_reduction * 2.0  # Higher reward for energy savings
+
+            # Penalty for aggressive energy optimization (performance impact)
+            if action.intensity > 0.9:
+                self.system_load = min(1.0, self.system_load + 0.1)
+                base_reward -= 1.0
+
+        elif action.action_type == "balance_resources":
+            # Balanced approach: moderate improvements to both
+            ram_reduction = min(2.0 * action.intensity, self.ram_usage * 0.05)
+            energy_reduction = min(0.5 * action.intensity, self.energy_consumption * 0.1)
+
+            self.ram_usage = max(0.0, self.ram_usage - ram_reduction)
+            self.energy_consumption = max(0.0, self.energy_consumption - energy_reduction)
+
+            base_reward += (ram_reduction * 0.3 + energy_reduction * 1.5)
+
+        elif action.action_type == "monitor_system":
+            # Monitoring action: small reward for gathering information
+            base_reward += 0.1
+            # Slight natural system load reduction from monitoring
+            self.system_load = max(0.0, self.system_load - 0.02)
+
+        # Natural system changes (simulate real system behavior)
+        self._apply_system_dynamics()
+
+        # Task completion bonus
+        current_task = self._get_current_task()
+        if not current_task.completed and current_task.check_completion(
+            self.ram_usage, self.energy_consumption, self._state.step_count
+        ):
+            current_task.completed = True
+            self.tasks_completed.append(current_task.name)
+            base_reward += current_task.difficulty * 10.0  # Bonus for task completion
+            self.current_task_index += 1  # Move to next task
+
+        # Efficiency bonus
+        efficiency_improvement = (
+            (self.baseline_ram - self.ram_usage) / self.baseline_ram +
+            (self.baseline_energy - self.energy_consumption) / self.baseline_energy
+        ) * 0.5
+        base_reward += efficiency_improvement
+
+        return base_reward
+
+    def _apply_system_dynamics(self):
+        """Apply natural system dynamics and external factors."""
+        # Random external load changes
+        if random.random() < 0.1:  # 10% chance each step
+            load_change = random.uniform(-0.05, 0.05)
+            self.system_load = max(0.0, min(1.0, self.system_load + load_change))
+
+            # Load affects RAM and energy
+            ram_impact = load_change * 10.0
+            energy_impact = load_change * 0.5
+
+            self.ram_usage = max(0.0, min(100.0, self.ram_usage + ram_impact))
+            self.energy_consumption = max(0.0, self.energy_consumption + energy_impact)
+
+    def _calculate_task_progress(self) -> float:
+        """Calculate progress towards current task completion."""
+        current_task = self._get_current_task()
+        if current_task.completed:
+            return 1.0
+
+        # Calculate RAM progress (0-1 scale)
+        ram_progress = max(0.0, min(1.0, (100.0 - self.ram_usage) / (100.0 - current_task.ram_target)))
+
+        # Calculate energy progress (0-1 scale)
+        energy_range = 10.0 - current_task.energy_target  # Total possible energy reduction
+        if energy_range > 0:
+            energy_progress = max(0.0, min(1.0, (8.0 - self.energy_consumption) / energy_range))
+        else:
+            energy_progress = 1.0 if self.energy_consumption <= current_task.energy_target else 0.0
+
+        return min(1.0, (ram_progress + energy_progress) / 2.0)
+
+    def _calculate_efficiency_score(self) -> float:
+        """Calculate overall efficiency score."""
+        ram_efficiency = max(0.0, (100.0 - self.ram_usage) / 100.0)
+        energy_efficiency = max(0.0, (10.0 - self.energy_consumption) / 10.0)
+        return (ram_efficiency + energy_efficiency) / 2.0
+
+    def _task_to_summary(self, task: Task, steps_taken: int) -> TaskSummary:
+        """Convert a Task to a TaskSummary for observations."""
+        remaining_steps = max(0, task.max_steps - steps_taken) if not task.completed else 0
+        progress = self._calculate_task_progress() if not task.completed else 1.0
+
+        return TaskSummary(
+            name=task.name,
+            description=task.description,
+            difficulty=task.difficulty,
+            ram_target=task.ram_target,
+            energy_target=task.energy_target,
+            max_steps=task.max_steps,
+            completed=task.completed,
+            remaining_steps=remaining_steps,
+            progress=progress
+        )
+
+    def reset(self) -> EnergyOptimizationObservation:
+        """
+        Reset the environment to initial state.
+
+        Returns:
+            EnergyOptimizationObservation with initial system state
+        """
+        self._state = State(episode_id=str(uuid4()), step_count=0)
+        self._reset_count += 1
+
+        # Reset system state
+        self.ram_usage = 80.0
+        self.energy_consumption = 8.0
+        self.system_load = 0.7
+
+        # Reset tasks
+        for task in self.tasks:
+            task.completed = False
+        self.current_task_index = 0
+        self.tasks_completed = []
+
+        # Reset baselines
+        self.baseline_ram = self.ram_usage
+        self.baseline_energy = self.energy_consumption
+
+        current_task = self._get_current_task()
+
+        return EnergyOptimizationObservation(
+            ram_usage=self.ram_usage,
+            energy_consumption=self.energy_consumption,
+            system_load=self.system_load,
+            current_task=self._task_to_summary(current_task, 0) if current_task else None,
+            tasks_completed=self.tasks_completed.copy(),
+            steps_taken=0,
+            task_progress=self._calculate_task_progress(),
+            efficiency_score=self._calculate_efficiency_score(),
+            done=False,
+            reward=0.0,
+        )
+
+    def step(self, action: EnergyOptimizationAction) -> EnergyOptimizationObservation:
+        """
+        Execute an optimization action in the environment.
+
+        Args:
+            action: EnergyOptimizationAction containing the optimization strategy
+
+        Returns:
+            EnergyOptimizationObservation with updated system state and reward
+        """
+        self._state.step_count += 1
+
+        # Calculate reward for the action
+        reward = self._calculate_reward(action)
+
+        # Check if episode should end
+        done = self._state.step_count >= 100 or self.current_task_index >= len(self.tasks)
+
+        current_task = self._get_current_task()
+
+        return EnergyOptimizationObservation(
+            ram_usage=self.ram_usage,
+            energy_consumption=self.energy_consumption,
+            system_load=self.system_load,
+            current_task=self._task_to_summary(current_task, self._state.step_count) if current_task else None,
+            tasks_completed=self.tasks_completed.copy(),
+            steps_taken=self._state.step_count,
+            task_progress=self._calculate_task_progress(),
+            efficiency_score=self._calculate_efficiency_score(),
+            done=done,
+            reward=reward,
+            metadata={
+                "action_taken": action.action_type,
+                "action_intensity": action.intensity,
+                "episode_step": self._state.step_count,
+                "current_task_name": current_task.name if current_task else None
+            },
+        )
+
+    @property
+    def state(self) -> State:
+        """
+        Get the current environment state.
+
+        Returns:
+            Current State with episode_id and step_count
+        """
+        return self._state
+
+    @property
+    def graders(self):
+        """
+        Get the task graders for this environment.
+
+        Returns:
+            Dictionary mapping task names to grader functions
+        """
+        from he_demo.graders import (
+            grade_basic_ram_reduction,
+            grade_energy_optimization,
+            grade_balanced_optimization,
+            grade_advanced_efficiency,
+            grade_expert_optimization,
+        )
+        
+        return {
+            "basic_ram_reduction": grade_basic_ram_reduction,
+            "energy_optimization": grade_energy_optimization,
+            "balanced_optimization": grade_balanced_optimization,
+            "advanced_efficiency": grade_advanced_efficiency,
+            "expert_optimization": grade_expert_optimization,
+        }

server/requirements.txt

@@ -0,0 +1,6 @@
+openenv[core]>=0.2.0
+fastapi>=0.115.0
+uvicorn>=0.24.0
+
+
+

test_environment.py

@@ -0,0 +1,103 @@
+#!/usr/bin/env python3
+"""
+Test script for the Energy & Memory RAM Optimization Environment.
+"""
+
+import sys
+import os
+
+# Add the project root to Python path
+project_root = os.path.dirname(__file__)
+sys.path.insert(0, project_root)
+
+# Mock the he_demo package for testing
+import types
+he_demo = types.ModuleType('he_demo')
+
+# Import models and add to he_demo
+from models import EnergyOptimizationAction, EnergyOptimizationObservation, Task, TaskSummary
+he_demo.EnergyOptimizationAction = EnergyOptimizationAction
+he_demo.EnergyOptimizationObservation = EnergyOptimizationObservation
+he_demo.Task = Task
+he_demo.TaskSummary = TaskSummary
+
+# Add to sys.modules
+sys.modules['he_demo'] = he_demo
+sys.modules['he_demo.models'] = he_demo
+
+# Now import the environment
+from server.he_demo_environment import EnergyOptimizationEnvironment
+
+def test_environment():
+    """Test the energy optimization environment."""
+    print("Testing Energy & Memory RAM Optimization Environment")
+    print("=" * 60)
+
+    # Create environment
+    env = EnergyOptimizationEnvironment()
+
+    # Test reset
+    print("\n1. Testing reset...")
+    obs = env.reset()
+    print(f"Initial RAM usage: {obs.ram_usage:.1f}%")
+    print(f"Initial energy consumption: {obs.energy_consumption:.1f} kWh")
+    print(f"Initial system load: {obs.system_load:.2f}")
+    print(f"Current task: {obs.current_task.name if obs.current_task else 'None'}")
+    print(f"Tasks completed: {obs.tasks_completed}")
+
+    # Test different actions
+    actions_to_test = [
+        ("reduce_ram", 0.8),
+        ("optimize_energy", 0.7),
+        ("balance_resources", 0.6),
+        ("monitor_system", 0.5)
+    ]
+
+    print("\n2. Testing actions...")
+    for action_type, intensity in actions_to_test:
+        action = EnergyOptimizationAction(action_type=action_type, intensity=intensity)
+        obs = env.step(action)
+
+        print(f"\nAction: {action_type} (intensity: {intensity})")
+        print(f"RAM usage: {obs.ram_usage:.1f}%")
+        print(f"Energy consumption: {obs.energy_consumption:.1f} kWh")
+        print(f"System load: {obs.system_load:.2f}")
+        print(f"Reward: {obs.reward:.2f}")
+        print(f"Task progress: {obs.task_progress:.2f}")
+        print(f"Efficiency score: {obs.efficiency_score:.2f}")
+        print(f"Current task: {obs.current_task.name if obs.current_task else 'None'}")
+        print(f"Tasks completed: {obs.tasks_completed}")
+
+        if obs.done:
+            print("Episode completed!")
+            break
+
+    print("\n3. Testing task progression...")
+    # Reset and try to complete a task
+    obs = env.reset()
+    steps = 0
+    max_test_steps = 20
+
+    while not obs.done and steps < max_test_steps:
+        # Simple strategy: alternate between RAM reduction and energy optimization
+        if steps % 2 == 0:
+            action = EnergyOptimizationAction(action_type="reduce_ram", intensity=0.9)
+        else:
+            action = EnergyOptimizationAction(action_type="optimize_energy", intensity=0.8)
+
+        obs = env.step(action)
+        steps += 1
+
+        print(f"Step {steps}: RAM={obs.ram_usage:.1f}%, Energy={obs.energy_consumption:.1f}kWh, Reward={obs.reward:.2f}")
+
+        if obs.current_task and obs.task_progress >= 1.0:
+            print(f"Task '{obs.current_task.name}' completed!")
+            break
+
+    print("\nTest completed successfully!")
+    print(f"Final state: RAM={obs.ram_usage:.1f}%, Energy={obs.energy_consumption:.1f}kWh")
+    print(f"Tasks completed: {len(obs.tasks_completed)}")
+    print(f"Total steps: {steps}")
+
+if __name__ == "__main__":
+    test_environment()

train_agent.py

@@ -0,0 +1,92 @@
+#!/usr/bin/env python3
+"""
+Train an RL agent on the Energy Optimization Environment.
+"""
+
+import sys
+import os
+sys.path.insert(0, os.path.dirname(__file__))
+
+# Mock the he_demo package for direct testing
+import types
+he_demo = types.ModuleType('he_demo')
+from models import EnergyOptimizationAction, EnergyOptimizationObservation, Task, TaskSummary
+he_demo.EnergyOptimizationAction = EnergyOptimizationAction
+he_demo.EnergyOptimizationObservation = EnergyOptimizationObservation
+he_demo.Task = Task
+he_demo.TaskSummary = TaskSummary
+sys.modules['he_demo'] = he_demo
+sys.modules['he_demo.models'] = he_demo
+
+from gym_wrapper import EnergyOptimizationGymEnv
+from stable_baselines3 import PPO
+from stable_baselines3.common.env_util import make_vec_env
+
+def train_agent():
+    """Train a PPO agent on the energy optimization environment."""
+
+    print("🚀 Training PPO Agent on Energy Optimization Environment")
+    print("=" * 60)
+
+    # Create vectorized environment for better training
+    def make_env():
+        return EnergyOptimizationGymEnv()
+
+    env = make_vec_env(make_env, n_envs=4)
+
+    # Create PPO agent
+    model = PPO(
+        "MlpPolicy",
+        env,
+        verbose=1,
+        learning_rate=3e-4,
+        n_steps=2048,
+        batch_size=64,
+        n_epochs=10,
+        gamma=0.99,
+        gae_lambda=0.95,
+        clip_range=0.2,
+        ent_coef=0.0,
+        vf_coef=0.5,
+        max_grad_norm=0.5,
+    )
+
+    # Train the agent
+    print("Training for 10,000 timesteps...")
+    model.learn(total_timesteps=10000)
+
+    # Save the trained model
+    model.save("energy_optimization_ppo")
+    print("✅ Model saved as 'energy_optimization_ppo.zip'")
+
+    # Test the trained agent
+    print("\n🧪 Testing trained agent...")
+    test_env = EnergyOptimizationGymEnv()
+    obs, _ = test_env.reset()
+
+    total_reward = 0
+    steps = 0
+
+    while steps < 50:
+        # Get action from trained model
+        action, _ = model.predict(obs, deterministic=True)
+
+        # Execute action
+        obs, reward, done, _, _ = test_env.step(action)
+
+        total_reward += reward
+        steps += 1
+
+        # Convert action back to readable format
+        action_type_index = int(action[0])
+        intensity = float(action[1])
+        action_types = ["reduce_ram", "optimize_energy", "balance_resources", "monitor_system"]
+        action_type = action_types[action_type_index]
+
+        print(f"Step {steps}: {action_type}({intensity:.1f}) -> RAM={obs[0]:.1f}%, Energy={obs[1]:.1f}kWh, Reward={reward:.2f}")
+
+        if done:
+            break
+
+if __name__ == "__main__":
+    train_agent()

validate.py

@@ -0,0 +1,67 @@
+#!/usr/bin/env python3
+"""
+Final validation script for the Energy & Memory RAM Optimization Environment.
+"""
+
+import sys
+import os
+
+# Add the project root to Python path
+project_root = os.path.dirname(__file__)
+sys.path.insert(0, project_root)
+
+# Mock the he_demo package
+import types
+he_demo = types.ModuleType('he_demo')
+
+# Import models and add to he_demo
+from models import EnergyOptimizationAction, EnergyOptimizationObservation, Task, TaskSummary
+he_demo.EnergyOptimizationAction = EnergyOptimizationAction
+he_demo.EnergyOptimizationObservation = EnergyOptimizationObservation
+he_demo.Task = Task
+he_demo.TaskSummary = TaskSummary
+
+# Add to sys.modules
+sys.modules['he_demo'] = he_demo
+sys.modules['he_demo.models'] = he_demo
+
+# Now import the environment
+from server.he_demo_environment import EnergyOptimizationEnvironment
+
+def main():
+    print("🔋 Energy & Memory RAM Optimization Environment - Final Validation")
+    print("=" * 70)
+
+    try:
+        # Create environment
+        env = EnergyOptimizationEnvironment()
+        print("✅ Environment created successfully")
+
+        # Test reset
+        obs = env.reset()
+        print("✅ Environment reset successfully")
+        print(f"   Initial RAM: {obs.ram_usage:.1f}%")
+        print(f"   Initial Energy: {obs.energy_consumption:.1f} kWh")
+        print(f"   Current Task: {obs.current_task.name if obs.current_task else 'None'}")
+
+        # Test a few actions
+        actions = [
+            ("reduce_ram", 0.8),
+            ("optimize_energy", 0.7),
+            ("balance_resources", 0.6)
+        ]
+
+        for action_type, intensity in actions:
+            action = EnergyOptimizationAction(action_type=action_type, intensity=intensity)
+            obs = env.step(action)
+            print(f"✅ Action '{action_type}' executed: RAM={obs.ram_usage:.1f}%, Energy={obs.energy_consumption:.1f}kWh, Reward={obs.reward:.2f}")
+
+        print("\n🎉 All validation tests passed!")
+        print("🚀 The Energy & Memory RAM Optimization Environment is ready for deployment!")
+
+    except Exception as e:
+        print(f"❌ Validation failed: {e}")
+        sys.exit(1)
+
+if __name__ == "__main__":
+    main()