From d8ee0b45841bb1a9cfaf253b8a60ab7d9163d246 Mon Sep 17 00:00:00 2001
From: Chenlei Hu <chenlei.hu@mail.utoronto.ca>
Date: Wed, 14 Aug 2024 14:17:18 -0400
Subject: [PATCH] Relax schema on node I/O link (#427)

---
 src/types/comfyWorkflow.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/types/comfyWorkflow.ts b/src/types/comfyWorkflow.ts
index d887ae978..d8b36bc06 100644
--- a/src/types/comfyWorkflow.ts
+++ b/src/types/comfyWorkflow.ts
@@ -42,7 +42,7 @@ const zNodeOutput = z
   .object({
     name: z.string(),
     type: zDataType,
-    links: z.array(z.number()).nullable(),
+    links: z.array(z.number()).nullable().optional(),
     slot_index: zSlotIndex.optional()
   })
   .passthrough()
@@ -51,7 +51,7 @@ const zNodeInput = z
   .object({
     name: z.string(),
     type: zDataType,
-    link: z.number().nullable(),
+    link: z.number().nullable().optional(),
     slot_index: zSlotIndex.optional()
   })
   .passthrough()