mirror of
https://github.com/jihe520/MathModelAgent.git
synced 2026-09-19 08:09:48 +08:00
EastonJiang
11f38624cd
* fix: 加固 Docker matplotlib 中文字体配置 Co-authored-by: Cursor <cursoragent@cursor.com> * fix: 提前读取路径 * perf: 进度时间线与代码执行区自动跟踪最新消息 Co-authored-by: Cursor <cursoragent@cursor.com> * fix: 未验证模型配置直接启动任务时无限重试刷爆日志 - 前端保存配置时始终推送到后端，不再依赖验证通过 - 后端 workflow 入口预校验四个 Agent 的 model/apiKey - _validate_config 改抛 LLMConfigError 避免被 JSON 重试捕获 - coordinator/modeler 的 JSON 解析重试加上限 (MAX_JSON_RETRIES=3) Co-authored-by: Cursor <cursoragent@cursor.com> --------- Co-authored-by: Cursor <cursoragent@cursor.com>
109 lines
4.2 KiB
Python
109 lines
4.2 KiB
Python
"""Matplotlib 引导脚本，供本地/E2B 解释器在 kernel 启动时注入字体与绘图常量。"""
|
||
|
||
from __future__ import annotations
|
||
|
||
# 竞赛向配色（与 CODER_PROMPT 保持同名，阶段 3 可在此统一调整）
|
||
COLORS: dict[str, str] = {
|
||
"primary": "#2E5B88",
|
||
"secondary": "#E85D4C",
|
||
"tertiary": "#4A9B7F",
|
||
"neutral": "#7F7F7F",
|
||
"light": "#B8D4E8",
|
||
}
|
||
|
||
FIG_SINGLE = (5, 4)
|
||
FIG_DOUBLE = (10, 4)
|
||
FIG_WIDE = (8, 3)
|
||
FIG_SQUARE = (6, 6)
|
||
|
||
|
||
def build_matplotlib_init_code(
|
||
work_dir: str,
|
||
*,
|
||
font_dir: str | None = None,
|
||
setup_chdir: bool = True,
|
||
) -> str:
|
||
"""生成在 Jupyter/E2B kernel 中执行的 matplotlib 初始化代码。
|
||
|
||
Args:
|
||
work_dir: 任务工作目录（本地解释器用于 chdir）。
|
||
font_dir: 字体文件目录，默认与 work_dir 相同。
|
||
setup_chdir: 是否切换到 work_dir。
|
||
|
||
Returns:
|
||
可在 kernel 中执行的 Python 代码字符串。
|
||
"""
|
||
font_dir = font_dir or work_dir
|
||
colors_repr = repr(COLORS)
|
||
fig_single = repr(FIG_SINGLE)
|
||
fig_double = repr(FIG_DOUBLE)
|
||
fig_wide = repr(FIG_WIDE)
|
||
fig_square = repr(FIG_SQUARE)
|
||
|
||
# 必须先解析为绝对路径：相对 work_dir 在 os.chdir 之后会导致 listdir/addfont 失败
|
||
chdir_block = (
|
||
"import os\n"
|
||
f"work_dir = os.path.abspath(r'{work_dir}')\n"
|
||
f"_font_dir = os.path.abspath(r'{font_dir}')\n"
|
||
)
|
||
if setup_chdir:
|
||
chdir_block += (
|
||
"os.makedirs(work_dir, exist_ok=True)\n"
|
||
"os.chdir(work_dir)\n"
|
||
"print('[matplotlib_setup] 当前工作目录:', os.getcwd())\n"
|
||
)
|
||
|
||
return (
|
||
chdir_block
|
||
+ "import matplotlib\n"
|
||
+ "import matplotlib.pyplot as plt\n"
|
||
+ "from matplotlib import font_manager\n"
|
||
+ "import glob as _glob, pathlib as _pl\n"
|
||
+ "_cache_dir = _pl.Path(matplotlib.get_cachedir())\n"
|
||
+ "for _cache_file in _glob.glob(str(_cache_dir / 'fontlist*.json')):\n"
|
||
+ " _pl.Path(_cache_file).unlink(missing_ok=True)\n"
|
||
+ "font_manager.fontManager.__init__()\n"
|
||
+ "_cjk_fonts = []\n"
|
||
+ "for _f in os.listdir(_font_dir):\n"
|
||
+ " if _f.lower().endswith(('.ttf', '.otf', '.ttc')):\n"
|
||
+ " _fp = os.path.join(_font_dir, _f)\n"
|
||
+ " font_manager.fontManager.addfont(_fp)\n"
|
||
+ " _name = font_manager.FontProperties(fname=_fp).get_name()\n"
|
||
+ " if _name not in _cjk_fonts:\n"
|
||
+ " _cjk_fonts.append(_name)\n"
|
||
+ "if _cjk_fonts:\n"
|
||
+ " CJK_FONT = _cjk_fonts[0]\n"
|
||
+ " _fallback = ['Heiti SC', 'STHeiti', 'PingFang SC', 'Noto Sans CJK SC', 'Noto Sans SC', 'WenQuanYi Micro Hei', 'Microsoft YaHei', 'sans-serif']\n"
|
||
+ " plt.rcParams['font.sans-serif'] = _cjk_fonts + [f for f in _fallback if f not in _cjk_fonts]\n"
|
||
+ " plt.rcParams['axes.unicode_minus'] = False\n"
|
||
+ " plt.rcParams['font.family'] = 'sans-serif'\n"
|
||
+ " print(f'[matplotlib_setup] 中文字体已加载: {CJK_FONT} (共 {len(_cjk_fonts)} 个)')\n"
|
||
+ "else:\n"
|
||
+ " CJK_FONT = None\n"
|
||
+ " print('[matplotlib_setup] 警告: 未找到中文字体文件，中文标签可能显示为方框')\n"
|
||
+ "plt.rcParams.update({\n"
|
||
+ " 'font.size': 11,\n"
|
||
+ " 'axes.titlesize': 12,\n"
|
||
+ " 'axes.titleweight': 'bold',\n"
|
||
+ " 'axes.labelsize': 11,\n"
|
||
+ " 'axes.linewidth': 1.2,\n"
|
||
+ " 'axes.spines.top': False,\n"
|
||
+ " 'axes.spines.right': False,\n"
|
||
+ " 'xtick.labelsize': 10,\n"
|
||
+ " 'ytick.labelsize': 10,\n"
|
||
+ " 'legend.fontsize': 10,\n"
|
||
+ " 'legend.frameon': False,\n"
|
||
+ " 'figure.dpi': 300,\n"
|
||
+ " 'savefig.dpi': 300,\n"
|
||
+ " 'savefig.bbox': 'tight',\n"
|
||
+ " 'savefig.pad_inches': 0.1,\n"
|
||
+ "})\n"
|
||
+ f"COLORS = {colors_repr}\n"
|
||
+ "DEFAULT_COLORS = list(COLORS.values())\n"
|
||
+ f"FIG_SINGLE = {fig_single}\n"
|
||
+ f"FIG_DOUBLE = {fig_double}\n"
|
||
+ f"FIG_WIDE = {fig_wide}\n"
|
||
+ f"FIG_SQUARE = {fig_square}\n"
|
||
+ "print('[matplotlib_setup] 绘图环境就绪 (COLORS, FIG_* 已注入)')\n"
|
||
)
|