Function loop_until_match

optillm/plugins/longcepo/utils.py:144–171

Repeatedly calls a function until its output matches one of the given patterns or max attempts is reached. Args: function (Callable): Function returning (answer: str, cb_log). pattern_list (Tuple[str]): Patterns to match in the answer. num_attempts (int): Max number

(
    function: Callable, pattern_list: Tuple[str], num_attempts: int = 10
)

def loop_until_match(
    function: Callable, pattern_list: Tuple[str], num_attempts: int = 10
):
    """
    Repeatedly calls a function until its output matches one of the given patterns or max attempts is reached.

    Args:
        function (Callable): Function returning (answer: str, cb_log).
        pattern_list (Tuple[str]): Patterns to match in the answer.
        num_attempts (int): Max number of attempts (default: 10).

    Returns:
        Tuple[str, Any]: The matching answer and its corresponding log object.
    """
    correct_format = False
    for _ in range(num_attempts):
        answer, cb_log = function()

        for pattern in pattern_list:
            if pattern in answer:
                correct_format = True

        if correct_format:
            break

        logger.info("Wrong output formatting, retrying...")

    return answer, cb_log


def longcepo_init(