小鱼开发
35 lines
741 B
Python
35 lines
741 B
Python
"""
|
|
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 as _VolcengineProvider
|
|
|
|
VolcengineProvider: type | None = _VolcengineProvider
|
|
VOLCENGINE_AVAILABLE = True
|
|
except ImportError:
|
|
VOLCENGINE_AVAILABLE = False
|
|
VolcengineProvider = None
|
|
|
|
__all__ = [
|
|
"LLMProvider",
|
|
"GenerationResult",
|
|
"ModelHealth",
|
|
"ProviderError",
|
|
"ModelUnavailableError",
|
|
]
|
|
|
|
if VOLCENGINE_AVAILABLE:
|
|
__all__.append("VolcengineProvider")
|