Merge pull request #167 from openqj/feat-imagegen-mcp

feat: 图片生成 MCP 工具 image_generate（中转站图片分组）

cloud/worker.py

@@ -167,6 +167,8 @@ from cloud.peer_messages import row_get as _row_get
 from cloud.sub2api import ensure_user_gateway_key as _ensure_sub2api_key
 from cloud.websearch import ensure_codex_home_websearch as _ensure_websearch_mcp
 from cloud.websearch import ensure_websearch_mcp_env as _ensure_websearch_mcp_env
+from cloud.imagegen import ensure_codex_home_imagegen as _ensure_imagegen_mcp
+from cloud.imagegen import ensure_imagegen_mcp_env as _ensure_imagegen_mcp_env
 from cloud.skill_seed import ensure_codex_home_skills as _ensure_skills
 from cloud.sub2api import gateway_name as _gateway_name
 from cloud.sub2api import gateway_origin as _gateway_origin
@@ -415,6 +417,7 @@ class Worker:
                  tenant_template_home=None, tenant_template_repo=None,
                  egress_proxy_port=0, claim_executor_only=False,
                  websearch_mcp=False, seed_skills=False,
+                 imagegen_mcp=False,
                  host_name=None, pr_jobs_only=False,
                  upload_dir=None,
                  digest_interval_seconds=3600,
@@ -567,6 +570,16 @@ class Worker:
         # [mcp_servers.websearch]（容器内路径，见 _tenant_dirs）。
         # 老租户下一次任务自动合并，已有 section 不重写。
         self.websearch_mcp = bool(websearch_mcp)
+        # 图片生成 MCP：默认关（老行为：租户 codex-home 只有
+        # 模板 config.toml + 可选的 websearch MCP）。开启后
+        # _tenant_dirs 在 websearch 播种之后调用 imagegen 的
+        # 播种助手，把 imagegen.py / imagegen_mcp.py 拷进租户
+        # codex-home 并追加 [mcp_servers.imagegen]；_prepare_tenant
+        # 在容器启动前写 [mcp_servers.imagegen.env]（系统级
+        # IMAGE_GEN_API_KEY + 代理 + NO_PROXY 网关 host）。
+        # 未配 IMAGE_GEN_API_KEY 时工具会明确报错“未配置”，
+        # 不会静默回退到任务自己的分组 2 key（那会 404）。
+        self.imagegen_mcp = bool(imagegen_mcp)
         # 技能播种：默认关（老行为：租户 codex-home 只有模板
         # config.toml + 可选的 websearch MCP）。开启后
         # _tenant_dirs 在 websearch 播种之后调用 skill_seed
@@ -2448,6 +2461,17 @@ class Worker:
                     "python3",
                     "/var/lib/codex-app-server/codex-home/"
                     "websearch_mcp.py"])
+        if self.imagegen_mcp:
+            # 与 websearch 同款：播种失败直接抛错、任务不派发
+            # ——静默无生图比 loud 失败更糟，尤其对已宣传
+            # “能生图”的任务。key 不在这里写，只写
+            # command/args；key 由 _prepare_tenant 的 env 块给。
+            _ensure_imagegen_mcp(
+                dirs["codex_home"],
+                command_args=[
+                    "python3",
+                    "/var/lib/codex-app-server/codex-home/"
+                    "imagegen_mcp.py"])
         if self.seed_skills:
             # 技能播种失败绝不阻塞任务：技能是可选的流程
             # 说明，不是 websearch 那种“已宣传就必须有”的
@@ -2875,6 +2899,35 @@ class Worker:
             except Exception as exc:
                 self.logger.warning(
                     "websearch MCP env write failed: %s" % exc)
+        if self.tenant_isolation and self.imagegen_mcp:
+            # 图片生成 MCP env：系统级 IMAGE_GEN_*（worker 环境，
+            # 即主机 env 文件）+ 代理（供下载返回的图片 URL）+
+            # NO_PROXY（供直连本地网关 origin，不走 egress 代理）。
+            # 未配 IMAGE_GEN_API_KEY 时仍写 base/model，工具会
+            # 明确报“未配置”；绝不把任务自己的分组 2 key 塞进来。
+            try:
+                imagegen_env = dict(proxy_env) if proxy_env else {}
+                for key in ("IMAGE_GEN_BASE_URL", "IMAGE_GEN_API_KEY",
+                            "IMAGE_GEN_MODEL"):
+                    if os.environ.get(key):
+                        imagegen_env[key] = os.environ[key]
+                base = (os.environ.get("IMAGE_GEN_BASE_URL", "")
+                        or os.environ.get("SUB2API_BASE_URL", ""))
+                host = base.split("://", 1)[-1].split("/", 1)[0]
+                host = host.split(":", 1)[0]
+                no_proxy = imagegen_env.get("NO_PROXY", "")
+                parts = [p for p in no_proxy.split(",") if p]
+                for item in ("localhost", "127.0.0.1", host):
+                    if item and item not in parts:
+                        parts.append(item)
+                imagegen_env["NO_PROXY"] = ",".join(parts)
+                imagegen_env["no_proxy"] = ",".join(parts)
+                home = self._tenant_dirs(
+                    task.get("project_id"))["codex_home"]
+                _ensure_imagegen_mcp_env(home, imagegen_env)
+            except Exception as exc:
+                self.logger.warning(
+                    "imagegen MCP env write failed: %s" % exc)
         return (True, tenant_mounts, tenant_network, proxy_env, None)
 
     def _run_fork(self, task, quota_check=None):
@@ -5928,6 +5981,14 @@ def main(argv=None):
                              " 拷进租户 codex-home/skills/；"
                              "默认关，老租户不受影响；租户改过的"
                              "技能文件绝不覆盖")
+    parser.add_argument("--imagegen-mcp", action="store_true",
+                        help="租户 codex-home 自动播种图片生成 MCP："
+                             "_tenant_dirs 追加 [mcp_servers.imagegen] 并拷入"
+                             " imagegen.py / imagegen_mcp.py，_prepare_tenant"
+                             " 写 [mcp_servers.imagegen.env]（系统级"
+                             " IMAGE_GEN_API_KEY + 代理 + NO_PROXY）；"
+                             "默认关，老租户不受影响；未配 key 时工具"
+                             "明确报未配置，不回退任务自己的网关 key")
     parser.add_argument("--task-timeout", type=float, default=1800,
                         help="每任务截止时间（秒，默认 1800）；0 表示不限时"
                              "（与 host --turn-timeout 0 同约定，worker 级"
@@ -5990,7 +6051,8 @@ def main(argv=None):
                     egress_proxy_port=args.egress_proxy_port,
                     claim_executor_only=args.claim_executor_only,
                     websearch_mcp=args.websearch_mcp,
-                    seed_skills=args.seed_skills)
+                    seed_skills=args.seed_skills,
+                    imagegen_mcp=args.imagegen_mcp)
     # CLI 启动清理兜底（常驻与单次都做）：清掉上一轮残留容器。
     worker.cleanup_stale_containers()
     # CLI 起 worker 前先建表/迁移：全新库直接起循环不应因缺表而炸。