__init__.py

"""
# Copyright (c) 2025  PaddlePaddle Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License"
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""

from .base import KVCacheBase
from .cache_controller import CacheController
from .cache_manager import CacheManager
from .cache_utils import LayerDoneCounter, LayerSwapTimeoutError, get_rdma_nics
from .metadata import (
    AsyncTaskHandler,
    BlockNode,
    CacheBlockMetadata,
    CacheStatus,
    MatchResult,
    PDTransferMetadata,
    PendingPrefetch,
    StorageConfig,
    StorageMetadata,
    StorageType,
    TransferConfig,
    TransferResult,
    TransferStatus,
    TransferTask,
    TransferType,
)
from .storage import create_storage_connector, create_storage_scheduler
from .transfer import create_transfer_connector
from .transfer_manager import CacheTransferManager

__all__ = [
    # Base classes
    "KVCacheBase",
    # Managers
    "CacheManager",
    "CacheController",
    "CacheTransferManager",
    # Exceptions
    "LayerSwapTimeoutError",
    # Utils
    "LayerDoneCounter",
    "get_rdma_nics",
    # Metadata
    "CacheBlockMetadata",
    "BlockNode",
    "CacheStatus",
    "TransferTask",
    "TransferStatus",
    "TransferConfig",
    "TransferResult",
    "AsyncTaskHandler",
    "MatchResult",
    "StorageMetadata",
    "PendingPrefetch",
    "PDTransferMetadata",
    "StorageConfig",
    "StorageType",
    "TransferType",
    # Factory functions
    "create_storage_scheduler",
    "create_storage_connector",
    "create_transfer_connector",
]