egutierrez
f12272d002
- docs/capabilities/INDEX.md - docs/capabilities/comfyui.md - python/functions/browser/comfyui_export_workflow_ui.md - python/functions/browser/comfyui_export_workflow_ui.py - python/functions/browser/comfyui_load_workflow_ui.md - python/functions/browser/comfyui_load_workflow_ui.py - python/functions/browser/comfyui_queue_prompt_ui.md - python/functions/browser/comfyui_queue_prompt_ui.py - python/functions/browser/comfyui_refresh_nodes_ui.md - python/functions/browser/comfyui_refresh_nodes_ui.py - ... Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
66 lines
2.5 KiB
Python
66 lines
2.5 KiB
Python
"""Consulta el catalogo de nodos de un servidor ComfyUI via GET /object_info.
|
|
|
|
Funcion impura: hace red (HTTP GET). Solo stdlib (urllib, json).
|
|
|
|
El catalogo describe cada class_type disponible: sus inputs requeridos y
|
|
opcionales, sus tipos, y los valores enumerados (ej. la lista de checkpoints
|
|
visibles para el servidor en CheckpointLoaderSimple). Util para validar un
|
|
workflow antes de enviarlo y para descubrir que checkpoints/samplers existen.
|
|
"""
|
|
import json
|
|
import urllib.error
|
|
import urllib.parse
|
|
import urllib.request
|
|
|
|
|
|
def comfyui_object_info(
|
|
server: str = "127.0.0.1:8188",
|
|
node_class: str | None = None,
|
|
timeout: float = 30.0,
|
|
) -> dict:
|
|
"""Recupera el catalogo de nodos (o uno concreto) de ComfyUI.
|
|
|
|
Args:
|
|
server: host:port del servidor ComfyUI (sin esquema).
|
|
node_class: si se pasa, consulta solo ese class_type via
|
|
GET /object_info/{node_class} (ej. "CheckpointLoaderSimple").
|
|
Si es None, devuelve el catalogo completo (GET /object_info).
|
|
timeout: timeout de la peticion HTTP en segundos.
|
|
|
|
Returns:
|
|
dict con el catalogo. Con node_class=None es {class_type: spec, ...}.
|
|
Con node_class set, ComfyUI devuelve {class_type: spec} (un solo item).
|
|
|
|
Raises:
|
|
RuntimeError: si la peticion HTTP falla (conexion rechazada, timeout,
|
|
HTTP de error) o si la respuesta no es JSON valido. El mensaje
|
|
incluye el cuerpo del error cuando ComfyUI lo provee.
|
|
"""
|
|
path = "/object_info"
|
|
if node_class is not None:
|
|
path = f"/object_info/{urllib.parse.quote(node_class)}"
|
|
url = f"http://{server}{path}"
|
|
try:
|
|
with urllib.request.urlopen(url, timeout=timeout) as resp:
|
|
return json.loads(resp.read())
|
|
except urllib.error.HTTPError as exc:
|
|
body = exc.read().decode(errors="replace")
|
|
raise RuntimeError(
|
|
f"comfyui_object_info: HTTP {exc.code} en {url}: {body}"
|
|
) from exc
|
|
except urllib.error.URLError as exc:
|
|
raise RuntimeError(
|
|
f"comfyui_object_info: no se pudo conectar a {url}: {exc.reason}"
|
|
) from exc
|
|
except json.JSONDecodeError as exc:
|
|
raise RuntimeError(
|
|
f"comfyui_object_info: respuesta no es JSON valido desde {url}: {exc}"
|
|
) from exc
|
|
|
|
|
|
if __name__ == "__main__":
|
|
info = comfyui_object_info()
|
|
print(f"nodos disponibles: {len(info)}")
|
|
ckpts = info["CheckpointLoaderSimple"]["input"]["required"]["ckpt_name"][0]
|
|
print(f"checkpoints visibles: {ckpts}")
|