这里提供一个模板:
def remove_thought(self, output):
output = output.strip()
pattern = re.compile(r'<think>([\s\S]*)</think>', flags=re.DOTALL) # DOTALL to match across multiple lines
matched = [m.strip() for m in pattern.findall(output)]
processed_text = pattern.sub("", output)
return processed_text.strip(), matched
因为一开始没有加DOTALL,结果白跑实验两天,破防了。。。