__init__.py
950 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
"""LLM 模块"""
from .base import BaseLLMClient, LLMResponse, LLMUsage
from .glm import GLMClient
from .doubao import DoubaoClient
from .anthropic_compat import AnthropicCompatClient
from .openai_compat import OpenAICompatClient
from ..config import get_settings
def get_llm_client() -> BaseLLMClient:
"""获取 LLM 客户端"""
settings = get_settings()
provider = settings.primary_llm_provider
if provider == "openai_compat":
return OpenAICompatClient()
elif provider == "anthropic_compat":
return AnthropicCompatClient()
elif provider == "glm":
return GLMClient()
elif provider == "doubao":
return DoubaoClient()
else:
raise ValueError(f"不支持的 LLM 供应商: {provider}")
__all__ = [
"BaseLLMClient",
"LLMResponse",
"LLMUsage",
"GLMClient",
"DoubaoClient",
"AnthropicCompatClient",
"OpenAICompatClient",
"get_llm_client",
]