15 lines
499 B
Python
15 lines
499 B
Python
|
|
||
|
|
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)
|