@@ -37,12 +115,6 @@
> - 由于持续迭代，可能存在一些已知或未知的bug
> - 由于开发中，可能消耗较多token
-## 💬交流群
-- [一群](https://qm.qq.com/q/VQ3XZrWgMs) 766798517 ，建议加下面的（开发和建议相关讨论）不一定有空回复，会优先写文档和代码
-- [二群](https://qm.qq.com/q/RzmCiRtHEW) 571780722 （开发和建议相关讨论）不一定有空回复，会优先写文档和代码
-- [三群](https://qm.qq.com/q/wlH5eT8OmQ) 1035228475（开发和建议相关讨论）不一定有空回复，会优先写文档和代码
-- [四群](https://qm.qq.com/q/wlH5eT8OmQ) 729957033（开发和建议相关讨论）不一定有空回复，会优先写文档和代码
-
**📚 有热心网友创作的wiki:** https://maimbot.pages.dev/
**📚 由SLAPQ制作的B站教程:** https://www.bilibili.com/opus/1041609335464001545
@@ -51,9 +123,17 @@
- (由 [CabLate](https://github.com/cablate) 贡献) [Telegram 与其他平台(未来可能会有)的版本](https://github.com/cablate/MaiMBot/tree/telegram) - [集中讨论串](https://github.com/SengokuCola/MaiMBot/discussions/149)
-## 📝 注意注意注意注意注意注意注意注意注意注意注意注意注意注意注意注意注意
-**如果你有想法想要提交pr**
-- 由于本项目在快速迭代和功能调整，并且有重构计划，目前不接受任何未经过核心开发组讨论的pr合并，谢谢！如您仍旧希望提交pr，可以详情请看置顶issue
+## ✍️如何给本项目报告BUG/提交建议/做贡献
+
+MaiMBot是一个开源项目，我们非常欢迎你的参与。你的贡献，无论是提交bug报告、功能需求还是代码pr，都对项目非常宝贵。我们非常感谢你的支持！🎉 但无序的讨论会降低沟通效率，进而影响问题的解决速度，因此在提交任何贡献前，请务必先阅读本项目的[贡献指南](CONTRIBUTE.md)
+
+### 💬交流群
+- [一群](https://qm.qq.com/q/VQ3XZrWgMs) 766798517 ，建议加下面的（开发和建议相关讨论）不一定有空回复，会优先写文档和代码
+- [二群](https://qm.qq.com/q/RzmCiRtHEW) 571780722 （开发和建议相关讨论）不一定有空回复，会优先写文档和代码
+- [三群](https://qm.qq.com/q/wlH5eT8OmQ) 1035228475（开发和建议相关讨论）不一定有空回复，会优先写文档和代码
+- [四群](https://qm.qq.com/q/wlH5eT8OmQ) 729957033（开发和建议相关讨论）不一定有空回复，会优先写文档和代码
+
+
📚 文档 ⬇️ 快速开始使用麦麦 ⬇️
@@ -148,23 +228,24 @@
## 设计理念
-- **千石可乐说：**
-- 这个项目最初只是为了给牛牛bot添加一点额外的功能，但是功能越写越多，最后决定重写。其目的是为了创造一个活跃在QQ群聊的"生命体"。可以目的并不是为了写一个功能齐全的机器人，而是一个尽可能让人感知到真实的类人存在.
-- 程序的功能设计理念基于一个核心的原则："最像而不是好"
-- 主打一个陪伴
-- 如果人类真的需要一个AI来陪伴自己，并不是所有人都需要一个完美的，能解决所有问题的helpful assistant，而是一个会犯错的，拥有自己感知和想法的"生命形式"。
-- 代码会保持开源和开放，但个人希望MaiMbot的运行时数据保持封闭，尽量避免以显式命令来对其进行控制和调试.我认为一个你无法完全掌控的个体才更能让你感觉到它的自主性，而视其成为一个对话机器.
+> **千石可乐说：**
+> - 这个项目最初只是为了给牛牛bot添加一点额外的功能，但是功能越写越多，最后决定重写。其目的是为了创造一个活跃在QQ群聊的"生命体"。可以目的并不是为了写一个功能齐全的机器人，而是一个尽可能让人感知到真实的类人存在.
+> - 程序的功能设计理念基于一个核心的原则："最像而不是好"
+> - 主打一个陪伴
+> - 如果人类真的需要一个AI来陪伴自己，并不是所有人都需要一个完美的，能解决所有问题的helpful assistant，而是一个会犯错的，拥有自己感知和想法的"生命形式"。
+> - 代码会保持开源和开放，但个人希望MaiMbot的运行时数据保持封闭，尽量避免以显式命令来对其进行控制和调试.我认为一个你无法完全掌控的个体才更能让你感觉到它的自主性，而视其成为一个对话机器.
## 📌 注意事项
SengokuCola~~纯编程外行，面向cursor编程，很多代码写得不好多多包涵~~已得到大脑升级
+
> [!WARNING]
> 本应用生成内容来自人工智能模型，由 AI 生成，请仔细甄别，请勿用于违反法律的用途，AI生成内容不代表本人观点和立场。
## 致谢
-[nonebot2](https://github.com/nonebot/nonebot2): 跨平台 Python 异步聊天机器人框架
-[NapCat](https://github.com/NapNeko/NapCatQQ): 现代化的基于 NTQQ 的 Bot 协议端实现
+- [nonebot2](https://github.com/nonebot/nonebot2): 跨平台 Python 异步聊天机器人框架
+- [NapCat](https://github.com/NapNeko/NapCatQQ): 现代化的基于 NTQQ 的 Bot 协议端实现
### 贡献者
diff --git a/bot.py b/bot.py
index bf853bc0c..54193b3d1 100644
--- a/bot.py
+++ b/bot.py
@@ -166,25 +166,57 @@ async def uvicorn_main():
await server.serve()
def check_eula():
- eula_file = Path("elua.confirmed")
+ eula_confirm_file = Path("elua.confirmed")
+ privacy_confirm_file = Path("privacy.confirmed")
+ eula_file = Path("EULA.md")
+ privacy_file = Path("PRIVACY.md")
- # 如果已经确认过EULA，直接返回
- if eula_file.exists():
- return
-
- print("使用MaiMBot前请先阅读ELUA协议，继续运行视为同意协议")
- print("协议内容：https://github.com/SengokuCola/MaiMBot/blob/main/EULA.md")
- print('输入"同意"或"confirmed"继续运行')
-
- while True:
- user_input = input().strip().lower() # 转换为小写以忽略大小写
- if user_input in ['同意', 'confirmed']:
- # 创建确认文件
- eula_file.touch()
- break
- else:
- print('请输入"同意"或"confirmed"以继续运行')
+ eula_updated = True
+ privacy_updated = True
+ eula_confirmed = False
+ privacy_confirmed = False
+
+ # 检查EULA确认文件是否存在
+ if eula_confirm_file.exists():
+ # 检查EULA文件版本是否更新（与elua.confirmed文件对比）
+ with open(eula_file, "r") as f:
+ eula_content = f.read()
+ with open(eula_confirm_file, "r") as f:
+ confirmed_content = f.read()
+ if eula_content == confirmed_content:
+ eula_confirmed = True
+ eula_updated = False
+
+
+ # 检查隐私条款确认文件是否存在
+ if privacy_confirm_file.exists():
+ # 检查隐私条款文件版本是否更新（与privacy.confirmed文件对比）
+ with open(privacy_file, "r") as f:
+ privacy_content = f.read()
+ with open(privacy_confirm_file, "r") as f:
+ confirmed_content = f.read()
+ if privacy_content == confirmed_content:
+ privacy_confirmed = True
+ privacy_updated = False
+
+ # 如果EULA或隐私条款有更新，提示用户重新确认
+ if eula_updated or privacy_updated:
+ print("EULA或隐私条款内容已更新，请在阅读后重新确认，继续运行视为同意更新后的以上两款协议")
+ print('输入"同意"或"confirmed"继续运行')
+ while True:
+ user_input = input().strip().lower()
+ if user_input in ['同意', 'confirmed']:
+ if eula_updated:
+ eula_confirm_file.write_text(eula_file.read_text(encoding="utf-8"),encoding="utf-8")
+ if privacy_updated:
+ privacy_confirm_file.write_text(privacy_file.read_text(encoding="utf-8"),encoding="utf-8")
+ break
+ else:
+ print('请输入"同意"或"confirmed"以继续运行')
+ return
+ elif eula_confirmed and privacy_confirmed:
+ return
def raw_main():
# 利用 TZ 环境变量设定程序工作的时区
diff --git a/run.sh b/run.sh
deleted file mode 100644
index 663fc8a67..000000000
--- a/run.sh
+++ /dev/null
@@ -1,280 +0,0 @@
-#!/bin/bash
-
-# Maimbot 一键安装脚本 by Cookie987
-# 适用于Debian系
-# 请小心使用任何一键脚本！
-
-# 如无法访问GitHub请修改此处镜像地址
-
-LANG=C.UTF-8
-
-GITHUB_REPO="https://ghfast.top/https://github.com/SengokuCola/MaiMBot.git"
-
-# 颜色输出
-GREEN="\e[32m"
-RED="\e[31m"
-RESET="\e[0m"
-
-# 需要的基本软件包
-REQUIRED_PACKAGES=("git" "sudo" "python3" "python3-venv" "curl" "gnupg" "python3-pip")
-
-# 默认项目目录
-DEFAULT_INSTALL_DIR="/opt/maimbot"
-
-# 服务名称
-SERVICE_NAME="maimbot"
-
-IS_INSTALL_MONGODB=false
-IS_INSTALL_NAPCAT=false
-
-# 1/6: 检测是否安装 whiptail
-if ! command -v whiptail &>/dev/null; then
- echo -e "${RED}[1/6] whiptail 未安装，正在安装...${RESET}"
- apt update && apt install -y whiptail
-fi
-
-get_os_info() {
- if command -v lsb_release &>/dev/null; then
- OS_INFO=$(lsb_release -d | cut -f2)
- elif [[ -f /etc/os-release ]]; then
- OS_INFO=$(grep "^PRETTY_NAME=" /etc/os-release | cut -d '"' -f2)
- else
- OS_INFO="Unknown OS"
- fi
- echo "$OS_INFO"
-}
-
-# 检查系统
-check_system() {
- # 检查是否为 root 用户
- if [[ "$(id -u)" -ne 0 ]]; then
- whiptail --title "🚫 权限不足" --msgbox "请使用 root 用户运行此脚本！\n执行方式: sudo bash $0" 10 60
- exit 1
- fi
-
- if [[ -f /etc/os-release ]]; then
- source /etc/os-release
- if [[ "$ID" != "debian" || "$VERSION_ID" != "12" ]]; then
- whiptail --title "🚫 不支持的系统" --msgbox "此脚本仅支持 Debian 12 (Bookworm)！\n当前系统: $PRETTY_NAME\n安装已终止。" 10 60
- exit 1
- fi
- else
- whiptail --title "⚠️ 无法检测系统" --msgbox "无法识别系统版本，安装已终止。" 10 60
- exit 1
- fi
-}
-
-# 3/6: 询问用户是否安装缺失的软件包
-install_packages() {
- missing_packages=()
- for package in "${REQUIRED_PACKAGES[@]}"; do
- if ! dpkg -s "$package" &>/dev/null; then
- missing_packages+=("$package")
- fi
- done
-
- if [[ ${#missing_packages[@]} -gt 0 ]]; then
- whiptail --title "📦 [3/6] 软件包检查" --yesno "检测到以下必须的依赖项目缺失:\n${missing_packages[*]}\n\n是否要自动安装？" 12 60
- if [[ $? -eq 0 ]]; then
- return 0
- else
- whiptail --title "⚠️ 注意" --yesno "某些必要的依赖项未安装，可能会影响运行！\n是否继续？" 10 60 || exit 1
- fi
- fi
-}
-
-# 4/6: Python 版本检查
-check_python() {
- PYTHON_VERSION=$(python3 -c 'import sys; print(f"{sys.version_info.major}.{sys.version_info.minor}")')
-
- python3 -c "import sys; exit(0) if sys.version_info >= (3,9) else exit(1)"
- if [[ $? -ne 0 ]]; then
- whiptail --title "⚠️ [4/6] Python 版本过低" --msgbox "检测到 Python 版本为 $PYTHON_VERSION，需要 3.9 或以上！\n请升级 Python 后重新运行本脚本。" 10 60
- exit 1
- fi
-}
-
-# 5/6: 选择分支
-choose_branch() {
- BRANCH=$(whiptail --title "🔀 [5/6] 选择 Maimbot 分支" --menu "请选择要安装的 Maimbot 分支：" 15 60 2 \
- "main" "稳定版本（推荐，供下载使用）" \
- "main-fix" "生产环境紧急修复" 3>&1 1>&2 2>&3)
-
- if [[ -z "$BRANCH" ]]; then
- BRANCH="main"
- whiptail --title "🔀 默认选择" --msgbox "未选择分支，默认安装稳定版本（main）" 10 60
- fi
-}
-
-# 6/6: 选择安装路径
-choose_install_dir() {
- INSTALL_DIR=$(whiptail --title "📂 [6/6] 选择安装路径" --inputbox "请输入 Maimbot 的安装目录：" 10 60 "$DEFAULT_INSTALL_DIR" 3>&1 1>&2 2>&3)
-
- if [[ -z "$INSTALL_DIR" ]]; then
- whiptail --title "⚠️ 取消输入" --yesno "未输入安装路径，是否退出安装？" 10 60
- if [[ $? -ne 0 ]]; then
- INSTALL_DIR="$DEFAULT_INSTALL_DIR"
- else
- exit 1
- fi
- fi
-}
-
-# 显示确认界面
-confirm_install() {
- local confirm_message="请确认以下更改:\n\n"
-
- if [[ ${#missing_packages[@]} -gt 0 ]]; then
- confirm_message+="📦 安装缺失的依赖项: ${missing_packages[*]}\n"
- else
- confirm_message+="✅ 所有依赖项已安装\n"
- fi
-
- confirm_message+="📂 安装麦麦Bot到: $INSTALL_DIR\n"
- confirm_message+="🔀 分支: $BRANCH\n"
-
- if [[ "$MONGODB_INSTALLED" == "true" ]]; then
- confirm_message+="✅ MongoDB 已安装\n"
- else
- if [[ "$IS_INSTALL_MONGODB" == "true" ]]; then
- confirm_message+="📦 安装 MongoDB\n"
- fi
- fi
-
- if [[ "$NAPCAT_INSTALLED" == "true" ]]; then
- confirm_message+="✅ NapCat 已安装\n"
- else
- if [[ "$IS_INSTALL_NAPCAT" == "true" ]]; then
- confirm_message+="📦 安装 NapCat\n"
- fi
- fi
-
- confirm_message+="🛠️ 添加麦麦Bot作为系统服务 ($SERVICE_NAME.service)\n"
-
- confitm_message+="\n\n注意：本脚本默认使用ghfast.top为GitHub进行加速，如不想使用请手动修改脚本开头的GITHUB_REPO变量。"
- whiptail --title "🔧 安装确认" --yesno "$confirm_message\n\n是否继续安装？" 15 60
- if [[ $? -ne 0 ]]; then
- whiptail --title "🚫 取消安装" --msgbox "安装已取消。" 10 60
- exit 1
- fi
-}
-
-check_mongodb() {
- if command -v mongod &>/dev/null; then
- MONGO_INSTALLED=true
- else
- MONGO_INSTALLED=false
- fi
-}
-
-# 安装 MongoDB
-install_mongodb() {
- if [[ "$MONGO_INSTALLED" == "true" ]]; then
- return 0
- fi
-
- whiptail --title "📦 [3/6] 软件包检查" --yesno "检测到未安装MongoDB，是否安装？\n如果您想使用远程数据库，请跳过此步。" 10 60
- if [[ $? -ne 0 ]]; then
- return 1
- fi
- IS_INSTALL_MONGODB=true
-}
-
-check_napcat() {
- if command -v napcat &>/dev/null; then
- NAPCAT_INSTALLED=true
- else
- NAPCAT_INSTALLED=false
- fi
-}
-
-install_napcat() {
- if [[ "$NAPCAT_INSTALLED" == "true" ]]; then
- return 0
- fi
-
- whiptail --title "📦 [3/6] 软件包检查" --yesno "检测到未安装NapCat，是否安装？\n如果您想使用远程NapCat，请跳过此步。" 10 60
- if [[ $? -ne 0 ]]; then
- return 1
- fi
- IS_INSTALL_NAPCAT=true
-}
-
-# 运行安装步骤
-whiptail --title "⚠️ 警告：安装前详阅" --msgbox "项目处于活跃开发阶段，代码可能随时更改\n文档未完善，有问题可以提交 Issue 或者 Discussion\nQQ机器人存在被限制风险，请自行了解，谨慎使用\n由于持续迭代，可能存在一些已知或未知的bug\n由于开发中，可能消耗较多token\n\n本脚本可能更新不及时，如遇到bug请优先尝试手动部署以确定是否为脚本问题" 14 60
-
-check_system
-check_mongodb
-check_napcat
-install_packages
-install_mongodb
-install_napcat
-check_python
-choose_branch
-choose_install_dir
-confirm_install
-
-# 开始安装
-whiptail --title "🚀 开始安装" --msgbox "所有环境检查完毕，即将开始安装麦麦Bot！" 10 60
-
-echo -e "${GREEN}安装依赖项...${RESET}"
-
-apt update && apt install -y "${missing_packages[@]}"
-
-
-if [[ "$IS_INSTALL_MONGODB" == "true" ]]; then
- echo -e "${GREEN}安装 MongoDB...${RESET}"
- curl -fsSL https://www.mongodb.org/static/pgp/server-8.0.asc | gpg -o /usr/share/keyrings/mongodb-server-8.0.gpg --dearmor
- echo "deb [ signed-by=/usr/share/keyrings/mongodb-server-8.0.gpg ] http://repo.mongodb.org/apt/debian bookworm/mongodb-org/8.0 main" | sudo tee /etc/apt/sources.list.d/mongodb-org-8.0.list
- apt-get update
- apt-get install -y mongodb-org
-
- systemctl enable mongod
- systemctl start mongod
-fi
-
-if [[ "$IS_INSTALL_NAPCAT" == "true" ]]; then
- echo -e "${GREEN}安装 NapCat...${RESET}"
- curl -o napcat.sh https://nclatest.znin.net/NapNeko/NapCat-Installer/main/script/install.sh && bash napcat.sh --cli y --docker n
-fi
-
-echo -e "${GREEN}创建 Python 虚拟环境...${RESET}"
-mkdir -p "$INSTALL_DIR"
-cd "$INSTALL_DIR" || exit
-python3 -m venv venv
-source venv/bin/activate
-
-echo -e "${GREEN}克隆仓库...${RESET}"
-# 安装 Maimbot
-mkdir -p "$INSTALL_DIR/repo"
-cd "$INSTALL_DIR/repo" || exit 1
-git clone -b "$BRANCH" $GITHUB_REPO .
-
-echo -e "${GREEN}安装 Python 依赖...${RESET}"
-pip install -r requirements.txt
-
-echo -e "${GREEN}设置服务...${RESET}"
-
-# 设置 Maimbot 服务
-cat <&1 1>&2 2>&3)
+
+ [[ $? -ne 0 ]] && exit 0
+
+ case "$choice" in
+ 1)
+ systemctl start ${SERVICE_NAME}
+ whiptail --msgbox "✅麦麦Bot已启动" 10 60
+ ;;
+ 2)
+ systemctl stop ${SERVICE_NAME}
+ whiptail --msgbox "🛑麦麦Bot已停止" 10 60
+ ;;
+ 3)
+ systemctl restart ${SERVICE_NAME}
+ whiptail --msgbox "🔄麦麦Bot已重启" 10 60
+ ;;
+ 4)
+ systemctl start ${SERVICE_NAME_WEB}
+ whiptail --msgbox "✅WebUI已启动" 10 60
+ ;;
+ 5)
+ systemctl stop ${SERVICE_NAME_WEB}
+ whiptail --msgbox "🛑WebUI已停止" 10 60
+ ;;
+ 6)
+ systemctl restart ${SERVICE_NAME_WEB}
+ whiptail --msgbox "🔄WebUI已重启" 10 60
+ ;;
+ 7)
+ update_dependencies
+ ;;
+ 8)
+ switch_branch
+ ;;
+ 9)
+ update_config
+ ;;
+ 10)
+ exit 0
+ ;;
+ *)
+ whiptail --msgbox "无效选项！" 10 60
+ ;;
+ esac
+ done
+}
+
+# 更新依赖
+update_dependencies() {
+ cd "${INSTALL_DIR}/repo" || {
+ whiptail --msgbox "🚫 无法进入安装目录！" 10 60
+ return 1
+ }
+ if ! git pull origin "${BRANCH}"; then
+ whiptail --msgbox "🚫 代码更新失败！" 10 60
+ return 1
+ fi
+ source "${INSTALL_DIR}/venv/bin/activate"
+ if ! pip install -r requirements.txt; then
+ whiptail --msgbox "🚫 依赖安装失败！" 10 60
+ deactivate
+ return 1
+ fi
+ deactivate
+ systemctl restart ${SERVICE_NAME}
+ whiptail --msgbox "✅ 依赖已更新并重启服务！" 10 60
+}
+
+# 切换分支
+switch_branch() {
+ new_branch=$(whiptail --inputbox "请输入要切换的分支名称：" 10 60 "${BRANCH}" 3>&1 1>&2 2>&3)
+ [[ -z "$new_branch" ]] && {
+ whiptail --msgbox "🚫 分支名称不能为空！" 10 60
+ return 1
+ }
+
+ cd "${INSTALL_DIR}/repo" || {
+ whiptail --msgbox "🚫 无法进入安装目录！" 10 60
+ return 1
+ }
+
+ if ! git ls-remote --exit-code --heads origin "${new_branch}" >/dev/null 2>&1; then
+ whiptail --msgbox "🚫 分支 ${new_branch} 不存在！" 10 60
+ return 1
+ fi
+
+ if ! git checkout "${new_branch}"; then
+ whiptail --msgbox "🚫 分支切换失败！" 10 60
+ return 1
+ fi
+
+ if ! git pull origin "${new_branch}"; then
+ whiptail --msgbox "🚫 代码拉取失败！" 10 60
+ return 1
+ fi
+
+ source "${INSTALL_DIR}/venv/bin/activate"
+ pip install -r requirements.txt
+ deactivate
+
+ sed -i "s/^BRANCH=.*/BRANCH=${new_branch}/" /etc/maimbot_install.conf
+ BRANCH="${new_branch}"
+ systemctl restart ${SERVICE_NAME}
+ touch "${INSTALL_DIR}/repo/elua.confirmed"
+ whiptail --msgbox "✅ 已切换到分支 ${new_branch} 并重启服务！" 10 60
+}
+
+# 更新配置文件
+update_config() {
+ cd "${INSTALL_DIR}/repo" || {
+ whiptail --msgbox "🚫 无法进入安装目录！" 10 60
+ return 1
+ }
+ if [[ -f config/bot_config.toml ]]; then
+ cp config/bot_config.toml config/bot_config.toml.bak
+ whiptail --msgbox "📁 原配置文件已备份为 bot_config.toml.bak" 10 60
+ source "${INSTALL_DIR}/venv/bin/activate"
+ python3 config/auto_update.py
+ deactivate
+ whiptail --msgbox "🆕 已更新配置文件，请重启麦麦Bot！" 10 60
+ return 0
+ else
+ whiptail --msgbox "🚫 未找到配置文件 bot_config.toml\n 请先运行一次麦麦Bot" 10 60
+ return 1
+ fi
+}
+
+# ----------- 主安装流程 -----------
+run_installation() {
+ # 1/6: 检测是否安装 whiptail
+ if ! command -v whiptail &>/dev/null; then
+ echo -e "${RED}[1/6] whiptail 未安装，正在安装...${RESET}"
+ apt update && apt install -y whiptail
+ fi
+
+ # 协议确认
+ if ! (whiptail --title "ℹ️ [1/6] 使用协议" --yes-button "我同意" --no-button "我拒绝" --yesno "使用麦麦Bot及此脚本前请先阅读ELUA协议\nhttps://github.com/SengokuCola/MaiMBot/blob/main/EULA.md\n\n您是否同意此协议？" 12 70); then
+ exit 1
+ fi
+
+ # 欢迎信息
+ whiptail --title "[2/6] 欢迎使用麦麦Bot一键安装脚本 by Cookie987" --msgbox "检测到您未安装麦麦Bot，将自动进入安装流程，安装完成后再次运行此脚本即可进入管理菜单。\n\n项目处于活跃开发阶段，代码可能随时更改\n文档未完善，有问题可以提交 Issue 或者 Discussion\nQQ机器人存在被限制风险，请自行了解，谨慎使用\n由于持续迭代，可能存在一些已知或未知的bug\n由于开发中，可能消耗较多token\n\n本脚本可能更新不及时，如遇到bug请优先尝试手动部署以确定是否为脚本问题" 17 60
+
+ # 系统检查
+ check_system() {
+ if [[ "$(id -u)" -ne 0 ]]; then
+ whiptail --title "🚫 权限不足" --msgbox "请使用 root 用户运行此脚本！\n执行方式: sudo bash $0" 10 60
+ exit 1
+ fi
+
+ if [[ -f /etc/os-release ]]; then
+ source /etc/os-release
+ if [[ "$ID" != "debian" || "$VERSION_ID" != "12" ]]; then
+ whiptail --title "🚫 不支持的系统" --msgbox "此脚本仅支持 Debian 12 (Bookworm)！\n当前系统: $PRETTY_NAME\n安装已终止。" 10 60
+ exit 1
+ fi
+ else
+ whiptail --title "⚠️ 无法检测系统" --msgbox "无法识别系统版本，安装已终止。" 10 60
+ exit 1
+ fi
+ }
+ check_system
+
+ # 检查MongoDB
+ check_mongodb() {
+ if command -v mongod &>/dev/null; then
+ MONGO_INSTALLED=true
+ else
+ MONGO_INSTALLED=false
+ fi
+ }
+ check_mongodb
+
+ # 检查NapCat
+ check_napcat() {
+ if command -v napcat &>/dev/null; then
+ NAPCAT_INSTALLED=true
+ else
+ NAPCAT_INSTALLED=false
+ fi
+ }
+ check_napcat
+
+ # 安装必要软件包
+ install_packages() {
+ missing_packages=()
+ for package in "${REQUIRED_PACKAGES[@]}"; do
+ if ! dpkg -s "$package" &>/dev/null; then
+ missing_packages+=("$package")
+ fi
+ done
+
+ if [[ ${#missing_packages[@]} -gt 0 ]]; then
+ whiptail --title "📦 [3/6] 软件包检查" --yesno "检测到以下必须的依赖项目缺失:\n${missing_packages[*]}\n\n是否要自动安装？" 12 60
+ if [[ $? -eq 0 ]]; then
+ IS_INSTALL_DEPENDENCIES=true
+ else
+ whiptail --title "⚠️ 注意" --yesno "某些必要的依赖项未安装，可能会影响运行！\n是否继续？" 10 60 || exit 1
+ fi
+ fi
+ }
+ install_packages
+
+ # 安装MongoDB
+ install_mongodb() {
+ [[ $MONGO_INSTALLED == true ]] && return
+ whiptail --title "📦 [3/6] 软件包检查" --yesno "检测到未安装MongoDB，是否安装？\n如果您想使用远程数据库，请跳过此步。" 10 60 && {
+ echo -e "${GREEN}安装 MongoDB...${RESET}"
+ curl -fsSL https://www.mongodb.org/static/pgp/server-8.0.asc | gpg -o /usr/share/keyrings/mongodb-server-8.0.gpg --dearmor
+ echo "deb [ signed-by=/usr/share/keyrings/mongodb-server-8.0.gpg ] http://repo.mongodb.org/apt/debian bookworm/mongodb-org/8.0 main" | tee /etc/apt/sources.list.d/mongodb-org-8.0.list
+ apt update
+ apt install -y mongodb-org
+ systemctl enable --now mongod
+ IS_INSTALL_MONGODB=true
+ }
+ }
+ install_mongodb
+
+ # 安装NapCat
+ install_napcat() {
+ [[ $NAPCAT_INSTALLED == true ]] && return
+ whiptail --title "📦 [3/6] 软件包检查" --yesno "检测到未安装NapCat，是否安装？\n如果您想使用远程NapCat，请跳过此步。" 10 60 && {
+ echo -e "${GREEN}安装 NapCat...${RESET}"
+ curl -o napcat.sh https://nclatest.znin.net/NapNeko/NapCat-Installer/main/script/install.sh && bash napcat.sh --cli y --docker n
+ IS_INSTALL_NAPCAT=true
+ }
+ }
+ install_napcat
+
+ # Python版本检查
+ check_python() {
+ PYTHON_VERSION=$(python3 -c 'import sys; print(f"{sys.version_info.major}.{sys.version_info.minor}")')
+ if ! python3 -c "import sys; exit(0) if sys.version_info >= (3,9) else exit(1)"; then
+ whiptail --title "⚠️ [4/6] Python 版本过低" --msgbox "检测到 Python 版本为 $PYTHON_VERSION，需要 3.9 或以上！\n请升级 Python 后重新运行本脚本。" 10 60
+ exit 1
+ fi
+ }
+ check_python
+
+ # 选择分支
+ choose_branch() {
+ BRANCH=$(whiptail --title "🔀 [5/6] 选择麦麦Bot分支" --menu "请选择要安装的麦麦Bot分支：" 15 60 2 \
+ "main" "稳定版本（推荐，供下载使用）" \
+ "main-fix" "生产环境紧急修复" 3>&1 1>&2 2>&3)
+ [[ -z "$BRANCH" ]] && BRANCH="main"
+ }
+ choose_branch
+
+ # 选择安装路径
+ choose_install_dir() {
+ INSTALL_DIR=$(whiptail --title "📂 [6/6] 选择安装路径" --inputbox "请输入麦麦Bot的安装目录：" 10 60 "$DEFAULT_INSTALL_DIR" 3>&1 1>&2 2>&3)
+ [[ -z "$INSTALL_DIR" ]] && {
+ whiptail --title "⚠️ 取消输入" --yesno "未输入安装路径，是否退出安装？" 10 60 && exit 1
+ INSTALL_DIR="$DEFAULT_INSTALL_DIR"
+ }
+ }
+ choose_install_dir
+
+ # 确认安装
+ confirm_install() {
+ local confirm_msg="请确认以下信息：\n\n"
+ confirm_msg+="📂 安装麦麦Bot到: $INSTALL_DIR\n"
+ confirm_msg+="🔀 分支: $BRANCH\n"
+ [[ $IS_INSTALL_DEPENDENCIES == true ]] && confirm_msg+="📦 安装依赖：${missing_packages}\n"
+ [[ $IS_INSTALL_MONGODB == true || $IS_INSTALL_NAPCAT == true ]] && confirm_msg+="📦 安装额外组件：\n"
+
+ [[ $IS_INSTALL_MONGODB == true ]] && confirm_msg+=" - MongoDB\n"
+ [[ $IS_INSTALL_NAPCAT == true ]] && confirm_msg+=" - NapCat\n"
+ confirm_msg+="\n注意：本脚本默认使用ghfast.top为GitHub进行加速，如不想使用请手动修改脚本开头的GITHUB_REPO变量。"
+
+ whiptail --title "🔧 安装确认" --yesno "$confirm_msg" 16 60 || exit 1
+ }
+ confirm_install
+
+ # 开始安装
+ echo -e "${GREEN}安装依赖...${RESET}"
+ [[ $IS_INSTALL_DEPENDENCIES == true ]] && apt update && apt install -y "${missing_packages[@]}"
+
+ echo -e "${GREEN}创建安装目录...${RESET}"
+ mkdir -p "$INSTALL_DIR"
+ cd "$INSTALL_DIR" || exit 1
+
+ echo -e "${GREEN}设置Python虚拟环境...${RESET}"
+ python3 -m venv venv
+ source venv/bin/activate
+
+ echo -e "${GREEN}克隆仓库...${RESET}"
+ git clone -b "$BRANCH" "$GITHUB_REPO" repo || {
+ echo -e "${RED}克隆仓库失败！${RESET}"
+ exit 1
+ }
+
+ echo -e "${GREEN}安装Python依赖...${RESET}"
+ pip install -r repo/requirements.txt
+
+ echo -e "${GREEN}同意协议...${RESET}"
+ touch repo/elua.confirmed
+
+ echo -e "${GREEN}创建系统服务...${RESET}"
+ cat > /etc/systemd/system/${SERVICE_NAME}.service < /etc/systemd/system/${SERVICE_NAME_WEB}.service < /etc/maimbot_install.conf
+ echo "INSTALL_DIR=${INSTALL_DIR}" >> /etc/maimbot_install.conf
+ echo "BRANCH=${BRANCH}" >> /etc/maimbot_install.conf
+
+ whiptail --title "🎉 安装完成" --msgbox "麦麦Bot安装完成！\n已创建系统服务：${SERVICE_NAME}，${SERVICE_NAME_WEB}\n\n使用以下命令管理服务：\n启动服务：systemctl start ${SERVICE_NAME}\n查看状态：systemctl status ${SERVICE_NAME}" 14 60
+}
+
+# ----------- 主执行流程 -----------
+# 检查root权限
+[[ $(id -u) -ne 0 ]] && {
+ echo -e "${RED}请使用root用户运行此脚本！${RESET}"
+ exit 1
+}
+
+# 如果已安装显示菜单
+if check_installed; then
+ load_install_info
+ show_menu
+else
+ run_installation
+ # 安装完成后询问是否启动
+ if whiptail --title "安装完成" --yesno "是否立即启动麦麦Bot服务？" 10 60; then
+ systemctl start ${SERVICE_NAME}
+ whiptail --msgbox "✅ 服务已启动！\n使用 systemctl status ${SERVICE_NAME} 查看状态" 10 60
+ fi
+fi
diff --git a/src/plugins/memory_system/manually_alter_memory.py b/src/plugins/memory_system/manually_alter_memory.py
new file mode 100644
index 000000000..e049bd2a9
--- /dev/null
+++ b/src/plugins/memory_system/manually_alter_memory.py
@@ -0,0 +1,319 @@
+# -*- coding: utf-8 -*-
+import os
+import sys
+import time
+from pathlib import Path
+import datetime
+from rich.console import Console
+
+from dotenv import load_dotenv
+
+
+'''
+我想 总有那么一个瞬间
+你会想和某天才变态少女助手一样
+往Bot的海马体里插上几个电极 不是吗
+
+Let's do some dirty job.
+'''
+
+# 获取当前文件的目录
+current_dir = Path(__file__).resolve().parent
+# 获取项目根目录（上三层目录）
+project_root = current_dir.parent.parent.parent
+# env.dev文件路径
+env_path = project_root / ".env.dev"
+
+# from chat.config import global_config
+root_path = os.path.abspath(os.path.join(os.path.dirname(__file__), "../../.."))
+sys.path.append(root_path)
+
+from src.common.logger import get_module_logger
+from src.common.database import db
+from src.plugins.memory_system.offline_llm import LLMModel
+
+logger = get_module_logger('mem_alter')
+console = Console()
+
+# 加载环境变量
+if env_path.exists():
+ logger.info(f"从 {env_path} 加载环境变量")
+ load_dotenv(env_path)
+else:
+ logger.warning(f"未找到环境变量文件: {env_path}")
+ logger.info("将使用默认配置")
+
+from memory_manual_build import Memory_graph, Hippocampus #海马体和记忆图
+
+# 查询节点信息
+def query_mem_info(memory_graph: Memory_graph):
+ while True:
+ query = input("\n请输入新的查询概念（输入'退出'以结束）：")
+ if query.lower() == '退出':
+ break
+
+ items_list = memory_graph.get_related_item(query)
+ if items_list:
+ have_memory = False
+ first_layer, second_layer = items_list
+ if first_layer:
+ have_memory = True
+ print("\n直接相关的记忆：")
+ for item in first_layer:
+ print(f"- {item}")
+ if second_layer:
+ have_memory = True
+ print("\n间接相关的记忆：")
+ for item in second_layer:
+ print(f"- {item}")
+ if not have_memory:
+ print("\n未找到相关记忆。")
+ else:
+ print("未找到相关记忆。")
+
+# 增加概念节点
+def add_mem_node(hippocampus: Hippocampus):
+ while True:
+ concept = input("请输入节点概念名:\n")
+ result = db.graph_data.nodes.count_documents({'concept': concept})
+
+ if result != 0:
+ console.print("[yellow]已存在名为“{concept}”的节点，行为已取消[/yellow]")
+ continue
+
+ memory_items = list()
+ while True:
+ context = input("请输入节点描述信息（输入'终止'以结束）")
+ if context.lower() == "终止": break
+ memory_items.append(context)
+
+ current_time = datetime.datetime.now().timestamp()
+ hippocampus.memory_graph.G.add_node(concept,
+ memory_items=memory_items,
+ created_time=current_time,
+ last_modified=current_time)
+# 删除概念节点（及连接到它的边）
+def remove_mem_node(hippocampus: Hippocampus):
+ concept = input("请输入节点概念名:\n")
+ result = db.graph_data.nodes.count_documents({'concept': concept})
+
+ if result == 0:
+ console.print(f"[red]不存在名为“{concept}”的节点[/red]")
+
+ edges = db.graph_data.edges.find({
+ '$or': [
+ {'source': concept},
+ {'target': concept}
+ ]
+ })
+
+ for edge in edges:
+ console.print(f"[yellow]存在边“{edge['source']} -> {edge['target']}”, 请慎重考虑[/yellow]")
+
+ console.print(f"[yellow]确定要移除名为“{concept}”的节点以及其相关边吗[/yellow]")
+ destory = console.input(f"[red]请输入“{concept}”以删除节点 其他输入将被视为取消操作[/red]\n")
+ if destory == concept:
+ hippocampus.memory_graph.G.remove_node(concept)
+ else:
+ logger.info("[green]删除操作已取消[/green]")
+# 增加节点间边
+def add_mem_edge(hippocampus: Hippocampus):
+ while True:
+ source = input("请输入 **第一个节点** 名称（输入'退出'以结束）：\n")
+ if source.lower() == "退出": break
+ if db.graph_data.nodes.count_documents({'concept': source}) == 0:
+ console.print(f"[yellow]“{source}”节点不存在，操作已取消。[/yellow]")
+ continue
+
+ target = input("请输入 **第二个节点** 名称：\n")
+ if db.graph_data.nodes.count_documents({'concept': target}) == 0:
+ console.print(f"[yellow]“{target}”节点不存在，操作已取消。[/yellow]")
+ continue
+
+ if source == target:
+ console.print(f"[yellow]试图创建“{source} <-> {target}”自环，操作已取消。[/yellow]")
+ continue
+
+ hippocampus.memory_graph.connect_dot(source, target)
+ edge = hippocampus.memory_graph.G.get_edge_data(source, target)
+ if edge['strength'] == 1:
+ console.print(f"[green]成功创建边“{source} <-> {target}”，默认权重1[/green]")
+ else:
+ console.print(f"[yellow]边“{source} <-> {target}”已存在，更新权重: {edge['strength']-1} <-> {edge['strength']}[/yellow]")
+# 删除节点间边
+def remove_mem_edge(hippocampus: Hippocampus):
+ while True:
+ source = input("请输入 **第一个节点** 名称（输入'退出'以结束）：\n")
+ if source.lower() == "退出": break
+ if db.graph_data.nodes.count_documents({'concept': source}) == 0:
+ console.print("[yellow]“{source}”节点不存在，操作已取消。[/yellow]")
+ continue
+
+ target = input("请输入 **第二个节点** 名称：\n")
+ if db.graph_data.nodes.count_documents({'concept': target}) == 0:
+ console.print("[yellow]“{target}”节点不存在，操作已取消。[/yellow]")
+ continue
+
+ if source == target:
+ console.print("[yellow]试图创建“{source} <-> {target}”自环，操作已取消。[/yellow]")
+ continue
+
+ edge = hippocampus.memory_graph.G.get_edge_data(source, target)
+ if edge is None:
+ console.print("[yellow]边“{source} <-> {target}”不存在，操作已取消。[/yellow]")
+ continue
+ else:
+ accept = console.input("[orange]请输入“确认”以确认删除操作（其他输入视为取消）[/orange]\n")
+ if accept.lower() == "确认":
+ hippocampus.memory_graph.G.remove_edge(source, target)
+ console.print(f"[green]边“{source} <-> {target}”已删除。[green]")
+
+# 修改节点信息
+def alter_mem_node(hippocampus: Hippocampus):
+ batchEnviroment = dict()
+ while True:
+ concept = input("请输入节点概念名（输入'终止'以结束）:\n")
+ if concept.lower() == "终止": break
+ _, node = hippocampus.memory_graph.get_dot(concept)
+ if node is None:
+ console.print(f"[yellow]“{concept}”节点不存在，操作已取消。[/yellow]")
+ continue
+
+ console.print("[yellow]注意，请确保你知道自己在做什么[/yellow]")
+ console.print("[yellow]你将获得一个执行任意代码的环境[/yellow]")
+ console.print("[red]你已经被警告过了。[/red]\n")
+
+ nodeEnviroment = {"concept": '<节点名>', 'memory_items': '<记忆文本数组>'}
+ console.print("[green]环境变量中会有env与batchEnv两个dict, env在切换节点时会清空, batchEnv在操作终止时才会清空[/green]")
+ console.print(f"[green] env 会被初始化为[/green]\n{nodeEnviroment}\n[green]且会在用户代码执行完毕后被提交 [/green]")
+ console.print("[yellow]为便于书写临时脚本，请手动在输入代码通过Ctrl+C等方式触发KeyboardInterrupt来结束代码执行[/yellow]")
+
+ # 拷贝数据以防操作炸了
+ nodeEnviroment = dict(node)
+ nodeEnviroment['concept'] = concept
+
+ while True:
+ userexec = lambda script, env, batchEnv: eval(script)
+ try:
+ command = console.input()
+ except KeyboardInterrupt:
+ # 稍微防一下小天才
+ try:
+ if isinstance(nodeEnviroment['memory_items'], list):
+ node['memory_items'] = nodeEnviroment['memory_items']
+ else:
+ raise Exception
+
+ except:
+ console.print("[red]我不知道你做了什么，但显然nodeEnviroment['memory_items']已经不是个数组了，操作已取消[/red]")
+ break
+
+ try:
+ userexec(command, nodeEnviroment, batchEnviroment)
+ except Exception as e:
+ console.print(e)
+ console.print("[red]自定义代码执行时发生异常，已捕获，请重试（可通过 console.print(locals()) 检查环境状态）[/red]")
+# 修改边信息
+def alter_mem_edge(hippocampus: Hippocampus):
+ batchEnviroment = dict()
+ while True:
+ source = input("请输入 **第一个节点** 名称（输入'终止'以结束）：\n")
+ if source.lower() == "终止": break
+ if hippocampus.memory_graph.get_dot(source) is None:
+ console.print(f"[yellow]“{source}”节点不存在，操作已取消。[/yellow]")
+ continue
+
+ target = input("请输入 **第二个节点** 名称：\n")
+ if hippocampus.memory_graph.get_dot(target) is None:
+ console.print(f"[yellow]“{target}”节点不存在，操作已取消。[/yellow]")
+ continue
+
+ edge = hippocampus.memory_graph.G.get_edge_data(source, target)
+ if edge is None:
+ console.print(f"[yellow]边“{source} <-> {target}”不存在，操作已取消。[/yellow]")
+ continue
+
+ console.print("[yellow]注意，请确保你知道自己在做什么[/yellow]")
+ console.print("[yellow]你将获得一个执行任意代码的环境[/yellow]")
+ console.print("[red]你已经被警告过了。[/red]\n")
+
+ edgeEnviroment = {"source": '<节点名>', "target": '<节点名>', 'strength': '<强度值,装在一个list里>'}
+ console.print("[green]环境变量中会有env与batchEnv两个dict, env在切换节点时会清空, batchEnv在操作终止时才会清空[/green]")
+ console.print(f"[green] env 会被初始化为[/green]\n{edgeEnviroment}\n[green]且会在用户代码执行完毕后被提交 [/green]")
+ console.print("[yellow]为便于书写临时脚本，请手动在输入代码通过Ctrl+C等方式触发KeyboardInterrupt来结束代码执行[/yellow]")
+
+ # 拷贝数据以防操作炸了
+ edgeEnviroment['strength'] = [edge["strength"]]
+ edgeEnviroment['source'] = source
+ edgeEnviroment['target'] = target
+
+ while True:
+ userexec = lambda script, env, batchEnv: eval(script)
+ try:
+ command = console.input()
+ except KeyboardInterrupt:
+ # 稍微防一下小天才
+ try:
+ if isinstance(edgeEnviroment['strength'][0], int):
+ edge['strength'] = edgeEnviroment['strength'][0]
+ else:
+ raise Exception
+
+ except:
+ console.print("[red]我不知道你做了什么，但显然edgeEnviroment['strength']已经不是个int了，操作已取消[/red]")
+ break
+
+ try:
+ userexec(command, edgeEnviroment, batchEnviroment)
+ except Exception as e:
+ console.print(e)
+ console.print("[red]自定义代码执行时发生异常，已捕获，请重试（可通过 console.print(locals()) 检查环境状态）[/red]")
+
+
+
+async def main():
+ start_time = time.time()
+
+ # 创建记忆图
+ memory_graph = Memory_graph()
+
+ # 创建海马体
+ hippocampus = Hippocampus(memory_graph)
+
+ # 从数据库同步数据
+ hippocampus.sync_memory_from_db()
+
+ end_time = time.time()
+ logger.info(f"\033[32m[加载海马体耗时: {end_time - start_time:.2f} 秒]\033[0m")
+
+ while True:
+ try:
+ query = int(input("请输入操作类型\n0 -> 查询节点; 1 -> 增加节点; 2 -> 移除节点; 3 -> 增加边; 4 -> 移除边;\n5 -> 修改节点; 6 -> 修改边; 其他任意输入 -> 退出\n"))
+ except:
+ query = -1
+
+ if query == 0:
+ query_mem_info(memory_graph)
+ elif query == 1:
+ add_mem_node(hippocampus)
+ elif query == 2:
+ remove_mem_node(hippocampus)
+ elif query == 3:
+ add_mem_edge(hippocampus)
+ elif query == 4:
+ remove_mem_edge(hippocampus)
+ elif query == 5:
+ alter_mem_node(hippocampus)
+ elif query == 6:
+ alter_mem_edge(hippocampus)
+ else:
+ print("已结束操作")
+ break
+
+ hippocampus.sync_memory_to_db()
+
+
+
+if __name__ == "__main__":
+ import asyncio
+ asyncio.run(main())
diff --git a/src/plugins/utils/statistic.py b/src/plugins/utils/statistic.py
index f1f53275e..6a5062567 100644
--- a/src/plugins/utils/statistic.py
+++ b/src/plugins/utils/statistic.py
@@ -50,7 +50,11 @@ class LLMStatistics:
"total_cost": 0.0,
"costs_by_user": defaultdict(float),
"costs_by_type": defaultdict(float),
- "costs_by_model": defaultdict(float)
+ "costs_by_model": defaultdict(float),
+ #新增token统计字段
+ "tokens_by_type": defaultdict(int),
+ "tokens_by_user": defaultdict(int),
+ "tokens_by_model": defaultdict(int),
}
cursor = db.llm_usage.find({
@@ -71,7 +75,11 @@ class LLMStatistics:
prompt_tokens = doc.get("prompt_tokens", 0)
completion_tokens = doc.get("completion_tokens", 0)
- stats["total_tokens"] += prompt_tokens + completion_tokens
+ total_tokens = prompt_tokens + completion_tokens # 根据数据库字段调整
+ stats["tokens_by_type"][request_type] += total_tokens
+ stats["tokens_by_user"][user_id] += total_tokens
+ stats["tokens_by_model"][model_name] += total_tokens
+ stats["total_tokens"] += total_tokens
cost = doc.get("cost", 0.0)
stats["total_cost"] += cost
@@ -98,31 +106,61 @@ class LLMStatistics:
}
def _format_stats_section(self, stats: Dict[str, Any], title: str) -> str:
- """格式化统计部分的输出
-
- Args:
- stats: 统计数据
- title: 部分标题
- """
+ """格式化统计部分的输出"""
output = []
- output.append(f"\n{title}")
- output.append("=" * len(title))
+
+ output.append("\n"+"-" * 84)
+ output.append(f"{title}")
+ output.append("-" * 84)
output.append(f"总请求数: {stats['total_requests']}")
if stats['total_requests'] > 0:
output.append(f"总Token数: {stats['total_tokens']}")
- output.append(f"总花费: ¥{stats['total_cost']:.4f}")
+ output.append(f"总花费: {stats['total_cost']:.4f}¥\n")
- output.append("\n按模型统计:")
+ data_fmt = "{:<32} {:>10} {:>14} {:>13.4f} ¥"
+
+ # 按模型统计
+ output.append("按模型统计:")
+ output.append(("模型名称 调用次数 Token总量 累计花费"))
for model_name, count in sorted(stats["requests_by_model"].items()):
+ tokens = stats["tokens_by_model"][model_name]
cost = stats["costs_by_model"][model_name]
- output.append(f"- {model_name}: {count}次 (花费: ¥{cost:.4f})")
+ output.append(data_fmt.format(
+ model_name[:32] + ".." if len(model_name) > 32 else model_name,
+ count,
+ tokens,
+ cost
+ ))
+ output.append("")
- output.append("\n按请求类型统计:")
+ # 按请求类型统计
+ output.append("按请求类型统计:")
+ output.append(("模型名称 调用次数 Token总量 累计花费"))
for req_type, count in sorted(stats["requests_by_type"].items()):
+ tokens = stats["tokens_by_type"][req_type]
cost = stats["costs_by_type"][req_type]
- output.append(f"- {req_type}: {count}次 (花费: ¥{cost:.4f})")
-
+ output.append(data_fmt.format(
+ req_type[:22] + ".." if len(req_type) > 24 else req_type,
+ count,
+ tokens,
+ cost
+ ))
+ output.append("")
+
+ # 修正用户统计列宽
+ output.append("按用户统计:")
+ output.append(("模型名称 调用次数 Token总量 累计花费"))
+ for user_id, count in sorted(stats["requests_by_user"].items()):
+ tokens = stats["tokens_by_user"][user_id]
+ cost = stats["costs_by_user"][user_id]
+ output.append(data_fmt.format(
+ user_id[:22], # 不再添加省略号，保持原始ID
+ count,
+ tokens,
+ cost
+ ))
+
return "\n".join(output)
def _save_statistics(self, all_stats: Dict[str, Dict[str, Any]]):
@@ -131,7 +169,7 @@ class LLMStatistics:
output = []
output.append(f"LLM请求统计报告 (生成时间: {current_time})")
- output.append("=" * 50)
+
# 添加各个时间段的统计
sections = [