OpenAI-compatible adapter for ComfyUI image generation and editing

2026-10-04 21:56:00 +03:00
commit c17ab5036e
29 changed files with 1730 additions and 0 deletions
@@ -0,0 +1,55 @@
+"""Environment settings and the parameter -> workflow node mapping."""
+import json
+from pathlib import Path
+
+from pydantic_settings import BaseSettings, SettingsConfigDict
+
+REPO_ROOT = Path(__file__).resolve().parent.parent
+
+CREATE_MODEL = "z-image-turbo"
+EDIT_MODEL = "qwen-image-edit"
+AVAILABLE_MODELS = [CREATE_MODEL, EDIT_MODEL]
+
+# Node ids refer to workflow_create.json / workflow_edit.json.
+# Override the whole map via MAPPING_FILE (JSON with "create" and "edit" keys).
+DEFAULT_MAPPING: dict[str, dict[str, dict[str, str]]] = {
+    "create": {
+        "prompt": {"node": "83:27", "field": "text"},
+        "size": {"node": "83:13", "width_field": "width", "height_field": "height"},
+        "seed": {"node": "83:3", "field": "seed"},
+    },
+    "edit": {
+        "prompt": {"node": "115:111", "field": "prompt"},
+        "negative": {"node": "115:110", "field": "prompt"},
+        "seed": {"node": "115:3", "field": "seed"},
+        "image": {"node": "78", "field": "image"},
+    },
+}
+
+
+class Settings(BaseSettings):
+    model_config = SettingsConfigDict(env_file=".env", extra="ignore")
+
+    comfy_base_url: str = "http://192.168.0.3:8188"
+    adapter_api_key: str = ""
+    execution_timeout: float = 300.0
+    poll_interval: float = 1.5
+    create_template_path: Path = REPO_ROOT / "workflow_create.json"
+    edit_template_path: Path = REPO_ROOT / "workflow_edit.json"
+    mapping_file: Path | None = None
+    # Validate graphs on the ComfyUI server (POST /prompt?front=false) at startup.
+    validate_templates_on_server: bool = True
+
+    def mapping(self) -> dict[str, dict[str, dict[str, str]]]:
+        if self.mapping_file is None:
+            return DEFAULT_MAPPING
+        with open(self.mapping_file, encoding="utf-8") as f:
+            loaded = json.load(f)
+        merged = {k: dict(v) for k, v in DEFAULT_MAPPING.items()}
+        for section, section_map in loaded.items():
+            merged.setdefault(section, {}).update(section_map)
+        return merged
+
+
+def get_settings() -> Settings:
+    return Settings()