feat: add explicit task registry and validation endpoints

- Created task_registry.py with explicit TASK_REGISTRY mapping tasks to graders
- Added /tasks, /tasks/{task_name}, and /validate endpoints to FastAPI server
- /validate endpoint now reports 5 tasks with graders (requirement: >=3)
- Ensures maximum visibility and discoverability of grader assignments
- Fixes validator 'Not enough tasks with graders' issue

__init__.py

@@ -19,6 +19,13 @@ from .task_graders import (
     task_4_advanced_efficiency_grader,
     task_5_expert_optimization_grader,
 )
+from .task_registry import (
+    TASK_REGISTRY,
+    get_all_tasks_with_graders,
+    get_task_grader,
+    get_tasks_count,
+    is_grader_requirement_met,
+)
 
 __all__ = [
     "EnergyOptimizationAction",
@@ -34,4 +41,9 @@ __all__ = [
     "task_3_balanced_optimization_grader",
     "task_4_advanced_efficiency_grader",
     "task_5_expert_optimization_grader",
+    "TASK_REGISTRY",
+    "get_all_tasks_with_graders",
+    "get_task_grader",
+    "get_tasks_count",
+    "is_grader_requirement_met",
 ]

server/app.py

@@ -39,6 +39,7 @@ from fastapi import FastAPI
 from he_demo.models import EnergyOptimizationAction, EnergyOptimizationObservation
 from he_demo.server.he_demo_environment import EnergyOptimizationEnvironment
 from he_demo.task_graders import get_grader_metadata, TASK_GRADERS
+from he_demo.task_registry import get_all_tasks_with_graders, get_tasks_count, is_grader_requirement_met
 
 
 # Create the app with web interface and README integration
@@ -114,6 +115,68 @@ def graders_info():
     }
 
 
+# ============================================================================
+# TASK REGISTRY ENDPOINTS FOR VALIDATOR DETECTION
+# ============================================================================
+
+@app.get("/tasks")
+def get_tasks():
+    """
+    Get all available tasks with their associated graders.
+    
+    Returns:
+        Dictionary of tasks with grader assignments
+    """
+    return {
+        "tasks": get_all_tasks_with_graders(),
+        "total_tasks_with_graders": get_tasks_count(),
+        "requirement_met": is_grader_requirement_met(),
+        "minimum_required": 3
+    }
+
+
+@app.get("/tasks/{task_name}")
+def get_task_info(task_name: str):
+    """
+    Get information about a specific task and its grader.
+    
+    Args:
+        task_name: Name of the task
+        
+    Returns:
+        Task information with grader metadata
+    """
+    tasks = get_all_tasks_with_graders()
+    if task_name not in tasks:
+        return {"error": f"Task '{task_name}' not found"}
+    task_info = tasks[task_name].copy()
+    # Remove the grader function from response (not JSON serializable)
+    task_info.pop("grader", None)
+    return {
+        "task_name": task_name,
+        "task_info": task_info,
+        "grader_metadata": get_grader_metadata(task_name)
+    }
+
+
+@app.get("/validate")
+def validate_graders():
+    """
+    Validation endpoint for OpenEnv compliance checking.
+    
+    Returns:
+        Validation status indicating whether requirements are met
+    """
+    return {
+        "requirement": "At least 3 tasks with graders",
+        "total_tasks_with_graders": get_tasks_count(),
+        "validation_passed": is_grader_requirement_met(),
+        "tasks": list(get_all_tasks_with_graders().keys()),
+        "status": "PASS" if is_grader_requirement_met() else "FAIL",
+        "message": f"Environment has {get_tasks_count()} tasks with graders (minimum required: 3)"
+    }
+
+
 def main(host: str = "0.0.0.0", port: int = 8000):
     """
     Entry point for direct execution via uv run or python -m.

task_registry.py

@@ -0,0 +1,90 @@
+"""
+Task Registry with Explicit Grader Mappings
+
+This module provides a registry of all tasks with their corresponding graders
+for easy discovery and validation.
+"""
+
+from typing import Dict, Callable, Any
+from he_demo.task_graders import (
+    task_1_basic_ram_reduction_grader,
+    task_2_energy_optimization_grader,
+    task_3_balanced_optimization_grader,
+    task_4_advanced_efficiency_grader,
+    task_5_expert_optimization_grader,
+)
+
+# Explicit registry of tasks with graders
+TASK_REGISTRY: Dict[str, Dict[str, Any]] = {
+    "basic_ram_reduction": {
+        "task_name": "basic_ram_reduction",
+        "display_name": "Basic RAM Reduction",
+        "difficulty": 1,
+        "grader": task_1_basic_ram_reduction_grader,
+        "grader_name": "task_1_basic_ram_reduction_grader",
+        "description": "Reduce RAM usage below 70%"
+    },
+    "energy_optimization": {
+        "task_name": "energy_optimization",
+        "display_name": "Energy Optimization",
+        "difficulty": 2,
+        "grader": task_2_energy_optimization_grader,
+        "grader_name": "task_2_energy_optimization_grader",
+        "description": "Reduce energy consumption below 6 kWh while maintaining RAM below 75%"
+    },
+    "balanced_optimization": {
+        "task_name": "balanced_optimization",
+        "display_name": "Balanced Optimization",
+        "difficulty": 3,
+        "grader": task_3_balanced_optimization_grader,
+        "grader_name": "task_3_balanced_optimization_grader",
+        "description": "Balance RAM below 60% and energy below 5 kWh"
+    },
+    "advanced_efficiency": {
+        "task_name": "advanced_efficiency",
+        "display_name": "Advanced Efficiency",
+        "difficulty": 4,
+        "grader": task_4_advanced_efficiency_grader,
+        "grader_name": "task_4_advanced_efficiency_grader",
+        "description": "Achieve RAM below 50% and energy below 4 kWh"
+    },
+    "expert_optimization": {
+        "task_name": "expert_optimization",
+        "display_name": "Expert Optimization",
+        "difficulty": 5,
+        "grader": task_5_expert_optimization_grader,
+        "grader_name": "task_5_expert_optimization_grader",
+        "description": "Master level: RAM below 40% and energy below 3 kWh"
+    }
+}
+
+
+def get_all_tasks_with_graders() -> Dict[str, Dict[str, Any]]:
+    """Get all tasks with their associated graders."""
+    return TASK_REGISTRY
+
+
+def get_task_grader(task_name: str) -> Callable:
+    """Get the grader for a specific task."""
+    if task_name not in TASK_REGISTRY:
+        raise ValueError(f"Unknown task: {task_name}")
+    return TASK_REGISTRY[task_name]["grader"]
+
+
+def get_tasks_count() -> int:
+    """Get the total number of tasks with graders."""
+    return len(TASK_REGISTRY)
+
+
+def is_grader_requirement_met() -> bool:
+    """Check if minimum grader requirement (3 tasks) is met."""
+    return len(TASK_REGISTRY) >= 3
+
+
+if __name__ == "__main__":
+    print(f"Total tasks with graders: {get_tasks_count()}")
+    print(f"Requirement met (≥3): {is_grader_requirement_met()}")
+    print("\nTasks:")
+    for task_name, info in TASK_REGISTRY.items():
+        print(f"  - {info['display_name']} (Difficulty {info['difficulty']})")
+        print(f"    Grader: {info['grader_name']}")