config.py

import os
from pathlib import Path

class Config:
    """Centralized configuration for the agentic neural network."""

    # API Configuration
    ANTHROPIC_API_KEY = os.getenv("ANTHROPIC_API_KEY")
    CLAUDE_MODEL = "claude-3-7-sonnet-20250219"
    MAX_TOKENS = 2048

    # Paths
    BASE_DIR = Path(__file__).parent
    PROMPTS_DIR = BASE_DIR / "prompts"
    SOLUTION_DIR = BASE_DIR / "solution"
    TEMP_DIR = BASE_DIR / "temp_images"
    ARC_DATA_DIR = BASE_DIR / "ARC" / "data" / "training"

    # Task Configuration
    DEFAULT_TASK = "0a938d79.json"
    
    # Agent Configuration
    MAX_LOOPS = 3
    BEAM_WIDTH = 5
    MAX_SEARCH_DEPTH = 3
    
    # Image Configuration
    IMAGE_SIZE = (3, 3)
    ARC_COLORS = ['#000000', '#0074D9', '#FF4136', '#2ECC40', '#FFDC00',
                  '#AAAAAA', '#F012BE', '#FF851B', '#7FDBFF', '#870C25']
    
    # Logging Configuration
    LOG_LEVEL = "INFO"
    LOG_FORMAT = "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
    
    @classmethod
    def validate(cls):
        # Ensure directories exist
        cls.SOLUTION_DIR.mkdir(exist_ok=True)
        cls.TEMP_DIR.mkdir(exist_ok=True)
        
        return True