feat: 支持插件市场

通过 Commit Hash 更新和仪表盘 UI 优化

.codecov.yml

@@ -0,0 +1,3 @@
+comment:
+    layout: "condensed_header, condensed_files, condensed_footer"
+    hide_project_coverage: TRUE

.coveragerc

@@ -0,0 +1,5 @@
+[run]
+omit =
+    */site-packages/*
+    */dist-packages/*
+    your_package_name/tests/*

.dockerignore

@@ -0,0 +1,18 @@
+# Covers JetBrains IDEs: IntelliJ, RubyMine, PhpStorm, AppCode, PyCharm, CLion, Android Studio and WebStorm
+# Reference: https://intellij-support.jetbrains.com/hc/en-us/articles/206544839
+# github acions
+.github/
+.*ignore
+.git/
+# User-specific stuff
+.idea/
+# Byte-compiled / optimized / DLL files
+__pycache__/
+# Environments
+.env
+.venv
+env/
+venv*/
+ENV/
+.conda/
+README*.md

.github/ISSUE_TEMPLATE/bug-report.yml

@@ -0,0 +1,82 @@
+name: '🐛 报告 Bug'
+title: '[Bug]'
+description: 提交报告帮助我们改进。
+labels: [ 'bug' ]
+body:
+  - type: markdown
+    attributes:
+      value: |
+        感谢您抽出时间报告问题！请准确解释您的问题。如果可能，请提供一个可复现的片段（这有助于更快地解决问题）。
+  - type: textarea
+    attributes:
+      label: 发生了什么
+      description: 描述你遇到的异常
+      placeholder: >
+        一个清晰且具体的描述这个异常是什么。
+    validations:
+      required: true
+
+  - type: textarea
+    attributes:
+      label: 如何复现？
+      description: >
+        复现该问题的步骤
+      placeholder: >
+        如: 1. 打开 '...'
+    validations:
+      required: true
+
+  - type: textarea
+    attributes:
+      label: AstrBot 版本与部署方式
+      description: >
+        请提供您的 AstrBot 版本和部署方式。
+      placeholder: >
+        如: 3.1.8 Docker, 3.1.7 Windows启动器
+    validations:
+      required: true
+
+  - type: dropdown
+    attributes:
+      label: 操作系统
+      description: |
+        你在哪个操作系统上遇到了这个问题？
+      multiple: false
+      options:
+        - 'Windows'
+        - 'macOS'
+        - 'Linux'
+        - 'Other'
+        - 'Not sure'
+    validations:
+      required: true
+
+  - type: textarea
+    attributes:
+      label: 额外信息
+      description: >
+        任何额外信息，如报错日志、截图等。
+      placeholder: >
+        请提供完整的报错日志或截图。
+    validations:
+      required: true
+
+  - type: checkboxes
+    attributes:
+      label: 你愿意提交 PR 吗？
+      description: >
+        这绝对不是必需的，但我们很乐意在贡献过程中为您提供指导特别是如果你已经很好地理解了如何实现修复。
+      options:
+        - label: 是的，我愿意提交 PR!
+
+  - type: checkboxes
+    attributes:
+      label: Code of Conduct
+      options:
+        - label: >
+            我已阅读并同意遵守该项目的 [行为准则](https://docs.github.com/zh/site-policy/github-terms/github-community-code-of-conduct)。
+          required: true
+
+  - type: markdown
+    attributes:
+      value: "感谢您填写我们的表单！"

.github/ISSUE_TEMPLATE/feature-request.yml

@@ -0,0 +1,42 @@
+
+name: '🎉 功能建议'
+title: "[Feature]"
+description: 提交建议帮助我们改进。
+labels: [ "enhancement" ]
+body:
+  - type: markdown
+    attributes:
+      value: |
+        感谢您抽出时间提出新功能建议，请准确解释您的想法。
+
+  - type: textarea
+    attributes:
+      label: 描述
+      description: 简短描述您的功能建议。
+
+  - type: textarea
+    attributes:
+      label: 使用场景
+      description: 你想要发生什么？
+      placeholder: >
+        一个清晰且具体的描述这个功能的使用场景。
+
+  - type: checkboxes
+    attributes:
+      label: 你愿意提交PR吗?
+      description: >
+        这不是必须的，但我们欢迎您的贡献。
+      options:
+        - label: 是的, 我愿意提交PR!
+
+  - type: checkboxes
+    attributes:
+      label: Code of Conduct
+      options:
+        - label: >
+            我已阅读并同意遵守该项目的 [行为准则](https://docs.github.com/zh/site-policy/github-terms/github-community-code-of-conduct)。
+          required: true
+
+  - type: markdown
+    attributes:
+      value: "感谢您填写我们的表单！"

.github/PULL_REQUEST_TEMPLATE.md

@@ -0,0 +1,10 @@
+<!-- 如果有的话，指定这个 PR 要解决的 ISSUE -->
+修复了 #XYZ
+
+### Motivation
+
+<!--解释为什么要改动-->
+
+### Modifications
+
+<!--简单解释你的改动-->

.github/workflows/codeql.yml

@@ -0,0 +1,93 @@
+# For most projects, this workflow file will not need changing; you simply need
+# to commit it to your repository.
+#
+# You may wish to alter this file to override the set of languages analyzed,
+# or to provide custom queries or build logic.
+#
+# ******** NOTE ********
+# We have attempted to detect the languages in your repository. Please check
+# the `language` matrix defined below to confirm you have the correct set of
+# supported CodeQL languages.
+#
+name: "CodeQL"
+
+on:
+  push:
+    branches: [ "master" ]
+  pull_request:
+    branches: [ "master" ]
+  schedule:
+    - cron: '21 15 * * 5'
+
+jobs:
+  analyze:
+    name: Analyze (${{ matrix.language }})
+    # Runner size impacts CodeQL analysis time. To learn more, please see:
+    #   - https://gh.io/recommended-hardware-resources-for-running-codeql
+    #   - https://gh.io/supported-runners-and-hardware-resources
+    #   - https://gh.io/using-larger-runners (GitHub.com only)
+    # Consider using larger runners or machines with greater resources for possible analysis time improvements.
+    runs-on: ${{ (matrix.language == 'swift' && 'macos-latest') || 'ubuntu-latest' }}
+    timeout-minutes: ${{ (matrix.language == 'swift' && 120) || 360 }}
+    permissions:
+      # required for all workflows
+      security-events: write
+
+      # required to fetch internal or private CodeQL packs
+      packages: read
+
+      # only required for workflows in private repositories
+      actions: read
+      contents: read
+
+    strategy:
+      fail-fast: false
+      matrix:
+        include:
+        - language: python
+          build-mode: none
+        # CodeQL supports the following values keywords for 'language': 'c-cpp', 'csharp', 'go', 'java-kotlin', 'javascript-typescript', 'python', 'ruby', 'swift'
+        # Use `c-cpp` to analyze code written in C, C++ or both
+        # Use 'java-kotlin' to analyze code written in Java, Kotlin or both
+        # Use 'javascript-typescript' to analyze code written in JavaScript, TypeScript or both
+        # To learn more about changing the languages that are analyzed or customizing the build mode for your analysis,
+        # see https://docs.github.com/en/code-security/code-scanning/creating-an-advanced-setup-for-code-scanning/customizing-your-advanced-setup-for-code-scanning.
+        # If you are analyzing a compiled language, you can modify the 'build-mode' for that language to customize how
+        # your codebase is analyzed, see https://docs.github.com/en/code-security/code-scanning/creating-an-advanced-setup-for-code-scanning/codeql-code-scanning-for-compiled-languages
+    steps:
+    - name: Checkout repository
+      uses: actions/checkout@v4
+
+    # Initializes the CodeQL tools for scanning.
+    - name: Initialize CodeQL
+      uses: github/codeql-action/init@v3
+      with:
+        languages: ${{ matrix.language }}
+        build-mode: ${{ matrix.build-mode }}
+        # If you wish to specify custom queries, you can do so here or in a config file.
+        # By default, queries listed here will override any specified in a config file.
+        # Prefix the list here with "+" to use these queries and those in the config file.
+
+        # For more details on CodeQL's query packs, refer to: https://docs.github.com/en/code-security/code-scanning/automatically-scanning-your-code-for-vulnerabilities-and-errors/configuring-code-scanning#using-queries-in-ql-packs
+        # queries: security-extended,security-and-quality
+
+    # If the analyze step fails for one of the languages you are analyzing with
+    # "We were unable to automatically build your code", modify the matrix above
+    # to set the build mode to "manual" for that language. Then modify this step
+    # to build your code.
+    # ℹ️ Command-line programs to run using the OS shell.
+    # 📚 See https://docs.github.com/en/actions/using-workflows/workflow-syntax-for-github-actions#jobsjob_idstepsrun
+    - if: matrix.build-mode == 'manual'
+      shell: bash
+      run: |
+        echo 'If you are using a "manual" build mode for one or more of the' \
+          'languages you are analyzing, replace this with the commands to build' \
+          'your code, for example:'
+        echo '  make bootstrap'
+        echo '  make release'
+        exit 1
+
+    - name: Perform CodeQL Analysis
+      uses: github/codeql-action/analyze@v3
+      with:
+        category: "/language:${{matrix.language}}"

.github/workflows/coverage_test.yml

@@ -0,0 +1,39 @@
+name: Run tests and upload coverage
+
+on: 
+  push
+
+jobs:
+  test:
+    name: Run tests and collect coverage
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v4
+        with:
+          fetch-depth: 0
+
+      - name: Set up Python
+        uses: actions/setup-python@v4
+      
+      - name: Install dependencies
+        run: |
+          python -m pip install --upgrade pip
+          pip install -r requirements.txt
+          pip install pytest pytest-cov pytest-asyncio
+          mkdir data
+          mkdir data/plugins
+          mkdir data/config
+          mkdir temp
+
+      - name: Run tests
+        run: |
+          export LLM_MODEL=${{ secrets.LLM_MODEL }}
+          export OPENAI_API_BASE=${{ secrets.OPENAI_API_BASE }}
+          export OPENAI_API_KEY=${{ secrets.OPENAI_API_KEY }}
+          PYTHONPATH=./ pytest --cov=. tests/ -v
+
+      - name: Upload results to Codecov
+        uses: codecov/codecov-action@v4
+        with:
+          token: ${{ secrets.CODECOV_TOKEN }}

.github/workflows/docker-image.yml

@@ -0,0 +1,42 @@
+name: Docker Image CI/CD
+
+on:
+  release:
+    types: [published]
+  workflow_dispatch:
+
+jobs:
+  publish-docker:
+    runs-on: ubuntu-latest
+
+    steps:
+      - name: 拉取源码
+        uses: actions/checkout@v3
+        with:
+          fetch-depth: 1
+
+      - name: 设置 QEMU
+        uses: docker/setup-qemu-action@v3
+
+      - name: 设置 Docker Buildx
+        uses: docker/setup-buildx-action@v3
+
+      - name: 登录到 DockerHub
+        uses: docker/login-action@v3
+        with:
+          username: ${{ secrets.DOCKER_HUB_USERNAME }}
+          password: ${{ secrets.DOCKER_HUB_PASSWORD }}
+
+      - name: 构建和推送 Docker hub
+        uses: docker/build-push-action@v6
+        with:
+          context: .
+          platforms: linux/amd64,linux/arm64
+          push: true
+          tags: |
+            ${{ secrets.DOCKER_HUB_USERNAME }}/astrbot:latest
+            ${{ secrets.DOCKER_HUB_USERNAME }}/astrbot:${{ github.event.release.tag_name }}
+
+      - name: Post build notifications
+        run: echo "Docker image has been built and pushed successfully"
+

.github/workflows/stale.yml

@@ -0,0 +1,27 @@
+# This workflow warns and then closes issues and PRs that have had no activity for a specified amount of time.
+#
+# You can adjust the behavior by modifying this file.
+# For more information, see:
+# https://github.com/actions/stale
+name: Mark stale issues and pull requests
+
+on:
+  schedule:
+  - cron: '21 23 * * *'
+
+jobs:
+  stale:
+
+    runs-on: ubuntu-latest
+    permissions:
+      issues: write
+      pull-requests: write
+
+    steps:
+    - uses: actions/stale@v5
+      with:
+        repo-token: ${{ secrets.GITHUB_TOKEN }}
+        stale-issue-message: 'Stale issue message'
+        stale-pr-message: 'Stale pull request message'
+        stale-issue-label: 'no-issue-activity'
+        stale-pr-label: 'no-pr-activity'

.gitignore

@@ -0,0 +1,14 @@
+__pycache__
+botpy.log
+.vscode
+data_v2.db
+configs/session
+configs/config.yaml
+**/.DS_Store
+temp
+cmd_config.json
+data/*
+cookies.json
+logs/
+addons/plugins
+.coverage

.vscode/settings.json

@@ -1,3 +0,0 @@
-{
-    "python.analysis.typeCheckingMode": "basic"
-}

CODE_OF_CONDUCT.md

@@ -0,0 +1,128 @@
+# Contributor Covenant Code of Conduct
+
+## Our Pledge
+
+We as members, contributors, and leaders pledge to make participation in our
+community a harassment-free experience for everyone, regardless of age, body
+size, visible or invisible disability, ethnicity, sex characteristics, gender
+identity and expression, level of experience, education, socio-economic status,
+nationality, personal appearance, race, religion, or sexual identity
+and orientation.
+
+We pledge to act and interact in ways that contribute to an open, welcoming,
+diverse, inclusive, and healthy community.
+
+## Our Standards
+
+Examples of behavior that contributes to a positive environment for our
+community include:
+
+* Demonstrating empathy and kindness toward other people
+* Being respectful of differing opinions, viewpoints, and experiences
+* Giving and gracefully accepting constructive feedback
+* Accepting responsibility and apologizing to those affected by our mistakes,
+  and learning from the experience
+* Focusing on what is best not just for us as individuals, but for the
+  overall community
+
+Examples of unacceptable behavior include:
+
+* The use of sexualized language or imagery, and sexual attention or
+  advances of any kind
+* Trolling, insulting or derogatory comments, and personal or political attacks
+* Public or private harassment
+* Publishing others' private information, such as a physical or email
+  address, without their explicit permission
+* Other conduct which could reasonably be considered inappropriate in a
+  professional setting
+
+## Enforcement Responsibilities
+
+Community leaders are responsible for clarifying and enforcing our standards of
+acceptable behavior and will take appropriate and fair corrective action in
+response to any behavior that they deem inappropriate, threatening, offensive,
+or harmful.
+
+Community leaders have the right and responsibility to remove, edit, or reject
+comments, commits, code, wiki edits, issues, and other contributions that are
+not aligned to this Code of Conduct, and will communicate reasons for moderation
+decisions when appropriate.
+
+## Scope
+
+This Code of Conduct applies within all community spaces, and also applies when
+an individual is officially representing the community in public spaces.
+Examples of representing our community include using an official e-mail address,
+posting via an official social media account, or acting as an appointed
+representative at an online or offline event.
+
+## Enforcement
+
+Instances of abusive, harassing, or otherwise unacceptable behavior may be
+reported to the community leaders responsible for enforcement at
+SoulterL@outlook.com.
+All complaints will be reviewed and investigated promptly and fairly.
+
+All community leaders are obligated to respect the privacy and security of the
+reporter of any incident.
+
+## Enforcement Guidelines
+
+Community leaders will follow these Community Impact Guidelines in determining
+the consequences for any action they deem in violation of this Code of Conduct:
+
+### 1. Correction
+
+**Community Impact**: Use of inappropriate language or other behavior deemed
+unprofessional or unwelcome in the community.
+
+**Consequence**: A private, written warning from community leaders, providing
+clarity around the nature of the violation and an explanation of why the
+behavior was inappropriate. A public apology may be requested.
+
+### 2. Warning
+
+**Community Impact**: A violation through a single incident or series
+of actions.
+
+**Consequence**: A warning with consequences for continued behavior. No
+interaction with the people involved, including unsolicited interaction with
+those enforcing the Code of Conduct, for a specified period of time. This
+includes avoiding interactions in community spaces as well as external channels
+like social media. Violating these terms may lead to a temporary or
+permanent ban.
+
+### 3. Temporary Ban
+
+**Community Impact**: A serious violation of community standards, including
+sustained inappropriate behavior.
+
+**Consequence**: A temporary ban from any sort of interaction or public
+communication with the community for a specified period of time. No public or
+private interaction with the people involved, including unsolicited interaction
+with those enforcing the Code of Conduct, is allowed during this period.
+Violating these terms may lead to a permanent ban.
+
+### 4. Permanent Ban
+
+**Community Impact**: Demonstrating a pattern of violation of community
+standards, including sustained inappropriate behavior,  harassment of an
+individual, or aggression toward or disparagement of classes of individuals.
+
+**Consequence**: A permanent ban from any sort of public interaction within
+the community.
+
+## Attribution
+
+This Code of Conduct is adapted from the [Contributor Covenant][homepage],
+version 2.0, available at
+https://www.contributor-covenant.org/version/2/0/code_of_conduct.html.
+
+Community Impact Guidelines were inspired by [Mozilla's code of conduct
+enforcement ladder](https://github.com/mozilla/diversity).
+
+[homepage]: https://www.contributor-covenant.org
+
+For answers to common questions about this code of conduct, see the FAQ at
+https://www.contributor-covenant.org/faq. Translations are available at
+https://www.contributor-covenant.org/translations.

Dockerfile

@@ -0,0 +1,20 @@
+FROM python:3.10-slim
+WORKDIR /AstrBot
+
+COPY . /AstrBot/
+
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    gcc \
+    build-essential \
+    python3-dev \
+    libffi-dev \
+    libssl-dev \
+    && apt-get clean \
+    && rm -rf /var/lib/apt/lists/*
+
+RUN python -m pip install -r requirements.txt
+
+EXPOSE 6185 
+EXPOSE 6186
+
+CMD [ "python", "main.py" ]

README.md

@@ -1,115 +1,67 @@
-# 2023/3/2 已支持官方ChatGPT API，性能更好！！！
+<p align="center">
+
+<img width="750" alt="image" src="https://github.com/Soulter/AstrBot/assets/37870767/c6f057d9-46d7-4144-8116-00a962941746">
+
+</p>
+<div align="center">
+
+[![GitHub release (latest by date)](https://img.shields.io/github/v/release/Soulter/AstrBot)](https://github.com/Soulter/AstrBot/releases/latest)
+<img src="https://img.shields.io/badge/python-3.9+-blue.svg" alt="python">
+<a href="https://hub.docker.com/r/soulter/astrbot"><img alt="Docker pull" src="https://img.shields.io/docker/pulls/soulter/astrbot.svg"/></a>
+[![codecov](https://codecov.io/gh/Soulter/AstrBot/graph/badge.svg?token=FF3P5967B8)](https://codecov.io/gh/Soulter/AstrBot)
+<a href="https://qm.qq.com/cgi-bin/qm/qr?k=EYGsuUTfe00_iOu9JTXS7_TEpMkXOvwv&jump_from=webapi&authKey=uUEMKCROfsseS+8IzqPjzV3y1tzy4AkykwTib2jNkOFdzezF9s9XknqnIaf3CDft">
+<img alt="Static Badge" src="https://img.shields.io/badge/QQ群-322154837-purple">
+</a>
+
+<a href="https://astrbot.soulter.top/docs/main">快速开始</a> ｜
+<a href="https://github.com/Soulter/AstrBot/issues">问题提交</a> ｜
+<a href="https://astrbot.soulter.top/docs/develop/plugin4p">插件开发</a>
+</div>
+
+## 🛠️ 功能
+
+🌍 支持的消息平台
+- QQ 群、QQ 频道（OneBot、QQ 官方接口）
+- Telegram（[astrbot_plugin_telegram](https://github.com/Soulter/astrbot_plugin_telegram) 插件）
+
+🌍 支持的大模型/底座：
+
+- OpenAI GPT、DallE 系列
+- Claude（由[LLMs插件](https://github.com/Soulter/llms)支持）
+- HuggingChat（由[LLMs插件](https://github.com/Soulter/llms)支持）
+- Gemini（由[LLMs插件](https://github.com/Soulter/llms)支持）
+- Ollama 
+- 几乎所有已知模型（可接入 [OneAPI](https://astrbot.soulter.top/docs/docs/adavanced/one-api)）
+
+🌍 机器人支持的能力一览：
+- 大模型对话、人格、网页搜索
+- 可视化仪表盘
+- 同时处理多平台消息
+- 精确到个人的会话隔离
+- 插件支持
+- 文本转图片回复（Markdown）
+
+## 🧩 插件
+
+有关插件的使用和列表请移步：[AstrBot 文档 - 插件](https://astrbot.soulter.top/docs/get-started/plugin)
+
+## 云部署
+
+[![Run on Repl.it](https://repl.it/badge/github/Soulter/AstrBot)](https://repl.it/github/Soulter/AstrBot)
+
+## ❤️ 贡献
+
+欢迎任何 Issues/Pull Requests！只需要将你的更改提交到此项目 ：)
+
+对于新功能的添加，请先通过 Issue 进行讨论。
+
+## 🔭 展望
+
+- [ ] 更多、更开放的 LLM Agent 能力
+
+## ✨ Demo
+
+<img width="900" alt="image" src="https://github.com/Soulter/AstrBot/assets/37870767/824d1ff3-7b85-481c-b795-8e62dedb9fd7">
 
 
-## ⭐体验
 
-使用手机QQ扫码加入QQ频道(频道名: GPT机器人 | 频道号: x42d56aki2)
-
-![image](https://user-images.githubusercontent.com/37870767/221722540-d4b0fada-4fa2-4063-bb00-1bb3e4d9a747.png)
-
-**Windows用户推荐Windows一键安装，请前往Release下载最新版本**
-
-**详细部署教程链接**https://soulter.top/posts/qpdg.html
-
-有任何问题请加频道反馈。
-
-## ⭐功能：
-
-### 基本功能
-<details> 
- <summary>✅ 回复符合上下文</summary>
-
-   -  程序向API发送近多次对话内容，模型根据上下文生成回复
-
-   -  你可在`configs/config.yaml`中修改`total_token_limit`来近似控制缓存大小。
- </details> 
-
-<details> 
- <summary>✅ 超额自动切换</summary>
-
-   -  超额时，程序自动切换openai的key，方便快捷
-   
-</details>
-
-<details> 
-
- <summary>✅ 支持统计频道、消息数量等信息</summary> 
-
-   -  实现了简单的统计功能
-
- </details>
-
-<details> 
- <summary>✅ 多并发处理，回复速度快</summary> 
-  
-   -  使用了协程，理论最高可以支持每个子频道每秒回复5条信息
-  
- </details>
-
-<details>
- <summary>✅ 持久化转储历史记录，重启不丢失</summary> 
-
-   -  使用内置的sqlite数据库存储历史记录到本地
-
-   -  方式为定时转储，可在`config.yaml`下修改`dump_history_interval`来修改间隔时间，单位为分钟。
-  
- </details>
-
-<details> 
- <summary>✅ 支持多种指令控制</summary> 
-  
-   -  详见下方`指令功能`
-  
- </details>
-
-<details>
-<summary>✅ 官方API，稳定</summary>
-
-   -  不使用ChatGPT逆向接口，而使用官方API接口，稳定方便。
-
-   -  QQ频道机器人框架为QQ官方开源的框架，稳定。
-
-</details>
-
-> 关于token：token就相当于是AI中的单词数（但是不等于单词数），`text-davinci-003`模型中最大可以支持`4097`个token。在发送信息时，这个机器人会将用户的历史聊天记录打包发送给ChatGPT，因此，`token`也会相应的累加，为了保证聊天的上下文的逻辑性，就有了缓存token。
-### 指令功能
-需要先`@`机器人之后再输入指令
-- `/reset`重置prompt
-- `/his`查看历史记录（每个用户都有独立的会话）
-- `/his [页码数]`查看不同页码的历史记录。例如`/his 2`查看第2页
-- `/token`查看当前缓存的总token数
-- `/count` 查看统计
-- `/status` 查看chatGPT的配置
-- `/help` 查看帮助
-- `/key` 动态添加key
-- `/set` 人格设置面板
-
-## 📰使用方法：
-
-**详细部署教程链接**https://soulter.top/posts/qpdg.html
-
-### 安装第三方库
-
-使用Python的pip工具安装
-- `qq-botpy` （QQ频道官方Python SDK）
-- `openai` (OpenAI Python SDK)
-
-```shell
-pip install -r requirements.txt
-```
-> ⚠注意，由于qq-botpy库需要运行在`Python 3.8+`的版本上，因此本项目也需要在此之上运行
-
-### 配置
-
-- 获得 OpenAI的key [OpenAI](https://beta.openai.com/)
-- 获得 QQ开放平台下QQ频道机器人的token和appid [QQ开放平台](https://q.qq.com/)，一个QQ频道机器人（很容易创建~）
-- 在configs/config.yaml下进行配置
-
-### 启动
-- 启动main.py
-
-
-## DEMO
-![1.jpg](screenshots/1.jpg)
-![3.jpg](screenshots/3.jpg)
-![2.jpg](screenshots/2.jpg)

addons/revChatGPT/revchatgpt.py

@@ -1,29 +0,0 @@
-from revChatGPT.V1 import Chatbot
-
-class revChatGPT:
-    def __init__(self, config):
-        
-        print("[RevChatGPT] 逆向库初始化："+str(config))
-        config['password'] = str(config['password'])
-        self.chatbot = Chatbot(config=config)
-
-    def chat(self, prompt):
-        resp = ''
-
-        err_count = 0
-        retry_count = 5
-
-        while err_count < retry_count:
-            try:
-                for data in self.chatbot.ask(prompt):
-                    resp = data["message"]
-                break
-            except BaseException as e:
-                print(e)
-                print("[RevChatGPT] 请求出现了一些问题, 正在重试。次数"+str(err_count))
-                err_count += 1
-                if err_count == retry_count:
-                    raise e
-        
-        print("[RevChatGPT] "+str(resp))
-        return resp

astrbot/bootstrap.py

@@ -0,0 +1,144 @@
+import asyncio
+import traceback
+import os
+from astrbot.message.handler import MessageHandler
+from astrbot.db.sqlite import SQLiteDatabase
+from dashboard.server import AstrBotDashboard
+from model.command.manager import CommandManager
+from model.command.internal_handler import InternalCommandHandler
+from model.plugin.manager import PluginManager
+from model.platform.manager import PlatformManager
+from typing import Union
+from type.types import Context
+from type.config import VERSION, DB_PATH
+from logging import Logger
+from util.cmd_config import AstrBotConfig, try_migrate
+from util.metrics import MetricUploader
+from util.updator.astrbot_updator import AstrBotUpdator
+from util.log import LogManager
+
+
+logger: Logger = LogManager.GetLogger(log_name='astrbot')
+
+class AstrBotBootstrap():
+    def __init__(self) -> None:
+        self.context = Context()
+        # load configs and ensure the backward compatibility
+        try_migrate()
+        self.config_helper = AstrBotConfig()
+        self.context.config_helper = self.config_helper
+        # set log queue handler 
+        LogManager.set_queue_handler(logger, self.context._log_queue)
+        logger.info("AstrBot v" + VERSION)
+        # set log level
+        logger.setLevel(self.config_helper.log_level)
+        # apply proxy settings
+        http_proxy = self.context.config_helper.http_proxy
+        https_proxy = self.context.config_helper.https_proxy
+        if http_proxy:
+            os.environ['HTTP_PROXY'] = http_proxy
+        if https_proxy:
+            os.environ['HTTPS_PROXY'] = https_proxy
+        os.environ['NO_PROXY'] = 'https://api.sgroup.qq.com'
+        
+        if http_proxy and https_proxy:
+            logger.info(f"使用代理: {http_proxy}, {https_proxy}")
+        else:
+            logger.info("未使用代理。")
+            
+        self.test_mode = os.environ.get('TEST_MODE', 'off') == 'on'
+
+        # set t2i endpoint
+        if self.context.config_helper.t2i_endpoint:
+            self.context.image_renderer.set_network_endpoint(
+                self.context.config_helper.t2i_endpoint
+            )
+    
+    async def run(self):
+        self.command_manager = CommandManager()
+        self.plugin_manager = PluginManager(self.context)
+        self.updator = AstrBotUpdator()
+        self.cmd_handler = InternalCommandHandler(self.command_manager, self.plugin_manager)
+        self.db_helper = SQLiteDatabase(DB_PATH)
+        
+        # load llm provider
+        self.load_llm()
+        
+        self.message_handler = MessageHandler(self.context, self.command_manager, self.db_helper)
+        self.platfrom_manager = PlatformManager(self.context, self.message_handler)
+        self.dashboard = AstrBotDashboard(self.context, 
+                                          plugin_manager=self.plugin_manager, 
+                                          astrbot_updator=self.updator,
+                                          db_helper=self.db_helper)
+        self.metrics_uploader = MetricUploader(self.context, self.db_helper)
+        
+        self.context.metrics_uploader = self.metrics_uploader
+        self.context.updator = self.updator
+        self.context.plugin_updator = self.plugin_manager.updator
+        self.context.message_handler = self.message_handler
+        self.context.command_manager = self.command_manager
+
+        # load dashboard
+        dashboard_server_task = asyncio.create_task(self.dashboard.run(), name="dashboard")
+
+        if self.test_mode:
+            return
+        
+        # load plugins, plugins' commands.
+        self.load_plugins()
+        self.command_manager.register_from_pcb(self.context.plugin_command_bridge)
+        
+        # load platforms
+        platform_tasks = self.load_platform()
+        # load metrics uploader
+        metrics_upload_task = asyncio.create_task(self.metrics_uploader.upload_metrics(), name="metrics-uploader")
+        
+        log_task = asyncio.create_task(self.dashboard.lr._receive_log_task(), name="log")
+        tasks = [metrics_upload_task, dashboard_server_task, log_task, *platform_tasks, *self.context.ext_tasks]
+        tasks = [self.handle_task(task) for task in tasks]
+        await asyncio.gather(*tasks)
+
+    async def handle_task(self, task: Union[asyncio.Task, asyncio.Future]):
+        while True:
+            try:
+                result = await task
+                return result
+            except asyncio.CancelledError:
+                logger.info(f"{task.get_name()} 任务已取消。")
+                return
+            except Exception as e:
+                logger.error(traceback.format_exc())
+                logger.error(f"{task.get_name()} 任务发生错误。")
+                return
+    
+    def load_llm(self):
+        f = False
+        llms = self.context.config_helper.llm
+        logger.info(f"加载 {len(llms)} 个 LLM Provider...")
+        for llm in llms:
+            if llm.enable:
+                if llm.name == "openai":
+                    if not llm.key or not llm.enable:
+                        logger.warning("没有开启 LLM Provider 或 API Key 未填写。")
+                        continue
+                    self.load_openai(llm)
+                    f = True
+                    logger.info(f"已启用 LLM Provider(OpenAI API): {llm.name}。")
+        if f:
+            from model.command.openai_official_handler import OpenAIOfficialCommandHandler
+            self.openai_command_handler = OpenAIOfficialCommandHandler(self.command_manager)
+            self.openai_command_handler.set_provider(self.context.llms[0].llm_instance)
+
+    def load_openai(self, llm_config):
+        from model.provider.openai_official import ProviderOpenAIOfficial
+        inst = ProviderOpenAIOfficial(llm_config, self.db_helper)
+        self.context.register_provider("internal_openai", inst)
+    
+    def load_plugins(self):
+        self.plugin_manager.plugin_reload()
+    
+    def load_platform(self):
+        platforms = self.platfrom_manager.load_platforms()
+        if not platforms:
+            logger.warning("未启用任何消息平台。")
+        return platforms

astrbot/db/__init__.py

@@ -0,0 +1,64 @@
+import abc
+from dataclasses import dataclass
+from typing import List
+from astrbot.db.po import Stats, LLMHistory
+
+@dataclass
+class BaseDatabase(abc.ABC):
+    '''
+    数据库基类
+    '''
+    def __init__(self) -> None:
+        pass
+    
+    def insert_base_metrics(self, metrics: dict):
+        '''插入基础指标数据'''
+        self.insert_platform_metrics(metrics['platform_stats'])
+        self.insert_plugin_metrics(metrics['plugin_stats'])
+        self.insert_command_metrics(metrics['command_stats'])
+        self.insert_llm_metrics(metrics['llm_stats'])
+    
+    @abc.abstractmethod
+    def insert_platform_metrics(self, metrics: dict):
+        '''插入平台指标数据'''
+        raise NotImplementedError
+    
+    @abc.abstractmethod
+    def insert_plugin_metrics(self, metrics: dict):
+        '''插入插件指标数据'''
+        raise NotImplementedError
+    
+    @abc.abstractmethod
+    def insert_command_metrics(self, metrics: dict):
+        '''插入指令指标数据'''
+        raise NotImplementedError
+    
+    @abc.abstractmethod
+    def insert_llm_metrics(self, metrics: dict):
+        '''插入 LLM 指标数据'''
+        raise NotImplementedError
+    
+    @abc.abstractmethod
+    def update_llm_history(self, session_id: str, content: str):
+        '''更新 LLM 历史记录。当不存在 session_id 时插入'''
+        raise NotImplementedError
+    
+    @abc.abstractmethod
+    def get_llm_history(self, session_id: str = None) -> List[LLMHistory]:
+        '''获取 LLM 历史记录, 如果 session_id 为 None, 返回所有'''
+        raise NotImplementedError
+    
+    @abc.abstractmethod
+    def get_base_stats(self, offset_sec: int = 86400) -> Stats:
+        '''获取基础统计数据'''
+        raise NotImplementedError
+    
+    @abc.abstractmethod
+    def get_total_message_count(self) -> int:
+        '''获取总消息数'''
+        raise NotImplementedError
+    
+    @abc.abstractmethod
+    def get_grouped_base_stats(self, offset_sec: int = 86400) -> Stats:
+        '''获取基础统计数据(合并)'''
+        raise NotImplementedError

astrbot/db/po.py

@@ -0,0 +1,42 @@
+'''指标数据'''
+
+from dataclasses import dataclass, field
+# default_factory
+from typing import List
+
+@dataclass
+class Platform():
+    name: str
+    count: int
+    timestamp: int
+    
+@dataclass
+class Provider():
+    name: str
+    count: int
+    timestamp: int
+    
+@dataclass
+class Plugin():
+    name: str
+    count: int
+    timestamp: int
+    
+@dataclass
+class Command():
+    name: str
+    count: int
+    timestamp: int
+
+@dataclass
+class Stats():
+    platform: List[Platform] = field(default_factory=list)
+    command: List[Command] = field(default_factory=list)
+    llm: List[Provider] = field(default_factory=list)
+    
+'''LLM 聊天时持久化的信息'''
+
+@dataclass
+class LLMHistory():
+    session_id: str
+    content: str

astrbot/db/sqlite.py

@@ -0,0 +1,211 @@
+import sqlite3
+import os
+import time
+from astrbot.db.po import (
+    Platform, 
+    Command, 
+    Provider,
+    Stats,
+    LLMHistory
+)
+from . import BaseDatabase
+from typing import Tuple
+
+
+class SQLiteDatabase(BaseDatabase):
+    def __init__(self, db_path: str) -> None:
+        super().__init__()
+        self.db_path = db_path
+        
+        with open(os.path.dirname(__file__) + "/sqlite_init.sql", "r") as f:
+            sql = f.read()
+        
+        # 初始化数据库
+        self.conn = self._get_conn(self.db_path)
+        c = self.conn.cursor()
+        c.executescript(sql)
+        self.conn.commit()
+    
+    def _get_conn(self, db_path: str) -> sqlite3.Connection:
+        conn = sqlite3.connect(self.db_path)
+        conn.text_factory = str
+        return conn
+    
+    def _exec_sql(self, sql: str, params: Tuple = None):
+        conn = self.conn
+        try:
+            c = self.conn.cursor()
+        except sqlite3.ProgrammingError:
+            conn = self._get_conn(self.db_path)
+            c = conn.cursor()
+            
+        if params:
+            c.execute(sql, params)
+            c.close()
+        else:
+            c.execute(sql)
+            c.close()
+        
+        conn.commit()
+        
+    def insert_platform_metrics(self, metrics: dict):
+        for k, v in metrics.items():
+            self._exec_sql(
+                '''
+                INSERT INTO platform(name, count, timestamp) VALUES (?, ?, ?)
+                ''', (k, v, int(time.time()))
+            )
+
+    def insert_plugin_metrics(self, metrics: dict):
+        pass
+
+    def insert_command_metrics(self, metrics: dict):
+        for k, v in metrics.items():
+            self._exec_sql(
+                '''
+                INSERT INTO command(name, count, timestamp) VALUES (?, ?, ?)
+                ''', (k, v, int(time.time()))
+            )
+
+    def insert_llm_metrics(self, metrics: dict):
+        for k, v in metrics.items():
+            self._exec_sql(
+                '''
+                INSERT INTO llm(name, count, timestamp) VALUES (?, ?, ?)
+                ''', (k, v, int(time.time()))
+            )
+
+    def update_llm_history(self, session_id: str, content: str):
+        res = self.get_llm_history(session_id)
+        if res:
+            self._exec_sql(
+                '''
+                UPDATE llm_history SET content = ? WHERE session_id = ?
+                ''', (content, session_id)
+            )
+        else:
+            self._exec_sql(
+                '''
+                INSERT INTO llm_history(session_id, content) VALUES (?, ?)
+                ''', (session_id, content)
+            )
+
+    def get_llm_history(self, session_id: str = None) -> Tuple:
+        try:
+            c = self.conn.cursor()
+        except sqlite3.ProgrammingError:
+            c = self._get_conn(self.db_path).cursor()
+            
+        where_clause = "" if session_id is None else f"WHERE session_id = '{session_id}'"
+        c.execute(
+            '''
+            SELECT * FROM llm_history
+            ''' + where_clause
+        )
+        res = c.fetchall()
+        histories = []
+        for row in res:
+            histories.append(LLMHistory(*row))
+        c.close()
+        return histories
+
+    def get_base_stats(self, offset_sec: int = 86400) -> Stats:
+        '''获取 offset_sec 秒前到现在的基础统计数据'''
+        where_clause = f" WHERE timestamp >= {int(time.time()) - offset_sec}"
+        
+        try:
+            c = self.conn.cursor()
+        except sqlite3.ProgrammingError:
+            c = self._get_conn(self.db_path).cursor()
+            
+        c.execute(
+            '''
+            SELECT * FROM platform
+            ''' + where_clause
+        )
+        
+        platform = []
+        for row in c.fetchall():
+            platform.append(Platform(*row))
+            
+        # c.execute(
+        #     '''
+        #     SELECT * FROM command
+        #     ''' + where_clause
+        # )
+        
+        # command = []
+        # for row in c.fetchall():
+        #     command.append(Command(*row))
+            
+        # c.execute(
+        #     '''
+        #     SELECT * FROM llm
+        #     ''' + where_clause
+        # )
+        
+        # llm = []
+        # for row in c.fetchall():
+        #     llm.append(Provider(*row))
+            
+        c.close()
+            
+        return Stats(platform, [], [])
+    
+    def get_total_message_count(self) -> int:
+        try:
+            c = self.conn.cursor()
+        except sqlite3.ProgrammingError:
+            c = self._get_conn(self.db_path).cursor()
+            
+        c.execute(
+            '''
+            SELECT SUM(count) FROM platform
+            '''
+        )
+        res = c.fetchone()
+        c.close()
+        return res[0]
+    
+    def get_grouped_base_stats(self, offset_sec: int = 86400) -> Stats:
+        '''获取 offset_sec 秒前到现在的基础统计数据(合并)'''
+        where_clause = f" WHERE timestamp >= {int(time.time()) - offset_sec}"
+        
+        try:
+            c = self.conn.cursor()
+        except sqlite3.ProgrammingError:
+            c = self._get_conn(self.db_path).cursor()
+            
+        c.execute(
+            '''
+            SELECT name, SUM(count), timestamp FROM platform
+            ''' + where_clause + " GROUP BY name"
+        )
+        
+        platform = []
+        for row in c.fetchall():
+            platform.append(Platform(*row))
+            
+        # c.execute(
+        #     '''
+        #     SELECT name, SUM(count), timestamp FROM command
+        #     ''' + where_clause + " GROUP BY name"
+        # )
+        
+        # command = []
+        # for row in c.fetchall():
+        #     command.append(Command(*row))
+            
+        # c.execute(
+        #     '''
+        #     SELECT name, SUM(count), timestamp FROM llm
+        #     ''' + where_clause + " GROUP BY name"
+        # )
+        
+        # llm = []
+        # for row in c.fetchall():
+        #     llm.append(Provider(*row))
+            
+        c.close()
+            
+        return Stats(platform, [], [])

astrbot/db/sqlite_init.sql

@@ -0,0 +1,24 @@
+CREATE TABLE IF NOT EXISTS platform(
+    name VARCHAR(32),
+    count INTEGER,
+    timestamp INTEGER
+);
+CREATE TABLE IF NOT EXISTS llm(
+    name VARCHAR(32),
+    count INTEGER,
+    timestamp INTEGER
+);
+CREATE TABLE IF NOT EXISTS plugin(
+    name VARCHAR(32),
+    count INTEGER,
+    timestamp INTEGER
+);
+CREATE TABLE IF NOT EXISTS command(
+    name VARCHAR(32),
+    count INTEGER,
+    timestamp INTEGER
+);
+CREATE TABLE IF NOT EXISTS llm_history(
+    session_id VARCHAR(32),
+    content TEXT
+);

astrbot/message/baidu_aip_judge.py

@@ -0,0 +1,28 @@
+from aip import AipContentCensor
+from util.cmd_config import BaiduAIPConfig
+
+
+class BaiduJudge:
+    def __init__(self, baidu_configs: BaiduAIPConfig) -> None:
+        self.app_id = baidu_configs.app_id
+        self.api_key = baidu_configs.api_key
+        self.secret_key = baidu_configs.secret_key
+        self.client = AipContentCensor(self.app_id, 
+                                       self.api_key, 
+                                       self.secret_key)
+
+    def judge(self, text):
+        res = self.client.textCensorUserDefined(text)
+        if 'conclusionType' not in res:
+            return False, "百度审核服务未知错误"
+        if res['conclusionType'] == 1:
+            return True, "合规"
+        else:
+            if 'data' not in res:
+                return False, "百度审核服务未知错误"
+            count = len(res['data'])
+            info = f"百度审核服务发现 {count} 处违规：\n"
+            for i in res['data']:
+                info += f"{i['msg']}；\n"
+            info += "\n判断结果："+res['conclusion']
+            return False, info

astrbot/message/handler.py

@@ -0,0 +1,290 @@
+import time, json
+import re, os
+import asyncio
+import traceback
+import astrbot.message.unfit_words as uw
+
+from typing import Dict
+from astrbot.db import BaseDatabase
+from model.provider.provider import Provider
+from model.command.manager import CommandManager
+from type.message_event import AstrMessageEvent, MessageResult
+from type.types import Context
+from type.command import CommandResult
+from util.log import LogManager
+from logging import Logger
+from nakuru.entities.components import Image
+from util.agent.func_call import FuncCall
+from openai._exceptions import *
+from openai.types.chat.chat_completion_message_tool_call import Function
+
+logger: Logger = LogManager.GetLogger(log_name='astrbot')
+
+
+class RateLimitHelper():
+    def __init__(self, context: Context) -> None:
+        self.user_rate_limit: Dict[int, int] = {}
+        rl = context.config_helper.platform_settings.rate_limit
+        self.rate_limit_time: int = rl.time
+        self.rate_limit_count: int = rl.count
+        self.user_frequency = {}
+
+    def check_frequency(self, session_id: str) -> bool:
+        '''
+        检查发言频率
+        '''
+        ts = int(time.time())
+        if session_id in self.user_frequency:
+            if ts-self.user_frequency[session_id]['time'] > self.rate_limit_time:
+                self.user_frequency[session_id]['time'] = ts
+                self.user_frequency[session_id]['count'] = 1
+                return True
+            else:
+                if self.user_frequency[session_id]['count'] >= self.rate_limit_count:
+                    return False
+                else:
+                    self.user_frequency[session_id]['count'] += 1
+                    return True
+        else:
+            t = {'time': ts, 'count': 1}
+            self.user_frequency[session_id] = t
+            return True
+
+class ContentSafetyHelper():
+    def __init__(self, context: Context) -> None:
+        self.baidu_judge = None
+        aip = context.config_helper.content_safety.baidu_aip
+        if aip.enable:
+            try:
+                from astrbot.message.baidu_aip_judge import BaiduJudge
+                self.baidu_judge = BaiduJudge(aip)
+                logger.info("已启用百度 AI 内容审核。")
+            except ImportError as e:
+                logger.error("检测到库依赖不完整，将不会启用百度 AI 内容审核。请先使用 pip 安装 `baidu_aip` 包。")
+                logger.error(e)
+            except BaseException as e:
+                logger.error("百度 AI 内容审核初始化失败。")
+                logger.error(e)
+                
+    async def check_content(self, content: str) -> bool:
+        '''
+        检查文本内容是否合法
+        '''
+        for i in uw.unfit_words_q:
+            matches = re.match(i, content.strip(), re.I | re.M)
+            if matches:
+                return False
+        if self.baidu_judge != None:
+            check, msg = await asyncio.to_thread(self.baidu_judge.judge, content)
+            if not check:
+                logger.info(f"百度 AI 内容审核发现以下违规：{msg}")
+                return False
+        return True
+    
+    def filter_content(self, content: str) -> str:
+        '''
+        过滤文本内容
+        '''
+        for i in uw.unfit_words_q:
+            content = re.sub(i, "*", content, flags=re.I)
+        return content
+    
+    def baidu_check(self, content: str) -> bool:
+        '''
+        使用百度 AI 内容审核检查文本内容是否合法
+        '''
+        if self.baidu_judge != None:
+            check, msg = self.baidu_judge.judge(content)
+            if not check:
+                logger.info(f"百度 AI 内容审核发现以下违规：{msg}")
+                return False
+        return True
+
+class MessageHandler():
+    def __init__(self, context: Context,
+                 command_manager: CommandManager,
+                 db_helper: BaseDatabase) -> None:
+        self.context = context
+        self.command_manager = command_manager
+        self.db_helper = db_helper
+        self.rate_limit_helper = RateLimitHelper(context)
+        self.content_safety_helper = ContentSafetyHelper(context)
+        self.llm_wake_prefix = self.context.config_helper.llm_settings.wake_prefix
+        self.llm_identifier = self.context.config_helper.llm_settings.identifier
+        if self.llm_wake_prefix:
+            self.llm_wake_prefix = self.llm_wake_prefix.strip()
+        self.provider = self.context.llms[0].llm_instance if len(self.context.llms) > 0 else None
+        self.reply_prefix = str(self.context.config_helper.platform_settings.reply_prefix)        
+        self.llm_tools = FuncCall(self.provider)
+
+    def set_provider(self, provider: Provider):
+        self.provider = provider
+
+    async def handle(self, message: AstrMessageEvent, llm_provider: Provider = None) -> MessageResult:
+        '''
+        Handle the message event, including commands, plugins, etc.
+        
+        `llm_provider`: the provider to use for LLM. If None, use the default provider
+        '''
+        msg_plain = message.message_str.strip()
+        provider = llm_provider if llm_provider else self.provider        
+        
+        # TODO: this should be configurable
+        # if not message.message_str:
+        #     return MessageResult("Hi~")
+        
+        # check the rate limit
+        if not self.rate_limit_helper.check_frequency(message.message_obj.sender.user_id):
+            logger.warning(f"用户 {message.message_obj.sender.user_id} 的发言频率超过限制，已忽略。")
+            return
+
+        # remove the nick prefix
+        for nick in self.context.config_helper.wake_prefix:
+            if msg_plain.startswith(nick):
+                msg_plain = msg_plain.removeprefix(nick)
+                break
+        message.message_str = msg_plain
+
+        # scan candidate commands
+        cmd_res = await self.command_manager.scan_command(message, self.context)
+        if cmd_res:
+            assert(isinstance(cmd_res, CommandResult))
+            return MessageResult(
+                cmd_res.message_chain,
+                is_command_call=True,
+                use_t2i=cmd_res.is_use_t2i
+            )
+            
+        # middlewares
+        for middleware in self.context.middlewares:
+            try:
+                logger.info(f"执行中间件 {middleware.origin}/{middleware.name}...")
+                await middleware.func(message, self.context)
+            except BaseException as e:
+                logger.error(f"中间件 {middleware.origin}/{middleware.name} 处理消息时发生异常：{e}，跳过。")
+                logger.error(traceback.format_exc())
+        
+        if message.only_command:
+            return
+        
+        # next is the LLM part
+        # check if the message is a llm-wake-up command
+        if self.llm_wake_prefix and not msg_plain.startswith(self.llm_wake_prefix):
+            logger.debug(f"消息 `{msg_plain}` 没有以 LLM 唤醒前缀 `{self.llm_wake_prefix}` 开头，忽略。")
+            return
+        
+        if not provider:
+            logger.debug("没有任何 LLM 可用，忽略。")
+            return
+        
+        # check the content safety
+        if not await self.content_safety_helper.check_content(msg_plain):
+            return MessageResult("信息包含违规内容，由于机器人管理者开启内容安全审核，你的此条消息已被停止继续处理。")
+        
+        image_url = None
+        for comp in message.message_obj.message:
+            if isinstance(comp, Image):
+                image_url = comp.url if comp.url else comp.file
+                break
+        try:
+            if not self.llm_tools.empty():
+                # tools-use
+                tool_use_flag = True
+                llm_result = await provider.text_chat(
+                    prompt=msg_plain, 
+                    session_id=message.session_id, 
+                    tools=self.llm_tools.get_func()
+                )
+                
+                if isinstance(llm_result, Function):
+                    logger.debug(f"function-calling: {llm_result}")
+                    func_obj = None
+                    for i in self.llm_tools.func_list:
+                        if i["name"] == llm_result.name:
+                            func_obj = i["func_obj"]
+                            break
+                    if not func_obj:
+                        return MessageResult("AstrBot Function-calling 异常：未找到请求的函数调用。")
+                    try:
+                        args = json.loads(llm_result.arguments)
+                        args['ame'] = message
+                        args['context'] = self.context
+                        try:
+                            cmd_res = await func_obj(**args)
+                        except TypeError as e:
+                            args.pop('ame')
+                            args.pop('context')
+                            cmd_res = await func_obj(**args)
+                        if isinstance(cmd_res, CommandResult):
+                            return MessageResult(
+                                cmd_res.message_chain,
+                                is_command_call=True,
+                                use_t2i=cmd_res.is_use_t2i
+                            )
+                        elif isinstance(cmd_res, str):
+                            return MessageResult(cmd_res)
+                        elif not cmd_res:
+                            return
+                        else:
+                            return MessageResult(f"AstrBot Function-calling 异常：调用：{llm_result} 时，返回了未知的返回值类型。")
+                    except BaseException as e:
+                        traceback.print_exc()
+                        return MessageResult("AstrBot Function-calling 异常：" + str(e))
+                else:
+                    return MessageResult(llm_result)
+        
+            else:
+                # normal chat
+                tool_use_flag = False
+                # add user info to the prompt
+                if self.llm_identifier:
+                    user_id = message.message_obj.sender.user_id
+                    user_nickname = message.message_obj.sender.nickname
+                    user_info = f"[User ID: {user_id}, Nickname: {user_nickname}]\n"
+                    msg_plain = user_info + msg_plain
+
+                llm_result = await provider.text_chat(
+                    prompt=msg_plain, 
+                    session_id=message.session_id, 
+                    image_url=image_url
+                )
+        except BadRequestError as e:
+            if tool_use_flag:
+                # seems like the model don't support function-calling
+                logger.error(f"error: {e}. Using local function-calling implementation")
+                
+                try:
+                    # use local function-calling implementation
+                    args = {
+                        'question': llm_result,
+                        'func_definition': self.llm_tools.func_dump(),
+                    }
+                    _, has_func = await self.llm_tools.func_call(**args)
+                    
+                    if not has_func:
+                        # normal chat
+                        llm_result = await provider.text_chat(
+                            prompt=msg_plain, 
+                            session_id=message.session_id, 
+                            image_url=image_url
+                        )
+                except BaseException as e:
+                    logger.error(traceback.format_exc())
+                    return CommandResult("AstrBot Function-calling 异常：" + str(e))
+
+        except BaseException as e:
+            logger.error(traceback.format_exc())
+            logger.error(f"LLM 调用失败。")
+            return MessageResult("AstrBot 请求 LLM 资源失败：" + str(e))
+
+        # concatenate reply prefix
+        if self.reply_prefix:
+            llm_result = self.reply_prefix + llm_result
+        
+        # mask unsafe content
+        llm_result = self.content_safety_helper.filter_content(llm_result)
+        check = self.content_safety_helper.baidu_check(llm_result)
+        if not check:
+            return MessageResult("LLM 输出的信息包含违规内容，由于机器人管理者开启了内容安全审核，该条消息已拦截。")
+        
+        return MessageResult(llm_result)

astrbot/message/unfit_words.py

@@ -0,0 +1,114 @@
+# 网上搜集的。
+# 来源词拦截
+# 支持正则
+unfit_words_q = ["习近平",
+                 "胡锦涛",
+                 "江泽民",
+                 "温家宝",
+                 "李克强",
+                 "李长春",
+                 "毛泽东",
+                 "邓小平",
+                 "周恩来",
+                 "社会主义",
+                 "共产党",
+                 "共产主义",
+                 "大陆官方",
+                 "北京政权",
+                 "中华帝国",
+                 "中国政府",
+                 "共狗",
+                 "六四事件",
+                 "天安门",
+                 "六四",
+                 "政治局常委",
+                 "学潮",
+                 "八九",
+                 "二十大",
+                 "民进党",
+                 "台独",
+                 "台湾独立",
+                 "台湾国",
+                 "国民党",
+                 "台湾民国",
+                 "中华民国",
+                 "pornhub",
+                 "Pornhub",
+                 "作爱",
+                 "做爱",
+                 "性交",
+                 "自慰",
+                 "阴茎",
+                 "淫妇",
+                 "肛交",
+                 "交配",
+                 "性关系",
+                 "性活动",
+                 "色情",
+                 "色图",
+                 "裸体",
+                 "小穴",
+                 "淫荡",
+                 "性爱",
+                 "港独",
+                 "法轮功",
+                 "六四"]
+
+# 回复词过滤
+unfit_words = ["习近平",
+               "胡锦涛",
+               "江泽民",
+               "温家宝",
+               "李克强",
+               "李长春",
+               "毛泽东",
+               "邓小平",
+               "周恩来",
+               "社会主义",
+               "共产党",
+               "共产主义",
+               "大陆官方",
+               "北京政权",
+               "中华帝国",
+               "中国政府",
+               "共狗",
+               "六四事件",
+               "天安门",
+               "六四",
+               "政治局常委",
+               "学潮",
+               "八九",
+               "二十大",
+               "民进党",
+               "台独",
+               "台湾独立",
+               "台湾国",
+               "国民党",
+               "台湾民国",
+               "中华民国",
+               "pornhub",
+               "Pornhub",
+               "作爱",
+               "做爱",
+               "性交",
+               "自慰",
+               "阴茎",
+               "淫妇",
+               "肛交",
+               "交配",
+               "性关系",
+               "性活动",
+               "色情",
+               "色图",
+               "涩图",
+               "裸体",
+               "小穴",
+               "淫荡",
+               "性爱",
+               "中华人民共和国",
+               "党中央",
+               "中央军委主席",
+               "台湾",
+               "港独",
+               "法轮功",
+               "PRC"]

bot24linux/configs/config.yaml

@@ -1,42 +0,0 @@
-openai:
-  # 注意：在1.7版本已支持多key自动切换，方法：
-  # key:
-  #   - xxxxx
-  #   - xxxxxx
-  # 在下方非注释的地方使用以上格式
-  key: 
-    - 
-    
-  chatGPTConfigs:
-    engine: "gpt-3.5-turbo"
-    max_tokens: 1000
-    temperature: 0.9
-    top_p: 1
-    frequency_penalty: 0
-    presence_penalty: 0
-    
-  total_tokens_limit: 2000
-qqbot:
-  appid: 
-  token: 
-
-# 设置是否一个人一个会话
-uniqueSessionMode: false
-
-# QChannelBot 的版本，请勿修改此字段，否则可能产生一些bug
-version: 2.4 RealChatGPT Ver.
-
-# [Beta] 转储历史记录时间间隔(分钟)
-dump_history_interval: 10
-
-# 一个用户只能在time秒内发送count条消息
-limit:
-  time: 60
-  count: 5
-# 公告
-notice: "此机器人由Github项目QQChannelChatGPT驱动。"
-
-# 是否打开私信功能
-# 设置为true则频道成员可以私聊机器人。
-# 设置为false则频道成员不能私聊机器人。
-direct_message_mode: true

botpy.log

@@ -1,323 +0,0 @@
-2022-12-08 14:29:09,486	[INFO]	(client.py:159)_bot_login	[botpy] 登录机器人账号中...
-2022-12-08 14:29:10,173	[INFO]	(client.py:178)_bot_init	[botpy] 程序启动...
-2022-12-08 14:29:10,174	[INFO]	(connection.py:59)multi_run	[botpy] 最大并发连接数: 1, 启动会话数: 1
-2022-12-08 14:29:10,175	[INFO]	(client.py:236)bot_connect	[botpy] 会话启动中...
-2022-12-08 14:29:10,175	[INFO]	(gateway.py:110)ws_connect	[botpy] 启动中...
-2022-12-08 14:29:10,335	[INFO]	(gateway.py:136)ws_identify	[botpy] 鉴权中...
-2022-12-08 14:29:10,460	[INFO]	(gateway.py:80)on_message	[botpy] 机器人「SoGPT-测试中」启动成功！
-2022-12-08 14:29:10,461	[INFO]	(gateway.py:217)_send_heart	[botpy] 心跳维持启动...
-2022-12-08 16:17:18,117	[INFO]	(client.py:159)_bot_login	[botpy] 登录机器人账号中...
-2022-12-08 16:17:18,355	[ERROR]	(http.py:73)_handle_response	[botpy] 接口请求异常，请求连接: https://api.sgroup.qq.com/users/@me, 错误代码: 401, 返回内容: {'message': 'wrong bot token', 'code': 11242}, trace_id:829d8c60d296a3edcfac3d776dbd8b47
-2022-12-08 16:18:59,759	[INFO]	(client.py:159)_bot_login	[botpy] 登录机器人账号中...
-2022-12-08 16:19:00,266	[INFO]	(client.py:178)_bot_init	[botpy] 程序启动...
-2022-12-08 16:19:00,267	[INFO]	(connection.py:59)multi_run	[botpy] 最大并发连接数: 1, 启动会话数: 1
-2022-12-08 16:19:00,268	[INFO]	(client.py:236)bot_connect	[botpy] 会话启动中...
-2022-12-08 16:19:00,268	[INFO]	(gateway.py:110)ws_connect	[botpy] 启动中...
-2022-12-08 16:19:00,412	[INFO]	(gateway.py:136)ws_identify	[botpy] 鉴权中...
-2022-12-08 16:19:00,522	[INFO]	(gateway.py:80)on_message	[botpy] 机器人「SoGPT-测试中」启动成功！
-2022-12-08 16:19:00,522	[INFO]	(gateway.py:217)_send_heart	[botpy] 心跳维持启动...
-2022-12-08 16:20:14,446	[INFO]	(client.py:159)_bot_login	[botpy] 登录机器人账号中...
-2022-12-08 16:20:15,035	[INFO]	(client.py:178)_bot_init	[botpy] 程序启动...
-2022-12-08 16:20:15,036	[INFO]	(connection.py:59)multi_run	[botpy] 最大并发连接数: 1, 启动会话数: 1
-2022-12-08 16:20:15,037	[INFO]	(client.py:236)bot_connect	[botpy] 会话启动中...
-2022-12-08 16:20:15,037	[INFO]	(gateway.py:110)ws_connect	[botpy] 启动中...
-2022-12-08 16:20:15,232	[INFO]	(gateway.py:136)ws_identify	[botpy] 鉴权中...
-2022-12-08 16:20:15,320	[INFO]	(gateway.py:80)on_message	[botpy] 机器人「SoGPT-测试中」启动成功！
-2022-12-08 16:20:15,321	[INFO]	(gateway.py:217)_send_heart	[botpy] 心跳维持启动...
-2022-12-08 16:42:06,957	[INFO]	(client.py:159)_bot_login	[botpy] 登录机器人账号中...
-2022-12-08 16:42:07,468	[INFO]	(client.py:178)_bot_init	[botpy] 程序启动...
-2022-12-08 16:42:07,469	[INFO]	(connection.py:59)multi_run	[botpy] 最大并发连接数: 1, 启动会话数: 1
-2022-12-08 16:42:07,469	[INFO]	(client.py:236)bot_connect	[botpy] 会话启动中...
-2022-12-08 16:42:07,470	[INFO]	(gateway.py:110)ws_connect	[botpy] 启动中...
-2022-12-08 16:42:07,672	[INFO]	(gateway.py:136)ws_identify	[botpy] 鉴权中...
-2022-12-08 16:42:07,862	[INFO]	(gateway.py:80)on_message	[botpy] 机器人「SoGPT-测试中」启动成功！
-2022-12-08 16:42:07,863	[INFO]	(gateway.py:217)_send_heart	[botpy] 心跳维持启动...
-2022-12-08 16:45:44,758	[INFO]	(client.py:159)_bot_login	[botpy] 登录机器人账号中...
-2022-12-08 16:45:45,439	[INFO]	(client.py:178)_bot_init	[botpy] 程序启动...
-2022-12-08 16:45:45,440	[INFO]	(connection.py:59)multi_run	[botpy] 最大并发连接数: 1, 启动会话数: 1
-2022-12-08 16:45:45,441	[INFO]	(client.py:236)bot_connect	[botpy] 会话启动中...
-2022-12-08 16:45:45,441	[INFO]	(gateway.py:110)ws_connect	[botpy] 启动中...
-2022-12-08 16:45:45,751	[INFO]	(gateway.py:136)ws_identify	[botpy] 鉴权中...
-2022-12-08 16:45:45,858	[INFO]	(gateway.py:80)on_message	[botpy] 机器人「SoGPT-测试中」启动成功！
-2022-12-08 16:45:45,859	[INFO]	(gateway.py:217)_send_heart	[botpy] 心跳维持启动...
-2022-12-08 16:47:16,567	[INFO]	(client.py:159)_bot_login	[botpy] 登录机器人账号中...
-2022-12-08 16:47:17,008	[INFO]	(client.py:178)_bot_init	[botpy] 程序启动...
-2022-12-08 16:47:17,009	[INFO]	(connection.py:59)multi_run	[botpy] 最大并发连接数: 1, 启动会话数: 1
-2022-12-08 16:47:17,009	[INFO]	(client.py:236)bot_connect	[botpy] 会话启动中...
-2022-12-08 16:47:17,010	[INFO]	(gateway.py:110)ws_connect	[botpy] 启动中...
-2022-12-08 16:47:17,187	[INFO]	(gateway.py:136)ws_identify	[botpy] 鉴权中...
-2022-12-08 16:47:17,284	[INFO]	(gateway.py:80)on_message	[botpy] 机器人「SoGPT-测试中」启动成功！
-2022-12-08 16:47:17,285	[INFO]	(gateway.py:217)_send_heart	[botpy] 心跳维持启动...
-2022-12-08 16:48:39,358	[INFO]	(client.py:159)_bot_login	[botpy] 登录机器人账号中...
-2022-12-08 16:48:45,366	[WARNING]	(http.py:188)request	请求超时，请求连接: https://api.sgroup.qq.com/gateway/bot
-2022-12-08 16:48:53,301	[INFO]	(client.py:159)_bot_login	[botpy] 登录机器人账号中...
-2022-12-08 16:48:53,789	[INFO]	(client.py:178)_bot_init	[botpy] 程序启动...
-2022-12-08 16:48:53,790	[INFO]	(connection.py:59)multi_run	[botpy] 最大并发连接数: 1, 启动会话数: 1
-2022-12-08 16:48:53,791	[INFO]	(client.py:236)bot_connect	[botpy] 会话启动中...
-2022-12-08 16:48:53,791	[INFO]	(gateway.py:110)ws_connect	[botpy] 启动中...
-2022-12-08 16:48:53,969	[INFO]	(gateway.py:136)ws_identify	[botpy] 鉴权中...
-2022-12-08 16:48:54,062	[INFO]	(gateway.py:80)on_message	[botpy] 机器人「SoGPT-测试中」启动成功！
-2022-12-08 16:48:54,063	[INFO]	(gateway.py:217)_send_heart	[botpy] 心跳维持启动...
-2022-12-08 16:49:26,466	[ERROR]	(http.py:73)_handle_response	[botpy] 接口请求异常，请求连接: https://api.sgroup.qq.com/channels/7150805/messages, 错误代码: 403, 返回内容: {'code': 304003, 'message': 'url not allowed'}, trace_id:b201dd7b37649dcf47b82d54c58bc5dd
-2022-12-08 16:51:59,155	[INFO]	(client.py:159)_bot_login	[botpy] 登录机器人账号中...
-2022-12-08 16:51:59,728	[INFO]	(client.py:178)_bot_init	[botpy] 程序启动...
-2022-12-08 16:51:59,729	[INFO]	(connection.py:59)multi_run	[botpy] 最大并发连接数: 1, 启动会话数: 1
-2022-12-08 16:51:59,730	[INFO]	(client.py:236)bot_connect	[botpy] 会话启动中...
-2022-12-08 16:51:59,730	[INFO]	(gateway.py:110)ws_connect	[botpy] 启动中...
-2022-12-08 16:51:59,887	[INFO]	(gateway.py:136)ws_identify	[botpy] 鉴权中...
-2022-12-08 16:52:00,022	[INFO]	(gateway.py:80)on_message	[botpy] 机器人「SoGPT-测试中」启动成功！
-2022-12-08 16:52:00,023	[INFO]	(gateway.py:217)_send_heart	[botpy] 心跳维持启动...
-2022-12-08 16:52:28,760	[ERROR]	(http.py:73)_handle_response	[botpy] 接口请求异常，请求连接: https://api.sgroup.qq.com/channels/7150805/messages, 错误代码: 403, 返回内容: {'code': 304003, 'message': 'url not allowed'}, trace_id:e62b072c9f0184f6bf7dd03a00fc0ef3
-2022-12-08 16:53:53,370	[INFO]	(client.py:159)_bot_login	[botpy] 登录机器人账号中...
-2022-12-08 16:53:53,890	[INFO]	(client.py:178)_bot_init	[botpy] 程序启动...
-2022-12-08 16:53:53,891	[INFO]	(connection.py:59)multi_run	[botpy] 最大并发连接数: 1, 启动会话数: 1
-2022-12-08 16:53:53,892	[INFO]	(client.py:236)bot_connect	[botpy] 会话启动中...
-2022-12-08 16:53:53,892	[INFO]	(gateway.py:110)ws_connect	[botpy] 启动中...
-2022-12-08 16:53:54,101	[INFO]	(gateway.py:136)ws_identify	[botpy] 鉴权中...
-2022-12-08 16:53:54,194	[INFO]	(gateway.py:80)on_message	[botpy] 机器人「SoGPT-测试中」启动成功！
-2022-12-08 16:53:54,195	[INFO]	(gateway.py:217)_send_heart	[botpy] 心跳维持启动...
-2022-12-08 16:54:26,287	[ERROR]	(http.py:73)_handle_response	[botpy] 接口请求异常，请求连接: https://api.sgroup.qq.com/channels/7150805/messages, 错误代码: 501, 返回内容: None, trace_id:None
-2022-12-08 16:55:01,062	[INFO]	(client.py:159)_bot_login	[botpy] 登录机器人账号中...
-2022-12-08 16:55:01,601	[INFO]	(client.py:178)_bot_init	[botpy] 程序启动...
-2022-12-08 16:55:01,601	[INFO]	(connection.py:59)multi_run	[botpy] 最大并发连接数: 1, 启动会话数: 1
-2022-12-08 16:55:01,602	[INFO]	(client.py:236)bot_connect	[botpy] 会话启动中...
-2022-12-08 16:55:01,603	[INFO]	(gateway.py:110)ws_connect	[botpy] 启动中...
-2022-12-08 16:55:01,742	[INFO]	(gateway.py:136)ws_identify	[botpy] 鉴权中...
-2022-12-08 16:55:01,816	[INFO]	(gateway.py:80)on_message	[botpy] 机器人「SoGPT-测试中」启动成功！
-2022-12-08 16:55:01,817	[INFO]	(gateway.py:217)_send_heart	[botpy] 心跳维持启动...
-2022-12-08 17:00:29,939	[INFO]	(client.py:159)_bot_login	[botpy] 登录机器人账号中...
-2022-12-08 17:00:30,583	[INFO]	(client.py:178)_bot_init	[botpy] 程序启动...
-2022-12-08 17:00:30,584	[INFO]	(connection.py:59)multi_run	[botpy] 最大并发连接数: 1, 启动会话数: 1
-2022-12-08 17:00:30,585	[INFO]	(client.py:236)bot_connect	[botpy] 会话启动中...
-2022-12-08 17:00:30,585	[INFO]	(gateway.py:110)ws_connect	[botpy] 启动中...
-2022-12-08 17:00:30,839	[INFO]	(gateway.py:136)ws_identify	[botpy] 鉴权中...
-2022-12-08 17:00:30,943	[INFO]	(gateway.py:80)on_message	[botpy] 机器人「SoGPT-测试中」启动成功！
-2022-12-08 17:00:30,944	[INFO]	(gateway.py:217)_send_heart	[botpy] 心跳维持启动...
-2022-12-08 17:00:53,548	[ERROR]	(http.py:73)_handle_response	[botpy] 接口请求异常，请求连接: https://api.sgroup.qq.com/channels/7150658/messages, 错误代码: 403, 返回内容: {'code': 304003, 'message': 'url not allowed'}, trace_id:2ea547d78364cea60583ed450e589a17
-2022-12-08 17:10:31,405	[INFO]	(client.py:159)_bot_login	[botpy] 登录机器人账号中...
-2022-12-08 17:10:32,081	[INFO]	(client.py:178)_bot_init	[botpy] 程序启动...
-2022-12-08 17:10:32,082	[INFO]	(connection.py:59)multi_run	[botpy] 最大并发连接数: 1, 启动会话数: 1
-2022-12-08 17:10:32,083	[INFO]	(client.py:236)bot_connect	[botpy] 会话启动中...
-2022-12-08 17:10:32,083	[INFO]	(gateway.py:110)ws_connect	[botpy] 启动中...
-2022-12-08 17:10:32,342	[INFO]	(gateway.py:136)ws_identify	[botpy] 鉴权中...
-2022-12-08 17:10:32,449	[INFO]	(gateway.py:80)on_message	[botpy] 机器人「SoGPT-测试中」启动成功！
-2022-12-08 17:10:32,450	[INFO]	(gateway.py:217)_send_heart	[botpy] 心跳维持启动...
-2022-12-08 17:13:47,495	[INFO]	(client.py:159)_bot_login	[botpy] 登录机器人账号中...
-2022-12-08 17:13:47,993	[INFO]	(client.py:178)_bot_init	[botpy] 程序启动...
-2022-12-08 17:13:47,994	[INFO]	(connection.py:59)multi_run	[botpy] 最大并发连接数: 1, 启动会话数: 1
-2022-12-08 17:13:47,995	[INFO]	(client.py:236)bot_connect	[botpy] 会话启动中...
-2022-12-08 17:13:47,995	[INFO]	(gateway.py:110)ws_connect	[botpy] 启动中...
-2022-12-08 17:13:48,198	[INFO]	(gateway.py:136)ws_identify	[botpy] 鉴权中...
-2022-12-08 17:13:48,300	[INFO]	(gateway.py:80)on_message	[botpy] 机器人「SoGPT-测试中」启动成功！
-2022-12-08 17:13:48,302	[INFO]	(gateway.py:217)_send_heart	[botpy] 心跳维持启动...
-2022-12-08 17:16:31,326	[ERROR]	(http.py:73)_handle_response	[botpy] 接口请求异常，请求连接: https://api.sgroup.qq.com/channels/7150658/messages, 错误代码: 501, 返回内容: None, trace_id:None
-2022-12-08 17:22:03,924	[INFO]	(client.py:159)_bot_login	[botpy] 登录机器人账号中...
-2022-12-08 17:22:04,671	[INFO]	(client.py:178)_bot_init	[botpy] 程序启动...
-2022-12-08 17:22:04,672	[INFO]	(connection.py:59)multi_run	[botpy] 最大并发连接数: 1, 启动会话数: 1
-2022-12-08 17:22:04,673	[INFO]	(client.py:236)bot_connect	[botpy] 会话启动中...
-2022-12-08 17:22:04,673	[INFO]	(gateway.py:110)ws_connect	[botpy] 启动中...
-2022-12-08 17:22:04,856	[INFO]	(gateway.py:136)ws_identify	[botpy] 鉴权中...
-2022-12-08 17:22:04,958	[INFO]	(gateway.py:80)on_message	[botpy] 机器人「SoGPT-测试中」启动成功！
-2022-12-08 17:22:04,959	[INFO]	(gateway.py:217)_send_heart	[botpy] 心跳维持启动...
-2022-12-08 17:52:04,868	[INFO]	(gateway.py:54)on_closed	[botpy] 关闭, 返回码: 4009, 返回信息: Session timed out
-2022-12-08 17:52:09,874	[INFO]	(connection.py:59)multi_run	[botpy] 最大并发连接数: 1, 启动会话数: 1
-2022-12-08 17:52:09,876	[INFO]	(client.py:236)bot_connect	[botpy] 会话启动中...
-2022-12-08 17:52:09,877	[INFO]	(gateway.py:110)ws_connect	[botpy] 启动中...
-2022-12-08 17:52:10,104	[INFO]	(gateway.py:169)ws_resume	[botpy] 重连启动...
-2022-12-08 17:52:10,170	[INFO]	(gateway.py:85)on_message	[botpy] 机器人重连成功! 
-2022-12-08 17:52:10,171	[INFO]	(gateway.py:217)_send_heart	[botpy] 心跳维持启动...
-2022-12-08 18:00:00,757	[INFO]	(client.py:159)_bot_login	[botpy] 登录机器人账号中...
-2022-12-08 18:00:01,443	[INFO]	(client.py:178)_bot_init	[botpy] 程序启动...
-2022-12-08 18:00:01,443	[INFO]	(connection.py:59)multi_run	[botpy] 最大并发连接数: 1, 启动会话数: 1
-2022-12-08 18:00:01,444	[INFO]	(client.py:236)bot_connect	[botpy] 会话启动中...
-2022-12-08 18:00:01,445	[INFO]	(gateway.py:110)ws_connect	[botpy] 启动中...
-2022-12-08 18:00:01,602	[INFO]	(gateway.py:136)ws_identify	[botpy] 鉴权中...
-2022-12-08 18:00:01,823	[INFO]	(gateway.py:80)on_message	[botpy] 机器人「SoGPT-测试中」启动成功！
-2022-12-08 18:00:01,824	[INFO]	(gateway.py:217)_send_heart	[botpy] 心跳维持启动...
-2022-12-08 18:03:13,211	[INFO]	(client.py:159)_bot_login	[botpy] 登录机器人账号中...
-2022-12-08 18:03:14,082	[INFO]	(client.py:178)_bot_init	[botpy] 程序启动...
-2022-12-08 18:03:14,084	[INFO]	(connection.py:59)multi_run	[botpy] 最大并发连接数: 1, 启动会话数: 1
-2022-12-08 18:03:14,084	[INFO]	(client.py:236)bot_connect	[botpy] 会话启动中...
-2022-12-08 18:03:14,085	[INFO]	(gateway.py:110)ws_connect	[botpy] 启动中...
-2022-12-08 18:03:14,289	[INFO]	(gateway.py:136)ws_identify	[botpy] 鉴权中...
-2022-12-08 18:03:14,498	[INFO]	(gateway.py:80)on_message	[botpy] 机器人「SoGPT-测试中」启动成功！
-2022-12-08 18:03:14,499	[INFO]	(gateway.py:217)_send_heart	[botpy] 心跳维持启动...
-2022-12-08 18:16:29,246	[INFO]	(client.py:159)_bot_login	[botpy] 登录机器人账号中...
-2022-12-08 18:16:30,054	[INFO]	(client.py:178)_bot_init	[botpy] 程序启动...
-2022-12-08 18:16:30,055	[INFO]	(connection.py:59)multi_run	[botpy] 最大并发连接数: 1, 启动会话数: 1
-2022-12-08 18:16:30,056	[INFO]	(client.py:236)bot_connect	[botpy] 会话启动中...
-2022-12-08 18:16:30,056	[INFO]	(gateway.py:110)ws_connect	[botpy] 启动中...
-2022-12-08 18:16:30,293	[INFO]	(gateway.py:136)ws_identify	[botpy] 鉴权中...
-2022-12-08 18:16:30,424	[INFO]	(gateway.py:80)on_message	[botpy] 机器人「SoGPT-测试中」启动成功！
-2022-12-08 18:16:30,424	[INFO]	(gateway.py:217)_send_heart	[botpy] 心跳维持启动...
-2022-12-08 18:16:41,898	[ERROR]	(http.py:73)_handle_response	[botpy] 接口请求异常，请求连接: https://api.sgroup.qq.com/channels/7150658/messages, 错误代码: 403, 返回内容: {'code': 304003, 'message': 'url not allowed'}, trace_id:3cc9fc95ac27cbbc1f1667f1aa11bf8d
-2022-12-08 18:18:16,615	[INFO]	(client.py:159)_bot_login	[botpy] 登录机器人账号中...
-2022-12-08 18:18:17,399	[INFO]	(client.py:178)_bot_init	[botpy] 程序启动...
-2022-12-08 18:18:17,400	[INFO]	(connection.py:59)multi_run	[botpy] 最大并发连接数: 1, 启动会话数: 1
-2022-12-08 18:18:17,400	[INFO]	(client.py:236)bot_connect	[botpy] 会话启动中...
-2022-12-08 18:18:17,401	[INFO]	(gateway.py:110)ws_connect	[botpy] 启动中...
-2022-12-08 18:18:17,606	[INFO]	(gateway.py:136)ws_identify	[botpy] 鉴权中...
-2022-12-08 18:18:17,731	[INFO]	(gateway.py:80)on_message	[botpy] 机器人「SoGPT-测试中」启动成功！
-2022-12-08 18:18:17,732	[INFO]	(gateway.py:217)_send_heart	[botpy] 心跳维持启动...
-2022-12-08 18:19:07,357	[INFO]	(client.py:159)_bot_login	[botpy] 登录机器人账号中...
-2022-12-08 18:19:07,941	[INFO]	(client.py:178)_bot_init	[botpy] 程序启动...
-2022-12-08 18:19:07,942	[INFO]	(connection.py:59)multi_run	[botpy] 最大并发连接数: 1, 启动会话数: 1
-2022-12-08 18:19:07,942	[INFO]	(client.py:236)bot_connect	[botpy] 会话启动中...
-2022-12-08 18:19:07,943	[INFO]	(gateway.py:110)ws_connect	[botpy] 启动中...
-2022-12-08 18:19:08,247	[INFO]	(gateway.py:136)ws_identify	[botpy] 鉴权中...
-2022-12-08 18:19:08,335	[INFO]	(gateway.py:80)on_message	[botpy] 机器人「SoGPT-测试中」启动成功！
-2022-12-08 18:19:08,336	[INFO]	(gateway.py:217)_send_heart	[botpy] 心跳维持启动...
-2022-12-08 18:20:46,416	[INFO]	(client.py:159)_bot_login	[botpy] 登录机器人账号中...
-2022-12-08 18:20:47,023	[INFO]	(client.py:178)_bot_init	[botpy] 程序启动...
-2022-12-08 18:20:47,023	[INFO]	(connection.py:59)multi_run	[botpy] 最大并发连接数: 1, 启动会话数: 1
-2022-12-08 18:20:47,024	[INFO]	(client.py:236)bot_connect	[botpy] 会话启动中...
-2022-12-08 18:20:47,025	[INFO]	(gateway.py:110)ws_connect	[botpy] 启动中...
-2022-12-08 18:20:47,475	[INFO]	(gateway.py:136)ws_identify	[botpy] 鉴权中...
-2022-12-08 18:20:47,640	[INFO]	(gateway.py:80)on_message	[botpy] 机器人「SoGPT-测试中」启动成功！
-2022-12-08 18:20:47,641	[INFO]	(gateway.py:217)_send_heart	[botpy] 心跳维持启动...
-2022-12-08 18:21:24,437	[INFO]	(client.py:159)_bot_login	[botpy] 登录机器人账号中...
-2022-12-08 18:21:24,991	[INFO]	(client.py:178)_bot_init	[botpy] 程序启动...
-2022-12-08 18:21:24,992	[INFO]	(connection.py:59)multi_run	[botpy] 最大并发连接数: 1, 启动会话数: 1
-2022-12-08 18:21:24,993	[INFO]	(client.py:236)bot_connect	[botpy] 会话启动中...
-2022-12-08 18:21:24,993	[INFO]	(gateway.py:110)ws_connect	[botpy] 启动中...
-2022-12-08 18:21:25,165	[INFO]	(gateway.py:136)ws_identify	[botpy] 鉴权中...
-2022-12-08 18:21:25,265	[INFO]	(gateway.py:80)on_message	[botpy] 机器人「SoGPT-测试中」启动成功！
-2022-12-08 18:21:25,266	[INFO]	(gateway.py:217)_send_heart	[botpy] 心跳维持启动...
-2022-12-08 18:24:18,469	[INFO]	(client.py:159)_bot_login	[botpy] 登录机器人账号中...
-2022-12-08 18:24:19,076	[INFO]	(client.py:178)_bot_init	[botpy] 程序启动...
-2022-12-08 18:24:19,077	[INFO]	(connection.py:59)multi_run	[botpy] 最大并发连接数: 1, 启动会话数: 1
-2022-12-08 18:24:19,077	[INFO]	(client.py:236)bot_connect	[botpy] 会话启动中...
-2022-12-08 18:24:19,078	[INFO]	(gateway.py:110)ws_connect	[botpy] 启动中...
-2022-12-08 18:24:19,252	[INFO]	(gateway.py:136)ws_identify	[botpy] 鉴权中...
-2022-12-08 18:24:19,339	[INFO]	(gateway.py:80)on_message	[botpy] 机器人「SoGPT-测试中」启动成功！
-2022-12-08 18:24:19,341	[INFO]	(gateway.py:217)_send_heart	[botpy] 心跳维持启动...
-2022-12-08 18:27:13,898	[INFO]	(client.py:159)_bot_login	[botpy] 登录机器人账号中...
-2022-12-08 18:27:14,553	[INFO]	(client.py:178)_bot_init	[botpy] 程序启动...
-2022-12-08 18:27:14,554	[INFO]	(connection.py:59)multi_run	[botpy] 最大并发连接数: 1, 启动会话数: 1
-2022-12-08 18:27:14,554	[INFO]	(client.py:236)bot_connect	[botpy] 会话启动中...
-2022-12-08 18:27:14,555	[INFO]	(gateway.py:110)ws_connect	[botpy] 启动中...
-2022-12-08 18:27:14,747	[INFO]	(gateway.py:136)ws_identify	[botpy] 鉴权中...
-2022-12-08 18:27:14,850	[INFO]	(gateway.py:80)on_message	[botpy] 机器人「SoGPT-测试中」启动成功！
-2022-12-08 18:27:14,851	[INFO]	(gateway.py:217)_send_heart	[botpy] 心跳维持启动...
-2022-12-08 18:30:47,647	[INFO]	(client.py:159)_bot_login	[botpy] 登录机器人账号中...
-2022-12-08 18:30:48,327	[INFO]	(client.py:178)_bot_init	[botpy] 程序启动...
-2022-12-08 18:30:48,328	[INFO]	(connection.py:59)multi_run	[botpy] 最大并发连接数: 1, 启动会话数: 1
-2022-12-08 18:30:48,329	[INFO]	(client.py:236)bot_connect	[botpy] 会话启动中...
-2022-12-08 18:30:48,330	[INFO]	(gateway.py:110)ws_connect	[botpy] 启动中...
-2022-12-08 18:30:48,736	[INFO]	(gateway.py:136)ws_identify	[botpy] 鉴权中...
-2022-12-08 18:30:48,873	[INFO]	(gateway.py:80)on_message	[botpy] 机器人「SoGPT-测试中」启动成功！
-2022-12-08 18:30:48,874	[INFO]	(gateway.py:217)_send_heart	[botpy] 心跳维持启动...
-2022-12-08 18:31:57,250	[INFO]	(client.py:159)_bot_login	[botpy] 登录机器人账号中...
-2022-12-08 18:31:57,955	[INFO]	(client.py:178)_bot_init	[botpy] 程序启动...
-2022-12-08 18:31:57,956	[INFO]	(connection.py:59)multi_run	[botpy] 最大并发连接数: 1, 启动会话数: 1
-2022-12-08 18:31:57,957	[INFO]	(client.py:236)bot_connect	[botpy] 会话启动中...
-2022-12-08 18:31:57,957	[INFO]	(gateway.py:110)ws_connect	[botpy] 启动中...
-2022-12-08 18:31:58,165	[INFO]	(gateway.py:136)ws_identify	[botpy] 鉴权中...
-2022-12-08 18:31:58,269	[INFO]	(gateway.py:80)on_message	[botpy] 机器人「SoGPT-测试中」启动成功！
-2022-12-08 18:31:58,270	[INFO]	(gateway.py:217)_send_heart	[botpy] 心跳维持启动...
-2022-12-08 19:01:58,099	[INFO]	(gateway.py:54)on_closed	[botpy] 关闭, 返回码: 4009, 返回信息: Session timed out
-2022-12-08 19:02:03,110	[INFO]	(connection.py:59)multi_run	[botpy] 最大并发连接数: 1, 启动会话数: 1
-2022-12-08 19:02:03,111	[INFO]	(client.py:236)bot_connect	[botpy] 会话启动中...
-2022-12-08 19:02:03,111	[INFO]	(gateway.py:110)ws_connect	[botpy] 启动中...
-2022-12-08 19:02:03,314	[INFO]	(gateway.py:169)ws_resume	[botpy] 重连启动...
-2022-12-08 19:02:03,381	[INFO]	(gateway.py:85)on_message	[botpy] 机器人重连成功! 
-2022-12-08 19:02:03,382	[INFO]	(gateway.py:217)_send_heart	[botpy] 心跳维持启动...
-2022-12-08 19:32:03,274	[INFO]	(gateway.py:54)on_closed	[botpy] 关闭, 返回码: 4009, 返回信息: Session timed out
-2022-12-08 19:32:08,271	[INFO]	(connection.py:59)multi_run	[botpy] 最大并发连接数: 1, 启动会话数: 1
-2022-12-08 19:32:08,272	[INFO]	(client.py:236)bot_connect	[botpy] 会话启动中...
-2022-12-08 19:32:08,272	[INFO]	(gateway.py:110)ws_connect	[botpy] 启动中...
-2022-12-08 19:32:08,534	[INFO]	(gateway.py:169)ws_resume	[botpy] 重连启动...
-2022-12-08 19:32:08,604	[INFO]	(gateway.py:85)on_message	[botpy] 机器人重连成功! 
-2022-12-08 19:32:08,605	[INFO]	(gateway.py:217)_send_heart	[botpy] 心跳维持启动...
-2022-12-08 20:02:08,495	[INFO]	(gateway.py:54)on_closed	[botpy] 关闭, 返回码: 4009, 返回信息: Session timed out
-2022-12-08 20:02:13,497	[INFO]	(connection.py:59)multi_run	[botpy] 最大并发连接数: 1, 启动会话数: 1
-2022-12-08 20:02:13,497	[INFO]	(client.py:236)bot_connect	[botpy] 会话启动中...
-2022-12-08 20:02:13,498	[INFO]	(gateway.py:110)ws_connect	[botpy] 启动中...
-2022-12-08 20:02:13,797	[INFO]	(gateway.py:169)ws_resume	[botpy] 重连启动...
-2022-12-08 20:02:13,914	[INFO]	(gateway.py:85)on_message	[botpy] 机器人重连成功! 
-2022-12-08 20:02:13,915	[INFO]	(gateway.py:217)_send_heart	[botpy] 心跳维持启动...
-2022-12-08 20:31:53,159	[INFO]	(client.py:159)_bot_login	[botpy] 登录机器人账号中...
-2022-12-08 20:31:58,306	[INFO]	(client.py:178)_bot_init	[botpy] 程序启动...
-2022-12-08 20:31:58,307	[INFO]	(connection.py:59)multi_run	[botpy] 最大并发连接数: 1, 启动会话数: 1
-2022-12-08 20:31:58,308	[INFO]	(client.py:236)bot_connect	[botpy] 会话启动中...
-2022-12-08 20:31:58,308	[INFO]	(gateway.py:110)ws_connect	[botpy] 启动中...
-2022-12-08 20:31:58,748	[INFO]	(gateway.py:136)ws_identify	[botpy] 鉴权中...
-2022-12-08 20:31:58,924	[INFO]	(gateway.py:80)on_message	[botpy] 机器人「SoGPT-测试中」启动成功！
-2022-12-08 20:31:58,925	[INFO]	(gateway.py:217)_send_heart	[botpy] 心跳维持启动...
-2022-12-08 20:34:26,596	[INFO]	(client.py:159)_bot_login	[botpy] 登录机器人账号中...
-2022-12-08 20:34:27,610	[INFO]	(client.py:178)_bot_init	[botpy] 程序启动...
-2022-12-08 20:34:27,611	[INFO]	(connection.py:59)multi_run	[botpy] 最大并发连接数: 1, 启动会话数: 1
-2022-12-08 20:34:27,613	[INFO]	(client.py:236)bot_connect	[botpy] 会话启动中...
-2022-12-08 20:34:27,613	[INFO]	(gateway.py:110)ws_connect	[botpy] 启动中...
-2022-12-08 20:34:27,919	[INFO]	(gateway.py:136)ws_identify	[botpy] 鉴权中...
-2022-12-08 20:34:28,022	[INFO]	(gateway.py:80)on_message	[botpy] 机器人「SoGPT-测试中」启动成功！
-2022-12-08 20:34:28,023	[INFO]	(gateway.py:217)_send_heart	[botpy] 心跳维持启动...
-2022-12-08 20:35:47,952	[INFO]	(client.py:159)_bot_login	[botpy] 登录机器人账号中...
-2022-12-08 20:35:48,499	[INFO]	(client.py:178)_bot_init	[botpy] 程序启动...
-2022-12-08 20:35:48,500	[INFO]	(connection.py:59)multi_run	[botpy] 最大并发连接数: 1, 启动会话数: 1
-2022-12-08 20:35:48,500	[INFO]	(client.py:236)bot_connect	[botpy] 会话启动中...
-2022-12-08 20:35:48,501	[INFO]	(gateway.py:110)ws_connect	[botpy] 启动中...
-2022-12-08 20:35:48,723	[INFO]	(gateway.py:136)ws_identify	[botpy] 鉴权中...
-2022-12-08 20:35:48,793	[INFO]	(gateway.py:80)on_message	[botpy] 机器人「SoGPT-测试中」启动成功！
-2022-12-08 20:35:48,794	[INFO]	(gateway.py:217)_send_heart	[botpy] 心跳维持启动...
-2022-12-08 20:55:03,450	[ERROR]	(http.py:73)_handle_response	[botpy] 接口请求异常，请求连接: https://api.sgroup.qq.com/channels/7150658/messages, 错误代码: 403, 返回内容: {'code': 304003, 'message': 'url not allowed'}, trace_id:c4c4d23f9b2829d03ae9a7dca0184fe9
-2022-12-08 21:05:48,678	[INFO]	(gateway.py:54)on_closed	[botpy] 关闭, 返回码: 4009, 返回信息: Session timed out
-2022-12-08 21:05:53,700	[INFO]	(connection.py:59)multi_run	[botpy] 最大并发连接数: 1, 启动会话数: 1
-2022-12-08 21:05:53,701	[INFO]	(client.py:236)bot_connect	[botpy] 会话启动中...
-2022-12-08 21:05:53,702	[INFO]	(gateway.py:110)ws_connect	[botpy] 启动中...
-2022-12-08 21:05:54,145	[INFO]	(gateway.py:169)ws_resume	[botpy] 重连启动...
-2022-12-08 21:05:54,232	[INFO]	(gateway.py:85)on_message	[botpy] 机器人重连成功! 
-2022-12-08 21:05:54,233	[INFO]	(gateway.py:217)_send_heart	[botpy] 心跳维持启动...
-2022-12-08 21:07:39,468	[INFO]	(client.py:159)_bot_login	[botpy] 登录机器人账号中...
-2022-12-08 21:07:40,244	[INFO]	(client.py:178)_bot_init	[botpy] 程序启动...
-2022-12-08 21:07:40,245	[INFO]	(connection.py:59)multi_run	[botpy] 最大并发连接数: 1, 启动会话数: 1
-2022-12-08 21:07:40,246	[INFO]	(client.py:236)bot_connect	[botpy] 会话启动中...
-2022-12-08 21:07:40,247	[INFO]	(gateway.py:110)ws_connect	[botpy] 启动中...
-2022-12-08 21:07:40,553	[INFO]	(gateway.py:136)ws_identify	[botpy] 鉴权中...
-2022-12-08 21:07:40,659	[INFO]	(gateway.py:80)on_message	[botpy] 机器人「SoGPT-测试中」启动成功！
-2022-12-08 21:07:40,660	[INFO]	(gateway.py:217)_send_heart	[botpy] 心跳维持启动...
-2022-12-08 21:30:11,249	[INFO]	(client.py:159)_bot_login	[botpy] 登录机器人账号中...
-2022-12-08 21:30:12,217	[INFO]	(client.py:178)_bot_init	[botpy] 程序启动...
-2022-12-08 21:30:12,218	[INFO]	(connection.py:59)multi_run	[botpy] 最大并发连接数: 1, 启动会话数: 1
-2022-12-08 21:30:12,219	[INFO]	(client.py:236)bot_connect	[botpy] 会话启动中...
-2022-12-08 21:30:12,219	[INFO]	(gateway.py:110)ws_connect	[botpy] 启动中...
-2022-12-08 21:30:12,627	[INFO]	(gateway.py:136)ws_identify	[botpy] 鉴权中...
-2022-12-08 21:30:12,730	[INFO]	(gateway.py:80)on_message	[botpy] 机器人「SoGPT-测试中」启动成功！
-2022-12-08 21:30:12,731	[INFO]	(gateway.py:217)_send_heart	[botpy] 心跳维持启动...
-2022-12-08 21:36:34,660	[ERROR]	(http.py:73)_handle_response	[botpy] 接口请求异常，请求连接: https://api.sgroup.qq.com/channels/7150658/messages, 错误代码: 501, 返回内容: None, trace_id:None
-2022-12-08 21:38:39,294	[ERROR]	(http.py:73)_handle_response	[botpy] 接口请求异常，请求连接: https://api.sgroup.qq.com/channels/7150658/messages, 错误代码: 501, 返回内容: None, trace_id:None
-2022-12-08 21:39:43,233	[INFO]	(client.py:159)_bot_login	[botpy] 登录机器人账号中...
-2022-12-08 21:39:43,878	[INFO]	(client.py:178)_bot_init	[botpy] 程序启动...
-2022-12-08 21:39:43,879	[INFO]	(connection.py:59)multi_run	[botpy] 最大并发连接数: 1, 启动会话数: 1
-2022-12-08 21:39:43,880	[INFO]	(client.py:236)bot_connect	[botpy] 会话启动中...
-2022-12-08 21:39:43,880	[INFO]	(gateway.py:110)ws_connect	[botpy] 启动中...
-2022-12-08 21:39:44,099	[INFO]	(gateway.py:136)ws_identify	[botpy] 鉴权中...
-2022-12-08 21:39:44,279	[INFO]	(gateway.py:80)on_message	[botpy] 机器人「SoGPT-测试中」启动成功！
-2022-12-08 21:39:44,281	[INFO]	(gateway.py:217)_send_heart	[botpy] 心跳维持启动...
-2022-12-08 21:40:38,103	[INFO]	(client.py:159)_bot_login	[botpy] 登录机器人账号中...
-2022-12-08 21:40:39,100	[INFO]	(client.py:178)_bot_init	[botpy] 程序启动...
-2022-12-08 21:40:39,101	[INFO]	(connection.py:59)multi_run	[botpy] 最大并发连接数: 1, 启动会话数: 1
-2022-12-08 21:40:39,102	[INFO]	(client.py:236)bot_connect	[botpy] 会话启动中...
-2022-12-08 21:40:39,102	[INFO]	(gateway.py:110)ws_connect	[botpy] 启动中...
-2022-12-08 21:40:39,402	[INFO]	(gateway.py:136)ws_identify	[botpy] 鉴权中...
-2022-12-08 21:40:39,502	[INFO]	(gateway.py:80)on_message	[botpy] 机器人「SoGPT-测试中」启动成功！
-2022-12-08 21:40:39,503	[INFO]	(gateway.py:217)_send_heart	[botpy] 心跳维持启动...
-2022-12-08 21:40:52,805	[ERROR]	(http.py:73)_handle_response	[botpy] 接口请求异常，请求连接: https://api.sgroup.qq.com/channels/7150658/messages, 错误代码: 501, 返回内容: None, trace_id:None
-2022-12-08 21:41:39,703	[INFO]	(client.py:159)_bot_login	[botpy] 登录机器人账号中...
-2022-12-08 21:41:40,399	[INFO]	(client.py:178)_bot_init	[botpy] 程序启动...
-2022-12-08 21:41:40,400	[INFO]	(connection.py:59)multi_run	[botpy] 最大并发连接数: 1, 启动会话数: 1
-2022-12-08 21:41:40,401	[INFO]	(client.py:236)bot_connect	[botpy] 会话启动中...
-2022-12-08 21:41:40,401	[INFO]	(gateway.py:110)ws_connect	[botpy] 启动中...
-2022-12-08 21:41:40,644	[INFO]	(gateway.py:136)ws_identify	[botpy] 鉴权中...
-2022-12-08 21:41:40,740	[INFO]	(gateway.py:80)on_message	[botpy] 机器人「SoGPT-测试中」启动成功！
-2022-12-08 21:41:40,741	[INFO]	(gateway.py:217)_send_heart	[botpy] 心跳维持启动...
-2022-12-08 21:41:56,705	[ERROR]	(http.py:73)_handle_response	[botpy] 接口请求异常，请求连接: https://api.sgroup.qq.com/channels/7150658/messages, 错误代码: 501, 返回内容: None, trace_id:None
-2022-12-08 21:42:55,625	[INFO]	(client.py:159)_bot_login	[botpy] 登录机器人账号中...
-2022-12-08 21:42:56,278	[INFO]	(client.py:178)_bot_init	[botpy] 程序启动...
-2022-12-08 21:42:56,279	[INFO]	(connection.py:59)multi_run	[botpy] 最大并发连接数: 1, 启动会话数: 1
-2022-12-08 21:42:56,280	[INFO]	(client.py:236)bot_connect	[botpy] 会话启动中...
-2022-12-08 21:42:56,281	[INFO]	(gateway.py:110)ws_connect	[botpy] 启动中...
-2022-12-08 21:42:56,718	[INFO]	(gateway.py:136)ws_identify	[botpy] 鉴权中...
-2022-12-08 21:42:56,819	[INFO]	(gateway.py:80)on_message	[botpy] 机器人「SoGPT-测试中」启动成功！
-2022-12-08 21:42:56,820	[INFO]	(gateway.py:217)_send_heart	[botpy] 心跳维持启动...
-2022-12-08 21:43:09,794	[ERROR]	(http.py:73)_handle_response	[botpy] 接口请求异常，请求连接: https://api.sgroup.qq.com/channels/7150658/messages, 错误代码: 501, 返回内容: None, trace_id:None

configs/config.yaml

@@ -1,76 +0,0 @@
-# 如果你不知道怎么部署，请务必查看https://soulter.top/posts/qpdg.html
-
-# 注意：已支持多key自动切换，方法：
-# key:
-#   - sk-xxxxxx
-#   - sk-xxxxxx
-# 在下方非注释的地方使用以上格式
-openai:
-  key: 
-    - 
-    
-  # 这里是GPT配置，语言模型默认使用gpt-3.5-turbo
-  chatGPTConfigs:
-    model: gpt-3.5-turbo
-    max_tokens: 1500
-    temperature: 0.9
-    top_p: 1
-    frequency_penalty: 0
-    presence_penalty: 0
-    
-  total_tokens_limit: 2800
-
-# QQ开放平台的appid和令牌
-# q.qq.com
-qqbot:
-  appid: 
-  token: 
-
-# 设置是否一个人一个会话
-uniqueSessionMode: false
-
-# QChannelBot 的版本，请勿修改此字段，否则可能产生一些bug
-version: 2.5 RealChatGPT Ver.
-
-# [Beta] 转储历史记录时间间隔(分钟)
-dump_history_interval: 10
-
-# 一个用户只能在time秒内发送count条消息
-limit:
-  time: 60
-  count: 5
-# 公告
-notice: "此机器人由Github项目QQChannelChatGPT驱动。"
-
-# 是否打开私信功能
-# 设置为true则频道成员可以私聊机器人。
-# 设置为false则频道成员不能私聊机器人。
-direct_message_mode: true
-
-################外带程序（插件）################
-
-# 逆向ChatGPT库
-# https://github.com/acheong08/ChatGPT
-# 优点：免费（无免费额度限制）；
-# 缺点：速度相对慢。OpenAI 速率限制：免费帐户每小时 50 个请求。您可以通过多帐户循环来绕过它
-# enable设置为true后，将会停止使用上面正常的官方API调用而使用本逆向项目
-#
-# 多账户可以保证每个请求都能得到及时的回复。
-# 关于account的格式
-# account:
-#   - email: 第1个账户
-#     password: 第1个账户密码
-#   - email: 第2个账户
-#     password: 第2个账户密码
-#   - ....
-# 请严格按照上面这个格式填写。
-# 这里我免费提供2个账号给大家，不过用的人一定会很多的，所以会造成一些bug，因此还是用自己的账号好一点。
-# 需要账号可以联系我。QQ905617992
-
-rev_ChatGPT:
-  enable: false
-  account:
-    - email: d.o.m.her.ry61.7@gmail.com
-      password: 11111111
-    - email: ca.it.li.nal.o.i.si.o91@gmail.com
-      password: 11111111

cores/database/conn.py

@@ -1,86 +0,0 @@
-import sqlite3
-import yaml
-
-# TODO: 数据库缓存prompt
-
-class dbConn():
-    def __init__(self):
-        # 读取参数,并支持中文
-        conn = sqlite3.connect("data.db")
-        conn.text_factory=str
-        self.conn = conn
-        c = conn.cursor()
-        c.execute(
-            '''
-            CREATE TABLE IF NOT EXISTS tb_session(
-                qq_id   VARCHAR(32) PRIMARY KEY,
-                history TEXT
-            )
-            '''
-        )
-        
-        conn.commit()
-
-    def insert_session(self, qq_id, history):
-        conn = self.conn
-        c = conn.cursor()
-        c.execute(
-            '''
-            INSERT INTO tb_session(qq_id, history) VALUES (?, ?)
-            ''', (qq_id, history)
-        )
-        conn.commit()
-
-    def update_session(self, qq_id, history):
-        conn = self.conn
-        c = conn.cursor()
-        c.execute(
-            '''
-            UPDATE tb_session SET history = ? WHERE qq_id = ?
-            ''', (history, qq_id)
-        )
-        conn.commit()
-
-    def get_session(self, qq_id):
-        conn = self.conn
-        c = conn.cursor()
-        c.execute(
-            '''
-            SELECT * FROM tb_session WHERE qq_id = ?
-            ''', (qq_id, )
-        )
-        return c.fetchone()
-    
-    def get_all_session(self):
-        conn = self.conn
-        c = conn.cursor()
-        c.execute(
-            '''
-            SELECT * FROM tb_session
-            '''
-        )
-        return c.fetchall()
-    
-    def check_session(self, qq_id):
-        conn = self.conn
-        c = conn.cursor()
-        c.execute(
-            '''
-            SELECT * FROM tb_session WHERE qq_id = ?
-            ''', (qq_id, )
-        )
-        return c.fetchone() is not None
-
-    def delete_session(self, qq_id):
-        conn = self.conn
-        c = conn.cursor()
-        c.execute(
-            '''
-            DELETE FROM tb_session WHERE qq_id = ?
-            ''', (qq_id, )
-        )
-        conn.commit()
-
-    def close(self):
-        self.conn.close()
-    

cores/openai/core.py

@@ -1,172 +0,0 @@
-import openai
-import yaml
-from util.errors.errors import PromptExceededError
-import json
-import time
-import os
-import sys
-
-inst = None
-# 适配pyinstaller
-abs_path = os.path.dirname(os.path.realpath(sys.argv[0])) + '/'
-key_record_path = abs_path+'chatgpt_key_record'
-
-class ChatGPT:
-    def __init__(self, cfg):
-        self.key_list = []
-        if cfg['key'] != '' or cfg['key'] != '修改我！！':
-            print("[System] 读取ChatGPT Key成功")
-            self.key_list = cfg['key']
-            # openai.api_key = cfg['key']
-        else:
-            input("[System] 请先去完善ChatGPT的Key。详情请前往https://beta.openai.com/account/api-keys")
-        
-        # init key record
-        self.init_key_record()
-
-        chatGPT_configs = cfg['chatGPTConfigs']
-        print(f'[System] 加载ChatGPTConfigs: {chatGPT_configs}')
-        self.chatGPT_configs = chatGPT_configs
-        self.openai_configs = cfg
-    
-    def chat(self, prompt, image_mode = False):
-        # ChatGPT API 2023/3/2
-        messages = [{"role": "user", "content": prompt}]
-        try:
-            if not image_mode:
-                
-                response = openai.ChatCompletion.create(
-                    messages=messages,
-                    **self.chatGPT_configs
-                )
-            else:
-                response = openai.Image.create(
-                    prompt=prompt,
-                    n=1,
-                    size="512x512",
-                )
-        except Exception as e:
-            print(e)
-            if 'You exceeded' in str(e) or 'Billing hard limit has been reached' in str(e) or 'No API key provided.' in str(e):
-                print("[System] 当前Key已超额,正在切换")
-                self.key_stat[openai.api_key]['exceed'] = True
-                self.save_key_record()
-
-                response, is_switched = self.handle_switch_key(prompt)
-                if not is_switched:
-                    # 所有Key都超额
-                    raise e
-            else:
-                if not image_mode:
-                    response = openai.ChatCompletion.create(
-                        messages=messages,
-                        **self.chatGPT_configs
-                    )
-                else:
-                    response = openai.Image.create(
-                        prompt=prompt,
-                        n=1,
-                        size="512x512",
-                    )
-        if not image_mode:
-            self.key_stat[openai.api_key]['used'] += response['usage']['total_tokens']
-            self.save_key_record()
-            print("[ChatGPT] "+str(response["choices"][0]["message"]["content"]))
-            return str(response["choices"][0]["message"]["content"]).strip(), response['usage']['total_tokens']
-        else:
-            return response['data'][0]['url']
-            
-    def handle_switch_key(self, prompt):
-        messages = [{"role": "user", "content": prompt}]
-        while True:
-            is_all_exceed = True
-            for key in self.key_stat:
-                if not self.key_stat[key]['exceed']:
-                    is_all_exceed = False
-                    openai.api_key = key
-                    print(f"[System] 切换到Key: {key}, 已使用token: {self.key_stat[key]['used']}")
-                    if prompt != '':
-                        try:
-                            response = openai.ChatCompletion.create(
-                                messages=messages,
-                                **self.chatGPT_configs
-                            )
-                            return response, True
-                        except Exception as e:
-                            print(e)
-                            if 'You exceeded' in str(e):
-                                print("[System] 当前Key已超额,正在切换")
-                                self.key_stat[openai.api_key]['exceed'] = True
-                                self.save_key_record()
-                                time.sleep(1)
-                                continue
-                    else:
-                        return True
-            if is_all_exceed:
-                print("[System] 所有Key已超额")
-                return None, False
-                
-    def getConfigs(self):
-        return self.openai_configs
-    
-    def save_key_record(self):
-        with open(key_record_path, 'w', encoding='utf-8') as f:
-            json.dump(self.key_stat, f)
-
-    def get_key_stat(self):
-        return self.key_stat
-    def get_key_list(self):
-        return self.key_list
-    
-    # 添加key
-    def append_key(self, key, sponsor):
-        self.key_list.append(key)
-        self.key_stat[key] = {'exceed': False, 'used': 0, 'sponsor': sponsor}
-        self.save_key_record()
-        self.init_key_record()
-    # 检查key是否可用
-    def check_key(self, key):
-        pre_key = openai.api_key
-        openai.api_key = key
-        messages = [{"role": "user", "content": "1"}]
-        try:
-            response = openai.ChatCompletion.create(
-                messages=messages,
-                **self.chatGPT_configs
-            )
-            openai.api_key = pre_key
-            return True
-        except Exception as e:
-            pass
-        openai.api_key = pre_key
-        return False
-
-    #将key_list的key转储到key_record中，并记录相关数据
-    def init_key_record(self):
-        if not os.path.exists(key_record_path):
-            with open(key_record_path, 'w', encoding='utf-8') as f:
-                json.dump({}, f)
-        with open(key_record_path, 'r', encoding='utf-8') as keyfile:
-            try:
-                self.key_stat = json.load(keyfile)
-            except Exception as e:
-                print(e)
-                self.key_stat = {}
-            finally:
-                for key in self.key_list:
-                    if key not in self.key_stat:
-                        self.key_stat[key] = {'exceed': False, 'used': 0}
-                        # if openai.api_key is None:
-                        #     openai.api_key = key
-                    else:
-                        # if self.key_stat[key]['exceed']:
-                        #     print(f"Key: {key} 已超额")
-                        #     continue
-                        # else:
-                        #     if openai.api_key is None:
-                        #         openai.api_key = key
-                        #         print(f"使用Key: {key}, 已使用token: {self.key_stat[key]['used']}")
-                        pass
-                if openai.api_key == None:
-                    self.handle_switch_key("")
-            self.save_key_record()

cores/qqbot/core.py

@@ -1,768 +0,0 @@
-import botpy
-from botpy.message import Message
-import yaml
-import re
-from util.errors.errors import PromptExceededError
-from botpy.message import DirectMessage
-import json
-import threading
-import asyncio
-import time
-from cores.database.conn import dbConn
-import requests
-import util.unfit_words as uw
-import os
-import sys
-from cores.qqbot.personality import personalities
-
-
-history_dump_interval = 10
-# QQBotClient实例
-client = ''
-# ChatGPT实例
-global chatgpt
-# 缓存的会话
-session_dict = {}
-# 最大缓存token（在配置里改 configs/config.yaml）
-max_tokens = 2000
-# 配置信息
-config = {}
-# 统计信息
-count = {}
-# 统计信息
-stat_file = ''
-# 是否独立会话默认值
-uniqueSession = False
-
-# 日志记录
-logf = open('log.log', 'a+', encoding='utf-8')
-# 是否上传日志,仅上传频道数量等数量的统计信息
-is_upload_log = True
-
-# 用户发言频率
-user_frequency = {}
-# 时间默认值
-frequency_time = 60
-# 计数默认值
-frequency_count = 2
-
-# 公告（可自定义）：
-announcement = ""
-
-# 人格信息
-now_personality = {}
-
-# 机器人私聊模式
-direct_message_mode = True
-
-# 适配pyinstaller
-abs_path = os.path.dirname(os.path.realpath(sys.argv[0])) + '/'
-
-# 版本
-version = '2.4 RealChatGPT Ver.'
-
-# 语言模型提供商
-REV_CHATGPT = 'rev_chatgpt'
-OPENAI_OFFICIAL = 'openai_official'
-provider = ''
-
-# 逆向库对象及负载均衡
-rev_chatgpt = []
-
-# gpt配置信息
-gpt_config = {}
-
-def new_sub_thread(func, args=()):
-    thread = threading.Thread(target=func, args=args, daemon=True)
-    thread.start() 
-
-class botClient(botpy.Client):
-    # 收到At消息
-    async def on_at_message_create(self, message: Message):
-        toggle_count(at=True, message=message)
-        # executor.submit(oper_msg, message, True)
-        new_sub_thread(oper_msg, (message, True))
-        # await oper_msg(message=message, at=True)
-
-    # 收到私聊消息
-    async def on_direct_message_create(self, message: DirectMessage):
-        if direct_message_mode:
-            toggle_count(at=False, message=message)
-            # executor.submit(oper_msg, message, True)
-            # await oper_msg(message=message, at=False)
-            new_sub_thread(oper_msg, (message, False))
-
-# 写入统计信息
-def toggle_count(at: bool, message):
-    global stat_file
-    try: 
-        if str(message.guild_id) not in count:
-            count[str(message.guild_id)] = {
-                'count': 1,
-                'direct_count': 1,
-            }
-        else:
-            count[str(message.guild_id)]['count'] += 1
-            if not at:
-                count[str(message.guild_id)]['direct_count'] += 1
-        stat_file = open(abs_path+"configs/stat", 'w', encoding='utf-8')
-        stat_file.write(json.dumps(count))
-        stat_file.flush()
-        stat_file.close()
-    except BaseException:
-        pass
-
-# 转储历史记录的定时器~ Soulter
-def dump_history():
-    time.sleep(10)
-    global session_dict, history_dump_interval
-    db = dbConn()
-    while True:
-        try:
-            # print("转储历史记录...")
-            for key in session_dict:
-                # print("TEST: "+str(db.get_session(key)))
-                data = session_dict[key]
-                data_json = {
-                    'data': data
-                }
-                if db.check_session(key):
-                    db.update_session(key, json.dumps(data_json))
-                else:
-                    db.insert_session(key, json.dumps(data_json))
-            # print("转储历史记录完毕")
-        except BaseException as e:
-            print(e)
-        # 每隔10分钟转储一次
-        time.sleep(10*history_dump_interval)
-
-# 上传统计信息并检查更新
-def upload():
-    global object_id
-    global version
-    while True:
-        addr = ''
-        try:
-            # 用户唯一性标识
-            addr = requests.get('http://myip.ipip.net', timeout=5).text
-        except BaseException:
-            pass
-        try:
-            ts = str(time.time())
-            guild_count, guild_msg_count, guild_direct_msg_count, session_count = get_stat()
-            headers = {
-                'X-LC-Id': 'UqfXTWW15nB7iMT0OHvYrDFb-gzGzoHsz',
-                'X-LC-Key': 'QAZ1rQLY1ZufHrZlpuUiNff7',
-                'Content-Type': 'application/json'
-            }
-            key_stat = chatgpt.get_key_stat()
-            d = {"data": {"guild_count": guild_count, "guild_msg_count": guild_msg_count, "guild_direct_msg_count": guild_direct_msg_count, "session_count": session_count, 'addr': addr, 'winver': version, 'key_stat':key_stat}}
-            d = json.dumps(d).encode("utf-8")
-            res = requests.put(f'https://uqfxtww1.lc-cn-n1-shared.com/1.1/classes/bot_record/{object_id}', headers = headers, data = d)
-            if json.loads(res.text)['code'] == 1:
-                print("[System] New User.")
-                res = requests.post(f'https://uqfxtww1.lc-cn-n1-shared.com/1.1/classes/bot_record', headers = headers, data = d)
-                object_id = json.loads(res.text)['objectId']
-                object_id_file = open(abs_path+"configs/object_id", 'w+', encoding='utf-8')
-                object_id_file.write(str(object_id))
-                object_id_file.flush()
-                object_id_file.close()
-        except BaseException as e:
-            pass
-        # 每隔2小时上传一次
-        time.sleep(60*60*2)
-
-'''
-初始化机器人
-'''
-def initBot(cfg, prov):
-    global chatgpt, provider, rev_chatgpt
-    global now_personality, gpt_config, config, uniqueSession, history_dump_interval, frequency_count, frequency_time,announcement, direct_message_mode, version
-
-    provider = prov
-    config = cfg
-
-    # 语言模型提供商
-    if prov == REV_CHATGPT:
-        if 'account' in cfg['rev_ChatGPT']:
-            from addons.revChatGPT.revchatgpt import revChatGPT
-            for i in range(0, len(cfg['rev_ChatGPT']['account'])):
-                print(f"[System] 正在创建rev_ChatGPT负载{str(i)}: " + cfg['rev_ChatGPT']['account'][i]['email'])
-                revstat = {
-                    'obj': revChatGPT(cfg['rev_ChatGPT']['account'][i]),
-                    'busy': False
-                }
-                rev_chatgpt.append(revstat)
-        else:
-            input("[System-err] 请退出本程序, 然后在配置文件中填写rev_ChatGPT的email和password")
-    elif prov == OPENAI_OFFICIAL:
-        from cores.openai.core import ChatGPT
-        chatgpt = ChatGPT(cfg['openai'])
-        global max_tokens
-        max_tokens = int(chatgpt.getConfigs()['total_tokens_limit'])
-        
-        # 读取历史记录 Soulter
-        try:
-            db1 = dbConn()
-            for session in db1.get_all_session():
-                session_dict[session[0]] = json.loads(session[1])['data']
-            print("[System] 历史记录读取成功喵")
-        except BaseException as e:
-            print("[System] 历史记录读取失败: " + str(e))
-
-        # 读统计信息
-        global stat_file
-        if not os.path.exists(abs_path+"configs/stat"):
-            with open(abs_path+"configs/stat", 'w', encoding='utf-8') as f:
-                    json.dump({}, f)
-        stat_file = open(abs_path+"configs/stat", 'r', encoding='utf-8')
-        global count
-        res = stat_file.read()
-        if res == '':
-            count = {}
-        else:
-            try: 
-                count = json.loads(res)
-            except BaseException:
-                pass
-        # 创建转储定时器线程
-        threading.Thread(target=dump_history, daemon=True).start()
-
-        # 得到GPT配置信息
-        if 'openai' in cfg and 'chatGPTConfigs' in cfg['openai']:
-            gpt_config = cfg['openai']['chatGPTConfigs']
-
-    # 统计上传
-    if is_upload_log:
-        # 读取object_id
-        global object_id
-        if not os.path.exists(abs_path+"configs/object_id"):
-            with open(abs_path+"configs/object_id", 'w', encoding='utf-8') as f:
-                f.write("")
-        object_id_file = open(abs_path+"configs/object_id", 'r', encoding='utf-8')
-        object_id = object_id_file.read()
-        object_id_file.close()
-        # 创建上传定时器线程
-        threading.Thread(target=upload, daemon=True).start()
-    
-    # 得到私聊模式配置
-    if 'direct_message_mode' in cfg:
-        direct_message_mode = cfg['direct_message_mode']
-        print("[System] 私聊功能: "+str(direct_message_mode))
-
-    # 得到版本
-    if 'version' in cfg:
-        version = cfg['version']
-        print("[System] QQChannelChatGPT版本: "+str(version))
-
-    # 得到发言频率配置
-    if 'limit' in cfg:
-        print('[System] 发言频率配置: '+str(cfg['limit']))
-        if 'count' in cfg['limit']:
-            frequency_count = cfg['limit']['count']
-        if 'time' in cfg['limit']:
-            frequency_time = cfg['limit']['time']
-    
-    announcement += '[QQChannelChatGPT项目]\n所有回答与腾讯公司无关。出现问题请前往[GPT机器人]官方频道\n\n'
-    # 得到公告配置
-    if 'notice' in cfg:
-        print('[System] 公告配置: '+cfg['notice'])
-        announcement += cfg['notice']
-    try:
-        if 'uniqueSessionMode' in cfg and cfg['uniqueSessionMode']:
-            uniqueSession = True
-        else:
-            uniqueSession = False
-        print("[System] 独立会话: " + str(uniqueSession))
-        if 'dump_history_interval' in cfg:
-            history_dump_interval = int(cfg['dump_history_interval'])
-            print("[System] 历史记录转储时间周期: " + str(history_dump_interval) + "分钟")
-    except BaseException:
-        print("[System-Error] 读取uniqueSessionMode/version/dump_history_interval配置文件失败, 使用默认值。")
-
-    print(f"[System] QQ开放平台AppID: {cfg['qqbot']['appid']} 令牌: {cfg['qqbot']['token']}")
-
-    print("\n[System] 如果有任何问题，请在https://github.com/Soulter/QQChannelChatGPT上提交issue说明问题！或者添加QQ：905617992")
-    print("[System] 请给https://github.com/Soulter/QQChannelChatGPT点个star!")
-    print("[System] 请给https://github.com/Soulter/QQChannelChatGPT点个star!")
-    input("\n仔细阅读完以上信息后，输入任意信息并回车以继续")
-    try:
-        run_bot(cfg['qqbot']['appid'], cfg['qqbot']['token'])
-    except BaseException as e:
-        input(f"\n[System-Error] 启动QQ机器人时出现错误，原因如下：{e}\n可能是没有填写QQBOT appid和token？请在config中完善你的appid和token\n配置教程：https://soulter.top/posts/qpdg.html\n")
-
-        
-'''
-启动机器人
-'''
-def run_bot(appid, token):
-    intents = botpy.Intents(public_guild_messages=True, direct_message=True) 
-    global client
-    client = botClient(intents=intents)
-    client.run(appid=appid, token=token)
-
-'''
-得到OpenAI官方API的回复
-'''
-def get_chatGPT_response(prompts_str, image_mode=False):
-    res = ''
-    usage = ''
-    if not image_mode:
-        res, usage = chatgpt.chat(prompts_str)
-        # 处理结果文本
-        chatgpt_res = res.strip()
-        return res, usage
-    else:
-        res = chatgpt.chat(prompts_str, image_mode = True)
-        return res
-
-'''
-负载均衡，得到逆向ChatGPT回复
-'''
-def get_rev_ChatGPT_response(prompts_str):
-    res = ''
-    print("[Debug] "+str(rev_chatgpt))
-    for revstat in rev_chatgpt:
-        if not revstat['busy']:
-            revstat['busy'] = True
-            print("[Debug] 使用逆向ChatGPT回复ing", end='')
-            res = revstat['obj'].chat(prompts_str)
-            print("OK")
-            revstat['busy'] = False
-            # 处理结果文本
-            chatgpt_res = res.strip()
-            return res
-    res = '所有的OpenAI账号都有负载, 请稍后再试~'
-    return res
-
-
-'''
-回复QQ消息
-'''
-def send_qq_msg(message, res, image_mode=False):
-    if not image_mode:
-        try:
-            asyncio.run_coroutine_threadsafe(message.reply(content=res), client.loop)
-        except:
-            raise
-    else:
-        asyncio.run_coroutine_threadsafe(message.reply(image=res, content=""), client.loop)
-
-
-'''
-获取缓存的会话
-'''
-def get_prompts_by_cache_list(cache_data_list, divide=False, paging=False, size=5, page=1):
-    prompts = ""
-    if paging:
-        page_begin = (page-1)*size
-        page_end = page*size
-        if page_begin < 0:
-            page_begin = 0
-        if page_end > len(cache_data_list):
-            page_end = len(cache_data_list)
-        cache_data_list = cache_data_list[page_begin:page_end]
-    for item in cache_data_list:
-        prompts += str(item['prompt'])
-        if divide:
-            prompts += "----------\n"
-    return prompts
-    
-def get_user_usage_tokens(cache_list):
-    usage_tokens = 0
-    for item in cache_list:
-        usage_tokens += int(item['single_tokens'])
-    return usage_tokens
-
-'''
-检查发言频率
-'''
-def check_frequency(id) -> bool:
-    ts = int(time.time())
-    if id in user_frequency:
-        if ts-user_frequency[id]['time'] > frequency_time:
-            user_frequency[id]['time'] = ts
-            user_frequency[id]['count'] = 1
-            return True
-        else:
-            if user_frequency[id]['count'] >= frequency_count:
-                return False
-            else:
-                user_frequency[id]['count']+=1
-                return True
-    else:
-        t = {'time':ts,'count':1}
-        user_frequency[id] = t
-        return True
-
-'''
-处理消息
-'''
-def oper_msg(message, at=False, loop=None):
-    global session_dict, provider
-    print("[QQBOT] 接收到消息："+ str(message.content))
-    qq_msg = ''
-    session_id = ''
-    name = ''
-    user_id = message.author.id
-    user_name = message.author.username
-    
-    # 检查发言频率
-    if not check_frequency(user_id):
-        send_qq_msg(message, f'{user_name}的发言超过频率限制(╯▔皿▔)╯。\n{frequency_time}秒内只能提问{frequency_count}次。')
-        return
-
-    logf.write("[QQBOT] "+ str(message.content)+'\n')
-    logf.flush()
-
-    if at:
-        qq_msg = message.content
-        lines = qq_msg.splitlines()
-        for i in range(len(lines)):
-            lines[i] = re.sub(r"<@!\d+>", "", lines[i])
-        qq_msg = "\n".join(lines).lstrip().strip()
-
-        if uniqueSession:
-            session_id = user_id
-        else:
-            session_id = message.channel_id
-    else:
-        qq_msg = message.content
-        session_id = user_id
-        
-    if uniqueSession:
-        name = user_name
-    else:
-        name = "频道"
-
-    command_type = -1
-    # 特殊指令
-    if qq_msg == "/继续":
-        qq_msg = "继续"
-    # 普通指令
-    else:
-        # 如果第一个字符是/，则为指令
-        if qq_msg[0] == "/":
-            res, go, command_type = command_oper(qq_msg, message, session_id, name, user_id, user_name, at)
-            send_qq_msg(message, res)
-            if not go:
-                return
-    if command_type == 1 and 'prompt' in now_personality:
-        # 设置人格
-        qq_msg = now_personality['prompt']
-
-    # 这里是预设，你可以按需更改
-    if qq_msg.strip() == 'hello' or qq_msg.strip() == '你好' or qq_msg.strip() == '':
-        send_qq_msg(message, f"你好呀~")
-        return
-    # if qq_msg.strip() == '傻逼' or qq_msg.strip() == 'sb':
-    #     send_qq_msg(message, f"好好好")
-    #     return
-    # if '喜多' in qq_msg:
-    #     send_qq_msg(message, 'http://rn2ztkeap.hn-bkt.clouddn.com/xd1.gif', True)
-    # if '台湾' in qq_msg or 'taiwan' in qq_msg:
-    #     send_qq_msg(message, "声明：台湾是中国的一部分。\n下面否定此结论的回答全部无效。")
-    
-    # 关键词拦截器
-    for i in uw.unfit_words_q:
-        if i in qq_msg.strip():
-            send_qq_msg(message, f"你的提问中有不太合适的内容😭\n请更换措辞~")
-            return
-        
-    # 会话机制
-    if session_id not in session_dict:
-        session_dict[session_id] = []
-
-        fjson = {}
-        try:
-            f = open(abs_path+"configs/session", "r", encoding="utf-8")
-            fjson = json.loads(f.read())
-            f.close()
-        except:
-            pass
-        finally:
-            fjson[session_id] = 'true'
-            f = open(abs_path+"configs/session", "w", encoding="utf-8")
-            f.write(json.dumps(fjson))
-            f.flush()
-            f.close()
-        
-    chatgpt_res = "[Error] 占位符"
-    if provider == OPENAI_OFFICIAL:
-
-        # 获取缓存
-        cache_prompt = ''
-        cache_data_list = session_dict[session_id]
-        cache_prompt = get_prompts_by_cache_list(cache_data_list)
-        cache_prompt += "\nHuman: "+ qq_msg + "\nAI: "
-    
-        # 请求chatGPT获得结果
-        try:
-            chatgpt_res, current_usage_tokens = get_chatGPT_response(prompts_str=cache_prompt)
-        except (BaseException) as e:
-            print("[System-Err] OpenAI API错误。原因如下:\n"+str(e))
-            if 'maximum context length' in str(e):
-                print("token超限, 清空对应缓存")
-                session_dict[session_id] = []
-                cache_data_list = []
-                cache_prompt = "Human: "+ qq_msg + "\nAI: "
-                chatgpt_res, current_usage_tokens = get_chatGPT_response(prompts_str=cache_prompt)
-            elif 'exceeded' in str(e):
-                send_qq_msg(message, f"OpenAI API错误。原因：\n{str(e)} \n超额了。可自己搭建一个机器人(Github仓库：QQChannelChatGPT)")
-            else:
-                send_qq_msg(message, f"OpenAI API错误。原因如下：\n{str(e)} \n前往官方频道反馈~")
-            return
-        
-        # 超过指定tokens， 尽可能的保留最多的条目，直到小于max_tokens
-        if current_usage_tokens > max_tokens:
-            t = current_usage_tokens
-            index = 0
-            while t > max_tokens:
-                if index >= len(cache_data_list):
-                    break
-                if 'level' in cache_data_list[index] and cache_data_list[index]['level'] != 'max':
-                    t -= int(cache_data_list[index]['single_tokens'])
-                    del cache_data_list[index]
-                else:
-                    index += 1
-            # 删除完后更新相关字段
-            session_dict[session_id] = cache_data_list
-            cache_prompt = get_prompts_by_cache_list(cache_data_list)
-
-        # 添加新条目进入缓存的prompt
-        if command_type == 1:
-            level = 'max'
-        else:
-            level = 'normal'
-        if len(cache_data_list) > 0: 
-            single_record = {
-                "prompt": f'Human: {qq_msg}\nAI: {chatgpt_res}\n',
-                "usage_tokens": current_usage_tokens,
-                "single_tokens": current_usage_tokens - int(cache_data_list[-1]['usage_tokens']),
-                "level": level
-            }
-        else:
-            single_record = {
-                "prompt": f'Human: {qq_msg}\nAI: {chatgpt_res}\n',
-                "usage_tokens": current_usage_tokens,
-                "single_tokens": current_usage_tokens,
-                "level": level
-            }
-        cache_data_list.append(single_record)
-        session_dict[session_id] = cache_data_list
-        
-    elif provider == REV_CHATGPT:
-        try:
-            chatgpt_res = "[Rev]"+str(get_rev_ChatGPT_response(qq_msg))
-        except BaseException as e:
-            print("[System-Err] Rev ChatGPT API错误。原因如下:\n"+str(e))
-            send_qq_msg(message, f"Rev ChatGPT API错误。原因如下：\n{str(e)} \n前往官方频道反馈~")
-            return
-
-    # 发送qq信息
-    try:
-        # 防止被qq频道过滤消息
-        gap_chatgpt_res = chatgpt_res.replace(".", " . ")
-        if '```' in gap_chatgpt_res:
-            chatgpt_res.replace('```', "")
-        # 过滤不合适的词
-        for i in uw.unfit_words:
-            if i in gap_chatgpt_res:
-                gap_chatgpt_res = gap_chatgpt_res.replace(i, "***")
-        # 发送信息
-        send_qq_msg(message, ''+gap_chatgpt_res)
-    except:
-        print("QQ频道API错误: \n"+str(e))
-        f_res = ""
-        for t in chatgpt_res:
-            f_res += t + ' '
-        try:
-            send_qq_msg(message, ''+f_res)
-            # send(message, f"QQ频道API错误：{str(e)}\n下面是格式化后的回答：\n{f_res}")
-        except:
-            # 如果还是不行则过滤url
-            f_res = re.sub(r'(https|http)?:\/\/(\w|\.|\/|\?|\=|\&|\%)*\b', '', f_res, flags=re.MULTILINE)
-            f_res = f_res.replace(".", "·")
-            send_qq_msg(message, ''+f_res)
-            # send(message, f"QQ频道API错误：{str(e)}\n下面是格式化后的回答：\n{f_res}")
-
-    # 记录日志
-    logf.write("[GPT] "+ str(chatgpt_res)+'\n')
-    logf.flush()
-
-
-'''
-获取统计信息
-'''
-def get_stat():
-    try:
-        f = open(abs_path+"configs/stat", "r", encoding="utf-8")
-        fjson = json.loads(f.read())
-        f.close()
-        guild_count = 0
-        guild_msg_count = 0
-        guild_direct_msg_count = 0
-
-        for k,v in fjson.items():
-            guild_count += 1
-            guild_msg_count += v['count']
-            guild_direct_msg_count += v['direct_count']
-        
-        session_count = 0
-
-        f = open(abs_path+"configs/session", "r", encoding="utf-8")
-        fjson = json.loads(f.read())
-        f.close()
-        for k,v in fjson.items():
-            session_count += 1
-        return guild_count, guild_msg_count, guild_direct_msg_count, session_count
-    except:
-        return -1, -1, -1, -1
-
-'''
-指令处理
-'''
-def command_oper(qq_msg, message, session_id, name, user_id, user_name, at):
-    go = False # 是否处理完指令后继续执行msg_oper后面的代码
-    msg = ''
-    global session_dict, now_personality, provider
-
-    # 指令返回值，/set设置人格是1
-    type = -1
-    
-    # 指令控制
-    if qq_msg == "/reset" or qq_msg == "/重置":
-        msg = ''
-        session_dict[session_id] = []
-        if at:
-            msg = f"{name}(id: {session_id})的历史记录重置成功\n\n{announcement}"
-        else:
-            msg = f"你的历史记录重置成功"
-    
-    if qq_msg[:4] == "/his":
-        if provider == REV_CHATGPT:
-            msg = "[QQChannelChatGPT]当前使用的语言模型提供商是Rev_ChatGPT, 不支持查看历史记录"
-            return msg, go, type
-        #分页，每页5条
-        msg = ''
-        size_per_page = 3
-        page = 1
-        if qq_msg[5:]:
-            page = int(qq_msg[5:])
-        # 检查是否有过历史记录
-        if session_id not in session_dict:
-            msg = f"{name} 的历史记录为空"
-        l = session_dict[session_id]
-        max_page = len(l)//size_per_page + 1 if len(l)%size_per_page != 0 else len(l)//size_per_page
-        p = get_prompts_by_cache_list(session_dict[session_id], divide=True, paging=True, size=size_per_page, page=page)
-        if at:
-            msg=f"{name}的历史记录如下：\n{p}\n第{page}页 | 共{max_page}页\n*输入/his 2跳转到第2页"
-        else:
-            msg=f"历史记录如下：\n{p}\n第{page}页 | 共{max_page}页\n*输入/his 2跳转到第2页\n\n{announcement}"
-    
-    if qq_msg == "/token":
-        if provider == REV_CHATGPT:
-            msg = "[QQChannelChatGPT]当前使用的语言模型提供商是Rev_ChatGPT, 不支持使用此指令"
-            return msg, go, type
-        msg = ''
-        if at:
-            msg=f"{name} 会话的token数: {get_user_usage_tokens(session_dict[session_id])}\n系统最大缓存token数: {max_tokens}"
-        else:
-            msg=f"会话的token数: {get_user_usage_tokens(session_dict[session_id])}\n系统最大缓存token数: {max_tokens}"
-    
-    if qq_msg == '/gpt':
-        if provider == REV_CHATGPT:
-            msg = "[QQChannelChatGPT]当前使用的语言模型提供商是Rev_ChatGPT, 不支持使用此指令"
-            return msg, go, type
-        global gpt_config
-        msg=f"OpenAI GPT配置:\n {gpt_config}"
-    
-    if qq_msg == "/status" or qq_msg == "/状态":
-        
-        chatgpt_cfg_str = ""
-        key_stat = chatgpt.get_key_stat()
-        key_list = chatgpt.get_key_list()
-        index = 1
-        max = 900000
-        gg_count = 0
-        total = 0
-        tag = ''
-        for key in key_stat.keys():
-            sponsor = ''
-            total += key_stat[key]['used']
-            if key_stat[key]['exceed']:
-                gg_count += 1
-                continue
-            if 'sponsor' in key_stat[key]:
-                sponsor = key_stat[key]['sponsor']
-            chatgpt_cfg_str += f"  |-{index}: {key_stat[key]['used']}/{max} {sponsor}赞助{tag}\n"
-            index += 1
-        msg = f"⭐使用情况({str(gg_count)}个已用):\n{chatgpt_cfg_str}⏰全频道已用{total}tokens\n{announcement}"
-    if qq_msg == "/count" or qq_msg == "/统计":
-        guild_count, guild_msg_count, guild_direct_msg_count, session_count = get_stat()
-        msg = f"当前会话数: {len(session_dict)}\n共有频道数: {guild_count} \n共有消息数: {guild_msg_count}\n私信数: {guild_direct_msg_count}\n历史会话数: {session_count}"
-    
-    if qq_msg == "/help":
-        ol_version = 'Unknown'
-        try:
-            global version
-            res = requests.get("https://soulter.top/channelbot/update.json")
-            res_obj = json.loads(res.text)
-            ol_version = res_obj['version']
-        except BaseException:
-            pass
-        msg = f"[Github项目名: QQChannelChatGPT，有问题请前往提交issue，欢迎Star此项目~]\n\n当前版本:{version}\n最新版本:{str(ol_version)}\n请及时更新！\n\n指令面板：\n/status 查看机器人key状态\n/count 查看机器人统计信息\n/reset 重置会话\n/his 查看历史记录\n/token 查看会话token数\n/help 查看帮助\n/set 人格指令菜单\n/key 动态添加key"
-
-    if qq_msg[:4] == "/key":
-        if len(qq_msg) == 4:
-            msg = "感谢您赞助key，key为官方API使用，请以以下格式赞助:\n/key xxxxx"
-        key = qq_msg[5:]
-        send_qq_msg(message, "收到！正在核验...")
-        if chatgpt.check_key(key):
-            msg = f"*★,°*:.☆(￣▽￣)/$:*.°★* 。\n该Key被验证为有效。感谢{user_name}赞助~"
-            chatgpt.append_key(key, user_name)
-        else:
-            msg = "该Key被验证为无效。也许是输入错误了，或者重试。"
-
-    if qq_msg[:6] == "/unset":
-        now_personality = {}
-        msg = "已清除人格"
-    
-    if qq_msg[:4] == "/set":
-        if len(qq_msg) == 4:
-            np = '无'
-            if "name" in now_personality:
-                np=now_personality["name"]
-            msg = f"【由Github项目QQChannelChatGPT支持】\n\n【人格文本由PlexPt开源项目awesome-chatgpt-prompts-zh提供】\n\n这个是人格设置指令。\n设置人格: \n/set 人格名。例如/set 编剧\n人格列表: /set list\n人格详细信息: /set view 人格名\n自定义人格: /set 人格文本\n清除人格: /unset\n【当前人格】: {np}"
-        elif qq_msg[5:] == "list":
-            per_dict = personalities
-            msg = "人格列表：\n"
-            for key in per_dict.keys():
-                msg += f"  |-{key}\n"
-            msg += '\n\n*输入/set view 人格名查看人格详细信息'
-            msg += '\n\n*不定时更新人格库，请及时更新本项目。'
-        elif qq_msg[5:9] == "view":
-            ps = qq_msg[10:]
-            ps = ps.strip()
-            per_dict = personalities
-            if ps in per_dict:
-                msg = f"人格{ps}的详细信息：\n"
-                msg += f"{per_dict[ps]}\n"
-            else:
-                msg = f"人格{ps}不存在"
-        else:
-            ps = qq_msg[5:]
-            ps = ps.strip()
-            per_dict = personalities
-            if ps in per_dict:
-                now_personality = {
-                    'name': ps,
-                    'prompt': per_dict[ps]
-                }
-                session_dict[session_id] = []
-                msg = f"人格{ps}已设置，请耐心等待机器人回复第一条信息。"
-                go = True
-                type = 1
-            else:
-                msg = f"人格{ps}不存在, 请使用/set list查看人格列表"
-    return msg, go, type

dashboard/__init__.py

@@ -0,0 +1,31 @@
+import logging
+from dataclasses import dataclass
+from quart import Quart
+from type.types import Context
+
+logger = logging.getLogger("astrbot")
+class Route():
+    def __init__(self, context: Context, app: Quart):
+        self.context = context
+        self.app = app
+    
+    def register_routes(self):
+        for route, (method, func) in self.routes.items():
+            self.app.add_url_rule(f"/api{route}", view_func=func, methods=[method])
+
+@dataclass
+class Response():
+    status: str = None
+    message: str = None
+    data: dict = None
+
+    def error(self, message: str):
+        self.status = "error"
+        self.message = message
+        return self
+
+    def ok(self, data: dict={}, message: str=None):
+        self.status = "ok"
+        self.data = data
+        self.message = message
+        return self

dashboard/dist/_redirects

@@ -0,0 +1 @@
+/*    /index.html   200

dashboard/dist/assets/BlankLayout-854c8615.js

@@ -0,0 +1 @@
+import{q as e,o as a,c as t,w as o,d as s,x as n,U as r,X as c}from"./index-b50bcc8e.js";const f=e({__name:"BlankLayout",setup(p){return(u,_)=>(a(),t(r,null,{default:o(()=>[s(n(c))]),_:1}))}});export{f as default};

dashboard/dist/assets/ConfigPage-f564cc69.css

@@ -0,0 +1 @@
+.v-tab{text-transform:none!important}

dashboard/dist/assets/ConsolePage-ff373be6.css

@@ -0,0 +1,32 @@
+/**
+ * Copyright (c) 2014 The xterm.js authors. All rights reserved.
+ * Copyright (c) 2012-2013, Christopher Jeffrey (MIT License)
+ * https://github.com/chjj/term.js
+ * @license MIT
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy
+ * of this software and associated documentation files (the "Software"), to deal
+ * in the Software without restriction, including without limitation the rights
+ * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ * copies of the Software, and to permit persons to whom the Software is
+ * furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in
+ * all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+ * THE SOFTWARE.
+ *
+ * Originally forked from (with the author's permission):
+ *   Fabrice Bellard's javascript vt100 for jslinux:
+ *   http://bellard.org/jslinux/
+ *   Copyright (c) 2011 Fabrice Bellard
+ *   The original design remains. The terminal itself
+ *   has been extended to include xterm CSI codes, among
+ *   other features.
+ */.xterm{cursor:text;position:relative;user-select:none;-ms-user-select:none;-webkit-user-select:none}.xterm.focus,.xterm:focus{outline:none}.xterm .xterm-helpers{position:absolute;top:0;z-index:5}.xterm .xterm-helper-textarea{padding:0;border:0;margin:0;position:absolute;opacity:0;left:-9999em;top:0;width:0;height:0;z-index:-5;white-space:nowrap;overflow:hidden;resize:none}.xterm .composition-view{background:#000;color:#fff;display:none;position:absolute;white-space:nowrap;z-index:1}.xterm .composition-view.active{display:block}.xterm .xterm-viewport{background-color:#000;overflow-y:scroll;cursor:default;position:absolute;right:0;left:0;top:0;bottom:0}.xterm .xterm-screen{position:relative}.xterm .xterm-screen canvas{position:absolute;left:0;top:0}.xterm .xterm-scroll-area{visibility:hidden}.xterm-char-measure-element{display:inline-block;visibility:hidden;position:absolute;top:0;left:-9999em;line-height:normal}.xterm.enable-mouse-events{cursor:default}.xterm.xterm-cursor-pointer,.xterm .xterm-cursor-pointer{cursor:pointer}.xterm.column-select.focus{cursor:crosshair}.xterm .xterm-accessibility,.xterm .xterm-message{position:absolute;left:0;top:0;bottom:0;right:0;z-index:10;color:transparent;pointer-events:none}.xterm .live-region{position:absolute;left:-9999px;width:1px;height:1px;overflow:hidden}.xterm-dim{opacity:1!important}.xterm-underline-1{text-decoration:underline}.xterm-underline-2{text-decoration:double underline}.xterm-underline-3{text-decoration:wavy underline}.xterm-underline-4{text-decoration:dotted underline}.xterm-underline-5{text-decoration:dashed underline}.xterm-overline{text-decoration:overline}.xterm-overline.xterm-underline-1{text-decoration:overline underline}.xterm-overline.xterm-underline-2{text-decoration:overline double underline}.xterm-overline.xterm-underline-3{text-decoration:overline wavy underline}.xterm-overline.xterm-underline-4{text-decoration:overline dotted underline}.xterm-overline.xterm-underline-5{text-decoration:overline dashed underline}.xterm-strikethrough{text-decoration:line-through}.xterm-screen .xterm-decoration-container .xterm-decoration{z-index:6;position:absolute}.xterm-screen .xterm-decoration-container .xterm-decoration.xterm-decoration-top-layer{z-index:7}.xterm-decoration-overview-ruler{z-index:8;position:absolute;top:0;right:0;pointer-events:none}.xterm-decoration-top{z-index:2;position:relative}

dashboard/dist/assets/LoginPage-74e85ca7.css

@@ -0,0 +1 @@
+.custom-devider{border-color:#00000014!important}.googleBtn{border-color:#00000014;margin:30px 0 20px}.outlinedInput .v-field{border:1px solid rgba(0,0,0,.08);box-shadow:none}.orbtn{padding:2px 40px;border-color:#00000014;margin:20px 15px}.pwdInput{position:relative}.pwdInput .v-input__append{position:absolute;right:10px;top:50%;transform:translateY(-50%)}.loginForm .v-text-field .v-field--active input{font-weight:500}.loginBox{max-width:475px;margin:0 auto}

dashboard/dist/assets/_plugin-vue_export-helper-c27b6911.js

@@ -0,0 +1 @@
+const s=(t,r)=>{const o=t.__vccOpts||t;for(const[c,e]of r)o[c]=e;return o};export{s as _};

dashboard/dist/favicon.svg

@@ -0,0 +1 @@
+<svg t="1702013028016" class="icon" viewBox="0 0 1024 1024" version="1.1" xmlns="http://www.w3.org/2000/svg" p-id="1541" width="200" height="200"><path d="M0 0m204.8 0l614.4 0q204.8 0 204.8 204.8l0 614.4q0 204.8-204.8 204.8l-614.4 0q-204.8 0-204.8-204.8l0-614.4q0-204.8 204.8-204.8Z" fill="#FFEC9C" p-id="1542"></path><path d="M819.2 0H534.272A756.48 756.48 0 0 0 0 483.584V819.2a204.8 204.8 0 0 0 204.8 204.8h614.4a204.8 204.8 0 0 0 204.8-204.8V204.8a204.8 204.8 0 0 0-204.8-204.8z" fill="#FFE98A" p-id="1543"></path><path d="M819.2 0h-3.84a755.2 755.2 0 0 0-539.392 1024H819.2a204.8 204.8 0 0 0 204.8-204.8V204.8a204.8 204.8 0 0 0-204.8-204.8z" fill="#FFE471" p-id="1544"></path><path d="M497.152 721.152A752.384 752.384 0 0 0 560.384 1024H819.2a204.8 204.8 0 0 0 204.8-204.8V204.8a204.8 204.8 0 0 0-89.088-168.96 755.2 755.2 0 0 0-437.76 685.312z" fill="#FFE161" p-id="1545"></path><path d="M526.08 140.032l98.304 199.168L844.8 371.2a15.616 15.616 0 0 1 8.704 25.6l-159.744 156.16 37.632 219.136a15.616 15.616 0 0 1-22.528 16.384l-196.608-102.4-196.608 102.4a15.616 15.616 0 0 1-22.528-16.384l37.12-219.136-159.232-155.136a15.616 15.616 0 0 1 8.704-25.6l219.904-32 98.304-199.168a15.616 15.616 0 0 1 28.16-1.024z" fill="#FFF5CC" p-id="1546"></path><path d="M665.6 409.6a444.16 444.16 0 0 0 25.6-61.44l-65.536-9.472-99.584-198.656a15.616 15.616 0 0 0-27.904 0l-98.304 199.168L179.2 371.2a15.616 15.616 0 0 0-8.704 25.6l159.744 156.16-15.104 87.04A407.808 407.808 0 0 0 665.6 409.6z" fill="#FFFFFF" p-id="1547"></path></svg>

dashboard/dist/index.html

@@ -0,0 +1,21 @@
+<!doctype html>
+<html lang="en">
+  <head>
+    <meta charset="UTF-8" />
+    <link rel="icon" href="/favicon.svg" />
+    <meta name="viewport" content="width=device-width, initial-scale=1.0" />
+    <meta name="keywords" content="AstrBot Soulter" />
+    <meta name="description" content="AstrBot Dashboard" />
+    <link
+      rel="stylesheet"
+      href="https://fonts.googleapis.com/css2?family=Inter:wght@400;500;600;700&family=Poppins:wght@400;500;600;700&family=Roboto:wght@400;500;700&display=swap"
+    />
+    <title>AstrBot - 仪表盘</title>
+    <script type="module" crossorigin src="/assets/index-b50bcc8e.js"></script>
+    <link rel="stylesheet" href="/assets/index-86dd25ba.css">
+  </head>
+  <body>
+    <div id="app"></div>
+    
+  </body>
+</html>

dashboard/routes/__init__.py

@@ -0,0 +1,17 @@
+from .auth import AuthRoute
+from .plugin import PluginRoute
+from .config import ConfigRoute
+from .update import UpdateRoute
+from .stat import StatRoute
+from .log import LogRoute
+from .static_file import StaticFileRoute
+
+__all__ = [
+    "AuthRoute",
+    "PluginRoute",
+    "ConfigRoute",
+    "UpdateRoute",
+    "StatRoute",
+    "LogRoute",
+    "StaticFileRoute"
+]

dashboard/routes/auth.py

@@ -0,0 +1,33 @@
+from .. import Route, Response
+from quart import Quart, request
+from type.types import Context
+
+class AuthRoute(Route):
+    def __init__(self, context: Context, app: Quart) -> None:
+        super().__init__(context, app)
+        self.routes = {
+            '/auth/login': ('POST', self.login),
+            '/auth/password/reset': ('POST', self.reset_password),
+        }
+        self.register_routes()
+    
+    async def login(self):
+        username = self.context.config_helper.dashboard.username
+        password = self.context.config_helper.dashboard.password
+        post_data = await request.json
+        if post_data["username"] == username and post_data["password"] == password:
+            return Response().ok({
+                "token": "astrbot-test-token",
+                "username": username
+            }).__dict__
+        else:
+            return Response().error("用户名或密码错误").__dict__
+        
+    async def reset_password(self):
+        password = self.context.config_helper.dashboard.password
+        post_data = await request.json
+        if post_data["password"] == password:
+            self.context.config_helper.dashboard.password = post_data['new_password']
+            return Response().ok(None).__dict__
+        else:
+            return Response().error("原密码错误").__dict__

dashboard/routes/config.py

@@ -0,0 +1,80 @@
+import os, json, threading
+from .. import Route, Response
+from ..utils.config import *
+from quart import Quart, request
+from type.types import Context
+from type.config import CONFIG_METADATA_2
+from util.updator.astrbot_updator import AstrBotUpdator
+
+
+class ConfigRoute(Route):
+    def __init__(self, context: Context, app: Quart, astrbot_updator: AstrBotUpdator) -> None:
+        super().__init__(context, app)
+        self.config_key_dont_show = ['dashboard', 'config_version']
+        self.astrbot_updator = astrbot_updator
+        self.routes = {
+            '/config/get': ('GET', self.get_configs),
+            '/config/astrbot/update': ('POST', self.post_astrbot_configs),
+            '/config/plugin/update': ('POST', self.post_extension_configs),
+        }
+        self.register_routes()
+
+    async def get_configs(self):
+        # namespace 为空时返回 AstrBot 配置
+        # 否则返回指定 namespace 的插件配置
+        namespace = "" if "namespace" not in request.args else request.args["namespace"]
+        if not namespace:
+            return Response().ok(await self._get_astrbot_config()).__dict__
+        return Response().ok(await self._get_extension_config(namespace)).__dict__
+
+    async def post_astrbot_configs(self):
+        post_configs = await request.json
+        try:
+            await self._save_astrbot_configs(post_configs)
+            return Response().ok(None, "保存成功~ 机器人将在 3 秒内重启以应用新的配置。").__dict__
+        except Exception as e:
+            return Response().error(str(e)).__dict__
+    
+    async def post_extension_configs(self):
+        post_configs = await request.json
+        try:
+            await self._save_extension_configs(post_configs)
+            return Response().ok(None, "保存成功~ 机器人将在 3 秒内重启以应用新的配置。").__dict__
+        except Exception as e:
+            return Response().error(str(e)).__dict__
+            
+    async def _get_astrbot_config(self):
+        config = self.context.config_helper.to_dict()
+        for key in self.config_key_dont_show:
+            if key in config:
+                del config[key]
+        return {
+            "metadata": CONFIG_METADATA_2,
+            "config": config,
+        }
+            
+    async def _get_extension_config(self, namespace: str):
+        path = f"data/config/{namespace}.json"
+        if not os.path.exists(path):
+            return []
+        with open(path, "r", encoding="utf-8-sig") as f:
+            return [{
+                "config_type": "group",
+                "name": namespace + " 插件配置",
+                "description": "",
+                "body": list(json.load(f).values())
+            },]
+
+    async def _save_astrbot_configs(self, post_configs: dict):
+        try:
+            save_astrbot_config(post_configs, self.context)
+            threading.Thread(target=self.astrbot_updator._reboot, args=(3, self.context), daemon=True).start()
+        except Exception as e:
+            raise e
+    
+    async def _save_extension_configs(self, post_configs: dict):
+        try:
+            save_extension_config(post_configs)
+            threading.Thread(target=self.astrbot_updator._reboot, args=(3, self.context), daemon=True).start()
+        except Exception as e:
+            raise e

dashboard/routes/log.py

@@ -0,0 +1,50 @@
+import asyncio
+from quart import websocket
+from quart import Quart
+from type.types import Context
+from .. import logger
+
+class Broker:
+    def __init__(self) -> None:
+        self.connections = set()
+        
+    async def send(self, message: str):
+        for connection in self.connections:
+            try:
+                await connection.send(message)
+            except Exception as e:
+                logger.warning(f"发送日志失败: {e.__str__()}")
+
+            
+class LogRoute:
+    def __init__(self, context: Context, app: Quart) -> None:
+        self.app = app
+        self.context = context
+        self.broker = Broker()
+        self.app.add_url_rule('/api/live-log', view_func=self.log, methods=['GET'], websocket=True)
+        
+    async def _receive_log_task(self):
+        while True:
+            message = await self.context._log_queue.get()
+            await self.broker.send(message)
+            
+    async def _get_log_history(self):
+        try:
+            dq = self.context._log_queue.get_cache()
+            ret = ""
+            for log in dq:
+                log = log.replace("\n", "\n\r")
+                ret += log + "\n\r"
+            return ret
+        except Exception as e:
+            logger.warning(f"读取日志历史失败: {e.__str__()}")
+            return ""
+            
+    async def log(self):
+        try:
+            await websocket.send(await self._get_log_history())
+            self.broker.connections.add(websocket)
+            while True:
+                await asyncio.sleep(1)
+        except asyncio.CancelledError:
+            self.broker.connections.remove(websocket)

dashboard/routes/plugin.py

@@ -0,0 +1,86 @@
+import threading, traceback, uuid
+from .. import Route, Response, logger
+from quart import Quart, request
+from type.types import Context
+from model.plugin.manager import PluginManager
+from util.updator.astrbot_updator import AstrBotUpdator
+
+class PluginRoute(Route):
+    def __init__(self, context: Context, app: Quart, astrbot_updator: AstrBotUpdator, plugin_manager: PluginManager) -> None:
+        super().__init__(context, app)
+        self.routes = {
+            '/plugin/get': ('GET', self.get_plugins),
+            '/plugin/install': ('POST', self.install_plugin),
+            '/plugin/install-upload': ('POST', self.install_plugin_upload),
+            '/plugin/update': ('POST', self.update_plugin),
+            '/plugin/uninstall': ('POST', self.uninstall_plugin),
+        }
+        self.astrbot_updator = astrbot_updator
+        self.plugin_manager = plugin_manager
+        self.register_routes()
+    
+    async def get_plugins(self):
+        _plugin_resp = []
+        for plugin in self.context.cached_plugins:
+            _p = plugin.metadata
+            _t = {
+                "name": _p.plugin_name,
+                "repo": '' if _p.repo is None else _p.repo,
+                "author": _p.author,
+                "desc": _p.desc,
+                "version": _p.version
+            }
+            _plugin_resp.append(_t)
+        return Response().ok(_plugin_resp).__dict__
+        
+    async def install_plugin(self):
+        post_data = await request.json
+        repo_url = post_data["url"]
+        try:
+            logger.info(f"正在安装插件 {repo_url}")
+            await self.plugin_manager.install_plugin(repo_url)
+            threading.Thread(target=self.astrbot_updator._reboot, args=(2, self.context)).start()
+            logger.info(f"安装插件 {repo_url} 成功, 2秒后重启")
+            return Response().ok(None, "安装成功，程序将在 2 秒内重启。").__dict__
+        except Exception as e:
+            logger.error(traceback.format_exc())
+            return Response().error(str(e)).__dict__
+        
+    async def install_plugin_upload(self):
+        try:
+            file = request.files['file']
+            print(file.filename)
+            logger.info(f"正在安装用户上传的插件 {file.filename}")
+            file_path = f"data/temp/{uuid.uuid4()}.zip"
+            file.save(file_path)
+            self.plugin_manager.install_plugin_from_file(file_path)
+            logger.info(f"安装插件 {file.filename} 成功")
+            return Response().ok(None, "安装成功！！").__dict__
+        except Exception as e:
+            logger.error(traceback.format_exc())
+            return Response().error(str(e)).__dict__
+        
+    async def uninstall_plugin(self):
+        post_data = await request.json
+        plugin_name = post_data["name"]
+        try:
+            logger.info(f"正在卸载插件 {plugin_name}")
+            self.plugin_manager.uninstall_plugin(plugin_name)
+            logger.info(f"卸载插件 {plugin_name} 成功")
+            return Response().ok(None, "卸载成功").__dict__
+        except Exception as e:
+            logger.error(traceback.format_exc())
+            return Response().error(str(e)).__dict__
+        
+    async def update_plugin(self):
+        post_data = await request.json
+        plugin_name = post_data["name"]
+        try:
+            logger.info(f"正在更新插件 {plugin_name}")
+            await self.plugin_manager.update_plugin(plugin_name)
+            threading.Thread(target=self.astrbot_updator._reboot, args=(2, self.context)).start()
+            logger.info(f"更新插件 {plugin_name} 成功，2秒后重启")
+            return Response().ok(None, "更新成功，程序将在 2 秒内重启。").__dict__
+        except Exception as e:
+            logger.error(f"/api/extensions/update: {traceback.format_exc()}")
+            return Response().error(str(e)).__dict__

dashboard/routes/stat.py

@@ -0,0 +1,62 @@
+import traceback, psutil, time, datetime
+from .. import Route, Response, logger
+from quart import Quart, request
+from type.types import Context
+from astrbot.db import BaseDatabase
+from type.config import VERSION
+
+class StatRoute(Route):
+    def __init__(self, context: Context, app: Quart, db_helper: BaseDatabase) -> None:
+        super().__init__(context, app)
+        self.routes = {
+            '/stat/get': ('GET', self.get_stat),
+            '/stat/version': ('GET', self.get_version),
+        }
+        self.db_helper = db_helper
+        self.register_routes()
+        
+    def format_sec(self, sec: int):
+        m, s = divmod(sec, 60)
+        h, m = divmod(m, 60)
+        return f"{h}小时{m}分{s}秒"
+        
+    async def get_version(self):
+        return Response().ok({
+            "version": VERSION
+        }).__dict__
+    
+    async def get_stat(self):
+        offset_sec = request.args.get('offset_sec', 86400)
+        offset_sec = int(offset_sec)
+        try:
+            stat = self.db_helper.get_base_stats(offset_sec)
+            now = int(time.time())
+            start_time = now - offset_sec
+            message_time_based_stats = []
+            
+            idx = 0
+            for bucket_end in range(start_time, now, 1800):
+                cnt = 0
+                while idx < len(stat.platform) and stat.platform[idx].timestamp < bucket_end:
+                    cnt += stat.platform[idx].count
+                    idx += 1
+                message_time_based_stats.append([bucket_end, cnt])
+            
+            stat_dict = stat.__dict__
+            
+            stat_dict.update({
+                "platform": self.db_helper.get_grouped_base_stats(offset_sec).platform,
+                "message_count": self.db_helper.get_total_message_count() or 0,
+                "platform_count": len(self.context.platforms),
+                "message_time_series": message_time_based_stats,
+                "running": self.format_sec(int(time.time() - self.context._start_running)),
+                "memory": {
+                    "process": psutil.Process().memory_info().rss >> 20,
+                    "system": psutil.virtual_memory().total >> 20
+                }
+            })
+            
+            return Response().ok(stat_dict).__dict__
+        except Exception as e:
+            logger.error(traceback.format_exc())
+            return Response().error(e.__str__()).__dict__

dashboard/routes/static_file.py

@@ -0,0 +1,14 @@
+from .. import Route
+from quart import Quart
+from type.types import Context
+
+class StaticFileRoute(Route):
+    def __init__(self, context: Context, app: Quart) -> None:
+        super().__init__(context, app)
+        
+        index_ = ['/', '/auth/login', '/config', '/logs', '/extension', '/dashboard/default']
+        for i in index_:
+            self.app.add_url_rule(i, view_func=self.index)
+    
+    async def index(self):
+        return await self.app.send_static_file('index.html')

dashboard/routes/update.py

@@ -0,0 +1,45 @@
+import threading, traceback
+from .. import Route, Response, logger
+from quart import Quart, request
+from type.types import Context
+from util.updator.astrbot_updator import AstrBotUpdator
+
+class UpdateRoute(Route):
+    def __init__(self, context: Context, app: Quart, astrbot_updator: AstrBotUpdator) -> None:
+        super().__init__(context, app)
+        self.routes = {
+            '/update/check': ('GET', self.check_update),
+            '/update/do': ('POST', self.update_project),
+        }
+        self.astrbot_updator = astrbot_updator
+        self.register_routes()
+    
+    async def check_update(self):
+        try:
+            ret = await self.astrbot_updator.check_update(None, None)
+            return Response(
+                status="success",
+                message=str(ret) if ret is not None else "已经是最新版本了。",
+                data={
+                    "has_new_version": ret is not None
+                }
+            ).__dict__
+        except Exception as e:
+            logger.error(traceback.format_exc())
+            return Response().error(e.__str__()).__dict__
+    
+    async def update_project(self):
+        data = await request.json
+        version = data.get('version', '')
+        if version == "" or version == "latest":
+            latest = True
+            version = ''
+        else:
+            latest = False
+        try:
+            await self.astrbot_updator.update(latest=latest, version=version)
+            threading.Thread(target=self.astrbot_updator._reboot, args=(2, self.context)).start()
+            return Response().ok(None, "更新成功，程序将在 2 秒内重启。").__dict__
+        except Exception as e:
+            logger.error(f"/api/update_project: {traceback.format_exc()}")
+            return Response().error(e.__str__()).__dict__

dashboard/server.py

@@ -0,0 +1,39 @@
+import logging
+import asyncio
+from quart import Quart
+from quart.logging import default_handler
+from type.types import Context
+from .routes import *
+from . import logger
+from astrbot.db import BaseDatabase
+from model.plugin.manager import PluginManager
+from util.updator.astrbot_updator import AstrBotUpdator
+from util.io import get_local_ip_addresses
+
+class AstrBotDashboard():
+    def __init__(self, context: Context, 
+                 plugin_manager: PluginManager, 
+                 astrbot_updator: AstrBotUpdator, 
+                 db_helper: BaseDatabase) -> None:
+        self.context = context
+        self.app = Quart("dashboard", static_folder="dist", static_url_path="/")
+        self.app.json.sort_keys = False
+        
+        logging.getLogger(self.app.name).removeHandler(default_handler)
+        
+        self.ar = AuthRoute(context, self.app)
+        self.ur = UpdateRoute(context, self.app, astrbot_updator)
+        self.sr = StatRoute(context, self.app, db_helper)
+        self.pr = PluginRoute(context, self.app, astrbot_updator, plugin_manager)
+        self.cr = ConfigRoute(context, self.app, astrbot_updator)
+        self.lr = LogRoute(context, self.app)
+        self.sfr = StaticFileRoute(context, self.app)
+    
+    async def shutdown_trigger_placeholder(self):
+        while self.context.running:
+            await asyncio.sleep(1)
+        
+    def run(self):
+        ip_addr = get_local_ip_addresses()
+        logger.info(f"仪表盘已启动，可访问 http://{ip_addr}:6185 登录。")
+        return self.app.run_task(host="0.0.0.0", port=6185, shutdown_trigger=self.shutdown_trigger_placeholder)

dashboard/utils/config.py

@@ -0,0 +1,98 @@
+from dataclasses import asdict
+from util.plugin_dev.api.v1.config import update_config
+from type.config import CONFIG_METADATA_2
+from type.types import Context
+
+def try_cast(value: str, type_: str):
+    if type_ == "int" and value.isdigit():
+        return int(value)
+    elif type_ == "float" and isinstance(value, str) \
+        and value.replace(".", "", 1).isdigit():
+        return float(value)
+    elif type_ == "float" and isinstance(value, int):
+        return float(value)
+
+def get_default_val_by_type(type_: str):
+    if type_ == "int":
+        return 0
+    elif type_ == "float":
+        return 0.0
+    elif type_ == "bool":
+        return False
+    elif type_ == "string":
+        return ""
+    elif type_ == "list":
+        return []
+    elif type_ == "object":
+        return {}
+        
+
+def validate_config(data, context: Context):
+    errors = []
+    def validate(data, metadata=CONFIG_METADATA_2, path=""):
+        for key, meta in metadata.items():
+            if key not in data:
+                continue
+            value = data[key]
+            # null 转换
+            if value is None:
+                data[key] = get_default_val_by_type(meta["type"])
+                continue
+            # 递归验证
+            if meta["type"] == "list" and isinstance(value, list):
+                for item in value:
+                    validate(item, meta["items"], path=f"{path}{key}.")
+            elif meta["type"] == "object" and isinstance(value, dict):
+                validate(value, meta["items"], path=f"{path}{key}.")
+
+            if meta["type"] == "int" and not isinstance(value, int):
+                casted = try_cast(value, "int")
+                if casted is None:
+                    errors.append(f"错误的类型 {path}{key}: 期望是 int, 得到了 {type(value).__name__}")
+                data[key] = casted
+            elif meta["type"] == "float" and not isinstance(value, float):
+                casted = try_cast(value, "float")
+                if casted is None:
+                    errors.append(f"错误的类型 {path}{key}: 期望是 float, 得到了 {type(value).__name__}")
+                data[key] = casted
+            elif meta["type"] == "bool" and not isinstance(value, bool):
+                errors.append(f"错误的类型 {path}{key}: 期望是 bool, 得到了 {type(value).__name__}")
+            elif meta["type"] == "string" and not isinstance(value, str):
+                errors.append(f"错误的类型 {path}{key}: 期望是 string, 得到了 {type(value).__name__}")
+            elif meta["type"] == "list" and not isinstance(value, list):
+                errors.append(f"错误的类型 {path}{key}: 期望是 list, 得到了 {type(value).__name__}")
+            elif meta["type"] == "object" and not isinstance(value, dict):
+                errors.append(f"错误的类型 {path}{key}: 期望是 dict, 得到了 {type(value).__name__}")
+                validate(value, meta["items"], path=f"{path}{key}.")
+    validate(data)
+    
+    # hardcode warning
+    data['config_version'] = context.config_helper.config_version
+    data['dashboard'] = asdict(context.config_helper.dashboard)
+    
+    return errors
+
+def save_astrbot_config(post_config: dict, context: Context):
+    '''验证并保存配置'''
+    errors = validate_config(post_config, context)
+    if errors:
+        raise ValueError(f"格式校验未通过: {errors}")
+    context.config_helper.flush_config(post_config)
+    
+def save_extension_config(post_config: dict):
+    if 'namespace' not in post_config:
+        raise ValueError("Missing key: namespace")
+    if 'config' not in post_config:
+        raise ValueError("Missing key: config")
+
+    namespace = post_config['namespace']
+    config: list = post_config['config'][0]['body']
+    for item in config:
+        key = item['path']
+        value = item['value']
+        typ = item['val_type']
+        if typ == 'int':
+            if not value.isdigit():
+                raise ValueError(f"错误的类型 {namespace}.{key}: 期望是 int, 得到了 {type(value).__name__}")
+            value = int(value)
+        update_config(namespace, key, value)

main.py

@@ -1,175 +1,57 @@
-import threading
-import time
+
+import os
 import asyncio
-import os, sys
-import signal
-import requests,json
+import sys
+import warnings
+import traceback
+import mimetypes
+from astrbot.bootstrap import AstrBotBootstrap
+from util.log import LogManager
 
-
-abs_path = os.path.dirname(os.path.realpath(sys.argv[0])) + '/'
-
-def main(loop, event):
-    import cores.qqbot.core as qqBot
-    import yaml
-
-    ymlfile =  open(abs_path+"configs/config.yaml", 'r', encoding='utf-8')
-    cfg = yaml.safe_load(ymlfile)
-    provider = privider_chooser(cfg)
-    print('[System] 当前语言模型提供商: ' + provider)
-    # 执行Bot
-    qqBot.initBot(cfg, provider)
-
-# 语言模型提供商选择器
-# 目前有：OpenAI官方API、逆向库
-def privider_chooser(cfg):
-    if 'rev_ChatGPT' in cfg and cfg['rev_ChatGPT']['enable']:
-        return 'rev_chatgpt'
-    else:
-        return 'openai_official'
-
-# 仅支持linux
-def hot_update(ver):
-    target = 'target.tar'
-    time.sleep(5)
-    while(True):
-        if os.path.exists('version.txt'):
-            version_file = open('version.txt', 'r', encoding='utf-8')
-            vs = version_file.read()
-            version = float(vs)
-        else:
-            version = 0
-        if not os.path.exists(target):
-            version = 0
-        try:
-            res = requests.get("https://soulter.top/channelbot/update.json")
-            res_obj = json.loads(res.text)
-            ol_version = float(res_obj['version'])
-            if ol_version > version:
-                print('发现新版本: ' + str(ol_version))
-                res = requests.get(res_obj['linux-url'], stream=True)
-                filesize = res.headers["Content-Length"]
-                print('文件大小: ' + str(int(filesize) / 1024 / 1024) + 'MB')
-                print('正在更新文件...')
-                chunk_size = 1024
-                times = int(filesize) // chunk_size
-                show = 1 / times
-                show2 = 1 / times
-                start = 1
-                with open(target, "wb") as pyFile:
-                    for chunk in res.iter_content(chunk_size=chunk_size):
-                        if chunk:
-                            pyFile.write(chunk)
-                        if start <= times:
-                            print(f"\r下载进度: {show:.2%}",end="",flush=True)
-                            start += 1
-                            show += show2
-                        else:
-                            sys.stdout.write(f"下载进度: 100%\n")
-                print('更新完成')
-                print('解压覆盖')
-                os.system(f"tar -zxvf {target}")
-                version = ol_version
-                version_file = open('version.txt', 'w+', encoding='utf-8')
-                version_file.write(str(res_obj['version']))
-                version_file.flush()
-                version_file.close()
-
-                try:
-                    update_version(version)
-                except BaseException as e:
-                    print(e)
-                
-                print('自启动')
-                py = sys.executable
-                os.execl(py, py, *sys.argv)
-            time.sleep(60*20)
-        except BaseException as e:
-            print(e)
-            print("upd出现异常, 请联系QQ905617992")
-            time.sleep(60*20)
-
-def update_version(ver):
-    if not os.path.exists('update_record'):
-        object_id = ''
-    else:
-        object_id = open("update_record", 'r', encoding='utf-8').read()
-    addr = 'unknown'
+warnings.filterwarnings("ignore")
+logo_tmpl = r"""
+     ___           _______.___________..______      .______     ______   .___________.
+    /   \         /       |           ||   _  \     |   _  \   /  __  \  |           |
+   /  ^  \       |   (----`---|  |----`|  |_)  |    |  |_)  | |  |  |  | `---|  |----`
+  /  /_\  \       \   \       |  |     |      /     |   _  <  |  |  |  |     |  |     
+ /  _____  \  .----)   |      |  |     |  |\  \----.|  |_)  | |  `--'  |     |  |     
+/__/     \__\ |_______/       |__|     | _| `._____||______/   \______/      |__|     
+                                                                                    
+"""
+    
+def main():
+    global logger
     try:
-        addr = requests.get('http://myip.ipip.net', timeout=5).text
-    except BaseException:
-        pass
-    try:
-        ts = str(time.time())
-        # md = hashlib.md5((ts+'QAZ1rQLY1ZufHrZlpuUiNff7').encode())
-        headers = {
-            'X-LC-Id': 'UqfXTWW15nB7iMT0OHvYrDFb-gzGzoHsz',
-            'X-LC-Key': 'QAZ1rQLY1ZufHrZlpuUiNff7',
-            'Content-Type': 'application/json'
-        }
-        d = {"data": {'version':'win-hot-update'+str(ver), 'addr': addr}}
-        d = json.dumps(d).encode("utf-8")
-        res = requests.put(f'https://uqfxtww1.lc-cn-n1-shared.com/1.1/classes/version_record/{object_id}', headers = headers, data = d)
-        if json.loads(res.text)['code'] == 1:
-            res = requests.post(f'https://uqfxtww1.lc-cn-n1-shared.com/1.1/classes/version_record', headers = headers, data = d)
-            object_id = json.loads(res.text)['objectId']
-            object_id_file = open("update_record", 'w+', encoding='utf-8')
-            object_id_file.write(str(object_id))
-            object_id_file.flush()
-            object_id_file.close()
+        import botpy, logging
+        # delete qqbotpy's logger
+        for handler in logging.root.handlers[:]:
+            logging.root.removeHandler(handler)
+        
+        logger.info(logo_tmpl)
+
+        bootstrap = AstrBotBootstrap() 
+        asyncio.run(bootstrap.run())
+    except KeyboardInterrupt:
+        logger.info("AstrBot 已退出。")
+
     except BaseException as e:
-        print(e)
+        logger.error(traceback.format_exc())
 
 def check_env():
-    if not (sys.version_info.major == 3 and sys.version_info.minor >= 8):
-        print("请使用Python3.8运行本项目")
-        input("按任意键退出...")
+    if not (sys.version_info.major == 3 and sys.version_info.minor >= 9):
+        logger.error("请使用 Python3.9+ 运行本项目。")
         exit()
-    try:
-        import openai
-        import botpy
-        import yaml
-    except Exception as e:
-        # print(e)
-        try:
-            print("安装依赖库中...")
-            os.system("pip3 install openai")
-            os.system("pip3 install qq-botpy")
-            os.system("pip3 install pyyaml")
-            print("安装依赖库完毕...")
-        except BaseException:
-            print("\n安装第三方库异常.请自行安装或者联系QQ905617992.")
         
-    # 检查key
-    with open(abs_path+"configs/config.yaml", 'r', encoding='utf-8') as ymlfile:
-        import yaml
-        cfg = yaml.safe_load(ymlfile)
-        if cfg['openai']['key'] == '' or cfg['openai']['key'] == None:
-            print("请先在configs/config.yaml下添加一个可用的OpenAI Key。详情请前往https://beta.openai.com/account/api-keys")
-        if cfg['qqbot']['appid'] == '' or cfg['qqbot']['token'] == '' or cfg['qqbot']['appid'] == None or cfg['qqbot']['token'] == None: 
-            print("请先在configs/config.yaml下完善appid和token令牌(在https://q.qq.com/上注册一个QQ机器人即可获得)")
-
-def get_platform():
-    import platform
-    sys_platform = platform.platform().lower()
-    if "windows" in sys_platform:
-        return "win"
-    elif "macos" in sys_platform:
-        return "mac"
-    elif "linux" in sys_platform:
-        return "linux"
-    else:
-        print("other")
+    os.makedirs("data/config", exist_ok=True)
+    os.makedirs("data/plugins", exist_ok=True)
+    os.makedirs("data/temp", exist_ok=True)
 
+    # workaround for issue #181
+    mimetypes.add_type("text/javascript", ".js") 
+    mimetypes.add_type("text/javascript", ".mjs")
+    mimetypes.add_type("application/json", ".json")
+    
 if __name__ == "__main__":
-    global pid
-    pid = os.getpid()
-    global ma_type
-    print("程序PID:"+str(pid))
     check_env()
-    bot_event = threading.Event()
-    loop = asyncio.get_event_loop()
-    ma_type = get_platform()
-    if ma_type == 'linux':
-        threading.Thread(target=hot_update).start()
-        
-    main(loop, bot_event)
+    logger = LogManager.GetLogger(log_name='astrbot')
+    main()

model/command/internal_handler.py

@@ -0,0 +1,276 @@
+import aiohttp, os
+
+from model.command.manager import CommandManager
+from model.plugin.manager import PluginManager
+from type.message_event import AstrMessageEvent
+from type.command import CommandResult
+from type.types import Context
+from type.config import VERSION
+from util.log import LogManager
+from logging import Logger
+from util.agent.web_searcher import search_from_bing, fetch_website_content
+
+logger: Logger = LogManager.GetLogger(log_name='astrbot')
+
+
+class InternalCommandHandler:
+    def __init__(self, manager: CommandManager, plugin_manager: PluginManager) -> None:
+        self.manager = manager
+        self.plugin_manager = plugin_manager
+        
+        self.manager.register("help", "查看帮助", 10, self.help)
+        self.manager.register("wake", "唤醒前缀", 10, self.set_nick)
+        self.manager.register("update", "更新管理", 10, self.update)
+        self.manager.register("plugin", "插件管理", 10, self.plugin)
+        self.manager.register("reboot", "重启 AstrBot", 10, self.reboot)
+        self.manager.register("websearch", "网页搜索", 10, self.web_search)
+        self.manager.register("t2i", "文转图", 10, self.t2i_toggle)
+        self.manager.register("myid", "用户ID", 10, self.myid)
+        self.manager.register("provider", "LLM 接入源", 10, self.provider)
+
+    def _check_auth(self, message: AstrMessageEvent, context: Context):
+        if os.environ.get("TEST_MODE", "off") == "on":
+            return
+        if message.role != "admin":
+            user_id = message.message_obj.sender.user_id
+            raise Exception(f"用户(ID: {user_id}) 没有足够的权限使用该指令。")
+        
+    def provider(self, message: AstrMessageEvent, context: Context):
+        if len(context.llms) == 0:
+            return CommandResult().message("当前没有加载任何 LLM 接入源。")
+
+        tokens = self.manager.command_parser.parse(message.message_str)
+        
+        if tokens.len == 1:
+            ret = "## 当前载入的 LLM 接入源\n"
+            for idx, llm in enumerate(context.llms):
+                ret += f"{idx}. {llm.llm_name}"
+                if llm.origin:
+                    ret += f" (来源: {llm.origin})"
+                if context.message_handler.provider == llm.llm_instance:
+                    ret += " (当前使用)"
+                ret += "\n"
+            
+            ret += "\n使用 provider <序号> 切换 LLM 接入源。" 
+            return CommandResult().message(ret)
+        else:
+            try:
+                idx = int(tokens.get(1))
+                if idx >= len(context.llms):
+                    return CommandResult().message("provider: 无效的序号。")
+                context.message_handler.set_provider(context.llms[idx].llm_instance)
+                return CommandResult().message(f"已经成功切换到 LLM 接入源 {context.llms[idx].llm_name}。")
+            except BaseException as e:
+                return CommandResult().message("provider: 参数错误。")
+
+    def set_nick(self, message: AstrMessageEvent, context: Context):
+        self._check_auth(message, context)
+        message_str = message.message_str
+        l = message_str.split(" ")
+        if len(l) == 1:
+            return CommandResult().message(f"设置机器人唤醒词。以唤醒词开头的消息会唤醒机器人处理，起到 @ 的效果。\n示例：wake 昵称。当前唤醒词是：{context.config_helper.wake_prefix[0]}")
+        nick = l[1].strip()
+        if not nick:
+            return CommandResult().message("wake: 请指定唤醒词。")
+        context.config_helper.wake_prefix = [nick]
+        context.config_helper.save_config()
+        return CommandResult(
+            hit=True,
+            success=True,
+            message_chain=f"已经成功将唤醒前缀设定为 {nick}。",
+        )
+        
+    async def update(self, message: AstrMessageEvent, context: Context):
+        self._check_auth(message, context)
+        tokens = self.manager.command_parser.parse(message.message_str)
+        update_info = await context.updator.check_update(None, None)
+        if tokens.len == 1:
+            ret = ""
+            if not update_info:
+                ret = f"当前已经是最新版本 v{VERSION}。"
+            else:
+                ret = f"发现新版本 {update_info.version}，更新内容如下:\n---\n{update_info.body}\n---\n- 使用 /update latest 更新到最新版本。\n- 使用 /update vX.X.X 更新到指定版本。"
+            return CommandResult().message(ret)
+        else:
+            if tokens.get(1) == "latest":
+                try:
+                    await context.updator.update()
+                    return CommandResult().message(f"已经成功更新到最新版本 v{update_info.version}。要应用更新，请重启 AstrBot。输入 /reboot 即可重启")
+                except BaseException as e:
+                    return CommandResult().message(f"更新失败。原因：{str(e)}")
+            elif tokens.get(1).startswith("v"):
+                try:
+                    await context.updator.update(version=tokens.get(1))
+                    return CommandResult().message(f"已经成功更新到版本 v{tokens.get(1)}。要应用更新，请重启 AstrBot。输入 /reboot 即可重启")
+                except BaseException as e:
+                    return CommandResult().message(f"更新失败。原因：{str(e)}")
+            else:
+                return CommandResult().message("update: 参数错误。")
+            
+    def reboot(self, message: AstrMessageEvent, context: Context):
+        self._check_auth(message, context)
+        context.updator._reboot(3, context)
+        return CommandResult(
+            hit=True,
+            success=True,
+            message_chain="AstrBot 将在 3s 后重启。",
+        )
+    
+    async def plugin(self, message: AstrMessageEvent, context: Context):
+        tokens = self.manager.command_parser.parse(message.message_str)
+        if tokens.len == 1:
+            ret = "# 插件指令面板 \n- 安装插件: `plugin i 插件Github地址`\n- 卸载插件: `plugin d 插件名`\n- 查看插件列表：`plugin l`\n - 更新插件: `plugin u 插件名`\n"
+            return CommandResult().message(ret)
+        
+        if tokens.get(1) == "l":
+            plugin_list_info = ""
+            for plugin in context.cached_plugins:
+                plugin_list_info += f"- `{plugin.metadata.plugin_name}` By {plugin.metadata.author}: {plugin.metadata.desc}\n"
+            if plugin_list_info.strip() == "":
+                return CommandResult().message("plugin v: 没有找到插件。")
+            return CommandResult().message(plugin_list_info)
+
+        self._check_auth(message, context)
+        
+        if tokens.get(1) == "d":
+            if tokens.len == 2:
+                return CommandResult().message("plugin d: 请指定要卸载的插件名。")
+            plugin_name = tokens.get(2)
+            try:
+                self.plugin_manager.uninstall_plugin(plugin_name)
+            except BaseException as e:
+                return CommandResult().message(f"plugin d: 卸载插件失败。原因：{str(e)}")
+            return CommandResult().message(f"plugin d: 已经成功卸载插件 {plugin_name}。")
+        
+        elif tokens.get(1) == "i":
+            if tokens.len == 2:
+                return CommandResult().message("plugin i: 请指定要安装的插件的 Github 地址，或者前往可视化面板安装。")
+            plugin_url = tokens.get(2)
+            try:
+                await self.plugin_manager.install_plugin(plugin_url)
+            except BaseException as e:
+                return CommandResult().message(f"plugin i: 安装插件失败。原因：{str(e)}")
+            return CommandResult().message("plugin i: 已经成功安装插件。")
+        
+        elif tokens.get(1) == "u":
+            if tokens.len == 2:
+                return CommandResult().message("plugin u: 请指定要更新的插件名。")
+            plugin_name = tokens.get(2)
+            try:
+                await context.plugin_updator.update(plugin_name)
+            except BaseException as e:
+                return CommandResult().message(f"plugin u: 更新插件失败。原因：{str(e)}")
+            return CommandResult().message(f"plugin u: 已经成功更新插件 {plugin_name}。")
+        
+        return CommandResult().message("plugin: 参数错误。")
+
+    async def help(self, message: AstrMessageEvent, context: Context):
+        notice = ""
+        try:
+            async with aiohttp.ClientSession() as session:
+                async with session.get("https://soulter.top/channelbot/notice.json") as resp:
+                    notice = (await resp.json())["notice"]
+        except BaseException as e:
+            logger.warning("An error occurred while fetching astrbot notice. Never mind, it's not important.")
+
+        msg = "# 帮助中心\n## 指令\n"
+        for key, value in self.manager.commands_handler.items():
+            if value.plugin_metadata:
+                msg += f"- `{key}` ({value.plugin_metadata.plugin_name}): {value.description}\n"
+            else: msg += f"- `{key}`: {value.description}\n"
+        # plugins
+        if context.cached_plugins:
+            plugin_list_info = ""
+            for plugin in context.cached_plugins:
+                plugin_list_info += f"- `{plugin.metadata.plugin_name}` {plugin.metadata.desc}\n"
+            if plugin_list_info.strip() != "":
+                msg += "\n## 插件\n> 使用plugin v 插件名 查看插件帮助\n"
+                msg += plugin_list_info
+        msg += notice
+
+        return CommandResult().message(msg)
+
+    def web_search(self, message: AstrMessageEvent, context: Context):
+        l = message.message_str.split(' ')
+        if len(l) == 1:
+            return CommandResult(
+                hit=True,
+                success=True,
+                message_chain=f"网页搜索功能当前状态: {context.config_helper.llm_settings.web_search}",
+            )
+        elif l[1] == 'on':
+            context.config_helper.llm_settings.web_search = True
+            context.config_helper.save_config()
+            context.register_llm_tool("web_search", [{
+                "type": "string",
+                "name": "keyword",
+                "description": "搜索关键词"
+            }],
+                "通过搜索引擎搜索。如果问题需要获取近期、实时的消息，在网页上搜索(如天气、新闻或任何需要通过网页获取信息的问题)，则调用此函数；如果没有，不要调用此函数。",
+                search_from_bing
+            )
+            context.register_llm_tool("fetch_website_content", [{
+                "type": "string",
+                "name": "url",
+                "description": "要获取内容的网页链接"
+            }],
+                "获取网页的内容。如果问题带有合法的网页链接并且用户有需求了解网页内容(例如: `帮我总结一下 https://github.com 的内容`), 就调用此函数。如果没有，不要调用此函数。",
+                fetch_website_content
+            )
+            
+            return CommandResult(
+                hit=True,
+                success=True,
+                message_chain="已开启网页搜索",
+            )
+        elif l[1] == 'off':
+            context.config_helper.llm_settings.web_search = False
+            context.config_helper.save_config()
+            context.unregister_llm_tool("web_search")
+            context.unregister_llm_tool("fetch_website_content")
+            
+            return CommandResult(
+                hit=True,
+                success=True,
+                message_chain="已关闭网页搜索",
+            )
+        else:
+            return CommandResult(
+                hit=True,
+                success=False,
+                message_chain="参数错误",
+            )
+    
+    def t2i_toggle(self, message: AstrMessageEvent, context: Context):
+        p = context.config_helper.t2i
+        if p:
+            context.config_helper.t2i = False
+            context.config_helper.save_config()
+            return CommandResult(
+                hit=True,
+                success=True,
+                message_chain="已关闭文本转图片模式。",
+            )
+        context.config_helper.t2i = True
+        context.config_helper.save_config()
+        
+        return CommandResult(
+            hit=True,
+            success=True,
+            message_chain="已开启文本转图片模式。",
+        )
+        
+    def myid(self, message: AstrMessageEvent, context: Context):
+        try:
+            user_id = str(message.message_obj.sender.user_id)
+            return CommandResult(
+                hit=True,
+                success=True,
+                message_chain=f"你在此平台上的ID：{user_id}",
+            )
+        except BaseException as e:
+            return CommandResult(
+                hit=True,
+                success=False,
+                message_chain=f"获取失败，原因: {str(e)}",
+            )

model/command/manager.py

@@ -0,0 +1,145 @@
+import heapq
+import inspect
+import traceback
+from typing import Dict
+from type.types import Context
+from type.plugin import PluginMetadata
+from type.message_event import AstrMessageEvent
+from type.command import CommandResult
+from type.register import RegisteredPlugins
+from model.command.parser import CommandParser
+from model.plugin.command import PluginCommandBridge
+from util.log import LogManager
+from logging import Logger
+from dataclasses import dataclass
+
+logger: Logger = LogManager.GetLogger(log_name='astrbot')
+
+@dataclass
+class CommandMetadata():
+    inner_command: bool
+    plugin_metadata: PluginMetadata
+    handler: callable
+    use_regex: bool = False
+    ignore_prefix: bool = False
+    description: str = ""
+
+class CommandManager():
+    def __init__(self):
+        self.commands = []
+        self.commands_handler: Dict[str, CommandMetadata] = {}
+        self.command_parser = CommandParser()
+        
+    def register(self, 
+                 command: str, 
+                 description: str, 
+                 priority: int,
+                 handler: callable,
+                 use_regex: bool = False,
+                 ignore_prefix: bool = False,
+                 plugin_metadata: PluginMetadata = None,
+                ):
+        '''
+        优先级越高，越先被处理。
+        
+        use_regex: 是否使用正则表达式匹配指令。
+        '''
+        if command in self.commands_handler:
+            raise ValueError(f"Command {command} already exists.")
+        if not handler:
+            raise ValueError(f"Handler of {command} is None.")
+        
+        heapq.heappush(self.commands, (-priority, command))
+        self.commands_handler[command] = CommandMetadata(
+            inner_command=plugin_metadata == None,
+            plugin_metadata=plugin_metadata,
+            handler=handler,
+            use_regex=use_regex,
+            ignore_prefix=ignore_prefix,
+            description=description
+        )
+        if plugin_metadata:
+            logger.info(f"已注册 {plugin_metadata.author}/{plugin_metadata.plugin_name} 的指令 {command}。")
+        else:
+            logger.info(f"已注册指令 {command}。")
+        
+    def register_from_pcb(self, pcb: PluginCommandBridge):
+        for request in pcb.plugin_commands_waitlist:
+            plugin = None
+            for registered_plugin in pcb.cached_plugins:
+                if registered_plugin.metadata.plugin_name == request.plugin_name:
+                    plugin = registered_plugin
+                    break
+            if not plugin:
+                logger.warning(f"插件 {request.plugin_name} 未找到，无法注册指令 {request.command_name}。")
+            else:
+                self.register(command=request.command_name, 
+                              description=request.description, 
+                              priority=request.priority, 
+                              handler=request.handler, 
+                              use_regex=request.use_regex,
+                              ignore_prefix=request.ignore_prefix,
+                              plugin_metadata=plugin.metadata)
+        self.plugin_commands_waitlist = []
+        
+    async def check_command_ignore_prefix(self, message_str: str) -> bool:
+        for _, command in self.commands:
+            command_metadata = self.commands_handler[command]
+            if command_metadata.ignore_prefix:
+                trig = False
+                if self.commands_handler[command].use_regex:
+                    trig = self.command_parser.regex_match(message_str, command)
+                else:
+                    trig = message_str.startswith(command)
+                if trig:
+                    return True
+        return False
+                
+    async def scan_command(self, message_event: AstrMessageEvent, context: Context) -> CommandResult:
+        message_str = message_event.message_str
+        for _, command in self.commands:
+            trig = False
+            if self.commands_handler[command].use_regex:
+                trig = self.command_parser.regex_match(message_str, command)
+            else:
+                trig = message_str.startswith(command)
+            if trig:
+                logger.info(f"触发 {command} 指令。")
+                command_result = await self.execute_handler(command, message_event, context)
+                if not command_result:
+                    continue
+                if command_result.hit:
+                    return command_result
+
+    async def execute_handler(self,
+                        command: str,
+                        message_event: AstrMessageEvent,
+                        context: Context) -> CommandResult:
+        command_metadata = self.commands_handler[command]
+        handler = command_metadata.handler
+        # call handler
+        try:
+            if inspect.iscoroutinefunction(handler):
+                command_result = await handler(message_event, context)
+            else:
+                command_result = handler(message_event, context)
+            
+            # if not isinstance(command_result, CommandResult):
+            #     raise ValueError(f"Command {command} handler should return CommandResult.")
+
+            if not command_result:
+                return
+            
+            context.metrics_uploader.command_stats[command] += 1
+            
+            return command_result
+        except BaseException as e:
+            logger.error(traceback.format_exc())
+            
+            if not command_metadata.inner_command:
+                text = f"执行 {command}/({command_metadata.plugin_metadata.plugin_name} By {command_metadata.plugin_metadata.author}) 指令时发生了异常。{e}"
+                logger.error(text)
+            else:
+                text = f"执行 {command} 指令时发生了异常。{e}"
+                logger.error(text)
+            return CommandResult().message(text)

model/command/openai_official_handler.py

@@ -0,0 +1,186 @@
+from model.command.manager import CommandManager
+from type.message_event import AstrMessageEvent
+from type.command import CommandResult
+from type.types import Context
+from util.log import LogManager
+from logging import Logger
+from nakuru.entities.components import Image
+from model.provider.openai_official import ProviderOpenAIOfficial, MODELS
+from util.personality import personalities
+from util.io import download_image_by_url
+
+logger: Logger = LogManager.GetLogger(log_name='astrbot')
+
+
+class OpenAIOfficialCommandHandler():
+    def __init__(self, manager: CommandManager) -> None:
+        self.manager = manager
+        
+        self.provider = None
+        
+        self.manager.register("reset", "重置会话", 10, self.reset)
+        self.manager.register("his", "查看历史记录", 10, self.his)
+        self.manager.register("status", "查看当前状态", 10, self.status)
+        self.manager.register("switch", "切换账号", 10, self.switch)
+        self.manager.register("unset", "清除个性化人格设置", 10, self.unset)
+        self.manager.register("set", "设置个性化人格", 10, self.set)
+        self.manager.register("draw", "调用 DallE 模型画图", 10, self.draw)
+        self.manager.register("model", "切换模型", 10, self.model)
+        self.manager.register("画", "调用 DallE 模型画图", 10, self.draw)
+        
+    def set_provider(self, provider):
+        self.provider = provider
+    
+    async def reset(self, message: AstrMessageEvent, context: Context):
+        tokens = self.manager.command_parser.parse(message.message_str)
+        if tokens.len == 1:
+            await self.provider.forget(message.session_id, keep_system_prompt=True)
+            return CommandResult().message("重置成功")
+        elif tokens.get(1) == 'p':
+            await self.provider.forget(message.session_id)
+    
+    async def model(self, message: AstrMessageEvent, context: Context):
+        tokens = self.manager.command_parser.parse(message.message_str)
+        if tokens.len == 1:
+            ret = await self._print_models()
+            return CommandResult().message(ret)
+        model = tokens.get(1)
+        if model.isdigit():
+            try:
+                models = await self.provider.get_models()
+            except BaseException as e:
+                logger.error(f"获取模型列表失败: {str(e)}")
+                return CommandResult().message("获取模型列表失败，无法使用编号切换模型。可以尝试直接输入模型名来切换，如 gpt-4o。")
+            models = list(models)
+            if int(model) <= len(models) and int(model) >= 1:
+                model = models[int(model)-1]
+            self.provider.set_model(model.id)
+            return CommandResult().message(f"模型已设置为 {model.id}")
+        else:
+            self.provider.set_model(model)
+            return CommandResult().message(f"模型已设置为 {model} (自定义)")
+
+    async def _print_models(self):
+        try:
+            models = await self.provider.get_models()
+        except BaseException as e:
+            return "获取模型列表失败: " + str(e)
+        i = 1
+        ret = "OpenAI GPT 类可用模型"
+        for model in models:
+            ret += f"\n{i}. {model.id}"
+            i += 1
+        ret += "\nTips: 使用 /model 模型名/编号，即可实时更换模型。如目标模型不存在于上表，请输入模型名。"
+        logger.debug(ret)
+        return ret
+        
+    def his(self, message: AstrMessageEvent, context: Context):
+        tokens = self.manager.command_parser.parse(message.message_str)
+        size_per_page = 3
+        page = 1
+        if tokens.len == 2:
+            try:
+                page = int(tokens.get(1))
+            except BaseException as e:
+                return CommandResult().message("页码格式错误")
+        contexts, total_num = self.provider.dump_contexts_page(message.session_id, size_per_page, page=page)
+        t_pages = total_num // size_per_page + 1
+        return CommandResult().message(f"历史记录如下：\n{contexts}\n第 {page} 页 | 共 {t_pages} 页\n*输入 /his 2 跳转到第 2 页")
+    
+    def status(self, message: AstrMessageEvent, context: Context):
+        keys_data = self.provider.get_keys_data()
+        ret = "OpenAI Key"
+        for k in keys_data:
+            status = "🟢" if keys_data[k] else "🔴"
+            ret += "\n|- " + k[:8] + " " + status
+
+        conf = self.provider.get_configs()
+        ret += "\n当前模型: " + conf['model']
+        if conf['model'] in MODELS:
+            ret += "\n最大上下文窗口: " + str(MODELS[conf['model']]) + " tokens"
+
+        if message.session_id in self.provider.session_memory and len(self.provider.session_memory[message.session_id]):
+            ret += "\n你的会话上下文: " + str(self.provider.session_memory[message.session_id][-1]['usage_tokens']) + " tokens"
+
+        return CommandResult().message(ret)
+    
+    async def switch(self, message: AstrMessageEvent, context: Context):
+        '''
+        切换账号
+        '''
+        tokens = self.manager.command_parser.parse(message.message_str)
+        if tokens.len == 1:
+            _, ret, _ = self.status()
+            curr_ = self.provider.get_curr_key()
+            if curr_ is None:
+                ret += "当前您未选择账号。输入/switch <账号序号>切换账号。"
+            else:
+                ret += f"当前您选择的账号为：{curr_[-8:]}。输入/switch <账号序号>切换账号。"
+            return CommandResult().message(ret)
+        elif tokens.len == 2:
+            try:
+                key_stat = self.provider.get_keys_data()
+                index = int(tokens.get(1))
+                if index > len(key_stat) or index < 1:
+                    return CommandResult().message("账号序号错误。")
+                else:
+                    try:
+                        new_key = list(key_stat.keys())[index-1]
+                        self.provider.set_key(new_key)
+                    except BaseException as e:
+                        return CommandResult().message("切换账号未知错误: "+str(e))
+                    return CommandResult().message("切换账号成功。")    
+            except BaseException as e:
+                return CommandResult().message("切换账号错误。")
+        else:
+            return CommandResult().message("参数过多。")
+
+    def unset(self, message: AstrMessageEvent, context: Context):
+        self.provider.curr_personality = {}
+        self.provider.forget(message.session_id)
+        return CommandResult().message("已清除个性化设置。")
+    
+    
+    def set(self, message: AstrMessageEvent, context: Context):
+        l = message.message_str.split(" ")
+        if len(l) == 1:
+            return CommandResult().message("- 设置人格: \nset 人格名。例如 set 编剧\n- 人格列表: set list\n- 人格详细信息: set view 人格名\n- 自定义人格: set 人格文本\n- 重置会话(清除人格): reset\n- 重置会话(保留人格): reset p\n\n【当前人格】: " + str(self.provider.curr_personality['prompt']))
+        elif l[1] == "list":
+            msg = "人格列表：\n"
+            for key in personalities.keys():
+                msg += f"- {key}\n"
+            msg += '\n\n*输入 set view 人格名 查看人格详细信息'
+            return CommandResult().message(msg)
+        elif l[1] == "view":
+            if len(l) == 2:
+                return CommandResult().message("请输入人格名")
+            ps = l[2].strip()
+            if ps in personalities:
+                msg = f"人格{ps}的详细信息：\n"
+                msg += f"{personalities[ps]}\n"
+            else:
+                msg = f"人格{ps}不存在"
+            return CommandResult().message(msg)
+        else:
+            ps = "".join(l[1:]).strip()
+            if ps in personalities:
+                self.provider.curr_personality = {
+                    'name': ps,
+                    'prompt': personalities[ps]
+                }
+                self.provider.personality_set(self.provider.curr_personality, message.session_id)
+                return CommandResult().message(f"人格已设置。 \n人格信息: {ps}")
+            else:
+                self.provider.curr_personality = {
+                    'name': '自定义人格',
+                    'prompt': ps
+                }
+                self.provider.personality_set(self.provider.curr_personality, message.session_id)
+                return CommandResult().message(f"人格已设置。 \n人格信息: {ps}")
+
+    async def draw(self, message: AstrMessageEvent, context: Context):
+        message = message.message_str.removeprefix("画")
+        img_url = await self.provider.image_generate(message)
+        return CommandResult(
+            message_chain=[Image.fromURL(img_url)],
+        )

model/command/parser.py

@@ -0,0 +1,25 @@
+import re
+
+class CommandTokens():
+    def __init__(self) -> None:
+        self.tokens = []
+        self.len = 0
+        
+    def get(self, idx: int):
+        if idx >= self.len:
+            return None
+        return self.tokens[idx].strip()
+
+class CommandParser():
+    def __init__(self):
+        pass
+        
+    def parse(self, message: str):
+        cmd_tokens = CommandTokens()
+        cmd_tokens.tokens = message.split(" ")
+        cmd_tokens.len = len(cmd_tokens.tokens)
+        return cmd_tokens
+    
+    def regex_match(self, message: str, command: str) -> bool:
+        return re.search(command, message, re.MULTILINE) is not None
+    

model/platform/__init__.py

@@ -0,0 +1,94 @@
+import abc
+from typing import Union, Any, List
+from nakuru.entities.components import Plain, At, Image, BaseMessageComponent
+from type.astrbot_message import AstrBotMessage
+from type.command import CommandResult
+from type.astrbot_message import MessageType
+
+class T2IException(Exception):
+    def __init__(self, message: str = "文本转图片时发生错误") -> None:
+        super().__init__(message)
+
+class Platform():
+    def __init__(self, platform_name: str, context) -> None:
+        self.PLATFORM_NAME = platform_name
+        self.context = context
+
+    @abc.abstractmethod
+    async def handle_msg(self, message: AstrBotMessage):
+        '''
+        处理到来的消息
+        '''
+        pass
+
+    @abc.abstractmethod
+    async def reply_msg(self, message: AstrBotMessage, 
+                        result_message: List[BaseMessageComponent]):
+        '''
+        回复用户唤醒机器人的消息。（被动回复）
+        '''
+        pass
+
+    @abc.abstractmethod
+    async def send_msg(self, target: Any, result_message: CommandResult):
+        '''
+        发送消息（主动）
+        '''
+        pass
+    
+    @abc.abstractmethod
+    async def send_msg_new(self, message_type: MessageType, target: str, result_message: CommandResult):
+        '''
+        发送消息（主动）
+        '''
+        pass
+
+    def parse_message_outline(self, message: Union[AstrBotMessage, list]) -> str:
+        '''
+        将消息解析成大纲消息形式，如: xxxxx[图片]xxxxx。用于输出日志等。
+        '''
+        ret = ''
+        if isinstance(message, list):
+            parsed = message
+        elif isinstance(message, AstrBotMessage):
+            parsed = message.message
+        elif isinstance(message, str):
+            return message
+        
+        try:
+            for node in parsed:
+                if isinstance(node, Plain):
+                    ret += node.text.replace('\n', ' ')
+                elif isinstance(node, At):
+                    ret += f'[At: {node.name}/{node.qq}]'
+                elif isinstance(node, Image):
+                    ret += '[图片]'
+        except Exception as e:
+            pass
+        return ret[:100] if len(ret) > 100 else ret
+    
+    def check_nick(self, message_str: str) -> bool:
+        w = self.context.config_helper.wake_prefix
+        if not w: return False
+        for nick in w:
+            if nick and message_str.strip().startswith(nick):
+                return True
+        return False
+
+    async def convert_to_t2i_chain(self, message_result: list) -> list:
+        plain_str = ""
+        rendered_images = []
+        for i in message_result:
+            if isinstance(i, Plain):
+                plain_str += i.text
+        if plain_str and len(plain_str) > 50:
+            p = await self.context.image_renderer.render(plain_str, return_url=True)
+            if p.startswith('http'):
+                rendered_images.append(Image.fromURL(p))
+            else:
+                rendered_images.append(Image.fromFileSystem(p))
+            return rendered_images
+        return message_result
+        
+    async def record_metrics(self):
+        self.context.metrics_uploader.increment_platform_stat(self.PLATFORM_NAME)

model/platform/manager.py

@@ -0,0 +1,97 @@
+import asyncio
+
+from util.io import port_checker
+from type.register import RegisteredPlatform
+from type.types import Context
+from util.log import LogManager
+from logging import Logger
+from astrbot.message.handler import MessageHandler
+from util.cmd_config import (
+    PlatformConfig, 
+    AiocqhttpPlatformConfig, 
+    NakuruPlatformConfig, 
+    QQOfficialPlatformConfig
+)
+
+logger: Logger = LogManager.GetLogger(log_name='astrbot')
+
+
+class PlatformManager():
+    def __init__(self, context: Context, message_handler: MessageHandler) -> None:
+        self.context = context
+        self.msg_handler = message_handler
+        
+    def load_platforms(self):
+        tasks = []
+        
+        platforms = self.context.config_helper.platform
+        logger.info(f"加载 {len(platforms)} 个机器人消息平台...")
+        for platform in platforms:
+            if not platform.enable:
+                continue
+            if platform.name == "qq_official":
+                assert isinstance(platform, QQOfficialPlatformConfig), "qq_official: 无法识别的配置类型。"
+                logger.info(f"加载 QQ官方 机器人消息平台 (appid: {platform.appid})")
+                tasks.append(asyncio.create_task(self.qqofficial_bot(platform), name="qqofficial-adapter"))
+            elif platform.name == "nakuru":
+                assert isinstance(platform, NakuruPlatformConfig), "nakuru: 无法识别的配置类型。"
+                logger.info(f"加载 QQ(nakuru) 机器人消息平台 ({platform.host}, {platform.websocket_port}, {platform.port})")
+                tasks.append(asyncio.create_task(self.nakuru_bot(platform), name="nakuru-adapter"))
+            elif platform.name == "aiocqhttp":
+                assert isinstance(platform, AiocqhttpPlatformConfig), "aiocqhttp: 无法识别的配置类型。"
+                logger.info("加载 QQ(aiocqhttp) 机器人消息平台")
+                tasks.append(asyncio.create_task(self.aiocq_bot(platform), name="aiocqhttp-adapter"))
+
+        return tasks
+
+    async def nakuru_bot(self, config: NakuruPlatformConfig):
+        '''
+        运行 QQ(nakuru 适配器) 
+        '''
+        from model.platform.qq_nakuru import QQNakuru
+        noticed = False
+        host = config.host
+        port = config.websocket_port
+        http_port = config.port
+        logger.info(
+            f"正在检查连接...host: {host}, ws port: {port}, http port: {http_port}")
+        while True:
+            if not port_checker(port=port, host=host) or not port_checker(port=http_port, host=host):
+                if not noticed:
+                    noticed = True
+                    logger.warning(
+                        f"连接到{host}:{port}（或{http_port}）失败。程序会每隔 5s 自动重试。")
+                await asyncio.sleep(5)
+            else:
+                logger.info("nakuru 适配器已连接。")
+                break
+        try:
+            qq_gocq = QQNakuru(self.context, self.msg_handler, config)
+            self.context.platforms.append(RegisteredPlatform(
+                platform_name="nakuru", platform_instance=qq_gocq, origin="internal"))
+            await qq_gocq.run()
+        except BaseException as e:
+            logger.error("启动 nakuru 适配器时出现错误: " + str(e))
+
+    def aiocq_bot(self, config):
+        '''
+        运行 QQ(aiocqhttp 适配器)
+        '''
+        from model.platform.qq_aiocqhttp import AIOCQHTTP
+        qq_aiocqhttp = AIOCQHTTP(self.context, self.msg_handler, config)
+        self.context.platforms.append(RegisteredPlatform(
+            platform_name="aiocqhttp", platform_instance=qq_aiocqhttp, origin="internal"))
+        return qq_aiocqhttp.run_aiocqhttp()
+
+    def qqofficial_bot(self, config):
+        '''
+        运行 QQ 官方机器人适配器
+        '''
+        try:
+            from model.platform.qq_official import QQOfficial
+            qqchannel_bot = QQOfficial(self.context, self.msg_handler, config)
+            self.context.platforms.append(RegisteredPlatform(
+                platform_name="qqofficial", platform_instance=qqchannel_bot, origin="internal"))
+            return qqchannel_bot.run()
+        except BaseException as e:
+            logger.error("启动 QQ官方机器人适配器时出现错误: " + str(e))

model/platform/qq_aiocqhttp.py

@@ -0,0 +1,268 @@
+import time
+import asyncio
+import traceback
+import logging
+from aiocqhttp import CQHttp, Event
+from aiocqhttp.exceptions import ActionFailed
+from . import Platform, T2IException
+from type.astrbot_message import *
+from type.message_event import *
+from type.command import *
+from typing import Union, List, Dict
+from nakuru.entities.components import *
+from util.log import LogManager
+from logging import Logger
+from astrbot.message.handler import MessageHandler
+from util.cmd_config import PlatformConfig, AiocqhttpPlatformConfig
+
+logger: Logger = LogManager.GetLogger(log_name='astrbot')
+
+class AIOCQHTTP(Platform):
+    def __init__(self, context: Context, 
+                 message_handler: MessageHandler, 
+                 platform_config: PlatformConfig) -> None:
+        super().__init__("aiocqhttp", context)
+        assert isinstance(platform_config, AiocqhttpPlatformConfig), "aiocqhttp: 无法识别的配置类型。"
+        
+        self.message_handler = message_handler
+        self.context = context
+        self.config = platform_config
+        self.unique_session = context.config_helper.platform_settings.unique_session
+        self.host = platform_config.ws_reverse_host
+        self.port = platform_config.ws_reverse_port
+        
+    def convert_message(self, event: Event) -> AstrBotMessage:
+        
+        abm = AstrBotMessage()
+        abm.self_id = str(event.self_id)
+        abm.tag = "aiocqhttp"
+        
+        abm.sender = MessageMember(str(event.sender['user_id']), event.sender['nickname'])        
+
+        if event['message_type'] == 'group':
+            abm.type = MessageType.GROUP_MESSAGE
+        elif event['message_type'] == 'private':
+            abm.type = MessageType.FRIEND_MESSAGE
+        
+        if self.unique_session:
+            abm.session_id = abm.sender.user_id
+        else:
+            abm.session_id = str(event.group_id) if abm.type == MessageType.GROUP_MESSAGE else abm.sender.user_id
+        
+        abm.message_id = str(event.message_id)
+        abm.message = []
+        
+        message_str = ""
+        if not isinstance(event.message, list):
+            err = f"aiocqhttp: 无法识别的消息类型: {str(event.message)}，此条消息将被忽略。如果您在使用 go-cqhttp，请将其配置文件中的 message.post-format 更改为 array。"
+            logger.critical(err)
+            try:
+                self.bot.send(event, err)
+            except BaseException as e:
+                logger.error(f"回复消息失败: {e}")
+            return
+        for m in event.message:
+            t = m['type']
+            a = None
+            if t == 'at':
+                a = At(**m['data'])
+                abm.message.append(a)
+            if t == 'text':
+                a = Plain(text=m['data']['text'])
+                message_str += m['data']['text'].strip()
+                abm.message.append(a)
+            if t == 'image':
+                file = m['data']['file'] if 'file' in m['data'] else None
+                url = m['data']['url'] if 'url' in m['data'] else None
+                a = Image(file=file, url=url)
+                abm.message.append(a)
+        abm.timestamp = int(time.time())
+        abm.message_str = message_str
+        abm.raw_message = event
+        return abm
+            
+    def run_aiocqhttp(self):
+        if not self.host or not self.port:
+            return
+        self.bot = CQHttp(use_ws_reverse=True, import_name='aiocqhttp', api_timeout_sec=180)
+        @self.bot.on_message('group')
+        async def group(event: Event):
+            abm = self.convert_message(event)
+            if abm:
+                await self.handle_msg(abm)
+        
+        @self.bot.on_message('private')
+        async def private(event: Event):
+            abm = self.convert_message(event)
+            if abm:
+                await self.handle_msg(abm)
+        
+        bot = self.bot.run_task(host=self.host, port=int(self.port), shutdown_trigger=self.shutdown_trigger_placeholder)
+        
+        for handler in logging.root.handlers[:]:
+            logging.root.removeHandler(handler)
+        logging.getLogger('aiocqhttp').setLevel(logging.ERROR)
+        
+        return bot
+    
+    async def shutdown_trigger_placeholder(self):
+        while self.context.running:
+            await asyncio.sleep(1)
+    
+    async def pre_check(self, message: AstrBotMessage) -> bool:
+        # if message chain contains Plain components or 
+        # At components which points to self_id, return True
+        if message.type == MessageType.FRIEND_MESSAGE:
+            return True, "friend"
+        for comp in message.message:
+            if isinstance(comp, At) and str(comp.qq) == message.self_id:
+                return True, "at"
+        # check commands which ignore prefix
+        if await self.context.command_manager.check_command_ignore_prefix(message.message_str):
+            return True, "command"
+        # check nicks
+        if self.check_nick(message.message_str):
+            return True, "nick"
+        return False, "none"
+        
+    async def handle_msg(self, message: AstrBotMessage):
+        logger.info(
+            f"{message.sender.nickname}/{message.sender.user_id} -> {self.parse_message_outline(message)}")
+        
+        ok, reason = await self.pre_check(message)
+        if not ok:
+            return
+            
+        # parse unified message origin
+        unified_msg_origin = None
+        assert isinstance(message.raw_message, Event)
+        if message.type == MessageType.GROUP_MESSAGE:
+            unified_msg_origin = f"aiocqhttp:{message.type.value}:{message.raw_message.group_id}"
+        elif message.type == MessageType.FRIEND_MESSAGE:
+            unified_msg_origin = f"aiocqhttp:{message.type.value}:{message.sender.user_id}"
+        
+        logger.debug(f"unified_msg_origin: {unified_msg_origin}")
+
+        # construct astrbot message event
+        ame = AstrMessageEvent.from_astrbot_message(message, 
+                                                    self.context, 
+                                                    "aiocqhttp", 
+                                                    message.session_id, 
+                                                    unified_msg_origin,
+                                                    reason == "command") # only_command
+        
+        # transfer control to message handler
+        message_result = await self.message_handler.handle(ame)
+        if not message_result: return
+        
+        await self.reply_msg(message, message_result.result_message, message_result.use_t2i)
+        if message_result.callback:
+            message_result.callback()
+
+        return message_result
+   
+    async def reply_msg(self,
+                        message: AstrBotMessage,
+                        result_message: list,
+                        use_t2i: bool = None):
+        """
+        回复用户唤醒机器人的消息。（被动回复）
+        """
+        try:
+            await self._reply(message, result_message, use_t2i)
+        except T2IException as e:
+            logger.error(traceback.format_exc())
+            logger.warning(f"文本转图片时发生错误，将使用纯文本发送。")
+            await self._reply(message, result_message, False)
+        return result_message
+            
+    async def _reply(self, message: Union[AstrBotMessage, Dict], message_chain: List[BaseMessageComponent], use_t2i: bool = None):
+        await self.record_metrics()
+        if isinstance(message_chain, str): 
+            message_chain = [Plain(text=message_chain), ]
+            
+        # 文转图处理
+        if (use_t2i or (use_t2i == None and self.context.config_helper.t2i)) and isinstance(message_chain, list):
+            try:
+                message_chain = await self.convert_to_t2i_chain(message_chain)
+                if not message_chain: raise T2IException()
+            except BaseException as e:
+                raise T2IException()
+        
+        # log
+        if isinstance(message, AstrBotMessage):
+            logger.info(
+                f"{message.sender.nickname}/{message.sender.user_id} <- {self.parse_message_outline(message_chain)}")
+        else:
+            logger.info(f"回复消息: {message_chain}")
+            
+        # 解析成 OneBot json 格式并发送
+        ret = []
+        image_idx = []
+        for idx, segment in enumerate(message_chain):
+            d = segment.toDict()
+            if isinstance(segment, Plain):
+                d['type'] = 'text'
+            if isinstance(segment, Image):
+                image_idx.append(idx)
+            ret.append(d)
+        if os.environ.get('TEST_MODE', 'off') == 'on':
+            logger.info(f"回复消息: {ret}")
+            return
+        try:
+            await self._reply_wrapper(message, ret)
+        except ActionFailed as e:
+            if e.retcode == 1200:
+                # ENOENT
+                if not image_idx:
+                    raise e
+                logger.warning("回复失败。检测到失败原因为文件未找到，猜测用户的协议端与 AstrBot 位于不同的文件系统上。尝试采用上传图片的方式发图。")
+                for idx in image_idx:
+                    if ret[idx]['data']['file'].startswith('file://'):
+                        logger.info(f"正在上传图片: {ret[idx]['data']['path']}")
+                        # 除了上传到图床，想不到更好的办法。
+                        image_url = await self.context.image_uploader.upload_image(ret[idx]['data']['path'])
+                        logger.info(f"上传成功。")
+                        ret[idx]['data']['file'] = image_url
+                        ret[idx]['data']['path'] = image_url
+                await self._reply_wrapper(message, ret)
+            else:
+                logger.error(traceback.format_exc())
+                logger.error(f"回复消息失败: {e}")
+                raise e
+                    
+    async def _reply_wrapper(self, message: Union[AstrBotMessage, Dict], ret: List):
+        if isinstance(message, AstrBotMessage):
+            await self.bot.send(message.raw_message, ret)
+        if isinstance(message, dict):
+            if 'group_id' in message:
+                await self.bot.send_group_msg(group_id=message['group_id'], message=ret)
+            elif 'user_id' in message:
+                await self.bot.send_private_msg(user_id=message['user_id'], message=ret)
+            else:
+                raise Exception("aiocqhttp: 无法识别的消息来源。仅支持 group_id 和 user_id。")
+
+    async def send_msg(self, target: Dict[str, int], result_message: CommandResult):
+        '''
+        以主动的方式给QQ用户、QQ群发送一条消息。
+        
+        `target` 接收一个 dict 类型的值引用。
+        
+        - 要发给 QQ 下的某个用户，请添加 key `user_id`，值为 int 类型的 qq 号；
+        - 要发给某个群聊，请添加 key `group_id`，值为 int 类型的 qq 群号；
+        
+        '''
+        try:
+            await self._reply(target, result_message.message_chain, result_message.is_use_t2i)
+        except T2IException as e:
+            logger.error(traceback.format_exc())
+            logger.warning(f"文本转图片时发生错误，将使用纯文本发送。")
+            await self._reply(target, result_message.message_chain, False)
+        
+    async def send_msg_new(self, message_type: MessageType, target: str, result_message: CommandResult):
+        if message_type == MessageType.GROUP_MESSAGE:
+            await self.send_msg({'group_id': int(target)}, result_message)
+        elif message_type == MessageType.FRIEND_MESSAGE:
+            await self.send_msg({'user_id': int(target)}, result_message)
+        else:
+            raise Exception("aiocqhttp: 无法识别的消息类型。")

model/platform/qq_nakuru.py

@@ -0,0 +1,287 @@
+import time, asyncio, traceback
+
+from nakuru.entities.components import Plain, At, Image, Node, BaseMessageComponent
+from nakuru import (
+    CQHTTP,
+    GuildMessage,
+    GroupMessage,
+    FriendMessage,
+    GroupMemberIncrease,
+    MessageItemType
+)
+from typing import Union, List, Dict
+from type.types import Context
+from . import Platform, T2IException
+from type.astrbot_message import *
+from type.message_event import *
+from type.command import *
+from util.log import LogManager
+from logging import Logger
+from astrbot.message.handler import MessageHandler
+from util.cmd_config import PlatformConfig, NakuruPlatformConfig
+
+logger: Logger = LogManager.GetLogger(log_name='astrbot')
+
+
+class FakeSource:
+    def __init__(self, type, group_id):
+        self.type = type
+        self.group_id = group_id
+
+
+class QQNakuru(Platform):
+    def __init__(self, context: Context, 
+                 message_handler: MessageHandler,
+                 platform_config: PlatformConfig) -> None:
+        super().__init__("nakuru", context)
+        assert isinstance(platform_config, NakuruPlatformConfig), "gocq: 无法识别的配置类型。"
+        
+        self.loop = asyncio.new_event_loop()
+        asyncio.set_event_loop(self.loop)
+        
+        self.message_handler = message_handler
+        self.context = context
+        self.unique_session = context.config_helper.platform_settings.unique_session
+        self.config = platform_config
+        
+        self.client = CQHTTP(
+            host=self.config.host,
+            port=self.config.websocket_port,
+            http_port=self.config.port
+        )
+        gocq_app = self.client
+
+        @gocq_app.receiver("GroupMessage")
+        async def _(app: CQHTTP, source: GroupMessage):
+            if self.config.enable_group:
+                abm = self.convert_message(source)
+                await self.handle_msg(abm)
+
+        @gocq_app.receiver("FriendMessage")
+        async def _(app: CQHTTP, source: FriendMessage):
+            if self.config.enable_direct_message:
+                abm = self.convert_message(source)
+                await self.handle_msg(abm)
+
+        @gocq_app.receiver("GuildMessage")
+        async def _(app: CQHTTP, source: GuildMessage):
+            if self.config.enable_guild:
+                abm = self.convert_message(source)
+                await self.handle_msg(abm)
+                        
+    def pre_check(self, message: AstrBotMessage) -> bool:
+        # if message chain contains Plain components or At components which points to self_id, return True
+        if message.type == MessageType.FRIEND_MESSAGE:
+            return True, "friend"
+        for comp in message.message:
+            if isinstance(comp, At) and str(comp.qq) == message.self_id:
+                return True, "at"
+        # check commands which ignore prefix
+        if self.context.command_manager.check_command_ignore_prefix(message.message_str):
+            return True, "command"
+        # check nicks
+        if self.check_nick(message.message_str):
+            return True, "nick"
+        return False, "none"
+
+    def run(self):
+        coro = self.client._run()
+        return coro
+
+    async def handle_msg(self, message: AstrBotMessage):
+        logger.info(
+            f"{message.sender.nickname}/{message.sender.user_id} -> {self.parse_message_outline(message)}")
+
+        assert isinstance(message.raw_message,
+                          (GroupMessage, FriendMessage, GuildMessage))
+
+        # 判断是否响应消息
+        ok, reason = self.pre_check(message)
+        if not ok:
+            return
+
+        # 解析 session_id
+        if self.unique_session or message.type == MessageType.FRIEND_MESSAGE:
+            session_id = message.raw_message.user_id
+        elif message.type == MessageType.GROUP_MESSAGE:
+            session_id = message.raw_message.group_id
+        elif message.type == MessageType.GUILD_MESSAGE:
+            session_id = message.raw_message.channel_id
+        else:
+            session_id = message.raw_message.user_id
+
+        message.session_id = session_id
+            
+        # parse unified message origin
+        unified_msg_origin = None
+        if message.type == MessageType.GROUP_MESSAGE:
+            assert isinstance(message.raw_message, GroupMessage)
+            unified_msg_origin = f"nakuru:{message.type.value}:{message.raw_message.group_id}"
+        elif message.type == MessageType.FRIEND_MESSAGE:
+            assert isinstance(message.raw_message, FriendMessage)
+            unified_msg_origin = f"nakuru:{message.type.value}:{message.sender.user_id}"
+        elif message.type == MessageType.GUILD_MESSAGE:
+            assert isinstance(message.raw_message, GuildMessage)
+            unified_msg_origin = f"nakuru:{message.type.value}:{message.raw_message.channel_id}"
+        
+        logger.debug(f"unified_msg_origin: {unified_msg_origin}")
+
+            
+        # construct astrbot message event
+        ame = AstrMessageEvent.from_astrbot_message(message, 
+                                                    self.context, 
+                                                    "nakuru", 
+                                                    session_id, 
+                                                    unified_msg_origin,
+                                                    reason == 'command') # only_command
+        
+        # transfer control to message handler
+        message_result = await self.message_handler.handle(ame)
+        if not message_result: return
+        
+        await self.reply_msg(message, message_result.result_message, message_result.use_t2i)
+        if message_result.callback:
+            message_result.callback()
+
+    async def reply_msg(self,
+                        message: AstrBotMessage,
+                        result_message: List[BaseMessageComponent],
+                        use_t2i: bool = None):
+        """
+        回复用户唤醒机器人的消息。（被动回复）
+        """        
+        assert isinstance(message.raw_message, (GroupMessage, FriendMessage, GuildMessage))
+
+        try:
+            await self._reply(message, result_message, use_t2i)
+        except T2IException as e:
+            logger.error(traceback.format_exc())
+            logger.warning(f"文本转图片时发生错误，将使用纯文本发送。")
+            await self._reply(message, result_message, False)
+        return result_message
+        
+    async def _reply(self, message: Union[AstrBotMessage, Dict], message_chain: List[BaseMessageComponent], use_t2i: bool = None):
+        await self.record_metrics()
+        if isinstance(message_chain, str): 
+            message_chain = [Plain(text=message_chain), ]
+            
+        # 文转图处理
+        if (use_t2i or (use_t2i == None and self.context.config_helper.t2i)) and isinstance(message_chain, list):
+            try:
+                message_chain = await self.convert_to_t2i_chain(message_chain)
+                if not message_chain: raise T2IException()
+            except BaseException as e:
+                raise T2IException()
+                
+        # log
+        if isinstance(message, AstrBotMessage):
+            logger.info(
+                f"{message.sender.nickname}/{message.sender.user_id} <- {self.parse_message_outline(message_chain)}")
+        else:
+            logger.info(f"回复消息: {message_chain}")
+        
+        source = message.raw_message
+        is_dict = isinstance(source, dict)
+        
+        # 发消息
+        typ = None
+        if is_dict:
+            if "group_id" in source:
+                typ = "GroupMessage"
+            elif "user_id" in source:
+                typ = "FriendMessage"
+            elif "guild_id" in source:
+                typ = "GuildMessage"
+        else:
+            typ = source.type
+        
+        if typ == "GuildMessage":
+            guild_id = source['guild_id'] if is_dict else source.guild_id
+            chan_id = source['channel_id'] if is_dict else source.channel_id
+            await self.client.sendGuildChannelMessage(guild_id, chan_id, message_chain)
+        elif typ == "FriendMessage":
+            user_id = source['user_id'] if is_dict else source.user_id
+            await self.client.sendFriendMessage(user_id, message_chain)
+        elif typ == "GroupMessage":
+            group_id = source['group_id'] if is_dict else source.group_id
+            # 过长时forward发送
+            plain_text_len = 0
+            image_num = 0
+            for i in message_chain:
+                if isinstance(i, Plain):
+                    plain_text_len += len(i.text)
+                elif isinstance(i, Image):
+                    image_num += 1
+            if plain_text_len > self.context.config_helper.platform_settings.forward_threshold or image_num > 1:
+                # 删除At
+                for i in message_chain:
+                    if isinstance(i, At):
+                        message_chain.remove(i)
+                node = Node(message_chain)
+                node.uin = 123456
+                node.name = f"bot"
+                node.time = int(time.time())
+                nodes = [node]
+                await self.client.sendGroupForwardMessage(group_id, nodes)
+                return
+            await self.client.sendGroupMessage(group_id, message_chain)
+        
+    async def send_msg(self, target: Dict[str, int], result_message: CommandResult):
+        '''
+        以主动的方式给用户、群或者频道发送一条消息。
+        
+        `target` 接收一个 dict 类型的值引用。
+        
+        - 要发给 QQ 下的某个用户，请添加 key `user_id`，值为 int 类型的 qq 号；
+        - 要发给某个群聊，请添加 key `group_id`，值为 int 类型的 qq 群号；
+        - 要发给某个频道，请添加 key `guild_id`, `channel_id`。均为 int 类型。
+        
+        guild_id 不是频道号。
+        '''
+        try:
+            await self._reply(target, result_message.message_chain, result_message.is_use_t2i)
+        except T2IException as e:
+            logger.error(traceback.format_exc())
+            logger.warning(f"文本转图片时发生错误，将使用纯文本发送。")
+            await self._reply(target, result_message.message_chain, False)
+        return result_message
+        
+    async def send_msg_new(self, message_type: MessageType, target: str, result_message: CommandResult):
+        '''
+        以主动的方式给用户、群或者频道发送一条消息。
+        
+        `message_type` 为 MessageType 枚举类型。
+        
+        - 要发给 QQ 下的某个用户，请使用 MessageType.FRIEND_MESSAGE；
+        - 要发给某个群聊，请使用 MessageType.GROUP_MESSAGE；
+        - 要发给某个频道，请使用 MessageType.GUILD_MESSAGE。
+        '''
+        if message_type == MessageType.FRIEND_MESSAGE:
+            await self.send_msg({"user_id": int(target)}, result_message)
+        elif message_type == MessageType.GROUP_MESSAGE:
+            await self.send_msg({"group_id": int(target)}, result_message)
+        elif message_type == MessageType.GUILD_MESSAGE:
+            await self.send_msg({"channel_id": int(target)}, result_message)
+
+    def convert_message(self, message: Union[GroupMessage, FriendMessage, GuildMessage]) -> AstrBotMessage:
+        abm = AstrBotMessage()
+        abm.type = MessageType(message.type)
+        abm.raw_message = message
+        abm.message_id = message.message_id
+
+        plain_content = ""
+        for i in message.message:
+            if isinstance(i, Plain):
+                plain_content += i.text
+        abm.message_str = plain_content.strip()
+        if message.type == MessageItemType.GuildMessage:
+            abm.self_id = str(message.self_tiny_id)
+        else:
+            abm.self_id = str(message.self_id)
+        abm.sender = MessageMember(
+            str(message.sender.user_id),
+            str(message.sender.nickname)
+        )
+        abm.tag = "nakuru"
+        abm.message = message.message
+        return abm

model/platform/qq_official.py

@@ -0,0 +1,380 @@
+import botpy
+import re
+import time
+import traceback
+import asyncio
+import botpy.message
+import botpy.types
+import botpy.types.message
+
+from botpy.types.message import Reference, Media
+from botpy import Client
+from util.io import save_temp_img, download_image_by_url
+from . import Platform
+from type.astrbot_message import *
+from type.message_event import *
+from type.command import *
+from typing import Union, List, Dict
+from nakuru.entities.components import *
+from util.log import LogManager
+from logging import Logger
+from astrbot.message.handler import MessageHandler
+from util.cmd_config import PlatformConfig, QQOfficialPlatformConfig
+
+logger: Logger = LogManager.GetLogger(log_name='astrbot')
+
+# QQ 机器人官方框架
+class botClient(Client):
+    def set_platform(self, platform: 'QQOfficial'):
+        self.platform = platform
+        
+    # 收到群消息
+    async def on_group_at_message_create(self, message: botpy.message.GroupMessage):
+        abm = self.platform._parse_from_qqofficial(message, MessageType.GROUP_MESSAGE)
+        await self.platform.handle_msg(abm)
+
+    # 收到频道消息
+    async def on_at_message_create(self, message: botpy.message.Message):
+        # 转换层
+        abm = self.platform._parse_from_qqofficial(message, MessageType.GUILD_MESSAGE)
+        await self.platform.handle_msg(abm)
+
+    # 收到私聊消息
+    async def on_direct_message_create(self, message: botpy.message.DirectMessage):
+        # 转换层
+        abm = self.platform._parse_from_qqofficial(message, MessageType.FRIEND_MESSAGE)
+        await self.platform.handle_msg(abm)
+        
+    # 收到 C2C 消息
+    async def on_c2c_message_create(self, message: botpy.message.C2CMessage):
+        abm = self.platform._parse_from_qqofficial(message, MessageType.FRIEND_MESSAGE)
+        await self.platform.handle_msg(abm)
+
+
+class QQOfficial(Platform):
+
+    def __init__(self, context: Context, 
+                 message_handler: MessageHandler, 
+                 platform_config: PlatformConfig,
+                 test_mode = False) -> None:
+        super().__init__("qqofficial", context)
+        assert isinstance(platform_config, QQOfficialPlatformConfig), "qq_official: 无法识别的配置类型。"
+        self.loop = asyncio.new_event_loop()
+        asyncio.set_event_loop(self.loop)
+        
+        self.message_handler = message_handler
+        self.context = context
+        self.config = platform_config
+        
+        self.appid = platform_config.appid
+        self.secret = platform_config.secret
+        self.unique_session = context.config_helper.platform_settings.unique_session
+        qq_group = platform_config.enable_group_c2c
+        guild_dm = platform_config.enable_guild_direct_message
+
+        if qq_group:
+            self.intents = botpy.Intents(
+                public_messages=True,
+                public_guild_messages=True,
+                direct_message=guild_dm
+            )
+        else:
+            self.intents = botpy.Intents(
+                public_guild_messages=True,
+                direct_message=guild_dm
+            )
+        self.client = botClient(
+            intents=self.intents,
+            bot_log=False,
+            timeout=20,
+        )
+
+        self.client.set_platform(self)
+        
+        self.test_mode = os.environ.get('TEST_MODE', 'off') == 'on'
+
+    async def _parse_to_qqofficial(self, message: List[BaseMessageComponent], is_group: bool = False):
+        plain_text = ""
+        image_path = None  # only one img supported
+        for i in message:
+            if isinstance(i, Plain):
+                plain_text += i.text
+            elif isinstance(i, Image) and not image_path:
+                if i.path:
+                    image_path = i.path
+                elif i.file and i.file.startswith("base64://"):
+                    img_data = base64.b64decode(i.file[9:])
+                    image_path = save_temp_img(img_data)
+                elif i.file and i.file.startswith("http"):
+                    # 如果是群消息，不需要下载
+                    image_path = await download_image_by_url(i.file) if not is_group else i.file
+        return plain_text, image_path
+
+    def _parse_from_qqofficial(self, message: Union[botpy.message.Message, botpy.message.GroupMessage],
+                                  message_type: MessageType):
+        abm = AstrBotMessage()
+        abm.type = message_type
+        abm.timestamp = int(time.time())
+        abm.raw_message = message
+        abm.message_id = message.id
+        abm.tag = "qqofficial"
+        msg: List[BaseMessageComponent] = []
+
+        if isinstance(message, botpy.message.GroupMessage) or isinstance(message, botpy.message.C2CMessage):
+            if isinstance(message, botpy.message.GroupMessage):
+                abm.sender = MessageMember(
+                    message.author.member_openid,
+                    ""
+                )
+            else:
+                abm.sender = MessageMember(
+                    message.author.user_openid,
+                    ""
+                )
+            abm.message_str = message.content.strip()
+            abm.self_id = "unknown_selfid"
+
+            msg.append(Plain(abm.message_str))
+            if message.attachments:
+                for i in message.attachments:
+                    if i.content_type.startswith("image"):
+                        url = i.url
+                        if not url.startswith("http"):
+                            url = "https://"+url
+                        img = Image.fromURL(url)
+                        msg.append(img)
+            abm.message = msg
+
+        elif isinstance(message, botpy.message.Message) or isinstance(message, botpy.message.DirectMessage):
+            try:
+                abm.self_id = str(message.mentions[0].id)
+            except:
+                abm.self_id = ""
+
+            plain_content = message.content.replace(
+                "<@!"+str(abm.self_id)+">", "").strip()
+            msg.append(Plain(plain_content))
+            if message.attachments:
+                for i in message.attachments:
+                    if i.content_type.startswith("image"):
+                        url = i.url
+                        if not url.startswith("http"):
+                            url = "https://"+url
+                        img = Image.fromURL(url)
+                        msg.append(img)
+            abm.message = msg
+            abm.message_str = plain_content
+            abm.sender = MessageMember(
+                str(message.author.id),
+                str(message.author.username)
+            )
+        else:
+            raise ValueError(f"Unknown message type: {message_type}")
+        return abm
+
+    def run(self):
+        return self.client.start(
+            appid=self.appid,
+            secret=self.secret
+        )
+
+    async def handle_msg(self, message: AstrBotMessage):
+        assert isinstance(message.raw_message, (botpy.message.Message,
+                          botpy.message.GroupMessage, botpy.message.DirectMessage, botpy.message.C2CMessage))
+        is_group = message.type != MessageType.FRIEND_MESSAGE
+
+        _t = "/私聊" if not is_group else ""
+        logger.info(
+            f"{message.sender.nickname}({message.sender.user_id}{_t}) -> {self.parse_message_outline(message)}")
+
+        # 解析出 session_id
+        if self.unique_session or not is_group:
+            session_id = message.sender.user_id
+        else:
+            if message.type == MessageType.GUILD_MESSAGE:
+                session_id = message.raw_message.channel_id
+            elif message.type == MessageType.GROUP_MESSAGE:
+                session_id = str(message.raw_message.group_openid)
+            else:
+                session_id = str(message.raw_message.author.id)
+        message.session_id = session_id
+
+        # construct astrbot message event
+        ame = AstrMessageEvent.from_astrbot_message(message, self.context, "qqofficial", session_id)
+        
+        message_result = await self.message_handler.handle(ame)
+        if not message_result:
+            return
+
+        ret = await self.reply_msg(message, message_result.result_message, message_result.use_t2i)
+        if message_result.callback:
+            message_result.callback()
+
+        return ret
+
+    async def reply_msg(self,
+                        message: AstrBotMessage,
+                        result_message: List[BaseMessageComponent],
+                        use_t2i: bool = None):
+        '''
+        回复频道消息
+        '''
+        source = message.raw_message
+        assert isinstance(source, (botpy.message.Message,
+                          botpy.message.GroupMessage, botpy.message.DirectMessage, botpy.message.C2CMessage))
+        logger.info(
+            f"{message.sender.nickname}({message.sender.user_id}) <- {self.parse_message_outline(result_message)}")
+
+        plain_text = ''
+        image_path = ''
+        msg_ref = None
+        rendered_images = None
+        
+        if use_t2i or (use_t2i == None and self.context.config_helper.t2i) and isinstance(result_message, list):
+            try:
+                rendered_images = await self.convert_to_t2i_chain(result_message)
+            except BaseException as e:
+                logger.warning(traceback.format_exc())
+                logger.warning(f"文本转图片时发生错误: {e}，将尝试默认方式。")
+                rendered_images = None
+        
+        if isinstance(result_message, list):
+            plain_text, image_path = await self._parse_to_qqofficial(result_message, message.type == MessageType.GROUP_MESSAGE)
+        else:
+            plain_text = result_message
+
+        if source and not image_path:  # file_image与message_reference不能同时传入
+            msg_ref = Reference(message_id=source.id,
+                                ignore_get_message_error=False)
+
+        # 到这里，我们得到了 plain_text，image_path，msg_ref
+        data = {
+            'content': plain_text,
+            'msg_id': message.message_id,
+            'message_reference': msg_ref
+        }
+        
+        if isinstance(message.raw_message, botpy.message.GroupMessage):
+            data['group_openid'] = str(source.group_openid)
+        elif isinstance(message.raw_message, botpy.message.Message):
+            data['channel_id'] = source.channel_id
+        elif isinstance(message.raw_message, botpy.message.DirectMessage):
+            data['guild_id'] = source.guild_id
+        elif isinstance(message.raw_message, botpy.message.C2CMessage):
+            data['openid'] = source.author.user_openid
+        if image_path:
+            data['file_image'] = image_path
+        if rendered_images:
+            # 文转图
+            _data = data.copy()
+            _data['content'] = ''
+            _data['file_image'] = rendered_images[0].file
+            _data['message_reference'] = None
+            
+            try:
+                return await self._reply(**_data)
+            except BaseException as e:
+                logger.warn(traceback.format_exc())
+                logger.warn(f"以文本转图片的形式回复消息时发生错误: {e}，将尝试默认方式。")
+
+        try:
+            return await self._reply(**data)
+        except BaseException as e:
+            logger.error(traceback.format_exc())
+            # 分割过长的消息
+            if "msg over length" in str(e):
+                split_res = []
+                split_res.append(plain_text[:len(plain_text)//2])
+                split_res.append(plain_text[len(plain_text)//2:])
+                for i in split_res:
+                    data['content'] = i
+                    return await self._reply(**data)
+            else:
+                try:
+                    # 防止被qq频道过滤消息
+                    plain_text = plain_text.replace(".", " . ")
+                    return await self._reply(**data)
+                except BaseException as e:
+                    try:
+                        data['content'] = str.join(" ", plain_text)
+                        return await self._reply(**data)
+                    except BaseException as e:
+                        plain_text = re.sub(
+                            r'(https|http)?:\/\/(\w|\.|\/|\?|\=|\&|\%)*\b', '[被隐藏的链接]', str(e), flags=re.MULTILINE)
+                        plain_text = plain_text.replace(".", "·")
+                        data['content'] = plain_text
+                        return await self._reply(**data)
+
+    async def _reply(self, **kwargs):
+        await self.record_metrics()
+        if 'group_openid' in kwargs or 'openid' in kwargs:
+            # QQ群组消息
+            if 'file_image' in kwargs and kwargs['file_image']:
+                file_image_path = kwargs['file_image'].replace("file:///", "")
+                if file_image_path:
+                    
+                    if file_image_path.startswith("http"):
+                        image_url = file_image_path
+                    else:
+                        logger.debug(f"上传图片: {file_image_path}")
+                        image_url = await self.context.image_uploader.upload_image(file_image_path)
+                        logger.debug(f"上传成功: {image_url}")
+                    if 'group_openid' in kwargs:
+                        media = await self.client.api.post_group_file(kwargs['group_openid'], 1, image_url)
+                    elif 'openid' in kwargs:
+                        media = await self.client.api.post_c2c_file(kwargs['openid'], 1, image_url)
+                    del kwargs['file_image']
+                    kwargs['media'] = media
+                    logger.debug(f"发送群图片: {media}")
+                    kwargs['msg_type'] = 7 # 富媒体
+            if self.test_mode: 
+                return kwargs
+            if 'group_openid' in kwargs:
+                await self.client.api.post_group_message(**kwargs)
+            elif 'openid' in kwargs:
+                await self.client.api.post_c2c_message(**kwargs)
+        elif 'channel_id' in kwargs:
+            # 频道消息
+            if 'file_image' in kwargs and kwargs['file_image']:
+                kwargs['file_image'] = kwargs['file_image'].replace("file:///", "")
+                # 频道消息发图只支持本地
+                if kwargs['file_image'].startswith("http"):
+                    kwargs['file_image'] = await download_image_by_url(kwargs['file_image'])
+            if self.test_mode:
+                return kwargs
+            await self.client.api.post_message(**kwargs)
+        elif 'guild_id' in kwargs:
+            # 频道私聊消息
+            if 'file_image' in kwargs and kwargs['file_image']:
+                kwargs['file_image'] = kwargs['file_image'].replace("file:///", "")
+                if kwargs['file_image'].startswith("http"):
+                    kwargs['file_image'] = await download_image_by_url(kwargs['file_image'])
+            if self.test_mode:
+                return kwargs
+            await self.client.api.post_dms(**kwargs)
+        else:
+            raise ValueError("Unknown target type.")
+
+    async def send_msg(self, target: Dict[str, str], result_message: CommandResult):
+        '''
+        以主动的方式给频道用户、群、频道或者消息列表用户（QQ用户）发送一条消息。
+        
+        `target` 接收一个 dict 类型的值引用。
+        
+        - 如果目标是 QQ 群，请添加 key `group_openid`。
+        - 如果目标是 频道消息，请添加 key `channel_id`。
+        - 如果目标是 频道私聊，请添加 key `guild_id`。
+        - 如果目标是 QQ 用户，请添加 key `openid`。
+        '''
+        plain_text, image_path = await self._parse_to_qqofficial(result_message.message_chain)
+
+        payload = {
+            'content': plain_text,
+            **target
+        }
+        if image_path:
+            payload['file_image'] = image_path
+        await self._reply(**payload)
+        
+    async def send_msg_new(self, message_type: MessageType, target: str, result_message: CommandResult):
+        raise NotImplementedError("qqofficial 不支持此方法。")

model/plugin/command.py

@@ -0,0 +1,26 @@
+from dataclasses import dataclass
+from type.register import RegisteredPlugins
+from typing import List, Union, Callable
+from util.log import LogManager
+from logging import Logger
+
+logger: Logger = LogManager.GetLogger(log_name='astrbot')
+
+
+@dataclass
+class CommandRegisterRequest():
+    command_name: str
+    description: str
+    priority: int
+    handler: Callable
+    use_regex: bool = False
+    plugin_name: str = None
+    ignore_prefix: bool = False
+
+class PluginCommandBridge():
+    def __init__(self, cached_plugins: RegisteredPlugins):
+        self.plugin_commands_waitlist: List[CommandRegisterRequest] = []
+        self.cached_plugins = cached_plugins
+        
+    def register_command(self, plugin_name, command_name, description, priority, handler, use_regex=False, ignore_prefix=False):
+        self.plugin_commands_waitlist.append(CommandRegisterRequest(command_name, description, priority, handler, use_regex, plugin_name, ignore_prefix))

model/plugin/manager.py

@@ -0,0 +1,293 @@
+import inspect
+import os
+import sys
+import traceback
+import uuid
+import shutil
+import yaml
+import subprocess
+
+from util.updator.plugin_updator import PluginUpdator
+from util.io import remove_dir, download_file
+from types import ModuleType
+from type.types import Context
+from type.plugin import *
+from type.register import *
+from util.log import LogManager
+from logging import Logger
+
+logger: Logger = LogManager.GetLogger(log_name='astrbot')
+
+class PluginManager():
+    def __init__(self, context: Context):
+        self.updator = PluginUpdator()
+        self.plugin_store_path = self.updator.get_plugin_store_path()
+        self.context = context
+
+    def get_classes(self, arg: ModuleType):
+        classes = []
+        clsmembers = inspect.getmembers(arg, inspect.isclass)
+        for (name, _) in clsmembers:
+            if name.lower().endswith("plugin") or name.lower() == "main":
+                classes.append(name)
+                break
+        return classes
+
+    def get_modules(self, path):
+        modules = []
+
+        dirs = os.listdir(path)
+        # 遍历文件夹，找到 main.py 或者和文件夹同名的文件
+        for d in dirs:
+            if os.path.isdir(os.path.join(path, d)):
+                if os.path.exists(os.path.join(path, d, "main.py")):
+                    module_str = 'main'
+                elif os.path.exists(os.path.join(path, d, d + ".py")):
+                    module_str = d
+                else:
+                    print(f"插件 {d} 未找到 main.py 或者 {d}.py，跳过。")
+                    continue
+                if os.path.exists(os.path.join(path, d, "main.py")) or os.path.exists(os.path.join(path, d, d + ".py")):
+                    modules.append({
+                        "pname": d,
+                        "module": module_str,
+                        "module_path": os.path.join(path, d, module_str)
+                    })
+        return modules
+    
+    def get_plugin_modules(self):
+        plugins = []
+        try:
+            plugin_dir = self.plugin_store_path
+            if os.path.exists(plugin_dir):
+                plugins = self.get_modules(plugin_dir)
+                return plugins
+        except BaseException as e:
+            raise e
+        
+    def check_plugin_dept_update(self, target_plugin: str = None):
+        plugin_dir = self.plugin_store_path
+        if not os.path.exists(plugin_dir):
+            return False
+        to_update = []
+        if target_plugin:
+            to_update.append(target_plugin)
+        else:
+            for p in self.context.cached_plugins:
+                to_update.append(p.root_dir_name)
+        for p in to_update:
+            plugin_path = os.path.join(plugin_dir, p)
+            if os.path.exists(os.path.join(plugin_path, "requirements.txt")):
+                pth = os.path.join(plugin_path, "requirements.txt")
+                logger.info(f"正在检查更新插件 {p} 的依赖: {pth}")
+                self.update_plugin_dept(os.path.join(plugin_path, "requirements.txt"))
+
+    def update_plugin_dept(self, path, break_system_package=True):
+        mirror = "https://mirrors.aliyun.com/pypi/simple/"
+        py = sys.executable
+        cmd = f"{py} -m pip install -r {path} -i {mirror} --trusted-host mirrors.aliyun.com"
+        if break_system_package:
+            cmd += " --break-system-package"
+        process = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True, universal_newlines=True)
+        
+        while True:
+            output = process.stdout.readline()
+            err = process.stderr.readline()
+            if err:
+                err = err.strip()
+                logger.error(err)
+                if "no such option: --break-system-package" in err:
+                    self.update_plugin_dept(path, break_system_package=False)
+                    break
+            if output == '' and process.poll() is not None:
+                break
+            if output:
+                output = output.strip()
+                if output.startswith("Requirement already satisfied"):
+                    continue
+                if output.startswith("Using cached"):
+                    continue
+                if output.startswith("Looking in indexes"):
+                    continue
+                logger.info(output)
+            
+        rc = process.poll()
+        
+        
+    async def install_plugin(self, repo_url: str):
+        ppath = self.plugin_store_path
+
+        # we no longer use Git anymore :)
+        # Repo.clone_from(repo_url, to_path=plugin_path, branch='master')
+        
+        plugin_path = await self.updator.update(repo_url)
+        with open(os.path.join(plugin_path, "REPO"), "w", encoding='utf-8') as f:
+            f.write(repo_url)
+            
+        self.check_plugin_dept_update()
+
+        return plugin_path
+        # ok, err = self.plugin_reload()
+        # if not ok:
+        #     raise Exception(err)
+        
+    async def download_from_repo_url(self, target_path: str, repo_url: str):
+        repo_namespace = repo_url.split("/")[-2:]
+        author = repo_namespace[0]
+        repo = repo_namespace[1]
+
+        logger.info(f"正在下载插件 {repo} ...")
+        release_url = f"https://api.github.com/repos/{author}/{repo}/releases"
+        releases = await self.updator.fetch_release_info(url=release_url)
+        if not releases:
+            # download from the default branch directly. 
+            logger.warn(f"未在插件 {author}/{repo} 中找到任何发布版本，将从默认分支下载。")
+            release_url = f"https://github.com/{author}/{repo}/archive/refs/heads/master.zip"
+        else:
+            release_url = releases[0]['zipball_url']
+
+        await download_file(release_url, target_path + ".zip")
+
+    def get_registered_plugin(self, plugin_name: str) -> RegisteredPlugin:
+        for p in self.context.cached_plugins:
+            if p.metadata.plugin_name == plugin_name:
+                return p
+    
+    def uninstall_plugin(self, plugin_name: str):
+        plugin = self.get_registered_plugin(plugin_name)
+        if not plugin:
+            raise Exception("插件不存在。")
+        root_dir_name = plugin.root_dir_name
+        ppath = self.plugin_store_path
+        self.context.cached_plugins.remove(plugin)
+        if not remove_dir(os.path.join(ppath, root_dir_name)):
+            raise Exception("移除插件成功，但是删除插件文件夹失败。您可以手动删除该文件夹，位于 addons/plugins/ 下。")
+
+    async def update_plugin(self, plugin_name: str):
+        plugin = self.get_registered_plugin(plugin_name)
+        if not plugin:
+            raise Exception("插件不存在。")
+        
+        await self.updator.update(plugin)
+        
+    def plugin_reload(self):
+        cached_plugins = self.context.cached_plugins
+        plugins = self.get_plugin_modules()
+        if plugins is None:
+            return False, "未找到任何插件模块"
+        fail_rec = ""
+
+        registered_map = {}
+        for p in cached_plugins:
+            registered_map[p.module_path] = None
+
+        for plugin in plugins:
+            try:
+                p = plugin['module']
+                module_path = plugin['module_path']
+                root_dir_name = plugin['pname']
+                
+                logger.info(f"正在加载插件 {root_dir_name} ...")
+
+                self.check_plugin_dept_update(target_plugin=root_dir_name)
+
+                module = __import__("data.plugins." +
+                                        root_dir_name + "." + p, fromlist=[p])
+
+                cls = self.get_classes(module)
+                
+                try:
+                    # 尝试传入 ctx
+                    obj = getattr(module, cls[0])(context=self.context)
+                except TypeError:
+                    obj = getattr(module, cls[0])()
+                except BaseException as e:
+                    raise e
+
+                metadata = None
+
+                plugin_path = os.path.join(self.plugin_store_path, root_dir_name)
+                metadata = self.load_plugin_metadata(plugin_path=plugin_path, plugin_obj=obj)
+                
+                logger.info(f"插件 {metadata.plugin_name}({metadata.author}) 加载成功。")
+
+                if module_path not in registered_map:
+                    cached_plugins.append(RegisteredPlugin(
+                        metadata=metadata,
+                        plugin_instance=obj,
+                        module=module,
+                        module_path=module_path,
+                        root_dir_name=root_dir_name
+                    ))
+            except BaseException as e:
+                traceback.print_exc()
+                fail_rec += f"加载{p}插件出现问题，原因 {str(e)}\n"
+
+        if not fail_rec:
+            return True, None
+        else:
+            return False, fail_rec
+        
+    def install_plugin_from_file(self, zip_file_path: str):
+        # try to unzip
+        temp_dir = os.path.join(os.path.dirname(zip_file_path), str(uuid.uuid4()))
+        self.updator.unzip_file(zip_file_path, temp_dir)
+        # check if the plugin has metadata.yaml
+        if not os.path.exists(os.path.join(temp_dir, "metadata.yaml")):
+            remove_dir(temp_dir)
+            raise Exception("插件缺少 metadata.yaml 文件。")
+        
+        metadata = self.load_plugin_metadata(temp_dir)
+        plugin_name = metadata.plugin_name
+        if not plugin_name: 
+            remove_dir(temp_dir)
+            raise Exception("插件 metadata.yaml 文件中 name 字段为空。")
+        plugin_name = self.updator.format_name(plugin_name)
+
+        ppath = self.plugin_store_path
+        plugin_path = os.path.join(ppath, plugin_name)
+        if os.path.exists(plugin_path): 
+            remove_dir(plugin_path)
+
+        # move to the target path
+        shutil.move(temp_dir, plugin_path)
+        
+        if metadata.repo:
+            with open(os.path.join(plugin_path, "REPO"), "w", encoding='utf-8') as f:
+                f.write(metadata.repo)
+
+        # remove the temp dir
+        remove_dir(temp_dir)
+        
+        self.check_plugin_dept_update()
+
+        # ok, err = self.plugin_reload()
+        # if not ok:
+        #     raise Exception(err)
+
+    def load_plugin_metadata(self, plugin_path: str, plugin_obj = None) -> PluginMetadata:
+        metadata = None
+        
+        if not os.path.exists(plugin_path):
+            raise Exception("插件不存在。")
+        
+        if os.path.exists(os.path.join(plugin_path, "metadata.yaml")):
+            with open(os.path.join(plugin_path, "metadata.yaml"), "r", encoding='utf-8') as f:
+                metadata = yaml.safe_load(f)
+        elif plugin_obj:
+            # 使用 info() 函数
+            metadata = plugin_obj.info()
+        
+        if isinstance(metadata, dict):
+            if 'name' not in metadata or 'desc' not in metadata or 'version' not in metadata or 'author' not in metadata:
+                raise Exception("插件元数据信息不完整。")
+            metadata = PluginMetadata(
+                plugin_name=metadata['name'],
+                plugin_type=PluginType.COMMON if 'plugin_type' not in metadata else PluginType(metadata['plugin_type']),
+                author=metadata['author'],
+                desc=metadata['desc'],
+                version=metadata['version'],
+                repo=metadata['repo'] if 'repo' in metadata else None
+            )
+            
+        return metadata

model/provider/openai_official.py

@@ -0,0 +1,500 @@
+import os
+import asyncio
+import json
+import time
+import tiktoken
+import threading
+import traceback
+import base64
+
+from openai import AsyncOpenAI
+from openai.types.chat.chat_completion import ChatCompletion
+from openai._exceptions import *
+from util.io import download_image_by_url
+
+from astrbot.db import BaseDatabase
+from model.provider.provider import Provider
+from util.cmd_config import LLMConfig
+from util.log import LogManager
+from logging import Logger
+from typing import List, Dict
+
+from dataclasses import asdict
+
+logger: Logger = LogManager.GetLogger(log_name='astrbot')
+
+MODELS = {
+    "gpt-4o": 128000,
+    "gpt-4o-2024-05-13": 128000,
+    "gpt-4-turbo": 128000,
+    "gpt-4-turbo-2024-04-09": 128000,
+    "gpt-4-turbo-preview": 128000,
+    "gpt-4-0125-preview": 128000,
+    "gpt-4-1106-preview": 128000,
+    "gpt-4-vision-preview": 128000,
+    "gpt-4-1106-vision-preview": 128000,
+    "gpt-4": 8192,
+    "gpt-4-0613": 8192,
+    "gpt-4-32k": 32768,
+    "gpt-4-32k-0613": 32768,
+    "gpt-3.5-turbo-0125": 16385,
+    "gpt-3.5-turbo": 16385,
+    "gpt-3.5-turbo-1106": 16385,
+    "gpt-3.5-turbo-instruct": 4096,
+    "gpt-3.5-turbo-16k": 16385,
+    "gpt-3.5-turbo-0613": 16385,
+    "gpt-3.5-turbo-16k-0613": 16385, 
+}
+
+class ProviderOpenAIOfficial(Provider):
+    def __init__(self, llm_config: LLMConfig, db_helper: BaseDatabase) -> None:
+        super().__init__()
+
+        self.api_keys = []
+        self.chosen_api_key = None
+        self.base_url = None
+        self.llm_config = llm_config
+        self.keys_data = {} # 记录超额
+        if llm_config.key: self.api_keys = llm_config.key
+        if llm_config.api_base: self.base_url = llm_config.api_base
+        if not self.api_keys:
+            logger.warn("看起来你没有添加 OpenAI 的 API 密钥，OpenAI LLM 能力将不会启用。")
+        else:
+            self.chosen_api_key = self.api_keys[0]
+
+        for key in self.api_keys:
+            self.keys_data[key] = True
+
+        self.client = AsyncOpenAI(
+            api_key=self.chosen_api_key,
+            base_url=self.base_url
+        )
+        super().set_curr_model(llm_config.model_config.model)
+        if llm_config.image_generation_model_config:
+            self.image_generator_model_configs: Dict = asdict(llm_config.image_generation_model_config)
+        self.session_memory: Dict[str, List]  = {} # 会话记忆
+        self.session_memory_lock = threading.Lock()
+        self.max_tokens = self.llm_config.model_config.max_tokens # 上下文窗口大小
+        
+        logger.info("正在载入分词器 cl100k_base...")
+        self.tokenizer = tiktoken.get_encoding("cl100k_base") # todo: 根据 model 切换分词器
+        logger.info("分词器载入完成。")
+        
+        self.DEFAULT_PERSONALITY = {
+            "prompt": self.llm_config.default_personality,
+            "name": "default"
+        }
+        self.curr_personality = self.DEFAULT_PERSONALITY
+        self.session_personality = {} # 记录了某个session是否已设置人格。
+        # 读取历史记录
+        self.db_helper = db_helper
+        try:
+            for history in db_helper.get_llm_history():
+                self.session_memory_lock.acquire()
+                self.session_memory[history.session_id] = json.loads(history.content)
+                self.session_memory_lock.release()
+        except BaseException as e:
+            logger.warning(f"读取 OpenAI LLM 对话历史记录 失败：{e}。仍可正常使用。")
+        
+        # 定时保存历史记录
+        threading.Thread(target=self.dump_history, daemon=True).start()
+
+    def dump_history(self):
+        '''转储历史记录'''
+        time.sleep(30)
+        while True:
+            try:
+                for session_id, content in self.session_memory.items():
+                    self.db_helper.update_llm_history(session_id, json.dumps(content))
+            except BaseException as e:
+                logger.error("保存 LLM 历史记录失败: " + str(e))
+            finally:
+                time.sleep(10*60)
+    
+    def personality_set(self, default_personality: dict, session_id: str):
+        if not default_personality: return
+        if session_id not in self.session_memory:
+            self.session_memory[session_id] = []
+        self.curr_personality = default_personality
+        self.session_personality = {} # 重置
+        encoded_prompt = self.tokenizer.encode(default_personality['prompt'])
+        tokens_num = len(encoded_prompt)
+        model = self.get_curr_model()
+        if model in MODELS and tokens_num > MODELS[model] - 500:
+            default_personality['prompt'] = self.tokenizer.decode(encoded_prompt[:MODELS[model] - 500])
+
+        new_record = {
+            "user": {
+                "role": "system",
+                "content": default_personality['prompt'],
+            },
+            'usage_tokens': 0, # 到该条目的总 token 数
+            'single-tokens': 0 # 该条目的 token 数
+        }
+
+        self.session_memory[session_id].append(new_record)
+
+    async def encode_image_bs64(self, image_url: str) -> str:
+        '''
+        将图片转换为 base64
+        '''
+        if image_url.startswith("http"):
+            image_url = await download_image_by_url(image_url)
+        
+        with open(image_url, "rb") as f:
+            image_bs64 = base64.b64encode(f.read()).decode()
+            return "data:image/jpeg;base64," + image_bs64
+
+    async def retrieve_context(self, session_id: str):
+        '''
+        根据 session_id 获取保存的 OpenAI 格式的上下文
+        '''
+        if session_id not in self.session_memory:
+            raise Exception("会话 ID 不存在")
+        
+        # 转换为 openai 要求的格式
+        context = []
+        is_lvm = await self.is_lvm()
+        for record in self.session_memory[session_id]:
+            if "user" in record and record['user']:
+                if not is_lvm and "content" in record['user'] and isinstance(record['user']['content'], list):
+                    logger.warn(f"由于当前模型 {self.get_curr_model()} 不支持视觉，将忽略上下文中的图片输入。如果一直弹出此警告，可以尝试 reset 指令。")
+                    continue
+                context.append(record['user'])
+            if "AI" in record and record['AI']:
+                context.append(record['AI'])
+
+        return context
+
+    async def is_lvm(self):
+        '''
+        是否是 LVM
+        '''
+        return self.get_curr_model().startswith("gpt-4")
+    
+    async def get_models(self):
+        try:
+            models = await self.client.models.list()
+        except NotFoundError as e:
+            bu = str(self.client.base_url)
+            self.client.base_url = bu + "/v1"
+            models = await self.client.models.list()
+        finally:
+            return filter(lambda x: x.id.startswith("gpt"), models.data)
+    
+    async def assemble_context(self, session_id: str, prompt: str, image_url: str = None):
+        '''
+        组装上下文，并且根据当前上下文窗口大小截断
+        '''
+        if session_id not in self.session_memory:
+            raise Exception("会话 ID 不存在")
+        
+        tokens_num = len(self.tokenizer.encode(prompt))
+        previous_total_tokens_num = 0 if not self.session_memory[session_id] else self.session_memory[session_id][-1]['usage_tokens']
+
+        message = {
+            "usage_tokens": previous_total_tokens_num + tokens_num,
+            "single_tokens": tokens_num,
+            "AI": None
+        }
+        if image_url:
+            user_content = {
+                "role": "user",
+                "content": [
+                    {
+                        "type": "text",
+                        "text": prompt
+                    },
+                    {
+                        "type": "image_url",
+                        "image_url": {
+                            "url": await self.encode_image_bs64(image_url)
+                        }
+                    }
+                ]
+            }
+        else:
+            user_content = {
+                "role": "user",
+                "content": prompt
+            }
+
+        message["user"] = user_content
+        self.session_memory[session_id].append(message)
+
+        # 根据 模型的上下文窗口 淘汰掉多余的记录
+        curr_model = self.get_curr_model()
+        if curr_model in MODELS:
+            maxium_tokens_num = MODELS[curr_model] - 300 # 至少预留 300 给 completion
+            # if message['usage_tokens'] > maxium_tokens_num:
+                # 淘汰多余的记录，使得最终的 usage_tokens 不超过 maxium_tokens_num - 300
+                # contexts = self.session_memory[session_id]
+                # need_to_remove_idx = 0
+                # freed_tokens_num = contexts[0]['single-tokens']
+                # while freed_tokens_num < message['usage_tokens'] - maxium_tokens_num:
+                #     need_to_remove_idx += 1
+                #     freed_tokens_num += contexts[need_to_remove_idx]['single-tokens']
+                # # 更新之后的所有记录的 usage_tokens
+                # for i in range(len(contexts)):
+                #     if i > need_to_remove_idx:
+                #         contexts[i]['usage_tokens'] -= freed_tokens_num
+                # logger.debug(f"淘汰上下文记录 {need_to_remove_idx+1} 条，释放 {freed_tokens_num} 个 token。当前上下文总 token 为 {contexts[-1]['usage_tokens']}。")
+                # self.session_memory[session_id] = contexts[need_to_remove_idx+1:]
+            while len(self.session_memory[session_id]) and self.session_memory[session_id][-1]['usage_tokens'] > maxium_tokens_num:
+                self.pop_record(session_id)
+
+
+    async def pop_record(self, session_id: str, pop_system_prompt: bool = False):
+        '''
+        弹出第一条记录
+        '''
+        if session_id not in self.session_memory:
+            raise Exception("会话 ID 不存在")
+        
+        if len(self.session_memory[session_id]) == 0:
+            return None
+        
+        for i in range(len(self.session_memory[session_id])):
+            # 检查是否是 system prompt
+            if not pop_system_prompt and self.session_memory[session_id][i]['user']['role'] == "system":
+                # 如果只有一个 system prompt，才不删掉
+                f = False
+                for j in range(i+1, len(self.session_memory[session_id])):
+                    if self.session_memory[session_id][j]['user']['role'] == "system":
+                        f = True
+                        break
+                if not f:
+                    continue
+            record = self.session_memory[session_id].pop(i)
+            break
+
+        # 更新之后所有记录的 usage_tokens
+        for i in range(len(self.session_memory[session_id])):
+            self.session_memory[session_id][i]['usage_tokens'] -= record['single-tokens']
+        logger.debug(f"淘汰上下文记录 1 条，释放 {record['single-tokens']} 个 token。当前上下文总 token 为 {self.session_memory[session_id][-1]['usage_tokens']}。")
+        return record
+
+    async def text_chat(self, 
+                        prompt: str, 
+                        session_id: str, 
+                        image_url: None=None, 
+                        tools: None=None, 
+                        extra_conf: Dict = None,
+                        **kwargs
+                    ) -> str:
+        if os.environ.get("TEST_LLM", "off") != "on" and os.environ.get("TEST_MODE", "off") == "on":
+            return "这是一个测试消息。"
+        
+        super().accu_model_stat()
+        if not session_id:
+            session_id = "unknown"
+            if "unknown" in self.session_memory:
+                del self.session_memory["unknown"]
+
+        if session_id not in self.session_memory:
+            self.session_memory[session_id] = []
+
+        if session_id not in self.session_personality or not self.session_personality[session_id]:
+            self.personality_set(self.curr_personality, session_id)
+            self.session_personality[session_id] = True
+        
+        # 如果 prompt 超过了最大窗口，截断。
+        # 1. 可以保证之后 pop 的时候不会出现问题
+        # 2. 可以保证不会超过最大 token 数
+        _encoded_prompt = self.tokenizer.encode(prompt)
+        curr_model = self.get_curr_model()
+        if curr_model in MODELS and len(_encoded_prompt) > MODELS[curr_model] - 300:
+            _encoded_prompt = _encoded_prompt[:MODELS[curr_model] - 300]
+            prompt = self.tokenizer.decode(_encoded_prompt)
+        
+        # 组装上下文，并且根据当前上下文窗口大小截断
+        await self.assemble_context(session_id, prompt, image_url)
+
+        # 获取上下文，openai 格式
+        contexts = await self.retrieve_context(session_id)
+
+        conf = asdict(self.llm_config.model_config)
+        if extra_conf: conf.update(extra_conf)
+
+        # start request
+        retry = 0
+        rate_limit_retry = 0
+        while retry < 3 or rate_limit_retry < 5:
+            logger.debug(conf)
+            logger.debug(contexts)
+            if tools:
+                completion_coro = self.client.chat.completions.create(
+                    messages=contexts,
+                    stream=False,
+                    tools=tools,
+                    **conf
+                )
+            else:
+                completion_coro = self.client.chat.completions.create(
+                    messages=contexts,
+                    stream=False,
+                    **conf
+                )
+            try:
+                completion = await completion_coro
+                break
+            except AuthenticationError as e:
+                api_key = self.chosen_api_key[10:] + "..."
+                logger.error(f"OpenAI API Key {api_key} 验证错误。详细原因：{e}。正在切换到下一个可用的 Key（如果有的话）")
+                self.keys_data[self.chosen_api_key] = False
+                ok = await self.switch_to_next_key()
+                if ok: continue
+                else: raise Exception("所有 OpenAI API Key 目前都不可用。")
+            except BadRequestError as e:
+                retry += 1
+                logger.warn(f"OpenAI 请求异常：{e}。")
+                if "image_url is only supported by certain models." in str(e):
+                    raise Exception(f"当前模型 { self.get_curr_model() } 不支持图片输入，请更换模型。")
+            except RateLimitError as e:
+                if "You exceeded your current quota" in str(e):
+                    self.keys_data[self.chosen_api_key] = False
+                    ok = await self.switch_to_next_key()
+                    if ok: continue
+                    else: raise Exception("所有 OpenAI API Key 目前都不可用。")
+                logger.error(f"OpenAI API Key {self.chosen_api_key} 达到请求速率限制或者官方服务器当前超载。详细原因：{e}")
+                await self.switch_to_next_key()
+                rate_limit_retry += 1
+                await asyncio.sleep(1)
+            except NotFoundError as e:
+                raise e
+            except Exception as e:
+                retry += 1
+                if retry >= 3:
+                    logger.error(traceback.format_exc())
+                    raise Exception(f"OpenAI 请求失败：{e}。重试次数已达到上限。")
+                if "maximum context length" in str(e):
+                    logger.warn(f"OpenAI 请求失败：{e}。上下文长度超过限制。尝试弹出最早的记录然后重试。")
+                    self.pop_record(session_id)
+                
+                logger.warning(traceback.format_exc())
+                logger.warning(f"OpenAI 请求失败：{e}。重试第 {retry} 次。")
+                await asyncio.sleep(1)
+
+        assert isinstance(completion, ChatCompletion)
+        logger.debug(f"openai completion: {completion.usage}")
+        
+        if len(completion.choices) == 0:
+            raise Exception("OpenAI API 返回的 completion 为空。")
+        choice = completion.choices[0]
+
+        usage_tokens = completion.usage.total_tokens
+        completion_tokens = completion.usage.completion_tokens
+        self.session_memory[session_id][-1]['usage_tokens'] = usage_tokens
+        self.session_memory[session_id][-1]['single_tokens'] += completion_tokens
+
+        if choice.message.content:
+            # 返回文本
+            completion_text = str(choice.message.content).strip()
+        elif choice.message.tool_calls and choice.message.tool_calls:
+            # tools call (function calling)
+            return choice.message.tool_calls[0].function
+        
+        self.session_memory[session_id][-1]['AI'] = {
+            "role": "assistant",
+            "content": completion_text
+        }
+
+        return completion_text
+    
+    async def switch_to_next_key(self):
+        '''
+        切换到下一个 API Key
+        '''
+        if not self.api_keys:
+            logger.error("OpenAI API Key 不存在。")
+            return False
+
+        for key in self.keys_data:
+            if self.keys_data[key]:
+                # 没超额
+                self.chosen_api_key = key
+                self.client.api_key = key
+                logger.info(f"OpenAI 切换到 API Key {key[:10]}... 成功。")
+                return True
+
+        return False
+    
+    async def image_generate(self, prompt: str, session_id: str = None, **kwargs) -> str:
+        '''
+        生成图片
+        '''
+        retry = 0
+        conf = self.image_generator_model_configs
+        if not conf:
+            logger.error("OpenAI 图片生成模型配置不存在。")
+            raise Exception("OpenAI 图片生成模型配置不存在。")
+        super().accu_model_stat(model=conf['model'])
+        while retry < 3:
+            try:
+                images_response = await self.client.images.generate(
+                    prompt=prompt,
+                    **conf
+                )
+                image_url = images_response.data[0].url
+                return image_url
+            except Exception as e:
+                retry += 1
+                if retry >= 3:
+                    logger.error(traceback.format_exc())
+                    raise Exception(f"OpenAI 图片生成请求失败：{e}。重试次数已达到上限。")
+                logger.warning(f"OpenAI 图片生成请求失败：{e}。重试第 {retry} 次。")
+                await asyncio.sleep(1)
+
+    async def forget(self, session_id=None, keep_system_prompt: bool=False) -> bool:
+        if session_id is None: return False
+        self.session_memory[session_id] = []
+        if keep_system_prompt:
+            self.personality_set(self.curr_personality, session_id)
+        else:
+            self.curr_personality = self.DEFAULT_PERSONALITY
+        return True
+    
+    def dump_contexts_page(self, session_id: str, size=5, page=1,):
+        '''
+        获取缓存的会话
+        '''
+        # contexts_str = ""
+        # for i, key in enumerate(self.session_memory):
+        #     if i < (page-1)*size or i >= page*size:
+        #         continue
+        #     contexts_str += f"Session ID: {key}\n"
+        #     for record in self.session_memory[key]:
+        #         if "user" in record:
+        #             contexts_str += f"User: {record['user']['content']}\n"
+        #         if "AI" in record:
+        #             contexts_str += f"AI: {record['AI']['content']}\n"
+        #     contexts_str += "---\n"
+        contexts_str = ""
+        if session_id in self.session_memory:
+            for record in self.session_memory[session_id]:
+                if "user" in record and record['user']:
+                    text = record['user']['content'][:100] + "..." if len(record['user']['content']) > 100 else record['user']['content']
+                    contexts_str += f"User: {text}\n"
+                if "AI" in record and record['AI']:
+                    text = record['AI']['content'][:100] + "..." if len(record['AI']['content']) > 100 else record['AI']['content']
+                    contexts_str += f"Assistant: {text}\n"
+        else:
+            contexts_str = "会话 ID 不存在。"
+
+        return contexts_str, len(self.session_memory[session_id])
+
+    def set_model(self, model: str):
+        # TODO: 更新配置文件
+        super().set_curr_model(model)
+    
+    def get_configs(self):
+        return asdict(self.llm_config)
+
+    def get_keys_data(self):
+        return self.keys_data
+
+    def get_curr_key(self):
+        return self.chosen_api_key
+
+    def set_key(self, key):
+        self.client.api_key = key

model/provider/provider.py

@@ -0,0 +1,58 @@
+from collections import defaultdict
+
+class Provider:
+    def __init__(self) -> None:
+        self.model_stat = defaultdict(int) # 用于记录 LLM Model 使用数据
+        self.curr_model_name = "unknown"
+        
+    def reset_model_stat(self):
+        self.model_stat.clear()
+        
+    def set_curr_model(self, model_name: str):
+        self.curr_model_name = model_name
+    
+    def get_curr_model(self):
+        '''
+        返回当前正在使用的 LLM
+        '''
+        return self.curr_model_name
+    
+    def accu_model_stat(self, model: str = None):
+        if not model:
+            model = self.get_curr_model()
+        self.model_stat[model] += 1
+        
+    async def text_chat(self,
+                        prompt: str,
+                        session_id: str,
+                        image_url: None = None,
+                        tools: None = None,
+                        extra_conf: dict = None,
+                        default_personality: dict = None,
+                        **kwargs) -> str:
+        '''
+        [require]
+        prompt: 提示词
+        session_id: 会话id
+
+        [optional]
+        image_url: 图片url（识图）
+        tools: 函数调用工具
+        extra_conf: 额外配置
+        default_personality: 默认人格
+        '''
+        raise NotImplementedError()
+
+    async def image_generate(self, prompt, session_id, **kwargs) -> str:
+        '''
+        [require]
+        prompt: 提示词
+        session_id: 会话id
+        '''
+        raise NotImplementedError()
+
+    async def forget(self, session_id=None) -> bool:
+        '''
+        重置会话
+        '''
+        raise NotImplementedError()

requirements.txt

@@ -1,3 +1,16 @@
-requests~=2.27.1
-openai~=0.27.0
-qq-botpy~=1.1.2
+pydantic~=1.10.4
+aiohttp
+openai
+qq-botpy
+chardet~=5.1.0
+Pillow
+nakuru-project
+beautifulsoup4
+googlesearch-python
+tiktoken
+readability-lxml
+quart
+psutil
+lxml_html_clean
+colorlog
+aiocqhttp

tests/mocks/onebot.py

@@ -0,0 +1,18 @@
+import copy
+from aiocqhttp import Event
+
+class MockOneBotMessage():
+    def __init__(self):
+        # 这些数据不是敏感的
+        self.group_event_sample = Event.from_payload({'self_id': 3430871669, 'user_id': 905617992, 'time': 1723882500, 'message_id': -2147480159, 'message_seq': -2147480159, 'real_id': -2147480159, 'message_type': 'group', 'sender': {'user_id': 905617992, 'nickname': 'Soulter', 'card': '', 'role': 'owner'}, 'raw_message': '[CQ:at,qq=3430871669] just reply me `ok`', 'font': 14, 'sub_type': 'normal', 'message': [{'data': {'qq': '3430871669'}, 'type': 'at'}, {'data': {'text': ' just reply me `ok`'}, 'type': 'text'}], 'message_format': 'array', 'post_type': 'message', 'group_id': 849750470})
+        self.friend_event_sample = Event.from_payload({'self_id': 3430871669, 'user_id': 905617992, 'time': 1723882599, 'message_id': -2147480157, 'message_seq': -2147480157, 'real_id': -2147480157, 'message_type': 'private', 'sender': {'user_id': 905617992, 'nickname': 'Soulter', 'card': ''}, 'raw_message': 'just reply me `ok`', 'font': 14, 'sub_type': 'friend', 'message': [{'data': {'text': 'just reply me `ok`'}, 'type': 'text'}], 'message_format': 'array', 'post_type': 'message'})
+
+    def create_random_group_message(self):
+        return self.group_event_sample
+        
+    def create_random_direct_message(self):
+        return self.friend_event_sample
+    
+    def create_msg(self, text: str):
+        self.group_event_sample.message = [{'data': {'qq': '3430871669'}, 'type': 'at'}, {'data': {'text': text}, 'type': 'text'}]
+        return self.group_event_sample

tests/mocks/qq_official.py

@@ -0,0 +1,54 @@
+import botpy.message
+
+class MockQQOfficialMessage():
+    def __init__(self):
+        # 这些数据已经经过去敏处理
+        self.group_plain_text_sample = {'author': {'id': '3E47ABD92415AFEF02DAD74FFAB592D1', 'member_openid': '3E47ABD92415AFEF02DAD74FFAB592D1'}, 'content': 'just reply me `ok`', 'group_id': 'BF5D5CA67932FFC4AFD18D4309DB759D', 'group_openid': 'BF5D5CA67932FFC4AFD18D4309DB759D', 'id': 'ROBOT1.0_test', 'timestamp': '2024-07-27T19:58:52+08:00'}
+        self.group_plain_image_sample = {'attachments': [{'content_type': 'image/png', 'filename': '165FCBF8BD6F42496B58A6C66C5D4255.png', 'height': 1034, 'size': 1440173, 'url': 'https://multimedia.nt.qq.com.cn/download?appid=1407&fileid=Cgk5MDU2MTc5OTISFBvbdDR6nYEHsqWEfYauN9wphLxlGK3zVyD_Cii9ibiql8eHA1CAvaMB&rkey=CAESKE4_cASDm1t162vI7q9gitU2u0SUciVRg1fbyn3zYe9f_XHL2vhiB0s&spec=0', 'width': 1186}], 'author': {'id': '3E47ABD92415AFEF02DAD74FFAB592D1', 'member_openid': '3E47ABD92415AFEF02DAD74FFAB592D1'}, 'content': ' ', 'group_id': 'BF5D5CA67932FFC4AFD18D4309DB759D', 'group_openid': 'BF5D5CA67932FFC4AFD18D4309DB759D', 'id': 'ROBOT1.0_test', 'timestamp': '2024-07-27T20:06:32+08:00'}
+        self.group_multimedia_sample = {'attachments': [{'content_type': 'image/png', 'filename': '165FCBF8BD6F42496B58A6C66C5D4255.png', 'height': 1034, 'size': 1440173, 'url': 'https://multimedia.nt.qq.com.cn/download?appid=1407&fileid=Cgk5MDU2MTc5OTISFBvbdDR6nYEHsqWEfYauN9wphLxlGK3zVyD_CiiMytyomceHA1CAvaMB&rkey=CAQSKDOc_jvbthUjVk7zSzPCqflD2XWA0OWzO5qCNsiRFY4RfQMuHYt8KDU&spec=0', 'width': 1186}], 'author': {'id': '3E47ABD92415AFEF02DAD74FFAB592D1', 'member_openid': '3E47ABD92415AFEF02DAD74FFAB592D1'}, 'content': " What's this", 'group_id': 'BF5D5CA67932FFC4AFD18D4309DB759D', 'group_openid': 'BF5D5CA67932FFC4AFD18D4309DB759D', 'id': 'ROBOT1.0_test', 'timestamp': '2024-07-27T20:15:24+08:00'}
+        self.group_event_id_sample = "GROUP_AT_MESSAGE_CREATE:ss6hqvpgtqv99eglilbjpsdzvudsjev64th8srgofxqkgxwpynhysl6q6ws849"
+        
+        self.guild_plain_text_sample = {'author': {'avatar': 'https://qqchannel-profile-1251316161.file.myqcloud.com/168087977775f0eae70da8e512?t=1680879777', 'bot': False, 'id': '6946931796791550499', 'username': 'Soulter'}, 'channel_id': '9941389', 'content': '<@!2519660939131724751> just reply me `ok`', 'guild_id': '7969749791337194879', 'id': '08ffca96ebdaa68fcd6e108de3de0438ef0e48a6c793b506', 'member': {'joined_at': '2022-08-13T13:13:56+08:00', 'nick': 'Soulter', 'roles': ['4', '23']}, 'mentions': [{'avatar': 'http://thirdqq.qlogo.cn/g?b=oidb&k=OUbv2LTECcjQt48ibDS4OcA&kti=ZqTjpgAAAAI&s=0&t=1708501824', 'bot': True, 'id': '2519660939131724751', 'username': '浅橙Bot'}], 'seq': 1903, 'seq_in_channel': '1903', 'timestamp': '2024-07-27T20:10:14+08:00'}
+        self.guild_plain_image_sample = {'attachments': [{'content_type': 'image/png', 'filename': '165FCBF8BD6F42496B58A6C66C5D4255.png', 'height': 1034, 'id': '2665728996', 'size': 1440173, 'url': 'gchat.qpic.cn/qmeetpic/75802001660367636/9941389-2665728996-165FCBF8BD6F42496B58A6C66C5D4255/0', 'width': 1186}], 'author': {'avatar': 'https://qqchannel-profile-1251316161.file.myqcloud.com/168087977775f0eae70da8e512?t=1680879777', 'bot': False, 'id': '6946931796791550499', 'username': 'Soulter'}, 'channel_id': '9941389', 'content': '<@!2519660939131724751> ', 'guild_id': '7969749791337194879', 'id': 'testid', 'member': {'joined_at': '2022-08-13T13:13:56+08:00', 'nick': 'Soulter', 'roles': ['4', '23']}, 'mentions': [{'avatar': 'http://thirdqq.qlogo.cn/g?b=oidb&k=mZ2Hn0BN5MLlBJTve0WIoA&kti=ZqTjnwAAAAA&s=0&t=1708501824', 'bot': True, 'id': '2519660939131724751', 'username': '浅橙Bot'}], 'seq': 1905, 'seq_in_channel': '1905', 'timestamp': '2024-07-27T20:11:07+08:00'}
+        self.guild_multimedia_sample = {'attachments': [{'content_type': 'image/png', 'filename': '165FCBF8BD6F42496B58A6C66C5D4255.png', 'height': 1034, 'id': '2501183002', 'size': 1440173, 'url': 'gchat.qpic.cn/qmeetpic/75802001660367636/9941389-2501183002-165FCBF8BD6F42496B58A6C66C5D4255/0', 'width': 1186}], 'author': {'avatar': 'https://qqchannel-profile-1251316161.file.myqcloud.com/168087977775f0eae70da8e512?t=1680879777', 'bot': False, 'id': '6946931796791550499', 'username': 'Soulter'}, 'channel_id': '9941389', 'content': "<@!2519660939131724751> What's this", 'guild_id': '7969749791337194879', 'id': 'testid', 'member': {'joined_at': '2022-08-13T13:13:56+08:00', 'nick': 'Soulter', 'roles': ['4', '23']}, 'mentions': [{'avatar': 'http://thirdqq.qlogo.cn/g?b=oidb&k=mZ2Hn0BN5MLlBJTve0WIoA&kti=ZqTjnwAAAAA&s=0&t=1708501824', 'bot': True, 'id': '2519660939131724751', 'username': '浅橙Bot'}], 'seq': 1907, 'seq_in_channel': '1907', 'timestamp': '2024-07-27T20:14:26+08:00'}
+        self.guild_event_id_sample = "AT_MESSAGE_CREATE:e4c09708-781d-44d0-b8cf-34bf3d4e2e64"
+        
+        self.direct_plain_text_sample = {'author': {'avatar': 'https://qqchannel-profile-1251316161.file.myqcloud.com/168087977775f0eae70da8e512?t=1680879777', 'id': '6946931796791550499', 'username': 'Soulter'}, 'channel_id': '33342831678707631', 'content': 'just reply me `ok`', 'direct_message': True, 'guild_id': '3398240095091349322', 'id': '08caaea38bcaabbe942f10afaf8fb08fa49d3b38a5014898c893b506', 'member': {'joined_at': '2023-03-13T19:40:31+08:00'}, 'seq': 165, 'seq_in_channel': '165', 'src_guild_id': '7969749791337194879', 'timestamp': '2024-07-27T20:12:08+08:00'}
+        self.direct_plain_image_sample = {'attachments': [{'content_type': 'image/png', 'filename': '165FCBF8BD6F42496B58A6C66C5D4255.png', 'height': 1034, 'id': '2658044992', 'size': 1440173, 'url': 'gchat.qpic.cn/qmeetpic/92265551678707631/33342831678707631-2658044992-165FCBF8BD6F42496B58A6C66C5D4255/0', 'width': 1186}], 'author': {'avatar': 'https://qqchannel-profile-1251316161.file.myqcloud.com/168087977775f0eae70da8e512?t=1680879777', 'id': '6946931796791550499', 'username': 'Soulter'}, 'channel_id': '33342831678707631', 'direct_message': True, 'guild_id': '3398240095091349322', 'id': 'testid', 'member': {'joined_at': '2023-03-13T19:40:31+08:00'}, 'seq': 167, 'seq_in_channel': '167', 'src_guild_id': '7969749791337194879', 'timestamp': '2024-07-27T20:12:29+08:00'}
+        self.direct_multimedia_sample = {'attachments': [{'content_type': 'image/png', 'filename': '165FCBF8BD6F42496B58A6C66C5D4255.png', 'height': 1034, 'id': '2526212938', 'size': 1440173, 'url': 'gchat.qpic.cn/qmeetpic/92265551678707631/33342831678707631-2526212938-165FCBF8BD6F42496B58A6C66C5D4255/0', 'width': 1186}], 'author': {'avatar': 'https://qqchannel-profile-1251316161.file.myqcloud.com/168087977775f0eae70da8e512?t=1680879777', 'id': '6946931796791550499', 'username': 'Soulter'}, 'channel_id': '33342831678707631', 'content': "What's this", 'direct_message': True, 'guild_id': '3398240095091349322', 'id': 'testid', 'member': {'joined_at': '2023-03-13T19:40:31+08:00'}, 'seq': 168, 'seq_in_channel': '168', 'src_guild_id': '7969749791337194879', 'timestamp': '2024-07-27T20:13:38+08:00'}
+        self.direct_event_id_sample = "DIRECT_MESSAGE_CREATE:e4c09708-781d-44d0-b8cf-34bf3d4e2e64"
+        
+    def create_random_group_message(self):
+        mocked = botpy.message.GroupMessage(
+            api=None,
+            event_id=self.group_event_id_sample,
+            data=self.group_plain_text_sample
+        )
+        return mocked
+        
+    def create_random_guild_message(self):
+        mocked = botpy.message.Message(
+            api=None,
+            event_id=self.guild_event_id_sample,
+            data=self.guild_plain_text_sample
+        )
+        return mocked
+        
+    def create_random_direct_message(self):
+        mocked = botpy.message.DirectMessage(
+            api=None,
+            event_id=self.direct_event_id_sample,
+            data=self.direct_plain_text_sample
+        )
+        return mocked
+    
+    def create_msg(self, text: str):
+        sample = self.group_plain_text_sample.copy()
+        sample['content'] = text
+        mocked = botpy.message.Message(
+            api=None,
+            event_id=self.group_event_id_sample,
+            data=sample
+        )
+        return mocked
+    

tests/test_http_server.py

@@ -0,0 +1,51 @@
+
+import aiohttp
+import pytest
+
+BASE_URL = "http://0.0.0.0:6185/api"
+
+async def get_url(url):
+    async with aiohttp.ClientSession() as session:
+        async with session.get(url) as response:
+            return await response.json()
+        
+async def post_url(url, data):
+    async with aiohttp.ClientSession() as session:
+        async with session.post(url, json=data) as response:
+            return await response.json()
+
+class TestHTTPServer:
+    @pytest.mark.asyncio
+    async def test_config(self):
+        configs = await get_url(f"{BASE_URL}/configs")
+        assert 'data' in configs and 'metadata' in configs['data'] \
+            and 'config' in configs['data']
+        config = configs['data']['config']
+        # test post config
+        await post_url(f"{BASE_URL}/astrbot-configs", config)
+        # text post config with invalid data
+        assert 'rate_limit' in config['platform_settings']
+        config['platform_settings']['rate_limit'] = "invalid"
+        ret = await post_url(f"{BASE_URL}/astrbot-configs", config)
+        assert 'status' in ret and ret['status'] == 'error'
+    
+    @pytest.mark.asyncio
+    async def test_update(self):
+        await get_url(f"{BASE_URL}/check_update")
+    
+    @pytest.mark.asyncio
+    async def test_plugins(self):
+        pname = "astrbot_plugin_bilibili"
+        url = f"https://github.com/Soulter/{pname}"
+
+        await get_url(f"{BASE_URL}/extensions")
+
+        # test install plugin
+        await post_url(f"{BASE_URL}/extensions/install", {
+            "url": url
+        })
+        
+        # test uninstall plugin
+        await post_url(f"{BASE_URL}/extensions/uninstall", {
+            "name": pname
+        })

tests/test_message.py

@@ -0,0 +1,155 @@
+import asyncio
+import pytest
+import os
+
+from tests.mocks.qq_official import MockQQOfficialMessage
+from tests.mocks.onebot import MockOneBotMessage
+
+from astrbot.bootstrap import AstrBotBootstrap
+from model.platform.qq_official import QQOfficial
+from model.platform.qq_aiocqhttp import AIOCQHTTP
+from model.provider.openai_official import ProviderOpenAIOfficial
+from type.astrbot_message import *
+from type.message_event import *
+from util.log import LogManager
+
+from util.cmd_config import QQOfficialPlatformConfig, AiocqhttpPlatformConfig
+
+logger = LogManager.GetLogger(log_name='astrbot')
+pytest_plugins = ('pytest_asyncio',)
+
+os.environ['TEST_MODE'] = 'on'
+bootstrap = AstrBotBootstrap()
+
+llm_config = bootstrap.context.config_helper.llm[0]
+llm_config.api_base = os.environ['OPENAI_API_BASE']
+llm_config.key = [os.environ['OPENAI_API_KEY']]
+llm_config.model_config.model = os.environ['LLM_MODEL']
+llm_config.model_config.max_tokens = 1000
+asyncio.run(bootstrap.run())
+llm_provider = ProviderOpenAIOfficial(llm_config, bootstrap.db_helper)
+bootstrap.message_handler.provider = llm_provider
+bootstrap.config_helper.wake_prefix = ["/"]
+bootstrap.config_helper.admins_id = ["905617992"]
+
+for p_config in bootstrap.context.config_helper.platform:
+    if isinstance(p_config, QQOfficialPlatformConfig):
+        qq_official = QQOfficial(bootstrap.context, bootstrap.message_handler, p_config)
+    elif isinstance(p_config, AiocqhttpPlatformConfig):
+        aiocqhttp = AIOCQHTTP(bootstrap.context, bootstrap.message_handler, p_config)
+
+class TestBasicMessageHandle():
+    @pytest.mark.asyncio
+    async def test_qqofficial_group_message(self):
+        group_message = MockQQOfficialMessage().create_random_group_message()
+        abm = qq_official._parse_from_qqofficial(group_message, MessageType.GROUP_MESSAGE)
+        ret = await qq_official.handle_msg(abm)
+        print(ret)
+        
+    @pytest.mark.asyncio
+    async def test_qqofficial_guild_message(self):
+        guild_message = MockQQOfficialMessage().create_random_guild_message()
+        abm = qq_official._parse_from_qqofficial(guild_message, MessageType.GUILD_MESSAGE)
+        ret = await qq_official.handle_msg(abm)
+        print(ret)
+
+    # 有共同性，为了节约开销，不测试频道私聊。
+    # @pytest.mark.asyncio
+    # async def test_qqofficial_private_message(self):
+    #     private_message = MockQQOfficialMessage().create_random_direct_message()
+    #     abm = qq_official._parse_from_qqofficial(private_message, MessageType.FRIEND_MESSAGE)
+    #     ret = await qq_official.handle_msg(abm)
+    #     print(ret)
+    
+    @pytest.mark.asyncio
+    async def test_aiocqhttp_group_message(self):
+        event = MockOneBotMessage().create_random_group_message()
+        abm = aiocqhttp.convert_message(event)
+        ret = await aiocqhttp.handle_msg(abm)
+        print(ret)
+
+    @pytest.mark.asyncio
+    async def test_aiocqhttp_direct_message(self):
+        event = MockOneBotMessage().create_random_direct_message()
+        abm = aiocqhttp.convert_message(event)
+        ret = await aiocqhttp.handle_msg(abm)
+        print(ret)
+        
+class TestInteralCommandHsandle():
+    def create(self, text: str):
+        event = MockOneBotMessage().create_msg(text)
+        abm = aiocqhttp.convert_message(event)
+        return abm
+    
+    async def fast_test(self, text: str):
+        abm = self.create(text)
+        ret = await aiocqhttp.handle_msg(abm)
+        print(f"Command: {text}, Result: {ret.result_message}")
+        return ret
+    
+    @pytest.mark.asyncio
+    async def test_config_save(self):
+        abm = self.create("/websearch on")
+        ret = await aiocqhttp.handle_msg(abm)
+        assert bootstrap.context.config_helper.llm_settings.web_search \
+            == bootstrap.config_helper.get("llm_settings")['web_search']
+    
+    @pytest.mark.asyncio
+    async def test_websearch(self):
+        await self.fast_test("/websearch")
+        await self.fast_test("/websearch on")
+        await self.fast_test("/websearch off")
+    
+    @pytest.mark.asyncio
+    async def test_help(self):
+        await self.fast_test("/help")
+        
+    @pytest.mark.asyncio
+    async def test_myid(self):
+        await self.fast_test("/myid")
+        
+    @pytest.mark.asyncio
+    async def test_wake(self):
+        await self.fast_test("/wake")
+        await self.fast_test("/wake #")
+        assert "#" in bootstrap.context.config_helper.wake_prefix 
+        assert "#" in bootstrap.context.config_helper.get("wake_prefix")
+        await self.fast_test("#wake /")
+        
+    @pytest.mark.asyncio
+    async def test_sleep(self):
+        await self.fast_test("/provider")
+    
+    @pytest.mark.asyncio
+    async def test_update(self):
+        await self.fast_test("/update")
+    
+    @pytest.mark.asyncio
+    async def test_t2i(self):
+        if not bootstrap.context.config_helper.t2i:
+            abm = self.create("/t2i")
+            await aiocqhttp.handle_msg(abm)
+        await self.fast_test("/help")
+
+    @pytest.mark.asyncio
+    async def test_plugin(self):
+        pname = "astrbot_plugin_bilibili"
+        url = f"https://github.com/Soulter/{pname}"
+        await self.fast_test("/plugin")
+        await self.fast_test(f"/plugin l")
+        await self.fast_test(f"/plugin i {url}")
+        await self.fast_test(f"/plugin u {url}")
+        await self.fast_test(f"/plugin d {pname}")
+
+class TestLLMChat():
+    @pytest.mark.asyncio
+    async def test_llm_chat(self):
+        os.environ["TEST_LLM"] = "on"
+        ret = await llm_provider.text_chat("Just reply `ok`", "test")
+        print(ret)
+        event = MockOneBotMessage().create_msg("Just reply `ok`")
+        abm = aiocqhttp.convert_message(event)
+        ret = await aiocqhttp.handle_msg(abm)
+        print(ret)
+        os.environ["TEST_LLM"] = "off"
+        

type/astrbot_message.py

@@ -0,0 +1,37 @@
+import time
+from enum import Enum
+from typing import List
+from dataclasses import dataclass
+from nakuru.entities.components import BaseMessageComponent
+
+class MessageType(Enum):
+    GROUP_MESSAGE = 'GroupMessage'  # 群组形式的消息
+    FRIEND_MESSAGE = 'FriendMessage'  # 私聊、好友等单聊消息
+    GUILD_MESSAGE = 'GuildMessage'  # 频道消息
+
+@dataclass
+class MessageMember():
+    user_id: str  # 发送者id
+    nickname: str = None
+
+
+class AstrBotMessage():
+    '''
+    AstrBot 的消息对象
+    '''
+    tag: str  # 消息来源标签
+    type: MessageType  # 消息类型
+    self_id: str  # 机器人的识别id
+    session_id: str  # 会话id
+    message_id: str  # 消息id
+    sender: MessageMember  # 发送者
+    message: List[BaseMessageComponent]  # 消息链使用 Nakuru 的消息链格式
+    message_str: str  # 最直观的纯文本消息字符串
+    raw_message: object
+    timestamp: int  # 消息时间戳
+    
+    def __init__(self) -> None:
+        self.timestamp = int(time.time())
+
+    def __str__(self) -> str:
+        return str(self.__dict__)

type/cached_queue.py

@@ -0,0 +1,28 @@
+from asyncio import Queue
+from collections import deque
+from typing import Deque
+
+class CachedQueue(Queue):
+    def __init__(self, maxsize: int = 0, cachesize: int = 200):
+        super().__init__(maxsize)
+        self.cache = deque(maxlen=cachesize)
+
+    def put_nowait(self, item):
+        self.cache.append(item)
+        super().put_nowait(item)
+
+    def get_nowait(self):
+        item = super().get_nowait()
+        return item
+
+    def get(self):
+        item = super().get()
+        return item
+
+    def clear(self):
+        self.cache.clear()
+        with self.mutex:
+            self._queue.clear()
+
+    def get_cache(self) -> Deque:
+        return self.cache

type/command.py

@@ -0,0 +1,78 @@
+from typing import Union, List, Callable
+from dataclasses import dataclass
+from nakuru.entities.components import Plain, Image
+
+@dataclass
+class CommandItem():
+    '''
+    用来描述单个指令
+    '''
+
+    command_name: Union[str, tuple]  # 指令名
+    callback: Callable  # 回调函数
+    description: str  # 描述
+    origin: str  # 注册来源
+
+class CommandResult():
+    '''
+    用于在Command中返回多个值
+    '''
+
+    def __init__(self, 
+                 hit: bool = True, 
+                 success: bool = True, 
+                 message_chain: list = [], 
+                 command_name: str = "unknown_command",
+                 use_t2i: bool = None) -> None:
+        self.hit = hit
+        self.success = success
+        self.message_chain = message_chain
+        self.command_name = command_name
+        self.is_use_t2i = use_t2i
+        
+    def message(self, message: str):
+        '''
+        快捷回复消息。
+        
+        CommandResult().message("Hello, world!")
+        '''
+        self.message_chain = [Plain(message), ]
+        return self
+    
+    def error(self, message: str):
+        '''
+        快捷回复消息。
+        
+        CommandResult().error("Hello, world!")
+        '''
+        self.success = False
+        self.message_chain = [Plain(message), ]
+        return self
+    
+    def url_image(self, url: str):
+        '''
+        快捷回复图片(网络url的格式)。
+        
+        CommandResult().image("https://example.com/image.jpg")
+        '''
+        self.message_chain = [Image.fromURL(url), ]
+        return self
+    
+    def file_image(self, path: str):
+        '''
+        快捷回复图片(本地文件路径的格式)。
+        
+        CommandResult().image("image.jpg")
+        '''
+        self.message_chain = [Image.fromFileSystem(path), ]
+        return self
+    
+    def use_t2i(self, use_t2i: bool):
+        '''
+        设置是否使用文本转图片服务。如果不设置，则跟随用户的设置。
+        '''
+        self.is_use_t2i = use_t2i
+        return self
+
+    def _result_tuple(self):
+        return (self.success, self.message_chain, self.command_name)

type/config.py

@@ -0,0 +1,287 @@
+VERSION = '3.3.17'
+DB_PATH = 'data/data_v2.db'
+
+# 新版本配置文件，摈弃旧版本令人困惑的配置项 :D
+DEFAULT_CONFIG_VERSION_2 = {
+    "config_version": 2,
+    "platform": [
+        {
+            "id": "default",
+            "name": "qq_official",
+            "enable": False,
+            "appid": "",
+            "secret": "",
+            "enable_group_c2c": True,
+            "enable_guild_direct_message": True,
+        },
+        {
+            "id": "default",
+            "name": "nakuru",
+            "enable": False,
+            "host": "172.0.0.1",
+            "port": 5700,
+            "websocket_port": 6700,
+            "enable_group": True,
+            "enable_guild": True,
+            "enable_direct_message": True,
+            "enable_group_increase": True,
+        },
+        {
+            "id": "default",
+            "name": "aiocqhttp",
+            "enable": False,
+            "ws_reverse_host": "",
+            "ws_reverse_port": 6199,
+        }
+    ],
+    "platform_settings": {
+        "unique_session": False,
+        "rate_limit": {
+            "time": 60,
+            "count": 30,
+        },
+        "reply_prefix": "",
+        "forward_threshold": 200, # 转发消息的阈值
+    },
+    "llm": [
+        {
+            "id": "default",
+            "name": "openai",
+            "enable": True,
+            "key": [],
+            "api_base": "",
+            "prompt_prefix": "",
+            "default_personality": "",
+            "model_config": {
+                "model": "gpt-4o",
+                "max_tokens": 6000,
+                "temperature": 0.9,
+                "top_p": 1,
+                "frequency_penalty": 0,
+                "presence_penalty": 0,
+            },
+            "image_generation_model_config": {
+                "enable": True,
+                "model": "dall-e-3",
+                "size": "1024x1024",
+                "style": "vivid",
+                "quality": "standard",
+            }
+        },
+    ],
+    "llm_settings": {
+        "wake_prefix": "",
+        "web_search": False,
+        "identifier": False,
+    },
+    "content_safety": {
+        "baidu_aip": {
+            "enable": False,
+            "app_id": "",
+            "api_key": "",
+            "secret_key": "",
+        },
+        "internal_keywords": {
+            "enable": True,
+            "extra_keywords": [],
+        }
+    },
+    "wake_prefix": ["/"],
+    "t2i": True,
+    "dump_history_interval": 10,
+    "admins_id": [],
+    "https_proxy": "",
+    "http_proxy": "",
+    "dashboard": {
+        "enable": True,
+        "username": "",
+        "password": "",
+    },
+    "log_level": "INFO",
+    "t2i_endpoint": "",
+}
+
+# 这个是用于迁移旧版本配置文件的映射表
+MAPPINGS_1_2 = [
+    [["qqbot", "enable"], ["platform", 0, "enable"]],
+    [["qqbot", "appid"], ["platform", 0, "appid"]],
+    [["qqbot", "token"], ["platform", 0, "secret"]],
+    [["qqofficial_enable_group_message"], ["platform", 0, "enable_group_c2c"]],
+    [["direct_message_mode"], ["platform", 0, "enable_guild_direct_message"]],
+    [["gocqbot", "enable"], ["platform", 1, "enable"]],
+    [["gocq_host"], ["platform", 1, "host"]],
+    [["gocq_http_port"], ["platform", 1, "port"]],
+    [["gocq_websocket_port"], ["platform", 1, "websocket_port"]],
+    [["gocq_react_group"], ["platform", 1, "enable_group"]],
+    [["gocq_react_guild"], ["platform", 1, "enable_guild"]],
+    [["gocq_react_friend"], ["platform", 1, "enable_direct_message"]],
+    [["gocq_react_group_increase"], ["platform", 1, "enable_group_increase"]],
+    [["aiocqhttp", "enable"], ["platform", 2, "enable"]],
+    [["aiocqhttp", "ws_reverse_host"], ["platform", 2, "ws_reverse_host"]],
+    [["aiocqhttp", "ws_reverse_port"], ["platform", 2, "ws_reverse_port"]],
+    [["uniqueSessionMode"], ["platform_settings", "unique_session"]],
+    [["limit", "time"], ["platform_settings", "rate_limit", "time"]],
+    [["limit", "count"], ["platform_settings", "rate_limit", "count"]],
+    [["reply_prefix"], ["platform_settings", "reply_prefix"]],
+    [["qq_forward_threshold"], ["platform_settings", "forward_threshold"]],
+    
+    [["openai", "key"], ["llm", 0, "key"]],
+    [["openai", "api_base"], ["llm", 0, "api_base"]],
+    [["openai", "chatGPTConfigs", "model"], ["llm", 0, "model_config", "model"]],
+    [["openai", "chatGPTConfigs", "max_tokens"], ["llm", 0, "model_config", "max_tokens"]],
+    [["openai", "chatGPTConfigs", "temperature"], ["llm", 0, "model_config", "temperature"]],
+    [["openai", "chatGPTConfigs", "top_p"], ["llm", 0, "model_config", "top_p"]],
+    [["openai", "chatGPTConfigs", "frequency_penalty"], ["llm", 0, "model_config", "frequency_penalty"]],
+    [["openai", "chatGPTConfigs", "presence_penalty"], ["llm", 0, "model_config", "presence_penalty"]],    
+    
+    [["default_personality_str"], ["llm", 0, "default_personality"]],
+    [["llm_env_prompt"], ["llm", 0, "prompt_prefix"]],
+    [["openai_image_generate", "model"], ["llm", 0, "image_generation_model_config", "model"]],
+    [["openai_image_generate", "size"], ["llm", 0, "image_generation_model_config", "size"]],
+    [["openai_image_generate", "style"], ["llm", 0, "image_generation_model_config", "style"]],
+    [["openai_image_generate", "quality"], ["llm", 0, "image_generation_model_config", "quality"]],
+    
+    [["llm_wake_prefix"], ["llm_settings", "wake_prefix"]],
+    
+    [["baidu_aip", "enable"], ["content_safety", "baidu_aip", "enable"]],
+    [["baidu_aip", "app_id"], ["content_safety", "baidu_aip", "app_id"]],
+    [["baidu_aip", "api_key"], ["content_safety", "baidu_aip", "api_key"]],
+    [["baidu_aip", "secret_key"], ["content_safety", "baidu_aip", "secret_key"]],
+    
+    [["qq_pic_mode"], ["t2i"]],
+    [["dump_history_interval"], ["dump_history_interval"]],
+    [["other_admins"], ["admins_id"]],
+    [["http_proxy"], ["http_proxy"]],
+    [["https_proxy"], ["https_proxy"]],
+    [["dashboard_username"], ["dashboard", "username"]],
+    [["dashboard_password"], ["dashboard", "password"]],
+    [["nick_qq"], ["wake_prefix"]],
+]
+
+# 配置项的中文描述、值类型
+CONFIG_METADATA_2 = {
+    "config_version": {"description": "配置版本", "type": "int"},
+    "platform": {
+        "description": "平台配置",
+        "type": "list",
+        "items": {
+            "name": {"description": "平台名称", "type": "string"},
+            "enable": {"description": "启用", "type": "bool"},
+            "appid": {"description": "appid", "type": "string"},
+            "secret": {"description": "secret", "type": "string"},
+            "enable_group_c2c": {"description": "启用消息列表单聊", "type": "bool"},
+            "enable_guild_direct_message": {"description": "启用频道私聊", "type": "bool"},
+            "host": {"description": "主机地址", "type": "string"},
+            "port": {"description": "端口", "type": "int"},
+            "websocket_port": {"description": "Websocket 端口", "type": "int"},
+            "ws_reverse_host": {"description": "反向 Websocket 主机地址", "type": "string"},
+            "ws_reverse_port": {"description": "反向 Websocket 端口", "type": "int"},
+            "enable_group": {"description": "接收群组消息", "type": "bool"},
+            "enable_guild": {"description": "接收频道消息", "type": "bool"},
+            "enable_direct_message": {"description": "接收频道私聊", "type": "bool"},
+            "enable_group_increase": {"description": "接收群组成员增加事件", "type": "bool"},
+        }
+    },
+    "platform_settings": {
+        "description": "平台设置",
+        "type": "object",
+        "items": {
+            "unique_session": {"description": "会话隔离到单个人", "type": "bool"},
+            "rate_limit": {
+                "description": "速率限制",
+                "type": "object",
+                "items": {
+                    "time": {"description": "消息速率限制时间", "type": "int"},
+                    "count": {"description": "消息速率限制计数", "type": "int"},
+                }
+            },
+            "reply_prefix": {"description": "回复前缀", "type": "string"},
+            "forward_threshold": {"description": "转发消息的字数阈值", "type": "int"},
+        }
+    },
+    "llm": {
+        "description": "大语言模型配置",
+        "type": "list",
+        "items": {
+            "name": {"description": "模型名称", "type": "string"},
+            "enable": {"description": "启用", "type": "bool"},
+            "key": {"description": "API Key", "type": "list", "items": {"type": "string"}},
+            "api_base": {"description": "API Base URL", "type": "string"},
+            "prompt_prefix": {"description": "Prompt 前缀", "type": "string"},
+            "default_personality": {"description": "默认人格", "type": "string"},
+            "model_config": {
+                "description": "模型配置",
+                "type": "object",
+                "items": {
+                    "model": {"description": "模型名称", "type": "string"},
+                    "max_tokens": {"description": "最大令牌数", "type": "int"},
+                    "temperature": {"description": "温度", "type": "float"},
+                    "top_p": {"description": "Top P值", "type": "float"},
+                    "frequency_penalty": {"description": "频率惩罚", "type": "float"},
+                    "presence_penalty": {"description": "存在惩罚", "type": "float"},
+                }
+            },
+            "image_generation_model_config": {
+                "description": "图像生成模型配置",
+                "type": "object",
+                "items": {
+                    "enable": {"description": "启用(需要该提供商支持图像生成模型)", "type": "bool"},
+                    "model": {"description": "模型名称", "type": "string"},
+                    "size": {"description": "图像尺寸", "type": "string"},
+                    "style": {"description": "图像风格", "type": "string"},
+                    "quality": {"description": "图像质量", "type": "string"},
+                }
+            },
+        }
+    },
+    "llm_settings": {
+        "description": "大语言模型设置",
+        "type": "object",
+        "items": {
+            "wake_prefix": {"description": "LLM 聊天额外唤醒前缀", "type": "string"},
+            "web_search": {"description": "启用网页搜索(能访问 Google 时效果最佳)", "type": "bool"},
+            "identifier": {"description": "启动识别群员(略微增加token开销)", "type": "bool"},
+        }
+    },
+    "content_safety": {
+        "description": "内容安全",
+        "type": "object",
+        "items": {
+            "baidu_aip": {
+                "description": "百度内容审核配置",
+                "type": "object",
+                "items": {
+                    "enable": {"description": "启用百度内容审核(需手动安装 baidu-aip 库)", "type": "bool"},
+                    "app_id": {"description": "APP ID", "type": "string"},
+                    "api_key": {"description": "API Key", "type": "string"},
+                    "secret_key": {"description": "Secret Key", "type": "string"},
+                }
+            },
+            "internal_keywords": {
+                "description": "内部关键词过滤",
+                "type": "object",
+                "items": {
+                    "enable": {"description": "启用内部关键词过滤", "type": "bool"},
+                    "extra_keywords": {"description": "额外关键词(支持正则)", "type": "list", "items": {"type": "string"}},
+                }
+            }
+        }
+    },
+    "wake_prefix": {"description": "唤醒前缀列表", "type": "list", "items": {"type": "string"}},
+    "t2i": {"description": "文本转图像功能", "type": "bool"},
+    "dump_history_interval": {"description": "历史记录转储间隔", "type": "int"},
+    "admins_id": {"description": "管理员ID列表", "type": "list", "items": {"type": "int"}},
+    "https_proxy": {"description": "HTTPS代理", "type": "string"},
+    "http_proxy": {"description": "HTTP代理", "type": "string"},
+    "dashboard": {
+        "description": "仪表盘配置",
+        "type": "object",
+        "items": {
+            "enable": {"description": "启用", "type": "bool"},
+            "username": {"description": "用户名", "type": "string"},
+            "password": {"description": "密码", "type": "string"},
+        }
+    },
+    "log_level": {"description": "控制台日志级别(DEBUG, INFO, WARNING, ERROR)", "type": "string"},
+    "t2i_endpoint": {"description": "文本转图像服务接口(为空时使用公共服务器)", "type": "string"},
+}

type/message_event.py

@@ -0,0 +1,70 @@
+from typing import List, Union, Optional
+from dataclasses import dataclass
+from type.register import RegisteredPlatform
+from type.types import Context
+from type.astrbot_message import AstrBotMessage, MessageType
+
+@dataclass
+class MessageResult():
+    result_message: Union[str, list]
+    is_command_call: Optional[bool] = False
+    use_t2i: Optional[bool] = None # None 为跟随用户设置
+    callback: Optional[callable] = None
+
+class AstrMessageEvent():
+
+    def __init__(self,
+                 message_str: str,
+                 message_obj: AstrBotMessage,
+                 platform: RegisteredPlatform,
+                 role: str,
+                 context: Context,
+                 session_id: str = None,
+                 unified_msg_origin: str = None,
+                 only_command: bool = False):
+        '''
+        AstrBot 消息事件。
+        
+        `message_str`: 纯消息字符串
+        `message_obj`: AstrBotMessage 对象
+        `platform`: 平台对象
+        `role`: 角色，`admin` or `member`
+        `context`: 全局对象
+        `session_id`: 会话id
+        `unified_msg_origin`: 统一消息来源
+        `only_command`: 是否只处理指令，而不使用 LLM 回复
+        '''
+        self.context = context
+        self.message_str = message_str
+        self.message_obj = message_obj
+        self.platform = platform
+        self.role = role
+        self.session_id = session_id
+        self.unified_msg_origin = unified_msg_origin
+        self.only_command = only_command
+        
+    def from_astrbot_message(message: AstrBotMessage, 
+                             context: Context,
+                             platform_name: str, 
+                             session_id: str,
+
+                             unified_msg_origin: str = None,
+                             only_command: bool = False):
+        
+        # 解析 role
+        sender_id = str(message.sender.user_id)
+        if sender_id in context.config_helper.admins_id:
+            role = 'admin'
+        else:
+            role = 'member'
+            
+        ame = AstrMessageEvent(message.message_str, 
+                               message, 
+                               context.find_platform(platform_name), 
+                               role, 
+                               context, 
+                               session_id,
+                               unified_msg_origin,
+                               only_command=only_command)
+        return ame
+

type/middleware.py

@@ -0,0 +1,8 @@
+from dataclasses import dataclass
+
+@dataclass
+class Middleware():
+    name: str = ""
+    description: str = ""
+    origin: str = ""  # 注册来源
+    func: callable = None

type/plugin.py

@@ -0,0 +1,53 @@
+from enum import Enum
+from types import ModuleType
+from typing import List
+from dataclasses import dataclass
+
+class PluginType(Enum):
+    PLATFORM = 'platform'  # 平台类插件。
+    LLM = 'llm'  # 大语言模型类插件
+    COMMON = 'common'  # 其他插件
+
+
+@dataclass
+class PluginMetadata:
+    '''
+    插件的元数据。
+    '''
+    # required
+    plugin_name: str
+    plugin_type: PluginType
+    author: str  # 插件作者
+    desc: str  # 插件简介
+    version: str  # 插件版本
+
+    # optional
+    repo: str = None  # 插件仓库地址
+
+    def __str__(self) -> str:
+        return f"PluginMetadata({self.plugin_name}, {self.plugin_type}, {self.desc}, {self.version}, {self.repo})"
+
+
+@dataclass
+class RegisteredPlugin:
+    '''
+    注册在 AstrBot 中的插件。
+    '''
+    metadata: PluginMetadata
+    plugin_instance: object
+    module_path: str
+    module: ModuleType
+    root_dir_name: str
+    trig_cnt: int = 0
+    
+    def reset_trig_cnt(self):
+        self.trig_cnt = 0
+    
+    def trig(self):
+        self.trig_cnt += 1
+
+    def __str__(self) -> str:
+        return f"RegisteredPlugin({self.metadata}, {self.module_path}, {self.root_dir_name})"
+
+
+RegisteredPlugins = List[RegisteredPlugin]

type/register.py

@@ -0,0 +1,27 @@
+from model.provider.provider import Provider as LLMProvider
+from model.platform import Platform
+from type.plugin import *
+from typing import List
+from dataclasses import dataclass
+
+@dataclass
+class RegisteredPlatform:
+    '''
+    注册在 AstrBot 中的平台。平台应当实现 Platform 接口。
+    '''
+    platform_name: str
+    platform_instance: Platform
+    origin: str = None  # 注册来源
+
+    def __str__(self) -> str:
+        return self.platform_name
+
+
+@dataclass
+class RegisteredLLM:
+    '''
+    注册在 AstrBot 中的大语言模型调用。大语言模型应当实现 LLMProvider 接口。
+    '''
+    llm_name: str
+    llm_instance: LLMProvider
+    origin: str = None  # 注册来源

type/types.py

@@ -0,0 +1,159 @@
+import asyncio, os, time
+from asyncio import Task
+from type.register import *
+from typing import List, Awaitable
+from logging import Logger
+from util.cmd_config import AstrBotConfig
+from util.t2i.renderer import TextToImageRenderer
+from util.updator.astrbot_updator import AstrBotUpdator
+from util.image_uploader import ImageUploader
+from util.updator.plugin_updator import PluginUpdator
+from type.command import CommandResult
+from type.middleware import Middleware
+from type.astrbot_message import MessageType
+from model.plugin.command import PluginCommandBridge
+from model.provider.provider import Provider
+from type.cached_queue import CachedQueue
+
+
+class Context:
+    '''
+    存放一些公用的数据，用于在不同模块(如core与command)之间传递
+    '''
+
+    def __init__(self):
+        self.running = True
+        self.logger: Logger = None
+        self.config_helper: AstrBotConfig = None
+        self.cached_plugins: List[RegisteredPlugin] = []  # 缓存的插件
+        self.platforms: List[RegisteredPlatform] = []
+        self.llms: List[RegisteredLLM] = []
+        self.default_personality: dict = None
+        
+        # self.unique_session = False # 独立会话
+        # self.version: str = None  # 机器人版本
+        # self.nick: tuple = None  # gocq 的唤醒词
+        # self.t2i_mode = False
+        # self.web_search = False  # 是否开启了网页搜索
+        
+        self.metrics_uploader = None
+        self.updator: AstrBotUpdator = None
+        self.plugin_updator: PluginUpdator = None
+        self.plugin_command_bridge = PluginCommandBridge(self.cached_plugins)
+        self.image_renderer = TextToImageRenderer()
+        self.image_uploader = ImageUploader()
+        self.message_handler = None # see astrbot/message/handler.py
+        self.ext_tasks: List[Task] = []
+        self.middlewares: List[Middleware] = []
+        
+        self.command_manager = None
+        self.running = True
+        self._loop = asyncio.get_event_loop()
+        self._start_running = int(time.time())
+
+        self._log_queue = CachedQueue()
+
+        # useless
+        # self.reply_prefix = ""
+
+    def register_commands(self, 
+                          plugin_name: str, 
+                          command_name: str, 
+                          description: str, 
+                          priority: int, 
+                          handler: callable,
+                          use_regex: bool = False,
+                          ignore_prefix: bool = False):
+        '''
+        注册插件指令。
+        
+        @param plugin_name: 插件名，注意需要和你的 metadata 中的一致。
+        @param command_name: 指令名，如 "help"。不需要带前缀。
+        @param description: 指令描述。
+        @param priority: 优先级越高，越先被处理。合理的优先级应该在 1-10 之间。
+        @param handler: 指令处理函数。函数参数：message: AstrMessageEvent, context: Context
+        @param use_regex: 是否使用正则表达式匹配指令名。
+        @param ignore_prefix: 是否忽略前缀。默认为 False。设置为 True 后，将不会检查用户设置的前缀。
+        
+        .. Example::
+        
+        ignore_prefix = False 时，用户输入 "/help" 时，会被识别为 "help" 指令。如果 ignore_prefix = True，则用户输入 "help" 也会被识别为 "help" 指令。
+        '''
+        self.plugin_command_bridge.register_command(plugin_name, 
+                                                    command_name, 
+                                                    description, 
+                                                    priority, 
+                                                    handler, 
+                                                    use_regex,
+                                                    ignore_prefix)
+        
+    def register_task(self, coro: Awaitable, task_name: str):
+        '''
+        注册任务。适用于需要长时间运行的插件。
+        
+        `coro`: 协程对象
+        `task_name`: 任务名，用于标识任务。自定义即可。
+        '''
+        task = asyncio.create_task(coro, name=task_name)
+        self.ext_tasks.append(task)
+        
+    def register_provider(self, llm_name: str, provider: Provider, origin: str = ''):
+        '''
+        注册一个提供 LLM 资源的 Provider。
+        
+        `provider`: Provider 对象。即你的实现需要继承 Provider 类。至少应该实现 text_chat() 方法。
+        '''
+        self.llms.append(RegisteredLLM(llm_name, provider, origin))
+        
+    def register_llm_tool(self, tool_name: str, params: list, desc: str, func: callable):
+        '''
+        为函数调用（function-calling / tools-use）添加工具。
+        
+        @param name: 函数名
+        @param func_args: 函数参数列表，格式为 [{"type": "string", "name": "arg_name", "description": "arg_description"}, ...]
+        @param desc: 函数描述
+        @param func_obj: 处理函数
+        '''
+        self.message_handler.llm_tools.add_func(tool_name, params, desc, func)
+    
+    def unregister_llm_tool(self, tool_name: str):
+        '''
+        删除一个函数调用工具。
+        '''
+        self.message_handler.llm_tools.remove_func(tool_name)
+        
+    def register_middleware(self, middleware: Middleware):
+        '''
+        注册一个中间件。所有的消息事件都会经过中间件处理，然后再进入 LLM 聊天模块。
+        
+        在 AstrBot 中，会对到来的消息事件首先检查指令，然后再检查中间件。触发指令后将不会进入 LLM 聊天模块，而中间件会。
+        '''
+        self.middlewares.append(middleware)
+     
+    def find_platform(self, platform_name: str) -> RegisteredPlatform:
+        for platform in self.platforms:
+            if platform_name == platform.platform_name:
+                return platform
+        
+        if not os.environ.get('TEST_MODE', 'off') == 'on': # 测试模式下不报错
+            raise ValueError("couldn't find the platform you specified")
+
+    async def send_message(self, unified_msg_origin: str, message: CommandResult):
+        '''
+        发送消息。
+        
+        `unified_msg_origin`: 统一消息来源
+        `message`: 消息内容
+        '''
+        l = unified_msg_origin.split(":")
+        if len(l) != 3:
+            raise ValueError("Invalid unified_msg_origin")
+        platform_name, message_type, id = l
+        platform = self.find_platform(platform_name)
+        await platform.platform_instance.send_msg_new(MessageType(message_type), id, message)
+    
+    def get_current_llm_provider(self) -> Provider:
+        '''
+        获取当前的 LLM Provider。
+        '''
+        return self.message_handler.provider