fix: input model generation no default for path inputs

Signed-off-by: Frost Ming <me@frostming.com>

pyproject.toml

@@ -5,8 +5,8 @@ authors = [{ name = "Frost Ming", email = "frost@bentoml.com" }]
 readme = "README.md"
 requires-python = ">=3.8"
 dependencies = [
+    "comfy-cli>=0.0.14",
     "pydantic>=2.9",
-    "comfy-cli>=1.2.8",
 ]
 dynamic = ["version"]
 

src/comfyui_idl/utils.py

@@ -21,6 +21,11 @@
     "BentoOutputImage",
 }
 
+BENTO_PATH_INPUT_NODES = {
+    "BentoInputPath",
+    "BentoInputImage",
+}
+
 
 def _get_node_value(node: dict) -> Any:
     return next(iter(node["inputs"].values()))
@@ -121,10 +126,10 @@ def generate_input_model(workflow: dict) -> type[BaseModel]:
         class_type = node["class_type"]
         if class_type in CLASS_TYPES:
             ann = CLASS_TYPES[class_type]
-            if class_type != "BentoInputPath":
-                field = (ann, Field(default=_get_node_value(node)))
-            else:
+            if class_type in BENTO_PATH_INPUT_NODES:
                 field = (ann, Field())
+            else:
+                field = (ann, Field(default=_get_node_value(node)))
             input_fields[name] = field
         else:
             raise ValueError(f"Unsupported class type: {class_type}")