digital-garbageandClaude Opus 5
7330ba0c18
ComfyUI's V3 entrypoint replaces NODE_CLASS_MAPPINGS, which must not exist at all: nodes.py checks it first and reaches comfy_entrypoint only in an elif, so an empty dict registers nothing and reports no error. A module declares NODES for graph structure and SETTINGS for preferences, and the two never mix -- a saved workflow stores class_type and input names, so letting a new setting add a socket would rot existing graphs. Traits now come from two places: facts core can read off any model, and providers announced by a model's own module. Core names no model, so supporting a new one is a folder rather than an edit here. Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
87 lines
3.8 KiB
Python
87 lines
3.8 KiB
Python
"""What a module declares, and the vocabulary it may use.
|
|
|
|
Core holds no list of features. A module announces itself and this file says
|
|
what a valid announcement looks like -- nothing here names an implementation.
|
|
"""
|
|
|
|
from dataclasses import dataclass, field
|
|
from typing import Any, Callable, Dict, List, Optional
|
|
|
|
# Bumped when the shape of an announcement changes. A module declaring an older
|
|
# version is not quietly adapted: it is refused, because guessing at what an old
|
|
# declaration meant is how silent behaviour changes happen.
|
|
CONTRACT_VERSION = 1
|
|
|
|
# Closed set. A type outside it has no renderer and no validation, so accepting
|
|
# one would mean rendering something approximate.
|
|
TYPES = frozenset({"bool", "int", "float", "enum", "text", "multiline", "path"})
|
|
|
|
NUMERIC = frozenset({"int", "float"})
|
|
|
|
# Which renderers each type allows, first being the default. A hint outside its
|
|
# type's list is refused at import: the alternative is a panel that silently
|
|
# renders something other than what was asked for.
|
|
UI_HINTS: Dict[str, List[str]] = {
|
|
"bool": ["checkboxRow", "toggle"],
|
|
"int": ["number", "slider", "stepper", "macroSlider"],
|
|
"float": ["number", "slider", "stepper", "macroSlider"],
|
|
"enum": ["select", "segmented", "radioGroup", "filterList", "wheel"],
|
|
"text": ["input", "search"],
|
|
"multiline": ["textarea", "autoTextarea"],
|
|
"path": ["filterList"],
|
|
}
|
|
|
|
# Ordering is coarse-grained by stage, then by declared relations within it.
|
|
# Numbers are deliberately absent: a priority number is a global namespace, so
|
|
# inserting one module means renumbering the others.
|
|
STAGES: List[str] = ["load", "conditioning", "latent", "guidance", "sampling", "post"]
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class ModuleSpec:
|
|
"""One module's announcement, after validation."""
|
|
|
|
id: str
|
|
title: str
|
|
mount: str
|
|
settings: Dict[str, dict] = field(default_factory=dict)
|
|
requires: List[str] = field(default_factory=list) # model traits
|
|
after: List[str] = field(default_factory=list) # module ids
|
|
before: List[str] = field(default_factory=list)
|
|
stage: str = "sampling"
|
|
# ComfyUI node classes this module contributes. Kept out of `settings` on
|
|
# purpose: a setting is a preference, a node is graph structure, and letting
|
|
# a new setting change a node's socket list would rot every saved workflow.
|
|
nodes: List[type] = field(default_factory=list)
|
|
# A callable(model) -> iterable[str], contributing traits core cannot read on
|
|
# its own. This is how a model's own module teaches the system to recognise
|
|
# it, so supporting a new model is a new folder and never an edit to core.
|
|
traits: Optional[Callable] = None
|
|
ui: Optional[str] = None # served path to its ui.js
|
|
status: str = "experimental" # or "proven"
|
|
source: str = "" # dotted import path
|
|
|
|
def defaults(self) -> Dict[str, Any]:
|
|
"""The values a headless run gets when no panel has been rendered.
|
|
|
|
Derived from the same declaration the panel renders, so the two cannot
|
|
disagree about what a setting means when nobody has touched it.
|
|
"""
|
|
return {key: spec["default"] for key, spec in self.settings.items()}
|
|
|
|
def to_manifest(self) -> dict:
|
|
# `nodes` is deliberately absent: it is not JSON, and the browser has no
|
|
# use for it. The graph is ComfyUI's surface, the manifest is the app's.
|
|
return {
|
|
"id": self.id,
|
|
"title": self.title,
|
|
"mount": self.mount,
|
|
"settings": self.settings,
|
|
"requires": list(self.requires),
|
|
"after": list(self.after),
|
|
"before": list(self.before),
|
|
"stage": self.stage,
|
|
"ui": self.ui,
|
|
"status": self.status,
|
|
}
|