From efedb19f452b8213aa004b65b0e1c11624949de4 Mon Sep 17 00:00:00 2001
From: QL-boy <46901341+QL-boy@users.noreply.github.com>
Date: Sat, 17 Jan 2026 19:56:03 +0800
Subject: [PATCH] Extend ComfyUI type definitions in __init__.py

Added support for extending ComfyUI type definitions with custom node type registration for TILE_CONFIG.
---
 __init__.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/__init__.py b/__init__.py
index a499d4c..ff2d041 100644
--- a/__init__.py
+++ b/__init__.py
@@ -11,4 +11,12 @@ NODE_DISPLAY_NAME_MAPPINGS = {
     "CustomTileMerger": "🔧 Advanced Tile Merger",
 }
 
+# Extend ComfyUI Type Definitions
+try:
+    from comfy.graph_utils import GraphBuilder
+    # Register custom dict type check
+    GraphBuilder.add_node_type("TILE_CONFIG", lambda x: isinstance(x, dict) and "splits" in x)
+except ImportError:
+    pass
+
 __all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]