GroupManager

def validate_transitions(self, agent_ids: List[str]) -> None:
    """Validate the transitions.

    If the selection mode is `transition`:

    - if `allow_repeat` is a list of agent_ids,
            make sure these ids exist.
    - make sure the `allowed_or_disallowed_transitions` mapping
            has valid agent ids.

    Parameters
    ----------
    agent_ids : List[str]
        The list of agent IDs.

    Raises
    ------
    ValueError
        If the transitions are invalid.
    """
    speakers: WaldiezGroupManagerSpeakers = self.data.speakers
    if speakers.selection_mode != "transition":
        return
    allow_repeat = speakers.allow_repeat
    if isinstance(allow_repeat, list):
        for agent_id in allow_repeat:
            if agent_id not in agent_ids:
                raise ValueError(f"Invalid agent id: {agent_id}")
    for (
        agent_id,
        transitions,
    ) in speakers.allowed_or_disallowed_transitions.items():
        if agent_id not in agent_ids:
            raise ValueError(f"Invalid agent id: {agent_id}")
        for agent_id in transitions:
            if agent_id not in agent_ids:
                raise ValueError(f"Invalid agent id: {agent_id}")

def get_custom_method_function(
    self,
    name_prefix: Optional[str] = None,
    name_suffix: Optional[str] = None,
) -> Tuple[str, str]:
    """Get the custom method function.

    Parameters
    ----------
    name_prefix : str
        The function name prefix.
    name_suffix : str
        The function name suffix.

    Returns
    -------
    Tuple[str, str]
        The custom method function and the function name.
    """
    function_name = CUSTOM_SPEAKER_SELECTION
    if name_prefix:
        function_name = f"{name_prefix}_{function_name}"
    if name_suffix:
        function_name = f"{function_name}_{name_suffix}"
    return (
        generate_function(
            function_name=function_name,
            function_args=CUSTOM_SPEAKER_SELECTION_ARGS,
            function_types=CUSTOM_SPEAKER_SELECTION_TYPES,
            function_body=self.custom_method_string or "",
        ),
        function_name,
    )

@model_validator(mode="after")
def validate_group_speakers_config(self) -> Self:
    """Validate the speakers config.

    Returns
    -------
    GroupManagerSpeakers
        The group manager speakers config.

    Raises
    ------
    ValueError
        If the custom method is invalid.
    """
    if self.selection_method == "custom":
        if not self.selection_custom_method:
            raise ValueError("No custom method provided.")
        is_valid, error_or_body = check_function(
            code_string=self.selection_custom_method,
            function_name=CUSTOM_SPEAKER_SELECTION,
            function_args=CUSTOM_SPEAKER_SELECTION_ARGS,
        )
        if not is_valid or not error_or_body:
            # pylint: disable=inconsistent-quotes
            raise ValueError(
                f"Invalid custom method: {error_or_body or 'no content'}"
            )
        self._custom_method_string = error_or_body
    return self