53 lines
1.9 KiB
Python
53 lines
1.9 KiB
Python
|
"""
|
|||
|
|
LLM Client 路由器：在 DeepSeek / OpenAI 之间切换。
|
|||
|
|
|
|||
|
|
约定：
|
|||
|
|
- 调用方只依赖 duck-typing：需要 ``chat`` / ``chat_with_json`` 等方法。
|
|||
|
|
- 通过环境变量 ``LLM_SERVICE_CODE``（openai|deepseek）决定默认路由；
|
|||
|
|
若未设置则按 Key 存在性自动选择（优先 deepseek）。
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
import os
|
|||
|
|
from typing import Any, Optional
|
|||
|
|
|
|||
|
|
from llm.deepseek_client import DeepSeekClient, DeepSeekConfig
|
|||
|
|
from llm.openai_client import create_openai_client
|
|||
|
|
|
|||
|
|
|
|||
|
|
def resolve_llm_service_code(service_code: Optional[str] = None) -> str:
|
|||
|
|
sc = (service_code or os.getenv("LLM_SERVICE_CODE") or "").strip().lower()
|
|||
|
|
if sc in ("openai", "deepseek"):
|
|||
|
|
return sc
|
|||
|
|
|
|||
|
|
# 自动选择：优先 DeepSeek（与历史默认一致）
|
|||
|
|
if (os.getenv("DEEPSEEK_API_KEY") or "").strip():
|
|||
|
|
return "deepseek"
|
|||
|
|
if (os.getenv("OPENAI_API_KEY") or "").strip():
|
|||
|
|
return "openai"
|
|||
|
|
return "deepseek"
|
|||
|
|
|
|||
|
|
|
|||
|
|
def create_llm_client(service_code: Optional[str] = None, **kwargs: Any) -> Any:
|
|||
|
|
"""
|
|||
|
|
创建 LLM Client。
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
DeepSeekClient 或 OpenAIClient（同形态接口）。
|
|||
|
|
"""
|
|||
|
|
sc = resolve_llm_service_code(service_code)
|
|||
|
|
if sc == "openai":
|
|||
|
|
# OpenAI 侧：默认从 OPENAI_* 读取
|
|||
|
|
return create_openai_client(**kwargs)
|
|||
|
|
|
|||
|
|
# DeepSeek 侧：从 DEEPSEEK_* 读取
|
|||
|
|
api_key = (kwargs.pop("api_key", None) or os.getenv("DEEPSEEK_API_KEY") or "").strip()
|
|||
|
|
if not api_key:
|
|||
|
|
raise ValueError("未配置 DEEPSEEK_API_KEY（LLM_SERVICE_CODE=deepseek）")
|
|||
|
|
base_url = (kwargs.pop("base_url", None) or os.getenv("DEEPSEEK_BASE_URL") or "https://api.deepseek.com").strip()
|
|||
|
|
model = (kwargs.pop("model_name", None) or os.getenv("MODEL_PRIMARY") or "deepseek-chat").strip()
|
|||
|
|
cfg = DeepSeekConfig(api_key=api_key, base_url=base_url, model_name=model, **kwargs)
|
|||
|
|
return DeepSeekClient(cfg)
|
|||
|
|
|