from pydantic import BaseModel


class ConfiguredProvider(BaseModel):
    name: str
    model: str


class TreeAiPolicyRead(BaseModel):
    # The model non-owners' assistant uses (null = none).
    member_provider: str | None
    # The model the association/recommendation engine uses (null = none).
    recommender_provider: str | None
    # Providers the operator has configured (from env). The owner may use any of
    # these; the two settings above restrict members and the recommender to one.
    configured_providers: list[ConfiguredProvider]
    default_provider: str


class TreeAiPolicyUpdate(BaseModel):
    member_provider: str | None = None
    recommender_provider: str | None = None