wehub-resource-sync
60 lines
1.7 KiB
Python
60 lines
1.7 KiB
Python
# llm/factory.py
|
|
|
|
from typing import Optional, Dict, Any
|
|
from .base import LLMClientInterface
|
|
from .openai_client import OpenAIClient
|
|
from .exceptions import LLMConfigException
|
|
from config.settings import settings
|
|
|
|
|
|
class LLMFactory:
|
|
"""LLM客户端工厂类"""
|
|
|
|
_clients: Dict[str, type] = {
|
|
'openai': OpenAIClient,
|
|
}
|
|
|
|
@classmethod
|
|
def create_client(
|
|
self,
|
|
provider: Optional[str] = None,
|
|
**kwargs
|
|
) -> LLMClientInterface:
|
|
"""
|
|
创建LLM客户端
|
|
|
|
Args:
|
|
provider: LLM提供商 ('openai')
|
|
**kwargs: 客户端配置参数
|
|
|
|
Returns:
|
|
LLMClientInterface: LLM客户端实例
|
|
"""
|
|
provider = provider or settings.DEFAULT_LLM_PROVIDER
|
|
|
|
if provider not in self._clients:
|
|
available_providers = ', '.join(self._clients.keys())
|
|
raise LLMConfigException(
|
|
f"Unsupported LLM provider: {provider}. "
|
|
f"Available providers: {available_providers}"
|
|
)
|
|
|
|
client_class = self._clients[provider]
|
|
|
|
try:
|
|
client = client_class(**kwargs)
|
|
print(f"✅ LLM客户端创建成功: {provider}")
|
|
return client
|
|
except Exception as e:
|
|
raise LLMConfigException(f"Failed to create {provider} client: {e}")
|
|
|
|
@classmethod
|
|
def get_available_providers(cls) -> list:
|
|
"""获取可用的LLM提供商列表"""
|
|
return list(cls._clients.keys())
|
|
|
|
@classmethod
|
|
def register_client(cls, provider: str, client_class: type):
|
|
"""注册新的LLM客户端类"""
|
|
cls._clients[provider] = client_class
|