src/utils/__init__.py

"""
SPARKNET Utilities Package
Provides configuration, caching, retry logic, and GPU management
"""

from .config import (
    SparknetConfig,
    GPUConfig,
    OllamaConfig,
    MemoryConfig,
    WorkflowConfig,
    LoggingConfig,
    load_config,
    save_config,
    get_config,
    set_config,
)

from .gpu_manager import (
    GPUManager,
    get_gpu_manager,
)

from .cache import (
    LLMResponseCache,
    EmbeddingCache,
    get_llm_cache,
    get_embedding_cache,
    cached_llm_call,
)

from .retry import (
    with_retry,
    with_fallback,
    CircuitBreaker,
    CircuitBreakerError,
    TRANSIENT_EXCEPTIONS,
)

__all__ = [
    # Config
    "SparknetConfig",
    "GPUConfig",
    "OllamaConfig",
    "MemoryConfig",
    "WorkflowConfig",
    "LoggingConfig",
    "load_config",
    "save_config",
    "get_config",
    "set_config",
    # GPU
    "GPUManager",
    "get_gpu_manager",
    # Cache
    "LLMResponseCache",
    "EmbeddingCache",
    "get_llm_cache",
    "get_embedding_cache",
    "cached_llm_call",
    # Retry
    "with_retry",
    "with_fallback",
    "CircuitBreaker",
    "CircuitBreakerError",
    "TRANSIENT_EXCEPTIONS",
]