小鱼开发
50 lines
1.1 KiB
Python
50 lines
1.1 KiB
Python
"""
|
|
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")
|