"""
配置文件 - 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": "个人规划"
}