main/utils/validation.py

from typing import Dict, Any
from pathlib import Path

def validate_model_path(model_path: Path) -> bool:
    """Validate that a model path exists and contains necessary files"""
    if not model_path.exists():
        return False
    required_files = ['config.json', 'pytorch_model.bin']
    return all((model_path / file).exists() for file in required_files)

def validate_generation_params(params: Dict[str, Any]) -> Dict[str, Any]:
    """Validate and normalize generation parameters"""
    validated = params.copy()

    # Ensure temperature is within bounds
    if 'temperature' in validated:
        validated['temperature'] = max(0.0, min(2.0, validated['temperature']))

    # Ensure max_new_tokens is reasonable
    if 'max_new_tokens' in validated:
        validated['max_new_tokens'] = max(1, min(4096, validated['max_new_tokens']))

    return validated