from typing import List, Optional
from pydantic import BaseModel


class CategorisationTriggerRequest(BaseModel):
    portfolio_id: int

    scenarios_to_consider: Optional[List[int]] = None
    scenario_priority_order: Optional[List[int]] = None


# {"portfolio_id": 556, "plans_to_consider": [1589319,1589320], "plan_priority_order": [1589319,1589320]}