from pydantic import BaseModel

from control_backend.schemas.program import Belief as ProgramBelief
from control_backend.schemas.program import Goal


class BeliefList(BaseModel):
    """
    Represents a list of beliefs, separated from a program. Useful in agents which need to
    communicate beliefs.

    :ivar: beliefs: The list of beliefs.
    """

    beliefs: list[ProgramBelief]


class GoalList(BaseModel):
    goals: list[Goal]