"""
LLM Provider 导出
=================
"""

from app.ai.providers.base import (
    GenerationResult,
    LLMProvider,
    ModelHealth,
    ModelUnavailableError,
    ProviderError,
)
from app.ai.providers.generic_llm_provider import GenericLLMProvider, MockProvider

# 火山方舟官方 SDK Provider
# 需要: pip install 'volcengine-python-sdk[ark]'
try:
    from app.ai.providers.volcengine_provider import VolcengineProvider

    VOLCENGINE_AVAILABLE = True
except ImportError:
    VOLCENGINE_AVAILABLE = False
    VolcengineProvider = None

# 可灵 AI Provider
# 需要: pip install pyjwt
try:
    from app.ai.providers.klingai_provider import KlingAIProvider

    KLINGAI_AVAILABLE = True
except ImportError:
    KLINGAI_AVAILABLE = False
    KlingAIProvider = None

__all__ = [
    "LLMProvider",
    "GenerationResult",
    "ModelHealth",
    "ProviderError",
    "ModelUnavailableError",
    "GenericLLMProvider",
    "MockProvider",
]

if VOLCENGINE_AVAILABLE:
    __all__.append("VolcengineProvider")

if KLINGAI_AVAILABLE:
    __all__.append("KlingAIProvider")