alphard/ComfyUI-Lora-Manager
1
0
Fork 0
mirror of https://github.com/willmiao/ComfyUI-Lora-Manager.git synced 2026-03-25 15:15:44 -03:00
Code Issues Packages Projects Releases Wiki Activity

feat: Integrate WebSocket routes for download progress tracking in standalone manager

Browse Source
This commit is contained in:
Will Miao
2025-07-23 18:02:38 +08:00
parent a834fc4b30
commit 298a95432d
2 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
py/routes/base_model_routes.py
View File

@@ -8,7 +8,6 @@ from typing import Dict
import jinja2 import jinja2
from ..utils.routes_common import ModelRouteUtils from ..utils.routes_common import ModelRouteUtils
from ..services.service_registry import ServiceRegistry
from ..services.websocket_manager import ws_manager from ..services.websocket_manager import ws_manager
from ..services.settings_manager import settings from ..services.settings_manager import settings
from ..config import config from ..config import config
@@ -516,8 +515,6 @@ class BaseModelRoutes(ABC):
'error': 'Download ID is required' 'error': 'Download ID is required'
}, status=400) }, status=400)
# Get progress information from websocket manager
from ..services.websocket_manager import ws_manager
progress_data = ws_manager.get_download_progress(download_id) progress_data = ws_manager.get_download_progress(download_id)
if progress_data is None: if progress_data is None:

6
standalone.py
View File

@@ -320,6 +320,7 @@ class StandaloneLoraManager(LoraManager):
from py.routes.misc_routes import MiscRoutes from py.routes.misc_routes import MiscRoutes
from py.routes.example_images_routes import ExampleImagesRoutes from py.routes.example_images_routes import ExampleImagesRoutes
from py.routes.stats_routes import StatsRoutes from py.routes.stats_routes import StatsRoutes
from py.services.websocket_manager import ws_manager
register_default_model_types() register_default_model_types()
@@ -335,6 +336,11 @@ class StandaloneLoraManager(LoraManager):
UpdateRoutes.setup_routes(app) UpdateRoutes.setup_routes(app)
MiscRoutes.setup_routes(app) MiscRoutes.setup_routes(app)
ExampleImagesRoutes.setup_routes(app) ExampleImagesRoutes.setup_routes(app)
# Setup WebSocket routes that are shared across all model types
app.router.add_get('/ws/fetch-progress', ws_manager.handle_connection)
app.router.add_get('/ws/download-progress', ws_manager.handle_download_connection)
app.router.add_get('/ws/init-progress', ws_manager.handle_init_connection)
# Schedule service initialization # Schedule service initialization
app.on_startup.append(lambda app: cls._initialize_services()) app.on_startup.append(lambda app: cls._initialize_services())