From 9f2aecf0529da0ced4ce016b999742db90679ee6 Mon Sep 17 00:00:00 2001
From: Chenlei Hu <huchenlei@proton.me>
Date: Mon, 5 May 2025 13:56:11 -0400
Subject: [PATCH] [API Nodes] Apply yellow color to api nodes by default
 (#3766)

---
 src/services/litegraphService.ts | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/src/services/litegraphService.ts b/src/services/litegraphService.ts
index 0cdd16139..999b9c4ad 100644
--- a/src/services/litegraphService.ts
+++ b/src/services/litegraphService.ts
@@ -1,5 +1,6 @@
 import {
   type IContextMenuValue,
+  LGraphCanvas,
   LGraphEventMode,
   LGraphNode,
   LiteGraph,
@@ -79,6 +80,13 @@ export const useLitegraphService = () => {
         this.#addOutputs(ComfyNode.nodeData.outputs)
         this.#setInitialSize()
         this.serialize_widgets = true
+
+        // Mark API Nodes yellow by default to distinguish with other nodes.
+        if (ComfyNode.nodeData.api_node) {
+          this.color = LGraphCanvas.node_colors.yellow.color
+          this.bgcolor = LGraphCanvas.node_colors.yellow.bgcolor
+        }
+
         void extensionService.invokeExtensionsAsync('nodeCreated', this)
       }