60 lines
2.0 KiB
Python
60 lines
2.0 KiB
Python
"""
|
||
llm/provider_factory.py
|
||
Provider 工厂：根据 settings.llm.provider 自动实例化对应 Provider
|
||
"""
|
||
|
||
from config.settings import LLMConfig, settings
|
||
from llm.providers.base_provider import BaseProvider
|
||
from utils.logger import get_logger
|
||
|
||
_logger = get_logger("LLM")
|
||
|
||
|
||
def create_provider(cfg: LLMConfig | None = None) -> BaseProvider:
|
||
"""
|
||
工厂函数：根据配置创建对应的 LLM Provider
|
||
|
||
Args:
|
||
cfg: LLMConfig 实例，None 时从全局 settings 读取
|
||
|
||
Returns:
|
||
BaseProvider 子类实例
|
||
|
||
支持的 provider:
|
||
- "openai" → OpenAIProvider（含兼容 OpenAI 协议的代理）
|
||
- "anthropic" → AnthropicProvider（预留）
|
||
- "ollama" → OllamaProvider（预留）
|
||
|
||
Raises:
|
||
ValueError: provider 名称不支持时
|
||
"""
|
||
cfg = cfg or settings.llm
|
||
_logger.info(f"🏭 Provider 工厂: 创建 [{cfg.provider}] Provider")
|
||
|
||
match cfg.provider.lower():
|
||
|
||
case "openai":
|
||
from llm.providers.openai_provider import OpenAIProvider
|
||
return OpenAIProvider(cfg)
|
||
|
||
case "anthropic":
|
||
# 预留：Anthropic Provider
|
||
# from llm.providers.anthropic_provider import AnthropicProvider
|
||
# return AnthropicProvider(cfg)
|
||
_logger.warning("⚠️ Anthropic Provider 尚未实现，回退到 OpenAI")
|
||
from llm.providers.openai_provider import OpenAIProvider
|
||
return OpenAIProvider(cfg)
|
||
|
||
case "ollama":
|
||
# 预留：Ollama 本地模型 Provider
|
||
# from llm.providers.ollama_provider import OllamaProvider
|
||
# return OllamaProvider(cfg)
|
||
_logger.warning("⚠️ Ollama Provider 尚未实现，回退到 OpenAI")
|
||
from llm.providers.openai_provider import OpenAIProvider
|
||
return OpenAIProvider(cfg)
|
||
|
||
case _:
|
||
raise ValueError(
|
||
f"不支持的 provider: '{cfg.provider}'，"
|
||
f"可选值: openai / anthropic / ollama"
|
||
) |