from pydantic import BaseModel
from typing import Optional

class ReflexRequest(BaseModel):
    scenario: str
    source: Optional[str] = "unknown"

class ReflexResponse(BaseModel):
    timestamp: str
    scenario_hash: str
    anomaly_score: float
    system_state: str
    action: str
    vault_size: int