"""
配置文件
"""
import os

# LLM 配置（兼容 OpenAI API 格式，包括 Ollama / vLLM / DeepSeek 等）
LLM_CONFIG = {
    "api_key": os.getenv("LLM_API_KEY", "sk-c44i1hy64xgzwox6x08o4zug93frq6rgn84oqugf2pje1tg4"),
    "base_url": os.getenv("LLM_BASE_URL", "https://api.xiaomimimo.com/v1"),
    "model": os.getenv("LLM_MODEL", "mimo-v2-flash"),
}

# 沙箱安全规则
SANDBOX_RULES = {
    "max_result_rows": 1000,
    "round_floats": 2,
    "suppress_small_n": 5,
    "banned_keywords": [
        "SELECT *", "INSERT", "UPDATE", "DELETE",
        "DROP", "ALTER", "CREATE", "ATTACH", "PRAGMA",
    ],
    "require_aggregation": True,
}

# 项目根目录
PROJECT_ROOT = os.path.dirname(os.path.dirname(__file__))

# 数据库路径
DB_PATH = os.getenv("DB_PATH", os.path.join(PROJECT_ROOT, "demo.db"))

# Playbook 目录
PLAYBOOK_DIR = os.getenv("PLAYBOOK_DIR", os.path.join(PROJECT_ROOT, "playbooks"))

# 图表输出目录
CHARTS_DIR = os.getenv("CHARTS_DIR", os.path.join(PROJECT_ROOT, "charts"))

# 分析控制
MAX_EXPLORATION_ROUNDS = int(os.getenv("MAX_ROUNDS", "6"))