Add comfy version to query (#1608)

* Add comfy version to query

* Add form factor detection for ComfyUI node query
This commit is contained in:
Yoland Yan 2025-03-05 04:18:45 -08:00 committed by GitHub
parent e8111527b4
commit d649ca47c6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1,12 +1,15 @@
import requests
from dataclasses import dataclass
from typing import List
import manager_util
import toml
import os
import asyncio import asyncio
import json import json
import os
import platform
import time import time
from dataclasses import dataclass
from typing import List
import manager_core
import manager_util
import requests
import toml
base_url = "https://api.comfy.org" base_url = "https://api.comfy.org"
@ -32,8 +35,34 @@ async def _get_cnr_data(cache_mode=True, dont_wait=True):
page = 1 page = 1
full_nodes = {} full_nodes = {}
# Get ComfyUI version tag
comfyui_tag = manager_core.get_comfyui_tag() or 'unknown'
# Determine form factor based on environment and platform
is_desktop = bool(os.environ.get('__COMFYUI_DESKTOP_VERSION__'))
system = platform.system().lower()
is_windows = system == 'windows'
is_mac = system == 'darwin'
if is_desktop:
if is_windows:
form_factor = 'desktop-win'
elif is_mac:
form_factor = 'desktop-mac'
else:
form_factor = 'other'
else:
if is_windows:
form_factor = 'git-windows'
elif is_mac:
form_factor = 'git-mac'
else:
form_factor = 'other'
while remained: while remained:
sub_uri = f'{base_url}/nodes?page={page}&limit=30' # Add comfyui_version and form_factor to the API request
sub_uri = f'{base_url}/nodes?page={page}&limit=30&comfyui_version={comfyui_tag}&form_factor={form_factor}'
sub_json_obj = await asyncio.wait_for(manager_util.get_data_with_cache(sub_uri, cache_mode=False, silent=True), timeout=30) sub_json_obj = await asyncio.wait_for(manager_util.get_data_with_cache(sub_uri, cache_mode=False, silent=True), timeout=30)
remained = page < sub_json_obj['totalPages'] remained = page < sub_json_obj['totalPages']