def enrich_input(message: str, input: dict) -> dict:
    pre_messages = input["messages"]
    problem = input["messages"][0][1]
    print("human", message + " " + problem)
    input["messages"] = pre_messages + [("human", message + " " + problem)]
    input["iterations"] = 0
    return input

def parse(res: str) -> str:
    """将优化后的问题提取出来"""
    return res.split("优化后的问题：",1)[-1]

def format_docs(docs):
    return "\n\n".join(doc.page_content for doc in docs)