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

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

# 沙箱安全规则
SANDBOX_RULES = {
    "max_result_rows": 1000,        # 聚合结果最大行数
    "round_floats": 2,              # 浮点数保留位数
    "suppress_small_n": 5,          # 分组样本 < n 时模糊处理
    "banned_keywords": [            # 禁止的 SQL 关键字
        "SELECT *",
        "INSERT",
        "UPDATE",
        "DELETE",
        "DROP",
        "ALTER",
        "CREATE",
        "ATTACH",
        "PRAGMA",
    ],
    "require_aggregation": True,     # 是否要求使用聚合函数
}

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

# 分析控制
MAX_EXPLORATION_ROUNDS = int(os.getenv("MAX_ROUNDS", "6"))  # 最大探索轮数