mirror of
https://github.com/jihe520/MathModelAgent.git
synced 2026-09-19 08:09:48 +08:00
Sanjin
60 lines
1.7 KiB
Python
60 lines
1.7 KiB
Python
import os
|
||
from utils.logger import log
|
||
from utils.data_recorder import DataRecorder
|
||
|
||
|
||
class UserOutput:
|
||
def __init__(self, work_dirs: dict, data_recorder: DataRecorder):
|
||
self.work_dirs = work_dirs
|
||
self.res: dict[str, str] = {
|
||
# "eda": "",
|
||
# "ques1": "",
|
||
}
|
||
self.data_recorder = data_recorder
|
||
self.cost_time = 0.0
|
||
self.initialized = True
|
||
|
||
def set_res(self, key: str, value: str):
|
||
self.res[key] = value # TODO： 换种数据类型有顺序
|
||
|
||
def get_res(self):
|
||
return self.res
|
||
|
||
def get_model_build_solve(self) -> str:
|
||
"""获取模型求解"""
|
||
model_build_solve = ",".join(
|
||
f"{key}-{value}"
|
||
for key, value in self.res.items()
|
||
if key.startswith("ques") and key != "ques_count"
|
||
)
|
||
|
||
return model_build_solve
|
||
|
||
def print_summary(self):
|
||
"""打印统计摘要"""
|
||
log.info("Token Usage Summary:")
|
||
pass
|
||
|
||
def get_result_to_save(self, ques_count):
|
||
# 动态顺序获取拼接res value，正确拼接顺序
|
||
ques_str = [f"ques{i}" for i in range(1, ques_count + 1)]
|
||
seq = [
|
||
"firstPage",
|
||
"RepeatQues",
|
||
"analysisQues",
|
||
"modelAssumption",
|
||
"symbol",
|
||
"eda",
|
||
*ques_str,
|
||
"sensitivity_analysis",
|
||
"judge",
|
||
"reference",
|
||
]
|
||
return "\n".join([self.res[key] for key in seq])
|
||
|
||
def save_result(self, ques_count):
|
||
res_path = os.path.join(self.work_dirs["res"], "res.md")
|
||
with open(res_path, "w", encoding="utf-8") as f:
|
||
f.write(self.get_result_to_save(ques_count))
|
||
log.info(f"结果已保存到 {res_path}")
|