39 lines
1.1 KiB
Python
39 lines
1.1 KiB
Python
|
"""
|
||
|
|
配置文件
|
||
|
|
"""
|
||
|
|
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"))
|