From 87a652d038f3b509e48ca086e30ac3a3fd017f4a Mon Sep 17 00:00:00 2001
From: "Dr.Lt.Data" <dr.lt.data@gmail.com>
Date: Mon, 24 Mar 2025 01:17:07 +0900
Subject: [PATCH] fixed: channel error when `DB: local` is selected

---
 glob/manager_core.py   | 4 ++--
 glob/manager_server.py | 6 ++++++
 pyproject.toml         | 2 +-
 3 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/glob/manager_core.py b/glob/manager_core.py
index 04dd5e98..3ee4c57f 100644
--- a/glob/manager_core.py
+++ b/glob/manager_core.py
@@ -43,7 +43,7 @@ import manager_downloader
 from node_package import InstalledNodePackage
 
 
-version_code = [3, 31, 7]
+version_code = [3, 31, 8]
 version_str = f"V{version_code[0]}.{version_code[1]}" + (f'.{version_code[2]}' if len(version_code) > 2 else '')
 
 
@@ -256,7 +256,7 @@ comfy_ui_revision = "Unknown"
 comfy_ui_commit_datetime = datetime(1900, 1, 1, 0, 0, 0)
 
 channel_dict = None
-valid_channels = set()
+valid_channels = {'default', 'local'}
 channel_list = None
 
 
diff --git a/glob/manager_server.py b/glob/manager_server.py
index 5054a098..97949832 100644
--- a/glob/manager_server.py
+++ b/glob/manager_server.py
@@ -1421,6 +1421,12 @@ async def disable_node(request):
 async def check_whitelist_for_model(item):
     json_obj = await core.get_data_by_mode('cache', 'model-list.json')
 
+    for x in json_obj.get('models', []):
+        if x['save_path'] == item['save_path'] and x['base'] == item['base'] and x['filename'] == item['filename']:
+            return True
+
+    json_obj = await core.get_data_by_mode('local', 'model-list.json')
+
     for x in json_obj.get('models', []):
         if x['save_path'] == item['save_path'] and x['base'] == item['base'] and x['filename'] == item['filename']:
             return True
diff --git a/pyproject.toml b/pyproject.toml
index 310265d9..72fe8cc4 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,7 +1,7 @@
 [project]
 name = "comfyui-manager"
 description = "ComfyUI-Manager provides features to install and manage custom nodes for ComfyUI, as well as various functionalities to assist with ComfyUI."
-version = "3.31.7"
+version = "3.31.8"
 license = { file = "LICENSE.txt" }
 dependencies = ["GitPython", "PyGithub", "matrix-client==0.4.0", "transformers", "huggingface-hub>0.20", "typer", "rich", "typing-extensions", "toml", "uv", "chardet"]