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

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

# 火山方舟官方 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

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

if VOLCENGINE_AVAILABLE:
    __all__.append("VolcengineProvider")
