提取操作使用千问plus大模型
This commit is contained in:
@@ -10,6 +10,9 @@ load_dotenv()
|
||||
|
||||
# Maps environment variable names to nested YAML key paths
|
||||
_ENV_OVERRIDES: dict[str, list[str]] = {
|
||||
"DASHSCOPE_API_KEY": ["dashscope", "api_key"],
|
||||
"DASHSCOPE_BASE_URL": ["dashscope", "base_url"],
|
||||
"DASHSCOPE_FINE_TUNE_BASE_URL": ["dashscope", "fine_tune_base_url"],
|
||||
"ZHIPUAI_API_KEY": ["zhipuai", "api_key"],
|
||||
"STORAGE_ACCESS_KEY": ["storage", "access_key"],
|
||||
"STORAGE_SECRET_KEY": ["storage", "secret_key"],
|
||||
|
||||
@@ -1,7 +1,7 @@
|
||||
from functools import lru_cache
|
||||
|
||||
from app.clients.llm.base import LLMClient
|
||||
from app.clients.llm.zhipuai_client import ZhipuAIClient
|
||||
from app.clients.llm.qwen_client import QwenClient
|
||||
from app.clients.storage.base import StorageClient
|
||||
from app.clients.storage.rustfs_client import RustFSClient
|
||||
from app.core.config import get_config
|
||||
@@ -10,7 +10,12 @@ from app.core.config import get_config
|
||||
@lru_cache(maxsize=1)
|
||||
def get_llm_client() -> LLMClient:
|
||||
cfg = get_config()
|
||||
return ZhipuAIClient(api_key=cfg["zhipuai"]["api_key"])
|
||||
dashscope_cfg = cfg["dashscope"]
|
||||
return QwenClient(
|
||||
api_key=dashscope_cfg["api_key"],
|
||||
base_url=dashscope_cfg.get("base_url", "https://dashscope.aliyuncs.com/compatible-mode/v1"),
|
||||
fine_tune_base_url=dashscope_cfg.get("fine_tune_base_url"),
|
||||
)
|
||||
|
||||
|
||||
@lru_cache(maxsize=1)
|
||||
|
||||
Reference in New Issue
Block a user