37 lines
1.0 KiB
Python
37 lines
1.0 KiB
Python
|
"""
|
||
|
|
配置文件
|
||
|
|
"""
|
||
|
|
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")) # 最大探索轮数
|