lollms_hub/app/schema/settings.py

# 📁 app/schema/settings.py
from pydantic import BaseModel, Field, field_validator, ConfigDict
from typing import List, Optional, Dict

class AppSettingsModel(BaseModel):
    # This prevents the Pydantic warning about "model_" prefixed fields.
    model_config = ConfigDict(protected_namespaces=())

    # --- BRANDING SETTINGS ---
    branding_title: str = "LoLLMs Hub"
    branding_logo_url: Optional[str] = Field(default=None, validate_default=True)

    # --- THEME SETTINGS ---
    ui_style: str = "dark-glass"  # 'dark-glass', 'dark-flat', 'light-glass', 'light-flat'
    selected_theme: str = "indigo"
    
    # Static property, not stored in DB, but available for the app
    @property
    def available_themes(self) -> Dict[str, Dict[str, str]]:
        return {
            "indigo": { "500": "#6366f1", "600": "#4f46e5", "700": "#4338ca", "800": "#3730a3" },
            "sky": { "500": "#0ea5e9", "600": "#0284c7", "700": "#0369a1", "800": "#075985" },
            "teal": { "500": "#14b8a6", "600": "#0d9488", "700": "#0f766e", "800": "#115e59" },
            "rose": { "500": "#f43f5e", "600": "#e11d48", "700": "#be123c", "800": "#9f1239" },
            "amber": { "500": "#f59e0b", "600": "#d97706", "700": "#b45309", "800": "#92400e" },
            "emerald": { "500": "#10b981", "600": "#059669", "700": "#047857", "800": "#065f46" },
            "fuchsia": { "500": "#d946ef", "600": "#c026d3", "700": "#a21caf", "800": "#86198f" },
            "orange": { "500": "#f97316", "600": "#ea580c", "700": "#c2410c", "800": "#9a3412" },
            "black": { "500": "#e5e7eb", "600": "#d1d5db", "700": "#9ca3af", "800": "#4b5563" },
            "white": { "500": "#4b5563", "600": "#374151", "700": "#1f2937", "800": "#111827" }
        }

    redis_host: str = "localhost"
    redis_port: int = 6379
    redis_username: Optional[str] = None
    redis_password: Optional[str] = None

    rate_limit_requests: int = 100
    rate_limit_window_minutes: int = 1

    allowed_ips: str = ""
    denied_ips: str = ""

    model_update_interval_minutes: int = 10

    # Model Storage
    default_models_path: str = Field(default="models", description="Default directory where HF models will be downloaded.")
    
    # Logging & Retention
    log_max_size_mb: int = 10
    log_backup_count: int = 5

    # Instance Management
    instance_scan_start_port: int = 11434
    instance_scan_end_port: int = 11445
    default_models_path: str = "models"

    # Administrative Hub Agent
    admin_agent_name: Optional[str] = None

    # Retry configuration for backend requests
    max_retries: int = Field(
        default=10, # Increased retries for busy queues
        ge=0,
        le=100,
        description="Maximum number of retry attempts when a backend server request fails"
    )
    retry_total_timeout_seconds: float = Field(
        default=600.0, # Wait up to 10 minutes total for a slot
        ge=0.1,
        le=3600.0,
        description="Total time budget (in seconds) for all retry attempts"
    )
    retry_base_delay_ms: int = Field(
        default=10,  # REDUCED from 50 to 10
        ge=1,
        le=5000,
        description="Base delay in milliseconds for exponential backoff between retries"
    )
    
    # --- HTTPS/SSL Settings ---
    ssl_keyfile: Optional[str] = Field(default=None, description="Path to the SSL private key file (e.g., key.pem). Requires a restart.")
    ssl_certfile: Optional[str] = Field(default=None, description="Path to the SSL certificate file (e.g., cert.pem). Requires a restart.")
    ssl_keyfile_content: Optional[str] = Field(default=None, description="Content of the uploaded SSL key file.", exclude=True) # Exclude from API responses
    ssl_certfile_content: Optional[str] = Field(default=None, description="Content of the uploaded SSL cert file.", exclude=True) # Exclude from API responses

    # --- SEARCH SETTINGS ---
    google_search_api_key: Optional[str] = Field(default=None, description="SerpApi or Google Custom Search Key")

    # --- GATEWAY SETTINGS ---
    wizard_completed: bool = False
    enable_tours: bool = True
    tour_dashboard: bool = True
    tour_models: bool = True
    tour_workflows: bool = True
    tour_datastores: bool = True
    tour_nodes: bool = True
    enable_bot_mode: bool = False
    enable_debug_mode: bool = False
    enable_ollama_api: bool = True
    enable_openai_api: bool = False
    openai_port: int = 8081 # Default secondary port

    # --- AUTO-ROUTING (SB-MRA) ---
    enable_sb_mra: bool = True
    routing_weight_priority: float = 1.0
    routing_weight_reliability: float = 2.0
    routing_weight_ecology: float = 4.5  # BOOSTED: Significant penalty for resource waste
    routing_weight_semantic: float = 3.0
    
    routing_vectorizer_name: str = "sentense_transformer"
    routing_vectorizer_model: Optional[str] = "sentence-transformers/all-MiniLM-L6-v2"
    routing_vectorizer_base_url: Optional[str] = None
    
    # Context Logic
    routing_context_margin: int = Field(default=512, description="Buffer tokens added to prompt length check.")
    routing_context_strategy: str = Field(default="crop", description="Strategy when user num_ctx > model context: 'forward', 'crop' (limit to model), or 'force' (always use model max).")

    # --- MEMORY RECOVERY ---
    memory_recovery_mode: str = Field(default="handles", description="How to recover deep memories: 'handles' (AI must search) or 'vector' (Auto-RAG).")
    memory_vector_top_k: int = 3
    memory_vector_threshold: float = 0.6

    # --- SECURITY ---
    blocked_ollama_endpoints: str = Field(
        default="pull,delete,create,copy,push",
        description="Comma-separated list of Ollama API paths to block for API key holders."
    )

    @field_validator('retry_total_timeout_seconds')
    @classmethod
    def validate_retry_timeout(cls, v: float, info) -> float:
        """Ensure retry timeout is reasonable."""
        if v <= 0:
            raise ValueError("retry_total_timeout_seconds must be positive")
        return v
    
    @field_validator('branding_logo_url', 'ssl_keyfile', 'ssl_certfile', 'selected_theme', 'ui_style')
    @classmethod
    def validate_empty_string_to_none(cls, v: Optional[str]) -> Optional[str]:
        """Ensures form 'empty' values are stored as null in DB."""
        if v == "" or v is None:
            return None
        return v