183 lines
6.2 KiB
Python
183 lines
6.2 KiB
Python
# config.py - 系统配置文件（添加了终端配置）
|
||
|
||
# API配置
|
||
API_BASE_URL = "https://ark.cn-beijing.volces.com/api/v3"
|
||
API_KEY = "3e96a682-919d-45c1-acb2-53bc4e9660d3" # 替换为你的API密钥
|
||
MODEL_ID = "kimi-k2-250905" #模型ID
|
||
|
||
#API_BASE_URL = "https://api.moonshot.cn/v1"
|
||
#API_KEY = "sk-xW0xjfQM6Mp9ZCWMLlnHiRJcpEOIZPTkXcN0dQ15xpZSuw2y" # 替换为你的API密钥
|
||
#MODEL_ID = "kimi-k2-thinking" #模型ID
|
||
|
||
# Tavily搜索配置
|
||
TAVILY_API_KEY = "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" # 替换为你的Tavily API密钥
|
||
|
||
# 系统配置
|
||
DEFAULT_PROJECT_PATH = "./project" # 默认项目文件夹
|
||
MAX_CONTEXT_SIZE = 100000 # 最大上下文字符数（约100K）
|
||
MAX_FILE_SIZE = 10 * 1024 * 1024 # 最大文件大小 10MB
|
||
MAX_OPEN_FILES = 20 # 最多同时打开的文件数
|
||
MAX_UPLOAD_SIZE = 50 * 1024 * 1024 # 单次上传最大50MB
|
||
UPLOAD_ALLOWED_EXTENSIONS = []
|
||
AGENT_VERSION = "v1.1" # 前端显示版本号
|
||
|
||
# 执行配置
|
||
CODE_EXECUTION_TIMEOUT = 60 # 代码执行超时（秒）
|
||
TERMINAL_COMMAND_TIMEOUT = 30 # 终端命令超时（秒）
|
||
SEARCH_MAX_RESULTS = 10 # 搜索最大结果数
|
||
|
||
# 持久化终端配置（新增）
|
||
MAX_TERMINALS = 3 # 最大同时开启的终端数量
|
||
TERMINAL_BUFFER_SIZE = 100000 # 每个终端的最大缓冲区大小（字符）
|
||
TERMINAL_DISPLAY_SIZE = 50000 # 终端显示大小限制（字符）
|
||
TERMINAL_TIMEOUT = 300 # 终端空闲超时（秒）
|
||
TERMINAL_OUTPUT_WAIT = 5 # 等待终端输出的默认时间（秒）
|
||
TERMINAL_SNAPSHOT_DEFAULT_LINES = 50 # 终端快照默认返回的行数
|
||
TERMINAL_SNAPSHOT_MAX_LINES = 200 # 终端快照允许的最大行数
|
||
TERMINAL_SNAPSHOT_MAX_CHARS = 60000 # 终端快照返回的最大字符数
|
||
TERMINAL_INPUT_MAX_CHARS = 20000 # terminal_input返回的最大字符数
|
||
|
||
# 在 config.py 中添加以下配置项
|
||
|
||
# 自动修复配置
|
||
AUTO_FIX_TOOL_CALL = False # 是否自动修复工具调用格式错误
|
||
AUTO_FIX_MAX_ATTEMPTS = 3 # 最大自动修复尝试次数
|
||
|
||
# 工具调用安全限制
|
||
MAX_ITERATIONS_PER_TASK = 100# 单个任务最大迭代次数
|
||
MAX_CONSECUTIVE_SAME_TOOL = 50 # 连续相同工具调用的最大次数
|
||
MAX_TOTAL_TOOL_CALLS = 100 #单个任务最大工具调用总数
|
||
TOOL_CALL_COOLDOWN = 0.5 # 工具调用之间的最小间隔（秒）
|
||
|
||
# 文件路径
|
||
PROMPTS_DIR = "./prompts"
|
||
DATA_DIR = "./data"
|
||
LOGS_DIR = "./logs"
|
||
# 多用户空间配置
|
||
USER_SPACE_DIR = "./users" # 每个用户的工作区根目录
|
||
USERS_DB_FILE = f"{DATA_DIR}/users.json" # 用户信息存储
|
||
INVITE_CODES_FILE = f"{DATA_DIR}/invite_codes.json" # 邀请码存储文件
|
||
|
||
# 记忆文件
|
||
MAIN_MEMORY_FILE = f"{DATA_DIR}/memory.md"
|
||
TASK_MEMORY_FILE = f"{DATA_DIR}/task_memory.md"
|
||
CONVERSATION_HISTORY_FILE = f"{DATA_DIR}/conversation_history.json"
|
||
|
||
# 日志配置
|
||
LOG_LEVEL = "INFO" # DEBUG, INFO, WARNING, ERROR
|
||
LOG_FORMAT = "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
|
||
|
||
# 安全配置
|
||
FORBIDDEN_COMMANDS = [
|
||
"rm -rf /",
|
||
"rm -rf ~",
|
||
"format",
|
||
"shutdown",
|
||
"reboot",
|
||
"kill -9",
|
||
"dd if=",
|
||
]
|
||
|
||
FORBIDDEN_PATHS = [
|
||
"/System",
|
||
"/usr",
|
||
"/bin",
|
||
"/sbin",
|
||
"/etc",
|
||
"/var",
|
||
"/tmp",
|
||
"/Applications",
|
||
"/Library",
|
||
"C:\\Windows",
|
||
"C:\\Program Files",
|
||
"C:\\Program Files (x86)",
|
||
"C:\\ProgramData"
|
||
]
|
||
|
||
# 这些是绝对不允许的根路径
|
||
FORBIDDEN_ROOT_PATHS = [
|
||
"/",
|
||
"C:\\",
|
||
"~"
|
||
]
|
||
|
||
# 确认操作列表
|
||
NEED_CONFIRMATION = [
|
||
"delete_file",
|
||
"delete_folder",
|
||
"clear_file",
|
||
"execute_terminal",
|
||
"batch_delete"
|
||
]
|
||
|
||
# 输出格式
|
||
OUTPUT_FORMATS = {
|
||
"thinking": "💭 [思考]",
|
||
"action": "🔧 [执行]",
|
||
"file": "📁 [文件]",
|
||
"search": "🔍 [搜索]",
|
||
"code": "💻 [代码]",
|
||
"terminal": "⚡ [终端]",
|
||
"memory": "📝 [记忆]",
|
||
"success": "✅ [成功]",
|
||
"error": "❌ [错误]",
|
||
"warning": "⚠️ [警告]",
|
||
"confirm": "❓ [确认]",
|
||
"info": "ℹ️ [信息]",
|
||
"session": "📺 [会话]" # 新增：终端会话标记
|
||
}
|
||
# 在 config.py 文件末尾添加以下对话持久化相关配置
|
||
|
||
# ==========================================
|
||
# 对话持久化配置（新增）
|
||
# ==========================================
|
||
|
||
# 对话存储配置
|
||
CONVERSATIONS_DIR = f"{DATA_DIR}/conversations" # 对话存储目录
|
||
CONVERSATION_INDEX_FILE = "index.json" # 对话索引文件名
|
||
CONVERSATION_FILE_PREFIX = "conv_" # 对话文件前缀
|
||
|
||
# 对话管理配置
|
||
DEFAULT_CONVERSATIONS_LIMIT = 20 # API默认返回的对话数量
|
||
MAX_CONVERSATIONS_LIMIT = 100 # API允许的最大对话数量限制
|
||
CONVERSATION_TITLE_MAX_LENGTH = 100 # 对话标题最大长度
|
||
CONVERSATION_SEARCH_MAX_RESULTS = 50 # 搜索结果最大数量
|
||
|
||
# 对话清理策略配置
|
||
CONVERSATION_AUTO_CLEANUP_ENABLED = False # 是否启用自动清理旧对话
|
||
CONVERSATION_RETENTION_DAYS = 30 # 对话保留天数（如果启用自动清理）
|
||
CONVERSATION_MAX_TOTAL = 1000 # 最大对话总数（超过时清理最旧的）
|
||
|
||
# 对话备份配置
|
||
CONVERSATION_BACKUP_ENABLED = True # 是否启用对话备份
|
||
CONVERSATION_BACKUP_INTERVAL_HOURS = 24 # 备份间隔（小时）
|
||
CONVERSATION_BACKUP_MAX_COUNT = 7 # 最多保留多少个备份文件
|
||
|
||
# 对话安全配置
|
||
CONVERSATION_MAX_MESSAGE_SIZE = 50000 # 单条消息最大字符数
|
||
CONVERSATION_MAX_MESSAGES_PER_CONVERSATION = 10000 # 每个对话最大消息数
|
||
CONVERSATION_EXPORT_MAX_SIZE = 10 * 1024 * 1024 # 导出文件最大大小（10MB）
|
||
|
||
# 对话性能配置
|
||
CONVERSATION_LAZY_LOADING = True # 是否启用懒加载（只加载对话元数据，不加载完整消息）
|
||
CONVERSATION_CACHE_SIZE = 50 # 内存中缓存的对话数量
|
||
CONVERSATION_INDEX_UPDATE_BATCH_SIZE = 100 # 批量更新索引的大小
|
||
|
||
# 工具输出字符数限制
|
||
MAX_READ_FILE_CHARS = 30000 # read_file工具限制
|
||
MAX_FOCUS_FILE_CHARS = 30000 # focus_file工具限制
|
||
MAX_RUN_COMMAND_CHARS = 10000 # run_command工具限制
|
||
MAX_EXTRACT_WEBPAGE_CHARS = 80000 # extract_webpage工具限制
|
||
|
||
# 待办事项配置
|
||
TODO_MAX_TASKS = 4
|
||
TODO_MAX_OVERVIEW_LENGTH = 999
|
||
TODO_MAX_TASK_LENGTH = 999
|
||
|
||
# 登录配置
|
||
ADMIN_USERNAME = "jojo"
|
||
ADMIN_PASSWORD_HASH = "pbkdf2:sha256:600000$FSNAVncPXW6CBtfj$b7f093f4256de9d1a16d588565d4b1e108a9c66b2901884dd118c515258d78c7"
|
||
|
||
# 模型调用相关
|
||
DEFAULT_RESPONSE_MAX_TOKENS = 32768 # 每次API响应的默认最大tokens，可在此调整
|