from llama_index.core.workflow import Event


class GetModulesEvent(Event):
    """
    Event to get modules. Outputs accepted modules for the task.
    """
    task: str
    modules: str


class RefineModulesEvent(Event):
    """
    Event to refine modules. Outputs refined and adapted modules.
    """
    task: str
    refined_modules: str


class ReasoningStructureEvent(Event):
    """
    Event to create reasoning structure. Outputs final reasoning structure.
    """
    task: str
    reasoning_structure: str


# TODO: Add JudgeEvent(Event) here which analyses context, judges if requirements complete,
#       and emits either loop or StopEvent.