61 lines
1.7 KiB
Python
61 lines
1.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
LLM配置文件 - 千问模型配置
|
|
"""
|
|
|
|
from src.agent.llm_client import LLMConfig
|
|
|
|
# 千问模型配置
|
|
QWEN_CONFIG = LLMConfig(
|
|
provider="qwen",
|
|
api_key="sk-c0dbefa1718d46eaa897199135066f00", # 请替换为您的千问API密钥
|
|
base_url="https://dashscope.aliyuncs.com/compatible-mode/v1",
|
|
model="qwen-plus-latest", # 可选: qwen-turbo, qwen-plus, qwen-max
|
|
temperature=0.7,
|
|
max_tokens=2000
|
|
)
|
|
|
|
# 其他模型配置示例
|
|
OPENAI_CONFIG = LLMConfig(
|
|
provider="openai",
|
|
api_key="sk-your-openai-api-key-here",
|
|
model="gpt-3.5-turbo",
|
|
temperature=0.7,
|
|
max_tokens=2000
|
|
)
|
|
|
|
ANTHROPIC_CONFIG = LLMConfig(
|
|
provider="anthropic",
|
|
api_key="sk-ant-your-anthropic-api-key-here",
|
|
model="claude-3-sonnet-20240229",
|
|
temperature=0.7,
|
|
max_tokens=2000
|
|
)
|
|
|
|
# 默认使用千问模型
|
|
DEFAULT_CONFIG = QWEN_CONFIG
|
|
|
|
|
|
def get_default_llm_config() -> LLMConfig:
|
|
"""
|
|
获取默认的LLM配置
|
|
优先从统一配置管理器获取，如果失败则使用本地配置
|
|
"""
|
|
try:
|
|
from src.config.unified_config import get_config
|
|
config = get_config()
|
|
llm_dict = config.get_llm_config()
|
|
|
|
# 创建LLMConfig对象
|
|
return LLMConfig(
|
|
provider=llm_dict.get("provider", "qwen"),
|
|
api_key=llm_dict.get("api_key", ""),
|
|
base_url=llm_dict.get("base_url", "https://dashscope.aliyuncs.com/compatible-mode/v1"),
|
|
model=llm_dict.get("model", "qwen-plus-latest"),
|
|
temperature=llm_dict.get("temperature", 0.7),
|
|
max_tokens=llm_dict.get("max_tokens", 2000)
|
|
)
|
|
except Exception:
|
|
# 如果统一配置不可用，使用本地配置
|
|
return DEFAULT_CONFIG
|