wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-03-11 20:27:44 +08:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

fix: inline registration logic in Caching class

Browse Source 
Follow the Execution/NodeReplacement pattern — the public API methods
contain the actual logic operating on cache_provider module state,
not wrapper functions delegating to free functions.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
Deep Mehta 2026-03-09 14:25:39 -07:00
parent 1e971fdc87
commit 2c34c89265
1 changed files with 36 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
comfy_api/latest/__init__.py
View File

@ -90,14 +90,26 @@ class ComfyAPI_latest(ComfyAPIBase):
from ._caching import CacheProvider, CacheContext, CacheValue from ._caching import CacheProvider, CacheContext, CacheValue
async def register_provider(self, provider: "ComfyAPI_latest.Caching.CacheProvider") -> None: async def register_provider(self, provider: "ComfyAPI_latest.Caching.CacheProvider") -> None:
"""Register an external cache provider.""" """Register an external cache provider. Providers are called in registration order."""
from comfy_execution.cache_provider import register_cache_provider from comfy_execution import cache_provider as _cp
register_cache_provider(provider) with _cp._providers_lock:
if provider in _cp._providers:
_cp._logger.warning(f"Provider {provider.__class__.__name__} already registered")
return
_cp._providers.append(provider)
_cp._providers_snapshot = tuple(_cp._providers)
_cp._logger.debug(f"Registered cache provider: {provider.__class__.__name__}")
async def unregister_provider(self, provider: "ComfyAPI_latest.Caching.CacheProvider") -> None: async def unregister_provider(self, provider: "ComfyAPI_latest.Caching.CacheProvider") -> None:
"""Unregister an external cache provider.""" """Unregister a previously registered cache provider."""
from comfy_execution.cache_provider import unregister_cache_provider from comfy_execution import cache_provider as _cp
unregister_cache_provider(provider) with _cp._providers_lock:
try:
_cp._providers.remove(provider)
_cp._providers_snapshot = tuple(_cp._providers)
_cp._logger.debug(f"Unregistered cache provider: {provider.__class__.__name__}")
except ValueError:
_cp._logger.warning(f"Provider {provider.__class__.__name__} was not registered")
class ComfyExtension(ABC): class ComfyExtension(ABC):
async def on_load(self) -> None: async def on_load(self) -> None:
@ -155,15 +167,27 @@ class Caching:
@staticmethod @staticmethod
def register_provider(provider: "Caching.CacheProvider") -> None: def register_provider(provider: "Caching.CacheProvider") -> None:
"""Register an external cache provider.""" """Register an external cache provider. Providers are called in registration order."""
from comfy_execution.cache_provider import register_cache_provider from comfy_execution import cache_provider as _cp
register_cache_provider(provider) with _cp._providers_lock:
if provider in _cp._providers:
_cp._logger.warning(f"Provider {provider.__class__.__name__} already registered")
return
_cp._providers.append(provider)
_cp._providers_snapshot = tuple(_cp._providers)
_cp._logger.debug(f"Registered cache provider: {provider.__class__.__name__}")
@staticmethod @staticmethod
def unregister_provider(provider: "Caching.CacheProvider") -> None: def unregister_provider(provider: "Caching.CacheProvider") -> None:
"""Unregister an external cache provider.""" """Unregister a previously registered cache provider."""
from comfy_execution.cache_provider import unregister_cache_provider from comfy_execution import cache_provider as _cp
unregister_cache_provider(provider) with _cp._providers_lock:
try:
_cp._providers.remove(provider)
_cp._providers_snapshot = tuple(_cp._providers)
_cp._logger.debug(f"Unregistered cache provider: {provider.__class__.__name__}")
except ValueError:
_cp._logger.warning(f"Provider {provider.__class__.__name__} was not registered")
ComfyAPI = ComfyAPI_latest ComfyAPI = ComfyAPI_latest