host
33bd14638f
- cloud/imagegen.py：POST /v1/images/generations，解析 url 或 b64_json，下载落盘到工作区；basename 校验防路径逃逸； 缺 IMAGE_GEN_API_KEY 时明确报未配置，绝不回退任务自己 的分组 2 key（分组 2 allow_image=False，回退只会 404）。 - cloud/imagegen_mcp.py：stdio MCP，工具 image_generate。 - worker --imagegen-mcp：租户 codex-home 播种脚本与 [mcp_servers.imagegen]，容器启动前写 env（系统级 IMAGE_GEN_* + 代理 + NO_PROXY 网关 host）。 - image-workflow 技能：有 image_generate 时用它生成， 无工具或报未配置时明说无法生成；改图暂无专用工具。 - 测试：imagegen 9 项 + worker 播种/env 3 项。 实测依据（2026-10-30）：分组 4 key 调 gpt-image-2.5-flare 返回 200，URL 图为真实 PNG 1024x1024；此前分组 4 账号 schedulable=False 时全模型 404，调度开启后恢复。 部署注意（主机侧，不在代码里）：env 需配 IMAGE_GEN_API_KEY； Tinyproxy 白名单需加 qinglanapi.top（下载返回图 URL 用）； autoscaler 需加 --imagegen-mcp。
97 lines
3.9 KiB
Python
97 lines
3.9 KiB
Python
"""Worker wiring for image generation MCP: tenant seeding + env.
|
|
|
|
No Docker, no network: _tenant_dirs runs against temp dirs, and
|
|
the MCP env helper only edits the tenant config.toml.
|
|
"""
|
|
|
|
import os
|
|
import sys
|
|
import tempfile
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
try:
|
|
from cloud.server import init_db
|
|
from cloud.worker import Worker
|
|
from cloud.imagegen import ensure_imagegen_mcp_env
|
|
except ImportError:
|
|
sys.path.insert(0, str(Path(__file__).resolve().parents[2]))
|
|
from cloud.server import init_db
|
|
from cloud.worker import Worker
|
|
from cloud.imagegen import ensure_imagegen_mcp_env
|
|
|
|
|
|
class WorkerImagegenTestCase(unittest.TestCase):
|
|
def setUp(self):
|
|
self.tmp = tempfile.TemporaryDirectory()
|
|
self.db_path = os.path.join(self.tmp.name, "control.db")
|
|
init_db(self.db_path)
|
|
self.template_home = os.path.join(self.tmp.name, "tpl-home")
|
|
os.makedirs(self.template_home)
|
|
open(os.path.join(self.template_home, "config.toml"),
|
|
"w").write('[model_provider.xcmapi]\n'
|
|
'base_url="https://xcmapi.org/v1"\n')
|
|
self.template_repo = os.path.join(self.tmp.name, "tpl-repo")
|
|
os.makedirs(os.path.join(self.template_repo, ".git"))
|
|
open(os.path.join(self.template_repo, "sample.py"),
|
|
"w").write("def add(a, b):\n return a + b\n")
|
|
|
|
def tearDown(self):
|
|
self.tmp.cleanup()
|
|
|
|
def _worker(self, **kwargs):
|
|
kwargs.setdefault("env", {"XCM_API_KEY": "test-placeholder"})
|
|
kwargs.setdefault("tenants_root",
|
|
os.path.join(self.tmp.name, "tenants"))
|
|
kwargs.setdefault("tenant_template_home", self.template_home)
|
|
kwargs.setdefault("tenant_template_repo", self.template_repo)
|
|
kwargs.setdefault("tenant_isolation", True)
|
|
return Worker(self.db_path, "/fake/host.py", **kwargs)
|
|
|
|
def test_flag_off_is_old_behaviour(self):
|
|
worker = self._worker()
|
|
self.assertFalse(worker.imagegen_mcp)
|
|
dirs = worker._tenant_dirs("projA")
|
|
config = open(os.path.join(dirs["codex_home"],
|
|
"config.toml")).read()
|
|
self.assertNotIn("[mcp_servers.imagegen]", config)
|
|
self.assertFalse(os.path.exists(
|
|
os.path.join(dirs["codex_home"], "imagegen_mcp.py")))
|
|
|
|
def test_flag_on_seeds_container_path_and_scripts(self):
|
|
worker = self._worker(imagegen_mcp=True)
|
|
dirs = worker._tenant_dirs("projA")
|
|
home = dirs["codex_home"]
|
|
config = open(os.path.join(home, "config.toml")).read()
|
|
self.assertIn("[mcp_servers.imagegen]", config)
|
|
self.assertIn("/var/lib/codex-app-server/codex-home/"
|
|
"imagegen_mcp.py", config)
|
|
self.assertNotIn(self.tmp.name, config)
|
|
self.assertTrue(os.path.exists(
|
|
os.path.join(home, "imagegen.py")))
|
|
self.assertTrue(os.path.exists(
|
|
os.path.join(home, "imagegen_mcp.py")))
|
|
script = open(os.path.join(home, "imagegen_mcp.py")).read()
|
|
self.assertIn("image_generate", script)
|
|
|
|
def test_env_helper_writes_imagegen_section_only(self):
|
|
worker = self._worker(imagegen_mcp=True)
|
|
dirs = worker._tenant_dirs("projA")
|
|
home = dirs["codex_home"]
|
|
self.assertTrue(ensure_imagegen_mcp_env(
|
|
home, {"IMAGE_GEN_API_KEY": "k",
|
|
"IMAGE_GEN_MODEL": "gpt-image-2.5-flare",
|
|
"NO_PROXY": "localhost,127.0.0.1"}))
|
|
config = open(os.path.join(home, "config.toml")).read()
|
|
self.assertIn("[mcp_servers.imagegen.env]", config)
|
|
self.assertIn("gpt-image-2.5-flare", config)
|
|
# Idempotent: same values again is a no-op.
|
|
self.assertFalse(ensure_imagegen_mcp_env(
|
|
home, {"IMAGE_GEN_API_KEY": "k",
|
|
"IMAGE_GEN_MODEL": "gpt-image-2.5-flare",
|
|
"NO_PROXY": "localhost,127.0.0.1"}))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|