validate.py

#!/usr/bin/env python
"""Validate OpenEnv project setup."""

import sys
import yaml
from pathlib import Path

def validate_project():
    """Validate the OpenEnv RL project setup."""
    
    print("[VALIDATION] Starting OpenEnv project validation...\n")
    
    errors = []
    warnings = []
    
    # Check 1: openenv.yaml exists and is valid
    print("[CHECK] openenv.yaml configuration...")
    openenv_file = Path("openenv.yaml")
    if not openenv_file.exists():
        errors.append("Missing openenv.yaml")
    else:
        try:
            with open(openenv_file) as f:
                config = yaml.safe_load(f)
            print(f"  ✓ openenv.yaml found")
            print(f"  ✓ Environment: {config.get('name')}")
            print(f"  ✓ Tasks: {', '.join(config.get('tasks', {}).keys())}")
        except Exception as e:
            errors.append(f"Invalid YAML: {e}")
    
    # Check 2: Required Python files exist
    print("\n[CHECK] Required source files...")
    required_files = ["main.py", "inference.py", "env.py", "tasks.py", "graders.py"]
    for fname in required_files:
        if Path(fname).exists():
            print(f"  ✓ {fname}")
        else:
            errors.append(f"Missing {fname}")
    
    # Check 3: FastAPI endpoints
    print("\n[CHECK] FastAPI endpoints...")
    try:
        from main import app
        routes = [route.path for route in app.routes]
        required_endpoints = ["/reset", "/step", "/state"]
        for endpoint in required_endpoints:
            if endpoint in routes:
                print(f"  ✓ {endpoint}")
            else:
                errors.append(f"Missing endpoint: {endpoint}")
    except Exception as e:
        warnings.append(f"Could not import main.py: {e}")
    
    # Check 4: Environment class
    print("\n[CHECK] Hospital environment...")
    try:
        from env import HospitalEnv
        env = HospitalEnv()
        state = env.reset()
        print(f"  ✓ HospitalEnv initialized")
        print(f"  ✓ Initial state: {state}")
    except Exception as e:
        errors.append(f"HospitalEnv error: {e}")
    
    # Check 5: Requirements.txt
    print("\n[CHECK] Dependencies...")
    required_packages = ["fastapi", "openai", "python-dotenv", "openenv-core"]
    with open("requirements.txt") as f:
        requirements = f.read()
    for pkg in required_packages:
        if pkg in requirements:
            print(f"  ✓ {pkg}")
        else:
            warnings.append(f"Package {pkg} not in requirements.txt")
    
    # Check 6: Docker setup
    print("\n[CHECK] Docker configuration...")
    if Path("Dockerfile").exists():
        print(f"  ✓ Dockerfile")
    else:
        errors.append("Missing Dockerfile")
    
    if Path(".dockerignore").exists():
        print(f"  ✓ .dockerignore")
    else:
        warnings.append("Missing .dockerignore")
    
    # Summary
    print("\n" + "="*50)
    if errors:
        print(f"❌ VALIDATION FAILED ({len(errors)} error(s))")
        for err in errors:
            print(f"   - {err}")
        return False
    else:
        print(f"✅ VALIDATION PASSED")
        if warnings:
            print(f"⚠️  Warnings ({len(warnings)}):")
            for warn in warnings:
                print(f"   - {warn}")
        return True

if __name__ == "__main__":
    success = validate_project()
    sys.exit(0 if success else 1)