From a4b8dbf1d32587c1eddb1ef80601527d64f64229 Mon Sep 17 00:00:00 2001
From: BennyKok <itechbenny@gmail.com>
Date: Thu, 11 Jan 2024 12:23:32 +0800
Subject: [PATCH] fix(builder): file custom nodes types

---
 builder/modal-builder/src/main.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/builder/modal-builder/src/main.py b/builder/modal-builder/src/main.py
index caa2459..f1dfc78 100644
--- a/builder/modal-builder/src/main.py
+++ b/builder/modal-builder/src/main.py
@@ -140,11 +140,15 @@ class GitCustomNodes(BaseModel):
     hash: str
     disabled: bool
 
+class FileCustomNodes(BaseModel):
+    filename: str
+    disabled: bool
+
 
 class Snapshot(BaseModel):
     comfyui: str
     git_custom_nodes: Dict[str, GitCustomNodes]
-
+    file_custom_nodes: List[FileCustomNodes]
 
 class Model(BaseModel):
     name: str