mirror of
https://git.datalinker.icu/ltdrdata/ComfyUI-Manager
synced 2025-12-13 08:04:24 +08:00
49549ddcb8
Enhances ComfyUI Manager with robust batch execution tracking and unified data model architecture: - Implemented automatic batch history serialization with before/after system state snapshots - Added comprehensive state management capturing installed nodes, models, and ComfyUI version info - Enhanced task queue with proper client ID handling and WebSocket notifications - Migrated all data models to OpenAPI-generated Pydantic models for consistency - Added documentation for new TaskQueue methods (done_count, total_count, finalize) - Fixed 64 linting errors with proper imports and code cleanup Technical improvements: - All models now auto-generated from openapi.yaml ensuring API/implementation consistency - Batch tracking captures complete system state at operation start and completion - Enhanced REST endpoints with comprehensive documentation - Removed manual model files in favor of single source of truth - Added helper methods for system state capture and batch lifecycle management
107 lines
2.3 KiB
Python
107 lines
2.3 KiB
Python
"""
|
|
Data models for ComfyUI Manager.
|
|
|
|
This package contains Pydantic models used throughout the ComfyUI Manager
|
|
for data validation, serialization, and type safety.
|
|
|
|
All models are auto-generated from the OpenAPI specification to ensure
|
|
consistency between the API and implementation.
|
|
"""
|
|
|
|
from .generated_models import (
|
|
# Core Task Queue Models
|
|
QueueTaskItem,
|
|
TaskHistoryItem,
|
|
TaskStateMessage,
|
|
TaskExecutionStatus,
|
|
|
|
# WebSocket Message Models
|
|
MessageTaskDone,
|
|
MessageTaskStarted,
|
|
MessageTaskFailed,
|
|
MessageUpdate,
|
|
ManagerMessageName,
|
|
|
|
# State Management Models
|
|
BatchExecutionRecord,
|
|
ComfyUISystemState,
|
|
BatchOperation,
|
|
InstalledNodeInfo,
|
|
InstalledModelInfo,
|
|
ComfyUIVersionInfo,
|
|
|
|
# Other models
|
|
Kind,
|
|
StatusStr,
|
|
ManagerPackInfo,
|
|
ManagerPackInstalled,
|
|
SelectedVersion,
|
|
ManagerChannel,
|
|
ManagerDatabaseSource,
|
|
ManagerPackState,
|
|
ManagerPackInstallType,
|
|
ManagerPack,
|
|
InstallPackParams,
|
|
UpdateAllPacksParams,
|
|
QueueStatus,
|
|
ManagerMappings,
|
|
ModelMetadata,
|
|
NodePackageMetadata,
|
|
SnapshotItem,
|
|
Error,
|
|
InstalledPacksResponse,
|
|
HistoryResponse,
|
|
HistoryListResponse,
|
|
InstallType,
|
|
OperationType,
|
|
Result,
|
|
)
|
|
|
|
__all__ = [
|
|
# Core Task Queue Models
|
|
"QueueTaskItem",
|
|
"TaskHistoryItem",
|
|
"TaskStateMessage",
|
|
"TaskExecutionStatus",
|
|
|
|
# WebSocket Message Models
|
|
"MessageTaskDone",
|
|
"MessageTaskStarted",
|
|
"MessageTaskFailed",
|
|
"MessageUpdate",
|
|
"ManagerMessageName",
|
|
|
|
# State Management Models
|
|
"BatchExecutionRecord",
|
|
"ComfyUISystemState",
|
|
"BatchOperation",
|
|
"InstalledNodeInfo",
|
|
"InstalledModelInfo",
|
|
"ComfyUIVersionInfo",
|
|
|
|
# Other models
|
|
"Kind",
|
|
"StatusStr",
|
|
"ManagerPackInfo",
|
|
"ManagerPackInstalled",
|
|
"SelectedVersion",
|
|
"ManagerChannel",
|
|
"ManagerDatabaseSource",
|
|
"ManagerPackState",
|
|
"ManagerPackInstallType",
|
|
"ManagerPack",
|
|
"InstallPackParams",
|
|
"UpdateAllPacksParams",
|
|
"QueueStatus",
|
|
"ManagerMappings",
|
|
"ModelMetadata",
|
|
"NodePackageMetadata",
|
|
"SnapshotItem",
|
|
"Error",
|
|
"InstalledPacksResponse",
|
|
"HistoryResponse",
|
|
"HistoryListResponse",
|
|
"InstallType",
|
|
"OperationType",
|
|
"Result",
|
|
] |