def extract_multi_choice()

in utils/extract_choice.py [0:0]

• 15 lines of code
• 3 McCabe index (conditional complexity)

def extract_multi_choice(gen, choice_list):
    gen = gen.strip()
    gen = gen.replace("、", "").replace(",", "").replace("，", "").replace(" ", "")

    # 严格遵循指令的情况
    res = re.search(
        r"答案[是：:]+([ABCDEF]+)",
        gen,
    )

    # 未严格遵循指令，直接输出答案的情况
    if res is None:
        res = re.search(
            r"(^[ABCDEF]+)",
            gen,
        )

    if res is None:
        return "X"

    return res.group(1)