Extract min_version_comfyUI and min_version_comfyUI_frontend to publish node api call

Author: coderfromthenorth93

comfy_cli/registry/api.py

@@ -55,6 +55,8 @@ def publish_node_version(self, node_config: PyProjectConfig, token) -> PublishNo
                 "license": license_json,
                 "repository": node_config.project.urls.repository,
                 "supported_os": node_config.project.supported_os,
+                "min_version_comfyUI_frontend": node_config.project.min_version_comfyUI_frontend,
+                "min_version_comfyUI": node_config.project.min_version_comfyUI,
             },
             "node_version": {
                 "version": node_config.project.version,

comfy_cli/registry/config_parser.py

@@ -156,6 +156,18 @@ def extract_node_configuration(
     project_data = data.get("project", {})
     urls_data = project_data.get("urls", {})
     comfy_data = data.get("tool", {}).get("comfy", {})
+    min_version_comfyUI = data.get("tool", {}).get("requires-comfyui", "")
+
+    dependencies = project_data.get("dependencies", [])
+    min_version_comfyUI_frontend = ""
+    for dep in dependencies:
+        if isinstance(dep, str) and dep.startswith("comfyui-frontend-package"):
+            parts = dep.split(">=")
+            if len(parts) > 1:
+                min_version_comfyUI_frontend = parts[1].strip()
+            break
+    
+    dependencies = [dep for dep in dependencies if not (isinstance(dep, str) and dep.startswith("comfyui-frontend-package"))]
 
     classifiers = project_data.get("classifiers", [])
     supported_os = [c for c in classifiers if c.startswith("Operating System :: ")]
@@ -185,7 +197,7 @@ def extract_node_configuration(
         description=project_data.get("description", ""),
         version=project_data.get("version", ""),
         requires_python=project_data.get("requires-python", ""),
-        dependencies=project_data.get("dependencies", []),
+        dependencies=dependencies,
         license=license,
         urls=URLs(
             homepage=urls_data.get("Homepage", ""),
@@ -194,6 +206,8 @@ def extract_node_configuration(
             issues=urls_data.get("Issues", ""),
         ),
         supported_os=supported_os,  # Add the OS classifiers
+        min_version_comfyUI_frontend=min_version_comfyUI_frontend,  # Add the ComfyUI version
+        min_version_comfyUI=min_version_comfyUI,  # Add the ComfyUI version from tools section
     )
 
     comfy = ComfyConfig(

comfy_cli/registry/types.py

@@ -70,6 +70,8 @@ class ProjectConfig:
     license: License = field(default_factory=License)
     urls: URLs = field(default_factory=URLs)
     supported_os: List[str] = field(default_factory=list)
+    min_version_comfyUI_frontend: str = ""
+    min_version_comfyUI: str = ""
 
 
 @dataclass

tests/comfy_cli/registry/test_config_parser.py

@@ -151,3 +151,46 @@ def test_extract_node_configuration_with_os_classifiers():
         assert len(result.project.supported_os) == 2
         assert "Operating System :: OS Independent" in result.project.supported_os
         assert "Operating System :: Microsoft :: Windows" in result.project.supported_os
+
+
+def test_extract_node_configuration_with_comfyui_version():
+    mock_data = {
+        "project": {
+            "dependencies": [
+                "requests>=2.0.0",
+                "comfyui-frontend-package>=1.2.3",
+                "numpy>=1.0.0"
+            ]
+        }
+    }
+    with (
+        patch("os.path.isfile", return_value=True),
+        patch("builtins.open", mock_open()),
+        patch("tomlkit.load", return_value=mock_data),
+    ):
+        result = extract_node_configuration("fake_path.toml")
+
+        assert result is not None
+        assert result.project.min_version_comfyUI_frontend == "1.2.3"
+        assert len(result.project.dependencies) == 2
+        assert "comfyui-frontend-package>=1.2.3" not in result.project.dependencies
+        assert "requests>=2.0.0" in result.project.dependencies
+        assert "numpy>=1.0.0" in result.project.dependencies
+
+
+def test_extract_node_configuration_with_requires_comfyui():
+    mock_data = {
+        "project": {},
+        "tool": {
+            "requires-comfyui": "2.0.0"
+        }
+    }
+    with (
+        patch("os.path.isfile", return_value=True),
+        patch("builtins.open", mock_open()),
+        patch("tomlkit.load", return_value=mock_data),
+    ):
+        result = extract_node_configuration("fake_path.toml")
+
+        assert result is not None
+        assert result.project.min_version_comfyUI == "2.0.0"