From b3038d717189c99b74848d3039865c20fbcf2458 Mon Sep 17 00:00:00 2001
From: Samuele Lorefice <samuelelorefice98@gmail.com>
Date: Sun, 24 Dec 2023 02:07:11 +0100
Subject: [PATCH] Added Output Type Scanning

---
 ComfySharp/NodeDBGenerator.cs | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/ComfySharp/NodeDBGenerator.cs b/ComfySharp/NodeDBGenerator.cs
index f8c655f..ff03cda 100644
--- a/ComfySharp/NodeDBGenerator.cs
+++ b/ComfySharp/NodeDBGenerator.cs
@@ -186,7 +186,10 @@ public class NodeDBGenerator {
         }
     }
 
-    private void ScanOutputs(JsonProperty output) { }
+    private void ScanOutputs(JsonProperty output) {
+        Logger.Debug($"Scanning outputs of node: {output.Name}");
+        foreach (var outputType in output.Value.EnumerateArray()) AddKnownType(outputType.GetString()!);
+    }
     
     [MethodImpl(MethodImplOptions.AggressiveInlining)]
     public void GenerateNode(JsonElement data) {