51 lines
1.3 KiB
Python
51 lines
1.3 KiB
Python
"""
|
|
配置文件 - API Keys 和模型设置
|
|
"""
|
|
import os
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv()
|
|
|
|
# API 配置
|
|
ANTHROPIC_API_KEY = os.getenv("ANTHROPIC_API_KEY", "")
|
|
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY", "")
|
|
AIHUBMIX_API_KEY = os.getenv("AIHUBMIX_API_KEY", "sk-yd8Tik0nFW5emKYcBdFc433b7c8b4dC182848f76819bBe73")
|
|
|
|
# AIHubMix 配置
|
|
AIHUBMIX_BASE_URL = "https://aihubmix.com/v1"
|
|
|
|
# 模型配置
|
|
DEFAULT_MODEL = "gpt-4o" # AIHubMix 支持的模型
|
|
LLM_PROVIDER = "aihubmix" # 默认使用 AIHubMix
|
|
|
|
# 辩论配置
|
|
MAX_DEBATE_ROUNDS = 3 # 最大辩论轮数
|
|
MAX_AGENTS = 6 # 最大参与 Agent 数量
|
|
|
|
# 研究模式模型角色配置
|
|
RESEARCH_MODEL_ROLES = {
|
|
"planner": {
|
|
"name": "Planner",
|
|
"default_model": "gpt-4o",
|
|
"description": "负责拆解问题，制定研究计划"
|
|
},
|
|
"researcher": {
|
|
"name": "Researcher",
|
|
"default_model": "gemini-1.5-pro",
|
|
"description": "负责执行具体的研究步骤，深度分析"
|
|
},
|
|
"writer": {
|
|
"name": "Writer",
|
|
"default_model": "claude-3-5-sonnet-20241022",
|
|
"description": "负责汇总信息，撰写最终报告"
|
|
}
|
|
}
|
|
|
|
# 决策类型
|
|
DECISION_TYPES = {
|
|
"product": "产品方案",
|
|
"business": "商业决策",
|
|
"tech": "技术选型",
|
|
"personal": "个人规划"
|
|
}
|