c62156af53
B1: 项目脚手架 + 数据模型 + 租户管理 - Task 1.1: FastAPI 项目脚手架、SQLite + async SQLAlchemy - Task 1.2: 7 个数据模型 (Tenant, TenantConfig, DigitalEmployee, Conversation, Message, KnowledgeBase, Document) - Task 1.3: 租户 CRUD API + LLM 配置（含 API Key AES 加密） B2: 数字员工配置 + LLM Provider 抽象层 - Task 2.1: 数字员工 CRUD API（关联知识库） - Task 2.2: BaseLLMProvider 抽象接口 + OpenAI/Qwen Provider - Task 2.3: Provider 动态实例化 + test-provider 端点 验证: 26 个测试全部通过 Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
23 lines
656 B
Python
23 lines
656 B
Python
"""LLM Provider 注册表"""
|
|
from app.providers.base import BaseLLMProvider
|
|
from app.providers.openai_provider import OpenAIProvider
|
|
from app.providers.qwen_provider import QwenProvider
|
|
|
|
PROVIDERS = {
|
|
"openai": OpenAIProvider,
|
|
"qwen": QwenProvider,
|
|
}
|
|
|
|
|
|
def get_provider(
|
|
provider_type: str,
|
|
api_key: str,
|
|
model: str,
|
|
base_url: str | None = None,
|
|
) -> BaseLLMProvider:
|
|
"""根据类型获取 Provider 实例"""
|
|
if provider_type not in PROVIDERS:
|
|
raise ValueError(f"Unknown provider: {provider_type}")
|
|
|
|
provider_cls = PROVIDERS[provider_type]
|
|
return provider_cls(api_key=api_key, model=model, base_url=base_url) |