Compare commits
97 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
3007f67cab | ||
|
|
ee08659f01 | ||
|
|
baf5ad0fab | ||
|
|
8bdd748aec | ||
|
|
cef0c22f52 | ||
|
|
13d3fc5cfe | ||
|
|
b91141e2be | ||
|
|
f8a4b54165 | ||
|
|
afe007ca0b | ||
|
|
8a9a044f95 | ||
|
|
5eaf03e227 | ||
|
|
a8437d9331 | ||
|
|
e0392fa98b | ||
|
|
68ff8951de | ||
|
|
9c6b31e71c | ||
|
|
50f74f5ba2 | ||
|
|
b9de2aef60 | ||
|
|
7a47598538 | ||
|
|
3c8c28ebd5 | ||
|
|
524285f767 | ||
|
|
c2a34475f1 | ||
|
|
a69195a02b | ||
|
|
19d7438499 | ||
|
|
ccb380ce06 | ||
|
|
a35c439bbd | ||
|
|
09d1f96603 | ||
|
|
26aa18d980 | ||
|
|
d10b542797 | ||
|
|
ce4e4fb8dd | ||
|
|
8f4a31cf8c | ||
|
|
23549f13d6 | ||
|
|
869d11f9a6 | ||
|
|
02e73b82ee | ||
|
|
f85f87f545 | ||
|
|
1fff5713f3 | ||
|
|
8453ec36f0 | ||
|
|
d5b3ce8424 | ||
|
|
80cbbfa5ca | ||
|
|
9177bb660f | ||
|
|
a3df39a01a | ||
|
|
25dce05cbb | ||
|
|
1542ea3e03 | ||
|
|
6084abbcfe | ||
|
|
ed19b63914 | ||
|
|
4efeb85296 | ||
|
|
fc76665615 | ||
|
|
3a044bb71a | ||
|
|
cddd606562 | ||
|
|
7a5bc51c11 | ||
|
|
9f939b4b6f | ||
|
|
80a86f5b1b | ||
|
|
a0ce1855ab | ||
|
|
a4b43b884a | ||
|
|
824c0f6667 | ||
|
|
a030fe8491 | ||
|
|
3a9429e8ef | ||
|
|
c4eb1ab748 | ||
|
|
29ed19d600 | ||
|
|
0cc65513a5 | ||
|
|
debc048659 | ||
|
|
92f5c918dd | ||
|
|
9519f1e8e2 | ||
|
|
a8f874bf05 | ||
|
|
9d9917e45b | ||
|
|
91ee0a870d | ||
|
|
6cbbffc5a9 | ||
|
|
8f26fd34d1 | ||
|
|
fda655f6d7 | ||
|
|
a663d6509b | ||
|
|
9ec8839efa | ||
|
|
a7a0350eb2 | ||
|
|
39a7a0d960 | ||
|
|
7740e1e131 | ||
|
|
9dce1ed47e | ||
|
|
e84a00d3a5 | ||
|
|
88a944cb57 | ||
|
|
20c32e72cc | ||
|
|
4788c20816 | ||
|
|
e83fc570a4 | ||
|
|
e841b6af88 | ||
|
|
ea6f209557 | ||
|
|
9bfa726107 | ||
|
|
d24902c66d | ||
|
|
72aea2d3f3 | ||
|
|
dc9612d564 | ||
|
|
1770556d56 | ||
|
|
888fb84aee | ||
|
|
d597fd056d | ||
|
|
dea0ab3974 | ||
|
|
da6facd7d7 | ||
|
|
bb8ab5f173 | ||
|
|
ac8a541059 | ||
|
|
0e66771f0e | ||
|
|
d3a295a801 | ||
|
|
f2df771771 | ||
|
|
7b72cd87a5 | ||
|
|
9431efc6d1 |
51
.github/PULL_REQUEST_TEMPLATE.md
vendored
51
.github/PULL_REQUEST_TEMPLATE.md
vendored
@@ -1,19 +1,46 @@
|
|||||||
<!-- 如果有的话,指定这个 PR 要解决的 ISSUE -->
|
<!-- 如果有的话,请指定此 PR 旨在解决的 ISSUE 编号。 -->
|
||||||
解决了 #XYZ
|
<!-- If applicable, please specify the ISSUE number this PR aims to resolve. -->
|
||||||
|
|
||||||
### Motivation
|
fixes #XYZ
|
||||||
|
|
||||||
<!--解释为什么要改动-->
|
---
|
||||||
|
|
||||||
### Modifications
|
### Motivation / 动机
|
||||||
|
|
||||||
<!--简单解释你的改动-->
|
<!--请描述此项更改的动机:它解决了什么问题?(例如:修复了 XX 错误,添加了 YY 功能)-->
|
||||||
|
<!--Please describe the motivation for this change: What problem does it solve? (e.g., Fixes XX bug, adds YY feature)-->
|
||||||
|
|
||||||
### Check
|
### Modifications / 改动点
|
||||||
|
|
||||||
<!--如果分支被合并,您的代码将服务于数万名用户!在提交前,请核查一下几点内容-->
|
<!--请总结你的改动:哪些核心文件被修改了?实现了什么功能?-->
|
||||||
|
<!--Please summarize your changes: What core files were modified? What functionality was implemented?-->
|
||||||
|
|
||||||
- [ ] 😊 我的 Commit Message 符合良好的[规范](https://www.conventionalcommits.org/en/v1.0.0/#summary)
|
### Verification Steps / 验证步骤
|
||||||
- [ ] 👀 我的更改经过良好的测试
|
|
||||||
- [ ] 🤓 我确保没有引入新依赖库,或者引入了新依赖库的同时将其添加到了 `requirements.txt` 和 `pyproject.toml` 文件相应位置。
|
<!--请为审查者 (Reviewer) 提供清晰、可复现的验证步骤(例如:1. 导航到... 2. 点击...)。-->
|
||||||
- [ ] 😮 我的更改没有引入恶意代码
|
<!--Please provide clear and reproducible verification steps for the Reviewer (e.g., 1. Navigate to... 2. Click...).-->
|
||||||
|
|
||||||
|
### Screenshots or Test Results / 运行截图或测试结果
|
||||||
|
|
||||||
|
<!--请粘贴截图、GIF 或测试日志,作为执行“验证步骤”的证据,证明此改动有效。-->
|
||||||
|
<!--Please paste screenshots, GIFs, or test logs here as evidence of executing the "Verification Steps" to prove this change is effective.-->
|
||||||
|
|
||||||
|
### Compatibility & Breaking Changes / 兼容性与破坏性变更
|
||||||
|
|
||||||
|
<!--请说明此变更的兼容性:哪些是破坏性变更?哪些地方做了向后兼容处理?是否提供了数据迁移方法?-->
|
||||||
|
<!--Please explain the compatibility of this change: What are the breaking changes? What backward-compatible measures were taken? Are data migration paths provided?-->
|
||||||
|
|
||||||
|
- [ ] 这是一个破坏性变更 (Breaking Change)。/ This is a breaking change.
|
||||||
|
- [ ] 这不是一个破坏性变更。/ This is NOT a breaking change.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
### Checklist / 检查清单
|
||||||
|
|
||||||
|
<!--如果分支被合并,您的代码将服务于数万名用户!在提交前,请核查一下几点内容。-->
|
||||||
|
<!--If merged, your code will serve tens of thousands of users! Please double-check the following items before submitting.-->
|
||||||
|
|
||||||
|
- [ ] 😊 如果 PR 中有新加入的功能,已经通过 Issue / 邮件等方式和作者讨论过。/ If there are new features added in the PR, I have discussed it with the authors through issues/emails, etc.
|
||||||
|
- [ ] 👀 我的更改经过了良好的测试,**并已在上方提供了“验证步骤”和“运行截图”**。/ My changes have been well-tested, **and "Verification Steps" and "Screenshots" have been provided above**.
|
||||||
|
- [ ] 🤓 我确保没有引入新依赖库,或者引入了新依赖库的同时将其添加到了 `requirements.txt` 和 `pyproject.toml` 文件相应位置。/ I have ensured that no new dependencies are introduced, OR if new dependencies are introduced, they have been added to the appropriate locations in `requirements.txt` and `pyproject.toml`.
|
||||||
|
- [ ] 😮 我的更改没有引入恶意代码。/ My changes do not introduce malicious code.
|
||||||
|
|||||||
36
.github/auto_assign.yml
vendored
Normal file
36
.github/auto_assign.yml
vendored
Normal file
@@ -0,0 +1,36 @@
|
|||||||
|
# Set to true to add reviewers to pull requests
|
||||||
|
addReviewers: true
|
||||||
|
|
||||||
|
# Set to true to add assignees to pull requests
|
||||||
|
addAssignees: false
|
||||||
|
|
||||||
|
# A list of reviewers to be added to pull requests (GitHub user name)
|
||||||
|
reviewers:
|
||||||
|
- Soulter
|
||||||
|
- Raven95676
|
||||||
|
- Larch-C
|
||||||
|
- anka-afk
|
||||||
|
- advent259141
|
||||||
|
# - zouyonghe
|
||||||
|
|
||||||
|
# A number of reviewers added to the pull request
|
||||||
|
# Set 0 to add all the reviewers (default: 0)
|
||||||
|
numberOfReviewers: 2
|
||||||
|
|
||||||
|
# A list of assignees, overrides reviewers if set
|
||||||
|
# assignees:
|
||||||
|
# - assigneeA
|
||||||
|
|
||||||
|
# A number of assignees to add to the pull request
|
||||||
|
# Set to 0 to add all of the assignees.
|
||||||
|
# Uses numberOfReviewers if unset.
|
||||||
|
# numberOfAssignees: 2
|
||||||
|
|
||||||
|
# A list of keywords to be skipped the process that add reviewers if pull requests include it
|
||||||
|
skipKeywords:
|
||||||
|
- wip
|
||||||
|
- draft
|
||||||
|
|
||||||
|
# A list of users to be skipped by both the add reviewers and add assignees processes
|
||||||
|
# skipUsers:
|
||||||
|
# - dependabot[bot]
|
||||||
2
.github/workflows/auto_release.yml
vendored
2
.github/workflows/auto_release.yml
vendored
@@ -73,7 +73,7 @@ jobs:
|
|||||||
uses: actions/checkout@v5
|
uses: actions/checkout@v5
|
||||||
|
|
||||||
- name: Set up Python
|
- name: Set up Python
|
||||||
uses: actions/setup-python@v5
|
uses: actions/setup-python@v6
|
||||||
with:
|
with:
|
||||||
python-version: '3.10'
|
python-version: '3.10'
|
||||||
|
|
||||||
|
|||||||
34
.github/workflows/code-format.yml
vendored
Normal file
34
.github/workflows/code-format.yml
vendored
Normal file
@@ -0,0 +1,34 @@
|
|||||||
|
name: Code Format Check
|
||||||
|
|
||||||
|
on:
|
||||||
|
pull_request:
|
||||||
|
branches: [ master ]
|
||||||
|
push:
|
||||||
|
branches: [ master ]
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
format-check:
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
|
||||||
|
steps:
|
||||||
|
- name: Checkout code
|
||||||
|
uses: actions/checkout@v5
|
||||||
|
|
||||||
|
- name: Set up Python
|
||||||
|
uses: actions/setup-python@v6
|
||||||
|
with:
|
||||||
|
python-version: '3.10'
|
||||||
|
|
||||||
|
- name: Install UV
|
||||||
|
run: pip install uv
|
||||||
|
|
||||||
|
- name: Install dependencies
|
||||||
|
run: uv sync
|
||||||
|
|
||||||
|
- name: Check code formatting with ruff
|
||||||
|
run: |
|
||||||
|
uv run ruff format --check .
|
||||||
|
|
||||||
|
- name: Check code style with ruff
|
||||||
|
run: |
|
||||||
|
uv run ruff check .
|
||||||
2
.github/workflows/coverage_test.yml
vendored
2
.github/workflows/coverage_test.yml
vendored
@@ -22,7 +22,7 @@ jobs:
|
|||||||
fetch-depth: 0
|
fetch-depth: 0
|
||||||
|
|
||||||
- name: Set up Python
|
- name: Set up Python
|
||||||
uses: actions/setup-python@v5
|
uses: actions/setup-python@v6
|
||||||
|
|
||||||
- name: Install dependencies
|
- name: Install dependencies
|
||||||
run: |
|
run: |
|
||||||
|
|||||||
1
.github/workflows/dashboard_ci.yml
vendored
1
.github/workflows/dashboard_ci.yml
vendored
@@ -37,6 +37,7 @@ jobs:
|
|||||||
!dist/**/*.md
|
!dist/**/*.md
|
||||||
|
|
||||||
- name: Create GitHub Release
|
- name: Create GitHub Release
|
||||||
|
if: github.event_name == 'push'
|
||||||
uses: ncipollo/release-action@v1
|
uses: ncipollo/release-action@v1
|
||||||
with:
|
with:
|
||||||
tag: release-${{ github.sha }}
|
tag: release-${{ github.sha }}
|
||||||
|
|||||||
2
.github/workflows/stale.yml
vendored
2
.github/workflows/stale.yml
vendored
@@ -18,7 +18,7 @@ jobs:
|
|||||||
pull-requests: write
|
pull-requests: write
|
||||||
|
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/stale@v9
|
- uses: actions/stale@v10
|
||||||
with:
|
with:
|
||||||
repo-token: ${{ secrets.GITHUB_TOKEN }}
|
repo-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
stale-issue-message: 'Stale issue message'
|
stale-issue-message: 'Stale issue message'
|
||||||
|
|||||||
2
.gitignore
vendored
2
.gitignore
vendored
@@ -31,3 +31,5 @@ packages/python_interpreter/workplace
|
|||||||
.idea
|
.idea
|
||||||
pytest.ini
|
pytest.ini
|
||||||
.astrbot
|
.astrbot
|
||||||
|
|
||||||
|
uv.lock
|
||||||
19
Dockerfile
19
Dockerfile
@@ -4,8 +4,6 @@ WORKDIR /AstrBot
|
|||||||
COPY . /AstrBot/
|
COPY . /AstrBot/
|
||||||
|
|
||||||
RUN apt-get update && apt-get install -y --no-install-recommends \
|
RUN apt-get update && apt-get install -y --no-install-recommends \
|
||||||
nodejs \
|
|
||||||
npm \
|
|
||||||
gcc \
|
gcc \
|
||||||
build-essential \
|
build-essential \
|
||||||
python3-dev \
|
python3-dev \
|
||||||
@@ -13,23 +11,20 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
|
|||||||
libssl-dev \
|
libssl-dev \
|
||||||
ca-certificates \
|
ca-certificates \
|
||||||
bash \
|
bash \
|
||||||
|
ffmpeg \
|
||||||
&& apt-get clean \
|
&& apt-get clean \
|
||||||
&& rm -rf /var/lib/apt/lists/*
|
&& rm -rf /var/lib/apt/lists/*
|
||||||
|
|
||||||
|
RUN apt-get update && apt-get install -y curl gnupg && \
|
||||||
|
curl -fsSL https://deb.nodesource.com/setup_lts.x | bash - && \
|
||||||
|
apt-get install -y nodejs && \
|
||||||
|
rm -rf /var/lib/apt/lists/*
|
||||||
|
|
||||||
RUN python -m pip install uv
|
RUN python -m pip install uv
|
||||||
RUN uv pip install -r requirements.txt --no-cache-dir --system
|
RUN uv pip install -r requirements.txt --no-cache-dir --system
|
||||||
RUN uv pip install socksio uv pyffmpeg pilk --no-cache-dir --system
|
RUN uv pip install socksio uv pilk --no-cache-dir --system
|
||||||
|
|
||||||
# 释出 ffmpeg
|
|
||||||
RUN python -c "from pyffmpeg import FFmpeg; ff = FFmpeg();"
|
|
||||||
|
|
||||||
# add /root/.pyffmpeg/bin/ffmpeg to PATH, inorder to use ffmpeg
|
|
||||||
RUN echo 'export PATH=$PATH:/root/.pyffmpeg/bin' >> ~/.bashrc
|
|
||||||
|
|
||||||
EXPOSE 6185
|
EXPOSE 6185
|
||||||
EXPOSE 6186
|
EXPOSE 6186
|
||||||
|
|
||||||
CMD [ "python", "main.py" ]
|
CMD [ "python", "main.py" ]
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
25
README.md
25
README.md
@@ -14,12 +14,12 @@
|
|||||||
<a href="https://qm.qq.com/cgi-bin/qm/qr?k=wtbaNx7EioxeaqS9z7RQWVXPIxg2zYr7&jump_from=webapi&authKey=vlqnv/AV2DbJEvGIcxdlNSpfxVy+8vVqijgreRdnVKOaydpc+YSw4MctmEbr0k5"><img alt="QQ_community" src="https://img.shields.io/badge/QQ群-775869627-purple?style=for-the-badge&color=76bad9"></a>
|
<a href="https://qm.qq.com/cgi-bin/qm/qr?k=wtbaNx7EioxeaqS9z7RQWVXPIxg2zYr7&jump_from=webapi&authKey=vlqnv/AV2DbJEvGIcxdlNSpfxVy+8vVqijgreRdnVKOaydpc+YSw4MctmEbr0k5"><img alt="QQ_community" src="https://img.shields.io/badge/QQ群-775869627-purple?style=for-the-badge&color=76bad9"></a>
|
||||||
<a href="https://t.me/+hAsD2Ebl5as3NmY1"><img alt="Telegram_community" src="https://img.shields.io/badge/Telegram-AstrBot-purple?style=for-the-badge&color=76bad9"></a>
|
<a href="https://t.me/+hAsD2Ebl5as3NmY1"><img alt="Telegram_community" src="https://img.shields.io/badge/Telegram-AstrBot-purple?style=for-the-badge&color=76bad9"></a>
|
||||||
[](https://wakatime.com/badge/user/915e5316-99c6-4563-a483-ef186cf000c9/project/018e705a-a1a7-409a-a849-3013485e6c8e)
|
[](https://wakatime.com/badge/user/915e5316-99c6-4563-a483-ef186cf000c9/project/018e705a-a1a7-409a-a849-3013485e6c8e)
|
||||||

|
|
||||||

|

|
||||||
|
|
||||||
<a href="https://github.com/Soulter/AstrBot/blob/master/README_en.md">English</a> |
|
<a href="https://github.com/Soulter/AstrBot/blob/master/README_en.md">English</a> |
|
||||||
<a href="https://github.com/Soulter/AstrBot/blob/master/README_ja.md">日本語</a> |
|
<a href="https://github.com/Soulter/AstrBot/blob/master/README_ja.md">日本語</a> |
|
||||||
<a href="https://astrbot.app/">查看文档</a> |
|
<a href="https://astrbot.app/">文档</a> |
|
||||||
|
<a href="https://blog.astrbot.app/">Blog</a> |
|
||||||
<a href="https://github.com/Soulter/AstrBot/issues">问题提交</a>
|
<a href="https://github.com/Soulter/AstrBot/issues">问题提交</a>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
@@ -100,7 +100,7 @@ uv run main.py
|
|||||||
- 3 群:630166526
|
- 3 群:630166526
|
||||||
- 5 群:822130018
|
- 5 群:822130018
|
||||||
- 6 群:753075035
|
- 6 群:753075035
|
||||||
- 开发者群:753075035
|
- 开发者群:975206796
|
||||||
- 开发者群(备份):295657329
|
- 开发者群(备份):295657329
|
||||||
|
|
||||||
### Telegram 群组
|
### Telegram 群组
|
||||||
@@ -111,7 +111,6 @@ uv run main.py
|
|||||||
|
|
||||||
<a href="https://discord.gg/hAVk6tgV36"><img alt="Discord_community" src="https://img.shields.io/badge/Discord-AstrBot-purple?style=for-the-badge&color=76bad9"></a>
|
<a href="https://discord.gg/hAVk6tgV36"><img alt="Discord_community" src="https://img.shields.io/badge/Discord-AstrBot-purple?style=for-the-badge&color=76bad9"></a>
|
||||||
|
|
||||||
|
|
||||||
## ⚡ 消息平台支持情况
|
## ⚡ 消息平台支持情况
|
||||||
|
|
||||||
| 平台 | 支持性 |
|
| 平台 | 支持性 |
|
||||||
@@ -128,6 +127,8 @@ uv run main.py
|
|||||||
| Discord | ✔ |
|
| Discord | ✔ |
|
||||||
| [KOOK](https://github.com/wuyan1003/astrbot_plugin_kook_adapter) | ✔ |
|
| [KOOK](https://github.com/wuyan1003/astrbot_plugin_kook_adapter) | ✔ |
|
||||||
| [VoceChat](https://github.com/HikariFroya/astrbot_plugin_vocechat) | ✔ |
|
| [VoceChat](https://github.com/HikariFroya/astrbot_plugin_vocechat) | ✔ |
|
||||||
|
| Satori | ✔ |
|
||||||
|
| Misskey | ✔ |
|
||||||
|
|
||||||
## ⚡ 提供商支持情况
|
## ⚡ 提供商支持情况
|
||||||
|
|
||||||
@@ -173,7 +174,6 @@ pip install pre-commit
|
|||||||
pre-commit install
|
pre-commit install
|
||||||
```
|
```
|
||||||
|
|
||||||
|
|
||||||
## ❤️ Special Thanks
|
## ❤️ Special Thanks
|
||||||
|
|
||||||
特别感谢所有 Contributors 和插件开发者对 AstrBot 的贡献 ❤️
|
特别感谢所有 Contributors 和插件开发者对 AstrBot 的贡献 ❤️
|
||||||
@@ -182,10 +182,18 @@ pre-commit install
|
|||||||
<img src="https://contrib.rocks/image?repo=AstrBotDevs/AstrBot" />
|
<img src="https://contrib.rocks/image?repo=AstrBotDevs/AstrBot" />
|
||||||
</a>
|
</a>
|
||||||
|
|
||||||
此外,本项目的诞生离不开以下开源项目:
|
此外,本项目的诞生离不开以下开源项目的帮助:
|
||||||
|
|
||||||
- [NapNeko/NapCatQQ](https://github.com/NapNeko/NapCatQQ) - 伟大的猫猫框架
|
- [NapNeko/NapCatQQ](https://github.com/NapNeko/NapCatQQ) - 伟大的猫猫框架
|
||||||
- [wechatpy/wechatpy](https://github.com/wechatpy/wechatpy)
|
|
||||||
|
另外,一些同类型其他的活跃开源 Bot 项目:
|
||||||
|
|
||||||
|
- [nonebot/nonebot2](https://github.com/nonebot/nonebot2) - 扩展性极强的 Bot 框架
|
||||||
|
- [koishijs/koishi](https://github.com/koishijs/koishi) - 扩展性极强的 Bot 框架
|
||||||
|
- [MaiM-with-u/MaiBot](https://github.com/MaiM-with-u/MaiBot) - 注重拟人功能的 ChatBot
|
||||||
|
- [langbot-app/LangBot](https://github.com/langbot-app/LangBot) - 功能丰富的 Bot 平台
|
||||||
|
- [KroMiose/nekro-agent](https://github.com/KroMiose/nekro-agent) - 注重 Agent 的 ChatBot
|
||||||
|
- [zhenxun-org/zhenxun_bot](https://github.com/zhenxun-org/zhenxun_bot) - 功能完善的 ChatBot
|
||||||
|
|
||||||
## ⭐ Star History
|
## ⭐ Star History
|
||||||
|
|
||||||
@@ -198,9 +206,6 @@ pre-commit install
|
|||||||
|
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
|
|
||||||
</details>
|
</details>
|
||||||
|
|
||||||
|
|
||||||
_私は、高性能ですから!_
|
_私は、高性能ですから!_
|
||||||
|
|
||||||
|
|||||||
@@ -7,6 +7,7 @@ from astrbot.core.star.register import (
|
|||||||
register_permission_type as permission_type,
|
register_permission_type as permission_type,
|
||||||
register_custom_filter as custom_filter,
|
register_custom_filter as custom_filter,
|
||||||
register_on_astrbot_loaded as on_astrbot_loaded,
|
register_on_astrbot_loaded as on_astrbot_loaded,
|
||||||
|
register_on_platform_loaded as on_platform_loaded,
|
||||||
register_on_llm_request as on_llm_request,
|
register_on_llm_request as on_llm_request,
|
||||||
register_on_llm_response as on_llm_response,
|
register_on_llm_response as on_llm_response,
|
||||||
register_llm_tool as llm_tool,
|
register_llm_tool as llm_tool,
|
||||||
@@ -41,6 +42,7 @@ __all__ = [
|
|||||||
"custom_filter",
|
"custom_filter",
|
||||||
"PermissionType",
|
"PermissionType",
|
||||||
"on_astrbot_loaded",
|
"on_astrbot_loaded",
|
||||||
|
"on_platform_loaded",
|
||||||
"on_llm_request",
|
"on_llm_request",
|
||||||
"llm_tool",
|
"llm_tool",
|
||||||
"on_decorating_result",
|
"on_decorating_result",
|
||||||
|
|||||||
@@ -124,7 +124,8 @@ def build_plug_list(plugins_dir: Path) -> list:
|
|||||||
if metadata and all(
|
if metadata and all(
|
||||||
k in metadata for k in ["name", "desc", "version", "author", "repo"]
|
k in metadata for k in ["name", "desc", "version", "author", "repo"]
|
||||||
):
|
):
|
||||||
result.append({
|
result.append(
|
||||||
|
{
|
||||||
"name": str(metadata.get("name", "")),
|
"name": str(metadata.get("name", "")),
|
||||||
"desc": str(metadata.get("desc", "")),
|
"desc": str(metadata.get("desc", "")),
|
||||||
"version": str(metadata.get("version", "")),
|
"version": str(metadata.get("version", "")),
|
||||||
@@ -132,7 +133,8 @@ def build_plug_list(plugins_dir: Path) -> list:
|
|||||||
"repo": str(metadata.get("repo", "")),
|
"repo": str(metadata.get("repo", "")),
|
||||||
"status": PluginStatus.INSTALLED,
|
"status": PluginStatus.INSTALLED,
|
||||||
"local_path": str(plugin_dir),
|
"local_path": str(plugin_dir),
|
||||||
})
|
}
|
||||||
|
)
|
||||||
|
|
||||||
# 获取在线插件列表
|
# 获取在线插件列表
|
||||||
online_plugins = []
|
online_plugins = []
|
||||||
@@ -142,7 +144,8 @@ def build_plug_list(plugins_dir: Path) -> list:
|
|||||||
resp.raise_for_status()
|
resp.raise_for_status()
|
||||||
data = resp.json()
|
data = resp.json()
|
||||||
for plugin_id, plugin_info in data.items():
|
for plugin_id, plugin_info in data.items():
|
||||||
online_plugins.append({
|
online_plugins.append(
|
||||||
|
{
|
||||||
"name": str(plugin_id),
|
"name": str(plugin_id),
|
||||||
"desc": str(plugin_info.get("desc", "")),
|
"desc": str(plugin_info.get("desc", "")),
|
||||||
"version": str(plugin_info.get("version", "")),
|
"version": str(plugin_info.get("version", "")),
|
||||||
@@ -150,7 +153,8 @@ def build_plug_list(plugins_dir: Path) -> list:
|
|||||||
"repo": str(plugin_info.get("repo", "")),
|
"repo": str(plugin_info.get("repo", "")),
|
||||||
"status": PluginStatus.NOT_INSTALLED,
|
"status": PluginStatus.NOT_INSTALLED,
|
||||||
"local_path": None,
|
"local_path": None,
|
||||||
})
|
}
|
||||||
|
)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
click.echo(f"获取在线插件列表失败: {e}", err=True)
|
click.echo(f"获取在线插件列表失败: {e}", err=True)
|
||||||
|
|
||||||
|
|||||||
@@ -9,5 +9,5 @@ from .hooks import BaseAgentRunHooks
|
|||||||
class Agent(Generic[TContext]):
|
class Agent(Generic[TContext]):
|
||||||
name: str
|
name: str
|
||||||
instructions: str | None = None
|
instructions: str | None = None
|
||||||
tools: list[str, FunctionTool] | None = None
|
tools: list[str | FunctionTool] | None = None
|
||||||
run_hooks: BaseAgentRunHooks[TContext] | None = None
|
run_hooks: BaseAgentRunHooks[TContext] | None = None
|
||||||
|
|||||||
@@ -92,7 +92,7 @@ class MCPClient:
|
|||||||
self.session: Optional[mcp.ClientSession] = None
|
self.session: Optional[mcp.ClientSession] = None
|
||||||
self.exit_stack = AsyncExitStack()
|
self.exit_stack = AsyncExitStack()
|
||||||
|
|
||||||
self.name = None
|
self.name: str | None = None
|
||||||
self.active: bool = True
|
self.active: bool = True
|
||||||
self.tools: list[mcp.Tool] = []
|
self.tools: list[mcp.Tool] = []
|
||||||
self.server_errlogs: list[str] = []
|
self.server_errlogs: list[str] = []
|
||||||
@@ -198,6 +198,8 @@ class MCPClient:
|
|||||||
|
|
||||||
async def list_tools_and_save(self) -> mcp.ListToolsResult:
|
async def list_tools_and_save(self) -> mcp.ListToolsResult:
|
||||||
"""List all tools from the server and save them to self.tools"""
|
"""List all tools from the server and save them to self.tools"""
|
||||||
|
if not self.session:
|
||||||
|
raise Exception("MCP Client is not initialized")
|
||||||
response = await self.session.list_tools()
|
response = await self.session.list_tools()
|
||||||
self.tools = response.tools
|
self.tools = response.tools
|
||||||
return response
|
return response
|
||||||
|
|||||||
@@ -2,6 +2,7 @@ from dataclasses import dataclass
|
|||||||
import typing as T
|
import typing as T
|
||||||
from astrbot.core.message.message_event_result import MessageChain
|
from astrbot.core.message.message_event_result import MessageChain
|
||||||
|
|
||||||
|
|
||||||
class AgentResponseData(T.TypedDict):
|
class AgentResponseData(T.TypedDict):
|
||||||
chain: MessageChain
|
chain: MessageChain
|
||||||
|
|
||||||
|
|||||||
@@ -14,4 +14,5 @@ class ContextWrapper(Generic[TContext]):
|
|||||||
context: TContext
|
context: TContext
|
||||||
event: AstrMessageEvent
|
event: AstrMessageEvent
|
||||||
|
|
||||||
|
|
||||||
NoContext = ContextWrapper[None]
|
NoContext = ContextWrapper[None]
|
||||||
|
|||||||
@@ -258,7 +258,7 @@ class ToolLoopAgentRunner(BaseAgentRunner[TContext]):
|
|||||||
)
|
)
|
||||||
yield MessageChain(
|
yield MessageChain(
|
||||||
type="tool_direct_result"
|
type="tool_direct_result"
|
||||||
).base64_image(res.content[0].data)
|
).base64_image(resource.blob)
|
||||||
else:
|
else:
|
||||||
tool_call_result_blocks.append(
|
tool_call_result_blocks.append(
|
||||||
ToolCallMessageSegment(
|
ToolCallMessageSegment(
|
||||||
@@ -269,17 +269,6 @@ class ToolLoopAgentRunner(BaseAgentRunner[TContext]):
|
|||||||
)
|
)
|
||||||
yield MessageChain().message("返回的数据类型不受支持。")
|
yield MessageChain().message("返回的数据类型不受支持。")
|
||||||
|
|
||||||
try:
|
|
||||||
await self.agent_hooks.on_tool_end(
|
|
||||||
self.run_context,
|
|
||||||
func_tool_name,
|
|
||||||
func_tool_args,
|
|
||||||
resp,
|
|
||||||
)
|
|
||||||
except Exception as e:
|
|
||||||
logger.error(
|
|
||||||
f"Error in on_tool_end hook: {e}", exc_info=True
|
|
||||||
)
|
|
||||||
elif resp is None:
|
elif resp is None:
|
||||||
# Tool 直接请求发送消息给用户
|
# Tool 直接请求发送消息给用户
|
||||||
# 这里我们将直接结束 Agent Loop。
|
# 这里我们将直接结束 Agent Loop。
|
||||||
@@ -289,14 +278,6 @@ class ToolLoopAgentRunner(BaseAgentRunner[TContext]):
|
|||||||
yield MessageChain(
|
yield MessageChain(
|
||||||
chain=res.chain, type="tool_direct_result"
|
chain=res.chain, type="tool_direct_result"
|
||||||
)
|
)
|
||||||
try:
|
|
||||||
await self.agent_hooks.on_tool_end(
|
|
||||||
self.run_context, func_tool_name, func_tool_args, None
|
|
||||||
)
|
|
||||||
except Exception as e:
|
|
||||||
logger.error(
|
|
||||||
f"Error in on_tool_end hook: {e}", exc_info=True
|
|
||||||
)
|
|
||||||
else:
|
else:
|
||||||
logger.warning(
|
logger.warning(
|
||||||
f"Tool 返回了不支持的类型: {type(resp)},将忽略。"
|
f"Tool 返回了不支持的类型: {type(resp)},将忽略。"
|
||||||
@@ -304,12 +285,10 @@ class ToolLoopAgentRunner(BaseAgentRunner[TContext]):
|
|||||||
|
|
||||||
try:
|
try:
|
||||||
await self.agent_hooks.on_tool_end(
|
await self.agent_hooks.on_tool_end(
|
||||||
self.run_context, func_tool_name, func_tool_args, None
|
self.run_context, func_tool, func_tool_args, None
|
||||||
)
|
)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.error(
|
logger.error(f"Error in on_tool_end hook: {e}", exc_info=True)
|
||||||
f"Error in on_tool_end hook: {e}", exc_info=True
|
|
||||||
)
|
|
||||||
|
|
||||||
self.run_context.event.clear_result()
|
self.run_context.event.clear_result()
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
from dataclasses import dataclass
|
from dataclasses import dataclass
|
||||||
from deprecated import deprecated
|
from deprecated import deprecated
|
||||||
from typing import Awaitable, Literal, Any, Optional
|
from typing import Awaitable, Callable, Literal, Any, Optional
|
||||||
from .mcp_client import MCPClient
|
from .mcp_client import MCPClient
|
||||||
|
|
||||||
|
|
||||||
@@ -8,10 +8,10 @@ from .mcp_client import MCPClient
|
|||||||
class FunctionTool:
|
class FunctionTool:
|
||||||
"""A class representing a function tool that can be used in function calling."""
|
"""A class representing a function tool that can be used in function calling."""
|
||||||
|
|
||||||
name: str | None = None
|
name: str
|
||||||
parameters: dict | None = None
|
parameters: dict | None = None
|
||||||
description: str | None = None
|
description: str | None = None
|
||||||
handler: Awaitable | None = None
|
handler: Callable[..., Awaitable[Any]] | None = None
|
||||||
"""处理函数, 当 origin 为 mcp 时,这个为空"""
|
"""处理函数, 当 origin 为 mcp 时,这个为空"""
|
||||||
handler_module_path: str | None = None
|
handler_module_path: str | None = None
|
||||||
"""处理函数的模块路径,当 origin 为 mcp 时,这个为空
|
"""处理函数的模块路径,当 origin 为 mcp 时,这个为空
|
||||||
@@ -51,7 +51,7 @@ class ToolSet:
|
|||||||
This class provides methods to add, remove, and retrieve tools, as well as
|
This class provides methods to add, remove, and retrieve tools, as well as
|
||||||
convert the tools to different API formats (OpenAI, Anthropic, Google GenAI)."""
|
convert the tools to different API formats (OpenAI, Anthropic, Google GenAI)."""
|
||||||
|
|
||||||
def __init__(self, tools: list[FunctionTool] = None):
|
def __init__(self, tools: list[FunctionTool] | None = None):
|
||||||
self.tools: list[FunctionTool] = tools or []
|
self.tools: list[FunctionTool] = tools or []
|
||||||
|
|
||||||
def empty(self) -> bool:
|
def empty(self) -> bool:
|
||||||
@@ -79,7 +79,13 @@ class ToolSet:
|
|||||||
return None
|
return None
|
||||||
|
|
||||||
@deprecated(reason="Use add_tool() instead", version="4.0.0")
|
@deprecated(reason="Use add_tool() instead", version="4.0.0")
|
||||||
def add_func(self, name: str, func_args: list, desc: str, handler: Awaitable):
|
def add_func(
|
||||||
|
self,
|
||||||
|
name: str,
|
||||||
|
func_args: list,
|
||||||
|
desc: str,
|
||||||
|
handler: Callable[..., Awaitable[Any]],
|
||||||
|
):
|
||||||
"""Add a function tool to the set."""
|
"""Add a function tool to the set."""
|
||||||
params = {
|
params = {
|
||||||
"type": "object", # hard-coded here
|
"type": "object", # hard-coded here
|
||||||
@@ -104,7 +110,7 @@ class ToolSet:
|
|||||||
self.remove_tool(name)
|
self.remove_tool(name)
|
||||||
|
|
||||||
@deprecated(reason="Use get_tool() instead", version="4.0.0")
|
@deprecated(reason="Use get_tool() instead", version="4.0.0")
|
||||||
def get_func(self, name: str) -> list[FunctionTool]:
|
def get_func(self, name: str) -> FunctionTool | None:
|
||||||
"""Get all function tools."""
|
"""Get all function tools."""
|
||||||
return self.get_tool(name)
|
return self.get_tool(name)
|
||||||
|
|
||||||
@@ -125,7 +131,11 @@ class ToolSet:
|
|||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
if tool.parameters.get("properties") or not omit_empty_parameter_field:
|
if (
|
||||||
|
tool.parameters
|
||||||
|
and tool.parameters.get("properties")
|
||||||
|
or not omit_empty_parameter_field
|
||||||
|
):
|
||||||
func_def["function"]["parameters"] = tool.parameters
|
func_def["function"]["parameters"] = tool.parameters
|
||||||
|
|
||||||
result.append(func_def)
|
result.append(func_def)
|
||||||
@@ -135,14 +145,14 @@ class ToolSet:
|
|||||||
"""Convert tools to Anthropic API format."""
|
"""Convert tools to Anthropic API format."""
|
||||||
result = []
|
result = []
|
||||||
for tool in self.tools:
|
for tool in self.tools:
|
||||||
|
input_schema = {"type": "object"}
|
||||||
|
if tool.parameters:
|
||||||
|
input_schema["properties"] = tool.parameters.get("properties", {})
|
||||||
|
input_schema["required"] = tool.parameters.get("required", [])
|
||||||
tool_def = {
|
tool_def = {
|
||||||
"name": tool.name,
|
"name": tool.name,
|
||||||
"description": tool.description,
|
"description": tool.description,
|
||||||
"input_schema": {
|
"input_schema": input_schema,
|
||||||
"type": "object",
|
|
||||||
"properties": tool.parameters.get("properties", {}),
|
|
||||||
"required": tool.parameters.get("required", []),
|
|
||||||
},
|
|
||||||
}
|
}
|
||||||
result.append(tool_def)
|
result.append(tool_def)
|
||||||
return result
|
return result
|
||||||
@@ -210,14 +220,15 @@ class ToolSet:
|
|||||||
|
|
||||||
return result
|
return result
|
||||||
|
|
||||||
tools = [
|
tools = []
|
||||||
{
|
for tool in self.tools:
|
||||||
|
d = {
|
||||||
"name": tool.name,
|
"name": tool.name,
|
||||||
"description": tool.description,
|
"description": tool.description,
|
||||||
"parameters": convert_schema(tool.parameters),
|
|
||||||
}
|
}
|
||||||
for tool in self.tools
|
if tool.parameters:
|
||||||
]
|
d["parameters"] = convert_schema(tool.parameters)
|
||||||
|
tools.append(d)
|
||||||
|
|
||||||
declarations = {}
|
declarations = {}
|
||||||
if tools:
|
if tools:
|
||||||
|
|||||||
@@ -6,7 +6,7 @@ import os
|
|||||||
|
|
||||||
from astrbot.core.utils.astrbot_path import get_astrbot_data_path
|
from astrbot.core.utils.astrbot_path import get_astrbot_data_path
|
||||||
|
|
||||||
VERSION = "4.0.0"
|
VERSION = "4.3.0"
|
||||||
DB_PATH = os.path.join(get_astrbot_data_path(), "data_v4.db")
|
DB_PATH = os.path.join(get_astrbot_data_path(), "data_v4.db")
|
||||||
|
|
||||||
# 默认配置
|
# 默认配置
|
||||||
@@ -56,14 +56,15 @@ DEFAULT_CONFIG = {
|
|||||||
"wake_prefix": "",
|
"wake_prefix": "",
|
||||||
"web_search": False,
|
"web_search": False,
|
||||||
"websearch_provider": "default",
|
"websearch_provider": "default",
|
||||||
"websearch_tavily_key": "",
|
"websearch_tavily_key": [],
|
||||||
"web_search_link": False,
|
"web_search_link": False,
|
||||||
"display_reasoning_text": False,
|
"display_reasoning_text": False,
|
||||||
"identifier": False,
|
"identifier": False,
|
||||||
|
"group_name_display": False,
|
||||||
"datetime_system_prompt": True,
|
"datetime_system_prompt": True,
|
||||||
"default_personality": "default",
|
"default_personality": "default",
|
||||||
"persona_pool": ["*"],
|
"persona_pool": ["*"],
|
||||||
"prompt_prefix": "",
|
"prompt_prefix": "{{prompt}}",
|
||||||
"max_context_length": -1,
|
"max_context_length": -1,
|
||||||
"dequeue_context_length": 1,
|
"dequeue_context_length": 1,
|
||||||
"streaming_response": False,
|
"streaming_response": False,
|
||||||
@@ -115,6 +116,15 @@ DEFAULT_CONFIG = {
|
|||||||
"port": 6185,
|
"port": 6185,
|
||||||
},
|
},
|
||||||
"platform": [],
|
"platform": [],
|
||||||
|
"platform_specific": {
|
||||||
|
# 平台特异配置:按平台分类,平台下按功能分组
|
||||||
|
"lark": {
|
||||||
|
"pre_ack_emoji": {"enable": False, "emojis": ["Typing"]},
|
||||||
|
},
|
||||||
|
"telegram": {
|
||||||
|
"pre_ack_emoji": {"enable": False, "emojis": ["✍️"]},
|
||||||
|
},
|
||||||
|
},
|
||||||
"wake_prefix": ["/"],
|
"wake_prefix": ["/"],
|
||||||
"log_level": "INFO",
|
"log_level": "INFO",
|
||||||
"pip_install_arg": "",
|
"pip_install_arg": "",
|
||||||
@@ -235,6 +245,16 @@ CONFIG_METADATA_2 = {
|
|||||||
"discord_guild_id_for_debug": "",
|
"discord_guild_id_for_debug": "",
|
||||||
"discord_activity_name": "",
|
"discord_activity_name": "",
|
||||||
},
|
},
|
||||||
|
"Misskey": {
|
||||||
|
"id": "misskey",
|
||||||
|
"type": "misskey",
|
||||||
|
"enable": False,
|
||||||
|
"misskey_instance_url": "https://misskey.example",
|
||||||
|
"misskey_token": "",
|
||||||
|
"misskey_default_visibility": "public",
|
||||||
|
"misskey_local_only": False,
|
||||||
|
"misskey_enable_chat": True,
|
||||||
|
},
|
||||||
"Slack": {
|
"Slack": {
|
||||||
"id": "slack",
|
"id": "slack",
|
||||||
"type": "slack",
|
"type": "slack",
|
||||||
@@ -252,7 +272,7 @@ CONFIG_METADATA_2 = {
|
|||||||
"type": "satori",
|
"type": "satori",
|
||||||
"enable": False,
|
"enable": False,
|
||||||
"satori_api_base_url": "http://localhost:5140/satori/v1",
|
"satori_api_base_url": "http://localhost:5140/satori/v1",
|
||||||
"satori_endpoint": "ws://127.0.0.1:5140/satori/v1/events",
|
"satori_endpoint": "ws://localhost:5140/satori/v1/events",
|
||||||
"satori_token": "",
|
"satori_token": "",
|
||||||
"satori_auto_reconnect": True,
|
"satori_auto_reconnect": True,
|
||||||
"satori_heartbeat_interval": 10,
|
"satori_heartbeat_interval": 10,
|
||||||
@@ -261,34 +281,34 @@ CONFIG_METADATA_2 = {
|
|||||||
},
|
},
|
||||||
"items": {
|
"items": {
|
||||||
"satori_api_base_url": {
|
"satori_api_base_url": {
|
||||||
"description": "Satori API Base URL",
|
"description": "Satori API 终结点",
|
||||||
"type": "string",
|
"type": "string",
|
||||||
"hint": "The base URL for the Satori API.",
|
"hint": "Satori API 的基础地址。",
|
||||||
},
|
},
|
||||||
"satori_endpoint": {
|
"satori_endpoint": {
|
||||||
"description": "Satori WebSocket Endpoint",
|
"description": "Satori WebSocket 终结点",
|
||||||
"type": "string",
|
"type": "string",
|
||||||
"hint": "The WebSocket endpoint for Satori events.",
|
"hint": "Satori 事件的 WebSocket 端点。",
|
||||||
},
|
},
|
||||||
"satori_token": {
|
"satori_token": {
|
||||||
"description": "Satori Token",
|
"description": "Satori 令牌",
|
||||||
"type": "string",
|
"type": "string",
|
||||||
"hint": "The token used for authenticating with the Satori API.",
|
"hint": "用于 Satori API 身份验证的令牌。",
|
||||||
},
|
},
|
||||||
"satori_auto_reconnect": {
|
"satori_auto_reconnect": {
|
||||||
"description": "Enable Auto Reconnect",
|
"description": "启用自动重连",
|
||||||
"type": "bool",
|
"type": "bool",
|
||||||
"hint": "Whether to automatically reconnect the WebSocket on disconnection.",
|
"hint": "断开连接时是否自动重新连接 WebSocket。",
|
||||||
},
|
},
|
||||||
"satori_heartbeat_interval": {
|
"satori_heartbeat_interval": {
|
||||||
"description": "Satori Heartbeat Interval",
|
"description": "Satori 心跳间隔",
|
||||||
"type": "int",
|
"type": "int",
|
||||||
"hint": "The interval (in seconds) for sending heartbeat messages.",
|
"hint": "发送心跳消息的间隔(秒)。",
|
||||||
},
|
},
|
||||||
"satori_reconnect_delay": {
|
"satori_reconnect_delay": {
|
||||||
"description": "Satori Reconnect Delay",
|
"description": "Satori 重连延迟",
|
||||||
"type": "int",
|
"type": "int",
|
||||||
"hint": "The delay (in seconds) before attempting to reconnect.",
|
"hint": "尝试重新连接前的延迟时间(秒)。",
|
||||||
},
|
},
|
||||||
"slack_connection_mode": {
|
"slack_connection_mode": {
|
||||||
"description": "Slack Connection Mode",
|
"description": "Slack Connection Mode",
|
||||||
@@ -336,6 +356,32 @@ CONFIG_METADATA_2 = {
|
|||||||
"type": "string",
|
"type": "string",
|
||||||
"hint": "如果你的网络环境为中国大陆,请在 `其他配置` 处设置代理或更改 api_base。",
|
"hint": "如果你的网络环境为中国大陆,请在 `其他配置` 处设置代理或更改 api_base。",
|
||||||
},
|
},
|
||||||
|
"misskey_instance_url": {
|
||||||
|
"description": "Misskey 实例 URL",
|
||||||
|
"type": "string",
|
||||||
|
"hint": "例如 https://misskey.example,填写 Bot 账号所在的 Misskey 实例地址",
|
||||||
|
},
|
||||||
|
"misskey_token": {
|
||||||
|
"description": "Misskey Access Token",
|
||||||
|
"type": "string",
|
||||||
|
"hint": "连接服务设置生成的 API 鉴权访问令牌(Access token)",
|
||||||
|
},
|
||||||
|
"misskey_default_visibility": {
|
||||||
|
"description": "默认帖子可见性",
|
||||||
|
"type": "string",
|
||||||
|
"options": ["public", "home", "followers"],
|
||||||
|
"hint": "机器人发帖时的默认可见性设置。public:公开,home:主页时间线,followers:仅关注者。",
|
||||||
|
},
|
||||||
|
"misskey_local_only": {
|
||||||
|
"description": "仅限本站(不参与联合)",
|
||||||
|
"type": "bool",
|
||||||
|
"hint": "启用后,机器人发出的帖子将仅在本实例可见,不会联合到其他实例",
|
||||||
|
},
|
||||||
|
"misskey_enable_chat": {
|
||||||
|
"description": "启用聊天消息响应",
|
||||||
|
"type": "bool",
|
||||||
|
"hint": "启用后,机器人将会监听和响应私信聊天消息",
|
||||||
|
},
|
||||||
"telegram_command_register": {
|
"telegram_command_register": {
|
||||||
"description": "Telegram 命令注册",
|
"description": "Telegram 命令注册",
|
||||||
"type": "bool",
|
"type": "bool",
|
||||||
@@ -599,6 +645,7 @@ CONFIG_METADATA_2 = {
|
|||||||
"api_base": "https://api.openai.com/v1",
|
"api_base": "https://api.openai.com/v1",
|
||||||
"timeout": 120,
|
"timeout": 120,
|
||||||
"model_config": {"model": "gpt-4o-mini", "temperature": 0.4},
|
"model_config": {"model": "gpt-4o-mini", "temperature": 0.4},
|
||||||
|
"custom_extra_body": {},
|
||||||
"modalities": ["text", "image", "tool_use"],
|
"modalities": ["text", "image", "tool_use"],
|
||||||
"hint": "也兼容所有与 OpenAI API 兼容的服务。",
|
"hint": "也兼容所有与 OpenAI API 兼容的服务。",
|
||||||
},
|
},
|
||||||
@@ -613,6 +660,7 @@ CONFIG_METADATA_2 = {
|
|||||||
"api_base": "",
|
"api_base": "",
|
||||||
"timeout": 120,
|
"timeout": 120,
|
||||||
"model_config": {"model": "gpt-4o-mini", "temperature": 0.4},
|
"model_config": {"model": "gpt-4o-mini", "temperature": 0.4},
|
||||||
|
"custom_extra_body": {},
|
||||||
"modalities": ["text", "image", "tool_use"],
|
"modalities": ["text", "image", "tool_use"],
|
||||||
},
|
},
|
||||||
"xAI": {
|
"xAI": {
|
||||||
@@ -625,6 +673,7 @@ CONFIG_METADATA_2 = {
|
|||||||
"api_base": "https://api.x.ai/v1",
|
"api_base": "https://api.x.ai/v1",
|
||||||
"timeout": 120,
|
"timeout": 120,
|
||||||
"model_config": {"model": "grok-2-latest", "temperature": 0.4},
|
"model_config": {"model": "grok-2-latest", "temperature": 0.4},
|
||||||
|
"custom_extra_body": {},
|
||||||
"modalities": ["text", "image", "tool_use"],
|
"modalities": ["text", "image", "tool_use"],
|
||||||
},
|
},
|
||||||
"Anthropic": {
|
"Anthropic": {
|
||||||
@@ -654,6 +703,7 @@ CONFIG_METADATA_2 = {
|
|||||||
"key": ["ollama"], # ollama 的 key 默认是 ollama
|
"key": ["ollama"], # ollama 的 key 默认是 ollama
|
||||||
"api_base": "http://localhost:11434/v1",
|
"api_base": "http://localhost:11434/v1",
|
||||||
"model_config": {"model": "llama3.1-8b", "temperature": 0.4},
|
"model_config": {"model": "llama3.1-8b", "temperature": 0.4},
|
||||||
|
"custom_extra_body": {},
|
||||||
"modalities": ["text", "image", "tool_use"],
|
"modalities": ["text", "image", "tool_use"],
|
||||||
},
|
},
|
||||||
"LM Studio": {
|
"LM Studio": {
|
||||||
@@ -667,6 +717,7 @@ CONFIG_METADATA_2 = {
|
|||||||
"model_config": {
|
"model_config": {
|
||||||
"model": "llama-3.1-8b",
|
"model": "llama-3.1-8b",
|
||||||
},
|
},
|
||||||
|
"custom_extra_body": {},
|
||||||
"modalities": ["text", "image", "tool_use"],
|
"modalities": ["text", "image", "tool_use"],
|
||||||
},
|
},
|
||||||
"Gemini(OpenAI兼容)": {
|
"Gemini(OpenAI兼容)": {
|
||||||
@@ -682,6 +733,7 @@ CONFIG_METADATA_2 = {
|
|||||||
"model": "gemini-1.5-flash",
|
"model": "gemini-1.5-flash",
|
||||||
"temperature": 0.4,
|
"temperature": 0.4,
|
||||||
},
|
},
|
||||||
|
"custom_extra_body": {},
|
||||||
"modalities": ["text", "image", "tool_use"],
|
"modalities": ["text", "image", "tool_use"],
|
||||||
},
|
},
|
||||||
"Gemini": {
|
"Gemini": {
|
||||||
@@ -722,6 +774,7 @@ CONFIG_METADATA_2 = {
|
|||||||
"api_base": "https://api.deepseek.com/v1",
|
"api_base": "https://api.deepseek.com/v1",
|
||||||
"timeout": 120,
|
"timeout": 120,
|
||||||
"model_config": {"model": "deepseek-chat", "temperature": 0.4},
|
"model_config": {"model": "deepseek-chat", "temperature": 0.4},
|
||||||
|
"custom_extra_body": {},
|
||||||
"modalities": ["text", "image", "tool_use"],
|
"modalities": ["text", "image", "tool_use"],
|
||||||
},
|
},
|
||||||
"302.AI": {
|
"302.AI": {
|
||||||
@@ -734,6 +787,7 @@ CONFIG_METADATA_2 = {
|
|||||||
"api_base": "https://api.302.ai/v1",
|
"api_base": "https://api.302.ai/v1",
|
||||||
"timeout": 120,
|
"timeout": 120,
|
||||||
"model_config": {"model": "gpt-4.1-mini", "temperature": 0.4},
|
"model_config": {"model": "gpt-4.1-mini", "temperature": 0.4},
|
||||||
|
"custom_extra_body": {},
|
||||||
"modalities": ["text", "image", "tool_use"],
|
"modalities": ["text", "image", "tool_use"],
|
||||||
},
|
},
|
||||||
"硅基流动": {
|
"硅基流动": {
|
||||||
@@ -749,6 +803,7 @@ CONFIG_METADATA_2 = {
|
|||||||
"model": "deepseek-ai/DeepSeek-V3",
|
"model": "deepseek-ai/DeepSeek-V3",
|
||||||
"temperature": 0.4,
|
"temperature": 0.4,
|
||||||
},
|
},
|
||||||
|
"custom_extra_body": {},
|
||||||
"modalities": ["text", "image", "tool_use"],
|
"modalities": ["text", "image", "tool_use"],
|
||||||
},
|
},
|
||||||
"PPIO派欧云": {
|
"PPIO派欧云": {
|
||||||
@@ -764,6 +819,7 @@ CONFIG_METADATA_2 = {
|
|||||||
"model": "deepseek/deepseek-r1",
|
"model": "deepseek/deepseek-r1",
|
||||||
"temperature": 0.4,
|
"temperature": 0.4,
|
||||||
},
|
},
|
||||||
|
"custom_extra_body": {},
|
||||||
},
|
},
|
||||||
"优云智算": {
|
"优云智算": {
|
||||||
"id": "compshare",
|
"id": "compshare",
|
||||||
@@ -777,6 +833,7 @@ CONFIG_METADATA_2 = {
|
|||||||
"model_config": {
|
"model_config": {
|
||||||
"model": "moonshotai/Kimi-K2-Instruct",
|
"model": "moonshotai/Kimi-K2-Instruct",
|
||||||
},
|
},
|
||||||
|
"custom_extra_body": {},
|
||||||
"modalities": ["text", "image", "tool_use"],
|
"modalities": ["text", "image", "tool_use"],
|
||||||
},
|
},
|
||||||
"Kimi": {
|
"Kimi": {
|
||||||
@@ -789,6 +846,7 @@ CONFIG_METADATA_2 = {
|
|||||||
"timeout": 120,
|
"timeout": 120,
|
||||||
"api_base": "https://api.moonshot.cn/v1",
|
"api_base": "https://api.moonshot.cn/v1",
|
||||||
"model_config": {"model": "moonshot-v1-8k", "temperature": 0.4},
|
"model_config": {"model": "moonshot-v1-8k", "temperature": 0.4},
|
||||||
|
"custom_extra_body": {},
|
||||||
"modalities": ["text", "image", "tool_use"],
|
"modalities": ["text", "image", "tool_use"],
|
||||||
},
|
},
|
||||||
"智谱 AI": {
|
"智谱 AI": {
|
||||||
@@ -820,6 +878,18 @@ CONFIG_METADATA_2 = {
|
|||||||
"timeout": 60,
|
"timeout": 60,
|
||||||
"hint": "请确保你在 AstrBot 里设置的 APP 类型和 Dify 里面创建的应用的类型一致!",
|
"hint": "请确保你在 AstrBot 里设置的 APP 类型和 Dify 里面创建的应用的类型一致!",
|
||||||
},
|
},
|
||||||
|
"Coze": {
|
||||||
|
"id": "coze",
|
||||||
|
"provider": "coze",
|
||||||
|
"provider_type": "chat_completion",
|
||||||
|
"type": "coze",
|
||||||
|
"enable": True,
|
||||||
|
"coze_api_key": "",
|
||||||
|
"bot_id": "",
|
||||||
|
"coze_api_base": "https://api.coze.cn",
|
||||||
|
"timeout": 60,
|
||||||
|
"auto_save_history": True,
|
||||||
|
},
|
||||||
"阿里云百炼应用": {
|
"阿里云百炼应用": {
|
||||||
"id": "dashscope",
|
"id": "dashscope",
|
||||||
"provider": "dashscope",
|
"provider": "dashscope",
|
||||||
@@ -847,6 +917,7 @@ CONFIG_METADATA_2 = {
|
|||||||
"timeout": 120,
|
"timeout": 120,
|
||||||
"api_base": "https://api-inference.modelscope.cn/v1",
|
"api_base": "https://api-inference.modelscope.cn/v1",
|
||||||
"model_config": {"model": "Qwen/Qwen3-32B", "temperature": 0.4},
|
"model_config": {"model": "Qwen/Qwen3-32B", "temperature": 0.4},
|
||||||
|
"custom_extra_body": {},
|
||||||
"modalities": ["text", "image", "tool_use"],
|
"modalities": ["text", "image", "tool_use"],
|
||||||
},
|
},
|
||||||
"FastGPT": {
|
"FastGPT": {
|
||||||
@@ -858,6 +929,7 @@ CONFIG_METADATA_2 = {
|
|||||||
"key": [],
|
"key": [],
|
||||||
"api_base": "https://api.fastgpt.in/api/v1",
|
"api_base": "https://api.fastgpt.in/api/v1",
|
||||||
"timeout": 60,
|
"timeout": 60,
|
||||||
|
"custom_extra_body": {},
|
||||||
},
|
},
|
||||||
"Whisper(API)": {
|
"Whisper(API)": {
|
||||||
"id": "whisper",
|
"id": "whisper",
|
||||||
@@ -1102,6 +1174,12 @@ CONFIG_METADATA_2 = {
|
|||||||
"render_type": "checkbox",
|
"render_type": "checkbox",
|
||||||
"hint": "模型支持的模态。如所填写的模型不支持图像,请取消勾选图像。",
|
"hint": "模型支持的模态。如所填写的模型不支持图像,请取消勾选图像。",
|
||||||
},
|
},
|
||||||
|
"custom_extra_body": {
|
||||||
|
"description": "自定义请求体参数",
|
||||||
|
"type": "dict",
|
||||||
|
"items": {},
|
||||||
|
"hint": "此处添加的键值对将被合并到发送给 API 的 extra_body 中。值可以是字符串、数字或布尔值。",
|
||||||
|
},
|
||||||
"provider": {
|
"provider": {
|
||||||
"type": "string",
|
"type": "string",
|
||||||
"invisible": True,
|
"invisible": True,
|
||||||
@@ -1678,6 +1756,26 @@ CONFIG_METADATA_2 = {
|
|||||||
"hint": "发送的消息文本内容对应的输入变量名。默认为 astrbot_text_query。",
|
"hint": "发送的消息文本内容对应的输入变量名。默认为 astrbot_text_query。",
|
||||||
"obvious": True,
|
"obvious": True,
|
||||||
},
|
},
|
||||||
|
"coze_api_key": {
|
||||||
|
"description": "Coze API Key",
|
||||||
|
"type": "string",
|
||||||
|
"hint": "Coze API 密钥,用于访问 Coze 服务。",
|
||||||
|
},
|
||||||
|
"bot_id": {
|
||||||
|
"description": "Bot ID",
|
||||||
|
"type": "string",
|
||||||
|
"hint": "Coze 机器人的 ID,在 Coze 平台上创建机器人后获得。",
|
||||||
|
},
|
||||||
|
"coze_api_base": {
|
||||||
|
"description": "API Base URL",
|
||||||
|
"type": "string",
|
||||||
|
"hint": "Coze API 的基础 URL 地址,默认为 https://api.coze.cn",
|
||||||
|
},
|
||||||
|
"auto_save_history": {
|
||||||
|
"description": "由 Coze 管理对话记录",
|
||||||
|
"type": "bool",
|
||||||
|
"hint": "启用后,将由 Coze 进行对话历史记录管理, 此时 AstrBot 本地保存的上下文不会生效(仅供浏览), 对 AstrBot 的上下文进行的操作也不会生效。如果为禁用, 则使用 AstrBot 管理上下文。",
|
||||||
|
},
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
"provider_settings": {
|
"provider_settings": {
|
||||||
@@ -1704,6 +1802,9 @@ CONFIG_METADATA_2 = {
|
|||||||
"identifier": {
|
"identifier": {
|
||||||
"type": "bool",
|
"type": "bool",
|
||||||
},
|
},
|
||||||
|
"group_name_display": {
|
||||||
|
"type": "bool",
|
||||||
|
},
|
||||||
"datetime_system_prompt": {
|
"datetime_system_prompt": {
|
||||||
"type": "bool",
|
"type": "bool",
|
||||||
},
|
},
|
||||||
@@ -1883,17 +1984,33 @@ CONFIG_METADATA_3 = {
|
|||||||
"_special": "select_provider",
|
"_special": "select_provider",
|
||||||
"hint": "留空代表不使用。可用于不支持视觉模态的聊天模型。",
|
"hint": "留空代表不使用。可用于不支持视觉模态的聊天模型。",
|
||||||
},
|
},
|
||||||
|
"provider_stt_settings.enable": {
|
||||||
|
"description": "启用语音转文本",
|
||||||
|
"type": "bool",
|
||||||
|
"hint": "STT 总开关。",
|
||||||
|
},
|
||||||
"provider_stt_settings.provider_id": {
|
"provider_stt_settings.provider_id": {
|
||||||
"description": "语音转文本模型",
|
"description": "默认语音转文本模型",
|
||||||
"type": "string",
|
"type": "string",
|
||||||
"hint": "留空代表不使用。",
|
"hint": "用户也可使用 /provider 指令单独选择会话的 STT 模型。",
|
||||||
"_special": "select_provider_stt",
|
"_special": "select_provider_stt",
|
||||||
|
"condition": {
|
||||||
|
"provider_stt_settings.enable": True,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
"provider_tts_settings.enable": {
|
||||||
|
"description": "启用文本转语音",
|
||||||
|
"type": "bool",
|
||||||
|
"hint": "TTS 总开关。当关闭时,会话启用 TTS 也不会生效。",
|
||||||
},
|
},
|
||||||
"provider_tts_settings.provider_id": {
|
"provider_tts_settings.provider_id": {
|
||||||
"description": "文本转语音模型",
|
"description": "默认文本转语音模型",
|
||||||
"type": "string",
|
"type": "string",
|
||||||
"hint": "留空代表不使用。",
|
"hint": "用户也可使用 /provider 单独选择会话的 TTS 模型。",
|
||||||
"_special": "select_provider_tts",
|
"_special": "select_provider_tts",
|
||||||
|
"condition": {
|
||||||
|
"provider_tts_settings.enable": True,
|
||||||
|
},
|
||||||
},
|
},
|
||||||
"provider_settings.image_caption_prompt": {
|
"provider_settings.image_caption_prompt": {
|
||||||
"description": "图片转述提示词",
|
"description": "图片转述提示词",
|
||||||
@@ -1938,7 +2055,9 @@ CONFIG_METADATA_3 = {
|
|||||||
},
|
},
|
||||||
"provider_settings.websearch_tavily_key": {
|
"provider_settings.websearch_tavily_key": {
|
||||||
"description": "Tavily API Key",
|
"description": "Tavily API Key",
|
||||||
"type": "string",
|
"type": "list",
|
||||||
|
"items": {"type": "string"},
|
||||||
|
"hint": "可添加多个 Key 进行轮询。",
|
||||||
"condition": {
|
"condition": {
|
||||||
"provider_settings.websearch_provider": "tavily",
|
"provider_settings.websearch_provider": "tavily",
|
||||||
},
|
},
|
||||||
@@ -1961,6 +2080,11 @@ CONFIG_METADATA_3 = {
|
|||||||
"description": "用户识别",
|
"description": "用户识别",
|
||||||
"type": "bool",
|
"type": "bool",
|
||||||
},
|
},
|
||||||
|
"provider_settings.group_name_display": {
|
||||||
|
"description": "显示群名称",
|
||||||
|
"type": "bool",
|
||||||
|
"hint": "启用后,在支持的平台(aiocqhttp)上会在 prompt 中包含群名称信息。",
|
||||||
|
},
|
||||||
"provider_settings.datetime_system_prompt": {
|
"provider_settings.datetime_system_prompt": {
|
||||||
"description": "现实世界时间感知",
|
"description": "现实世界时间感知",
|
||||||
"type": "bool",
|
"type": "bool",
|
||||||
@@ -1994,12 +2118,14 @@ CONFIG_METADATA_3 = {
|
|||||||
"provider_settings.wake_prefix": {
|
"provider_settings.wake_prefix": {
|
||||||
"description": "LLM 聊天额外唤醒前缀 ",
|
"description": "LLM 聊天额外唤醒前缀 ",
|
||||||
"type": "string",
|
"type": "string",
|
||||||
|
"hint": "例子: 如果唤醒前缀为 `/`, 额外聊天唤醒前缀为 `chat`,则需要 `/chat` 才会触发 LLM 请求。默认为空。",
|
||||||
},
|
},
|
||||||
"provider_settings.prompt_prefix": {
|
"provider_settings.prompt_prefix": {
|
||||||
"description": "额外前缀提示词",
|
"description": "用户提示词",
|
||||||
"type": "string",
|
"type": "string",
|
||||||
|
"hint": "可使用 {{prompt}} 作为用户输入的占位符。如果不输入占位符则代表添加在用户输入的前面。",
|
||||||
},
|
},
|
||||||
"provider_settings.dual_output": {
|
"provider_tts_settings.dual_output": {
|
||||||
"description": "开启 TTS 时同时输出语音和文字内容",
|
"description": "开启 TTS 时同时输出语音和文字内容",
|
||||||
"type": "bool",
|
"type": "bool",
|
||||||
},
|
},
|
||||||
@@ -2108,41 +2234,41 @@ CONFIG_METADATA_3 = {
|
|||||||
"description": "内容安全",
|
"description": "内容安全",
|
||||||
"type": "object",
|
"type": "object",
|
||||||
"items": {
|
"items": {
|
||||||
"platform_settings.content_safety.also_use_in_response": {
|
"content_safety.also_use_in_response": {
|
||||||
"description": "同时检查模型的响应内容",
|
"description": "同时检查模型的响应内容",
|
||||||
"type": "bool",
|
"type": "bool",
|
||||||
},
|
},
|
||||||
"platform_settings.content_safety.baidu_aip.enable": {
|
"content_safety.baidu_aip.enable": {
|
||||||
"description": "使用百度内容安全审核",
|
"description": "使用百度内容安全审核",
|
||||||
"type": "bool",
|
"type": "bool",
|
||||||
"hint": "您需要手动安装 baidu-aip 库。",
|
"hint": "您需要手动安装 baidu-aip 库。",
|
||||||
},
|
},
|
||||||
"platform_settings.content_safety.baidu_aip.app_id": {
|
"content_safety.baidu_aip.app_id": {
|
||||||
"description": "App ID",
|
"description": "App ID",
|
||||||
"type": "string",
|
"type": "string",
|
||||||
"condition": {
|
"condition": {
|
||||||
"platform_settings.content_safety.baidu_aip.enable": True,
|
"content_safety.baidu_aip.enable": True,
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
"platform_settings.content_safety.baidu_aip.api_key": {
|
"content_safety.baidu_aip.api_key": {
|
||||||
"description": "API Key",
|
"description": "API Key",
|
||||||
"type": "string",
|
"type": "string",
|
||||||
"condition": {
|
"condition": {
|
||||||
"platform_settings.content_safety.baidu_aip.enable": True,
|
"content_safety.baidu_aip.enable": True,
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
"platform_settings.content_safety.baidu_aip.secret_key": {
|
"content_safety.baidu_aip.secret_key": {
|
||||||
"description": "Secret Key",
|
"description": "Secret Key",
|
||||||
"type": "string",
|
"type": "string",
|
||||||
"condition": {
|
"condition": {
|
||||||
"platform_settings.content_safety.baidu_aip.enable": True,
|
"content_safety.baidu_aip.enable": True,
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
"platform_settings.content_safety.internal_keywords.enable": {
|
"content_safety.internal_keywords.enable": {
|
||||||
"description": "关键词检查",
|
"description": "关键词检查",
|
||||||
"type": "bool",
|
"type": "bool",
|
||||||
},
|
},
|
||||||
"platform_settings.content_safety.internal_keywords.extra_keywords": {
|
"content_safety.internal_keywords.extra_keywords": {
|
||||||
"description": "额外关键词",
|
"description": "额外关键词",
|
||||||
"type": "list",
|
"type": "list",
|
||||||
"items": {"type": "string"},
|
"items": {"type": "string"},
|
||||||
@@ -2180,6 +2306,32 @@ CONFIG_METADATA_3 = {
|
|||||||
"description": "用户权限不足时是否回复",
|
"description": "用户权限不足时是否回复",
|
||||||
"type": "bool",
|
"type": "bool",
|
||||||
},
|
},
|
||||||
|
"platform_specific.lark.pre_ack_emoji.enable": {
|
||||||
|
"description": "[飞书] 启用预回应表情",
|
||||||
|
"type": "bool",
|
||||||
|
},
|
||||||
|
"platform_specific.lark.pre_ack_emoji.emojis": {
|
||||||
|
"description": "表情列表(飞书表情枚举名)",
|
||||||
|
"type": "list",
|
||||||
|
"items": {"type": "string"},
|
||||||
|
"hint": "表情枚举名参考:https://open.feishu.cn/document/server-docs/im-v1/message-reaction/emojis-introduce",
|
||||||
|
"condition": {
|
||||||
|
"platform_specific.lark.pre_ack_emoji.enable": True,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
"platform_specific.telegram.pre_ack_emoji.enable": {
|
||||||
|
"description": "[Telegram] 启用预回应表情",
|
||||||
|
"type": "bool",
|
||||||
|
},
|
||||||
|
"platform_specific.telegram.pre_ack_emoji.emojis": {
|
||||||
|
"description": "表情列表(Unicode)",
|
||||||
|
"type": "list",
|
||||||
|
"items": {"type": "string"},
|
||||||
|
"hint": "Telegram 仅支持固定反应集合,参考:https://gist.github.com/Soulter/3f22c8e5f9c7e152e967e8bc28c97fc9",
|
||||||
|
"condition": {
|
||||||
|
"platform_specific.telegram.pre_ack_emoji.enable": True,
|
||||||
|
},
|
||||||
|
},
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
|
|||||||
@@ -87,14 +87,22 @@ class ConversationManager:
|
|||||||
unified_msg_origin (str): 统一的消息来源字符串。格式为 platform_name:message_type:session_id
|
unified_msg_origin (str): 统一的消息来源字符串。格式为 platform_name:message_type:session_id
|
||||||
conversation_id (str): 对话 ID, 是 uuid 格式的字符串
|
conversation_id (str): 对话 ID, 是 uuid 格式的字符串
|
||||||
"""
|
"""
|
||||||
f = False
|
|
||||||
if not conversation_id:
|
if not conversation_id:
|
||||||
conversation_id = self.session_conversations.get(unified_msg_origin)
|
conversation_id = self.session_conversations.get(unified_msg_origin)
|
||||||
if conversation_id:
|
|
||||||
f = True
|
|
||||||
if conversation_id:
|
if conversation_id:
|
||||||
await self.db.delete_conversation(cid=conversation_id)
|
await self.db.delete_conversation(cid=conversation_id)
|
||||||
if f:
|
curr_cid = await self.get_curr_conversation_id(unified_msg_origin)
|
||||||
|
if curr_cid == conversation_id:
|
||||||
|
self.session_conversations.pop(unified_msg_origin, None)
|
||||||
|
await sp.session_remove(unified_msg_origin, "sel_conv_id")
|
||||||
|
|
||||||
|
async def delete_conversations_by_user_id(self, unified_msg_origin: str):
|
||||||
|
"""删除会话的所有对话
|
||||||
|
|
||||||
|
Args:
|
||||||
|
unified_msg_origin (str): 统一的消息来源字符串。格式为 platform_name:message_type:session_id
|
||||||
|
"""
|
||||||
|
await self.db.delete_conversations_by_user_id(user_id=unified_msg_origin)
|
||||||
self.session_conversations.pop(unified_msg_origin, None)
|
self.session_conversations.pop(unified_msg_origin, None)
|
||||||
await sp.session_remove(unified_msg_origin, "sel_conv_id")
|
await sp.session_remove(unified_msg_origin, "sel_conv_id")
|
||||||
|
|
||||||
|
|||||||
@@ -154,12 +154,17 @@ class BaseDatabase(abc.ABC):
|
|||||||
"""Delete a conversation by its ID."""
|
"""Delete a conversation by its ID."""
|
||||||
...
|
...
|
||||||
|
|
||||||
|
@abc.abstractmethod
|
||||||
|
async def delete_conversations_by_user_id(self, user_id: str) -> None:
|
||||||
|
"""Delete all conversations for a specific user."""
|
||||||
|
...
|
||||||
|
|
||||||
@abc.abstractmethod
|
@abc.abstractmethod
|
||||||
async def insert_platform_message_history(
|
async def insert_platform_message_history(
|
||||||
self,
|
self,
|
||||||
platform_id: str,
|
platform_id: str,
|
||||||
user_id: str,
|
user_id: str,
|
||||||
content: list[dict],
|
content: dict,
|
||||||
sender_id: str | None = None,
|
sender_id: str | None = None,
|
||||||
sender_name: str | None = None,
|
sender_name: str | None = None,
|
||||||
) -> None:
|
) -> None:
|
||||||
@@ -282,3 +287,14 @@ class BaseDatabase(abc.ABC):
|
|||||||
# async def get_llm_messages(self, cid: str) -> list[LLMMessage]:
|
# async def get_llm_messages(self, cid: str) -> list[LLMMessage]:
|
||||||
# """Get all LLM messages for a specific conversation."""
|
# """Get all LLM messages for a specific conversation."""
|
||||||
# ...
|
# ...
|
||||||
|
|
||||||
|
@abc.abstractmethod
|
||||||
|
async def get_session_conversations(
|
||||||
|
self,
|
||||||
|
page: int = 1,
|
||||||
|
page_size: int = 20,
|
||||||
|
search_query: str | None = None,
|
||||||
|
platform: str | None = None,
|
||||||
|
) -> tuple[list[dict], int]:
|
||||||
|
"""Get paginated session conversations with joined conversation and persona details, support search and platform filter."""
|
||||||
|
...
|
||||||
|
|||||||
@@ -5,6 +5,7 @@ from astrbot.core.utils.astrbot_path import get_astrbot_data_path
|
|||||||
|
|
||||||
_VT = TypeVar("_VT")
|
_VT = TypeVar("_VT")
|
||||||
|
|
||||||
|
|
||||||
class SharedPreferences:
|
class SharedPreferences:
|
||||||
def __init__(self, path=None):
|
def __init__(self, path=None):
|
||||||
if path is None:
|
if path is None:
|
||||||
@@ -42,4 +43,5 @@ class SharedPreferences:
|
|||||||
self._data.clear()
|
self._data.clear()
|
||||||
self._save_preferences()
|
self._save_preferences()
|
||||||
|
|
||||||
|
|
||||||
sp = SharedPreferences()
|
sp = SharedPreferences()
|
||||||
|
|||||||
@@ -4,6 +4,7 @@ from astrbot.core.db.po import Platform, Stats
|
|||||||
from typing import Tuple, List, Dict, Any
|
from typing import Tuple, List, Dict, Any
|
||||||
from dataclasses import dataclass
|
from dataclasses import dataclass
|
||||||
|
|
||||||
|
|
||||||
@dataclass
|
@dataclass
|
||||||
class Conversation:
|
class Conversation:
|
||||||
"""LLM 对话存储
|
"""LLM 对话存储
|
||||||
@@ -76,7 +77,7 @@ PRAGMA encoding = 'UTF-8';
|
|||||||
"""
|
"""
|
||||||
|
|
||||||
|
|
||||||
class SQLiteDatabase():
|
class SQLiteDatabase:
|
||||||
def __init__(self, db_path: str) -> None:
|
def __init__(self, db_path: str) -> None:
|
||||||
super().__init__()
|
super().__init__()
|
||||||
self.db_path = db_path
|
self.db_path = db_path
|
||||||
|
|||||||
@@ -75,7 +75,9 @@ class Persona(SQLModel, table=True):
|
|||||||
|
|
||||||
__tablename__ = "personas"
|
__tablename__ = "personas"
|
||||||
|
|
||||||
id: int = Field(primary_key=True, sa_column_kwargs={"autoincrement": True})
|
id: int | None = Field(
|
||||||
|
primary_key=True, sa_column_kwargs={"autoincrement": True}, default=None
|
||||||
|
)
|
||||||
persona_id: str = Field(max_length=255, nullable=False)
|
persona_id: str = Field(max_length=255, nullable=False)
|
||||||
system_prompt: str = Field(sa_type=Text, nullable=False)
|
system_prompt: str = Field(sa_type=Text, nullable=False)
|
||||||
begin_dialogs: Optional[list] = Field(default=None, sa_type=JSON)
|
begin_dialogs: Optional[list] = Field(default=None, sa_type=JSON)
|
||||||
@@ -135,7 +137,9 @@ class PlatformMessageHistory(SQLModel, table=True):
|
|||||||
|
|
||||||
__tablename__ = "platform_message_history"
|
__tablename__ = "platform_message_history"
|
||||||
|
|
||||||
id: int = Field(primary_key=True, sa_column_kwargs={"autoincrement": True})
|
id: int | None = Field(
|
||||||
|
primary_key=True, sa_column_kwargs={"autoincrement": True}, default=None
|
||||||
|
)
|
||||||
platform_id: str = Field(nullable=False)
|
platform_id: str = Field(nullable=False)
|
||||||
user_id: str = Field(nullable=False) # An id of group, user in platform
|
user_id: str = Field(nullable=False) # An id of group, user in platform
|
||||||
sender_id: Optional[str] = Field(default=None) # ID of the sender in the platform
|
sender_id: Optional[str] = Field(default=None) # ID of the sender in the platform
|
||||||
@@ -158,8 +162,8 @@ class Attachment(SQLModel, table=True):
|
|||||||
|
|
||||||
__tablename__ = "attachments"
|
__tablename__ = "attachments"
|
||||||
|
|
||||||
inner_attachment_id: int = Field(
|
inner_attachment_id: int | None = Field(
|
||||||
primary_key=True, sa_column_kwargs={"autoincrement": True}
|
primary_key=True, sa_column_kwargs={"autoincrement": True}, default=None
|
||||||
)
|
)
|
||||||
attachment_id: str = Field(
|
attachment_id: str = Field(
|
||||||
max_length=36,
|
max_length=36,
|
||||||
|
|||||||
@@ -15,9 +15,8 @@ from astrbot.core.db.po import (
|
|||||||
SQLModel,
|
SQLModel,
|
||||||
)
|
)
|
||||||
|
|
||||||
from sqlalchemy import select, update, delete, text
|
from sqlmodel import select, update, delete, text, func, or_, desc, col
|
||||||
from sqlalchemy.ext.asyncio import AsyncSession
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||||||
from sqlalchemy.sql import func
|
|
||||||
|
|
||||||
NOT_GIVEN = T.TypeVar("NOT_GIVEN")
|
NOT_GIVEN = T.TypeVar("NOT_GIVEN")
|
||||||
|
|
||||||
@@ -41,10 +40,10 @@ class SQLiteDatabase(BaseDatabase):
|
|||||||
|
|
||||||
async def insert_platform_stats(
|
async def insert_platform_stats(
|
||||||
self,
|
self,
|
||||||
platform_id: str,
|
platform_id,
|
||||||
platform_type: str,
|
platform_type,
|
||||||
count: int = 1,
|
count=1,
|
||||||
timestamp: datetime = None,
|
timestamp=None,
|
||||||
) -> None:
|
) -> None:
|
||||||
"""Insert a new platform statistic record."""
|
"""Insert a new platform statistic record."""
|
||||||
async with self.get_db() as session:
|
async with self.get_db() as session:
|
||||||
@@ -75,7 +74,9 @@ class SQLiteDatabase(BaseDatabase):
|
|||||||
async with self.get_db() as session:
|
async with self.get_db() as session:
|
||||||
session: AsyncSession
|
session: AsyncSession
|
||||||
result = await session.execute(
|
result = await session.execute(
|
||||||
select(func.count(PlatformStat.platform_id)).select_from(PlatformStat)
|
select(func.count(col(PlatformStat.platform_id))).select_from(
|
||||||
|
PlatformStat
|
||||||
|
)
|
||||||
)
|
)
|
||||||
count = result.scalar_one_or_none()
|
count = result.scalar_one_or_none()
|
||||||
return count if count is not None else 0
|
return count if count is not None else 0
|
||||||
@@ -95,7 +96,7 @@ class SQLiteDatabase(BaseDatabase):
|
|||||||
"""),
|
"""),
|
||||||
{"start_time": start_time},
|
{"start_time": start_time},
|
||||||
)
|
)
|
||||||
return result.scalars().all()
|
return list(result.scalars().all())
|
||||||
|
|
||||||
# ====
|
# ====
|
||||||
# Conversation Management
|
# Conversation Management
|
||||||
@@ -111,7 +112,7 @@ class SQLiteDatabase(BaseDatabase):
|
|||||||
if platform_id:
|
if platform_id:
|
||||||
query = query.where(ConversationV2.platform_id == platform_id)
|
query = query.where(ConversationV2.platform_id == platform_id)
|
||||||
# order by
|
# order by
|
||||||
query = query.order_by(ConversationV2.created_at.desc())
|
query = query.order_by(desc(ConversationV2.created_at))
|
||||||
result = await session.execute(query)
|
result = await session.execute(query)
|
||||||
|
|
||||||
return result.scalars().all()
|
return result.scalars().all()
|
||||||
@@ -129,7 +130,7 @@ class SQLiteDatabase(BaseDatabase):
|
|||||||
offset = (page - 1) * page_size
|
offset = (page - 1) * page_size
|
||||||
result = await session.execute(
|
result = await session.execute(
|
||||||
select(ConversationV2)
|
select(ConversationV2)
|
||||||
.order_by(ConversationV2.created_at.desc())
|
.order_by(desc(ConversationV2.created_at))
|
||||||
.offset(offset)
|
.offset(offset)
|
||||||
.limit(page_size)
|
.limit(page_size)
|
||||||
)
|
)
|
||||||
@@ -150,11 +151,25 @@ class SQLiteDatabase(BaseDatabase):
|
|||||||
|
|
||||||
if platform_ids:
|
if platform_ids:
|
||||||
base_query = base_query.where(
|
base_query = base_query.where(
|
||||||
ConversationV2.platform_id.in_(platform_ids)
|
col(ConversationV2.platform_id).in_(platform_ids)
|
||||||
)
|
)
|
||||||
if search_query:
|
if search_query:
|
||||||
|
search_query = search_query.encode("unicode_escape").decode("utf-8")
|
||||||
base_query = base_query.where(
|
base_query = base_query.where(
|
||||||
ConversationV2.title.ilike(f"%{search_query}%")
|
or_(
|
||||||
|
col(ConversationV2.title).ilike(f"%{search_query}%"),
|
||||||
|
col(ConversationV2.content).ilike(f"%{search_query}%"),
|
||||||
|
col(ConversationV2.user_id).ilike(f"%{search_query}%"),
|
||||||
|
)
|
||||||
|
)
|
||||||
|
if "message_types" in kwargs and len(kwargs["message_types"]) > 0:
|
||||||
|
for msg_type in kwargs["message_types"]:
|
||||||
|
base_query = base_query.where(
|
||||||
|
col(ConversationV2.user_id).ilike(f"%:{msg_type}:%")
|
||||||
|
)
|
||||||
|
if "platforms" in kwargs and len(kwargs["platforms"]) > 0:
|
||||||
|
base_query = base_query.where(
|
||||||
|
col(ConversationV2.platform_id).in_(kwargs["platforms"])
|
||||||
)
|
)
|
||||||
|
|
||||||
# Get total count matching the filters
|
# Get total count matching the filters
|
||||||
@@ -165,7 +180,7 @@ class SQLiteDatabase(BaseDatabase):
|
|||||||
# Get paginated results
|
# Get paginated results
|
||||||
offset = (page - 1) * page_size
|
offset = (page - 1) * page_size
|
||||||
result_query = (
|
result_query = (
|
||||||
base_query.order_by(ConversationV2.created_at.desc())
|
base_query.order_by(desc(ConversationV2.created_at))
|
||||||
.offset(offset)
|
.offset(offset)
|
||||||
.limit(page_size)
|
.limit(page_size)
|
||||||
)
|
)
|
||||||
@@ -211,7 +226,7 @@ class SQLiteDatabase(BaseDatabase):
|
|||||||
session: AsyncSession
|
session: AsyncSession
|
||||||
async with session.begin():
|
async with session.begin():
|
||||||
query = update(ConversationV2).where(
|
query = update(ConversationV2).where(
|
||||||
ConversationV2.conversation_id == cid
|
col(ConversationV2.conversation_id) == cid
|
||||||
)
|
)
|
||||||
values = {}
|
values = {}
|
||||||
if title is not None:
|
if title is not None:
|
||||||
@@ -231,8 +246,125 @@ class SQLiteDatabase(BaseDatabase):
|
|||||||
session: AsyncSession
|
session: AsyncSession
|
||||||
async with session.begin():
|
async with session.begin():
|
||||||
await session.execute(
|
await session.execute(
|
||||||
delete(ConversationV2).where(ConversationV2.conversation_id == cid)
|
delete(ConversationV2).where(
|
||||||
|
col(ConversationV2.conversation_id) == cid
|
||||||
)
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
async def delete_conversations_by_user_id(self, user_id: str) -> None:
|
||||||
|
async with self.get_db() as session:
|
||||||
|
session: AsyncSession
|
||||||
|
async with session.begin():
|
||||||
|
await session.execute(
|
||||||
|
delete(ConversationV2).where(col(ConversationV2.user_id) == user_id)
|
||||||
|
)
|
||||||
|
|
||||||
|
async def get_session_conversations(
|
||||||
|
self,
|
||||||
|
page=1,
|
||||||
|
page_size=20,
|
||||||
|
search_query=None,
|
||||||
|
platform=None,
|
||||||
|
) -> tuple[list[dict], int]:
|
||||||
|
"""Get paginated session conversations with joined conversation and persona details."""
|
||||||
|
async with self.get_db() as session:
|
||||||
|
session: AsyncSession
|
||||||
|
offset = (page - 1) * page_size
|
||||||
|
|
||||||
|
base_query = (
|
||||||
|
select(
|
||||||
|
col(Preference.scope_id).label("session_id"),
|
||||||
|
func.json_extract(Preference.value, "$.val").label(
|
||||||
|
"conversation_id"
|
||||||
|
), # type: ignore
|
||||||
|
col(ConversationV2.persona_id).label("persona_id"),
|
||||||
|
col(ConversationV2.title).label("title"),
|
||||||
|
col(Persona.persona_id).label("persona_name"),
|
||||||
|
)
|
||||||
|
.select_from(Preference)
|
||||||
|
.outerjoin(
|
||||||
|
ConversationV2,
|
||||||
|
func.json_extract(Preference.value, "$.val")
|
||||||
|
== ConversationV2.conversation_id,
|
||||||
|
)
|
||||||
|
.outerjoin(
|
||||||
|
Persona, col(ConversationV2.persona_id) == Persona.persona_id
|
||||||
|
)
|
||||||
|
.where(Preference.scope == "umo", Preference.key == "sel_conv_id")
|
||||||
|
)
|
||||||
|
|
||||||
|
# 搜索筛选
|
||||||
|
if search_query:
|
||||||
|
search_pattern = f"%{search_query}%"
|
||||||
|
base_query = base_query.where(
|
||||||
|
or_(
|
||||||
|
col(Preference.scope_id).ilike(search_pattern),
|
||||||
|
col(ConversationV2.title).ilike(search_pattern),
|
||||||
|
col(Persona.persona_id).ilike(search_pattern),
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
# 平台筛选
|
||||||
|
if platform:
|
||||||
|
platform_pattern = f"{platform}:%"
|
||||||
|
base_query = base_query.where(
|
||||||
|
col(Preference.scope_id).like(platform_pattern)
|
||||||
|
)
|
||||||
|
|
||||||
|
# 排序
|
||||||
|
base_query = base_query.order_by(Preference.scope_id)
|
||||||
|
|
||||||
|
# 分页结果
|
||||||
|
result_query = base_query.offset(offset).limit(page_size)
|
||||||
|
result = await session.execute(result_query)
|
||||||
|
rows = result.fetchall()
|
||||||
|
|
||||||
|
# 查询总数(应用相同的筛选条件)
|
||||||
|
count_base_query = (
|
||||||
|
select(func.count(col(Preference.scope_id)))
|
||||||
|
.select_from(Preference)
|
||||||
|
.outerjoin(
|
||||||
|
ConversationV2,
|
||||||
|
func.json_extract(Preference.value, "$.val")
|
||||||
|
== ConversationV2.conversation_id,
|
||||||
|
)
|
||||||
|
.outerjoin(
|
||||||
|
Persona, col(ConversationV2.persona_id) == Persona.persona_id
|
||||||
|
)
|
||||||
|
.where(Preference.scope == "umo", Preference.key == "sel_conv_id")
|
||||||
|
)
|
||||||
|
|
||||||
|
# 应用相同的搜索和平台筛选条件到计数查询
|
||||||
|
if search_query:
|
||||||
|
search_pattern = f"%{search_query}%"
|
||||||
|
count_base_query = count_base_query.where(
|
||||||
|
or_(
|
||||||
|
col(Preference.scope_id).ilike(search_pattern),
|
||||||
|
col(ConversationV2.title).ilike(search_pattern),
|
||||||
|
col(Persona.persona_id).ilike(search_pattern),
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
if platform:
|
||||||
|
platform_pattern = f"{platform}:%"
|
||||||
|
count_base_query = count_base_query.where(
|
||||||
|
col(Preference.scope_id).like(platform_pattern)
|
||||||
|
)
|
||||||
|
|
||||||
|
total_result = await session.execute(count_base_query)
|
||||||
|
total = total_result.scalar() or 0
|
||||||
|
|
||||||
|
sessions_data = [
|
||||||
|
{
|
||||||
|
"session_id": row.session_id,
|
||||||
|
"conversation_id": row.conversation_id,
|
||||||
|
"persona_id": row.persona_id,
|
||||||
|
"title": row.title,
|
||||||
|
"persona_name": row.persona_name,
|
||||||
|
}
|
||||||
|
for row in rows
|
||||||
|
]
|
||||||
|
return sessions_data, total
|
||||||
|
|
||||||
async def insert_platform_message_history(
|
async def insert_platform_message_history(
|
||||||
self,
|
self,
|
||||||
@@ -267,9 +399,9 @@ class SQLiteDatabase(BaseDatabase):
|
|||||||
cutoff_time = now - timedelta(seconds=offset_sec)
|
cutoff_time = now - timedelta(seconds=offset_sec)
|
||||||
await session.execute(
|
await session.execute(
|
||||||
delete(PlatformMessageHistory).where(
|
delete(PlatformMessageHistory).where(
|
||||||
PlatformMessageHistory.platform_id == platform_id,
|
col(PlatformMessageHistory.platform_id) == platform_id,
|
||||||
PlatformMessageHistory.user_id == user_id,
|
col(PlatformMessageHistory.user_id) == user_id,
|
||||||
PlatformMessageHistory.created_at < cutoff_time,
|
col(PlatformMessageHistory.created_at) < cutoff_time,
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -286,7 +418,7 @@ class SQLiteDatabase(BaseDatabase):
|
|||||||
PlatformMessageHistory.platform_id == platform_id,
|
PlatformMessageHistory.platform_id == platform_id,
|
||||||
PlatformMessageHistory.user_id == user_id,
|
PlatformMessageHistory.user_id == user_id,
|
||||||
)
|
)
|
||||||
.order_by(PlatformMessageHistory.created_at.desc())
|
.order_by(desc(PlatformMessageHistory.created_at))
|
||||||
)
|
)
|
||||||
result = await session.execute(query.offset(offset).limit(page_size))
|
result = await session.execute(query.offset(offset).limit(page_size))
|
||||||
return result.scalars().all()
|
return result.scalars().all()
|
||||||
@@ -308,7 +440,7 @@ class SQLiteDatabase(BaseDatabase):
|
|||||||
"""Get an attachment by its ID."""
|
"""Get an attachment by its ID."""
|
||||||
async with self.get_db() as session:
|
async with self.get_db() as session:
|
||||||
session: AsyncSession
|
session: AsyncSession
|
||||||
query = select(Attachment).where(Attachment.id == attachment_id)
|
query = select(Attachment).where(Attachment.attachment_id == attachment_id)
|
||||||
result = await session.execute(query)
|
result = await session.execute(query)
|
||||||
return result.scalar_one_or_none()
|
return result.scalar_one_or_none()
|
||||||
|
|
||||||
@@ -351,7 +483,7 @@ class SQLiteDatabase(BaseDatabase):
|
|||||||
async with self.get_db() as session:
|
async with self.get_db() as session:
|
||||||
session: AsyncSession
|
session: AsyncSession
|
||||||
async with session.begin():
|
async with session.begin():
|
||||||
query = update(Persona).where(Persona.persona_id == persona_id)
|
query = update(Persona).where(col(Persona.persona_id) == persona_id)
|
||||||
values = {}
|
values = {}
|
||||||
if system_prompt is not None:
|
if system_prompt is not None:
|
||||||
values["system_prompt"] = system_prompt
|
values["system_prompt"] = system_prompt
|
||||||
@@ -371,7 +503,7 @@ class SQLiteDatabase(BaseDatabase):
|
|||||||
session: AsyncSession
|
session: AsyncSession
|
||||||
async with session.begin():
|
async with session.begin():
|
||||||
await session.execute(
|
await session.execute(
|
||||||
delete(Persona).where(Persona.persona_id == persona_id)
|
delete(Persona).where(col(Persona.persona_id) == persona_id)
|
||||||
)
|
)
|
||||||
|
|
||||||
async def insert_preference_or_update(self, scope, scope_id, key, value):
|
async def insert_preference_or_update(self, scope, scope_id, key, value):
|
||||||
@@ -426,9 +558,9 @@ class SQLiteDatabase(BaseDatabase):
|
|||||||
async with session.begin():
|
async with session.begin():
|
||||||
await session.execute(
|
await session.execute(
|
||||||
delete(Preference).where(
|
delete(Preference).where(
|
||||||
Preference.scope == scope,
|
col(Preference.scope) == scope,
|
||||||
Preference.scope_id == scope_id,
|
col(Preference.scope_id) == scope_id,
|
||||||
Preference.key == key,
|
col(Preference.key) == key,
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
await session.commit()
|
await session.commit()
|
||||||
@@ -440,7 +572,8 @@ class SQLiteDatabase(BaseDatabase):
|
|||||||
async with session.begin():
|
async with session.begin():
|
||||||
await session.execute(
|
await session.execute(
|
||||||
delete(Preference).where(
|
delete(Preference).where(
|
||||||
Preference.scope == scope, Preference.scope_id == scope_id
|
col(Preference.scope) == scope,
|
||||||
|
col(Preference.scope_id) == scope_id,
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
await session.commit()
|
await session.commit()
|
||||||
@@ -467,7 +600,7 @@ class SQLiteDatabase(BaseDatabase):
|
|||||||
DeprecatedPlatformStat(
|
DeprecatedPlatformStat(
|
||||||
name=data.platform_id,
|
name=data.platform_id,
|
||||||
count=data.count,
|
count=data.count,
|
||||||
timestamp=data.timestamp.timestamp(),
|
timestamp=int(data.timestamp.timestamp()),
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
return deprecated_stats
|
return deprecated_stats
|
||||||
@@ -525,7 +658,7 @@ class SQLiteDatabase(BaseDatabase):
|
|||||||
DeprecatedPlatformStat(
|
DeprecatedPlatformStat(
|
||||||
name=platform_id,
|
name=platform_id,
|
||||||
count=count,
|
count=count,
|
||||||
timestamp=start_time.timestamp(),
|
timestamp=int(start_time.timestamp()),
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
return deprecated_stats
|
return deprecated_stats
|
||||||
|
|||||||
@@ -113,7 +113,8 @@ class FaissVecDB(BaseVecDB):
|
|||||||
reranked_results, key=lambda x: x.relevance_score, reverse=True
|
reranked_results, key=lambda x: x.relevance_score, reverse=True
|
||||||
)
|
)
|
||||||
top_k_results = [
|
top_k_results = [
|
||||||
top_k_results[reranked_result.index] for reranked_result in reranked_results
|
top_k_results[reranked_result.index]
|
||||||
|
for reranked_result in reranked_results
|
||||||
]
|
]
|
||||||
|
|
||||||
return top_k_results
|
return top_k_results
|
||||||
|
|||||||
@@ -22,6 +22,7 @@ class InitialLoader:
|
|||||||
self.db = db
|
self.db = db
|
||||||
self.logger = logger
|
self.logger = logger
|
||||||
self.log_broker = log_broker
|
self.log_broker = log_broker
|
||||||
|
self.webui_dir: str | None = None
|
||||||
|
|
||||||
async def start(self):
|
async def start(self):
|
||||||
core_lifecycle = AstrBotCoreLifecycle(self.log_broker, self.db)
|
core_lifecycle = AstrBotCoreLifecycle(self.log_broker, self.db)
|
||||||
@@ -35,8 +36,10 @@ class InitialLoader:
|
|||||||
|
|
||||||
core_task = core_lifecycle.start()
|
core_task = core_lifecycle.start()
|
||||||
|
|
||||||
|
webui_dir = self.webui_dir
|
||||||
|
|
||||||
self.dashboard_server = AstrBotDashboard(
|
self.dashboard_server = AstrBotDashboard(
|
||||||
core_lifecycle, self.db, core_lifecycle.dashboard_shutdown_event
|
core_lifecycle, self.db, core_lifecycle.dashboard_shutdown_event, webui_dir
|
||||||
)
|
)
|
||||||
task = asyncio.gather(
|
task = asyncio.gather(
|
||||||
core_task, self.dashboard_server.run()
|
core_task, self.dashboard_server.run()
|
||||||
|
|||||||
@@ -19,7 +19,7 @@ class ContentSafetyCheckStage(Stage):
|
|||||||
self.strategy_selector = StrategySelector(config)
|
self.strategy_selector = StrategySelector(config)
|
||||||
|
|
||||||
async def process(
|
async def process(
|
||||||
self, event: AstrMessageEvent, check_text: str = None
|
self, event: AstrMessageEvent, check_text: str | None = None
|
||||||
) -> Union[None, AsyncGenerator[None, None]]:
|
) -> Union[None, AsyncGenerator[None, None]]:
|
||||||
"""检查内容安全"""
|
"""检查内容安全"""
|
||||||
text = check_text if check_text else event.get_message_str()
|
text = check_text if check_text else event.get_message_str()
|
||||||
|
|||||||
@@ -13,7 +13,7 @@ class BaiduAipStrategy(ContentSafetyStrategy):
|
|||||||
self.secret_key = sk
|
self.secret_key = sk
|
||||||
self.client = AipContentCensor(self.app_id, self.api_key, self.secret_key)
|
self.client = AipContentCensor(self.app_id, self.api_key, self.secret_key)
|
||||||
|
|
||||||
def check(self, content: str):
|
def check(self, content: str) -> tuple[bool, str]:
|
||||||
res = self.client.textCensorUserDefined(content)
|
res = self.client.textCensorUserDefined(content)
|
||||||
if "conclusionType" not in res:
|
if "conclusionType" not in res:
|
||||||
return False, ""
|
return False, ""
|
||||||
|
|||||||
@@ -16,7 +16,7 @@ class KeywordsStrategy(ContentSafetyStrategy):
|
|||||||
# json.loads(base64.b64decode(f.read()).decode("utf-8"))["keywords"]
|
# json.loads(base64.b64decode(f.read()).decode("utf-8"))["keywords"]
|
||||||
# )
|
# )
|
||||||
|
|
||||||
def check(self, content: str) -> bool:
|
def check(self, content: str) -> tuple[bool, str]:
|
||||||
for keyword in self.keywords:
|
for keyword in self.keywords:
|
||||||
if re.search(keyword, content):
|
if re.search(keyword, content):
|
||||||
return False, "内容安全检查不通过,匹配到敏感词。"
|
return False, "内容安全检查不通过,匹配到敏感词。"
|
||||||
|
|||||||
@@ -10,7 +10,7 @@ from astrbot.core.platform.astr_message_event import AstrMessageEvent
|
|||||||
|
|
||||||
async def call_handler(
|
async def call_handler(
|
||||||
event: AstrMessageEvent,
|
event: AstrMessageEvent,
|
||||||
handler: T.Awaitable,
|
handler: T.Callable[..., T.Awaitable[T.Any]],
|
||||||
*args,
|
*args,
|
||||||
**kwargs,
|
**kwargs,
|
||||||
) -> T.AsyncGenerator[T.Any, None]:
|
) -> T.AsyncGenerator[T.Any, None]:
|
||||||
@@ -36,6 +36,9 @@ async def call_handler(
|
|||||||
except TypeError:
|
except TypeError:
|
||||||
logger.error("处理函数参数不匹配,请检查 handler 的定义。", exc_info=True)
|
logger.error("处理函数参数不匹配,请检查 handler 的定义。", exc_info=True)
|
||||||
|
|
||||||
|
if not ready_to_call:
|
||||||
|
return
|
||||||
|
|
||||||
if inspect.isasyncgen(ready_to_call):
|
if inspect.isasyncgen(ready_to_call):
|
||||||
_has_yielded = False
|
_has_yielded = False
|
||||||
try:
|
try:
|
||||||
|
|||||||
@@ -1,5 +1,6 @@
|
|||||||
import traceback
|
import traceback
|
||||||
import asyncio
|
import asyncio
|
||||||
|
import random
|
||||||
from typing import Union, AsyncGenerator
|
from typing import Union, AsyncGenerator
|
||||||
from ..stage import Stage, register_stage
|
from ..stage import Stage, register_stage
|
||||||
from ..context import PipelineContext
|
from ..context import PipelineContext
|
||||||
@@ -22,6 +23,26 @@ class PreProcessStage(Stage):
|
|||||||
self, event: AstrMessageEvent
|
self, event: AstrMessageEvent
|
||||||
) -> Union[None, AsyncGenerator[None, None]]:
|
) -> Union[None, AsyncGenerator[None, None]]:
|
||||||
"""在处理事件之前的预处理"""
|
"""在处理事件之前的预处理"""
|
||||||
|
# 平台特异配置:platform_specific.<platform>.pre_ack_emoji
|
||||||
|
supported = {"telegram", "lark"}
|
||||||
|
platform = event.get_platform_name()
|
||||||
|
cfg = (
|
||||||
|
self.config.get("platform_specific", {})
|
||||||
|
.get(platform, {})
|
||||||
|
.get("pre_ack_emoji", {})
|
||||||
|
) or {}
|
||||||
|
emojis = cfg.get("emojis") or []
|
||||||
|
if (
|
||||||
|
cfg.get("enable", False)
|
||||||
|
and platform in supported
|
||||||
|
and emojis
|
||||||
|
and event.is_at_or_wake_command
|
||||||
|
):
|
||||||
|
try:
|
||||||
|
await event.react(random.choice(emojis))
|
||||||
|
except Exception as e:
|
||||||
|
logger.warning(f"{platform} 预回应表情发送失败: {e}")
|
||||||
|
|
||||||
# 路径映射
|
# 路径映射
|
||||||
if mappings := self.platform_settings.get("path_mapping", []):
|
if mappings := self.platform_settings.get("path_mapping", []):
|
||||||
# 支持 Record,Image 消息段的路径映射。
|
# 支持 Record,Image 消息段的路径映射。
|
||||||
@@ -46,6 +67,9 @@ class PreProcessStage(Stage):
|
|||||||
ctx = self.plugin_manager.context
|
ctx = self.plugin_manager.context
|
||||||
stt_provider = ctx.get_using_stt_provider(event.unified_msg_origin)
|
stt_provider = ctx.get_using_stt_provider(event.unified_msg_origin)
|
||||||
if not stt_provider:
|
if not stt_provider:
|
||||||
|
logger.warning(
|
||||||
|
f"会话 {event.unified_msg_origin} 未配置语音转文本模型。"
|
||||||
|
)
|
||||||
return
|
return
|
||||||
message_chain = event.get_messages()
|
message_chain = event.get_messages()
|
||||||
for idx, component in enumerate(message_chain):
|
for idx, component in enumerate(message_chain):
|
||||||
|
|||||||
@@ -7,6 +7,7 @@ import copy
|
|||||||
import json
|
import json
|
||||||
import traceback
|
import traceback
|
||||||
from typing import AsyncGenerator, Union
|
from typing import AsyncGenerator, Union
|
||||||
|
from astrbot.core.conversation_mgr import Conversation
|
||||||
from astrbot.core import logger
|
from astrbot.core import logger
|
||||||
from astrbot.core.message.components import Image
|
from astrbot.core.message.components import Image
|
||||||
from astrbot.core.message.message_event_result import (
|
from astrbot.core.message.message_event_result import (
|
||||||
@@ -133,6 +134,15 @@ class FunctionToolExecutor(BaseFunctionToolExecutor[AstrAgentContext]):
|
|||||||
|
|
||||||
if agent_runner.done():
|
if agent_runner.done():
|
||||||
llm_response = agent_runner.get_final_llm_resp()
|
llm_response = agent_runner.get_final_llm_resp()
|
||||||
|
|
||||||
|
if not llm_response:
|
||||||
|
text_content = mcp.types.TextContent(
|
||||||
|
type="text",
|
||||||
|
text=f"error when deligate task to {tool.agent.name}",
|
||||||
|
)
|
||||||
|
yield mcp.types.CallToolResult(content=[text_content])
|
||||||
|
return
|
||||||
|
|
||||||
logger.debug(
|
logger.debug(
|
||||||
f"Agent {tool.agent.name} 任务完成, response: {llm_response.completion_text}"
|
f"Agent {tool.agent.name} 任务完成, response: {llm_response.completion_text}"
|
||||||
)
|
)
|
||||||
@@ -148,7 +158,7 @@ class FunctionToolExecutor(BaseFunctionToolExecutor[AstrAgentContext]):
|
|||||||
)
|
)
|
||||||
yield mcp.types.CallToolResult(content=[text_content])
|
yield mcp.types.CallToolResult(content=[text_content])
|
||||||
else:
|
else:
|
||||||
yield mcp.types.TextContent(
|
text_content = mcp.types.TextContent(
|
||||||
type="text",
|
type="text",
|
||||||
text=f"error when deligate task to {tool.agent.name}",
|
text=f"error when deligate task to {tool.agent.name}",
|
||||||
)
|
)
|
||||||
@@ -200,7 +210,11 @@ class FunctionToolExecutor(BaseFunctionToolExecutor[AstrAgentContext]):
|
|||||||
):
|
):
|
||||||
if not tool.mcp_client:
|
if not tool.mcp_client:
|
||||||
raise ValueError("MCP client is not available for MCP function tools.")
|
raise ValueError("MCP client is not available for MCP function tools.")
|
||||||
res = await tool.mcp_client.session.call_tool(
|
|
||||||
|
session = tool.mcp_client.session
|
||||||
|
if not session:
|
||||||
|
raise ValueError("MCP session is not available for MCP function tools.")
|
||||||
|
res = await session.call_tool(
|
||||||
name=tool.name,
|
name=tool.name,
|
||||||
arguments=tool_args,
|
arguments=tool_args,
|
||||||
)
|
)
|
||||||
@@ -271,19 +285,12 @@ async def run_agent(
|
|||||||
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.error(traceback.format_exc())
|
logger.error(traceback.format_exc())
|
||||||
astr_event.set_result(
|
err_msg = f"\n\nAstrBot 请求失败。\n错误类型: {type(e).__name__}\n错误信息: {str(e)}\n\n请在控制台查看和分享错误详情。\n"
|
||||||
MessageEventResult().message(
|
if agent_runner.streaming:
|
||||||
f"AstrBot 请求失败。\n错误类型: {type(e).__name__}\n错误信息: {str(e)}\n\n请在控制台查看和分享错误详情。\n"
|
yield MessageChain().message(err_msg)
|
||||||
)
|
else:
|
||||||
)
|
astr_event.set_result(MessageEventResult().message(err_msg))
|
||||||
return
|
return
|
||||||
asyncio.create_task(
|
|
||||||
Metric.upload(
|
|
||||||
llm_tick=1,
|
|
||||||
model_name=agent_runner.provider.get_model(),
|
|
||||||
provider_type=agent_runner.provider.meta().type,
|
|
||||||
)
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
class LLMRequestSubStage(Stage):
|
class LLMRequestSubStage(Stage):
|
||||||
@@ -325,7 +332,7 @@ class LLMRequestSubStage(Stage):
|
|||||||
|
|
||||||
return _ctx.get_using_provider(umo=event.unified_msg_origin)
|
return _ctx.get_using_provider(umo=event.unified_msg_origin)
|
||||||
|
|
||||||
async def _get_session_conv(self, event: AstrMessageEvent):
|
async def _get_session_conv(self, event: AstrMessageEvent) -> Conversation:
|
||||||
umo = event.unified_msg_origin
|
umo = event.unified_msg_origin
|
||||||
conv_mgr = self.conv_manager
|
conv_mgr = self.conv_manager
|
||||||
|
|
||||||
@@ -337,6 +344,8 @@ class LLMRequestSubStage(Stage):
|
|||||||
if not conversation:
|
if not conversation:
|
||||||
cid = await conv_mgr.new_conversation(umo, event.get_platform_id())
|
cid = await conv_mgr.new_conversation(umo, event.get_platform_id())
|
||||||
conversation = await conv_mgr.get_conversation(umo, cid)
|
conversation = await conv_mgr.get_conversation(umo, cid)
|
||||||
|
if not conversation:
|
||||||
|
raise RuntimeError("无法创建新的对话。")
|
||||||
return conversation
|
return conversation
|
||||||
|
|
||||||
async def process(
|
async def process(
|
||||||
@@ -444,7 +453,10 @@ class LLMRequestSubStage(Stage):
|
|||||||
if event.plugins_name is not None and req.func_tool:
|
if event.plugins_name is not None and req.func_tool:
|
||||||
new_tool_set = ToolSet()
|
new_tool_set = ToolSet()
|
||||||
for tool in req.func_tool.tools:
|
for tool in req.func_tool.tools:
|
||||||
plugin = star_map.get(tool.handler_module_path)
|
mp = tool.handler_module_path
|
||||||
|
if not mp:
|
||||||
|
continue
|
||||||
|
plugin = star_map.get(mp)
|
||||||
if not plugin:
|
if not plugin:
|
||||||
continue
|
continue
|
||||||
if plugin.name in event.plugins_name or plugin.reserved:
|
if plugin.name in event.plugins_name or plugin.reserved:
|
||||||
@@ -505,6 +517,14 @@ class LLMRequestSubStage(Stage):
|
|||||||
if event.get_platform_name() == "webchat":
|
if event.get_platform_name() == "webchat":
|
||||||
asyncio.create_task(self._handle_webchat(event, req, provider))
|
asyncio.create_task(self._handle_webchat(event, req, provider))
|
||||||
|
|
||||||
|
asyncio.create_task(
|
||||||
|
Metric.upload(
|
||||||
|
llm_tick=1,
|
||||||
|
model_name=agent_runner.provider.get_model(),
|
||||||
|
provider_type=agent_runner.provider.meta().type,
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
async def _handle_webchat(
|
async def _handle_webchat(
|
||||||
self, event: AstrMessageEvent, req: ProviderRequest, prov: Provider
|
self, event: AstrMessageEvent, req: ProviderRequest, prov: Provider
|
||||||
):
|
):
|
||||||
@@ -517,7 +537,23 @@ class LLMRequestSubStage(Stage):
|
|||||||
latest_pair = messages[-2:]
|
latest_pair = messages[-2:]
|
||||||
if not latest_pair:
|
if not latest_pair:
|
||||||
return
|
return
|
||||||
cleaned_text = "User: " + latest_pair[0].get("content", "").strip()
|
content = latest_pair[0].get("content", "")
|
||||||
|
if isinstance(content, list):
|
||||||
|
# 多模态
|
||||||
|
text_parts = []
|
||||||
|
for item in content:
|
||||||
|
if isinstance(item, dict):
|
||||||
|
if item.get("type") == "text":
|
||||||
|
text_parts.append(item.get("text", ""))
|
||||||
|
elif item.get("type") == "image":
|
||||||
|
text_parts.append("[图片]")
|
||||||
|
elif isinstance(item, str):
|
||||||
|
text_parts.append(item)
|
||||||
|
cleaned_text = "User: " + " ".join(text_parts).strip()
|
||||||
|
elif isinstance(content, str):
|
||||||
|
cleaned_text = "User: " + content.strip()
|
||||||
|
else:
|
||||||
|
return
|
||||||
logger.debug(f"WebChat 对话标题生成请求,清理后的文本: {cleaned_text}")
|
logger.debug(f"WebChat 对话标题生成请求,清理后的文本: {cleaned_text}")
|
||||||
llm_resp = await prov.text_chat(
|
llm_resp = await prov.text_chat(
|
||||||
system_prompt="You are expert in summarizing user's query.",
|
system_prompt="You are expert in summarizing user's query.",
|
||||||
|
|||||||
@@ -34,12 +34,14 @@ class StarRequestSubStage(Stage):
|
|||||||
|
|
||||||
for handler in activated_handlers:
|
for handler in activated_handlers:
|
||||||
params = handlers_parsed_params.get(handler.handler_full_name, {})
|
params = handlers_parsed_params.get(handler.handler_full_name, {})
|
||||||
try:
|
md = star_map.get(handler.handler_module_path)
|
||||||
if handler.handler_module_path not in star_map:
|
if not md:
|
||||||
continue
|
logger.warning(
|
||||||
logger.debug(
|
f"Cannot find plugin for given handler module path: {handler.handler_module_path}"
|
||||||
f"plugin -> {star_map.get(handler.handler_module_path).name} - {handler.handler_name}"
|
|
||||||
)
|
)
|
||||||
|
continue
|
||||||
|
logger.debug(f"plugin -> {md.name} - {handler.handler_name}")
|
||||||
|
try:
|
||||||
wrapper = call_handler(event, handler.handler, **params)
|
wrapper = call_handler(event, handler.handler, **params)
|
||||||
async for ret in wrapper:
|
async for ret in wrapper:
|
||||||
yield ret
|
yield ret
|
||||||
@@ -49,7 +51,7 @@ class StarRequestSubStage(Stage):
|
|||||||
logger.error(f"Star {handler.handler_full_name} handle error: {e}")
|
logger.error(f"Star {handler.handler_full_name} handle error: {e}")
|
||||||
|
|
||||||
if event.is_at_or_wake_command:
|
if event.is_at_or_wake_command:
|
||||||
ret = f":(\n\n在调用插件 {star_map.get(handler.handler_module_path).name} 的处理函数 {handler.handler_name} 时出现异常:{e}"
|
ret = f":(\n\n在调用插件 {md.name} 的处理函数 {handler.handler_name} 时出现异常:{e}"
|
||||||
event.set_result(MessageEventResult().message(ret))
|
event.set_result(MessageEventResult().message(ret))
|
||||||
yield
|
yield
|
||||||
event.clear_result()
|
event.clear_result()
|
||||||
|
|||||||
@@ -1,17 +1,15 @@
|
|||||||
import random
|
import random
|
||||||
import asyncio
|
import asyncio
|
||||||
import math
|
import math
|
||||||
import traceback
|
|
||||||
import astrbot.core.message.components as Comp
|
import astrbot.core.message.components as Comp
|
||||||
from typing import Union, AsyncGenerator
|
from typing import Union, AsyncGenerator
|
||||||
from ..stage import register_stage, Stage
|
from ..stage import register_stage, Stage
|
||||||
from ..context import PipelineContext
|
from ..context import PipelineContext, call_event_hook
|
||||||
from astrbot.core.platform.astr_message_event import AstrMessageEvent
|
from astrbot.core.platform.astr_message_event import AstrMessageEvent
|
||||||
from astrbot.core.message.message_event_result import MessageChain, ResultContentType
|
from astrbot.core.message.message_event_result import MessageChain, ResultContentType
|
||||||
from astrbot.core import logger
|
from astrbot.core import logger
|
||||||
from astrbot.core.message.message_event_result import BaseMessageComponent
|
from astrbot.core.message.components import BaseMessageComponent, ComponentType
|
||||||
from astrbot.core.star.star_handler import star_handlers_registry, EventType
|
from astrbot.core.star.star_handler import EventType
|
||||||
from astrbot.core.star.star import star_map
|
|
||||||
from astrbot.core.utils.path_util import path_Mapping
|
from astrbot.core.utils.path_util import path_Mapping
|
||||||
from astrbot.core.utils.session_lock import session_lock_manager
|
from astrbot.core.utils.session_lock import session_lock_manager
|
||||||
|
|
||||||
@@ -114,6 +112,43 @@ class RespondStage(Stage):
|
|||||||
# 如果所有组件都为空
|
# 如果所有组件都为空
|
||||||
return True
|
return True
|
||||||
|
|
||||||
|
def is_seg_reply_required(self, event: AstrMessageEvent) -> bool:
|
||||||
|
"""检查是否需要分段回复"""
|
||||||
|
if not self.enable_seg:
|
||||||
|
return False
|
||||||
|
|
||||||
|
if self.only_llm_result and not event.get_result().is_llm_result():
|
||||||
|
return False
|
||||||
|
|
||||||
|
if event.get_platform_name() in [
|
||||||
|
"qq_official",
|
||||||
|
"weixin_official_account",
|
||||||
|
"dingtalk",
|
||||||
|
]:
|
||||||
|
return False
|
||||||
|
|
||||||
|
return True
|
||||||
|
|
||||||
|
def _extract_comp(
|
||||||
|
self,
|
||||||
|
raw_chain: list[BaseMessageComponent],
|
||||||
|
extract_types: set[ComponentType],
|
||||||
|
modify_raw_chain: bool = True,
|
||||||
|
):
|
||||||
|
extracted = []
|
||||||
|
if modify_raw_chain:
|
||||||
|
remaining = []
|
||||||
|
for comp in raw_chain:
|
||||||
|
if comp.type in extract_types:
|
||||||
|
extracted.append(comp)
|
||||||
|
else:
|
||||||
|
remaining.append(comp)
|
||||||
|
raw_chain[:] = remaining
|
||||||
|
else:
|
||||||
|
extracted = [comp for comp in raw_chain if comp.type in extract_types]
|
||||||
|
|
||||||
|
return extracted
|
||||||
|
|
||||||
async def process(
|
async def process(
|
||||||
self, event: AstrMessageEvent
|
self, event: AstrMessageEvent
|
||||||
) -> Union[None, AsyncGenerator[None, None]]:
|
) -> Union[None, AsyncGenerator[None, None]]:
|
||||||
@@ -123,7 +158,14 @@ class RespondStage(Stage):
|
|||||||
if result.result_content_type == ResultContentType.STREAMING_FINISH:
|
if result.result_content_type == ResultContentType.STREAMING_FINISH:
|
||||||
return
|
return
|
||||||
|
|
||||||
|
logger.info(
|
||||||
|
f"Prepare to send - {event.get_sender_name()}/{event.get_sender_id()}: {event._outline_chain(result.chain)}"
|
||||||
|
)
|
||||||
|
|
||||||
if result.result_content_type == ResultContentType.STREAMING_RESULT:
|
if result.result_content_type == ResultContentType.STREAMING_RESULT:
|
||||||
|
if result.async_stream is None:
|
||||||
|
logger.warning("async_stream 为空,跳过发送。")
|
||||||
|
return
|
||||||
# 流式结果直接交付平台适配器处理
|
# 流式结果直接交付平台适配器处理
|
||||||
use_fallback = self.config.get("provider_settings", {}).get(
|
use_fallback = self.config.get("provider_settings", {}).get(
|
||||||
"streaming_segmented", False
|
"streaming_segmented", False
|
||||||
@@ -148,87 +190,71 @@ class RespondStage(Stage):
|
|||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.warning(f"空内容检查异常: {e}")
|
logger.warning(f"空内容检查异常: {e}")
|
||||||
|
|
||||||
record_comps = [c for c in result.chain if isinstance(c, Comp.Record)]
|
# 发送消息链
|
||||||
non_record_comps = [
|
# Record 需要强制单独发送
|
||||||
c for c in result.chain if not isinstance(c, Comp.Record)
|
need_separately = {ComponentType.Record}
|
||||||
]
|
if self.is_seg_reply_required(event):
|
||||||
|
header_comps = self._extract_comp(
|
||||||
if (
|
result.chain,
|
||||||
self.enable_seg
|
{ComponentType.Reply, ComponentType.At},
|
||||||
and (
|
modify_raw_chain=True,
|
||||||
(self.only_llm_result and result.is_llm_result())
|
|
||||||
or not self.only_llm_result
|
|
||||||
)
|
)
|
||||||
and event.get_platform_name()
|
if not result.chain or len(result.chain) == 0:
|
||||||
not in ["qq_official", "weixin_official_account", "dingtalk"]
|
# may fix #2670
|
||||||
):
|
logger.warning(
|
||||||
decorated_comps = []
|
f"实际消息链为空, 跳过发送阶段。header_chain: {header_comps}, actual_chain: {result.chain}"
|
||||||
if self.reply_with_mention:
|
)
|
||||||
for comp in result.chain:
|
return
|
||||||
if isinstance(comp, Comp.At):
|
|
||||||
decorated_comps.append(comp)
|
|
||||||
result.chain.remove(comp)
|
|
||||||
break
|
|
||||||
if self.reply_with_quote:
|
|
||||||
for comp in result.chain:
|
|
||||||
if isinstance(comp, Comp.Reply):
|
|
||||||
decorated_comps.append(comp)
|
|
||||||
result.chain.remove(comp)
|
|
||||||
break
|
|
||||||
|
|
||||||
# leverage lock to guarentee the order of message sending among different events
|
|
||||||
async with session_lock_manager.acquire_lock(event.unified_msg_origin):
|
async with session_lock_manager.acquire_lock(event.unified_msg_origin):
|
||||||
for rcomp in record_comps:
|
for comp in result.chain:
|
||||||
i = await self._calc_comp_interval(rcomp)
|
|
||||||
await asyncio.sleep(i)
|
|
||||||
try:
|
|
||||||
await event.send(MessageChain([rcomp]))
|
|
||||||
except Exception as e:
|
|
||||||
logger.error(f"发送消息失败: {e} chain: {result.chain}")
|
|
||||||
break
|
|
||||||
# 分段回复
|
|
||||||
for comp in non_record_comps:
|
|
||||||
i = await self._calc_comp_interval(comp)
|
i = await self._calc_comp_interval(comp)
|
||||||
await asyncio.sleep(i)
|
await asyncio.sleep(i)
|
||||||
try:
|
try:
|
||||||
await event.send(MessageChain([*decorated_comps, comp]))
|
if comp.type in need_separately:
|
||||||
decorated_comps = [] # 清空已发送的装饰组件
|
await event.send(MessageChain([comp]))
|
||||||
except Exception as e:
|
|
||||||
logger.error(f"发送消息失败: {e} chain: {result.chain}")
|
|
||||||
break
|
|
||||||
else:
|
else:
|
||||||
for rcomp in record_comps:
|
await event.send(MessageChain([*header_comps, comp]))
|
||||||
try:
|
header_comps.clear()
|
||||||
await event.send(MessageChain([rcomp]))
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.error(f"发送消息失败: {e} chain: {result.chain}")
|
logger.error(
|
||||||
|
f"发送消息链失败: chain = {MessageChain([comp])}, error = {e}",
|
||||||
|
exc_info=True,
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
if all(
|
||||||
|
comp.type in {ComponentType.Reply, ComponentType.At}
|
||||||
|
for comp in result.chain
|
||||||
|
):
|
||||||
|
# may fix #2670
|
||||||
|
logger.warning(
|
||||||
|
f"消息链全为 Reply 和 At 消息段, 跳过发送阶段。chain: {result.chain}"
|
||||||
|
)
|
||||||
|
return
|
||||||
|
sep_comps = self._extract_comp(
|
||||||
|
result.chain,
|
||||||
|
need_separately,
|
||||||
|
modify_raw_chain=True,
|
||||||
|
)
|
||||||
|
for comp in sep_comps:
|
||||||
|
chain = MessageChain([comp])
|
||||||
try:
|
try:
|
||||||
await event.send(MessageChain(non_record_comps))
|
await event.send(chain)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.error(traceback.format_exc())
|
logger.error(
|
||||||
logger.error(f"发送消息失败: {e} chain: {result.chain}")
|
f"发送消息链失败: chain = {chain}, error = {e}",
|
||||||
|
exc_info=True,
|
||||||
logger.info(
|
|
||||||
f"AstrBot -> {event.get_sender_name()}/{event.get_sender_id()}: {event._outline_chain(result.chain)}"
|
|
||||||
)
|
)
|
||||||
|
chain = MessageChain(result.chain)
|
||||||
handlers = star_handlers_registry.get_handlers_by_event_type(
|
if result.chain and len(result.chain) > 0:
|
||||||
EventType.OnAfterMessageSentEvent, plugins_name=event.plugins_name
|
|
||||||
)
|
|
||||||
for handler in handlers:
|
|
||||||
try:
|
try:
|
||||||
logger.debug(
|
await event.send(chain)
|
||||||
f"hook(on_after_message_sent) -> {star_map[handler.handler_module_path].name} - {handler.handler_name}"
|
except Exception as e:
|
||||||
|
logger.error(
|
||||||
|
f"发送消息链失败: chain = {chain}, error = {e}",
|
||||||
|
exc_info=True,
|
||||||
)
|
)
|
||||||
await handler.handler(event)
|
|
||||||
except BaseException:
|
|
||||||
logger.error(traceback.format_exc())
|
|
||||||
|
|
||||||
if event.is_stopped():
|
if await call_event_hook(event, EventType.OnAfterMessageSentEvent):
|
||||||
logger.info(
|
|
||||||
f"{star_map[handler.handler_module_path].name} - {handler.handler_name} 终止了事件传播。"
|
|
||||||
)
|
|
||||||
return
|
return
|
||||||
|
|
||||||
event.clear_result()
|
event.clear_result()
|
||||||
|
|||||||
@@ -183,9 +183,13 @@ class ResultDecorateStage(Stage):
|
|||||||
if (
|
if (
|
||||||
self.ctx.astrbot_config["provider_tts_settings"]["enable"]
|
self.ctx.astrbot_config["provider_tts_settings"]["enable"]
|
||||||
and result.is_llm_result()
|
and result.is_llm_result()
|
||||||
and tts_provider
|
|
||||||
and SessionServiceManager.should_process_tts_request(event)
|
and SessionServiceManager.should_process_tts_request(event)
|
||||||
):
|
):
|
||||||
|
if not tts_provider:
|
||||||
|
logger.warning(
|
||||||
|
f"会话 {event.unified_msg_origin} 未配置文本转语音模型。"
|
||||||
|
)
|
||||||
|
return
|
||||||
new_chain = []
|
new_chain = []
|
||||||
for comp in result.chain:
|
for comp in result.chain:
|
||||||
if isinstance(comp, Plain) and len(comp.text) > 1:
|
if isinstance(comp, Plain) and len(comp.text) > 1:
|
||||||
|
|||||||
@@ -11,7 +11,8 @@ class SessionStatusCheckStage(Stage):
|
|||||||
"""检查会话是否整体启用"""
|
"""检查会话是否整体启用"""
|
||||||
|
|
||||||
async def initialize(self, ctx: PipelineContext) -> None:
|
async def initialize(self, ctx: PipelineContext) -> None:
|
||||||
pass
|
self.ctx = ctx
|
||||||
|
self.conv_mgr = ctx.plugin_manager.context.conversation_manager
|
||||||
|
|
||||||
async def process(
|
async def process(
|
||||||
self, event: AstrMessageEvent
|
self, event: AstrMessageEvent
|
||||||
@@ -19,4 +20,14 @@ class SessionStatusCheckStage(Stage):
|
|||||||
# 检查会话是否整体启用
|
# 检查会话是否整体启用
|
||||||
if not SessionServiceManager.is_session_enabled(event.unified_msg_origin):
|
if not SessionServiceManager.is_session_enabled(event.unified_msg_origin):
|
||||||
logger.debug(f"会话 {event.unified_msg_origin} 已被关闭,已终止事件传播。")
|
logger.debug(f"会话 {event.unified_msg_origin} 已被关闭,已终止事件传播。")
|
||||||
|
|
||||||
|
# workaround for #2309
|
||||||
|
conv_id = await self.conv_mgr.get_curr_conversation_id(
|
||||||
|
event.unified_msg_origin
|
||||||
|
)
|
||||||
|
if not conv_id:
|
||||||
|
await self.conv_mgr.new_conversation(
|
||||||
|
event.unified_msg_origin, platform_id=event.get_platform_id()
|
||||||
|
)
|
||||||
|
|
||||||
event.stop_event()
|
event.stop_event()
|
||||||
|
|||||||
@@ -5,6 +5,7 @@ from astrbot.core.message.components import At, AtAll, Reply
|
|||||||
from astrbot.core.message.message_event_result import MessageChain, MessageEventResult
|
from astrbot.core.message.message_event_result import MessageChain, MessageEventResult
|
||||||
from astrbot.core.platform.astr_message_event import AstrMessageEvent
|
from astrbot.core.platform.astr_message_event import AstrMessageEvent
|
||||||
from astrbot.core.star.filter.permission import PermissionTypeFilter
|
from astrbot.core.star.filter.permission import PermissionTypeFilter
|
||||||
|
from astrbot.core.star.filter.command_group import CommandGroupFilter
|
||||||
from astrbot.core.star.session_plugin_manager import SessionPluginManager
|
from astrbot.core.star.session_plugin_manager import SessionPluginManager
|
||||||
from astrbot.core.star.star import star_map
|
from astrbot.core.star.star import star_map
|
||||||
from astrbot.core.star.star_handler import EventType, star_handlers_registry
|
from astrbot.core.star.star_handler import EventType, star_handlers_registry
|
||||||
@@ -170,10 +171,14 @@ class WakingCheckStage(Stage):
|
|||||||
is_wake = True
|
is_wake = True
|
||||||
event.is_wake = True
|
event.is_wake = True
|
||||||
|
|
||||||
|
is_group_cmd_handler = any(
|
||||||
|
isinstance(f, CommandGroupFilter) for f in handler.event_filters
|
||||||
|
)
|
||||||
|
if not is_group_cmd_handler:
|
||||||
activated_handlers.append(handler)
|
activated_handlers.append(handler)
|
||||||
if "parsed_params" in event.get_extra():
|
if "parsed_params" in event.get_extra(default={}):
|
||||||
handlers_parsed_params[handler.handler_full_name] = event.get_extra(
|
handlers_parsed_params[handler.handler_full_name] = (
|
||||||
"parsed_params"
|
event.get_extra("parsed_params")
|
||||||
)
|
)
|
||||||
|
|
||||||
event._extras.pop("parsed_params", None)
|
event._extras.pop("parsed_params", None)
|
||||||
|
|||||||
@@ -4,7 +4,7 @@ import re
|
|||||||
import hashlib
|
import hashlib
|
||||||
import uuid
|
import uuid
|
||||||
|
|
||||||
from typing import List, Union, Optional, AsyncGenerator
|
from typing import List, Union, Optional, AsyncGenerator, TypeVar, Any
|
||||||
|
|
||||||
from astrbot import logger
|
from astrbot import logger
|
||||||
from astrbot.core.db.po import Conversation
|
from astrbot.core.db.po import Conversation
|
||||||
@@ -26,6 +26,8 @@ from .astrbot_message import AstrBotMessage, Group
|
|||||||
from .platform_metadata import PlatformMetadata
|
from .platform_metadata import PlatformMetadata
|
||||||
from .message_session import MessageSession, MessageSesion # noqa
|
from .message_session import MessageSession, MessageSesion # noqa
|
||||||
|
|
||||||
|
_VT = TypeVar("_VT")
|
||||||
|
|
||||||
|
|
||||||
class AstrMessageEvent(abc.ABC):
|
class AstrMessageEvent(abc.ABC):
|
||||||
def __init__(
|
def __init__(
|
||||||
@@ -49,7 +51,7 @@ class AstrMessageEvent(abc.ABC):
|
|||||||
"""是否唤醒(是否通过 WakingStage)"""
|
"""是否唤醒(是否通过 WakingStage)"""
|
||||||
self.is_at_or_wake_command = False
|
self.is_at_or_wake_command = False
|
||||||
"""是否是 At 机器人或者带有唤醒词或者是私聊(插件注册的事件监听器会让 is_wake 设为 True, 但是不会让这个属性置为 True)"""
|
"""是否是 At 机器人或者带有唤醒词或者是私聊(插件注册的事件监听器会让 is_wake 设为 True, 但是不会让这个属性置为 True)"""
|
||||||
self._extras = {}
|
self._extras: dict[str, Any] = {}
|
||||||
self.session = MessageSesion(
|
self.session = MessageSesion(
|
||||||
platform_name=platform_meta.id,
|
platform_name=platform_meta.id,
|
||||||
message_type=message_obj.type,
|
message_type=message_obj.type,
|
||||||
@@ -57,7 +59,7 @@ class AstrMessageEvent(abc.ABC):
|
|||||||
)
|
)
|
||||||
self.unified_msg_origin = str(self.session)
|
self.unified_msg_origin = str(self.session)
|
||||||
"""统一的消息来源字符串。格式为 platform_name:message_type:session_id"""
|
"""统一的消息来源字符串。格式为 platform_name:message_type:session_id"""
|
||||||
self._result: MessageEventResult = None
|
self._result: MessageEventResult | None = None
|
||||||
"""消息事件的结果"""
|
"""消息事件的结果"""
|
||||||
|
|
||||||
self._has_send_oper = False
|
self._has_send_oper = False
|
||||||
@@ -173,13 +175,15 @@ class AstrMessageEvent(abc.ABC):
|
|||||||
"""
|
"""
|
||||||
self._extras[key] = value
|
self._extras[key] = value
|
||||||
|
|
||||||
def get_extra(self, key=None):
|
def get_extra(
|
||||||
|
self, key: str | None = None, default: _VT = None
|
||||||
|
) -> dict[str, Any] | _VT:
|
||||||
"""
|
"""
|
||||||
获取额外的信息。
|
获取额外的信息。
|
||||||
"""
|
"""
|
||||||
if key is None:
|
if key is None:
|
||||||
return self._extras
|
return self._extras
|
||||||
return self._extras.get(key, None)
|
return self._extras.get(key, default)
|
||||||
|
|
||||||
def clear_extra(self):
|
def clear_extra(self):
|
||||||
"""
|
"""
|
||||||
@@ -412,6 +416,16 @@ class AstrMessageEvent(abc.ABC):
|
|||||||
)
|
)
|
||||||
self._has_send_oper = True
|
self._has_send_oper = True
|
||||||
|
|
||||||
|
async def react(self, emoji: str):
|
||||||
|
"""
|
||||||
|
对消息添加表情回应。
|
||||||
|
|
||||||
|
默认实现为发送一条包含该表情的消息。
|
||||||
|
注意:此实现并不一定符合所有平台的原生“表情回应”行为。
|
||||||
|
如需支持平台原生的消息反应功能,请在对应平台的子类中重写本方法。
|
||||||
|
"""
|
||||||
|
await self.send(MessageChain([Plain(emoji)]))
|
||||||
|
|
||||||
async def get_group(self, group_id: str = None, **kwargs) -> Optional[Group]:
|
async def get_group(self, group_id: str = None, **kwargs) -> Optional[Group]:
|
||||||
"""获取一个群聊的数据, 如果不填写 group_id: 如果是私聊消息,返回 None。如果是群聊消息,返回当前群聊的数据。
|
"""获取一个群聊的数据, 如果不填写 group_id: 如果是私聊消息,返回 None。如果是群聊消息,返回当前群聊的数据。
|
||||||
|
|
||||||
|
|||||||
@@ -55,7 +55,7 @@ class AstrBotMessage:
|
|||||||
self_id: str # 机器人的识别id
|
self_id: str # 机器人的识别id
|
||||||
session_id: str # 会话id。取决于 unique_session 的设置。
|
session_id: str # 会话id。取决于 unique_session 的设置。
|
||||||
message_id: str # 消息id
|
message_id: str # 消息id
|
||||||
group_id: str = "" # 群组id,如果为私聊,则为空
|
group: Group # 群组
|
||||||
sender: MessageMember # 发送者
|
sender: MessageMember # 发送者
|
||||||
message: List[BaseMessageComponent] # 消息链使用 Nakuru 的消息链格式
|
message: List[BaseMessageComponent] # 消息链使用 Nakuru 的消息链格式
|
||||||
message_str: str # 最直观的纯文本消息字符串
|
message_str: str # 最直观的纯文本消息字符串
|
||||||
@@ -64,6 +64,28 @@ class AstrBotMessage:
|
|||||||
|
|
||||||
def __init__(self) -> None:
|
def __init__(self) -> None:
|
||||||
self.timestamp = int(time.time())
|
self.timestamp = int(time.time())
|
||||||
|
self.group = None
|
||||||
|
|
||||||
def __str__(self) -> str:
|
def __str__(self) -> str:
|
||||||
return str(self.__dict__)
|
return str(self.__dict__)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def group_id(self) -> str:
|
||||||
|
"""
|
||||||
|
向后兼容的 group_id 属性
|
||||||
|
群组id,如果为私聊,则为空
|
||||||
|
"""
|
||||||
|
if self.group:
|
||||||
|
return self.group.group_id
|
||||||
|
return ""
|
||||||
|
|
||||||
|
@group_id.setter
|
||||||
|
def group_id(self, value: str):
|
||||||
|
"""设置 group_id"""
|
||||||
|
if value:
|
||||||
|
if self.group:
|
||||||
|
self.group.group_id = value
|
||||||
|
else:
|
||||||
|
self.group = Group(group_id=value)
|
||||||
|
else:
|
||||||
|
self.group = None
|
||||||
|
|||||||
@@ -6,6 +6,7 @@ from typing import List
|
|||||||
from asyncio import Queue
|
from asyncio import Queue
|
||||||
from .register import platform_cls_map
|
from .register import platform_cls_map
|
||||||
from astrbot.core import logger
|
from astrbot.core import logger
|
||||||
|
from astrbot.core.star.star_handler import star_handlers_registry, star_map, EventType
|
||||||
from .sources.webchat.webchat_adapter import WebChatAdapter
|
from .sources.webchat.webchat_adapter import WebChatAdapter
|
||||||
|
|
||||||
|
|
||||||
@@ -66,27 +67,39 @@ class PlatformManager:
|
|||||||
WeChatPadProAdapter, # noqa: F401
|
WeChatPadProAdapter, # noqa: F401
|
||||||
)
|
)
|
||||||
case "lark":
|
case "lark":
|
||||||
from .sources.lark.lark_adapter import LarkPlatformAdapter # noqa: F401
|
from .sources.lark.lark_adapter import (
|
||||||
|
LarkPlatformAdapter, # noqa: F401
|
||||||
|
)
|
||||||
case "dingtalk":
|
case "dingtalk":
|
||||||
from .sources.dingtalk.dingtalk_adapter import (
|
from .sources.dingtalk.dingtalk_adapter import (
|
||||||
DingtalkPlatformAdapter, # noqa: F401
|
DingtalkPlatformAdapter, # noqa: F401
|
||||||
)
|
)
|
||||||
case "telegram":
|
case "telegram":
|
||||||
from .sources.telegram.tg_adapter import TelegramPlatformAdapter # noqa: F401
|
from .sources.telegram.tg_adapter import (
|
||||||
|
TelegramPlatformAdapter, # noqa: F401
|
||||||
|
)
|
||||||
case "wecom":
|
case "wecom":
|
||||||
from .sources.wecom.wecom_adapter import WecomPlatformAdapter # noqa: F401
|
from .sources.wecom.wecom_adapter import (
|
||||||
|
WecomPlatformAdapter, # noqa: F401
|
||||||
|
)
|
||||||
case "weixin_official_account":
|
case "weixin_official_account":
|
||||||
from .sources.weixin_official_account.weixin_offacc_adapter import (
|
from .sources.weixin_official_account.weixin_offacc_adapter import (
|
||||||
WeixinOfficialAccountPlatformAdapter, # noqa
|
WeixinOfficialAccountPlatformAdapter, # noqa: F401
|
||||||
)
|
)
|
||||||
case "discord":
|
case "discord":
|
||||||
from .sources.discord.discord_platform_adapter import (
|
from .sources.discord.discord_platform_adapter import (
|
||||||
DiscordPlatformAdapter, # noqa: F401
|
DiscordPlatformAdapter, # noqa: F401
|
||||||
)
|
)
|
||||||
|
case "misskey":
|
||||||
|
from .sources.misskey.misskey_adapter import (
|
||||||
|
MisskeyPlatformAdapter, # noqa: F401
|
||||||
|
)
|
||||||
case "slack":
|
case "slack":
|
||||||
from .sources.slack.slack_adapter import SlackAdapter # noqa: F401
|
from .sources.slack.slack_adapter import SlackAdapter # noqa: F401
|
||||||
case "satori":
|
case "satori":
|
||||||
from .sources.satori.satori_adapter import SatoriPlatformAdapter # noqa: F401
|
from .sources.satori.satori_adapter import (
|
||||||
|
SatoriPlatformAdapter, # noqa: F401
|
||||||
|
)
|
||||||
except (ImportError, ModuleNotFoundError) as e:
|
except (ImportError, ModuleNotFoundError) as e:
|
||||||
logger.error(
|
logger.error(
|
||||||
f"加载平台适配器 {platform_config['type']} 失败,原因:{e}。请检查依赖库是否安装。提示:可以在 管理面板->控制台->安装Pip库 中安装依赖库。"
|
f"加载平台适配器 {platform_config['type']} 失败,原因:{e}。请检查依赖库是否安装。提示:可以在 管理面板->控制台->安装Pip库 中安装依赖库。"
|
||||||
@@ -115,6 +128,17 @@ class PlatformManager:
|
|||||||
)
|
)
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
handlers = star_handlers_registry.get_handlers_by_event_type(
|
||||||
|
EventType.OnPlatformLoadedEvent
|
||||||
|
)
|
||||||
|
for handler in handlers:
|
||||||
|
try:
|
||||||
|
logger.info(
|
||||||
|
f"hook(on_platform_loaded) -> {star_map[handler.handler_module_path].name} - {handler.handler_name}"
|
||||||
|
)
|
||||||
|
await handler.handler()
|
||||||
|
except Exception:
|
||||||
|
logger.error(traceback.format_exc())
|
||||||
|
|
||||||
async def _task_wrapper(self, task: asyncio.Task):
|
async def _task_wrapper(self, task: asyncio.Task):
|
||||||
try:
|
try:
|
||||||
|
|||||||
@@ -14,3 +14,5 @@ class PlatformMetadata:
|
|||||||
"""平台的默认配置模板"""
|
"""平台的默认配置模板"""
|
||||||
adapter_display_name: str = None
|
adapter_display_name: str = None
|
||||||
"""显示在 WebUI 配置页中的平台名称,如空则是 name"""
|
"""显示在 WebUI 配置页中的平台名称,如空则是 name"""
|
||||||
|
logo_path: str = None
|
||||||
|
"""平台适配器的 logo 文件路径(相对于插件目录)"""
|
||||||
|
|||||||
@@ -13,10 +13,12 @@ def register_platform_adapter(
|
|||||||
desc: str,
|
desc: str,
|
||||||
default_config_tmpl: dict = None,
|
default_config_tmpl: dict = None,
|
||||||
adapter_display_name: str = None,
|
adapter_display_name: str = None,
|
||||||
|
logo_path: str = None,
|
||||||
):
|
):
|
||||||
"""用于注册平台适配器的带参装饰器。
|
"""用于注册平台适配器的带参装饰器。
|
||||||
|
|
||||||
default_config_tmpl 指定了平台适配器的默认配置模板。用户填写好后将会作为 platform_config 传入你的 Platform 类的实现类。
|
default_config_tmpl 指定了平台适配器的默认配置模板。用户填写好后将会作为 platform_config 传入你的 Platform 类的实现类。
|
||||||
|
logo_path 指定了平台适配器的 logo 文件路径,是相对于插件目录的路径。
|
||||||
"""
|
"""
|
||||||
|
|
||||||
def decorator(cls):
|
def decorator(cls):
|
||||||
@@ -39,6 +41,7 @@ def register_platform_adapter(
|
|||||||
description=desc,
|
description=desc,
|
||||||
default_config_tmpl=default_config_tmpl,
|
default_config_tmpl=default_config_tmpl,
|
||||||
adapter_display_name=adapter_display_name,
|
adapter_display_name=adapter_display_name,
|
||||||
|
logo_path=logo_path,
|
||||||
)
|
)
|
||||||
platform_registry.append(pm)
|
platform_registry.append(pm)
|
||||||
platform_cls_map[adapter_name] = cls
|
platform_cls_map[adapter_name] = cls
|
||||||
|
|||||||
@@ -182,11 +182,13 @@ class AiocqhttpAdapter(Platform):
|
|||||||
abm = AstrBotMessage()
|
abm = AstrBotMessage()
|
||||||
abm.self_id = str(event.self_id)
|
abm.self_id = str(event.self_id)
|
||||||
abm.sender = MessageMember(
|
abm.sender = MessageMember(
|
||||||
str(event.sender["user_id"]), event.sender["nickname"]
|
str(event.sender["user_id"]),
|
||||||
|
event.sender.get("card") or event.sender.get("nickname", "N/A"),
|
||||||
)
|
)
|
||||||
if event["message_type"] == "group":
|
if event["message_type"] == "group":
|
||||||
abm.type = MessageType.GROUP_MESSAGE
|
abm.type = MessageType.GROUP_MESSAGE
|
||||||
abm.group_id = str(event.group_id)
|
abm.group_id = str(event.group_id)
|
||||||
|
abm.group.group_name = event.get("group_name", "N/A")
|
||||||
elif event["message_type"] == "private":
|
elif event["message_type"] == "private":
|
||||||
abm.type = MessageType.FRIEND_MESSAGE
|
abm.type = MessageType.FRIEND_MESSAGE
|
||||||
if self.unique_session and abm.type == MessageType.GROUP_MESSAGE:
|
if self.unique_session and abm.type == MessageType.GROUP_MESSAGE:
|
||||||
@@ -321,7 +323,9 @@ class AiocqhttpAdapter(Platform):
|
|||||||
user_id=int(m["data"]["qq"]),
|
user_id=int(m["data"]["qq"]),
|
||||||
no_cache=False,
|
no_cache=False,
|
||||||
)
|
)
|
||||||
nickname = at_info.get("nick", "") or at_info.get("nickname", "")
|
nickname = at_info.get("nick", "") or at_info.get(
|
||||||
|
"nickname", ""
|
||||||
|
)
|
||||||
is_at_self = str(m["data"]["qq"]) in {abm.self_id, "all"}
|
is_at_self = str(m["data"]["qq"]) in {abm.self_id, "all"}
|
||||||
|
|
||||||
abm.message.append(
|
abm.message.append(
|
||||||
|
|||||||
@@ -54,9 +54,9 @@ class DingtalkMessageEvent(AstrMessageEvent):
|
|||||||
logger.debug(f"send image: {ret}")
|
logger.debug(f"send image: {ret}")
|
||||||
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.error(f"钉钉图片处理失败: {e}")
|
logger.warning(f"钉钉图片处理失败: {e}, 跳过图片发送")
|
||||||
logger.warning(f"跳过图片发送: {image_path}")
|
|
||||||
continue
|
continue
|
||||||
|
|
||||||
async def send(self, message: MessageChain):
|
async def send(self, message: MessageChain):
|
||||||
await self.send_with_client(self.client, message)
|
await self.send_with_client(self.client, message)
|
||||||
await super().send(message)
|
await super().send(message)
|
||||||
|
|||||||
@@ -41,7 +41,8 @@ class DiscordBotClient(discord.Bot):
|
|||||||
await self.on_ready_once_callback()
|
await self.on_ready_once_callback()
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.error(
|
logger.error(
|
||||||
f"[Discord] on_ready_once_callback 执行失败: {e}", exc_info=True)
|
f"[Discord] on_ready_once_callback 执行失败: {e}", exc_info=True
|
||||||
|
)
|
||||||
|
|
||||||
def _create_message_data(self, message: discord.Message) -> dict:
|
def _create_message_data(self, message: discord.Message) -> dict:
|
||||||
"""从 discord.Message 创建数据字典"""
|
"""从 discord.Message 创建数据字典"""
|
||||||
@@ -90,7 +91,6 @@ class DiscordBotClient(discord.Bot):
|
|||||||
message_data = self._create_message_data(message)
|
message_data = self._create_message_data(message)
|
||||||
await self.on_message_received(message_data)
|
await self.on_message_received(message_data)
|
||||||
|
|
||||||
|
|
||||||
def _extract_interaction_content(self, interaction: discord.Interaction) -> str:
|
def _extract_interaction_content(self, interaction: discord.Interaction) -> str:
|
||||||
"""从交互中提取内容"""
|
"""从交互中提取内容"""
|
||||||
interaction_type = interaction.type
|
interaction_type = interaction.type
|
||||||
|
|||||||
@@ -79,9 +79,12 @@ class DiscordButton(BaseMessageComponent):
|
|||||||
self.url = url
|
self.url = url
|
||||||
self.disabled = disabled
|
self.disabled = disabled
|
||||||
|
|
||||||
|
|
||||||
class DiscordReference(BaseMessageComponent):
|
class DiscordReference(BaseMessageComponent):
|
||||||
"""Discord引用组件"""
|
"""Discord引用组件"""
|
||||||
|
|
||||||
type: str = "discord_reference"
|
type: str = "discord_reference"
|
||||||
|
|
||||||
def __init__(self, message_id: str, channel_id: str):
|
def __init__(self, message_id: str, channel_id: str):
|
||||||
self.message_id = message_id
|
self.message_id = message_id
|
||||||
self.channel_id = channel_id
|
self.channel_id = channel_id
|
||||||
@@ -98,7 +101,6 @@ class DiscordView(BaseMessageComponent):
|
|||||||
self.components = components or []
|
self.components = components or []
|
||||||
self.timeout = timeout
|
self.timeout = timeout
|
||||||
|
|
||||||
|
|
||||||
def to_discord_view(self) -> discord.ui.View:
|
def to_discord_view(self) -> discord.ui.View:
|
||||||
"""转换为Discord View对象"""
|
"""转换为Discord View对象"""
|
||||||
view = discord.ui.View(timeout=self.timeout)
|
view = discord.ui.View(timeout=self.timeout)
|
||||||
|
|||||||
@@ -53,7 +53,13 @@ class DiscordPlatformEvent(AstrMessageEvent):
|
|||||||
|
|
||||||
# 解析消息链为 Discord 所需的对象
|
# 解析消息链为 Discord 所需的对象
|
||||||
try:
|
try:
|
||||||
content, files, view, embeds, reference_message_id = await self._parse_to_discord(message)
|
(
|
||||||
|
content,
|
||||||
|
files,
|
||||||
|
view,
|
||||||
|
embeds,
|
||||||
|
reference_message_id,
|
||||||
|
) = await self._parse_to_discord(message)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.error(f"[Discord] 解析消息链时失败: {e}", exc_info=True)
|
logger.error(f"[Discord] 解析消息链时失败: {e}", exc_info=True)
|
||||||
return
|
return
|
||||||
@@ -206,8 +212,7 @@ class DiscordPlatformEvent(AstrMessageEvent):
|
|||||||
if await asyncio.to_thread(path.exists):
|
if await asyncio.to_thread(path.exists):
|
||||||
file_bytes = await asyncio.to_thread(path.read_bytes)
|
file_bytes = await asyncio.to_thread(path.read_bytes)
|
||||||
files.append(
|
files.append(
|
||||||
discord.File(BytesIO(file_bytes),
|
discord.File(BytesIO(file_bytes), filename=i.name)
|
||||||
filename=i.name)
|
|
||||||
)
|
)
|
||||||
else:
|
else:
|
||||||
logger.warning(
|
logger.warning(
|
||||||
|
|||||||
@@ -107,6 +107,22 @@ class LarkMessageEvent(AstrMessageEvent):
|
|||||||
|
|
||||||
await super().send(message)
|
await super().send(message)
|
||||||
|
|
||||||
|
async def react(self, emoji: str):
|
||||||
|
request = (
|
||||||
|
CreateMessageReactionRequest.builder()
|
||||||
|
.message_id(self.message_obj.message_id)
|
||||||
|
.request_body(
|
||||||
|
CreateMessageReactionRequestBody.builder()
|
||||||
|
.reaction_type(Emoji.builder().emoji_type(emoji).build())
|
||||||
|
.build()
|
||||||
|
)
|
||||||
|
.build()
|
||||||
|
)
|
||||||
|
response = await self.bot.im.v1.message_reaction.acreate(request)
|
||||||
|
if not response.success():
|
||||||
|
logger.error(f"发送飞书表情回应失败({response.code}): {response.msg}")
|
||||||
|
return None
|
||||||
|
|
||||||
async def send_streaming(self, generator, use_fallback: bool = False):
|
async def send_streaming(self, generator, use_fallback: bool = False):
|
||||||
buffer = None
|
buffer = None
|
||||||
async for chain in generator:
|
async for chain in generator:
|
||||||
|
|||||||
391
astrbot/core/platform/sources/misskey/misskey_adapter.py
Normal file
391
astrbot/core/platform/sources/misskey/misskey_adapter.py
Normal file
@@ -0,0 +1,391 @@
|
|||||||
|
import asyncio
|
||||||
|
import json
|
||||||
|
from typing import Dict, Any, Optional, Awaitable
|
||||||
|
|
||||||
|
from astrbot.api import logger
|
||||||
|
from astrbot.api.event import MessageChain
|
||||||
|
from astrbot.api.platform import (
|
||||||
|
AstrBotMessage,
|
||||||
|
Platform,
|
||||||
|
PlatformMetadata,
|
||||||
|
register_platform_adapter,
|
||||||
|
)
|
||||||
|
from astrbot.core.platform.astr_message_event import MessageSession
|
||||||
|
import astrbot.api.message_components as Comp
|
||||||
|
|
||||||
|
from .misskey_api import MisskeyAPI
|
||||||
|
from .misskey_event import MisskeyPlatformEvent
|
||||||
|
from .misskey_utils import (
|
||||||
|
serialize_message_chain,
|
||||||
|
resolve_message_visibility,
|
||||||
|
is_valid_user_session_id,
|
||||||
|
is_valid_room_session_id,
|
||||||
|
add_at_mention_if_needed,
|
||||||
|
process_files,
|
||||||
|
extract_sender_info,
|
||||||
|
create_base_message,
|
||||||
|
process_at_mention,
|
||||||
|
cache_user_info,
|
||||||
|
cache_room_info,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@register_platform_adapter("misskey", "Misskey 平台适配器")
|
||||||
|
class MisskeyPlatformAdapter(Platform):
|
||||||
|
def __init__(
|
||||||
|
self, platform_config: dict, platform_settings: dict, event_queue: asyncio.Queue
|
||||||
|
) -> None:
|
||||||
|
super().__init__(event_queue)
|
||||||
|
self.config = platform_config or {}
|
||||||
|
self.settings = platform_settings or {}
|
||||||
|
self.instance_url = self.config.get("misskey_instance_url", "")
|
||||||
|
self.access_token = self.config.get("misskey_token", "")
|
||||||
|
self.max_message_length = self.config.get("max_message_length", 3000)
|
||||||
|
self.default_visibility = self.config.get(
|
||||||
|
"misskey_default_visibility", "public"
|
||||||
|
)
|
||||||
|
self.local_only = self.config.get("misskey_local_only", False)
|
||||||
|
self.enable_chat = self.config.get("misskey_enable_chat", True)
|
||||||
|
|
||||||
|
self.unique_session = platform_settings["unique_session"]
|
||||||
|
|
||||||
|
self.api: Optional[MisskeyAPI] = None
|
||||||
|
self._running = False
|
||||||
|
self.client_self_id = ""
|
||||||
|
self._bot_username = ""
|
||||||
|
self._user_cache = {}
|
||||||
|
|
||||||
|
def meta(self) -> PlatformMetadata:
|
||||||
|
default_config = {
|
||||||
|
"misskey_instance_url": "",
|
||||||
|
"misskey_token": "",
|
||||||
|
"max_message_length": 3000,
|
||||||
|
"misskey_default_visibility": "public",
|
||||||
|
"misskey_local_only": False,
|
||||||
|
"misskey_enable_chat": True,
|
||||||
|
}
|
||||||
|
default_config.update(self.config)
|
||||||
|
|
||||||
|
return PlatformMetadata(
|
||||||
|
name="misskey",
|
||||||
|
description="Misskey 平台适配器",
|
||||||
|
id=self.config.get("id", "misskey"),
|
||||||
|
default_config_tmpl=default_config,
|
||||||
|
)
|
||||||
|
|
||||||
|
async def run(self):
|
||||||
|
if not self.instance_url or not self.access_token:
|
||||||
|
logger.error("[Misskey] 配置不完整,无法启动")
|
||||||
|
return
|
||||||
|
|
||||||
|
self.api = MisskeyAPI(self.instance_url, self.access_token)
|
||||||
|
self._running = True
|
||||||
|
|
||||||
|
try:
|
||||||
|
user_info = await self.api.get_current_user()
|
||||||
|
self.client_self_id = str(user_info.get("id", ""))
|
||||||
|
self._bot_username = user_info.get("username", "")
|
||||||
|
logger.info(
|
||||||
|
f"[Misskey] 已连接用户: {self._bot_username} (ID: {self.client_self_id})"
|
||||||
|
)
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"[Misskey] 获取用户信息失败: {e}")
|
||||||
|
self._running = False
|
||||||
|
return
|
||||||
|
|
||||||
|
await self._start_websocket_connection()
|
||||||
|
|
||||||
|
async def _start_websocket_connection(self):
|
||||||
|
backoff_delay = 1.0
|
||||||
|
max_backoff = 300.0
|
||||||
|
backoff_multiplier = 1.5
|
||||||
|
connection_attempts = 0
|
||||||
|
|
||||||
|
while self._running:
|
||||||
|
try:
|
||||||
|
connection_attempts += 1
|
||||||
|
if not self.api:
|
||||||
|
logger.error("[Misskey] API 客户端未初始化")
|
||||||
|
break
|
||||||
|
|
||||||
|
streaming = self.api.get_streaming_client()
|
||||||
|
streaming.add_message_handler("notification", self._handle_notification)
|
||||||
|
if self.enable_chat:
|
||||||
|
streaming.add_message_handler(
|
||||||
|
"newChatMessage", self._handle_chat_message
|
||||||
|
)
|
||||||
|
streaming.add_message_handler("_debug", self._debug_handler)
|
||||||
|
|
||||||
|
if await streaming.connect():
|
||||||
|
logger.info(
|
||||||
|
f"[Misskey] WebSocket 已连接 (尝试 #{connection_attempts})"
|
||||||
|
)
|
||||||
|
connection_attempts = 0 # 重置计数器
|
||||||
|
await streaming.subscribe_channel("main")
|
||||||
|
if self.enable_chat:
|
||||||
|
await streaming.subscribe_channel("messaging")
|
||||||
|
await streaming.subscribe_channel("messagingIndex")
|
||||||
|
logger.info("[Misskey] 聊天频道已订阅")
|
||||||
|
|
||||||
|
backoff_delay = 1.0 # 重置延迟
|
||||||
|
await streaming.listen()
|
||||||
|
else:
|
||||||
|
logger.error(
|
||||||
|
f"[Misskey] WebSocket 连接失败 (尝试 #{connection_attempts})"
|
||||||
|
)
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(
|
||||||
|
f"[Misskey] WebSocket 异常 (尝试 #{connection_attempts}): {e}"
|
||||||
|
)
|
||||||
|
|
||||||
|
if self._running:
|
||||||
|
logger.info(
|
||||||
|
f"[Misskey] {backoff_delay:.1f}秒后重连 (下次尝试 #{connection_attempts + 1})"
|
||||||
|
)
|
||||||
|
await asyncio.sleep(backoff_delay)
|
||||||
|
backoff_delay = min(backoff_delay * backoff_multiplier, max_backoff)
|
||||||
|
|
||||||
|
async def _handle_notification(self, data: Dict[str, Any]):
|
||||||
|
try:
|
||||||
|
logger.debug(
|
||||||
|
f"[Misskey] 收到通知事件:\n{json.dumps(data, indent=2, ensure_ascii=False)}"
|
||||||
|
)
|
||||||
|
notification_type = data.get("type")
|
||||||
|
if notification_type in ["mention", "reply", "quote"]:
|
||||||
|
note = data.get("note")
|
||||||
|
if note and self._is_bot_mentioned(note):
|
||||||
|
logger.info(
|
||||||
|
f"[Misskey] 处理贴文提及: {note.get('text', '')[:50]}..."
|
||||||
|
)
|
||||||
|
message = await self.convert_message(note)
|
||||||
|
event = MisskeyPlatformEvent(
|
||||||
|
message_str=message.message_str,
|
||||||
|
message_obj=message,
|
||||||
|
platform_meta=self.meta(),
|
||||||
|
session_id=message.session_id,
|
||||||
|
client=self.api,
|
||||||
|
)
|
||||||
|
self.commit_event(event)
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"[Misskey] 处理通知失败: {e}")
|
||||||
|
|
||||||
|
async def _handle_chat_message(self, data: Dict[str, Any]):
|
||||||
|
try:
|
||||||
|
logger.debug(
|
||||||
|
f"[Misskey] 收到聊天事件数据:\n{json.dumps(data, indent=2, ensure_ascii=False)}"
|
||||||
|
)
|
||||||
|
|
||||||
|
sender_id = str(
|
||||||
|
data.get("fromUserId", "") or data.get("fromUser", {}).get("id", "")
|
||||||
|
)
|
||||||
|
if sender_id == self.client_self_id:
|
||||||
|
return
|
||||||
|
|
||||||
|
room_id = data.get("toRoomId")
|
||||||
|
if room_id:
|
||||||
|
raw_text = data.get("text", "")
|
||||||
|
logger.debug(
|
||||||
|
f"[Misskey] 检查群聊消息: '{raw_text}', 机器人用户名: '{self._bot_username}'"
|
||||||
|
)
|
||||||
|
|
||||||
|
message = await self.convert_room_message(data)
|
||||||
|
logger.info(f"[Misskey] 处理群聊消息: {message.message_str[:50]}...")
|
||||||
|
else:
|
||||||
|
message = await self.convert_chat_message(data)
|
||||||
|
logger.info(f"[Misskey] 处理私聊消息: {message.message_str[:50]}...")
|
||||||
|
|
||||||
|
event = MisskeyPlatformEvent(
|
||||||
|
message_str=message.message_str,
|
||||||
|
message_obj=message,
|
||||||
|
platform_meta=self.meta(),
|
||||||
|
session_id=message.session_id,
|
||||||
|
client=self.api,
|
||||||
|
)
|
||||||
|
self.commit_event(event)
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"[Misskey] 处理聊天消息失败: {e}")
|
||||||
|
|
||||||
|
async def _debug_handler(self, data: Dict[str, Any]):
|
||||||
|
logger.debug(
|
||||||
|
f"[Misskey] 收到未处理事件:\n{json.dumps(data, indent=2, ensure_ascii=False)}"
|
||||||
|
)
|
||||||
|
|
||||||
|
def _is_bot_mentioned(self, note: Dict[str, Any]) -> bool:
|
||||||
|
text = note.get("text", "")
|
||||||
|
if not text:
|
||||||
|
return False
|
||||||
|
|
||||||
|
mentions = note.get("mentions", [])
|
||||||
|
if self._bot_username and f"@{self._bot_username}" in text:
|
||||||
|
return True
|
||||||
|
if self.client_self_id in [str(uid) for uid in mentions]:
|
||||||
|
return True
|
||||||
|
|
||||||
|
reply = note.get("reply")
|
||||||
|
if reply and isinstance(reply, dict):
|
||||||
|
reply_user_id = str(reply.get("user", {}).get("id", ""))
|
||||||
|
if reply_user_id == self.client_self_id:
|
||||||
|
return bool(self._bot_username and f"@{self._bot_username}" in text)
|
||||||
|
|
||||||
|
return False
|
||||||
|
|
||||||
|
async def send_by_session(
|
||||||
|
self, session: MessageSession, message_chain: MessageChain
|
||||||
|
) -> Awaitable[Any]:
|
||||||
|
if not self.api:
|
||||||
|
logger.error("[Misskey] API 客户端未初始化")
|
||||||
|
return await super().send_by_session(session, message_chain)
|
||||||
|
|
||||||
|
try:
|
||||||
|
session_id = session.session_id
|
||||||
|
text, has_at_user = serialize_message_chain(message_chain.chain)
|
||||||
|
|
||||||
|
if not has_at_user and session_id:
|
||||||
|
user_info = self._user_cache.get(session_id)
|
||||||
|
text = add_at_mention_if_needed(text, user_info, has_at_user)
|
||||||
|
|
||||||
|
if not text or not text.strip():
|
||||||
|
logger.warning("[Misskey] 消息内容为空,跳过发送")
|
||||||
|
return await super().send_by_session(session, message_chain)
|
||||||
|
|
||||||
|
if len(text) > self.max_message_length:
|
||||||
|
text = text[: self.max_message_length] + "..."
|
||||||
|
|
||||||
|
if session_id and is_valid_user_session_id(session_id):
|
||||||
|
from .misskey_utils import extract_user_id_from_session_id
|
||||||
|
|
||||||
|
user_id = extract_user_id_from_session_id(session_id)
|
||||||
|
await self.api.send_message(user_id, text)
|
||||||
|
elif session_id and is_valid_room_session_id(session_id):
|
||||||
|
from .misskey_utils import extract_room_id_from_session_id
|
||||||
|
|
||||||
|
room_id = extract_room_id_from_session_id(session_id)
|
||||||
|
await self.api.send_room_message(room_id, text)
|
||||||
|
else:
|
||||||
|
visibility, visible_user_ids = resolve_message_visibility(
|
||||||
|
user_id=session_id,
|
||||||
|
user_cache=self._user_cache,
|
||||||
|
self_id=self.client_self_id,
|
||||||
|
default_visibility=self.default_visibility,
|
||||||
|
)
|
||||||
|
|
||||||
|
await self.api.create_note(
|
||||||
|
text,
|
||||||
|
visibility=visibility,
|
||||||
|
visible_user_ids=visible_user_ids,
|
||||||
|
local_only=self.local_only,
|
||||||
|
)
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"[Misskey] 发送消息失败: {e}")
|
||||||
|
|
||||||
|
return await super().send_by_session(session, message_chain)
|
||||||
|
|
||||||
|
async def convert_message(self, raw_data: Dict[str, Any]) -> AstrBotMessage:
|
||||||
|
"""将 Misskey 贴文数据转换为 AstrBotMessage 对象"""
|
||||||
|
sender_info = extract_sender_info(raw_data, is_chat=False)
|
||||||
|
message = create_base_message(
|
||||||
|
raw_data,
|
||||||
|
sender_info,
|
||||||
|
self.client_self_id,
|
||||||
|
is_chat=False,
|
||||||
|
unique_session=self.unique_session,
|
||||||
|
)
|
||||||
|
cache_user_info(
|
||||||
|
self._user_cache, sender_info, raw_data, self.client_self_id, is_chat=False
|
||||||
|
)
|
||||||
|
|
||||||
|
message_parts = []
|
||||||
|
raw_text = raw_data.get("text", "")
|
||||||
|
|
||||||
|
if raw_text:
|
||||||
|
text_parts, processed_text = process_at_mention(
|
||||||
|
message, raw_text, self._bot_username, self.client_self_id
|
||||||
|
)
|
||||||
|
message_parts.extend(text_parts)
|
||||||
|
|
||||||
|
files = raw_data.get("files", [])
|
||||||
|
file_parts = process_files(message, files)
|
||||||
|
message_parts.extend(file_parts)
|
||||||
|
|
||||||
|
message.message_str = (
|
||||||
|
" ".join(part for part in message_parts if part.strip())
|
||||||
|
if message_parts
|
||||||
|
else ""
|
||||||
|
)
|
||||||
|
return message
|
||||||
|
|
||||||
|
async def convert_chat_message(self, raw_data: Dict[str, Any]) -> AstrBotMessage:
|
||||||
|
"""将 Misskey 聊天消息数据转换为 AstrBotMessage 对象"""
|
||||||
|
sender_info = extract_sender_info(raw_data, is_chat=True)
|
||||||
|
message = create_base_message(
|
||||||
|
raw_data,
|
||||||
|
sender_info,
|
||||||
|
self.client_self_id,
|
||||||
|
is_chat=True,
|
||||||
|
unique_session=self.unique_session,
|
||||||
|
)
|
||||||
|
cache_user_info(
|
||||||
|
self._user_cache, sender_info, raw_data, self.client_self_id, is_chat=True
|
||||||
|
)
|
||||||
|
|
||||||
|
raw_text = raw_data.get("text", "")
|
||||||
|
if raw_text:
|
||||||
|
message.message.append(Comp.Plain(raw_text))
|
||||||
|
|
||||||
|
files = raw_data.get("files", [])
|
||||||
|
process_files(message, files, include_text_parts=False)
|
||||||
|
|
||||||
|
message.message_str = raw_text if raw_text else ""
|
||||||
|
return message
|
||||||
|
|
||||||
|
async def convert_room_message(self, raw_data: Dict[str, Any]) -> AstrBotMessage:
|
||||||
|
"""将 Misskey 群聊消息数据转换为 AstrBotMessage 对象"""
|
||||||
|
sender_info = extract_sender_info(raw_data, is_chat=True)
|
||||||
|
room_id = raw_data.get("toRoomId", "")
|
||||||
|
message = create_base_message(
|
||||||
|
raw_data,
|
||||||
|
sender_info,
|
||||||
|
self.client_self_id,
|
||||||
|
is_chat=False,
|
||||||
|
room_id=room_id,
|
||||||
|
unique_session=self.unique_session,
|
||||||
|
)
|
||||||
|
|
||||||
|
cache_user_info(
|
||||||
|
self._user_cache, sender_info, raw_data, self.client_self_id, is_chat=False
|
||||||
|
)
|
||||||
|
cache_room_info(self._user_cache, raw_data, self.client_self_id)
|
||||||
|
|
||||||
|
raw_text = raw_data.get("text", "")
|
||||||
|
message_parts = []
|
||||||
|
|
||||||
|
if raw_text:
|
||||||
|
if self._bot_username and f"@{self._bot_username}" in raw_text:
|
||||||
|
text_parts, processed_text = process_at_mention(
|
||||||
|
message, raw_text, self._bot_username, self.client_self_id
|
||||||
|
)
|
||||||
|
message_parts.extend(text_parts)
|
||||||
|
else:
|
||||||
|
message.message.append(Comp.Plain(raw_text))
|
||||||
|
message_parts.append(raw_text)
|
||||||
|
|
||||||
|
files = raw_data.get("files", [])
|
||||||
|
file_parts = process_files(message, files)
|
||||||
|
message_parts.extend(file_parts)
|
||||||
|
|
||||||
|
message.message_str = (
|
||||||
|
" ".join(part for part in message_parts if part.strip())
|
||||||
|
if message_parts
|
||||||
|
else ""
|
||||||
|
)
|
||||||
|
return message
|
||||||
|
|
||||||
|
async def terminate(self):
|
||||||
|
self._running = False
|
||||||
|
if self.api:
|
||||||
|
await self.api.close()
|
||||||
|
|
||||||
|
def get_client(self) -> Any:
|
||||||
|
return self.api
|
||||||
404
astrbot/core/platform/sources/misskey/misskey_api.py
Normal file
404
astrbot/core/platform/sources/misskey/misskey_api.py
Normal file
@@ -0,0 +1,404 @@
|
|||||||
|
import json
|
||||||
|
from typing import Any, Optional, Dict, List, Callable, Awaitable
|
||||||
|
import uuid
|
||||||
|
|
||||||
|
try:
|
||||||
|
import aiohttp
|
||||||
|
import websockets
|
||||||
|
except ImportError as e:
|
||||||
|
raise ImportError(
|
||||||
|
"aiohttp and websockets are required for Misskey API. Please install them with: pip install aiohttp websockets"
|
||||||
|
) from e
|
||||||
|
|
||||||
|
from astrbot.api import logger
|
||||||
|
|
||||||
|
# Constants
|
||||||
|
API_MAX_RETRIES = 3
|
||||||
|
HTTP_OK = 200
|
||||||
|
|
||||||
|
|
||||||
|
class APIError(Exception):
|
||||||
|
"""Misskey API 基础异常"""
|
||||||
|
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
class APIConnectionError(APIError):
|
||||||
|
"""网络连接异常"""
|
||||||
|
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
class APIRateLimitError(APIError):
|
||||||
|
"""API 频率限制异常"""
|
||||||
|
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
class AuthenticationError(APIError):
|
||||||
|
"""认证失败异常"""
|
||||||
|
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
class WebSocketError(APIError):
|
||||||
|
"""WebSocket 连接异常"""
|
||||||
|
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
class StreamingClient:
|
||||||
|
def __init__(self, instance_url: str, access_token: str):
|
||||||
|
self.instance_url = instance_url.rstrip("/")
|
||||||
|
self.access_token = access_token
|
||||||
|
self.websocket: Optional[Any] = None
|
||||||
|
self.is_connected = False
|
||||||
|
self.message_handlers: Dict[str, Callable] = {}
|
||||||
|
self.channels: Dict[str, str] = {}
|
||||||
|
self._running = False
|
||||||
|
self._last_pong = None
|
||||||
|
|
||||||
|
async def connect(self) -> bool:
|
||||||
|
try:
|
||||||
|
ws_url = self.instance_url.replace("https://", "wss://").replace(
|
||||||
|
"http://", "ws://"
|
||||||
|
)
|
||||||
|
ws_url += f"/streaming?i={self.access_token}"
|
||||||
|
|
||||||
|
self.websocket = await websockets.connect(
|
||||||
|
ws_url, ping_interval=30, ping_timeout=10
|
||||||
|
)
|
||||||
|
self.is_connected = True
|
||||||
|
self._running = True
|
||||||
|
|
||||||
|
logger.info("[Misskey WebSocket] 已连接")
|
||||||
|
return True
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"[Misskey WebSocket] 连接失败: {e}")
|
||||||
|
self.is_connected = False
|
||||||
|
return False
|
||||||
|
|
||||||
|
async def disconnect(self):
|
||||||
|
self._running = False
|
||||||
|
if self.websocket:
|
||||||
|
await self.websocket.close()
|
||||||
|
self.websocket = None
|
||||||
|
self.is_connected = False
|
||||||
|
logger.info("[Misskey WebSocket] 连接已断开")
|
||||||
|
|
||||||
|
async def subscribe_channel(
|
||||||
|
self, channel_type: str, params: Optional[Dict] = None
|
||||||
|
) -> str:
|
||||||
|
if not self.is_connected or not self.websocket:
|
||||||
|
raise WebSocketError("WebSocket 未连接")
|
||||||
|
|
||||||
|
channel_id = str(uuid.uuid4())
|
||||||
|
message = {
|
||||||
|
"type": "connect",
|
||||||
|
"body": {"channel": channel_type, "id": channel_id, "params": params or {}},
|
||||||
|
}
|
||||||
|
|
||||||
|
await self.websocket.send(json.dumps(message))
|
||||||
|
self.channels[channel_id] = channel_type
|
||||||
|
return channel_id
|
||||||
|
|
||||||
|
async def unsubscribe_channel(self, channel_id: str):
|
||||||
|
if (
|
||||||
|
not self.is_connected
|
||||||
|
or not self.websocket
|
||||||
|
or channel_id not in self.channels
|
||||||
|
):
|
||||||
|
return
|
||||||
|
|
||||||
|
message = {"type": "disconnect", "body": {"id": channel_id}}
|
||||||
|
|
||||||
|
await self.websocket.send(json.dumps(message))
|
||||||
|
del self.channels[channel_id]
|
||||||
|
|
||||||
|
def add_message_handler(
|
||||||
|
self, event_type: str, handler: Callable[[Dict], Awaitable[None]]
|
||||||
|
):
|
||||||
|
self.message_handlers[event_type] = handler
|
||||||
|
|
||||||
|
async def listen(self):
|
||||||
|
if not self.is_connected or not self.websocket:
|
||||||
|
raise WebSocketError("WebSocket 未连接")
|
||||||
|
|
||||||
|
try:
|
||||||
|
async for message in self.websocket:
|
||||||
|
if not self._running:
|
||||||
|
break
|
||||||
|
|
||||||
|
try:
|
||||||
|
data = json.loads(message)
|
||||||
|
await self._handle_message(data)
|
||||||
|
except json.JSONDecodeError as e:
|
||||||
|
logger.warning(f"[Misskey WebSocket] 无法解析消息: {e}")
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"[Misskey WebSocket] 处理消息失败: {e}")
|
||||||
|
|
||||||
|
except websockets.exceptions.ConnectionClosedError as e:
|
||||||
|
logger.warning(f"[Misskey WebSocket] 连接意外关闭: {e}")
|
||||||
|
self.is_connected = False
|
||||||
|
except websockets.exceptions.ConnectionClosed as e:
|
||||||
|
logger.warning(
|
||||||
|
f"[Misskey WebSocket] 连接已关闭 (代码: {e.code}, 原因: {e.reason})"
|
||||||
|
)
|
||||||
|
self.is_connected = False
|
||||||
|
except websockets.exceptions.InvalidHandshake as e:
|
||||||
|
logger.error(f"[Misskey WebSocket] 握手失败: {e}")
|
||||||
|
self.is_connected = False
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"[Misskey WebSocket] 监听消息失败: {e}")
|
||||||
|
self.is_connected = False
|
||||||
|
|
||||||
|
async def _handle_message(self, data: Dict[str, Any]):
|
||||||
|
message_type = data.get("type")
|
||||||
|
body = data.get("body", {})
|
||||||
|
|
||||||
|
logger.debug(
|
||||||
|
f"[Misskey WebSocket] 收到消息类型: {message_type}\n数据: {json.dumps(data, indent=2, ensure_ascii=False)}"
|
||||||
|
)
|
||||||
|
|
||||||
|
if message_type == "channel":
|
||||||
|
channel_id = body.get("id")
|
||||||
|
event_type = body.get("type")
|
||||||
|
event_body = body.get("body", {})
|
||||||
|
|
||||||
|
logger.debug(
|
||||||
|
f"[Misskey WebSocket] 频道消息: {channel_id}, 事件类型: {event_type}"
|
||||||
|
)
|
||||||
|
|
||||||
|
if channel_id in self.channels:
|
||||||
|
channel_type = self.channels[channel_id]
|
||||||
|
handler_key = f"{channel_type}:{event_type}"
|
||||||
|
|
||||||
|
if handler_key in self.message_handlers:
|
||||||
|
logger.debug(f"[Misskey WebSocket] 使用处理器: {handler_key}")
|
||||||
|
await self.message_handlers[handler_key](event_body)
|
||||||
|
elif event_type in self.message_handlers:
|
||||||
|
logger.debug(f"[Misskey WebSocket] 使用事件处理器: {event_type}")
|
||||||
|
await self.message_handlers[event_type](event_body)
|
||||||
|
else:
|
||||||
|
logger.debug(
|
||||||
|
f"[Misskey WebSocket] 未找到处理器: {handler_key} 或 {event_type}"
|
||||||
|
)
|
||||||
|
if "_debug" in self.message_handlers:
|
||||||
|
await self.message_handlers["_debug"](
|
||||||
|
{
|
||||||
|
"type": event_type,
|
||||||
|
"body": event_body,
|
||||||
|
"channel": channel_type,
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
elif message_type in self.message_handlers:
|
||||||
|
logger.debug(f"[Misskey WebSocket] 直接消息处理器: {message_type}")
|
||||||
|
await self.message_handlers[message_type](body)
|
||||||
|
else:
|
||||||
|
logger.debug(f"[Misskey WebSocket] 未处理的消息类型: {message_type}")
|
||||||
|
if "_debug" in self.message_handlers:
|
||||||
|
await self.message_handlers["_debug"](data)
|
||||||
|
|
||||||
|
|
||||||
|
def retry_async(max_retries: int = 3, retryable_exceptions: tuple = ()):
|
||||||
|
def decorator(func):
|
||||||
|
async def wrapper(*args, **kwargs):
|
||||||
|
last_exc = None
|
||||||
|
for _ in range(max_retries):
|
||||||
|
try:
|
||||||
|
return await func(*args, **kwargs)
|
||||||
|
except retryable_exceptions as e:
|
||||||
|
last_exc = e
|
||||||
|
continue
|
||||||
|
if last_exc:
|
||||||
|
raise last_exc
|
||||||
|
|
||||||
|
return wrapper
|
||||||
|
|
||||||
|
return decorator
|
||||||
|
|
||||||
|
|
||||||
|
class MisskeyAPI:
|
||||||
|
def __init__(self, instance_url: str, access_token: str):
|
||||||
|
self.instance_url = instance_url.rstrip("/")
|
||||||
|
self.access_token = access_token
|
||||||
|
self._session: Optional[aiohttp.ClientSession] = None
|
||||||
|
self.streaming: Optional[StreamingClient] = None
|
||||||
|
|
||||||
|
async def __aenter__(self):
|
||||||
|
return self
|
||||||
|
|
||||||
|
async def __aexit__(self, exc_type, exc_val, exc_tb):
|
||||||
|
await self.close()
|
||||||
|
return False
|
||||||
|
|
||||||
|
async def close(self) -> None:
|
||||||
|
if self.streaming:
|
||||||
|
await self.streaming.disconnect()
|
||||||
|
self.streaming = None
|
||||||
|
if self._session:
|
||||||
|
await self._session.close()
|
||||||
|
self._session = None
|
||||||
|
logger.debug("[Misskey API] 客户端已关闭")
|
||||||
|
|
||||||
|
def get_streaming_client(self) -> StreamingClient:
|
||||||
|
if not self.streaming:
|
||||||
|
self.streaming = StreamingClient(self.instance_url, self.access_token)
|
||||||
|
return self.streaming
|
||||||
|
|
||||||
|
@property
|
||||||
|
def session(self) -> aiohttp.ClientSession:
|
||||||
|
if self._session is None or self._session.closed:
|
||||||
|
headers = {"Authorization": f"Bearer {self.access_token}"}
|
||||||
|
self._session = aiohttp.ClientSession(headers=headers)
|
||||||
|
return self._session
|
||||||
|
|
||||||
|
def _handle_response_status(self, status: int, endpoint: str):
|
||||||
|
"""处理 HTTP 响应状态码"""
|
||||||
|
if status == 400:
|
||||||
|
logger.error(f"API 请求错误: {endpoint} (状态码: {status})")
|
||||||
|
raise APIError(f"Bad request for {endpoint}")
|
||||||
|
elif status in (401, 403):
|
||||||
|
logger.error(f"API 认证失败: {endpoint} (状态码: {status})")
|
||||||
|
raise AuthenticationError(f"Authentication failed for {endpoint}")
|
||||||
|
elif status == 429:
|
||||||
|
logger.warning(f"API 频率限制: {endpoint} (状态码: {status})")
|
||||||
|
raise APIRateLimitError(f"Rate limit exceeded for {endpoint}")
|
||||||
|
else:
|
||||||
|
logger.error(f"API 请求失败: {endpoint} (状态码: {status})")
|
||||||
|
raise APIConnectionError(f"HTTP {status} for {endpoint}")
|
||||||
|
|
||||||
|
async def _process_response(
|
||||||
|
self, response: aiohttp.ClientResponse, endpoint: str
|
||||||
|
) -> Any:
|
||||||
|
"""处理 API 响应"""
|
||||||
|
if response.status == HTTP_OK:
|
||||||
|
try:
|
||||||
|
result = await response.json()
|
||||||
|
if endpoint == "i/notifications":
|
||||||
|
notifications_data = (
|
||||||
|
result
|
||||||
|
if isinstance(result, list)
|
||||||
|
else result.get("notifications", [])
|
||||||
|
if isinstance(result, dict)
|
||||||
|
else []
|
||||||
|
)
|
||||||
|
if notifications_data:
|
||||||
|
logger.debug(f"获取到 {len(notifications_data)} 条新通知")
|
||||||
|
else:
|
||||||
|
logger.debug(f"API 请求成功: {endpoint}")
|
||||||
|
return result
|
||||||
|
except json.JSONDecodeError as e:
|
||||||
|
logger.error(f"响应不是有效的 JSON 格式: {e}")
|
||||||
|
raise APIConnectionError("Invalid JSON response") from e
|
||||||
|
else:
|
||||||
|
try:
|
||||||
|
error_text = await response.text()
|
||||||
|
logger.error(
|
||||||
|
f"API 请求失败: {endpoint} - 状态码: {response.status}, 响应: {error_text}"
|
||||||
|
)
|
||||||
|
except Exception:
|
||||||
|
logger.error(f"API 请求失败: {endpoint} - 状态码: {response.status}")
|
||||||
|
|
||||||
|
self._handle_response_status(response.status, endpoint)
|
||||||
|
raise APIConnectionError(f"Request failed for {endpoint}")
|
||||||
|
|
||||||
|
@retry_async(
|
||||||
|
max_retries=API_MAX_RETRIES,
|
||||||
|
retryable_exceptions=(APIConnectionError, APIRateLimitError),
|
||||||
|
)
|
||||||
|
async def _make_request(
|
||||||
|
self, endpoint: str, data: Optional[Dict[str, Any]] = None
|
||||||
|
) -> Any:
|
||||||
|
url = f"{self.instance_url}/api/{endpoint}"
|
||||||
|
payload = {"i": self.access_token}
|
||||||
|
if data:
|
||||||
|
payload.update(data)
|
||||||
|
|
||||||
|
try:
|
||||||
|
async with self.session.post(url, json=payload) as response:
|
||||||
|
return await self._process_response(response, endpoint)
|
||||||
|
except aiohttp.ClientError as e:
|
||||||
|
logger.error(f"HTTP 请求错误: {e}")
|
||||||
|
raise APIConnectionError(f"HTTP request failed: {e}") from e
|
||||||
|
|
||||||
|
async def create_note(
|
||||||
|
self,
|
||||||
|
text: str,
|
||||||
|
visibility: str = "public",
|
||||||
|
reply_id: Optional[str] = None,
|
||||||
|
visible_user_ids: Optional[List[str]] = None,
|
||||||
|
local_only: bool = False,
|
||||||
|
) -> Dict[str, Any]:
|
||||||
|
"""创建新贴文"""
|
||||||
|
data: Dict[str, Any] = {
|
||||||
|
"text": text,
|
||||||
|
"visibility": visibility,
|
||||||
|
"localOnly": local_only,
|
||||||
|
}
|
||||||
|
if reply_id:
|
||||||
|
data["replyId"] = reply_id
|
||||||
|
if visible_user_ids and visibility == "specified":
|
||||||
|
data["visibleUserIds"] = visible_user_ids
|
||||||
|
|
||||||
|
result = await self._make_request("notes/create", data)
|
||||||
|
note_id = result.get("createdNote", {}).get("id", "unknown")
|
||||||
|
logger.debug(f"发帖成功,note_id: {note_id}")
|
||||||
|
return result
|
||||||
|
|
||||||
|
async def get_current_user(self) -> Dict[str, Any]:
|
||||||
|
"""获取当前用户信息"""
|
||||||
|
return await self._make_request("i", {})
|
||||||
|
|
||||||
|
async def send_message(self, user_id: str, text: str) -> Dict[str, Any]:
|
||||||
|
"""发送聊天消息"""
|
||||||
|
result = await self._make_request(
|
||||||
|
"chat/messages/create-to-user", {"toUserId": user_id, "text": text}
|
||||||
|
)
|
||||||
|
message_id = result.get("id", "unknown")
|
||||||
|
logger.debug(f"聊天发送成功,message_id: {message_id}")
|
||||||
|
return result
|
||||||
|
|
||||||
|
async def send_room_message(self, room_id: str, text: str) -> Dict[str, Any]:
|
||||||
|
"""发送房间消息"""
|
||||||
|
result = await self._make_request(
|
||||||
|
"chat/messages/create-to-room", {"toRoomId": room_id, "text": text}
|
||||||
|
)
|
||||||
|
message_id = result.get("id", "unknown")
|
||||||
|
logger.debug(f"房间消息发送成功,message_id: {message_id}")
|
||||||
|
return result
|
||||||
|
|
||||||
|
async def get_messages(
|
||||||
|
self, user_id: str, limit: int = 10, since_id: Optional[str] = None
|
||||||
|
) -> List[Dict[str, Any]]:
|
||||||
|
"""获取聊天消息历史"""
|
||||||
|
data: Dict[str, Any] = {"userId": user_id, "limit": limit}
|
||||||
|
if since_id:
|
||||||
|
data["sinceId"] = since_id
|
||||||
|
|
||||||
|
result = await self._make_request("chat/messages/user-timeline", data)
|
||||||
|
if isinstance(result, list):
|
||||||
|
return result
|
||||||
|
else:
|
||||||
|
logger.warning(f"获取聊天消息响应格式异常: {type(result)}")
|
||||||
|
return []
|
||||||
|
|
||||||
|
async def get_mentions(
|
||||||
|
self, limit: int = 10, since_id: Optional[str] = None
|
||||||
|
) -> List[Dict[str, Any]]:
|
||||||
|
"""获取提及通知"""
|
||||||
|
data: Dict[str, Any] = {"limit": limit}
|
||||||
|
if since_id:
|
||||||
|
data["sinceId"] = since_id
|
||||||
|
data["includeTypes"] = ["mention", "reply", "quote"]
|
||||||
|
|
||||||
|
result = await self._make_request("i/notifications", data)
|
||||||
|
if isinstance(result, list):
|
||||||
|
return result
|
||||||
|
elif isinstance(result, dict) and "notifications" in result:
|
||||||
|
return result["notifications"]
|
||||||
|
else:
|
||||||
|
logger.warning(f"获取提及通知响应格式异常: {type(result)}")
|
||||||
|
return []
|
||||||
123
astrbot/core/platform/sources/misskey/misskey_event.py
Normal file
123
astrbot/core/platform/sources/misskey/misskey_event.py
Normal file
@@ -0,0 +1,123 @@
|
|||||||
|
import asyncio
|
||||||
|
import re
|
||||||
|
from typing import AsyncGenerator
|
||||||
|
from astrbot.api import logger
|
||||||
|
from astrbot.api.event import AstrMessageEvent, MessageChain
|
||||||
|
from astrbot.api.platform import PlatformMetadata, AstrBotMessage
|
||||||
|
from astrbot.api.message_components import Plain
|
||||||
|
|
||||||
|
from .misskey_utils import (
|
||||||
|
serialize_message_chain,
|
||||||
|
resolve_visibility_from_raw_message,
|
||||||
|
is_valid_user_session_id,
|
||||||
|
is_valid_room_session_id,
|
||||||
|
add_at_mention_if_needed,
|
||||||
|
extract_user_id_from_session_id,
|
||||||
|
extract_room_id_from_session_id,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
class MisskeyPlatformEvent(AstrMessageEvent):
|
||||||
|
def __init__(
|
||||||
|
self,
|
||||||
|
message_str: str,
|
||||||
|
message_obj: AstrBotMessage,
|
||||||
|
platform_meta: PlatformMetadata,
|
||||||
|
session_id: str,
|
||||||
|
client,
|
||||||
|
):
|
||||||
|
super().__init__(message_str, message_obj, platform_meta, session_id)
|
||||||
|
self.client = client
|
||||||
|
|
||||||
|
def _is_system_command(self, message_str: str) -> bool:
|
||||||
|
"""检测是否为系统指令"""
|
||||||
|
if not message_str or not message_str.strip():
|
||||||
|
return False
|
||||||
|
|
||||||
|
system_prefixes = ["/", "!", "#", ".", "^"]
|
||||||
|
message_trimmed = message_str.strip()
|
||||||
|
|
||||||
|
return any(message_trimmed.startswith(prefix) for prefix in system_prefixes)
|
||||||
|
|
||||||
|
async def send(self, message: MessageChain):
|
||||||
|
content, has_at = serialize_message_chain(message.chain)
|
||||||
|
|
||||||
|
if not content:
|
||||||
|
logger.debug("[MisskeyEvent] 内容为空,跳过发送")
|
||||||
|
return
|
||||||
|
|
||||||
|
try:
|
||||||
|
original_message_id = getattr(self.message_obj, "message_id", None)
|
||||||
|
raw_message = getattr(self.message_obj, "raw_message", {})
|
||||||
|
|
||||||
|
if raw_message and not has_at:
|
||||||
|
user_data = raw_message.get("user", {})
|
||||||
|
user_info = {
|
||||||
|
"username": user_data.get("username", ""),
|
||||||
|
"nickname": user_data.get("name", user_data.get("username", "")),
|
||||||
|
}
|
||||||
|
content = add_at_mention_if_needed(content, user_info, has_at)
|
||||||
|
|
||||||
|
# 根据会话类型选择发送方式
|
||||||
|
if hasattr(self.client, "send_message") and is_valid_user_session_id(
|
||||||
|
self.session_id
|
||||||
|
):
|
||||||
|
user_id = extract_user_id_from_session_id(self.session_id)
|
||||||
|
await self.client.send_message(user_id, content)
|
||||||
|
elif hasattr(self.client, "send_room_message") and is_valid_room_session_id(
|
||||||
|
self.session_id
|
||||||
|
):
|
||||||
|
room_id = extract_room_id_from_session_id(self.session_id)
|
||||||
|
await self.client.send_room_message(room_id, content)
|
||||||
|
elif original_message_id and hasattr(self.client, "create_note"):
|
||||||
|
visibility, visible_user_ids = resolve_visibility_from_raw_message(
|
||||||
|
raw_message
|
||||||
|
)
|
||||||
|
await self.client.create_note(
|
||||||
|
content,
|
||||||
|
reply_id=original_message_id,
|
||||||
|
visibility=visibility,
|
||||||
|
visible_user_ids=visible_user_ids,
|
||||||
|
)
|
||||||
|
elif hasattr(self.client, "create_note"):
|
||||||
|
logger.debug("[MisskeyEvent] 创建新帖子")
|
||||||
|
await self.client.create_note(content)
|
||||||
|
|
||||||
|
await super().send(message)
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"[MisskeyEvent] 发送失败: {e}")
|
||||||
|
|
||||||
|
async def send_streaming(
|
||||||
|
self, generator: AsyncGenerator[MessageChain, None], use_fallback: bool = False
|
||||||
|
):
|
||||||
|
if not use_fallback:
|
||||||
|
buffer = None
|
||||||
|
async for chain in generator:
|
||||||
|
if not buffer:
|
||||||
|
buffer = chain
|
||||||
|
else:
|
||||||
|
buffer.chain.extend(chain.chain)
|
||||||
|
if not buffer:
|
||||||
|
return
|
||||||
|
buffer.squash_plain()
|
||||||
|
await self.send(buffer)
|
||||||
|
return await super().send_streaming(generator, use_fallback)
|
||||||
|
|
||||||
|
buffer = ""
|
||||||
|
pattern = re.compile(r"[^。?!~…]+[。?!~…]+")
|
||||||
|
|
||||||
|
async for chain in generator:
|
||||||
|
if isinstance(chain, MessageChain):
|
||||||
|
for comp in chain.chain:
|
||||||
|
if isinstance(comp, Plain):
|
||||||
|
buffer += comp.text
|
||||||
|
if any(p in buffer for p in "。?!~…"):
|
||||||
|
buffer = await self.process_buffer(buffer, pattern)
|
||||||
|
else:
|
||||||
|
await self.send(MessageChain(chain=[comp]))
|
||||||
|
await asyncio.sleep(1.5) # 限速
|
||||||
|
|
||||||
|
if buffer.strip():
|
||||||
|
await self.send(MessageChain([Plain(buffer)]))
|
||||||
|
return await super().send_streaming(generator, use_fallback)
|
||||||
327
astrbot/core/platform/sources/misskey/misskey_utils.py
Normal file
327
astrbot/core/platform/sources/misskey/misskey_utils.py
Normal file
@@ -0,0 +1,327 @@
|
|||||||
|
"""Misskey 平台适配器通用工具函数"""
|
||||||
|
|
||||||
|
from typing import Dict, Any, List, Tuple, Optional, Union
|
||||||
|
import astrbot.api.message_components as Comp
|
||||||
|
from astrbot.api.platform import AstrBotMessage, MessageMember, MessageType
|
||||||
|
|
||||||
|
|
||||||
|
def serialize_message_chain(chain: List[Any]) -> Tuple[str, bool]:
|
||||||
|
"""将消息链序列化为文本字符串"""
|
||||||
|
text_parts = []
|
||||||
|
has_at = False
|
||||||
|
|
||||||
|
def process_component(component):
|
||||||
|
nonlocal has_at
|
||||||
|
if isinstance(component, Comp.Plain):
|
||||||
|
return component.text
|
||||||
|
elif isinstance(component, Comp.File):
|
||||||
|
file_name = getattr(component, "name", "文件")
|
||||||
|
return f"[文件: {file_name}]"
|
||||||
|
elif isinstance(component, Comp.At):
|
||||||
|
has_at = True
|
||||||
|
return f"@{component.qq}"
|
||||||
|
elif hasattr(component, "text"):
|
||||||
|
text = getattr(component, "text", "")
|
||||||
|
if "@" in text:
|
||||||
|
has_at = True
|
||||||
|
return text
|
||||||
|
else:
|
||||||
|
return str(component)
|
||||||
|
|
||||||
|
for component in chain:
|
||||||
|
if isinstance(component, Comp.Node) and component.content:
|
||||||
|
for node_comp in component.content:
|
||||||
|
result = process_component(node_comp)
|
||||||
|
if result:
|
||||||
|
text_parts.append(result)
|
||||||
|
else:
|
||||||
|
result = process_component(component)
|
||||||
|
if result:
|
||||||
|
text_parts.append(result)
|
||||||
|
|
||||||
|
return "".join(text_parts), has_at
|
||||||
|
|
||||||
|
|
||||||
|
def resolve_message_visibility(
|
||||||
|
user_id: Optional[str],
|
||||||
|
user_cache: Dict[str, Any],
|
||||||
|
self_id: Optional[str],
|
||||||
|
default_visibility: str = "public",
|
||||||
|
) -> Tuple[str, Optional[List[str]]]:
|
||||||
|
"""解析 Misskey 消息的可见性设置"""
|
||||||
|
visibility = default_visibility
|
||||||
|
visible_user_ids = None
|
||||||
|
|
||||||
|
if user_id and user_cache:
|
||||||
|
user_info = user_cache.get(user_id)
|
||||||
|
if user_info:
|
||||||
|
original_visibility = user_info.get("visibility", default_visibility)
|
||||||
|
if original_visibility == "specified":
|
||||||
|
visibility = "specified"
|
||||||
|
original_visible_users = user_info.get("visible_user_ids", [])
|
||||||
|
users_to_include = [user_id]
|
||||||
|
if self_id:
|
||||||
|
users_to_include.append(self_id)
|
||||||
|
visible_user_ids = list(set(original_visible_users + users_to_include))
|
||||||
|
visible_user_ids = [uid for uid in visible_user_ids if uid]
|
||||||
|
else:
|
||||||
|
visibility = original_visibility
|
||||||
|
|
||||||
|
return visibility, visible_user_ids
|
||||||
|
|
||||||
|
|
||||||
|
def resolve_visibility_from_raw_message(
|
||||||
|
raw_message: Dict[str, Any], self_id: Optional[str] = None
|
||||||
|
) -> Tuple[str, Optional[List[str]]]:
|
||||||
|
"""从原始消息数据中解析可见性设置"""
|
||||||
|
visibility = "public"
|
||||||
|
visible_user_ids = None
|
||||||
|
|
||||||
|
if not raw_message:
|
||||||
|
return visibility, visible_user_ids
|
||||||
|
|
||||||
|
original_visibility = raw_message.get("visibility", "public")
|
||||||
|
if original_visibility == "specified":
|
||||||
|
visibility = "specified"
|
||||||
|
original_visible_users = raw_message.get("visibleUserIds", [])
|
||||||
|
sender_id = raw_message.get("userId", "")
|
||||||
|
|
||||||
|
users_to_include = []
|
||||||
|
if sender_id:
|
||||||
|
users_to_include.append(sender_id)
|
||||||
|
if self_id:
|
||||||
|
users_to_include.append(self_id)
|
||||||
|
|
||||||
|
visible_user_ids = list(set(original_visible_users + users_to_include))
|
||||||
|
visible_user_ids = [uid for uid in visible_user_ids if uid]
|
||||||
|
else:
|
||||||
|
visibility = original_visibility
|
||||||
|
|
||||||
|
return visibility, visible_user_ids
|
||||||
|
|
||||||
|
|
||||||
|
def is_valid_user_session_id(session_id: Union[str, Any]) -> bool:
|
||||||
|
"""检查 session_id 是否是有效的聊天用户 session_id (仅限chat%前缀)"""
|
||||||
|
if not isinstance(session_id, str) or "%" not in session_id:
|
||||||
|
return False
|
||||||
|
|
||||||
|
parts = session_id.split("%")
|
||||||
|
return (
|
||||||
|
len(parts) == 2
|
||||||
|
and parts[0] == "chat"
|
||||||
|
and bool(parts[1])
|
||||||
|
and parts[1] != "unknown"
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def is_valid_room_session_id(session_id: Union[str, Any]) -> bool:
|
||||||
|
"""检查 session_id 是否是有效的房间 session_id (仅限room%前缀)"""
|
||||||
|
if not isinstance(session_id, str) or "%" not in session_id:
|
||||||
|
return False
|
||||||
|
|
||||||
|
parts = session_id.split("%")
|
||||||
|
return (
|
||||||
|
len(parts) == 2
|
||||||
|
and parts[0] == "room"
|
||||||
|
and bool(parts[1])
|
||||||
|
and parts[1] != "unknown"
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def extract_user_id_from_session_id(session_id: str) -> str:
|
||||||
|
"""从 session_id 中提取用户 ID"""
|
||||||
|
if "%" in session_id:
|
||||||
|
parts = session_id.split("%")
|
||||||
|
if len(parts) >= 2:
|
||||||
|
return parts[1]
|
||||||
|
return session_id
|
||||||
|
|
||||||
|
|
||||||
|
def extract_room_id_from_session_id(session_id: str) -> str:
|
||||||
|
"""从 session_id 中提取房间 ID"""
|
||||||
|
if "%" in session_id:
|
||||||
|
parts = session_id.split("%")
|
||||||
|
if len(parts) >= 2 and parts[0] == "room":
|
||||||
|
return parts[1]
|
||||||
|
return session_id
|
||||||
|
|
||||||
|
|
||||||
|
def add_at_mention_if_needed(
|
||||||
|
text: str, user_info: Optional[Dict[str, Any]], has_at: bool = False
|
||||||
|
) -> str:
|
||||||
|
"""如果需要且没有@用户,则添加@用户"""
|
||||||
|
if has_at or not user_info:
|
||||||
|
return text
|
||||||
|
|
||||||
|
username = user_info.get("username")
|
||||||
|
nickname = user_info.get("nickname")
|
||||||
|
|
||||||
|
if username:
|
||||||
|
mention = f"@{username}"
|
||||||
|
if not text.startswith(mention):
|
||||||
|
text = f"{mention}\n{text}".strip()
|
||||||
|
elif nickname:
|
||||||
|
mention = f"@{nickname}"
|
||||||
|
if not text.startswith(mention):
|
||||||
|
text = f"{mention}\n{text}".strip()
|
||||||
|
|
||||||
|
return text
|
||||||
|
|
||||||
|
|
||||||
|
def create_file_component(file_info: Dict[str, Any]) -> Tuple[Any, str]:
|
||||||
|
"""创建文件组件和描述文本"""
|
||||||
|
file_url = file_info.get("url", "")
|
||||||
|
file_name = file_info.get("name", "未知文件")
|
||||||
|
file_type = file_info.get("type", "")
|
||||||
|
|
||||||
|
if file_type.startswith("image/"):
|
||||||
|
return Comp.Image(url=file_url, file=file_name), f"图片[{file_name}]"
|
||||||
|
elif file_type.startswith("audio/"):
|
||||||
|
return Comp.Record(url=file_url, file=file_name), f"音频[{file_name}]"
|
||||||
|
elif file_type.startswith("video/"):
|
||||||
|
return Comp.Video(url=file_url, file=file_name), f"视频[{file_name}]"
|
||||||
|
else:
|
||||||
|
return Comp.File(name=file_name, url=file_url), f"文件[{file_name}]"
|
||||||
|
|
||||||
|
|
||||||
|
def process_files(
|
||||||
|
message: AstrBotMessage, files: list, include_text_parts: bool = True
|
||||||
|
) -> list:
|
||||||
|
"""处理文件列表,添加到消息组件中并返回文本描述"""
|
||||||
|
file_parts = []
|
||||||
|
for file_info in files:
|
||||||
|
component, part_text = create_file_component(file_info)
|
||||||
|
message.message.append(component)
|
||||||
|
if include_text_parts:
|
||||||
|
file_parts.append(part_text)
|
||||||
|
return file_parts
|
||||||
|
|
||||||
|
|
||||||
|
def extract_sender_info(
|
||||||
|
raw_data: Dict[str, Any], is_chat: bool = False
|
||||||
|
) -> Dict[str, Any]:
|
||||||
|
"""提取发送者信息"""
|
||||||
|
if is_chat:
|
||||||
|
sender = raw_data.get("fromUser", {})
|
||||||
|
sender_id = str(sender.get("id", "") or raw_data.get("fromUserId", ""))
|
||||||
|
else:
|
||||||
|
sender = raw_data.get("user", {})
|
||||||
|
sender_id = str(sender.get("id", ""))
|
||||||
|
|
||||||
|
return {
|
||||||
|
"sender": sender,
|
||||||
|
"sender_id": sender_id,
|
||||||
|
"nickname": sender.get("name", sender.get("username", "")),
|
||||||
|
"username": sender.get("username", ""),
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
def create_base_message(
|
||||||
|
raw_data: Dict[str, Any],
|
||||||
|
sender_info: Dict[str, Any],
|
||||||
|
client_self_id: str,
|
||||||
|
is_chat: bool = False,
|
||||||
|
room_id: Optional[str] = None,
|
||||||
|
unique_session: bool = False,
|
||||||
|
) -> AstrBotMessage:
|
||||||
|
"""创建基础消息对象"""
|
||||||
|
message = AstrBotMessage()
|
||||||
|
message.raw_message = raw_data
|
||||||
|
message.message = []
|
||||||
|
|
||||||
|
message.sender = MessageMember(
|
||||||
|
user_id=sender_info["sender_id"],
|
||||||
|
nickname=sender_info["nickname"],
|
||||||
|
)
|
||||||
|
|
||||||
|
if room_id:
|
||||||
|
session_prefix = "room"
|
||||||
|
session_id = f"{session_prefix}%{room_id}"
|
||||||
|
if unique_session:
|
||||||
|
session_id += f"_{sender_info['sender_id']}"
|
||||||
|
message.type = MessageType.GROUP_MESSAGE
|
||||||
|
message.group_id = room_id
|
||||||
|
elif is_chat:
|
||||||
|
session_prefix = "chat"
|
||||||
|
session_id = f"{session_prefix}%{sender_info['sender_id']}"
|
||||||
|
message.type = MessageType.FRIEND_MESSAGE
|
||||||
|
else:
|
||||||
|
session_prefix = "note"
|
||||||
|
session_id = f"{session_prefix}%{sender_info['sender_id']}"
|
||||||
|
message.type = MessageType.FRIEND_MESSAGE
|
||||||
|
|
||||||
|
message.session_id = (
|
||||||
|
session_id if sender_info["sender_id"] else f"{session_prefix}%unknown"
|
||||||
|
)
|
||||||
|
message.message_id = str(raw_data.get("id", ""))
|
||||||
|
message.self_id = client_self_id
|
||||||
|
|
||||||
|
return message
|
||||||
|
|
||||||
|
|
||||||
|
def process_at_mention(
|
||||||
|
message: AstrBotMessage, raw_text: str, bot_username: str, client_self_id: str
|
||||||
|
) -> Tuple[List[str], str]:
|
||||||
|
"""处理@提及逻辑,返回消息部分列表和处理后的文本"""
|
||||||
|
message_parts = []
|
||||||
|
|
||||||
|
if not raw_text:
|
||||||
|
return message_parts, ""
|
||||||
|
|
||||||
|
if bot_username and raw_text.startswith(f"@{bot_username}"):
|
||||||
|
at_mention = f"@{bot_username}"
|
||||||
|
message.message.append(Comp.At(qq=client_self_id))
|
||||||
|
remaining_text = raw_text[len(at_mention) :].strip()
|
||||||
|
if remaining_text:
|
||||||
|
message.message.append(Comp.Plain(remaining_text))
|
||||||
|
message_parts.append(remaining_text)
|
||||||
|
return message_parts, remaining_text
|
||||||
|
else:
|
||||||
|
message.message.append(Comp.Plain(raw_text))
|
||||||
|
message_parts.append(raw_text)
|
||||||
|
return message_parts, raw_text
|
||||||
|
|
||||||
|
|
||||||
|
def cache_user_info(
|
||||||
|
user_cache: Dict[str, Any],
|
||||||
|
sender_info: Dict[str, Any],
|
||||||
|
raw_data: Dict[str, Any],
|
||||||
|
client_self_id: str,
|
||||||
|
is_chat: bool = False,
|
||||||
|
):
|
||||||
|
"""缓存用户信息"""
|
||||||
|
if is_chat:
|
||||||
|
user_cache_data = {
|
||||||
|
"username": sender_info["username"],
|
||||||
|
"nickname": sender_info["nickname"],
|
||||||
|
"visibility": "specified",
|
||||||
|
"visible_user_ids": [client_self_id, sender_info["sender_id"]],
|
||||||
|
}
|
||||||
|
else:
|
||||||
|
user_cache_data = {
|
||||||
|
"username": sender_info["username"],
|
||||||
|
"nickname": sender_info["nickname"],
|
||||||
|
"visibility": raw_data.get("visibility", "public"),
|
||||||
|
"visible_user_ids": raw_data.get("visibleUserIds", []),
|
||||||
|
}
|
||||||
|
|
||||||
|
user_cache[sender_info["sender_id"]] = user_cache_data
|
||||||
|
|
||||||
|
|
||||||
|
def cache_room_info(
|
||||||
|
user_cache: Dict[str, Any], raw_data: Dict[str, Any], client_self_id: str
|
||||||
|
):
|
||||||
|
"""缓存房间信息"""
|
||||||
|
room_data = raw_data.get("toRoom")
|
||||||
|
room_id = raw_data.get("toRoomId")
|
||||||
|
|
||||||
|
if room_data and room_id:
|
||||||
|
room_cache_key = f"room:{room_id}"
|
||||||
|
user_cache[room_cache_key] = {
|
||||||
|
"room_id": room_id,
|
||||||
|
"room_name": room_data.get("name", ""),
|
||||||
|
"room_description": room_data.get("description", ""),
|
||||||
|
"owner_id": room_data.get("ownerId", ""),
|
||||||
|
"visibility": "specified",
|
||||||
|
"visible_user_ids": [client_self_id],
|
||||||
|
}
|
||||||
@@ -94,10 +94,15 @@ class QQOfficialMessageEvent(AstrMessageEvent):
|
|||||||
plain_text,
|
plain_text,
|
||||||
image_base64,
|
image_base64,
|
||||||
image_path,
|
image_path,
|
||||||
record_file_path
|
record_file_path,
|
||||||
) = await QQOfficialMessageEvent._parse_to_qqofficial(self.send_buffer)
|
) = await QQOfficialMessageEvent._parse_to_qqofficial(self.send_buffer)
|
||||||
|
|
||||||
if not plain_text and not image_base64 and not image_path and not record_file_path:
|
if (
|
||||||
|
not plain_text
|
||||||
|
and not image_base64
|
||||||
|
and not image_path
|
||||||
|
and not record_file_path
|
||||||
|
):
|
||||||
return
|
return
|
||||||
|
|
||||||
payload = {
|
payload = {
|
||||||
@@ -190,28 +195,21 @@ class QQOfficialMessageEvent(AstrMessageEvent):
|
|||||||
return await self.bot.api._http.request(route, json=payload)
|
return await self.bot.api._http.request(route, json=payload)
|
||||||
|
|
||||||
async def upload_group_and_c2c_record(
|
async def upload_group_and_c2c_record(
|
||||||
self,
|
self, file_source: str, file_type: int, srv_send_msg: bool = False, **kwargs
|
||||||
file_source: str,
|
|
||||||
file_type: int,
|
|
||||||
srv_send_msg: bool = False,
|
|
||||||
**kwargs
|
|
||||||
) -> Optional[Media]:
|
) -> Optional[Media]:
|
||||||
"""
|
"""
|
||||||
上传媒体文件
|
上传媒体文件
|
||||||
"""
|
"""
|
||||||
# 构建基础payload
|
# 构建基础payload
|
||||||
payload = {
|
payload = {"file_type": file_type, "srv_send_msg": srv_send_msg}
|
||||||
"file_type": file_type,
|
|
||||||
"srv_send_msg": srv_send_msg
|
|
||||||
}
|
|
||||||
|
|
||||||
# 处理文件数据
|
# 处理文件数据
|
||||||
if os.path.exists(file_source):
|
if os.path.exists(file_source):
|
||||||
# 读取本地文件
|
# 读取本地文件
|
||||||
async with aiofiles.open(file_source, 'rb') as f:
|
async with aiofiles.open(file_source, "rb") as f:
|
||||||
file_content = await f.read()
|
file_content = await f.read()
|
||||||
# use base64 encode
|
# use base64 encode
|
||||||
payload["file_data"] = base64.b64encode(file_content).decode('utf-8')
|
payload["file_data"] = base64.b64encode(file_content).decode("utf-8")
|
||||||
else:
|
else:
|
||||||
# 使用URL
|
# 使用URL
|
||||||
payload["url"] = file_source
|
payload["url"] = file_source
|
||||||
@@ -222,7 +220,11 @@ class QQOfficialMessageEvent(AstrMessageEvent):
|
|||||||
route = Route("POST", "/v2/users/{openid}/files", openid=kwargs["openid"])
|
route = Route("POST", "/v2/users/{openid}/files", openid=kwargs["openid"])
|
||||||
elif "group_openid" in kwargs:
|
elif "group_openid" in kwargs:
|
||||||
payload["group_openid"] = kwargs["group_openid"]
|
payload["group_openid"] = kwargs["group_openid"]
|
||||||
route = Route("POST", "/v2/groups/{group_openid}/files", group_openid=kwargs["group_openid"])
|
route = Route(
|
||||||
|
"POST",
|
||||||
|
"/v2/groups/{group_openid}/files",
|
||||||
|
group_openid=kwargs["group_openid"],
|
||||||
|
)
|
||||||
else:
|
else:
|
||||||
return None
|
return None
|
||||||
|
|
||||||
@@ -235,7 +237,7 @@ class QQOfficialMessageEvent(AstrMessageEvent):
|
|||||||
file_uuid=result.get("file_uuid"),
|
file_uuid=result.get("file_uuid"),
|
||||||
file_info=result.get("file_info"),
|
file_info=result.get("file_info"),
|
||||||
ttl=result.get("ttl", 0),
|
ttl=result.get("ttl", 0),
|
||||||
file_id=result.get("id", "")
|
file_id=result.get("id", ""),
|
||||||
)
|
)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.error(f"上传请求错误: {e}")
|
logger.error(f"上传请求错误: {e}")
|
||||||
@@ -288,9 +290,13 @@ class QQOfficialMessageEvent(AstrMessageEvent):
|
|||||||
if i.file:
|
if i.file:
|
||||||
record_wav_path = await i.convert_to_file_path() # wav 路径
|
record_wav_path = await i.convert_to_file_path() # wav 路径
|
||||||
temp_dir = os.path.join(get_astrbot_data_path(), "temp")
|
temp_dir = os.path.join(get_astrbot_data_path(), "temp")
|
||||||
record_tecent_silk_path = os.path.join(temp_dir, f"{uuid.uuid4()}.silk")
|
record_tecent_silk_path = os.path.join(
|
||||||
|
temp_dir, f"{uuid.uuid4()}.silk"
|
||||||
|
)
|
||||||
try:
|
try:
|
||||||
duration = await wav_to_tencent_silk(record_wav_path, record_tecent_silk_path)
|
duration = await wav_to_tencent_silk(
|
||||||
|
record_wav_path, record_tecent_silk_path
|
||||||
|
)
|
||||||
if duration > 0:
|
if duration > 0:
|
||||||
record_file_path = record_tecent_silk_path
|
record_file_path = record_tecent_silk_path
|
||||||
else:
|
else:
|
||||||
|
|||||||
@@ -17,7 +17,14 @@ from astrbot.api.platform import (
|
|||||||
register_platform_adapter,
|
register_platform_adapter,
|
||||||
)
|
)
|
||||||
from astrbot.core.platform.astr_message_event import MessageSession
|
from astrbot.core.platform.astr_message_event import MessageSession
|
||||||
from astrbot.api.message_components import Plain, Image, At, File, Record
|
from astrbot.api.message_components import (
|
||||||
|
Plain,
|
||||||
|
Image,
|
||||||
|
At,
|
||||||
|
File,
|
||||||
|
Record,
|
||||||
|
Reply,
|
||||||
|
)
|
||||||
from xml.etree import ElementTree as ET
|
from xml.etree import ElementTree as ET
|
||||||
|
|
||||||
|
|
||||||
@@ -38,12 +45,18 @@ class SatoriPlatformAdapter(Platform):
|
|||||||
)
|
)
|
||||||
self.token = self.config.get("satori_token", "")
|
self.token = self.config.get("satori_token", "")
|
||||||
self.endpoint = self.config.get(
|
self.endpoint = self.config.get(
|
||||||
"satori_endpoint", "ws://127.0.0.1:5140/satori/v1/events"
|
"satori_endpoint", "ws://localhost:5140/satori/v1/events"
|
||||||
)
|
)
|
||||||
self.auto_reconnect = self.config.get("satori_auto_reconnect", True)
|
self.auto_reconnect = self.config.get("satori_auto_reconnect", True)
|
||||||
self.heartbeat_interval = self.config.get("satori_heartbeat_interval", 10)
|
self.heartbeat_interval = self.config.get("satori_heartbeat_interval", 10)
|
||||||
self.reconnect_delay = self.config.get("satori_reconnect_delay", 5)
|
self.reconnect_delay = self.config.get("satori_reconnect_delay", 5)
|
||||||
|
|
||||||
|
self.metadata = PlatformMetadata(
|
||||||
|
name="satori",
|
||||||
|
description="Satori 通用协议适配器",
|
||||||
|
id=self.config["id"],
|
||||||
|
)
|
||||||
|
|
||||||
self.ws: Optional[ClientConnection] = None
|
self.ws: Optional[ClientConnection] = None
|
||||||
self.session: Optional[ClientSession] = None
|
self.session: Optional[ClientSession] = None
|
||||||
self.sequence = 0
|
self.sequence = 0
|
||||||
@@ -63,7 +76,7 @@ class SatoriPlatformAdapter(Platform):
|
|||||||
await super().send_by_session(session, message_chain)
|
await super().send_by_session(session, message_chain)
|
||||||
|
|
||||||
def meta(self) -> PlatformMetadata:
|
def meta(self) -> PlatformMetadata:
|
||||||
return PlatformMetadata(name="satori", description="Satori 通用协议适配器")
|
return self.metadata
|
||||||
|
|
||||||
def _is_websocket_closed(self, ws) -> bool:
|
def _is_websocket_closed(self, ws) -> bool:
|
||||||
"""检查WebSocket连接是否已关闭"""
|
"""检查WebSocket连接是否已关闭"""
|
||||||
@@ -312,12 +325,52 @@ class SatoriPlatformAdapter(Platform):
|
|||||||
|
|
||||||
abm.self_id = login.get("user", {}).get("id", "")
|
abm.self_id = login.get("user", {}).get("id", "")
|
||||||
|
|
||||||
content = message.get("content", "")
|
# 消息链
|
||||||
abm.message = await self.parse_satori_elements(content)
|
abm.message = []
|
||||||
|
|
||||||
|
content = message.get("content", "")
|
||||||
|
|
||||||
|
quote = message.get("quote")
|
||||||
|
content_for_parsing = content # 副本
|
||||||
|
|
||||||
|
# 提取<quote>标签
|
||||||
|
if "<quote" in content:
|
||||||
|
try:
|
||||||
|
quote_info = await self._extract_quote_element(content)
|
||||||
|
if quote_info:
|
||||||
|
quote = quote_info["quote"]
|
||||||
|
content_for_parsing = quote_info["content_without_quote"]
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"解析<quote>标签时发生错误: {e}, 错误内容: {content}")
|
||||||
|
|
||||||
|
if quote:
|
||||||
|
# 引用消息
|
||||||
|
quote_abm = await self._convert_quote_message(quote)
|
||||||
|
if quote_abm:
|
||||||
|
sender_id = quote_abm.sender.user_id
|
||||||
|
if isinstance(sender_id, str) and sender_id.isdigit():
|
||||||
|
sender_id = int(sender_id)
|
||||||
|
elif not isinstance(sender_id, int):
|
||||||
|
sender_id = 0 # 默认值
|
||||||
|
|
||||||
|
reply_component = Reply(
|
||||||
|
id=quote_abm.message_id,
|
||||||
|
chain=quote_abm.message,
|
||||||
|
sender_id=quote_abm.sender.user_id,
|
||||||
|
sender_nickname=quote_abm.sender.nickname,
|
||||||
|
time=quote_abm.timestamp,
|
||||||
|
message_str=quote_abm.message_str,
|
||||||
|
text=quote_abm.message_str,
|
||||||
|
qq=sender_id,
|
||||||
|
)
|
||||||
|
abm.message.append(reply_component)
|
||||||
|
|
||||||
|
# 解析消息内容
|
||||||
|
content_elements = await self.parse_satori_elements(content_for_parsing)
|
||||||
|
abm.message.extend(content_elements)
|
||||||
|
|
||||||
# parse message_str
|
|
||||||
abm.message_str = ""
|
abm.message_str = ""
|
||||||
for comp in abm.message:
|
for comp in content_elements:
|
||||||
if isinstance(comp, Plain):
|
if isinstance(comp, Plain):
|
||||||
abm.message_str += comp.text
|
abm.message_str += comp.text
|
||||||
|
|
||||||
@@ -333,6 +386,163 @@ class SatoriPlatformAdapter(Platform):
|
|||||||
logger.error(f"转换 Satori 消息失败: {e}")
|
logger.error(f"转换 Satori 消息失败: {e}")
|
||||||
return None
|
return None
|
||||||
|
|
||||||
|
def _extract_namespace_prefixes(self, content: str) -> set:
|
||||||
|
"""提取XML内容中的命名空间前缀"""
|
||||||
|
prefixes = set()
|
||||||
|
|
||||||
|
# 查找所有标签
|
||||||
|
i = 0
|
||||||
|
while i < len(content):
|
||||||
|
# 查找开始标签
|
||||||
|
if content[i] == "<" and i + 1 < len(content) and content[i + 1] != "/":
|
||||||
|
# 找到标签结束位置
|
||||||
|
tag_end = content.find(">", i)
|
||||||
|
if tag_end != -1:
|
||||||
|
# 提取标签内容
|
||||||
|
tag_content = content[i + 1 : tag_end]
|
||||||
|
# 检查是否有命名空间前缀
|
||||||
|
if ":" in tag_content and "xmlns:" not in tag_content:
|
||||||
|
# 分割标签名
|
||||||
|
parts = tag_content.split()
|
||||||
|
if parts:
|
||||||
|
tag_name = parts[0]
|
||||||
|
if ":" in tag_name:
|
||||||
|
prefix = tag_name.split(":")[0]
|
||||||
|
# 确保是有效的命名空间前缀
|
||||||
|
if (
|
||||||
|
prefix.isalnum()
|
||||||
|
or prefix.replace("_", "").isalnum()
|
||||||
|
):
|
||||||
|
prefixes.add(prefix)
|
||||||
|
i = tag_end + 1
|
||||||
|
else:
|
||||||
|
i += 1
|
||||||
|
# 查找结束标签
|
||||||
|
elif content[i] == "<" and i + 1 < len(content) and content[i + 1] == "/":
|
||||||
|
# 找到标签结束位置
|
||||||
|
tag_end = content.find(">", i)
|
||||||
|
if tag_end != -1:
|
||||||
|
# 提取标签内容
|
||||||
|
tag_content = content[i + 2 : tag_end]
|
||||||
|
# 检查是否有命名空间前缀
|
||||||
|
if ":" in tag_content:
|
||||||
|
prefix = tag_content.split(":")[0]
|
||||||
|
# 确保是有效的命名空间前缀
|
||||||
|
if prefix.isalnum() or prefix.replace("_", "").isalnum():
|
||||||
|
prefixes.add(prefix)
|
||||||
|
i = tag_end + 1
|
||||||
|
else:
|
||||||
|
i += 1
|
||||||
|
else:
|
||||||
|
i += 1
|
||||||
|
|
||||||
|
return prefixes
|
||||||
|
|
||||||
|
async def _extract_quote_element(self, content: str) -> Optional[dict]:
|
||||||
|
"""提取<quote>标签信息"""
|
||||||
|
try:
|
||||||
|
# 处理命名空间前缀问题
|
||||||
|
processed_content = content
|
||||||
|
if ":" in content and not content.startswith("<root"):
|
||||||
|
prefixes = self._extract_namespace_prefixes(content)
|
||||||
|
|
||||||
|
# 构建命名空间声明
|
||||||
|
ns_declarations = " ".join(
|
||||||
|
[
|
||||||
|
f'xmlns:{prefix}="http://temp.uri/{prefix}"'
|
||||||
|
for prefix in prefixes
|
||||||
|
]
|
||||||
|
)
|
||||||
|
|
||||||
|
# 包装内容
|
||||||
|
processed_content = f"<root {ns_declarations}>{content}</root>"
|
||||||
|
elif not content.startswith("<root"):
|
||||||
|
processed_content = f"<root>{content}</root>"
|
||||||
|
else:
|
||||||
|
processed_content = content
|
||||||
|
|
||||||
|
root = ET.fromstring(processed_content)
|
||||||
|
|
||||||
|
# 查找<quote>标签
|
||||||
|
quote_element = None
|
||||||
|
for elem in root.iter():
|
||||||
|
tag_name = elem.tag
|
||||||
|
if "}" in tag_name:
|
||||||
|
tag_name = tag_name.split("}")[1]
|
||||||
|
if tag_name.lower() == "quote":
|
||||||
|
quote_element = elem
|
||||||
|
break
|
||||||
|
|
||||||
|
if quote_element is not None:
|
||||||
|
# 提取quote标签的属性
|
||||||
|
quote_id = quote_element.get("id", "")
|
||||||
|
|
||||||
|
# 提取<quote>标签内部的内容
|
||||||
|
inner_content = ""
|
||||||
|
if quote_element.text:
|
||||||
|
inner_content += quote_element.text
|
||||||
|
for child in quote_element:
|
||||||
|
inner_content += ET.tostring(
|
||||||
|
child, encoding="unicode", method="xml"
|
||||||
|
)
|
||||||
|
if child.tail:
|
||||||
|
inner_content += child.tail
|
||||||
|
|
||||||
|
# 构造移除了<quote>标签的内容
|
||||||
|
content_without_quote = content.replace(
|
||||||
|
ET.tostring(quote_element, encoding="unicode", method="xml"), ""
|
||||||
|
)
|
||||||
|
|
||||||
|
return {
|
||||||
|
"quote": {"id": quote_id, "content": inner_content},
|
||||||
|
"content_without_quote": content_without_quote,
|
||||||
|
}
|
||||||
|
|
||||||
|
return None
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"提取<quote>标签时发生错误: {e}")
|
||||||
|
return None
|
||||||
|
|
||||||
|
async def _convert_quote_message(self, quote: dict) -> Optional[AstrBotMessage]:
|
||||||
|
"""转换引用消息"""
|
||||||
|
try:
|
||||||
|
quote_abm = AstrBotMessage()
|
||||||
|
quote_abm.message_id = quote.get("id", "")
|
||||||
|
|
||||||
|
# 解析引用消息的发送者
|
||||||
|
quote_author = quote.get("author", {})
|
||||||
|
if quote_author:
|
||||||
|
quote_abm.sender = MessageMember(
|
||||||
|
user_id=quote_author.get("id", ""),
|
||||||
|
nickname=quote_author.get("nick", quote_author.get("name", "")),
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
# 如果没有作者信息,使用默认值
|
||||||
|
quote_abm.sender = MessageMember(
|
||||||
|
user_id=quote.get("user_id", ""),
|
||||||
|
nickname="内容",
|
||||||
|
)
|
||||||
|
|
||||||
|
# 解析引用消息内容
|
||||||
|
quote_content = quote.get("content", "")
|
||||||
|
quote_abm.message = await self.parse_satori_elements(quote_content)
|
||||||
|
|
||||||
|
quote_abm.message_str = ""
|
||||||
|
for comp in quote_abm.message:
|
||||||
|
if isinstance(comp, Plain):
|
||||||
|
quote_abm.message_str += comp.text
|
||||||
|
|
||||||
|
quote_abm.timestamp = int(quote.get("timestamp", time.time()))
|
||||||
|
|
||||||
|
# 如果没有任何内容,使用默认文本
|
||||||
|
if not quote_abm.message_str.strip():
|
||||||
|
quote_abm.message_str = "[引用消息]"
|
||||||
|
|
||||||
|
return quote_abm
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"转换引用消息失败: {e}")
|
||||||
|
return None
|
||||||
|
|
||||||
async def parse_satori_elements(self, content: str) -> list:
|
async def parse_satori_elements(self, content: str) -> list:
|
||||||
"""解析 Satori 消息元素"""
|
"""解析 Satori 消息元素"""
|
||||||
elements = []
|
elements = []
|
||||||
@@ -341,12 +551,35 @@ class SatoriPlatformAdapter(Platform):
|
|||||||
return elements
|
return elements
|
||||||
|
|
||||||
try:
|
try:
|
||||||
wrapped_content = f"<root>{content}</root>"
|
# 处理命名空间前缀问题
|
||||||
root = ET.fromstring(wrapped_content)
|
processed_content = content
|
||||||
|
if ":" in content and not content.startswith("<root"):
|
||||||
|
prefixes = self._extract_namespace_prefixes(content)
|
||||||
|
|
||||||
|
# 构建命名空间声明
|
||||||
|
ns_declarations = " ".join(
|
||||||
|
[
|
||||||
|
f'xmlns:{prefix}="http://temp.uri/{prefix}"'
|
||||||
|
for prefix in prefixes
|
||||||
|
]
|
||||||
|
)
|
||||||
|
|
||||||
|
# 包装内容
|
||||||
|
processed_content = f"<root {ns_declarations}>{content}</root>"
|
||||||
|
elif not content.startswith("<root"):
|
||||||
|
processed_content = f"<root>{content}</root>"
|
||||||
|
else:
|
||||||
|
processed_content = content
|
||||||
|
|
||||||
|
root = ET.fromstring(processed_content)
|
||||||
await self._parse_xml_node(root, elements)
|
await self._parse_xml_node(root, elements)
|
||||||
except ET.ParseError as e:
|
except ET.ParseError as e:
|
||||||
raise ValueError(f"解析 Satori 元素时发生解析错误: {e}")
|
logger.error(f"解析 Satori 元素时发生解析错误: {e}, 错误内容: {content}")
|
||||||
|
# 如果解析失败,将整个内容当作纯文本
|
||||||
|
if content.strip():
|
||||||
|
elements.append(Plain(text=content))
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
|
logger.error(f"解析 Satori 元素时发生未知错误: {e}")
|
||||||
raise e
|
raise e
|
||||||
|
|
||||||
# 如果没有解析到任何元素,将整个内容当作纯文本
|
# 如果没有解析到任何元素,将整个内容当作纯文本
|
||||||
@@ -361,7 +594,12 @@ class SatoriPlatformAdapter(Platform):
|
|||||||
elements.append(Plain(text=node.text))
|
elements.append(Plain(text=node.text))
|
||||||
|
|
||||||
for child in node:
|
for child in node:
|
||||||
tag_name = child.tag.lower()
|
# 获取标签名,去除命名空间前缀
|
||||||
|
tag_name = child.tag
|
||||||
|
if "}" in tag_name:
|
||||||
|
tag_name = tag_name.split("}")[1]
|
||||||
|
tag_name = tag_name.lower()
|
||||||
|
|
||||||
attrs = child.attrib
|
attrs = child.attrib
|
||||||
|
|
||||||
if tag_name == "at":
|
if tag_name == "at":
|
||||||
@@ -372,31 +610,59 @@ class SatoriPlatformAdapter(Platform):
|
|||||||
src = attrs.get("src", "")
|
src = attrs.get("src", "")
|
||||||
if not src:
|
if not src:
|
||||||
continue
|
continue
|
||||||
if src.startswith("data:image/"):
|
elements.append(Image(file=src))
|
||||||
src = src.split(",")[1]
|
|
||||||
elements.append(Image.fromBase64(src))
|
|
||||||
elif src.startswith("http"):
|
|
||||||
elements.append(Image.fromURL(src))
|
|
||||||
else:
|
|
||||||
logger.error(f"未知的图片 src 格式: {str(src)[:16]}")
|
|
||||||
|
|
||||||
elif tag_name == "file":
|
elif tag_name == "file":
|
||||||
src = attrs.get("src", "")
|
src = attrs.get("src", "")
|
||||||
name = attrs.get("name", "文件")
|
name = attrs.get("name", "文件")
|
||||||
if src:
|
if src:
|
||||||
elements.append(File(file=src, name=name))
|
elements.append(File(name=name, file=src))
|
||||||
|
|
||||||
elif tag_name in ("audio", "record"):
|
elif tag_name in ("audio", "record"):
|
||||||
src = attrs.get("src", "")
|
src = attrs.get("src", "")
|
||||||
if not src:
|
if not src:
|
||||||
continue
|
continue
|
||||||
if src.startswith("data:audio/"):
|
elements.append(Record(file=src))
|
||||||
src = src.split(",")[1]
|
|
||||||
elements.append(Record.fromBase64(src))
|
elif tag_name == "quote":
|
||||||
elif src.startswith("http"):
|
# quote标签已经被特殊处理
|
||||||
elements.append(Record.fromURL(src))
|
pass
|
||||||
|
|
||||||
|
elif tag_name == "face":
|
||||||
|
face_id = attrs.get("id", "")
|
||||||
|
face_name = attrs.get("name", "")
|
||||||
|
face_type = attrs.get("type", "")
|
||||||
|
|
||||||
|
if face_name:
|
||||||
|
elements.append(Plain(text=f"[表情:{face_name}]"))
|
||||||
|
elif face_id and face_type:
|
||||||
|
elements.append(Plain(text=f"[表情ID:{face_id},类型:{face_type}]"))
|
||||||
|
elif face_id:
|
||||||
|
elements.append(Plain(text=f"[表情ID:{face_id}]"))
|
||||||
else:
|
else:
|
||||||
logger.error(f"未知的音频 src 格式: {str(src)[:16]}")
|
elements.append(Plain(text="[表情]"))
|
||||||
|
|
||||||
|
elif tag_name == "ark":
|
||||||
|
# 作为纯文本添加到消息链中
|
||||||
|
data = attrs.get("data", "")
|
||||||
|
if data:
|
||||||
|
import html
|
||||||
|
|
||||||
|
decoded_data = html.unescape(data)
|
||||||
|
elements.append(Plain(text=f"[ARK卡片数据: {decoded_data}]"))
|
||||||
|
else:
|
||||||
|
elements.append(Plain(text="[ARK卡片]"))
|
||||||
|
|
||||||
|
elif tag_name == "json":
|
||||||
|
# JSON标签 视为ARK卡片消息
|
||||||
|
data = attrs.get("data", "")
|
||||||
|
if data:
|
||||||
|
import html
|
||||||
|
|
||||||
|
decoded_data = html.unescape(data)
|
||||||
|
elements.append(Plain(text=f"[ARK卡片数据: {decoded_data}]"))
|
||||||
|
else:
|
||||||
|
elements.append(Plain(text="[JSON卡片]"))
|
||||||
|
|
||||||
else:
|
else:
|
||||||
# 未知标签,递归处理其内容
|
# 未知标签,递归处理其内容
|
||||||
|
|||||||
@@ -17,6 +17,15 @@ class SatoriPlatformEvent(AstrMessageEvent):
|
|||||||
session_id: str,
|
session_id: str,
|
||||||
adapter: "SatoriPlatformAdapter",
|
adapter: "SatoriPlatformAdapter",
|
||||||
):
|
):
|
||||||
|
# 更新平台元数据
|
||||||
|
if adapter and hasattr(adapter, "logins") and adapter.logins:
|
||||||
|
current_login = adapter.logins[0]
|
||||||
|
platform_name = current_login.get("platform", "satori")
|
||||||
|
user = current_login.get("user", {})
|
||||||
|
user_id = user.get("id", "") if user else ""
|
||||||
|
if not platform_meta.id and user_id:
|
||||||
|
platform_meta.id = f"{platform_name}({user_id})"
|
||||||
|
|
||||||
super().__init__(message_str, message_obj, platform_meta, session_id)
|
super().__init__(message_str, message_obj, platform_meta, session_id)
|
||||||
self.adapter = adapter
|
self.adapter = adapter
|
||||||
self.platform = None
|
self.platform = None
|
||||||
|
|||||||
@@ -308,7 +308,9 @@ class SlackAdapter(Platform):
|
|||||||
base64_content = base64.b64encode(content).decode("utf-8")
|
base64_content = base64.b64encode(content).decode("utf-8")
|
||||||
return base64_content
|
return base64_content
|
||||||
else:
|
else:
|
||||||
logger.error(f"Failed to download slack file: {resp.status} {await resp.text()}")
|
logger.error(
|
||||||
|
f"Failed to download slack file: {resp.status} {await resp.text()}"
|
||||||
|
)
|
||||||
raise Exception(f"下载文件失败: {resp.status}")
|
raise Exception(f"下载文件失败: {resp.status}")
|
||||||
|
|
||||||
async def run(self) -> Awaitable[Any]:
|
async def run(self) -> Awaitable[Any]:
|
||||||
|
|||||||
@@ -75,7 +75,13 @@ class SlackMessageEvent(AstrMessageEvent):
|
|||||||
"text": {"type": "mrkdwn", "text": "文件上传失败"},
|
"text": {"type": "mrkdwn", "text": "文件上传失败"},
|
||||||
}
|
}
|
||||||
file_url = response["files"][0]["permalink"]
|
file_url = response["files"][0]["permalink"]
|
||||||
return {"type": "section", "text": {"type": "mrkdwn", "text": f"文件: <{file_url}|{segment.name or '文件'}>"}}
|
return {
|
||||||
|
"type": "section",
|
||||||
|
"text": {
|
||||||
|
"type": "mrkdwn",
|
||||||
|
"text": f"文件: <{file_url}|{segment.name or '文件'}>",
|
||||||
|
},
|
||||||
|
}
|
||||||
else:
|
else:
|
||||||
return {"type": "section", "text": {"type": "mrkdwn", "text": str(segment)}}
|
return {"type": "section", "text": {"type": "mrkdwn", "text": str(segment)}}
|
||||||
|
|
||||||
|
|||||||
@@ -95,9 +95,8 @@ class TelegramPlatformAdapter(Platform):
|
|||||||
|
|
||||||
@override
|
@override
|
||||||
def meta(self) -> PlatformMetadata:
|
def meta(self) -> PlatformMetadata:
|
||||||
return PlatformMetadata(
|
id_ = self.config.get("id") or "telegram"
|
||||||
name="telegram", description="telegram 适配器", id=self.config.get("id")
|
return PlatformMetadata(name="telegram", description="telegram 适配器", id=id_)
|
||||||
)
|
|
||||||
|
|
||||||
@override
|
@override
|
||||||
async def run(self):
|
async def run(self):
|
||||||
@@ -117,6 +116,10 @@ class TelegramPlatformAdapter(Platform):
|
|||||||
)
|
)
|
||||||
self.scheduler.start()
|
self.scheduler.start()
|
||||||
|
|
||||||
|
if not self.application.updater:
|
||||||
|
logger.error("Telegram Updater is not initialized. Cannot start polling.")
|
||||||
|
return
|
||||||
|
|
||||||
queue = self.application.updater.start_polling()
|
queue = self.application.updater.start_polling()
|
||||||
logger.info("Telegram Platform Adapter is running.")
|
logger.info("Telegram Platform Adapter is running.")
|
||||||
await queue
|
await queue
|
||||||
@@ -194,6 +197,11 @@ class TelegramPlatformAdapter(Platform):
|
|||||||
return cmd_name, description
|
return cmd_name, description
|
||||||
|
|
||||||
async def start(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
|
async def start(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
|
||||||
|
if not update.effective_chat:
|
||||||
|
logger.warning(
|
||||||
|
"Received a start command without an effective chat, skipping /start reply."
|
||||||
|
)
|
||||||
|
return
|
||||||
await context.bot.send_message(
|
await context.bot.send_message(
|
||||||
chat_id=update.effective_chat.id, text=self.config["start_message"]
|
chat_id=update.effective_chat.id, text=self.config["start_message"]
|
||||||
)
|
)
|
||||||
@@ -206,15 +214,20 @@ class TelegramPlatformAdapter(Platform):
|
|||||||
|
|
||||||
async def convert_message(
|
async def convert_message(
|
||||||
self, update: Update, context: ContextTypes.DEFAULT_TYPE, get_reply=True
|
self, update: Update, context: ContextTypes.DEFAULT_TYPE, get_reply=True
|
||||||
) -> AstrBotMessage:
|
) -> AstrBotMessage | None:
|
||||||
"""转换 Telegram 的消息对象为 AstrBotMessage 对象。
|
"""转换 Telegram 的消息对象为 AstrBotMessage 对象。
|
||||||
|
|
||||||
@param update: Telegram 的 Update 对象。
|
@param update: Telegram 的 Update 对象。
|
||||||
@param context: Telegram 的 Context 对象。
|
@param context: Telegram 的 Context 对象。
|
||||||
@param get_reply: 是否获取回复消息。这个参数是为了防止多个回复嵌套。
|
@param get_reply: 是否获取回复消息。这个参数是为了防止多个回复嵌套。
|
||||||
"""
|
"""
|
||||||
|
if not update.message:
|
||||||
|
logger.warning("Received an update without a message.")
|
||||||
|
return None
|
||||||
|
|
||||||
message = AstrBotMessage()
|
message = AstrBotMessage()
|
||||||
message.session_id = str(update.message.chat.id)
|
message.session_id = str(update.message.chat.id)
|
||||||
|
|
||||||
# 获得是群聊还是私聊
|
# 获得是群聊还是私聊
|
||||||
if update.message.chat.type == ChatType.PRIVATE:
|
if update.message.chat.type == ChatType.PRIVATE:
|
||||||
message.type = MessageType.FRIEND_MESSAGE
|
message.type = MessageType.FRIEND_MESSAGE
|
||||||
@@ -225,10 +238,13 @@ class TelegramPlatformAdapter(Platform):
|
|||||||
# Topic Group
|
# Topic Group
|
||||||
message.group_id += "#" + str(update.message.message_thread_id)
|
message.group_id += "#" + str(update.message.message_thread_id)
|
||||||
message.session_id = message.group_id
|
message.session_id = message.group_id
|
||||||
|
|
||||||
message.message_id = str(update.message.message_id)
|
message.message_id = str(update.message.message_id)
|
||||||
|
_from_user = update.message.from_user
|
||||||
|
if not _from_user:
|
||||||
|
logger.warning("[Telegram] Received a message without a from_user.")
|
||||||
|
return None
|
||||||
message.sender = MessageMember(
|
message.sender = MessageMember(
|
||||||
str(update.message.from_user.id), update.message.from_user.username
|
str(_from_user.id), _from_user.username or "Unknown"
|
||||||
)
|
)
|
||||||
message.self_id = str(context.bot.username)
|
message.self_id = str(context.bot.username)
|
||||||
message.raw_message = update
|
message.raw_message = update
|
||||||
@@ -247,6 +263,7 @@ class TelegramPlatformAdapter(Platform):
|
|||||||
)
|
)
|
||||||
reply_abm = await self.convert_message(reply_update, context, False)
|
reply_abm = await self.convert_message(reply_update, context, False)
|
||||||
|
|
||||||
|
if reply_abm:
|
||||||
message.message.append(
|
message.message.append(
|
||||||
Comp.Reply(
|
Comp.Reply(
|
||||||
id=reply_abm.message_id,
|
id=reply_abm.message_id,
|
||||||
@@ -263,6 +280,15 @@ class TelegramPlatformAdapter(Platform):
|
|||||||
if update.message.text:
|
if update.message.text:
|
||||||
# 处理文本消息
|
# 处理文本消息
|
||||||
plain_text = update.message.text
|
plain_text = update.message.text
|
||||||
|
if (
|
||||||
|
message.type == MessageType.GROUP_MESSAGE
|
||||||
|
and update.message
|
||||||
|
and update.message.reply_to_message
|
||||||
|
and update.message.reply_to_message.from_user
|
||||||
|
and update.message.reply_to_message.from_user.id == context.bot.id
|
||||||
|
):
|
||||||
|
plain_text2 = f"/@{context.bot.username} " + plain_text
|
||||||
|
plain_text = plain_text2
|
||||||
|
|
||||||
# 群聊场景命令特殊处理
|
# 群聊场景命令特殊处理
|
||||||
if plain_text.startswith("/"):
|
if plain_text.startswith("/"):
|
||||||
@@ -328,15 +354,25 @@ class TelegramPlatformAdapter(Platform):
|
|||||||
|
|
||||||
elif update.message.document:
|
elif update.message.document:
|
||||||
file = await update.message.document.get_file()
|
file = await update.message.document.get_file()
|
||||||
message.message = [
|
file_name = update.message.document.file_name or uuid.uuid4().hex
|
||||||
Comp.File(file=file.file_path, name=update.message.document.file_name),
|
file_path = file.file_path
|
||||||
]
|
if file_path is None:
|
||||||
|
logger.warning(
|
||||||
|
f"Telegram document file_path is None, cannot save the file {file_name}."
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
message.message.append(Comp.File(file=file_path, name=file_name))
|
||||||
|
|
||||||
elif update.message.video:
|
elif update.message.video:
|
||||||
file = await update.message.video.get_file()
|
file = await update.message.video.get_file()
|
||||||
message.message = [
|
file_name = update.message.video.file_name or uuid.uuid4().hex
|
||||||
Comp.Video(file=file.file_path, path=file.file_path),
|
file_path = file.file_path
|
||||||
]
|
if file_path is None:
|
||||||
|
logger.warning(
|
||||||
|
f"Telegram video file_path is None, cannot save the file {file_name}."
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
message.message.append(Comp.Video(file=file_path, path=file.file_path))
|
||||||
|
|
||||||
return message
|
return message
|
||||||
|
|
||||||
|
|||||||
@@ -16,6 +16,7 @@ from telegram.ext import ExtBot
|
|||||||
from astrbot.core.utils.io import download_file
|
from astrbot.core.utils.io import download_file
|
||||||
from astrbot import logger
|
from astrbot import logger
|
||||||
from astrbot.core.utils.astrbot_path import get_astrbot_data_path
|
from astrbot.core.utils.astrbot_path import get_astrbot_data_path
|
||||||
|
from telegram import ReactionTypeEmoji, ReactionTypeCustomEmoji
|
||||||
|
|
||||||
|
|
||||||
class TelegramPlatformEvent(AstrMessageEvent):
|
class TelegramPlatformEvent(AstrMessageEvent):
|
||||||
@@ -66,7 +67,9 @@ class TelegramPlatformEvent(AstrMessageEvent):
|
|||||||
return chunks
|
return chunks
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
async def send_with_client(cls, client: ExtBot, message: MessageChain, user_name: str):
|
async def send_with_client(
|
||||||
|
cls, client: ExtBot, message: MessageChain, user_name: str
|
||||||
|
):
|
||||||
image_path = None
|
image_path = None
|
||||||
|
|
||||||
has_reply = False
|
has_reply = False
|
||||||
@@ -133,6 +136,39 @@ class TelegramPlatformEvent(AstrMessageEvent):
|
|||||||
await self.send_with_client(self.client, message, self.get_sender_id())
|
await self.send_with_client(self.client, message, self.get_sender_id())
|
||||||
await super().send(message)
|
await super().send(message)
|
||||||
|
|
||||||
|
async def react(self, emoji: str | None, big: bool = False):
|
||||||
|
"""
|
||||||
|
给原消息添加 Telegram 反应:
|
||||||
|
- 普通 emoji:传入 '👍'、'😂' 等
|
||||||
|
- 自定义表情:传入其 custom_emoji_id(纯数字字符串)
|
||||||
|
- 取消本机器人的反应:传入 None 或空字符串
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
# 解析 chat_id(去掉超级群的 "#<thread_id>" 片段)
|
||||||
|
if self.get_message_type() == MessageType.GROUP_MESSAGE:
|
||||||
|
chat_id = (self.message_obj.group_id or "").split("#")[0]
|
||||||
|
else:
|
||||||
|
chat_id = self.get_sender_id()
|
||||||
|
|
||||||
|
message_id = int(self.message_obj.message_id)
|
||||||
|
|
||||||
|
# 组装 reaction 参数(必须是 ReactionType 的列表)
|
||||||
|
if not emoji: # 清空本 bot 的反应
|
||||||
|
reaction_param = [] # 空列表表示移除本 bot 的反应
|
||||||
|
elif emoji.isdigit(): # 自定义表情:传 custom_emoji_id
|
||||||
|
reaction_param = [ReactionTypeCustomEmoji(emoji)]
|
||||||
|
else: # 普通 emoji
|
||||||
|
reaction_param = [ReactionTypeEmoji(emoji)]
|
||||||
|
|
||||||
|
await self.client.set_message_reaction(
|
||||||
|
chat_id=chat_id,
|
||||||
|
message_id=message_id,
|
||||||
|
reaction=reaction_param, # 注意是列表
|
||||||
|
is_big=big, # 可选:大动画
|
||||||
|
)
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"[Telegram] 添加反应失败: {e}")
|
||||||
|
|
||||||
async def send_streaming(self, generator, use_fallback: bool = False):
|
async def send_streaming(self, generator, use_fallback: bool = False):
|
||||||
message_thread_id = None
|
message_thread_id = None
|
||||||
|
|
||||||
@@ -216,7 +252,6 @@ class TelegramPlatformEvent(AstrMessageEvent):
|
|||||||
try:
|
try:
|
||||||
msg = await self.client.send_message(text=delta, **payload)
|
msg = await self.client.send_message(text=delta, **payload)
|
||||||
current_content = delta
|
current_content = delta
|
||||||
delta = ""
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.warning(f"发送消息失败(streaming): {e!s}")
|
logger.warning(f"发送消息失败(streaming): {e!s}")
|
||||||
message_id = msg.message_id
|
message_id = msg.message_id
|
||||||
|
|||||||
@@ -1,5 +1,6 @@
|
|||||||
import asyncio
|
import asyncio
|
||||||
|
|
||||||
|
|
||||||
class WebChatQueueMgr:
|
class WebChatQueueMgr:
|
||||||
def __init__(self) -> None:
|
def __init__(self) -> None:
|
||||||
self.queues = {}
|
self.queues = {}
|
||||||
@@ -30,4 +31,5 @@ class WebChatQueueMgr:
|
|||||||
"""Check if a queue exists for the given conversation ID"""
|
"""Check if a queue exists for the given conversation ID"""
|
||||||
return conversation_id in self.queues
|
return conversation_id in self.queues
|
||||||
|
|
||||||
|
|
||||||
webchat_queue_mgr = WebChatQueueMgr()
|
webchat_queue_mgr = WebChatQueueMgr()
|
||||||
|
|||||||
@@ -234,7 +234,9 @@ class WeChatPadProAdapter(Platform):
|
|||||||
try:
|
try:
|
||||||
async with session.post(url, params=params, json=payload) as response:
|
async with session.post(url, params=params, json=payload) as response:
|
||||||
if response.status != 200:
|
if response.status != 200:
|
||||||
logger.error(f"生成授权码失败: {response.status}, {await response.text()}")
|
logger.error(
|
||||||
|
f"生成授权码失败: {response.status}, {await response.text()}"
|
||||||
|
)
|
||||||
return
|
return
|
||||||
|
|
||||||
response_data = await response.json()
|
response_data = await response.json()
|
||||||
@@ -245,7 +247,9 @@ class WeChatPadProAdapter(Platform):
|
|||||||
if self.auth_key:
|
if self.auth_key:
|
||||||
logger.info("成功获取授权码")
|
logger.info("成功获取授权码")
|
||||||
else:
|
else:
|
||||||
logger.error(f"生成授权码成功但未找到授权码: {response_data}")
|
logger.error(
|
||||||
|
f"生成授权码成功但未找到授权码: {response_data}"
|
||||||
|
)
|
||||||
else:
|
else:
|
||||||
logger.error(f"生成授权码失败: {response_data}")
|
logger.error(f"生成授权码失败: {response_data}")
|
||||||
except aiohttp.ClientConnectorError as e:
|
except aiohttp.ClientConnectorError as e:
|
||||||
|
|||||||
@@ -185,6 +185,7 @@ class WecomPlatformAdapter(Platform):
|
|||||||
return PlatformMetadata(
|
return PlatformMetadata(
|
||||||
"wecom",
|
"wecom",
|
||||||
"wecom 适配器",
|
"wecom 适配器",
|
||||||
|
id=self.config.get("id", "wecom"),
|
||||||
)
|
)
|
||||||
|
|
||||||
@override
|
@override
|
||||||
|
|||||||
@@ -48,7 +48,12 @@ class WeChatKF(BaseWeChatAPI):
|
|||||||
注意:可能会出现返回条数少于limit的情况,需结合返回的has_more字段判断是否继续请求。
|
注意:可能会出现返回条数少于limit的情况,需结合返回的has_more字段判断是否继续请求。
|
||||||
:return: 接口调用结果
|
:return: 接口调用结果
|
||||||
"""
|
"""
|
||||||
data = {"token": token, "cursor": cursor, "limit": limit, "open_kfid": open_kfid}
|
data = {
|
||||||
|
"token": token,
|
||||||
|
"cursor": cursor,
|
||||||
|
"limit": limit,
|
||||||
|
"open_kfid": open_kfid,
|
||||||
|
}
|
||||||
return self._post("kf/sync_msg", data=data)
|
return self._post("kf/sync_msg", data=data)
|
||||||
|
|
||||||
def get_service_state(self, open_kfid, external_userid):
|
def get_service_state(self, open_kfid, external_userid):
|
||||||
@@ -72,7 +77,9 @@ class WeChatKF(BaseWeChatAPI):
|
|||||||
}
|
}
|
||||||
return self._post("kf/service_state/get", data=data)
|
return self._post("kf/service_state/get", data=data)
|
||||||
|
|
||||||
def trans_service_state(self, open_kfid, external_userid, service_state, servicer_userid=""):
|
def trans_service_state(
|
||||||
|
self, open_kfid, external_userid, service_state, servicer_userid=""
|
||||||
|
):
|
||||||
"""
|
"""
|
||||||
变更会话状态
|
变更会话状态
|
||||||
|
|
||||||
@@ -180,7 +187,9 @@ class WeChatKF(BaseWeChatAPI):
|
|||||||
"""
|
"""
|
||||||
return self._get("kf/customer/get_upgrade_service_config")
|
return self._get("kf/customer/get_upgrade_service_config")
|
||||||
|
|
||||||
def upgrade_service(self, open_kfid, external_userid, service_type, member=None, groupchat=None):
|
def upgrade_service(
|
||||||
|
self, open_kfid, external_userid, service_type, member=None, groupchat=None
|
||||||
|
):
|
||||||
"""
|
"""
|
||||||
为客户升级为专员或客户群服务
|
为客户升级为专员或客户群服务
|
||||||
|
|
||||||
@@ -246,7 +255,9 @@ class WeChatKF(BaseWeChatAPI):
|
|||||||
data = {"open_kfid": open_kfid, "start_time": start_time, "end_time": end_time}
|
data = {"open_kfid": open_kfid, "start_time": start_time, "end_time": end_time}
|
||||||
return self._post("kf/get_corp_statistic", data=data)
|
return self._post("kf/get_corp_statistic", data=data)
|
||||||
|
|
||||||
def get_servicer_statistic(self, start_time, end_time, open_kfid=None, servicer_userid=None):
|
def get_servicer_statistic(
|
||||||
|
self, start_time, end_time, open_kfid=None, servicer_userid=None
|
||||||
|
):
|
||||||
"""
|
"""
|
||||||
获取「客户数据统计」接待人员明细数据
|
获取「客户数据统计」接待人员明细数据
|
||||||
|
|
||||||
|
|||||||
@@ -26,6 +26,7 @@ from optionaldict import optionaldict
|
|||||||
|
|
||||||
from wechatpy.client.api.base import BaseWeChatAPI
|
from wechatpy.client.api.base import BaseWeChatAPI
|
||||||
|
|
||||||
|
|
||||||
class WeChatKFMessage(BaseWeChatAPI):
|
class WeChatKFMessage(BaseWeChatAPI):
|
||||||
"""
|
"""
|
||||||
发送微信客服消息
|
发送微信客服消息
|
||||||
@@ -125,35 +126,55 @@ class WeChatKFMessage(BaseWeChatAPI):
|
|||||||
msg={"msgtype": "news", "link": {"link": articles_data}},
|
msg={"msgtype": "news", "link": {"link": articles_data}},
|
||||||
)
|
)
|
||||||
|
|
||||||
def send_msgmenu(self, user_id, open_kfid, head_content, menu_list, tail_content, msgid=""):
|
def send_msgmenu(
|
||||||
|
self, user_id, open_kfid, head_content, menu_list, tail_content, msgid=""
|
||||||
|
):
|
||||||
return self.send(
|
return self.send(
|
||||||
user_id,
|
user_id,
|
||||||
open_kfid,
|
open_kfid,
|
||||||
msgid,
|
msgid,
|
||||||
msg={
|
msg={
|
||||||
"msgtype": "msgmenu",
|
"msgtype": "msgmenu",
|
||||||
"msgmenu": {"head_content": head_content, "list": menu_list, "tail_content": tail_content},
|
"msgmenu": {
|
||||||
|
"head_content": head_content,
|
||||||
|
"list": menu_list,
|
||||||
|
"tail_content": tail_content,
|
||||||
|
},
|
||||||
},
|
},
|
||||||
)
|
)
|
||||||
|
|
||||||
def send_location(self, user_id, open_kfid, name, address, latitude, longitude, msgid=""):
|
def send_location(
|
||||||
|
self, user_id, open_kfid, name, address, latitude, longitude, msgid=""
|
||||||
|
):
|
||||||
return self.send(
|
return self.send(
|
||||||
user_id,
|
user_id,
|
||||||
open_kfid,
|
open_kfid,
|
||||||
msgid,
|
msgid,
|
||||||
msg={
|
msg={
|
||||||
"msgtype": "location",
|
"msgtype": "location",
|
||||||
"msgmenu": {"name": name, "address": address, "latitude": latitude, "longitude": longitude},
|
"msgmenu": {
|
||||||
|
"name": name,
|
||||||
|
"address": address,
|
||||||
|
"latitude": latitude,
|
||||||
|
"longitude": longitude,
|
||||||
|
},
|
||||||
},
|
},
|
||||||
)
|
)
|
||||||
|
|
||||||
def send_miniprogram(self, user_id, open_kfid, appid, title, thumb_media_id, pagepath, msgid=""):
|
def send_miniprogram(
|
||||||
|
self, user_id, open_kfid, appid, title, thumb_media_id, pagepath, msgid=""
|
||||||
|
):
|
||||||
return self.send(
|
return self.send(
|
||||||
user_id,
|
user_id,
|
||||||
open_kfid,
|
open_kfid,
|
||||||
msgid,
|
msgid,
|
||||||
msg={
|
msg={
|
||||||
"msgtype": "miniprogram",
|
"msgtype": "miniprogram",
|
||||||
"msgmenu": {"appid": appid, "title": title, "thumb_media_id": thumb_media_id, "pagepath": pagepath},
|
"msgmenu": {
|
||||||
|
"appid": appid,
|
||||||
|
"title": title,
|
||||||
|
"thumb_media_id": thumb_media_id,
|
||||||
|
"pagepath": pagepath,
|
||||||
|
},
|
||||||
},
|
},
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -160,7 +160,9 @@ class WeixinOfficialAccountPlatformAdapter(Platform):
|
|||||||
self.wexin_event_workers[msg.id] = future
|
self.wexin_event_workers[msg.id] = future
|
||||||
await self.convert_message(msg, future)
|
await self.convert_message(msg, future)
|
||||||
# I love shield so much!
|
# I love shield so much!
|
||||||
result = await asyncio.wait_for(asyncio.shield(future), 60) # wait for 60s
|
result = await asyncio.wait_for(
|
||||||
|
asyncio.shield(future), 60
|
||||||
|
) # wait for 60s
|
||||||
logger.debug(f"Got future result: {result}")
|
logger.debug(f"Got future result: {result}")
|
||||||
self.wexin_event_workers.pop(msg.id, None)
|
self.wexin_event_workers.pop(msg.id, None)
|
||||||
return result # xml. see weixin_offacc_event.py
|
return result # xml. see weixin_offacc_event.py
|
||||||
@@ -182,6 +184,7 @@ class WeixinOfficialAccountPlatformAdapter(Platform):
|
|||||||
return PlatformMetadata(
|
return PlatformMetadata(
|
||||||
"weixin_official_account",
|
"weixin_official_account",
|
||||||
"微信公众平台 适配器",
|
"微信公众平台 适配器",
|
||||||
|
id=self.config.get("id", "weixin_official_account"),
|
||||||
)
|
)
|
||||||
|
|
||||||
@override
|
@override
|
||||||
|
|||||||
@@ -150,7 +150,6 @@ class WeixinOfficialAccountPlatformEvent(AstrMessageEvent):
|
|||||||
return
|
return
|
||||||
logger.info(f"微信公众平台上传语音返回: {response}")
|
logger.info(f"微信公众平台上传语音返回: {response}")
|
||||||
|
|
||||||
|
|
||||||
if active_send_mode:
|
if active_send_mode:
|
||||||
self.client.message.send_voice(
|
self.client.message.send_voice(
|
||||||
message_obj.sender.user_id,
|
message_obj.sender.user_id,
|
||||||
|
|||||||
@@ -65,13 +65,16 @@ class AssistantMessageSegment:
|
|||||||
role: str = "assistant"
|
role: str = "assistant"
|
||||||
|
|
||||||
def to_dict(self):
|
def to_dict(self):
|
||||||
ret = {
|
ret: dict[str, str | list[dict]] = {
|
||||||
"role": self.role,
|
"role": self.role,
|
||||||
}
|
}
|
||||||
if self.content:
|
if self.content:
|
||||||
ret["content"] = self.content
|
ret["content"] = self.content
|
||||||
if self.tool_calls:
|
if self.tool_calls:
|
||||||
ret["tool_calls"] = self.tool_calls
|
tool_calls_dict = [
|
||||||
|
tc if isinstance(tc, dict) else tc.to_dict() for tc in self.tool_calls
|
||||||
|
]
|
||||||
|
ret["tool_calls"] = tool_calls_dict
|
||||||
return ret
|
return ret
|
||||||
|
|
||||||
|
|
||||||
@@ -117,7 +120,14 @@ class ProviderRequest:
|
|||||||
"""模型名称,为 None 时使用提供商的默认模型"""
|
"""模型名称,为 None 时使用提供商的默认模型"""
|
||||||
|
|
||||||
def __repr__(self):
|
def __repr__(self):
|
||||||
return f"ProviderRequest(prompt={self.prompt}, session_id={self.session_id}, image_urls={self.image_urls}, func_tool={self.func_tool}, contexts={self._print_friendly_context()}, system_prompt={self.system_prompt.strip()}, tool_calls_result={self.tool_calls_result})"
|
return (
|
||||||
|
f"ProviderRequest(prompt={self.prompt}, session_id={self.session_id}, "
|
||||||
|
f"image_count={len(self.image_urls or [])}, "
|
||||||
|
f"func_tool={self.func_tool}, "
|
||||||
|
f"contexts={self._print_friendly_context()}, "
|
||||||
|
f"system_prompt={self.system_prompt}, "
|
||||||
|
f"conversation_id={self.conversation.cid if self.conversation else 'N/A'}, "
|
||||||
|
)
|
||||||
|
|
||||||
def __str__(self):
|
def __str__(self):
|
||||||
return self.__repr__()
|
return self.__repr__()
|
||||||
@@ -297,6 +307,7 @@ class LLMResponse:
|
|||||||
)
|
)
|
||||||
return ret
|
return ret
|
||||||
|
|
||||||
|
|
||||||
@dataclass
|
@dataclass
|
||||||
class RerankResult:
|
class RerankResult:
|
||||||
index: int
|
index: int
|
||||||
|
|||||||
@@ -4,7 +4,7 @@ import os
|
|||||||
import asyncio
|
import asyncio
|
||||||
import aiohttp
|
import aiohttp
|
||||||
|
|
||||||
from typing import Dict, List, Awaitable
|
from typing import Dict, List, Awaitable, Callable, Any
|
||||||
from astrbot import logger
|
from astrbot import logger
|
||||||
from astrbot.core import sp
|
from astrbot.core import sp
|
||||||
|
|
||||||
@@ -109,7 +109,7 @@ class FunctionToolManager:
|
|||||||
name: str,
|
name: str,
|
||||||
func_args: list,
|
func_args: list,
|
||||||
desc: str,
|
desc: str,
|
||||||
handler: Awaitable,
|
handler: Callable[..., Awaitable[Any]],
|
||||||
) -> FuncTool:
|
) -> FuncTool:
|
||||||
params = {
|
params = {
|
||||||
"type": "object", # hard-coded here
|
"type": "object", # hard-coded here
|
||||||
@@ -132,7 +132,7 @@ class FunctionToolManager:
|
|||||||
name: str,
|
name: str,
|
||||||
func_args: list,
|
func_args: list,
|
||||||
desc: str,
|
desc: str,
|
||||||
handler: Awaitable,
|
handler: Callable[..., Awaitable[Any]],
|
||||||
) -> None:
|
) -> None:
|
||||||
"""添加函数调用工具
|
"""添加函数调用工具
|
||||||
|
|
||||||
@@ -220,7 +220,7 @@ class FunctionToolManager:
|
|||||||
name: str,
|
name: str,
|
||||||
cfg: dict,
|
cfg: dict,
|
||||||
event: asyncio.Event,
|
event: asyncio.Event,
|
||||||
ready_future: asyncio.Future = None,
|
ready_future: asyncio.Future | None = None,
|
||||||
) -> None:
|
) -> None:
|
||||||
"""初始化 MCP 客户端的包装函数,用于捕获异常"""
|
"""初始化 MCP 客户端的包装函数,用于捕获异常"""
|
||||||
try:
|
try:
|
||||||
|
|||||||
@@ -7,7 +7,13 @@ from astrbot.core.astrbot_config_mgr import AstrBotConfigManager
|
|||||||
from astrbot.core.db import BaseDatabase
|
from astrbot.core.db import BaseDatabase
|
||||||
|
|
||||||
from .entities import ProviderType
|
from .entities import ProviderType
|
||||||
from .provider import Provider, STTProvider, TTSProvider, EmbeddingProvider
|
from .provider import (
|
||||||
|
Provider,
|
||||||
|
STTProvider,
|
||||||
|
TTSProvider,
|
||||||
|
EmbeddingProvider,
|
||||||
|
RerankProvider,
|
||||||
|
)
|
||||||
from .register import llm_tools, provider_cls_map
|
from .register import llm_tools, provider_cls_map
|
||||||
from ..persona_mgr import PersonaManager
|
from ..persona_mgr import PersonaManager
|
||||||
|
|
||||||
@@ -38,7 +44,12 @@ class ProviderManager:
|
|||||||
"""加载的 Text To Speech Provider 的实例"""
|
"""加载的 Text To Speech Provider 的实例"""
|
||||||
self.embedding_provider_insts: List[EmbeddingProvider] = []
|
self.embedding_provider_insts: List[EmbeddingProvider] = []
|
||||||
"""加载的 Embedding Provider 的实例"""
|
"""加载的 Embedding Provider 的实例"""
|
||||||
self.inst_map: dict[str, Provider] = {}
|
self.rerank_provider_insts: List[RerankProvider] = []
|
||||||
|
"""加载的 Rerank Provider 的实例"""
|
||||||
|
self.inst_map: dict[
|
||||||
|
str,
|
||||||
|
Provider | STTProvider | TTSProvider | EmbeddingProvider | RerankProvider,
|
||||||
|
] = {}
|
||||||
"""Provider 实例映射. key: provider_id, value: Provider 实例"""
|
"""Provider 实例映射. key: provider_id, value: Provider 实例"""
|
||||||
self.llm_tools = llm_tools
|
self.llm_tools = llm_tools
|
||||||
|
|
||||||
@@ -87,19 +98,31 @@ class ProviderManager:
|
|||||||
)
|
)
|
||||||
return
|
return
|
||||||
# 不启用提供商会话隔离模式的情况
|
# 不启用提供商会话隔离模式的情况
|
||||||
self.curr_provider_inst = self.inst_map[provider_id]
|
|
||||||
if provider_type == ProviderType.TEXT_TO_SPEECH:
|
prov = self.inst_map[provider_id]
|
||||||
|
if provider_type == ProviderType.TEXT_TO_SPEECH and isinstance(
|
||||||
|
prov, TTSProvider
|
||||||
|
):
|
||||||
|
self.curr_tts_provider_inst = prov
|
||||||
sp.put("curr_provider_tts", provider_id, scope="global", scope_id="global")
|
sp.put("curr_provider_tts", provider_id, scope="global", scope_id="global")
|
||||||
elif provider_type == ProviderType.SPEECH_TO_TEXT:
|
elif provider_type == ProviderType.SPEECH_TO_TEXT and isinstance(
|
||||||
|
prov, STTProvider
|
||||||
|
):
|
||||||
|
self.curr_stt_provider_inst = prov
|
||||||
sp.put("curr_provider_stt", provider_id, scope="global", scope_id="global")
|
sp.put("curr_provider_stt", provider_id, scope="global", scope_id="global")
|
||||||
elif provider_type == ProviderType.CHAT_COMPLETION:
|
elif provider_type == ProviderType.CHAT_COMPLETION and isinstance(
|
||||||
|
prov, Provider
|
||||||
|
):
|
||||||
|
self.curr_provider_inst = prov
|
||||||
sp.put("curr_provider", provider_id, scope="global", scope_id="global")
|
sp.put("curr_provider", provider_id, scope="global", scope_id="global")
|
||||||
|
|
||||||
async def get_provider_by_id(self, provider_id: str) -> Provider | None:
|
async def get_provider_by_id(self, provider_id: str) -> Provider | None:
|
||||||
"""根据提供商 ID 获取提供商实例"""
|
"""根据提供商 ID 获取提供商实例"""
|
||||||
return self.inst_map.get(provider_id)
|
return self.inst_map.get(provider_id)
|
||||||
|
|
||||||
def get_using_provider(self, provider_type: ProviderType, umo=None):
|
def get_using_provider(
|
||||||
|
self, provider_type: ProviderType, umo=None
|
||||||
|
) -> Provider | STTProvider | TTSProvider | None:
|
||||||
"""获取正在使用的提供商实例。
|
"""获取正在使用的提供商实例。
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
@@ -211,6 +234,8 @@ class ProviderManager:
|
|||||||
)
|
)
|
||||||
case "dify":
|
case "dify":
|
||||||
from .sources.dify_source import ProviderDify as ProviderDify
|
from .sources.dify_source import ProviderDify as ProviderDify
|
||||||
|
case "coze":
|
||||||
|
from .sources.coze_source import ProviderCoze as ProviderCoze
|
||||||
case "dashscope":
|
case "dashscope":
|
||||||
from .sources.dashscope_source import (
|
from .sources.dashscope_source import (
|
||||||
ProviderDashscope as ProviderDashscope,
|
ProviderDashscope as ProviderDashscope,
|
||||||
@@ -303,12 +328,14 @@ class ProviderManager:
|
|||||||
provider_metadata = provider_cls_map[provider_config["type"]]
|
provider_metadata = provider_cls_map[provider_config["type"]]
|
||||||
try:
|
try:
|
||||||
# 按任务实例化提供商
|
# 按任务实例化提供商
|
||||||
|
cls_type = provider_metadata.cls_type
|
||||||
|
if not cls_type:
|
||||||
|
logger.error(f"无法找到 {provider_metadata.type} 的类")
|
||||||
|
return
|
||||||
|
|
||||||
if provider_metadata.provider_type == ProviderType.SPEECH_TO_TEXT:
|
if provider_metadata.provider_type == ProviderType.SPEECH_TO_TEXT:
|
||||||
# STT 任务
|
# STT 任务
|
||||||
inst = provider_metadata.cls_type(
|
inst = cls_type(provider_config, self.provider_settings)
|
||||||
provider_config, self.provider_settings
|
|
||||||
)
|
|
||||||
|
|
||||||
if getattr(inst, "initialize", None):
|
if getattr(inst, "initialize", None):
|
||||||
await inst.initialize()
|
await inst.initialize()
|
||||||
@@ -327,9 +354,7 @@ class ProviderManager:
|
|||||||
|
|
||||||
elif provider_metadata.provider_type == ProviderType.TEXT_TO_SPEECH:
|
elif provider_metadata.provider_type == ProviderType.TEXT_TO_SPEECH:
|
||||||
# TTS 任务
|
# TTS 任务
|
||||||
inst = provider_metadata.cls_type(
|
inst = cls_type(provider_config, self.provider_settings)
|
||||||
provider_config, self.provider_settings
|
|
||||||
)
|
|
||||||
|
|
||||||
if getattr(inst, "initialize", None):
|
if getattr(inst, "initialize", None):
|
||||||
await inst.initialize()
|
await inst.initialize()
|
||||||
@@ -345,7 +370,7 @@ class ProviderManager:
|
|||||||
|
|
||||||
elif provider_metadata.provider_type == ProviderType.CHAT_COMPLETION:
|
elif provider_metadata.provider_type == ProviderType.CHAT_COMPLETION:
|
||||||
# 文本生成任务
|
# 文本生成任务
|
||||||
inst = provider_metadata.cls_type(
|
inst = cls_type(
|
||||||
provider_config,
|
provider_config,
|
||||||
self.provider_settings,
|
self.provider_settings,
|
||||||
self.selected_default_persona,
|
self.selected_default_persona,
|
||||||
@@ -366,13 +391,16 @@ class ProviderManager:
|
|||||||
if not self.curr_provider_inst:
|
if not self.curr_provider_inst:
|
||||||
self.curr_provider_inst = inst
|
self.curr_provider_inst = inst
|
||||||
|
|
||||||
elif provider_metadata.provider_type in [ProviderType.EMBEDDING, ProviderType.RERANK]:
|
elif provider_metadata.provider_type == ProviderType.EMBEDDING:
|
||||||
inst = provider_metadata.cls_type(
|
inst = cls_type(provider_config, self.provider_settings)
|
||||||
provider_config, self.provider_settings
|
|
||||||
)
|
|
||||||
if getattr(inst, "initialize", None):
|
if getattr(inst, "initialize", None):
|
||||||
await inst.initialize()
|
await inst.initialize()
|
||||||
self.embedding_provider_insts.append(inst)
|
self.embedding_provider_insts.append(inst)
|
||||||
|
elif provider_metadata.provider_type == ProviderType.RERANK:
|
||||||
|
inst = cls_type(provider_config, self.provider_settings)
|
||||||
|
if getattr(inst, "initialize", None):
|
||||||
|
await inst.initialize()
|
||||||
|
self.rerank_provider_insts.append(inst)
|
||||||
|
|
||||||
self.inst_map[provider_config["id"]] = inst
|
self.inst_map[provider_config["id"]] = inst
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
@@ -388,6 +416,7 @@ class ProviderManager:
|
|||||||
|
|
||||||
# 和配置文件保持同步
|
# 和配置文件保持同步
|
||||||
config_ids = [provider["id"] for provider in self.providers_config]
|
config_ids = [provider["id"] for provider in self.providers_config]
|
||||||
|
logger.debug(f"providers in user's config: {config_ids}")
|
||||||
for key in list(self.inst_map.keys()):
|
for key in list(self.inst_map.keys()):
|
||||||
if key not in config_ids:
|
if key not in config_ids:
|
||||||
await self.terminate_provider(key)
|
await self.terminate_provider(key)
|
||||||
@@ -426,11 +455,17 @@ class ProviderManager:
|
|||||||
)
|
)
|
||||||
|
|
||||||
if self.inst_map[provider_id] in self.provider_insts:
|
if self.inst_map[provider_id] in self.provider_insts:
|
||||||
self.provider_insts.remove(self.inst_map[provider_id])
|
prov_inst = self.inst_map[provider_id]
|
||||||
|
if isinstance(prov_inst, Provider):
|
||||||
|
self.provider_insts.remove(prov_inst)
|
||||||
if self.inst_map[provider_id] in self.stt_provider_insts:
|
if self.inst_map[provider_id] in self.stt_provider_insts:
|
||||||
self.stt_provider_insts.remove(self.inst_map[provider_id])
|
prov_inst = self.inst_map[provider_id]
|
||||||
|
if isinstance(prov_inst, STTProvider):
|
||||||
|
self.stt_provider_insts.remove(prov_inst)
|
||||||
if self.inst_map[provider_id] in self.tts_provider_insts:
|
if self.inst_map[provider_id] in self.tts_provider_insts:
|
||||||
self.tts_provider_insts.remove(self.inst_map[provider_id])
|
prov_inst = self.inst_map[provider_id]
|
||||||
|
if isinstance(prov_inst, TTSProvider):
|
||||||
|
self.tts_provider_insts.remove(prov_inst)
|
||||||
|
|
||||||
if self.inst_map[provider_id] == self.curr_provider_inst:
|
if self.inst_map[provider_id] == self.curr_provider_inst:
|
||||||
self.curr_provider_inst = None
|
self.curr_provider_inst = None
|
||||||
|
|||||||
@@ -75,7 +75,7 @@ class Provider(AbstractProvider):
|
|||||||
raise NotImplementedError()
|
raise NotImplementedError()
|
||||||
|
|
||||||
@abc.abstractmethod
|
@abc.abstractmethod
|
||||||
def get_models(self) -> List[str]:
|
async def get_models(self) -> List[str]:
|
||||||
"""获得支持的模型列表"""
|
"""获得支持的模型列表"""
|
||||||
raise NotImplementedError()
|
raise NotImplementedError()
|
||||||
|
|
||||||
|
|||||||
314
astrbot/core/provider/sources/coze_api_client.py
Normal file
314
astrbot/core/provider/sources/coze_api_client.py
Normal file
@@ -0,0 +1,314 @@
|
|||||||
|
import json
|
||||||
|
import asyncio
|
||||||
|
import aiohttp
|
||||||
|
import io
|
||||||
|
from typing import Dict, List, Any, AsyncGenerator
|
||||||
|
from astrbot.core import logger
|
||||||
|
|
||||||
|
|
||||||
|
class CozeAPIClient:
|
||||||
|
def __init__(self, api_key: str, api_base: str = "https://api.coze.cn"):
|
||||||
|
self.api_key = api_key
|
||||||
|
self.api_base = api_base
|
||||||
|
self.session = None
|
||||||
|
|
||||||
|
async def _ensure_session(self):
|
||||||
|
"""确保HTTP session存在"""
|
||||||
|
if self.session is None:
|
||||||
|
connector = aiohttp.TCPConnector(
|
||||||
|
ssl=False if self.api_base.startswith("http://") else True,
|
||||||
|
limit=100,
|
||||||
|
limit_per_host=30,
|
||||||
|
keepalive_timeout=30,
|
||||||
|
enable_cleanup_closed=True,
|
||||||
|
)
|
||||||
|
timeout = aiohttp.ClientTimeout(
|
||||||
|
total=120, # 默认超时时间
|
||||||
|
connect=30,
|
||||||
|
sock_read=120,
|
||||||
|
)
|
||||||
|
headers = {
|
||||||
|
"Authorization": f"Bearer {self.api_key}",
|
||||||
|
"Accept": "text/event-stream",
|
||||||
|
}
|
||||||
|
self.session = aiohttp.ClientSession(
|
||||||
|
headers=headers, timeout=timeout, connector=connector
|
||||||
|
)
|
||||||
|
return self.session
|
||||||
|
|
||||||
|
async def upload_file(
|
||||||
|
self,
|
||||||
|
file_data: bytes,
|
||||||
|
) -> str:
|
||||||
|
"""上传文件到 Coze 并返回 file_id
|
||||||
|
|
||||||
|
Args:
|
||||||
|
file_data (bytes): 文件的二进制数据
|
||||||
|
Returns:
|
||||||
|
str: 上传成功后返回的 file_id
|
||||||
|
"""
|
||||||
|
session = await self._ensure_session()
|
||||||
|
url = f"{self.api_base}/v1/files/upload"
|
||||||
|
|
||||||
|
try:
|
||||||
|
file_io = io.BytesIO(file_data)
|
||||||
|
async with session.post(
|
||||||
|
url,
|
||||||
|
data={
|
||||||
|
"file": file_io,
|
||||||
|
},
|
||||||
|
timeout=aiohttp.ClientTimeout(total=60),
|
||||||
|
) as response:
|
||||||
|
if response.status == 401:
|
||||||
|
raise Exception("Coze API 认证失败,请检查 API Key 是否正确")
|
||||||
|
|
||||||
|
response_text = await response.text()
|
||||||
|
logger.debug(
|
||||||
|
f"文件上传响应状态: {response.status}, 内容: {response_text}"
|
||||||
|
)
|
||||||
|
|
||||||
|
if response.status != 200:
|
||||||
|
raise Exception(
|
||||||
|
f"文件上传失败,状态码: {response.status}, 响应: {response_text}"
|
||||||
|
)
|
||||||
|
|
||||||
|
try:
|
||||||
|
result = await response.json()
|
||||||
|
except json.JSONDecodeError:
|
||||||
|
raise Exception(f"文件上传响应解析失败: {response_text}")
|
||||||
|
|
||||||
|
if result.get("code") != 0:
|
||||||
|
raise Exception(f"文件上传失败: {result.get('msg', '未知错误')}")
|
||||||
|
|
||||||
|
file_id = result["data"]["id"]
|
||||||
|
logger.debug(f"[Coze] 图片上传成功,file_id: {file_id}")
|
||||||
|
return file_id
|
||||||
|
|
||||||
|
except asyncio.TimeoutError:
|
||||||
|
logger.error("文件上传超时")
|
||||||
|
raise Exception("文件上传超时")
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"文件上传失败: {str(e)}")
|
||||||
|
raise Exception(f"文件上传失败: {str(e)}")
|
||||||
|
|
||||||
|
async def download_image(self, image_url: str) -> bytes:
|
||||||
|
"""下载图片并返回字节数据
|
||||||
|
|
||||||
|
Args:
|
||||||
|
image_url (str): 图片的URL
|
||||||
|
Returns:
|
||||||
|
bytes: 图片的二进制数据
|
||||||
|
"""
|
||||||
|
session = await self._ensure_session()
|
||||||
|
|
||||||
|
try:
|
||||||
|
async with session.get(image_url) as response:
|
||||||
|
if response.status != 200:
|
||||||
|
raise Exception(f"下载图片失败,状态码: {response.status}")
|
||||||
|
|
||||||
|
image_data = await response.read()
|
||||||
|
return image_data
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"下载图片失败 {image_url}: {str(e)}")
|
||||||
|
raise Exception(f"下载图片失败: {str(e)}")
|
||||||
|
|
||||||
|
async def chat_messages(
|
||||||
|
self,
|
||||||
|
bot_id: str,
|
||||||
|
user_id: str,
|
||||||
|
additional_messages: List[Dict] | None = None,
|
||||||
|
conversation_id: str | None = None,
|
||||||
|
auto_save_history: bool = True,
|
||||||
|
stream: bool = True,
|
||||||
|
timeout: float = 120,
|
||||||
|
) -> AsyncGenerator[Dict[str, Any], None]:
|
||||||
|
"""发送聊天消息并返回流式响应
|
||||||
|
|
||||||
|
Args:
|
||||||
|
bot_id: Bot ID
|
||||||
|
user_id: 用户ID
|
||||||
|
additional_messages: 额外消息列表
|
||||||
|
conversation_id: 会话ID
|
||||||
|
auto_save_history: 是否自动保存历史
|
||||||
|
stream: 是否流式响应
|
||||||
|
timeout: 超时时间
|
||||||
|
"""
|
||||||
|
session = await self._ensure_session()
|
||||||
|
url = f"{self.api_base}/v3/chat"
|
||||||
|
|
||||||
|
payload = {
|
||||||
|
"bot_id": bot_id,
|
||||||
|
"user_id": user_id,
|
||||||
|
"stream": stream,
|
||||||
|
"auto_save_history": auto_save_history,
|
||||||
|
}
|
||||||
|
|
||||||
|
if additional_messages:
|
||||||
|
payload["additional_messages"] = additional_messages
|
||||||
|
|
||||||
|
params = {}
|
||||||
|
if conversation_id:
|
||||||
|
params["conversation_id"] = conversation_id
|
||||||
|
|
||||||
|
logger.debug(f"Coze chat_messages payload: {payload}, params: {params}")
|
||||||
|
|
||||||
|
try:
|
||||||
|
async with session.post(
|
||||||
|
url,
|
||||||
|
json=payload,
|
||||||
|
params=params,
|
||||||
|
timeout=aiohttp.ClientTimeout(total=timeout),
|
||||||
|
) as response:
|
||||||
|
if response.status == 401:
|
||||||
|
raise Exception("Coze API 认证失败,请检查 API Key 是否正确")
|
||||||
|
|
||||||
|
if response.status != 200:
|
||||||
|
raise Exception(f"Coze API 流式请求失败,状态码: {response.status}")
|
||||||
|
|
||||||
|
# SSE
|
||||||
|
buffer = ""
|
||||||
|
event_type = None
|
||||||
|
event_data = None
|
||||||
|
|
||||||
|
async for chunk in response.content:
|
||||||
|
if chunk:
|
||||||
|
buffer += chunk.decode("utf-8", errors="ignore")
|
||||||
|
lines = buffer.split("\n")
|
||||||
|
buffer = lines[-1]
|
||||||
|
|
||||||
|
for line in lines[:-1]:
|
||||||
|
line = line.strip()
|
||||||
|
|
||||||
|
if not line:
|
||||||
|
if event_type and event_data:
|
||||||
|
yield {"event": event_type, "data": event_data}
|
||||||
|
event_type = None
|
||||||
|
event_data = None
|
||||||
|
elif line.startswith("event:"):
|
||||||
|
event_type = line[6:].strip()
|
||||||
|
elif line.startswith("data:"):
|
||||||
|
data_str = line[5:].strip()
|
||||||
|
if data_str and data_str != "[DONE]":
|
||||||
|
try:
|
||||||
|
event_data = json.loads(data_str)
|
||||||
|
except json.JSONDecodeError:
|
||||||
|
event_data = {"content": data_str}
|
||||||
|
|
||||||
|
except asyncio.TimeoutError:
|
||||||
|
raise Exception(f"Coze API 流式请求超时 ({timeout}秒)")
|
||||||
|
except Exception as e:
|
||||||
|
raise Exception(f"Coze API 流式请求失败: {str(e)}")
|
||||||
|
|
||||||
|
async def clear_context(self, conversation_id: str):
|
||||||
|
"""清空会话上下文
|
||||||
|
|
||||||
|
Args:
|
||||||
|
conversation_id: 会话ID
|
||||||
|
Returns:
|
||||||
|
dict: API响应结果
|
||||||
|
"""
|
||||||
|
session = await self._ensure_session()
|
||||||
|
url = f"{self.api_base}/v3/conversation/message/clear_context"
|
||||||
|
payload = {"conversation_id": conversation_id}
|
||||||
|
|
||||||
|
try:
|
||||||
|
async with session.post(url, json=payload) as response:
|
||||||
|
response_text = await response.text()
|
||||||
|
|
||||||
|
if response.status == 401:
|
||||||
|
raise Exception("Coze API 认证失败,请检查 API Key 是否正确")
|
||||||
|
|
||||||
|
if response.status != 200:
|
||||||
|
raise Exception(f"Coze API 请求失败,状态码: {response.status}")
|
||||||
|
|
||||||
|
try:
|
||||||
|
return json.loads(response_text)
|
||||||
|
except json.JSONDecodeError:
|
||||||
|
raise Exception("Coze API 返回非JSON格式")
|
||||||
|
|
||||||
|
except asyncio.TimeoutError:
|
||||||
|
raise Exception("Coze API 请求超时")
|
||||||
|
except aiohttp.ClientError as e:
|
||||||
|
raise Exception(f"Coze API 请求失败: {str(e)}")
|
||||||
|
|
||||||
|
async def get_message_list(
|
||||||
|
self,
|
||||||
|
conversation_id: str,
|
||||||
|
order: str = "desc",
|
||||||
|
limit: int = 10,
|
||||||
|
offset: int = 0,
|
||||||
|
):
|
||||||
|
"""获取消息列表
|
||||||
|
|
||||||
|
Args:
|
||||||
|
conversation_id: 会话ID
|
||||||
|
order: 排序方式 (asc/desc)
|
||||||
|
limit: 限制数量
|
||||||
|
offset: 偏移量
|
||||||
|
Returns:
|
||||||
|
dict: API响应结果
|
||||||
|
"""
|
||||||
|
session = await self._ensure_session()
|
||||||
|
url = f"{self.api_base}/v3/conversation/message/list"
|
||||||
|
params = {
|
||||||
|
"conversation_id": conversation_id,
|
||||||
|
"order": order,
|
||||||
|
"limit": limit,
|
||||||
|
"offset": offset,
|
||||||
|
}
|
||||||
|
|
||||||
|
try:
|
||||||
|
async with session.get(url, params=params) as response:
|
||||||
|
response.raise_for_status()
|
||||||
|
return await response.json()
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"获取Coze消息列表失败: {str(e)}")
|
||||||
|
raise Exception(f"获取Coze消息列表失败: {str(e)}")
|
||||||
|
|
||||||
|
async def close(self):
|
||||||
|
"""关闭会话"""
|
||||||
|
if self.session:
|
||||||
|
await self.session.close()
|
||||||
|
self.session = None
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
import os
|
||||||
|
import asyncio
|
||||||
|
|
||||||
|
async def test_coze_api_client():
|
||||||
|
api_key = os.getenv("COZE_API_KEY", "")
|
||||||
|
bot_id = os.getenv("COZE_BOT_ID", "")
|
||||||
|
client = CozeAPIClient(api_key=api_key)
|
||||||
|
|
||||||
|
try:
|
||||||
|
with open("README.md", "rb") as f:
|
||||||
|
file_data = f.read()
|
||||||
|
file_id = await client.upload_file(file_data)
|
||||||
|
print(f"Uploaded file_id: {file_id}")
|
||||||
|
async for event in client.chat_messages(
|
||||||
|
bot_id=bot_id,
|
||||||
|
user_id="test_user",
|
||||||
|
additional_messages=[
|
||||||
|
{
|
||||||
|
"role": "user",
|
||||||
|
"content": json.dumps(
|
||||||
|
[
|
||||||
|
{"type": "text", "text": "这是什么"},
|
||||||
|
{"type": "file", "file_id": file_id},
|
||||||
|
],
|
||||||
|
ensure_ascii=False,
|
||||||
|
),
|
||||||
|
"content_type": "object_string",
|
||||||
|
},
|
||||||
|
],
|
||||||
|
stream=True,
|
||||||
|
):
|
||||||
|
print(f"Event: {event}")
|
||||||
|
|
||||||
|
finally:
|
||||||
|
await client.close()
|
||||||
|
|
||||||
|
asyncio.run(test_coze_api_client())
|
||||||
635
astrbot/core/provider/sources/coze_source.py
Normal file
635
astrbot/core/provider/sources/coze_source.py
Normal file
@@ -0,0 +1,635 @@
|
|||||||
|
import json
|
||||||
|
import os
|
||||||
|
import base64
|
||||||
|
import hashlib
|
||||||
|
from typing import AsyncGenerator, Dict
|
||||||
|
from astrbot.core.message.message_event_result import MessageChain
|
||||||
|
import astrbot.core.message.components as Comp
|
||||||
|
from astrbot.api.provider import Provider
|
||||||
|
from astrbot import logger
|
||||||
|
from astrbot.core.provider.entities import LLMResponse
|
||||||
|
from ..register import register_provider_adapter
|
||||||
|
from .coze_api_client import CozeAPIClient
|
||||||
|
|
||||||
|
|
||||||
|
@register_provider_adapter("coze", "Coze (扣子) 智能体适配器")
|
||||||
|
class ProviderCoze(Provider):
|
||||||
|
def __init__(
|
||||||
|
self,
|
||||||
|
provider_config,
|
||||||
|
provider_settings,
|
||||||
|
default_persona=None,
|
||||||
|
) -> None:
|
||||||
|
super().__init__(
|
||||||
|
provider_config,
|
||||||
|
provider_settings,
|
||||||
|
default_persona,
|
||||||
|
)
|
||||||
|
self.api_key = provider_config.get("coze_api_key", "")
|
||||||
|
if not self.api_key:
|
||||||
|
raise Exception("Coze API Key 不能为空。")
|
||||||
|
self.bot_id = provider_config.get("bot_id", "")
|
||||||
|
if not self.bot_id:
|
||||||
|
raise Exception("Coze Bot ID 不能为空。")
|
||||||
|
self.api_base: str = provider_config.get("coze_api_base", "https://api.coze.cn")
|
||||||
|
|
||||||
|
if not isinstance(self.api_base, str) or not self.api_base.startswith(
|
||||||
|
("http://", "https://")
|
||||||
|
):
|
||||||
|
raise Exception(
|
||||||
|
"Coze API Base URL 格式不正确,必须以 http:// 或 https:// 开头。"
|
||||||
|
)
|
||||||
|
|
||||||
|
self.timeout = provider_config.get("timeout", 120)
|
||||||
|
if isinstance(self.timeout, str):
|
||||||
|
self.timeout = int(self.timeout)
|
||||||
|
self.auto_save_history = provider_config.get("auto_save_history", True)
|
||||||
|
self.conversation_ids: Dict[str, str] = {}
|
||||||
|
self.file_id_cache: Dict[str, Dict[str, str]] = {}
|
||||||
|
|
||||||
|
# 创建 API 客户端
|
||||||
|
self.api_client = CozeAPIClient(api_key=self.api_key, api_base=self.api_base)
|
||||||
|
|
||||||
|
def _generate_cache_key(self, data: str, is_base64: bool = False) -> str:
|
||||||
|
"""生成统一的缓存键
|
||||||
|
|
||||||
|
Args:
|
||||||
|
data: 图片数据或路径
|
||||||
|
is_base64: 是否是 base64 数据
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
str: 缓存键
|
||||||
|
"""
|
||||||
|
|
||||||
|
try:
|
||||||
|
if is_base64 and data.startswith("data:image/"):
|
||||||
|
try:
|
||||||
|
header, encoded = data.split(",", 1)
|
||||||
|
image_bytes = base64.b64decode(encoded)
|
||||||
|
cache_key = hashlib.md5(image_bytes).hexdigest()
|
||||||
|
return cache_key
|
||||||
|
except Exception:
|
||||||
|
cache_key = hashlib.md5(encoded.encode("utf-8")).hexdigest()
|
||||||
|
return cache_key
|
||||||
|
else:
|
||||||
|
if data.startswith(("http://", "https://")):
|
||||||
|
# URL图片,使用URL作为缓存键
|
||||||
|
cache_key = hashlib.md5(data.encode("utf-8")).hexdigest()
|
||||||
|
return cache_key
|
||||||
|
else:
|
||||||
|
clean_path = (
|
||||||
|
data.split("_")[0]
|
||||||
|
if "_" in data and len(data.split("_")) >= 3
|
||||||
|
else data
|
||||||
|
)
|
||||||
|
|
||||||
|
if os.path.exists(clean_path):
|
||||||
|
with open(clean_path, "rb") as f:
|
||||||
|
file_content = f.read()
|
||||||
|
cache_key = hashlib.md5(file_content).hexdigest()
|
||||||
|
return cache_key
|
||||||
|
else:
|
||||||
|
cache_key = hashlib.md5(clean_path.encode("utf-8")).hexdigest()
|
||||||
|
return cache_key
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
cache_key = hashlib.md5(data.encode("utf-8")).hexdigest()
|
||||||
|
logger.debug(f"[Coze] 异常文件缓存键: {cache_key}, error={e}")
|
||||||
|
return cache_key
|
||||||
|
|
||||||
|
async def _upload_file(
|
||||||
|
self,
|
||||||
|
file_data: bytes,
|
||||||
|
session_id: str | None = None,
|
||||||
|
cache_key: str | None = None,
|
||||||
|
) -> str:
|
||||||
|
"""上传文件到 Coze 并返回 file_id"""
|
||||||
|
# 使用 API 客户端上传文件
|
||||||
|
file_id = await self.api_client.upload_file(file_data)
|
||||||
|
|
||||||
|
# 缓存 file_id
|
||||||
|
if session_id and cache_key:
|
||||||
|
if session_id not in self.file_id_cache:
|
||||||
|
self.file_id_cache[session_id] = {}
|
||||||
|
self.file_id_cache[session_id][cache_key] = file_id
|
||||||
|
logger.debug(f"[Coze] 图片上传成功并缓存,file_id: {file_id}")
|
||||||
|
|
||||||
|
return file_id
|
||||||
|
|
||||||
|
async def _download_and_upload_image(
|
||||||
|
self, image_url: str, session_id: str | None = None
|
||||||
|
) -> str:
|
||||||
|
"""下载图片并上传到 Coze,返回 file_id"""
|
||||||
|
# 计算哈希实现缓存
|
||||||
|
cache_key = self._generate_cache_key(image_url) if session_id else None
|
||||||
|
|
||||||
|
if session_id and cache_key:
|
||||||
|
if session_id not in self.file_id_cache:
|
||||||
|
self.file_id_cache[session_id] = {}
|
||||||
|
|
||||||
|
if cache_key in self.file_id_cache[session_id]:
|
||||||
|
file_id = self.file_id_cache[session_id][cache_key]
|
||||||
|
return file_id
|
||||||
|
|
||||||
|
try:
|
||||||
|
image_data = await self.api_client.download_image(image_url)
|
||||||
|
|
||||||
|
file_id = await self._upload_file(image_data, session_id, cache_key)
|
||||||
|
|
||||||
|
if session_id and cache_key:
|
||||||
|
self.file_id_cache[session_id][cache_key] = file_id
|
||||||
|
|
||||||
|
return file_id
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"处理图片失败 {image_url}: {str(e)}")
|
||||||
|
raise Exception(f"处理图片失败: {str(e)}")
|
||||||
|
|
||||||
|
async def _process_context_images(
|
||||||
|
self, content: str | list, session_id: str
|
||||||
|
) -> str:
|
||||||
|
"""处理上下文中的图片内容,将 base64 图片上传并替换为 file_id"""
|
||||||
|
|
||||||
|
try:
|
||||||
|
if isinstance(content, str):
|
||||||
|
return content
|
||||||
|
|
||||||
|
processed_content = []
|
||||||
|
if session_id not in self.file_id_cache:
|
||||||
|
self.file_id_cache[session_id] = {}
|
||||||
|
|
||||||
|
for item in content:
|
||||||
|
if not isinstance(item, dict):
|
||||||
|
processed_content.append(item)
|
||||||
|
continue
|
||||||
|
if item.get("type") == "text":
|
||||||
|
processed_content.append(item)
|
||||||
|
elif item.get("type") == "image_url":
|
||||||
|
# 处理图片逻辑
|
||||||
|
if "file_id" in item:
|
||||||
|
# 已经有 file_id
|
||||||
|
logger.debug(f"[Coze] 图片已有file_id: {item['file_id']}")
|
||||||
|
processed_content.append(item)
|
||||||
|
else:
|
||||||
|
# 获取图片数据
|
||||||
|
image_data = ""
|
||||||
|
if "image_url" in item and isinstance(item["image_url"], dict):
|
||||||
|
image_data = item["image_url"].get("url", "")
|
||||||
|
elif "data" in item:
|
||||||
|
image_data = item.get("data", "")
|
||||||
|
elif "url" in item:
|
||||||
|
image_data = item.get("url", "")
|
||||||
|
|
||||||
|
if not image_data:
|
||||||
|
continue
|
||||||
|
# 计算哈希用于缓存
|
||||||
|
cache_key = self._generate_cache_key(
|
||||||
|
image_data, is_base64=image_data.startswith("data:image/")
|
||||||
|
)
|
||||||
|
|
||||||
|
# 检查缓存
|
||||||
|
if cache_key in self.file_id_cache[session_id]:
|
||||||
|
file_id = self.file_id_cache[session_id][cache_key]
|
||||||
|
processed_content.append(
|
||||||
|
{"type": "image", "file_id": file_id}
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
# 上传图片并缓存
|
||||||
|
if image_data.startswith("data:image/"):
|
||||||
|
# base64 处理
|
||||||
|
_, encoded = image_data.split(",", 1)
|
||||||
|
image_bytes = base64.b64decode(encoded)
|
||||||
|
file_id = await self._upload_file(
|
||||||
|
image_bytes,
|
||||||
|
session_id,
|
||||||
|
cache_key,
|
||||||
|
)
|
||||||
|
elif image_data.startswith(("http://", "https://")):
|
||||||
|
# URL 图片
|
||||||
|
file_id = await self._download_and_upload_image(
|
||||||
|
image_data, session_id
|
||||||
|
)
|
||||||
|
# 为URL图片也添加缓存
|
||||||
|
self.file_id_cache[session_id][cache_key] = file_id
|
||||||
|
elif os.path.exists(image_data):
|
||||||
|
# 本地文件
|
||||||
|
with open(image_data, "rb") as f:
|
||||||
|
image_bytes = f.read()
|
||||||
|
file_id = await self._upload_file(
|
||||||
|
image_bytes,
|
||||||
|
session_id,
|
||||||
|
cache_key,
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
logger.warning(
|
||||||
|
f"无法处理的图片格式: {image_data[:50]}..."
|
||||||
|
)
|
||||||
|
continue
|
||||||
|
|
||||||
|
processed_content.append(
|
||||||
|
{"type": "image", "file_id": file_id}
|
||||||
|
)
|
||||||
|
|
||||||
|
result = json.dumps(processed_content, ensure_ascii=False)
|
||||||
|
return result
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"处理上下文图片失败: {str(e)}")
|
||||||
|
if isinstance(content, str):
|
||||||
|
return content
|
||||||
|
else:
|
||||||
|
return json.dumps(content, ensure_ascii=False)
|
||||||
|
|
||||||
|
async def text_chat(
|
||||||
|
self,
|
||||||
|
prompt: str,
|
||||||
|
session_id=None,
|
||||||
|
image_urls=None,
|
||||||
|
func_tool=None,
|
||||||
|
contexts=None,
|
||||||
|
system_prompt=None,
|
||||||
|
tool_calls_result=None,
|
||||||
|
model=None,
|
||||||
|
**kwargs,
|
||||||
|
) -> LLMResponse:
|
||||||
|
"""文本对话, 内部使用流式接口实现非流式
|
||||||
|
|
||||||
|
Args:
|
||||||
|
prompt (str): 用户提示词
|
||||||
|
session_id (str): 会话ID
|
||||||
|
image_urls (List[str]): 图片URL列表
|
||||||
|
func_tool (FuncCall): 函数调用工具(不支持)
|
||||||
|
contexts (List): 上下文列表
|
||||||
|
system_prompt (str): 系统提示语
|
||||||
|
tool_calls_result (ToolCallsResult | List[ToolCallsResult]): 工具调用结果(不支持)
|
||||||
|
model (str): 模型名称(不支持)
|
||||||
|
Returns:
|
||||||
|
LLMResponse: LLM响应对象
|
||||||
|
"""
|
||||||
|
accumulated_content = ""
|
||||||
|
final_response = None
|
||||||
|
|
||||||
|
async for llm_response in self.text_chat_stream(
|
||||||
|
prompt=prompt,
|
||||||
|
session_id=session_id,
|
||||||
|
image_urls=image_urls,
|
||||||
|
func_tool=func_tool,
|
||||||
|
contexts=contexts,
|
||||||
|
system_prompt=system_prompt,
|
||||||
|
tool_calls_result=tool_calls_result,
|
||||||
|
model=model,
|
||||||
|
**kwargs,
|
||||||
|
):
|
||||||
|
if llm_response.is_chunk:
|
||||||
|
if llm_response.completion_text:
|
||||||
|
accumulated_content += llm_response.completion_text
|
||||||
|
else:
|
||||||
|
final_response = llm_response
|
||||||
|
|
||||||
|
if final_response:
|
||||||
|
return final_response
|
||||||
|
|
||||||
|
if accumulated_content:
|
||||||
|
chain = MessageChain(chain=[Comp.Plain(accumulated_content)])
|
||||||
|
return LLMResponse(role="assistant", result_chain=chain)
|
||||||
|
else:
|
||||||
|
return LLMResponse(role="assistant", completion_text="")
|
||||||
|
|
||||||
|
async def text_chat_stream(
|
||||||
|
self,
|
||||||
|
prompt: str,
|
||||||
|
session_id=None,
|
||||||
|
image_urls=None,
|
||||||
|
func_tool=None,
|
||||||
|
contexts=None,
|
||||||
|
system_prompt=None,
|
||||||
|
tool_calls_result=None,
|
||||||
|
model=None,
|
||||||
|
**kwargs,
|
||||||
|
) -> AsyncGenerator[LLMResponse, None]:
|
||||||
|
"""流式对话接口"""
|
||||||
|
# 用户ID参数(参考文档, 可以自定义)
|
||||||
|
user_id = session_id or kwargs.get("user", "default_user")
|
||||||
|
|
||||||
|
# 获取或创建会话ID
|
||||||
|
conversation_id = self.conversation_ids.get(user_id)
|
||||||
|
|
||||||
|
# 构建消息
|
||||||
|
additional_messages = []
|
||||||
|
|
||||||
|
if system_prompt:
|
||||||
|
if not self.auto_save_history or not conversation_id:
|
||||||
|
additional_messages.append(
|
||||||
|
{"role": "system", "content": system_prompt, "content_type": "text"}
|
||||||
|
)
|
||||||
|
|
||||||
|
if not self.auto_save_history and contexts:
|
||||||
|
# 如果关闭了自动保存历史,传入上下文
|
||||||
|
for ctx in contexts:
|
||||||
|
if isinstance(ctx, dict) and "role" in ctx and "content" in ctx:
|
||||||
|
content = ctx["content"]
|
||||||
|
content_type = ctx.get("content_type", "text")
|
||||||
|
|
||||||
|
# 处理可能包含图片的上下文
|
||||||
|
if (
|
||||||
|
content_type == "object_string"
|
||||||
|
or (isinstance(content, str) and content.startswith("["))
|
||||||
|
or (
|
||||||
|
isinstance(content, list)
|
||||||
|
and any(
|
||||||
|
isinstance(item, dict)
|
||||||
|
and item.get("type") == "image_url"
|
||||||
|
for item in content
|
||||||
|
)
|
||||||
|
)
|
||||||
|
):
|
||||||
|
processed_content = await self._process_context_images(
|
||||||
|
content, user_id
|
||||||
|
)
|
||||||
|
additional_messages.append(
|
||||||
|
{
|
||||||
|
"role": ctx["role"],
|
||||||
|
"content": processed_content,
|
||||||
|
"content_type": "object_string",
|
||||||
|
}
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
# 纯文本
|
||||||
|
additional_messages.append(
|
||||||
|
{
|
||||||
|
"role": ctx["role"],
|
||||||
|
"content": (
|
||||||
|
content
|
||||||
|
if isinstance(content, str)
|
||||||
|
else json.dumps(content, ensure_ascii=False)
|
||||||
|
),
|
||||||
|
"content_type": "text",
|
||||||
|
}
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
logger.info(f"[Coze] 跳过格式不正确的上下文: {ctx}")
|
||||||
|
|
||||||
|
if prompt or image_urls:
|
||||||
|
if image_urls:
|
||||||
|
# 多模态
|
||||||
|
object_string_content = []
|
||||||
|
if prompt:
|
||||||
|
object_string_content.append({"type": "text", "text": prompt})
|
||||||
|
|
||||||
|
for url in image_urls:
|
||||||
|
try:
|
||||||
|
if url.startswith(("http://", "https://")):
|
||||||
|
# 网络图片
|
||||||
|
file_id = await self._download_and_upload_image(
|
||||||
|
url, user_id
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
# 本地文件或 base64
|
||||||
|
if url.startswith("data:image/"):
|
||||||
|
# base64
|
||||||
|
_, encoded = url.split(",", 1)
|
||||||
|
image_data = base64.b64decode(encoded)
|
||||||
|
cache_key = self._generate_cache_key(
|
||||||
|
url, is_base64=True
|
||||||
|
)
|
||||||
|
file_id = await self._upload_file(
|
||||||
|
image_data, user_id, cache_key
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
# 本地文件
|
||||||
|
if os.path.exists(url):
|
||||||
|
with open(url, "rb") as f:
|
||||||
|
image_data = f.read()
|
||||||
|
# 用文件路径和修改时间来缓存
|
||||||
|
file_stat = os.stat(url)
|
||||||
|
cache_key = self._generate_cache_key(
|
||||||
|
f"{url}_{file_stat.st_mtime}_{file_stat.st_size}",
|
||||||
|
is_base64=False,
|
||||||
|
)
|
||||||
|
file_id = await self._upload_file(
|
||||||
|
image_data, user_id, cache_key
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
logger.warning(f"图片文件不存在: {url}")
|
||||||
|
continue
|
||||||
|
|
||||||
|
object_string_content.append(
|
||||||
|
{
|
||||||
|
"type": "image",
|
||||||
|
"file_id": file_id,
|
||||||
|
}
|
||||||
|
)
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"处理图片失败 {url}: {str(e)}")
|
||||||
|
continue
|
||||||
|
|
||||||
|
if object_string_content:
|
||||||
|
content = json.dumps(object_string_content, ensure_ascii=False)
|
||||||
|
additional_messages.append(
|
||||||
|
{
|
||||||
|
"role": "user",
|
||||||
|
"content": content,
|
||||||
|
"content_type": "object_string",
|
||||||
|
}
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
# 纯文本
|
||||||
|
if prompt:
|
||||||
|
additional_messages.append(
|
||||||
|
{
|
||||||
|
"role": "user",
|
||||||
|
"content": prompt,
|
||||||
|
"content_type": "text",
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
try:
|
||||||
|
accumulated_content = ""
|
||||||
|
message_started = False
|
||||||
|
|
||||||
|
async for chunk in self.api_client.chat_messages(
|
||||||
|
bot_id=self.bot_id,
|
||||||
|
user_id=user_id,
|
||||||
|
additional_messages=additional_messages,
|
||||||
|
conversation_id=conversation_id,
|
||||||
|
auto_save_history=self.auto_save_history,
|
||||||
|
stream=True,
|
||||||
|
timeout=self.timeout,
|
||||||
|
):
|
||||||
|
event_type = chunk.get("event")
|
||||||
|
data = chunk.get("data", {})
|
||||||
|
|
||||||
|
if event_type == "conversation.chat.created":
|
||||||
|
if isinstance(data, dict) and "conversation_id" in data:
|
||||||
|
self.conversation_ids[user_id] = data["conversation_id"]
|
||||||
|
|
||||||
|
elif event_type == "conversation.message.delta":
|
||||||
|
if isinstance(data, dict):
|
||||||
|
content = data.get("content", "")
|
||||||
|
if not content and "delta" in data:
|
||||||
|
content = data["delta"].get("content", "")
|
||||||
|
if not content and "text" in data:
|
||||||
|
content = data.get("text", "")
|
||||||
|
|
||||||
|
if content:
|
||||||
|
message_started = True
|
||||||
|
accumulated_content += content
|
||||||
|
yield LLMResponse(
|
||||||
|
role="assistant",
|
||||||
|
completion_text=content,
|
||||||
|
is_chunk=True,
|
||||||
|
)
|
||||||
|
|
||||||
|
elif event_type == "conversation.message.completed":
|
||||||
|
if isinstance(data, dict):
|
||||||
|
msg_type = data.get("type")
|
||||||
|
if msg_type == "answer" and data.get("role") == "assistant":
|
||||||
|
final_content = data.get("content", "")
|
||||||
|
if not accumulated_content and final_content:
|
||||||
|
chain = MessageChain(chain=[Comp.Plain(final_content)])
|
||||||
|
yield LLMResponse(
|
||||||
|
role="assistant",
|
||||||
|
result_chain=chain,
|
||||||
|
is_chunk=False,
|
||||||
|
)
|
||||||
|
|
||||||
|
elif event_type == "conversation.chat.completed":
|
||||||
|
if accumulated_content:
|
||||||
|
chain = MessageChain(chain=[Comp.Plain(accumulated_content)])
|
||||||
|
yield LLMResponse(
|
||||||
|
role="assistant",
|
||||||
|
result_chain=chain,
|
||||||
|
is_chunk=False,
|
||||||
|
)
|
||||||
|
break
|
||||||
|
|
||||||
|
elif event_type == "done":
|
||||||
|
break
|
||||||
|
|
||||||
|
elif event_type == "error":
|
||||||
|
error_msg = (
|
||||||
|
data.get("message", "未知错误")
|
||||||
|
if isinstance(data, dict)
|
||||||
|
else str(data)
|
||||||
|
)
|
||||||
|
logger.error(f"Coze 流式响应错误: {error_msg}")
|
||||||
|
yield LLMResponse(
|
||||||
|
role="err",
|
||||||
|
completion_text=f"Coze 错误: {error_msg}",
|
||||||
|
is_chunk=False,
|
||||||
|
)
|
||||||
|
break
|
||||||
|
|
||||||
|
if not message_started and not accumulated_content:
|
||||||
|
yield LLMResponse(
|
||||||
|
role="assistant",
|
||||||
|
completion_text="LLM 未响应任何内容。",
|
||||||
|
is_chunk=False,
|
||||||
|
)
|
||||||
|
elif message_started and accumulated_content:
|
||||||
|
chain = MessageChain(chain=[Comp.Plain(accumulated_content)])
|
||||||
|
yield LLMResponse(
|
||||||
|
role="assistant",
|
||||||
|
result_chain=chain,
|
||||||
|
is_chunk=False,
|
||||||
|
)
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"Coze 流式请求失败: {str(e)}")
|
||||||
|
yield LLMResponse(
|
||||||
|
role="err",
|
||||||
|
completion_text=f"Coze 流式请求失败: {str(e)}",
|
||||||
|
is_chunk=False,
|
||||||
|
)
|
||||||
|
|
||||||
|
async def forget(self, session_id: str):
|
||||||
|
"""清空指定会话的上下文"""
|
||||||
|
user_id = session_id
|
||||||
|
conversation_id = self.conversation_ids.get(user_id)
|
||||||
|
|
||||||
|
if user_id in self.file_id_cache:
|
||||||
|
self.file_id_cache.pop(user_id, None)
|
||||||
|
|
||||||
|
if not conversation_id:
|
||||||
|
return True
|
||||||
|
|
||||||
|
try:
|
||||||
|
response = await self.api_client.clear_context(conversation_id)
|
||||||
|
|
||||||
|
if "code" in response and response["code"] == 0:
|
||||||
|
self.conversation_ids.pop(user_id, None)
|
||||||
|
return True
|
||||||
|
else:
|
||||||
|
logger.warning(f"清空 Coze 会话上下文失败: {response}")
|
||||||
|
return False
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"清空 Coze 会话失败: {str(e)}")
|
||||||
|
return False
|
||||||
|
|
||||||
|
async def get_current_key(self):
|
||||||
|
"""获取当前API Key"""
|
||||||
|
return self.api_key
|
||||||
|
|
||||||
|
async def set_key(self, key: str):
|
||||||
|
"""设置新的API Key"""
|
||||||
|
raise NotImplementedError("Coze 适配器不支持设置 API Key。")
|
||||||
|
|
||||||
|
async def get_models(self):
|
||||||
|
"""获取可用模型列表"""
|
||||||
|
return [f"bot_{self.bot_id}"]
|
||||||
|
|
||||||
|
def get_model(self):
|
||||||
|
"""获取当前模型"""
|
||||||
|
return f"bot_{self.bot_id}"
|
||||||
|
|
||||||
|
def set_model(self, model: str):
|
||||||
|
"""设置模型(在Coze中是Bot ID)"""
|
||||||
|
if model.startswith("bot_"):
|
||||||
|
self.bot_id = model[4:]
|
||||||
|
else:
|
||||||
|
self.bot_id = model
|
||||||
|
|
||||||
|
async def get_human_readable_context(
|
||||||
|
self, session_id: str, page: int = 1, page_size: int = 10
|
||||||
|
):
|
||||||
|
"""获取人类可读的上下文历史"""
|
||||||
|
user_id = session_id
|
||||||
|
conversation_id = self.conversation_ids.get(user_id)
|
||||||
|
|
||||||
|
if not conversation_id:
|
||||||
|
return []
|
||||||
|
|
||||||
|
try:
|
||||||
|
data = await self.api_client.get_message_list(
|
||||||
|
conversation_id=conversation_id,
|
||||||
|
order="desc",
|
||||||
|
limit=page_size,
|
||||||
|
offset=(page - 1) * page_size,
|
||||||
|
)
|
||||||
|
|
||||||
|
if data.get("code") != 0:
|
||||||
|
logger.warning(f"获取 Coze 消息历史失败: {data}")
|
||||||
|
return []
|
||||||
|
|
||||||
|
messages = data.get("data", {}).get("messages", [])
|
||||||
|
|
||||||
|
readable_history = []
|
||||||
|
for msg in messages:
|
||||||
|
role = msg.get("role", "unknown")
|
||||||
|
content = msg.get("content", "")
|
||||||
|
msg_type = msg.get("type", "")
|
||||||
|
|
||||||
|
if role == "user":
|
||||||
|
readable_history.append(f"用户: {content}")
|
||||||
|
elif role == "assistant" and msg_type == "answer":
|
||||||
|
readable_history.append(f"助手: {content}")
|
||||||
|
|
||||||
|
return readable_history
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"获取 Coze 消息历史失败: {str(e)}")
|
||||||
|
return []
|
||||||
|
|
||||||
|
async def terminate(self):
|
||||||
|
"""清理资源"""
|
||||||
|
await self.api_client.close()
|
||||||
@@ -98,7 +98,7 @@ class ProviderFishAudioTTSAPI(TTSProvider):
|
|||||||
|
|
||||||
# FishAudio的reference_id通常是32位十六进制字符串
|
# FishAudio的reference_id通常是32位十六进制字符串
|
||||||
# 例如: 626bb6d3f3364c9cbc3aa6a67300a664
|
# 例如: 626bb6d3f3364c9cbc3aa6a67300a664
|
||||||
pattern = r'^[a-fA-F0-9]{32}$'
|
pattern = r"^[a-fA-F0-9]{32}$"
|
||||||
return bool(re.match(pattern, reference_id.strip()))
|
return bool(re.match(pattern, reference_id.strip()))
|
||||||
|
|
||||||
async def _generate_request(self, text: str) -> dict:
|
async def _generate_request(self, text: str) -> dict:
|
||||||
|
|||||||
@@ -99,12 +99,15 @@ class ProviderOpenAIOfficial(Provider):
|
|||||||
for key in to_del:
|
for key in to_del:
|
||||||
del payloads[key]
|
del payloads[key]
|
||||||
|
|
||||||
model = payloads.get("model", "")
|
# 读取并合并 custom_extra_body 配置
|
||||||
# 针对 qwen3 非 thinking 模型的特殊处理:非流式调用必须设置 enable_thinking=false
|
custom_extra_body = self.provider_config.get("custom_extra_body", {})
|
||||||
if "qwen3" in model.lower() and "thinking" not in model.lower():
|
if isinstance(custom_extra_body, dict):
|
||||||
extra_body["enable_thinking"] = False
|
extra_body.update(custom_extra_body)
|
||||||
|
|
||||||
|
model = payloads.get("model", "").lower()
|
||||||
|
|
||||||
# 针对 deepseek 模型的特殊处理:deepseek-reasoner调用必须移除 tools ,否则将被切换至 deepseek-chat
|
# 针对 deepseek 模型的特殊处理:deepseek-reasoner调用必须移除 tools ,否则将被切换至 deepseek-chat
|
||||||
elif model == "deepseek-reasoner" and "tools" in payloads:
|
if model == "deepseek-reasoner" and "tools" in payloads:
|
||||||
del payloads["tools"]
|
del payloads["tools"]
|
||||||
|
|
||||||
completion = await self.client.chat.completions.create(
|
completion = await self.client.chat.completions.create(
|
||||||
@@ -137,6 +140,12 @@ class ProviderOpenAIOfficial(Provider):
|
|||||||
|
|
||||||
# 不在默认参数中的参数放在 extra_body 中
|
# 不在默认参数中的参数放在 extra_body 中
|
||||||
extra_body = {}
|
extra_body = {}
|
||||||
|
|
||||||
|
# 读取并合并 custom_extra_body 配置
|
||||||
|
custom_extra_body = self.provider_config.get("custom_extra_body", {})
|
||||||
|
if isinstance(custom_extra_body, dict):
|
||||||
|
extra_body.update(custom_extra_body)
|
||||||
|
|
||||||
to_del = []
|
to_del = []
|
||||||
for key in payloads.keys():
|
for key in payloads.keys():
|
||||||
if key not in self.default_params:
|
if key not in self.default_params:
|
||||||
|
|||||||
@@ -1,4 +1,5 @@
|
|||||||
import aiohttp
|
import aiohttp
|
||||||
|
from astrbot import logger
|
||||||
from ..provider import RerankProvider
|
from ..provider import RerankProvider
|
||||||
from ..register import register_provider_adapter
|
from ..register import register_provider_adapter
|
||||||
from ..entities import ProviderType, RerankResult
|
from ..entities import ProviderType, RerankResult
|
||||||
@@ -44,6 +45,11 @@ class VLLMRerankProvider(RerankProvider):
|
|||||||
response_data = await response.json()
|
response_data = await response.json()
|
||||||
results = response_data.get("results", [])
|
results = response_data.get("results", [])
|
||||||
|
|
||||||
|
if not results:
|
||||||
|
logger.warning(
|
||||||
|
f"Rerank API 返回了空的列表数据。原始响应: {response_data}"
|
||||||
|
)
|
||||||
|
|
||||||
return [
|
return [
|
||||||
RerankResult(
|
RerankResult(
|
||||||
index=result["index"],
|
index=result["index"],
|
||||||
|
|||||||
@@ -1,12 +1,12 @@
|
|||||||
from astrbot import logger
|
# This file was originally created to adapt to glm-4v-flash, which only supports one image in the context.
|
||||||
from astrbot.core.provider.func_tool_manager import FuncCall
|
# It is no longer specifically adapted to Zhipu's models. To ensure compatibility, this
|
||||||
from typing import List
|
|
||||||
|
|
||||||
from ..register import register_provider_adapter
|
from ..register import register_provider_adapter
|
||||||
from astrbot.core.provider.entities import LLMResponse
|
|
||||||
from .openai_source import ProviderOpenAIOfficial
|
from .openai_source import ProviderOpenAIOfficial
|
||||||
|
|
||||||
|
|
||||||
@register_provider_adapter("zhipu_chat_completion", "智浦 Chat Completion 提供商适配器")
|
@register_provider_adapter("zhipu_chat_completion", "智谱 Chat Completion 提供商适配器")
|
||||||
class ProviderZhipu(ProviderOpenAIOfficial):
|
class ProviderZhipu(ProviderOpenAIOfficial):
|
||||||
def __init__(
|
def __init__(
|
||||||
self,
|
self,
|
||||||
@@ -19,63 +19,3 @@ class ProviderZhipu(ProviderOpenAIOfficial):
|
|||||||
provider_settings,
|
provider_settings,
|
||||||
default_persona,
|
default_persona,
|
||||||
)
|
)
|
||||||
|
|
||||||
async def text_chat(
|
|
||||||
self,
|
|
||||||
prompt: str,
|
|
||||||
session_id: str = None,
|
|
||||||
image_urls: List[str] = None,
|
|
||||||
func_tool: FuncCall = None,
|
|
||||||
contexts=None,
|
|
||||||
system_prompt=None,
|
|
||||||
model=None,
|
|
||||||
**kwargs,
|
|
||||||
) -> LLMResponse:
|
|
||||||
if contexts is None:
|
|
||||||
contexts = []
|
|
||||||
new_record = await self.assemble_context(prompt, image_urls)
|
|
||||||
context_query = []
|
|
||||||
|
|
||||||
context_query = [*contexts, new_record]
|
|
||||||
|
|
||||||
model_cfgs: dict = self.provider_config.get("model_config", {})
|
|
||||||
model = model or self.get_model()
|
|
||||||
# glm-4v-flash 只支持一张图片
|
|
||||||
if model.lower() == "glm-4v-flash" and image_urls and len(context_query) > 1:
|
|
||||||
logger.debug("glm-4v-flash 只支持一张图片,将只保留最后一张图片")
|
|
||||||
logger.debug(context_query)
|
|
||||||
new_context_query_ = []
|
|
||||||
for i in range(0, len(context_query) - 1, 2):
|
|
||||||
if isinstance(context_query[i].get("content", ""), list):
|
|
||||||
continue
|
|
||||||
new_context_query_.append(context_query[i])
|
|
||||||
new_context_query_.append(context_query[i + 1])
|
|
||||||
new_context_query_.append(context_query[-1]) # 保留最后一条记录
|
|
||||||
context_query = new_context_query_
|
|
||||||
logger.debug(context_query)
|
|
||||||
|
|
||||||
if system_prompt:
|
|
||||||
context_query.insert(0, {"role": "system", "content": system_prompt})
|
|
||||||
|
|
||||||
payloads = {"messages": context_query, **model_cfgs}
|
|
||||||
try:
|
|
||||||
llm_response = await self._query(payloads, func_tool)
|
|
||||||
return llm_response
|
|
||||||
except Exception as e:
|
|
||||||
if "maximum context length" in str(e):
|
|
||||||
retry_cnt = 10
|
|
||||||
while retry_cnt > 0:
|
|
||||||
logger.warning(
|
|
||||||
f"请求失败:{e}。上下文长度超过限制。尝试弹出最早的记录然后重试。"
|
|
||||||
)
|
|
||||||
try:
|
|
||||||
self.pop_record(session_id)
|
|
||||||
llm_response = await self._query(payloads, func_tool)
|
|
||||||
break
|
|
||||||
except Exception as e:
|
|
||||||
if "maximum context length" in str(e):
|
|
||||||
retry_cnt -= 1
|
|
||||||
else:
|
|
||||||
raise e
|
|
||||||
else:
|
|
||||||
raise e
|
|
||||||
|
|||||||
@@ -6,6 +6,7 @@ from astrbot.core.provider.provider import (
|
|||||||
TTSProvider,
|
TTSProvider,
|
||||||
STTProvider,
|
STTProvider,
|
||||||
EmbeddingProvider,
|
EmbeddingProvider,
|
||||||
|
RerankProvider,
|
||||||
)
|
)
|
||||||
from astrbot.core.provider.entities import ProviderType
|
from astrbot.core.provider.entities import ProviderType
|
||||||
from astrbot.core.db import BaseDatabase
|
from astrbot.core.db import BaseDatabase
|
||||||
@@ -23,7 +24,7 @@ from .star import star_registry, StarMetadata, star_map
|
|||||||
from .star_handler import star_handlers_registry, StarHandlerMetadata, EventType
|
from .star_handler import star_handlers_registry, StarHandlerMetadata, EventType
|
||||||
from .filter.command import CommandFilter
|
from .filter.command import CommandFilter
|
||||||
from .filter.regex import RegexFilter
|
from .filter.regex import RegexFilter
|
||||||
from typing import Awaitable
|
from typing import Awaitable, Any, Callable
|
||||||
from astrbot.core.conversation_mgr import ConversationManager
|
from astrbot.core.conversation_mgr import ConversationManager
|
||||||
from astrbot.core.star.filter.platform_adapter_type import (
|
from astrbot.core.star.filter.platform_adapter_type import (
|
||||||
PlatformAdapterType,
|
PlatformAdapterType,
|
||||||
@@ -103,9 +104,14 @@ class Context:
|
|||||||
"""
|
"""
|
||||||
self.provider_manager.provider_insts.append(provider)
|
self.provider_manager.provider_insts.append(provider)
|
||||||
|
|
||||||
def get_provider_by_id(self, provider_id: str) -> Provider | None:
|
def get_provider_by_id(
|
||||||
"""通过 ID 获取对应的 LLM Provider(Chat_Completion 类型)。"""
|
self, provider_id: str
|
||||||
return self.provider_manager.inst_map.get(provider_id)
|
) -> (
|
||||||
|
Provider | TTSProvider | STTProvider | EmbeddingProvider | RerankProvider | None
|
||||||
|
):
|
||||||
|
"""通过 ID 获取对应的 LLM Provider。"""
|
||||||
|
prov = self.provider_manager.inst_map.get(provider_id)
|
||||||
|
return prov
|
||||||
|
|
||||||
def get_all_providers(self) -> List[Provider]:
|
def get_all_providers(self) -> List[Provider]:
|
||||||
"""获取所有用于文本生成任务的 LLM Provider(Chat_Completion 类型)。"""
|
"""获取所有用于文本生成任务的 LLM Provider(Chat_Completion 类型)。"""
|
||||||
@@ -130,34 +136,43 @@ class Context:
|
|||||||
Args:
|
Args:
|
||||||
umo(str): unified_message_origin 值,如果传入并且用户启用了提供商会话隔离,则使用该会话偏好的提供商。
|
umo(str): unified_message_origin 值,如果传入并且用户启用了提供商会话隔离,则使用该会话偏好的提供商。
|
||||||
"""
|
"""
|
||||||
return self.provider_manager.get_using_provider(
|
prov = self.provider_manager.get_using_provider(
|
||||||
provider_type=ProviderType.CHAT_COMPLETION,
|
provider_type=ProviderType.CHAT_COMPLETION,
|
||||||
umo=umo,
|
umo=umo,
|
||||||
)
|
)
|
||||||
|
if prov and not isinstance(prov, Provider):
|
||||||
|
raise ValueError("返回的 Provider 不是 Provider 类型")
|
||||||
|
return prov
|
||||||
|
|
||||||
def get_using_tts_provider(self, umo: str | None = None) -> TTSProvider:
|
def get_using_tts_provider(self, umo: str | None = None) -> TTSProvider | None:
|
||||||
"""
|
"""
|
||||||
获取当前使用的用于 TTS 任务的 Provider。
|
获取当前使用的用于 TTS 任务的 Provider。
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
umo(str): unified_message_origin 值,如果传入,则使用该会话偏好的提供商。
|
umo(str): unified_message_origin 值,如果传入,则使用该会话偏好的提供商。
|
||||||
"""
|
"""
|
||||||
return self.provider_manager.get_using_provider(
|
prov = self.provider_manager.get_using_provider(
|
||||||
provider_type=ProviderType.TEXT_TO_SPEECH,
|
provider_type=ProviderType.TEXT_TO_SPEECH,
|
||||||
umo=umo,
|
umo=umo,
|
||||||
)
|
)
|
||||||
|
if prov and not isinstance(prov, TTSProvider):
|
||||||
|
raise ValueError("返回的 Provider 不是 TTSProvider 类型")
|
||||||
|
return prov
|
||||||
|
|
||||||
def get_using_stt_provider(self, umo: str | None = None) -> STTProvider:
|
def get_using_stt_provider(self, umo: str | None = None) -> STTProvider | None:
|
||||||
"""
|
"""
|
||||||
获取当前使用的用于 STT 任务的 Provider。
|
获取当前使用的用于 STT 任务的 Provider。
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
umo(str): unified_message_origin 值,如果传入,则使用该会话偏好的提供商。
|
umo(str): unified_message_origin 值,如果传入,则使用该会话偏好的提供商。
|
||||||
"""
|
"""
|
||||||
return self.provider_manager.get_using_provider(
|
prov = self.provider_manager.get_using_provider(
|
||||||
provider_type=ProviderType.SPEECH_TO_TEXT,
|
provider_type=ProviderType.SPEECH_TO_TEXT,
|
||||||
umo=umo,
|
umo=umo,
|
||||||
)
|
)
|
||||||
|
if prov and not isinstance(prov, STTProvider):
|
||||||
|
raise ValueError("返回的 Provider 不是 STTProvider 类型")
|
||||||
|
return prov
|
||||||
|
|
||||||
def get_config(self, umo: str | None = None) -> AstrBotConfig:
|
def get_config(self, umo: str | None = None) -> AstrBotConfig:
|
||||||
"""获取 AstrBot 的配置。"""
|
"""获取 AstrBot 的配置。"""
|
||||||
@@ -245,7 +260,11 @@ class Context:
|
|||||||
"""
|
"""
|
||||||
|
|
||||||
def register_llm_tool(
|
def register_llm_tool(
|
||||||
self, name: str, func_args: list, desc: str, func_obj: Awaitable
|
self,
|
||||||
|
name: str,
|
||||||
|
func_args: list,
|
||||||
|
desc: str,
|
||||||
|
func_obj: Callable[..., Awaitable[Any]],
|
||||||
) -> None:
|
) -> None:
|
||||||
"""
|
"""
|
||||||
为函数调用(function-calling / tools-use)添加工具。
|
为函数调用(function-calling / tools-use)添加工具。
|
||||||
@@ -267,9 +286,7 @@ class Context:
|
|||||||
desc=desc,
|
desc=desc,
|
||||||
)
|
)
|
||||||
star_handlers_registry.append(md)
|
star_handlers_registry.append(md)
|
||||||
self.provider_manager.llm_tools.add_func(
|
self.provider_manager.llm_tools.add_func(name, func_args, desc, func_obj)
|
||||||
name, func_args, desc, func_obj, func_obj
|
|
||||||
)
|
|
||||||
|
|
||||||
def unregister_llm_tool(self, name: str) -> None:
|
def unregister_llm_tool(self, name: str) -> None:
|
||||||
"""删除一个函数调用工具。如果再要启用,需要重新注册。"""
|
"""删除一个函数调用工具。如果再要启用,需要重新注册。"""
|
||||||
@@ -281,7 +298,7 @@ class Context:
|
|||||||
command_name: str,
|
command_name: str,
|
||||||
desc: str,
|
desc: str,
|
||||||
priority: int,
|
priority: int,
|
||||||
awaitable: Awaitable,
|
awaitable: Callable[..., Awaitable[Any]],
|
||||||
use_regex=False,
|
use_regex=False,
|
||||||
ignore_prefix=False,
|
ignore_prefix=False,
|
||||||
):
|
):
|
||||||
|
|||||||
@@ -1,5 +1,7 @@
|
|||||||
import re
|
import re
|
||||||
import inspect
|
import inspect
|
||||||
|
import types
|
||||||
|
import typing
|
||||||
from typing import List, Any, Type, Dict
|
from typing import List, Any, Type, Dict
|
||||||
from . import HandlerFilter
|
from . import HandlerFilter
|
||||||
from astrbot.core.platform.astr_message_event import AstrMessageEvent
|
from astrbot.core.platform.astr_message_event import AstrMessageEvent
|
||||||
@@ -14,6 +16,18 @@ class GreedyStr(str):
|
|||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
def unwrap_optional(annotation) -> tuple:
|
||||||
|
"""去掉 Optional[T] / Union[T, None] / T|None,返回 T"""
|
||||||
|
args = typing.get_args(annotation)
|
||||||
|
non_none_args = [a for a in args if a is not type(None)]
|
||||||
|
if len(non_none_args) == 1:
|
||||||
|
return (non_none_args[0],)
|
||||||
|
elif len(non_none_args) > 1:
|
||||||
|
return tuple(non_none_args)
|
||||||
|
else:
|
||||||
|
return ()
|
||||||
|
|
||||||
|
|
||||||
# 标准指令受到 wake_prefix 的制约。
|
# 标准指令受到 wake_prefix 的制约。
|
||||||
class CommandFilter(HandlerFilter):
|
class CommandFilter(HandlerFilter):
|
||||||
"""标准指令过滤器"""
|
"""标准指令过滤器"""
|
||||||
@@ -32,11 +46,16 @@ class CommandFilter(HandlerFilter):
|
|||||||
self.init_handler_md(handler_md)
|
self.init_handler_md(handler_md)
|
||||||
self.custom_filter_list: List[CustomFilter] = []
|
self.custom_filter_list: List[CustomFilter] = []
|
||||||
|
|
||||||
|
# Cache for complete command names list
|
||||||
|
self._cmpl_cmd_names: list | None = None
|
||||||
|
|
||||||
def print_types(self):
|
def print_types(self):
|
||||||
result = ""
|
result = ""
|
||||||
for k, v in self.handler_params.items():
|
for k, v in self.handler_params.items():
|
||||||
if isinstance(v, type):
|
if isinstance(v, type):
|
||||||
result += f"{k}({v.__name__}),"
|
result += f"{k}({v.__name__}),"
|
||||||
|
elif isinstance(v, types.UnionType) or typing.get_origin(v) is typing.Union:
|
||||||
|
result += f"{k}({v}),"
|
||||||
else:
|
else:
|
||||||
result += f"{k}({type(v).__name__})={v},"
|
result += f"{k}({type(v).__name__})={v},"
|
||||||
result = result.rstrip(",")
|
result = result.rstrip(",")
|
||||||
@@ -92,7 +111,8 @@ class CommandFilter(HandlerFilter):
|
|||||||
# 没有 GreedyStr 的情况
|
# 没有 GreedyStr 的情况
|
||||||
if i >= len(params):
|
if i >= len(params):
|
||||||
if (
|
if (
|
||||||
isinstance(param_type_or_default_val, Type)
|
isinstance(param_type_or_default_val, (Type, types.UnionType))
|
||||||
|
or typing.get_origin(param_type_or_default_val) is typing.Union
|
||||||
or param_type_or_default_val is inspect.Parameter.empty
|
or param_type_or_default_val is inspect.Parameter.empty
|
||||||
):
|
):
|
||||||
# 是类型
|
# 是类型
|
||||||
@@ -128,6 +148,19 @@ class CommandFilter(HandlerFilter):
|
|||||||
result[param_name] = int(params[i])
|
result[param_name] = int(params[i])
|
||||||
elif isinstance(param_type_or_default_val, float):
|
elif isinstance(param_type_or_default_val, float):
|
||||||
result[param_name] = float(params[i])
|
result[param_name] = float(params[i])
|
||||||
|
else:
|
||||||
|
origin = typing.get_origin(param_type_or_default_val)
|
||||||
|
if origin in (typing.Union, types.UnionType):
|
||||||
|
# 注解是联合类型
|
||||||
|
# NOTE: 目前没有处理联合类型嵌套相关的注解写法
|
||||||
|
nn_types = unwrap_optional(param_type_or_default_val)
|
||||||
|
if len(nn_types) == 1:
|
||||||
|
# 只有一个非 NoneType 类型
|
||||||
|
result[param_name] = nn_types[0](params[i])
|
||||||
|
else:
|
||||||
|
# 没有或者有多个非 NoneType 类型,这里我们暂时直接赋值为原始值。
|
||||||
|
# NOTE: 目前还没有做类型校验
|
||||||
|
result[param_name] = params[i]
|
||||||
else:
|
else:
|
||||||
result[param_name] = param_type_or_default_val(params[i])
|
result[param_name] = param_type_or_default_val(params[i])
|
||||||
except ValueError:
|
except ValueError:
|
||||||
@@ -136,6 +169,22 @@ class CommandFilter(HandlerFilter):
|
|||||||
)
|
)
|
||||||
return result
|
return result
|
||||||
|
|
||||||
|
def get_complete_command_names(self):
|
||||||
|
if self._cmpl_cmd_names is not None:
|
||||||
|
return self._cmpl_cmd_names
|
||||||
|
self._cmpl_cmd_names = [
|
||||||
|
f"{parent} {cmd}" if parent else cmd
|
||||||
|
for cmd in [self.command_name] + list(self.alias)
|
||||||
|
for parent in self.parent_command_names or [""]
|
||||||
|
]
|
||||||
|
return self._cmpl_cmd_names
|
||||||
|
|
||||||
|
def equals(self, message_str: str) -> bool:
|
||||||
|
for full_cmd in self.get_complete_command_names():
|
||||||
|
if message_str == full_cmd:
|
||||||
|
return True
|
||||||
|
return False
|
||||||
|
|
||||||
def filter(self, event: AstrMessageEvent, cfg: AstrBotConfig) -> bool:
|
def filter(self, event: AstrMessageEvent, cfg: AstrBotConfig) -> bool:
|
||||||
if not event.is_at_or_wake_command:
|
if not event.is_at_or_wake_command:
|
||||||
return False
|
return False
|
||||||
@@ -145,18 +194,11 @@ class CommandFilter(HandlerFilter):
|
|||||||
|
|
||||||
# 检查是否以指令开头
|
# 检查是否以指令开头
|
||||||
message_str = re.sub(r"\s+", " ", event.get_message_str().strip())
|
message_str = re.sub(r"\s+", " ", event.get_message_str().strip())
|
||||||
candidates = [self.command_name] + list(self.alias)
|
|
||||||
ok = False
|
ok = False
|
||||||
for candidate in candidates:
|
for full_cmd in self.get_complete_command_names():
|
||||||
for parent_command_name in self.parent_command_names:
|
if message_str.startswith(f"{full_cmd} ") or message_str == full_cmd:
|
||||||
if parent_command_name:
|
|
||||||
_full = f"{parent_command_name} {candidate}"
|
|
||||||
else:
|
|
||||||
_full = candidate
|
|
||||||
if message_str.startswith(f"{_full} ") or message_str == _full:
|
|
||||||
message_str = message_str[len(_full) :].strip()
|
|
||||||
ok = True
|
ok = True
|
||||||
break
|
message_str = message_str[len(full_cmd) :].strip()
|
||||||
if not ok:
|
if not ok:
|
||||||
return False
|
return False
|
||||||
|
|
||||||
|
|||||||
@@ -13,8 +13,8 @@ class CommandGroupFilter(HandlerFilter):
|
|||||||
def __init__(
|
def __init__(
|
||||||
self,
|
self,
|
||||||
group_name: str,
|
group_name: str,
|
||||||
alias: set = None,
|
alias: set | None = None,
|
||||||
parent_group: CommandGroupFilter = None,
|
parent_group: CommandGroupFilter | None = None,
|
||||||
):
|
):
|
||||||
self.group_name = group_name
|
self.group_name = group_name
|
||||||
self.alias = alias if alias else set()
|
self.alias = alias if alias else set()
|
||||||
@@ -22,6 +22,9 @@ class CommandGroupFilter(HandlerFilter):
|
|||||||
self.custom_filter_list: List[CustomFilter] = []
|
self.custom_filter_list: List[CustomFilter] = []
|
||||||
self.parent_group = parent_group
|
self.parent_group = parent_group
|
||||||
|
|
||||||
|
# Cache for complete command names list
|
||||||
|
self._cmpl_cmd_names: list | None = None
|
||||||
|
|
||||||
def add_sub_command_filter(
|
def add_sub_command_filter(
|
||||||
self, sub_command_filter: Union[CommandFilter, CommandGroupFilter]
|
self, sub_command_filter: Union[CommandFilter, CommandGroupFilter]
|
||||||
):
|
):
|
||||||
@@ -34,6 +37,9 @@ class CommandGroupFilter(HandlerFilter):
|
|||||||
"""遍历父节点获取完整的指令名。
|
"""遍历父节点获取完整的指令名。
|
||||||
|
|
||||||
新版本 v3.4.29 采用预编译指令,不再从指令组递归遍历子指令,因此这个方法是返回包括别名在内的整个指令名列表。"""
|
新版本 v3.4.29 采用预编译指令,不再从指令组递归遍历子指令,因此这个方法是返回包括别名在内的整个指令名列表。"""
|
||||||
|
if self._cmpl_cmd_names is not None:
|
||||||
|
return self._cmpl_cmd_names
|
||||||
|
|
||||||
parent_cmd_names = (
|
parent_cmd_names = (
|
||||||
self.parent_group.get_complete_command_names() if self.parent_group else []
|
self.parent_group.get_complete_command_names() if self.parent_group else []
|
||||||
)
|
)
|
||||||
@@ -47,6 +53,7 @@ class CommandGroupFilter(HandlerFilter):
|
|||||||
for parent_cmd_name in parent_cmd_names:
|
for parent_cmd_name in parent_cmd_names:
|
||||||
for candidate in candidates:
|
for candidate in candidates:
|
||||||
result.append(parent_cmd_name + " " + candidate)
|
result.append(parent_cmd_name + " " + candidate)
|
||||||
|
self._cmpl_cmd_names = result
|
||||||
return result
|
return result
|
||||||
|
|
||||||
# 以树的形式打印出来
|
# 以树的形式打印出来
|
||||||
@@ -54,8 +61,8 @@ class CommandGroupFilter(HandlerFilter):
|
|||||||
self,
|
self,
|
||||||
sub_command_filters: List[Union[CommandFilter, CommandGroupFilter]],
|
sub_command_filters: List[Union[CommandFilter, CommandGroupFilter]],
|
||||||
prefix: str = "",
|
prefix: str = "",
|
||||||
event: AstrMessageEvent = None,
|
event: AstrMessageEvent | None = None,
|
||||||
cfg: AstrBotConfig = None,
|
cfg: AstrBotConfig | None = None,
|
||||||
) -> str:
|
) -> str:
|
||||||
result = ""
|
result = ""
|
||||||
for sub_filter in sub_command_filters:
|
for sub_filter in sub_command_filters:
|
||||||
@@ -97,6 +104,12 @@ class CommandGroupFilter(HandlerFilter):
|
|||||||
return False
|
return False
|
||||||
return True
|
return True
|
||||||
|
|
||||||
|
def startswith(self, message_str: str) -> bool:
|
||||||
|
return message_str.startswith(tuple(self.get_complete_command_names()))
|
||||||
|
|
||||||
|
def equals(self, message_str: str) -> bool:
|
||||||
|
return message_str in self.get_complete_command_names()
|
||||||
|
|
||||||
def filter(self, event: AstrMessageEvent, cfg: AstrBotConfig) -> bool:
|
def filter(self, event: AstrMessageEvent, cfg: AstrBotConfig) -> bool:
|
||||||
if not event.is_at_or_wake_command:
|
if not event.is_at_or_wake_command:
|
||||||
return False
|
return False
|
||||||
@@ -105,18 +118,14 @@ class CommandGroupFilter(HandlerFilter):
|
|||||||
if not self.custom_filter_ok(event, cfg):
|
if not self.custom_filter_ok(event, cfg):
|
||||||
return False
|
return False
|
||||||
|
|
||||||
complete_command_names = self.get_complete_command_names()
|
if self.equals(event.message_str.strip()):
|
||||||
if event.message_str.strip() in complete_command_names:
|
|
||||||
tree = (
|
tree = (
|
||||||
self.group_name
|
self.group_name
|
||||||
+ "\n"
|
+ "\n"
|
||||||
+ self.print_cmd_tree(self.sub_command_filters, event=event, cfg=cfg)
|
+ self.print_cmd_tree(self.sub_command_filters, event=event, cfg=cfg)
|
||||||
)
|
)
|
||||||
raise ValueError(
|
raise ValueError(
|
||||||
f"参数不足。{self.group_name} 指令组下有如下指令,请参考:\n"
|
f"参数不足。{self.group_name} 指令组下有如下指令,请参考:\n" + tree
|
||||||
+ tree
|
|
||||||
)
|
)
|
||||||
|
|
||||||
# complete_command_names = [name + " " for name in complete_command_names]
|
return self.startswith(event.message_str)
|
||||||
# return event.message_str.startswith(tuple(complete_command_names))
|
|
||||||
return False
|
|
||||||
|
|||||||
@@ -2,7 +2,6 @@ import enum
|
|||||||
from . import HandlerFilter
|
from . import HandlerFilter
|
||||||
from astrbot.core.platform.astr_message_event import AstrMessageEvent
|
from astrbot.core.platform.astr_message_event import AstrMessageEvent
|
||||||
from astrbot.core.config import AstrBotConfig
|
from astrbot.core.config import AstrBotConfig
|
||||||
from typing import Union
|
|
||||||
|
|
||||||
|
|
||||||
class PlatformAdapterType(enum.Flag):
|
class PlatformAdapterType(enum.Flag):
|
||||||
@@ -19,6 +18,7 @@ class PlatformAdapterType(enum.Flag):
|
|||||||
VOCECHAT = enum.auto()
|
VOCECHAT = enum.auto()
|
||||||
WEIXIN_OFFICIAL_ACCOUNT = enum.auto()
|
WEIXIN_OFFICIAL_ACCOUNT = enum.auto()
|
||||||
SATORI = enum.auto()
|
SATORI = enum.auto()
|
||||||
|
MISSKEY = enum.auto()
|
||||||
ALL = (
|
ALL = (
|
||||||
AIOCQHTTP
|
AIOCQHTTP
|
||||||
| QQOFFICIAL
|
| QQOFFICIAL
|
||||||
@@ -33,6 +33,7 @@ class PlatformAdapterType(enum.Flag):
|
|||||||
| VOCECHAT
|
| VOCECHAT
|
||||||
| WEIXIN_OFFICIAL_ACCOUNT
|
| WEIXIN_OFFICIAL_ACCOUNT
|
||||||
| SATORI
|
| SATORI
|
||||||
|
| MISSKEY
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
@@ -50,15 +51,19 @@ ADAPTER_NAME_2_TYPE = {
|
|||||||
"vocechat": PlatformAdapterType.VOCECHAT,
|
"vocechat": PlatformAdapterType.VOCECHAT,
|
||||||
"weixin_official_account": PlatformAdapterType.WEIXIN_OFFICIAL_ACCOUNT,
|
"weixin_official_account": PlatformAdapterType.WEIXIN_OFFICIAL_ACCOUNT,
|
||||||
"satori": PlatformAdapterType.SATORI,
|
"satori": PlatformAdapterType.SATORI,
|
||||||
|
"misskey": PlatformAdapterType.MISSKEY,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
class PlatformAdapterTypeFilter(HandlerFilter):
|
class PlatformAdapterTypeFilter(HandlerFilter):
|
||||||
def __init__(self, platform_adapter_type_or_str: Union[PlatformAdapterType, str]):
|
def __init__(self, platform_adapter_type_or_str: PlatformAdapterType | str):
|
||||||
self.type_or_str = platform_adapter_type_or_str
|
if isinstance(platform_adapter_type_or_str, str):
|
||||||
|
self.platform_type = ADAPTER_NAME_2_TYPE.get(platform_adapter_type_or_str)
|
||||||
|
else:
|
||||||
|
self.platform_type = platform_adapter_type_or_str
|
||||||
|
|
||||||
def filter(self, event: AstrMessageEvent, cfg: AstrBotConfig) -> bool:
|
def filter(self, event: AstrMessageEvent, cfg: AstrBotConfig) -> bool:
|
||||||
adapter_name = event.get_platform_name()
|
adapter_name = event.get_platform_name()
|
||||||
if adapter_name in ADAPTER_NAME_2_TYPE:
|
if adapter_name in ADAPTER_NAME_2_TYPE and self.platform_type is not None:
|
||||||
return ADAPTER_NAME_2_TYPE[adapter_name] & self.type_or_str
|
return bool(ADAPTER_NAME_2_TYPE[adapter_name] & self.platform_type)
|
||||||
return False
|
return False
|
||||||
|
|||||||
@@ -8,6 +8,7 @@ from .star_handler import (
|
|||||||
register_permission_type,
|
register_permission_type,
|
||||||
register_custom_filter,
|
register_custom_filter,
|
||||||
register_on_astrbot_loaded,
|
register_on_astrbot_loaded,
|
||||||
|
register_on_platform_loaded,
|
||||||
register_on_llm_request,
|
register_on_llm_request,
|
||||||
register_on_llm_response,
|
register_on_llm_response,
|
||||||
register_llm_tool,
|
register_llm_tool,
|
||||||
@@ -26,6 +27,7 @@ __all__ = [
|
|||||||
"register_permission_type",
|
"register_permission_type",
|
||||||
"register_custom_filter",
|
"register_custom_filter",
|
||||||
"register_on_astrbot_loaded",
|
"register_on_astrbot_loaded",
|
||||||
|
"register_on_platform_loaded",
|
||||||
"register_on_llm_request",
|
"register_on_llm_request",
|
||||||
"register_on_llm_response",
|
"register_on_llm_response",
|
||||||
"register_llm_tool",
|
"register_llm_tool",
|
||||||
|
|||||||
@@ -5,7 +5,9 @@ from astrbot.core.star import StarMetadata, star_map
|
|||||||
_warned_register_star = False
|
_warned_register_star = False
|
||||||
|
|
||||||
|
|
||||||
def register_star(name: str, author: str, desc: str, version: str, repo: str = None):
|
def register_star(
|
||||||
|
name: str, author: str, desc: str, version: str, repo: str | None = None
|
||||||
|
):
|
||||||
"""注册一个插件(Star)。
|
"""注册一个插件(Star)。
|
||||||
|
|
||||||
[DEPRECATED] 该装饰器已废弃,将在未来版本中移除。
|
[DEPRECATED] 该装饰器已废弃,将在未来版本中移除。
|
||||||
|
|||||||
@@ -12,7 +12,7 @@ from ..filter.platform_adapter_type import (
|
|||||||
from ..filter.permission import PermissionTypeFilter, PermissionType
|
from ..filter.permission import PermissionTypeFilter, PermissionType
|
||||||
from ..filter.custom_filter import CustomFilterAnd, CustomFilterOr
|
from ..filter.custom_filter import CustomFilterAnd, CustomFilterOr
|
||||||
from ..filter.regex import RegexFilter
|
from ..filter.regex import RegexFilter
|
||||||
from typing import Awaitable
|
from typing import Awaitable, Any, Callable
|
||||||
from astrbot.core.provider.func_tool_manager import SUPPORTED_TYPES
|
from astrbot.core.provider.func_tool_manager import SUPPORTED_TYPES
|
||||||
from astrbot.core.provider.register import llm_tools
|
from astrbot.core.provider.register import llm_tools
|
||||||
from astrbot.core.agent.agent import Agent
|
from astrbot.core.agent.agent import Agent
|
||||||
@@ -20,15 +20,19 @@ from astrbot.core.agent.tool import FunctionTool
|
|||||||
from astrbot.core.agent.handoff import HandoffTool
|
from astrbot.core.agent.handoff import HandoffTool
|
||||||
from astrbot.core.agent.hooks import BaseAgentRunHooks
|
from astrbot.core.agent.hooks import BaseAgentRunHooks
|
||||||
from astrbot.core.astr_agent_context import AstrAgentContext
|
from astrbot.core.astr_agent_context import AstrAgentContext
|
||||||
|
from astrbot.core import logger
|
||||||
|
|
||||||
|
|
||||||
def get_handler_full_name(awaitable: Awaitable) -> str:
|
def get_handler_full_name(awaitable: Callable[..., Awaitable[Any]]) -> str:
|
||||||
"""获取 Handler 的全名"""
|
"""获取 Handler 的全名"""
|
||||||
return f"{awaitable.__module__}_{awaitable.__name__}"
|
return f"{awaitable.__module__}_{awaitable.__name__}"
|
||||||
|
|
||||||
|
|
||||||
def get_handler_or_create(
|
def get_handler_or_create(
|
||||||
handler: Awaitable, event_type: EventType, dont_add=False, **kwargs
|
handler: Callable[..., Awaitable[Any]],
|
||||||
|
event_type: EventType,
|
||||||
|
dont_add=False,
|
||||||
|
**kwargs,
|
||||||
) -> StarHandlerMetadata:
|
) -> StarHandlerMetadata:
|
||||||
"""获取 Handler 或者创建一个新的 Handler"""
|
"""获取 Handler 或者创建一个新的 Handler"""
|
||||||
handler_full_name = get_handler_full_name(handler)
|
handler_full_name = get_handler_full_name(handler)
|
||||||
@@ -59,20 +63,33 @@ def get_handler_or_create(
|
|||||||
|
|
||||||
|
|
||||||
def register_command(
|
def register_command(
|
||||||
command_name: str = None, sub_command: str = None, alias: set = None, **kwargs
|
command_name: str | None = None,
|
||||||
|
sub_command: str | None = None,
|
||||||
|
alias: set | None = None,
|
||||||
|
**kwargs,
|
||||||
):
|
):
|
||||||
"""注册一个 Command."""
|
"""注册一个 Command."""
|
||||||
new_command = None
|
new_command = None
|
||||||
add_to_event_filters = False
|
add_to_event_filters = False
|
||||||
if isinstance(command_name, RegisteringCommandable):
|
if isinstance(command_name, RegisteringCommandable):
|
||||||
# 子指令
|
# 子指令
|
||||||
parent_command_names = command_name.parent_group.get_complete_command_names()
|
if sub_command is not None:
|
||||||
|
parent_command_names = (
|
||||||
|
command_name.parent_group.get_complete_command_names()
|
||||||
|
)
|
||||||
new_command = CommandFilter(
|
new_command = CommandFilter(
|
||||||
sub_command, alias, None, parent_command_names=parent_command_names
|
sub_command, alias, None, parent_command_names=parent_command_names
|
||||||
)
|
)
|
||||||
command_name.parent_group.add_sub_command_filter(new_command)
|
command_name.parent_group.add_sub_command_filter(new_command)
|
||||||
|
else:
|
||||||
|
logger.warning(
|
||||||
|
f"注册指令{command_name} 的子指令时未提供 sub_command 参数。"
|
||||||
|
)
|
||||||
else:
|
else:
|
||||||
# 裸指令
|
# 裸指令
|
||||||
|
if command_name is None:
|
||||||
|
logger.warning("注册裸指令时未提供 command_name 参数。")
|
||||||
|
else:
|
||||||
new_command = CommandFilter(command_name, alias, None)
|
new_command = CommandFilter(command_name, alias, None)
|
||||||
add_to_event_filters = True
|
add_to_event_filters = True
|
||||||
|
|
||||||
@@ -84,6 +101,7 @@ def register_command(
|
|||||||
handler_md = get_handler_or_create(
|
handler_md = get_handler_or_create(
|
||||||
awaitable, EventType.AdapterMessageEvent, **kwargs
|
awaitable, EventType.AdapterMessageEvent, **kwargs
|
||||||
)
|
)
|
||||||
|
if new_command:
|
||||||
new_command.init_handler_md(handler_md)
|
new_command.init_handler_md(handler_md)
|
||||||
handler_md.event_filters.append(new_command)
|
handler_md.event_filters.append(new_command)
|
||||||
return awaitable
|
return awaitable
|
||||||
@@ -163,26 +181,38 @@ def register_custom_filter(custom_type_filter, *args, **kwargs):
|
|||||||
|
|
||||||
|
|
||||||
def register_command_group(
|
def register_command_group(
|
||||||
command_group_name: str = None, sub_command: str = None, alias: set = None, **kwargs
|
command_group_name: str | None = None,
|
||||||
|
sub_command: str | None = None,
|
||||||
|
alias: set | None = None,
|
||||||
|
**kwargs,
|
||||||
):
|
):
|
||||||
"""注册一个 CommandGroup"""
|
"""注册一个 CommandGroup"""
|
||||||
new_group = None
|
new_group = None
|
||||||
if isinstance(command_group_name, RegisteringCommandable):
|
if isinstance(command_group_name, RegisteringCommandable):
|
||||||
# 子指令组
|
# 子指令组
|
||||||
|
if sub_command is None:
|
||||||
|
logger.warning(f"{command_group_name} 指令组的子指令组 sub_command 未指定")
|
||||||
|
else:
|
||||||
new_group = CommandGroupFilter(
|
new_group = CommandGroupFilter(
|
||||||
sub_command, alias, parent_group=command_group_name.parent_group
|
sub_command, alias, parent_group=command_group_name.parent_group
|
||||||
)
|
)
|
||||||
command_group_name.parent_group.add_sub_command_filter(new_group)
|
command_group_name.parent_group.add_sub_command_filter(new_group)
|
||||||
else:
|
else:
|
||||||
# 根指令组
|
# 根指令组
|
||||||
|
if command_group_name is None:
|
||||||
|
logger.warning("根指令组的名称未指定")
|
||||||
|
else:
|
||||||
new_group = CommandGroupFilter(command_group_name, alias)
|
new_group = CommandGroupFilter(command_group_name, alias)
|
||||||
|
|
||||||
def decorator(obj):
|
def decorator(obj):
|
||||||
# 根指令组
|
if new_group:
|
||||||
handler_md = get_handler_or_create(obj, EventType.AdapterMessageEvent, **kwargs)
|
handler_md = get_handler_or_create(
|
||||||
|
obj, EventType.AdapterMessageEvent, **kwargs
|
||||||
|
)
|
||||||
handler_md.event_filters.append(new_group)
|
handler_md.event_filters.append(new_group)
|
||||||
|
|
||||||
return RegisteringCommandable(new_group)
|
return RegisteringCommandable(new_group)
|
||||||
|
raise ValueError("注册指令组失败。")
|
||||||
|
|
||||||
return decorator
|
return decorator
|
||||||
|
|
||||||
@@ -190,9 +220,11 @@ def register_command_group(
|
|||||||
class RegisteringCommandable:
|
class RegisteringCommandable:
|
||||||
"""用于指令组级联注册"""
|
"""用于指令组级联注册"""
|
||||||
|
|
||||||
group: CommandGroupFilter = register_command_group
|
group: Callable[..., Callable[..., "RegisteringCommandable"]] = (
|
||||||
command: CommandFilter = register_command
|
register_command_group
|
||||||
custom_filter = register_custom_filter
|
)
|
||||||
|
command: Callable[..., Callable[..., None]] = register_command
|
||||||
|
custom_filter: Callable[..., Callable[..., None]] = register_custom_filter
|
||||||
|
|
||||||
def __init__(self, parent_group: CommandGroupFilter):
|
def __init__(self, parent_group: CommandGroupFilter):
|
||||||
self.parent_group = parent_group
|
self.parent_group = parent_group
|
||||||
@@ -267,6 +299,18 @@ def register_on_astrbot_loaded(**kwargs):
|
|||||||
return decorator
|
return decorator
|
||||||
|
|
||||||
|
|
||||||
|
def register_on_platform_loaded(**kwargs):
|
||||||
|
"""
|
||||||
|
当平台加载完成时
|
||||||
|
"""
|
||||||
|
|
||||||
|
def decorator(awaitable):
|
||||||
|
_ = get_handler_or_create(awaitable, EventType.OnPlatformLoadedEvent, **kwargs)
|
||||||
|
return awaitable
|
||||||
|
|
||||||
|
return decorator
|
||||||
|
|
||||||
|
|
||||||
def register_on_llm_request(**kwargs):
|
def register_on_llm_request(**kwargs):
|
||||||
"""当有 LLM 请求时的事件
|
"""当有 LLM 请求时的事件
|
||||||
|
|
||||||
@@ -311,7 +355,7 @@ def register_on_llm_response(**kwargs):
|
|||||||
return decorator
|
return decorator
|
||||||
|
|
||||||
|
|
||||||
def register_llm_tool(name: str = None, **kwargs):
|
def register_llm_tool(name: str | None = None, **kwargs):
|
||||||
"""为函数调用(function-calling / tools-use)添加工具。
|
"""为函数调用(function-calling / tools-use)添加工具。
|
||||||
|
|
||||||
请务必按照以下格式编写一个工具(包括函数注释,AstrBot 会尝试解析该函数注释)
|
请务必按照以下格式编写一个工具(包括函数注释,AstrBot 会尝试解析该函数注释)
|
||||||
@@ -349,9 +393,10 @@ def register_llm_tool(name: str = None, **kwargs):
|
|||||||
if kwargs.get("registering_agent"):
|
if kwargs.get("registering_agent"):
|
||||||
registering_agent = kwargs["registering_agent"]
|
registering_agent = kwargs["registering_agent"]
|
||||||
|
|
||||||
def decorator(awaitable: Awaitable):
|
def decorator(awaitable: Callable[..., Awaitable[Any]]):
|
||||||
llm_tool_name = name_ if name_ else awaitable.__name__
|
llm_tool_name = name_ if name_ else awaitable.__name__
|
||||||
docstring = docstring_parser.parse(awaitable.__doc__)
|
func_doc = awaitable.__doc__ or ""
|
||||||
|
docstring = docstring_parser.parse(func_doc)
|
||||||
args = []
|
args = []
|
||||||
for arg in docstring.params:
|
for arg in docstring.params:
|
||||||
if arg.type_name not in SUPPORTED_TYPES:
|
if arg.type_name not in SUPPORTED_TYPES:
|
||||||
@@ -367,18 +412,18 @@ def register_llm_tool(name: str = None, **kwargs):
|
|||||||
)
|
)
|
||||||
# print(llm_tool_name, registering_agent)
|
# print(llm_tool_name, registering_agent)
|
||||||
if not registering_agent:
|
if not registering_agent:
|
||||||
|
doc_desc = docstring.description.strip() if docstring.description else ""
|
||||||
md = get_handler_or_create(awaitable, EventType.OnCallingFuncToolEvent)
|
md = get_handler_or_create(awaitable, EventType.OnCallingFuncToolEvent)
|
||||||
llm_tools.add_func(
|
llm_tools.add_func(llm_tool_name, args, doc_desc, md.handler)
|
||||||
llm_tool_name, args, docstring.description.strip(), md.handler
|
|
||||||
)
|
|
||||||
else:
|
else:
|
||||||
assert isinstance(registering_agent, RegisteringAgent)
|
assert isinstance(registering_agent, RegisteringAgent)
|
||||||
# print(f"Registering tool {llm_tool_name} for agent", registering_agent._agent.name)
|
# print(f"Registering tool {llm_tool_name} for agent", registering_agent._agent.name)
|
||||||
if registering_agent._agent.tools is None:
|
if registering_agent._agent.tools is None:
|
||||||
registering_agent._agent.tools = []
|
registering_agent._agent.tools = []
|
||||||
registering_agent._agent.tools.append(llm_tools.spec_to_func(
|
|
||||||
llm_tool_name, args, docstring.description.strip(), awaitable
|
desc = docstring.description.strip() if docstring.description else ""
|
||||||
))
|
tool = llm_tools.spec_to_func(llm_tool_name, args, desc, awaitable)
|
||||||
|
registering_agent._agent.tools.append(tool)
|
||||||
|
|
||||||
return awaitable
|
return awaitable
|
||||||
|
|
||||||
@@ -399,8 +444,8 @@ class RegisteringAgent:
|
|||||||
def register_agent(
|
def register_agent(
|
||||||
name: str,
|
name: str,
|
||||||
instruction: str,
|
instruction: str,
|
||||||
tools: list[str | FunctionTool] = None,
|
tools: list[str | FunctionTool] | None = None,
|
||||||
run_hooks: BaseAgentRunHooks[AstrAgentContext] = None,
|
run_hooks: BaseAgentRunHooks[AstrAgentContext] | None = None,
|
||||||
):
|
):
|
||||||
"""注册一个 Agent
|
"""注册一个 Agent
|
||||||
|
|
||||||
@@ -412,7 +457,7 @@ def register_agent(
|
|||||||
"""
|
"""
|
||||||
tools_ = tools or []
|
tools_ = tools or []
|
||||||
|
|
||||||
def decorator(awaitable: Awaitable):
|
def decorator(awaitable: Callable[..., Awaitable[Any]]):
|
||||||
AstrAgent = Agent[AstrAgentContext]
|
AstrAgent = Agent[AstrAgentContext]
|
||||||
agent = AstrAgent(
|
agent = AstrAgent(
|
||||||
name=name,
|
name=name,
|
||||||
|
|||||||
@@ -52,10 +52,6 @@ class SessionServiceManager:
|
|||||||
"session_service_config", session_config, scope="umo", scope_id=session_id
|
"session_service_config", session_config, scope="umo", scope_id=session_id
|
||||||
)
|
)
|
||||||
|
|
||||||
logger.info(
|
|
||||||
f"会话 {session_id} 的LLM状态已更新为: {'启用' if enabled else '禁用'}"
|
|
||||||
)
|
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def should_process_llm_request(event: AstrMessageEvent) -> bool:
|
def should_process_llm_request(event: AstrMessageEvent) -> bool:
|
||||||
"""检查是否应该处理LLM请求
|
"""检查是否应该处理LLM请求
|
||||||
|
|||||||
@@ -84,7 +84,10 @@ class SessionPluginManager:
|
|||||||
session_config["disabled_plugins"] = disabled_plugins
|
session_config["disabled_plugins"] = disabled_plugins
|
||||||
session_plugin_config[session_id] = session_config
|
session_plugin_config[session_id] = session_config
|
||||||
sp.put(
|
sp.put(
|
||||||
"session_plugin_config", session_plugin_config, scope="umo", scope_id=session_id
|
"session_plugin_config",
|
||||||
|
session_plugin_config,
|
||||||
|
scope="umo",
|
||||||
|
scope_id=session_id,
|
||||||
)
|
)
|
||||||
|
|
||||||
logger.info(
|
logger.info(
|
||||||
@@ -137,6 +140,9 @@ class SessionPluginManager:
|
|||||||
filtered_handlers.append(handler)
|
filtered_handlers.append(handler)
|
||||||
continue
|
continue
|
||||||
|
|
||||||
|
if plugin.name is None:
|
||||||
|
continue
|
||||||
|
|
||||||
# 检查插件是否在当前会话中启用
|
# 检查插件是否在当前会话中启用
|
||||||
if SessionPluginManager.is_plugin_enabled_for_session(
|
if SessionPluginManager.is_plugin_enabled_for_session(
|
||||||
session_id, plugin.name
|
session_id, plugin.name
|
||||||
|
|||||||
@@ -1,7 +1,7 @@
|
|||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
import enum
|
import enum
|
||||||
from dataclasses import dataclass, field
|
from dataclasses import dataclass, field
|
||||||
from typing import Awaitable, List, Dict, TypeVar, Generic
|
from typing import Callable, Awaitable, Any, List, Dict, TypeVar, Generic
|
||||||
from .filter import HandlerFilter
|
from .filter import HandlerFilter
|
||||||
from .star import star_map
|
from .star import star_map
|
||||||
|
|
||||||
@@ -34,26 +34,33 @@ class StarHandlerRegistry(Generic[T]):
|
|||||||
) -> List[StarHandlerMetadata]:
|
) -> List[StarHandlerMetadata]:
|
||||||
handlers = []
|
handlers = []
|
||||||
for handler in self._handlers:
|
for handler in self._handlers:
|
||||||
|
# 过滤事件类型
|
||||||
if handler.event_type != event_type:
|
if handler.event_type != event_type:
|
||||||
continue
|
continue
|
||||||
|
# 过滤启用状态
|
||||||
if only_activated:
|
if only_activated:
|
||||||
plugin = star_map.get(handler.handler_module_path)
|
plugin = star_map.get(handler.handler_module_path)
|
||||||
if not (plugin and plugin.activated):
|
if not (plugin and plugin.activated):
|
||||||
continue
|
continue
|
||||||
|
# 过滤插件白名单
|
||||||
if plugins_name is not None and plugins_name != ["*"]:
|
if plugins_name is not None and plugins_name != ["*"]:
|
||||||
plugin = star_map.get(handler.handler_module_path)
|
plugin = star_map.get(handler.handler_module_path)
|
||||||
if not plugin:
|
if not plugin:
|
||||||
continue
|
continue
|
||||||
if (
|
if (
|
||||||
plugin.name not in plugins_name
|
plugin.name not in plugins_name
|
||||||
and event_type != EventType.OnAstrBotLoadedEvent
|
and event_type
|
||||||
|
not in (
|
||||||
|
EventType.OnAstrBotLoadedEvent,
|
||||||
|
EventType.OnPlatformLoadedEvent,
|
||||||
|
)
|
||||||
and not plugin.reserved
|
and not plugin.reserved
|
||||||
):
|
):
|
||||||
continue
|
continue
|
||||||
handlers.append(handler)
|
handlers.append(handler)
|
||||||
return handlers
|
return handlers
|
||||||
|
|
||||||
def get_handler_by_full_name(self, full_name: str) -> StarHandlerMetadata:
|
def get_handler_by_full_name(self, full_name: str) -> StarHandlerMetadata | None:
|
||||||
return self.star_handlers_map.get(full_name, None)
|
return self.star_handlers_map.get(full_name, None)
|
||||||
|
|
||||||
def get_handlers_by_module_name(
|
def get_handlers_by_module_name(
|
||||||
@@ -80,7 +87,7 @@ class StarHandlerRegistry(Generic[T]):
|
|||||||
return len(self._handlers)
|
return len(self._handlers)
|
||||||
|
|
||||||
|
|
||||||
star_handlers_registry = StarHandlerRegistry()
|
star_handlers_registry = StarHandlerRegistry() # type: ignore
|
||||||
|
|
||||||
|
|
||||||
class EventType(enum.Enum):
|
class EventType(enum.Enum):
|
||||||
@@ -90,6 +97,7 @@ class EventType(enum.Enum):
|
|||||||
"""
|
"""
|
||||||
|
|
||||||
OnAstrBotLoadedEvent = enum.auto() # AstrBot 加载完成
|
OnAstrBotLoadedEvent = enum.auto() # AstrBot 加载完成
|
||||||
|
OnPlatformLoadedEvent = enum.auto() # 平台加载完成
|
||||||
|
|
||||||
AdapterMessageEvent = enum.auto() # 收到适配器发来的消息
|
AdapterMessageEvent = enum.auto() # 收到适配器发来的消息
|
||||||
OnLLMRequestEvent = enum.auto() # 收到 LLM 请求(可以是用户也可以是插件)
|
OnLLMRequestEvent = enum.auto() # 收到 LLM 请求(可以是用户也可以是插件)
|
||||||
@@ -115,7 +123,7 @@ class StarHandlerMetadata:
|
|||||||
handler_module_path: str
|
handler_module_path: str
|
||||||
"""Handler 所在的模块路径。"""
|
"""Handler 所在的模块路径。"""
|
||||||
|
|
||||||
handler: Awaitable
|
handler: Callable[..., Awaitable[Any]]
|
||||||
"""Handler 的函数对象,应当是一个异步函数"""
|
"""Handler 的函数对象,应当是一个异步函数"""
|
||||||
|
|
||||||
event_filters: List[HandlerFilter]
|
event_filters: List[HandlerFilter]
|
||||||
|
|||||||
@@ -43,7 +43,7 @@ class PluginManager:
|
|||||||
self.updator = PluginUpdator()
|
self.updator = PluginUpdator()
|
||||||
|
|
||||||
self.context = context
|
self.context = context
|
||||||
self.context._star_manager = self
|
self.context._star_manager = self # type: ignore
|
||||||
|
|
||||||
self.config = config
|
self.config = config
|
||||||
self.plugin_store_path = get_astrbot_plugin_path()
|
self.plugin_store_path = get_astrbot_plugin_path()
|
||||||
@@ -478,6 +478,7 @@ class PluginManager:
|
|||||||
if isinstance(func_tool, HandoffTool):
|
if isinstance(func_tool, HandoffTool):
|
||||||
need_apply = []
|
need_apply = []
|
||||||
sub_tools = func_tool.agent.tools
|
sub_tools = func_tool.agent.tools
|
||||||
|
if sub_tools:
|
||||||
for sub_tool in sub_tools:
|
for sub_tool in sub_tools:
|
||||||
if isinstance(sub_tool, FunctionTool):
|
if isinstance(sub_tool, FunctionTool):
|
||||||
need_apply.append(sub_tool)
|
need_apply.append(sub_tool)
|
||||||
@@ -686,6 +687,9 @@ class PluginManager:
|
|||||||
)
|
)
|
||||||
|
|
||||||
# 从 star_registry 和 star_map 中删除
|
# 从 star_registry 和 star_map 中删除
|
||||||
|
if plugin.module_path is None or root_dir_name is None:
|
||||||
|
raise Exception(f"插件 {plugin_name} 数据不完整,无法卸载。")
|
||||||
|
|
||||||
await self._unbind_plugin(plugin_name, plugin.module_path)
|
await self._unbind_plugin(plugin_name, plugin.module_path)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
@@ -791,15 +795,17 @@ class PluginManager:
|
|||||||
if star_metadata.star_cls is None:
|
if star_metadata.star_cls is None:
|
||||||
return
|
return
|
||||||
|
|
||||||
if '__del__' in star_metadata.star_cls_type.__dict__:
|
if "__del__" in star_metadata.star_cls_type.__dict__:
|
||||||
asyncio.get_event_loop().run_in_executor(
|
asyncio.get_event_loop().run_in_executor(
|
||||||
None, star_metadata.star_cls.__del__
|
None, star_metadata.star_cls.__del__
|
||||||
)
|
)
|
||||||
elif 'terminate' in star_metadata.star_cls_type.__dict__:
|
elif "terminate" in star_metadata.star_cls_type.__dict__:
|
||||||
await star_metadata.star_cls.terminate()
|
await star_metadata.star_cls.terminate()
|
||||||
|
|
||||||
async def turn_on_plugin(self, plugin_name: str):
|
async def turn_on_plugin(self, plugin_name: str):
|
||||||
plugin = self.context.get_registered_star(plugin_name)
|
plugin = self.context.get_registered_star(plugin_name)
|
||||||
|
if plugin is None:
|
||||||
|
raise Exception(f"插件 {plugin_name} 不存在。")
|
||||||
inactivated_plugins: list = await sp.global_get("inactivated_plugins", [])
|
inactivated_plugins: list = await sp.global_get("inactivated_plugins", [])
|
||||||
inactivated_llm_tools: list = await sp.global_get("inactivated_llm_tools", [])
|
inactivated_llm_tools: list = await sp.global_get("inactivated_llm_tools", [])
|
||||||
if plugin.module_path in inactivated_plugins:
|
if plugin.module_path in inactivated_plugins:
|
||||||
|
|||||||
@@ -22,7 +22,7 @@ import inspect
|
|||||||
import os
|
import os
|
||||||
import uuid
|
import uuid
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from typing import Union, Awaitable, List, Optional, ClassVar
|
from typing import Union, Awaitable, Callable, Any, List, Optional, ClassVar
|
||||||
from astrbot.core.message.components import BaseMessageComponent
|
from astrbot.core.message.components import BaseMessageComponent
|
||||||
from astrbot.core.message.message_event_result import MessageChain
|
from astrbot.core.message.message_event_result import MessageChain
|
||||||
from astrbot.api.platform import MessageMember, AstrBotMessage, MessageType
|
from astrbot.api.platform import MessageMember, AstrBotMessage, MessageType
|
||||||
@@ -30,8 +30,13 @@ from astrbot.core.platform.astr_message_event import MessageSesion
|
|||||||
from astrbot.core.star.context import Context
|
from astrbot.core.star.context import Context
|
||||||
from astrbot.core.star.star import star_map
|
from astrbot.core.star.star import star_map
|
||||||
from astrbot.core.utils.astrbot_path import get_astrbot_data_path
|
from astrbot.core.utils.astrbot_path import get_astrbot_data_path
|
||||||
from astrbot.core.platform.sources.aiocqhttp.aiocqhttp_message_event import AiocqhttpMessageEvent
|
from astrbot.core.platform.sources.aiocqhttp.aiocqhttp_message_event import (
|
||||||
from astrbot.core.platform.sources.aiocqhttp.aiocqhttp_platform_adapter import AiocqhttpAdapter
|
AiocqhttpMessageEvent,
|
||||||
|
)
|
||||||
|
from astrbot.core.platform.sources.aiocqhttp.aiocqhttp_platform_adapter import (
|
||||||
|
AiocqhttpAdapter,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
class StarTools:
|
class StarTools:
|
||||||
"""
|
"""
|
||||||
@@ -77,7 +82,11 @@ class StarTools:
|
|||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
async def send_message_by_id(
|
async def send_message_by_id(
|
||||||
cls, type: str, id: str, message_chain: MessageChain, platform: str = "aiocqhttp"
|
cls,
|
||||||
|
type: str,
|
||||||
|
id: str,
|
||||||
|
message_chain: MessageChain,
|
||||||
|
platform: str = "aiocqhttp",
|
||||||
):
|
):
|
||||||
"""
|
"""
|
||||||
根据 id(例如qq号, 群号等) 直接, 主动地发送消息
|
根据 id(例如qq号, 群号等) 直接, 主动地发送消息
|
||||||
@@ -92,7 +101,9 @@ class StarTools:
|
|||||||
raise ValueError("StarTools not initialized")
|
raise ValueError("StarTools not initialized")
|
||||||
platforms = cls._context.platform_manager.get_insts()
|
platforms = cls._context.platform_manager.get_insts()
|
||||||
if platform == "aiocqhttp":
|
if platform == "aiocqhttp":
|
||||||
adapter = next((p for p in platforms if isinstance(p, AiocqhttpAdapter)), None)
|
adapter = next(
|
||||||
|
(p for p in platforms if isinstance(p, AiocqhttpAdapter)), None
|
||||||
|
)
|
||||||
if adapter is None:
|
if adapter is None:
|
||||||
raise ValueError("未找到适配器: AiocqhttpAdapter")
|
raise ValueError("未找到适配器: AiocqhttpAdapter")
|
||||||
await AiocqhttpMessageEvent.send_message(
|
await AiocqhttpMessageEvent.send_message(
|
||||||
@@ -115,7 +126,7 @@ class StarTools:
|
|||||||
message_str: str,
|
message_str: str,
|
||||||
message_id: str = "",
|
message_id: str = "",
|
||||||
raw_message: object = None,
|
raw_message: object = None,
|
||||||
group_id: str = ""
|
group_id: str = "",
|
||||||
) -> AstrBotMessage:
|
) -> AstrBotMessage:
|
||||||
"""
|
"""
|
||||||
创建一个AstrBot消息对象
|
创建一个AstrBot消息对象
|
||||||
@@ -152,7 +163,6 @@ class StarTools:
|
|||||||
@classmethod
|
@classmethod
|
||||||
async def create_event(
|
async def create_event(
|
||||||
cls, abm: AstrBotMessage, platform: str = "aiocqhttp", is_wake: bool = True
|
cls, abm: AstrBotMessage, platform: str = "aiocqhttp", is_wake: bool = True
|
||||||
|
|
||||||
) -> None:
|
) -> None:
|
||||||
"""
|
"""
|
||||||
创建并提交事件到指定平台
|
创建并提交事件到指定平台
|
||||||
@@ -167,7 +177,9 @@ class StarTools:
|
|||||||
raise ValueError("StarTools not initialized")
|
raise ValueError("StarTools not initialized")
|
||||||
platforms = cls._context.platform_manager.get_insts()
|
platforms = cls._context.platform_manager.get_insts()
|
||||||
if platform == "aiocqhttp":
|
if platform == "aiocqhttp":
|
||||||
adapter = next((p for p in platforms if isinstance(p, AiocqhttpAdapter)), None)
|
adapter = next(
|
||||||
|
(p for p in platforms if isinstance(p, AiocqhttpAdapter)), None
|
||||||
|
)
|
||||||
if adapter is None:
|
if adapter is None:
|
||||||
raise ValueError("未找到适配器: AiocqhttpAdapter")
|
raise ValueError("未找到适配器: AiocqhttpAdapter")
|
||||||
event = AiocqhttpMessageEvent(
|
event = AiocqhttpMessageEvent(
|
||||||
@@ -209,7 +221,11 @@ class StarTools:
|
|||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def register_llm_tool(
|
def register_llm_tool(
|
||||||
cls, name: str, func_args: list, desc: str, func_obj: Awaitable
|
cls,
|
||||||
|
name: str,
|
||||||
|
func_args: list,
|
||||||
|
desc: str,
|
||||||
|
func_obj: Callable[..., Awaitable[Any]],
|
||||||
) -> None:
|
) -> None:
|
||||||
"""
|
"""
|
||||||
为函数调用(function-calling/tools-use)添加工具
|
为函数调用(function-calling/tools-use)添加工具
|
||||||
@@ -277,7 +293,9 @@ class StarTools:
|
|||||||
if not plugin_name:
|
if not plugin_name:
|
||||||
raise ValueError("无法获取插件名称")
|
raise ValueError("无法获取插件名称")
|
||||||
|
|
||||||
data_dir = Path(os.path.join(get_astrbot_data_path(), "plugin_data", plugin_name))
|
data_dir = Path(
|
||||||
|
os.path.join(get_astrbot_data_path(), "plugin_data", plugin_name)
|
||||||
|
)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
data_dir.mkdir(parents=True, exist_ok=True)
|
data_dir.mkdir(parents=True, exist_ok=True)
|
||||||
|
|||||||
@@ -32,6 +32,9 @@ class PluginUpdator(RepoZipUpdator):
|
|||||||
if not repo_url:
|
if not repo_url:
|
||||||
raise Exception(f"插件 {plugin.name} 没有指定仓库地址。")
|
raise Exception(f"插件 {plugin.name} 没有指定仓库地址。")
|
||||||
|
|
||||||
|
if not plugin.root_dir_name:
|
||||||
|
raise Exception(f"插件 {plugin.name} 的根目录名未指定。")
|
||||||
|
|
||||||
plugin_path = os.path.join(self.plugin_store_path, plugin.root_dir_name)
|
plugin_path = os.path.join(self.plugin_store_path, plugin.root_dir_name)
|
||||||
|
|
||||||
logger.info(f"正在更新插件,路径: {plugin_path},仓库地址: {repo_url}")
|
logger.info(f"正在更新插件,路径: {plugin_path},仓库地址: {repo_url}")
|
||||||
|
|||||||
@@ -6,7 +6,7 @@ class CommandTokens:
|
|||||||
self.tokens = []
|
self.tokens = []
|
||||||
self.len = 0
|
self.len = 0
|
||||||
|
|
||||||
def get(self, idx: int):
|
def get(self, idx: int) -> str | None:
|
||||||
if idx >= self.len:
|
if idx >= self.len:
|
||||||
return None
|
return None
|
||||||
return self.tokens[idx].strip()
|
return self.tokens[idx].strip()
|
||||||
|
|||||||
@@ -1,9 +1,33 @@
|
|||||||
|
import codecs
|
||||||
import json
|
import json
|
||||||
from astrbot.core import logger
|
from astrbot.core import logger
|
||||||
from aiohttp import ClientSession
|
from aiohttp import ClientSession, ClientResponse
|
||||||
from typing import Dict, List, Any, AsyncGenerator
|
from typing import Dict, List, Any, AsyncGenerator
|
||||||
|
|
||||||
|
|
||||||
|
async def _stream_sse(resp: ClientResponse) -> AsyncGenerator[dict, None]:
|
||||||
|
decoder = codecs.getincrementaldecoder("utf-8")()
|
||||||
|
buffer = ""
|
||||||
|
async for chunk in resp.content.iter_chunked(8192):
|
||||||
|
buffer += decoder.decode(chunk)
|
||||||
|
while "\n\n" in buffer:
|
||||||
|
block, buffer = buffer.split("\n\n", 1)
|
||||||
|
if block.strip().startswith("data:"):
|
||||||
|
try:
|
||||||
|
yield json.loads(block[5:])
|
||||||
|
except json.JSONDecodeError:
|
||||||
|
logger.warning(f"Drop invalid dify json data: {block[5:]}")
|
||||||
|
continue
|
||||||
|
# flush any remaining text
|
||||||
|
buffer += decoder.decode(b"", final=True)
|
||||||
|
if buffer.strip().startswith("data:"):
|
||||||
|
try:
|
||||||
|
yield json.loads(buffer[5:])
|
||||||
|
except json.JSONDecodeError:
|
||||||
|
logger.warning(f"Drop invalid dify json data: {buffer[5:]}")
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
class DifyAPIClient:
|
class DifyAPIClient:
|
||||||
def __init__(self, api_key: str, api_base: str = "https://api.dify.ai/v1"):
|
def __init__(self, api_key: str, api_base: str = "https://api.dify.ai/v1"):
|
||||||
self.api_key = api_key
|
self.api_key = api_key
|
||||||
@@ -33,31 +57,11 @@ class DifyAPIClient:
|
|||||||
) as resp:
|
) as resp:
|
||||||
if resp.status != 200:
|
if resp.status != 200:
|
||||||
text = await resp.text()
|
text = await resp.text()
|
||||||
raise Exception(f"chat_messages 请求失败:{resp.status}. {text}")
|
raise Exception(
|
||||||
|
f"Dify /chat-messages 接口请求失败:{resp.status}. {text}"
|
||||||
buffer = ""
|
)
|
||||||
while True:
|
async for event in _stream_sse(resp):
|
||||||
# 保持原有的8192字节限制,防止数据过大导致高水位报错
|
yield event
|
||||||
chunk = await resp.content.read(8192)
|
|
||||||
if not chunk:
|
|
||||||
break
|
|
||||||
|
|
||||||
buffer += chunk.decode("utf-8")
|
|
||||||
blocks = buffer.split("\n\n")
|
|
||||||
|
|
||||||
# 处理完整的数据块
|
|
||||||
for block in blocks[:-1]:
|
|
||||||
if block.strip() and block.startswith("data:"):
|
|
||||||
try:
|
|
||||||
json_str = block[5:] # 移除 "data:" 前缀
|
|
||||||
json_obj = json.loads(json_str)
|
|
||||||
yield json_obj
|
|
||||||
except json.JSONDecodeError as e:
|
|
||||||
logger.error(f"JSON解析错误: {str(e)}")
|
|
||||||
logger.error(f"原始数据块: {json_str}")
|
|
||||||
|
|
||||||
# 保留最后一个可能不完整的块
|
|
||||||
buffer = blocks[-1] if blocks else ""
|
|
||||||
|
|
||||||
async def workflow_run(
|
async def workflow_run(
|
||||||
self,
|
self,
|
||||||
@@ -77,31 +81,11 @@ class DifyAPIClient:
|
|||||||
) as resp:
|
) as resp:
|
||||||
if resp.status != 200:
|
if resp.status != 200:
|
||||||
text = await resp.text()
|
text = await resp.text()
|
||||||
raise Exception(f"workflow_run 请求失败:{resp.status}. {text}")
|
raise Exception(
|
||||||
|
f"Dify /workflows/run 接口请求失败:{resp.status}. {text}"
|
||||||
buffer = ""
|
)
|
||||||
while True:
|
async for event in _stream_sse(resp):
|
||||||
# 保持原有的8192字节限制,防止数据过大导致高水位报错
|
yield event
|
||||||
chunk = await resp.content.read(8192)
|
|
||||||
if not chunk:
|
|
||||||
break
|
|
||||||
|
|
||||||
buffer += chunk.decode("utf-8")
|
|
||||||
blocks = buffer.split("\n\n")
|
|
||||||
|
|
||||||
# 处理完整的数据块
|
|
||||||
for block in blocks[:-1]:
|
|
||||||
if block.strip() and block.startswith("data:"):
|
|
||||||
try:
|
|
||||||
json_str = block[5:] # 移除 "data:" 前缀
|
|
||||||
json_obj = json.loads(json_str)
|
|
||||||
yield json_obj
|
|
||||||
except json.JSONDecodeError as e:
|
|
||||||
logger.error(f"JSON解析错误: {str(e)}")
|
|
||||||
logger.error(f"原始数据块: {json_str}")
|
|
||||||
|
|
||||||
# 保留最后一个可能不完整的块
|
|
||||||
buffer = blocks[-1] if blocks else ""
|
|
||||||
|
|
||||||
async def file_upload(
|
async def file_upload(
|
||||||
self,
|
self,
|
||||||
@@ -109,11 +93,14 @@ class DifyAPIClient:
|
|||||||
user: str,
|
user: str,
|
||||||
) -> Dict[str, Any]:
|
) -> Dict[str, Any]:
|
||||||
url = f"{self.api_base}/files/upload"
|
url = f"{self.api_base}/files/upload"
|
||||||
|
with open(file_path, "rb") as f:
|
||||||
payload = {
|
payload = {
|
||||||
"user": user,
|
"user": user,
|
||||||
"file": open(file_path, "rb"),
|
"file": f,
|
||||||
}
|
}
|
||||||
async with self.session.post(url, data=payload, headers=self.headers) as resp:
|
async with self.session.post(
|
||||||
|
url, data=payload, headers=self.headers
|
||||||
|
) as resp:
|
||||||
return await resp.json() # {"id": "xxx", ...}
|
return await resp.json() # {"id": "xxx", ...}
|
||||||
|
|
||||||
async def close(self):
|
async def close(self):
|
||||||
|
|||||||
@@ -227,9 +227,11 @@ async def download_dashboard(
|
|||||||
path = os.path.join(get_astrbot_data_path(), "dashboard.zip")
|
path = os.path.join(get_astrbot_data_path(), "dashboard.zip")
|
||||||
|
|
||||||
if latest or len(str(version)) != 40:
|
if latest or len(str(version)) != 40:
|
||||||
logger.info(f"准备下载 {version} 发行版本的 AstrBot WebUI 文件")
|
|
||||||
ver_name = "latest" if latest else version
|
ver_name = "latest" if latest else version
|
||||||
dashboard_release_url = f"https://astrbot-registry.soulter.top/download/astrbot-dashboard/{ver_name}/dist.zip"
|
dashboard_release_url = f"https://astrbot-registry.soulter.top/download/astrbot-dashboard/{ver_name}/dist.zip"
|
||||||
|
logger.info(
|
||||||
|
f"准备下载指定发行版本的 AstrBot WebUI 文件: {dashboard_release_url}"
|
||||||
|
)
|
||||||
try:
|
try:
|
||||||
await download_file(dashboard_release_url, path, show_progress=True)
|
await download_file(dashboard_release_url, path, show_progress=True)
|
||||||
except BaseException as _:
|
except BaseException as _:
|
||||||
@@ -241,24 +243,10 @@ async def download_dashboard(
|
|||||||
dashboard_release_url = f"{proxy}/{dashboard_release_url}"
|
dashboard_release_url = f"{proxy}/{dashboard_release_url}"
|
||||||
await download_file(dashboard_release_url, path, show_progress=True)
|
await download_file(dashboard_release_url, path, show_progress=True)
|
||||||
else:
|
else:
|
||||||
logger.info(f"准备下载指定版本的 AstrBot WebUI: {version}")
|
url = f"https://github.com/AstrBotDevs/astrbot-release-harbour/releases/download/release-{version}/dist.zip"
|
||||||
|
logger.info(f"准备下载指定版本的 AstrBot WebUI: {url}")
|
||||||
url = (
|
|
||||||
"https://api.github.com/repos/AstrBotDevs/astrbot-release-harbour/releases"
|
|
||||||
)
|
|
||||||
if proxy:
|
if proxy:
|
||||||
url = f"{proxy}/{url}"
|
url = f"{proxy}/{url}"
|
||||||
async with aiohttp.ClientSession(trust_env=True) as session:
|
await download_file(url, path, show_progress=True)
|
||||||
async with session.get(url) as resp:
|
|
||||||
if resp.status == 200:
|
|
||||||
releases = await resp.json()
|
|
||||||
for release in releases:
|
|
||||||
if version in release["tag_name"]:
|
|
||||||
download_url = release["assets"][0]["browser_download_url"]
|
|
||||||
await download_file(download_url, path, show_progress=True)
|
|
||||||
else:
|
|
||||||
logger.warning(f"未找到指定的版本的 Dashboard 构建文件: {version}")
|
|
||||||
return
|
|
||||||
|
|
||||||
with zipfile.ZipFile(path, "r") as z:
|
with zipfile.ZipFile(path, "r") as z:
|
||||||
z.extractall(extract_path)
|
z.extractall(extract_path)
|
||||||
|
|||||||
@@ -1,247 +0,0 @@
|
|||||||
<!doctype html>
|
|
||||||
<html>
|
|
||||||
<head>
|
|
||||||
<meta charset="utf-8"/>
|
|
||||||
<link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/katex@0.16.10/dist/katex.min.css" integrity="sha384-wcIxkf4k558AjM3Yz3BBFQUbk/zgIYC2R0QpeeYb+TwlBVMrlgLqwRjRtGZiK7ww" crossorigin="anonymous">
|
|
||||||
<link rel="stylesheet" href="/path/to/styles/default.min.css">
|
|
||||||
<script src="/path/to/highlight.min.js"></script>
|
|
||||||
<script>hljs.highlightAll();</script>
|
|
||||||
<script defer src="https://cdn.jsdelivr.net/npm/katex@0.16.10/dist/katex.min.js" integrity="sha384-hIoBPJpTUs74ddyc4bFZSM1TVlQDA60VBbJS0oA934VSz82sBx1X7kSx2ATBDIyd" crossorigin="anonymous"></script>
|
|
||||||
<script defer src="https://cdn.jsdelivr.net/npm/katex@0.16.10/dist/contrib/auto-render.min.js" integrity="sha384-43gviWU0YVjaDtb/GhzOouOXtZMP/7XUzwPTstBeZFe/+rCMvRwr4yROQP43s0Xk" crossorigin="anonymous"
|
|
||||||
onload="renderMathInElement(document.getElementById('content'),{delimiters: [{left: '$$', right: '$$', display: true},{left: '$', right: '$', display: false}]});"></script>
|
|
||||||
</head>
|
|
||||||
<body>
|
|
||||||
<div style="background-color: #3276dc; color: #fff; font-size: 64px; ">
|
|
||||||
<span style="font-weight: bold; margin-left: 16px"># AstrBot</span>
|
|
||||||
<span>{{ version }}</span>
|
|
||||||
</div>
|
|
||||||
<article style="margin-top: 32px" id="content"></article>
|
|
||||||
<script src="https://cdn.jsdelivr.net/npm/marked/marked.min.js"></script>
|
|
||||||
<script>
|
|
||||||
document.getElementById('content').innerHTML = marked.parse(`{{ text | safe}}`);
|
|
||||||
</script>
|
|
||||||
|
|
||||||
</body>
|
|
||||||
</html>
|
|
||||||
<style>
|
|
||||||
#content {
|
|
||||||
min-width: 200px;
|
|
||||||
max-width: 85%;
|
|
||||||
margin: 0 auto;
|
|
||||||
padding: 2rem 1em 1em;
|
|
||||||
}
|
|
||||||
|
|
||||||
body {
|
|
||||||
word-break: break-word;
|
|
||||||
line-height: 1.75;
|
|
||||||
font-weight: 400;
|
|
||||||
font-size: 32px;
|
|
||||||
margin: 0;
|
|
||||||
padding: 0;
|
|
||||||
overflow-x: hidden;
|
|
||||||
color: #333;
|
|
||||||
font-family: -apple-system,BlinkMacSystemFont,Segoe UI,Helvetica,Arial,sans-serif,Apple Color Emoji,Segoe UI Emoji;
|
|
||||||
}
|
|
||||||
h1, h2, h3, h4, h5, h6 {
|
|
||||||
line-height: 1.5;
|
|
||||||
margin-top: 35px;
|
|
||||||
margin-bottom: 10px;
|
|
||||||
padding-bottom: 5px;
|
|
||||||
}
|
|
||||||
h1:first-child, h2:first-child, h3:first-child, h4:first-child, h5:first-child, h6:first-child {
|
|
||||||
margin-top: -1.5rem;
|
|
||||||
margin-bottom: 1rem;
|
|
||||||
}
|
|
||||||
h1::before, h2::before, h3::before, h4::before, h5::before, h6::before {
|
|
||||||
content: "#";
|
|
||||||
display: inline-block;
|
|
||||||
color: #3eaf7c;
|
|
||||||
padding-right: 0.23em;
|
|
||||||
}
|
|
||||||
h1 {
|
|
||||||
position: relative;
|
|
||||||
font-size: 2.5rem;
|
|
||||||
margin-bottom: 5px;
|
|
||||||
}
|
|
||||||
h1::before {
|
|
||||||
font-size: 2.5rem;
|
|
||||||
}
|
|
||||||
h2 {
|
|
||||||
padding-bottom: 0.5rem;
|
|
||||||
font-size: 2.2rem;
|
|
||||||
border-bottom: 1px solid #ececec;
|
|
||||||
}
|
|
||||||
h3 {
|
|
||||||
font-size: 1.5rem;
|
|
||||||
padding-bottom: 0;
|
|
||||||
}
|
|
||||||
h4 {
|
|
||||||
font-size: 1.25rem;
|
|
||||||
}
|
|
||||||
h5 {
|
|
||||||
font-size: 1rem;
|
|
||||||
}
|
|
||||||
h6 {
|
|
||||||
margin-top: 5px;
|
|
||||||
}
|
|
||||||
p {
|
|
||||||
line-height: inherit;
|
|
||||||
margin-top: 22px;
|
|
||||||
margin-bottom: 22px;
|
|
||||||
}
|
|
||||||
strong {
|
|
||||||
color: #3eaf7c;
|
|
||||||
}
|
|
||||||
img {
|
|
||||||
max-width: 100%;
|
|
||||||
border-radius: 2px;
|
|
||||||
display: block;
|
|
||||||
margin: auto;
|
|
||||||
border: 3px solid rgba(62, 175, 124, 0.2);
|
|
||||||
}
|
|
||||||
hr {
|
|
||||||
border-top: 1px solid #3eaf7c;
|
|
||||||
border-bottom: none;
|
|
||||||
border-left: none;
|
|
||||||
border-right: none;
|
|
||||||
margin-top: 32px;
|
|
||||||
margin-bottom: 32px;
|
|
||||||
}
|
|
||||||
code {
|
|
||||||
font-family: Menlo, Monaco, Consolas, "Courier New", monospace;
|
|
||||||
word-break: break-word;
|
|
||||||
overflow-x: auto;
|
|
||||||
padding: 0.2rem 0.5rem;
|
|
||||||
margin: 0;
|
|
||||||
color: #3eaf7c;
|
|
||||||
font-size: 0.85em;
|
|
||||||
background-color: rgba(27, 31, 35, 0.05);
|
|
||||||
border-radius: 3px;
|
|
||||||
}
|
|
||||||
pre {
|
|
||||||
font-family: Menlo, Monaco, Consolas, "Courier New", monospace;
|
|
||||||
overflow: auto;
|
|
||||||
position: relative;
|
|
||||||
line-height: 1.75;
|
|
||||||
border-radius: 6px;
|
|
||||||
border: 2px solid #3eaf7c;
|
|
||||||
}
|
|
||||||
pre > code {
|
|
||||||
font-size: 12px;
|
|
||||||
padding: 15px 12px;
|
|
||||||
margin: 0;
|
|
||||||
word-break: normal;
|
|
||||||
display: block;
|
|
||||||
overflow-x: auto;
|
|
||||||
color: #333;
|
|
||||||
background: #f8f8f8;
|
|
||||||
}
|
|
||||||
a {
|
|
||||||
font-weight: 500;
|
|
||||||
text-decoration: none;
|
|
||||||
color: #3eaf7c;
|
|
||||||
}
|
|
||||||
a:hover, a:active {
|
|
||||||
border-bottom: 1.5px solid #3eaf7c;
|
|
||||||
}
|
|
||||||
a:before {
|
|
||||||
content: "⇲";
|
|
||||||
}
|
|
||||||
table {
|
|
||||||
display: inline-block !important;
|
|
||||||
font-size: 12px;
|
|
||||||
width: auto;
|
|
||||||
max-width: 100%;
|
|
||||||
overflow: auto;
|
|
||||||
border: solid 1px #3eaf7c;
|
|
||||||
}
|
|
||||||
thead {
|
|
||||||
background: #3eaf7c;
|
|
||||||
color: #fff;
|
|
||||||
text-align: left;
|
|
||||||
}
|
|
||||||
tr:nth-child(2n) {
|
|
||||||
background-color: rgba(62, 175, 124, 0.2);
|
|
||||||
}
|
|
||||||
th, td {
|
|
||||||
padding: 12px 7px;
|
|
||||||
line-height: 24px;
|
|
||||||
}
|
|
||||||
td {
|
|
||||||
min-width: 120px;
|
|
||||||
}
|
|
||||||
blockquote {
|
|
||||||
color: #666;
|
|
||||||
padding: 1px 23px;
|
|
||||||
margin: 22px 0;
|
|
||||||
border-left: 0.5rem solid rgba(62, 175, 124, 0.6);
|
|
||||||
border-color: #42b983;
|
|
||||||
background-color: #f8f8f8;
|
|
||||||
}
|
|
||||||
blockquote::after {
|
|
||||||
display: block;
|
|
||||||
content: "";
|
|
||||||
}
|
|
||||||
blockquote > p {
|
|
||||||
margin: 10px 0;
|
|
||||||
}
|
|
||||||
details {
|
|
||||||
border: none;
|
|
||||||
outline: none;
|
|
||||||
border-left: 4px solid #3eaf7c;
|
|
||||||
padding-left: 10px;
|
|
||||||
margin-left: 4px;
|
|
||||||
}
|
|
||||||
details summary {
|
|
||||||
cursor: pointer;
|
|
||||||
border: none;
|
|
||||||
outline: none;
|
|
||||||
background: white;
|
|
||||||
margin: 0px -17px;
|
|
||||||
}
|
|
||||||
details summary::-webkit-details-marker {
|
|
||||||
color: #3eaf7c;
|
|
||||||
}
|
|
||||||
ol, ul {
|
|
||||||
padding-left: 28px;
|
|
||||||
}
|
|
||||||
ol li, ul li {
|
|
||||||
margin-bottom: 0;
|
|
||||||
list-style: inherit;
|
|
||||||
}
|
|
||||||
ol li .task-list-item, ul li .task-list-item {
|
|
||||||
list-style: none;
|
|
||||||
}
|
|
||||||
ol li .task-list-item ul, ul li .task-list-item ul, ol li .task-list-item ol, ul li .task-list-item ol {
|
|
||||||
margin-top: 0;
|
|
||||||
}
|
|
||||||
ol ul, ul ul, ol ol, ul ol {
|
|
||||||
margin-top: 3px;
|
|
||||||
}
|
|
||||||
ol li {
|
|
||||||
padding-left: 6px;
|
|
||||||
}
|
|
||||||
ol li::marker {
|
|
||||||
color: #3eaf7c;
|
|
||||||
}
|
|
||||||
ul li {
|
|
||||||
list-style: none;
|
|
||||||
}
|
|
||||||
ul li:before {
|
|
||||||
content: "•";
|
|
||||||
margin-right: 4px;
|
|
||||||
color: #3eaf7c;
|
|
||||||
}
|
|
||||||
@media (max-width: 720px) {
|
|
||||||
h1 {
|
|
||||||
font-size: 24px;
|
|
||||||
}
|
|
||||||
h2 {
|
|
||||||
font-size: 20px;
|
|
||||||
}
|
|
||||||
h3 {
|
|
||||||
font-size: 18px;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
</style>
|
|
||||||
@@ -2,94 +2,111 @@
|
|||||||
|
|
||||||
import os
|
import os
|
||||||
import shutil
|
import shutil
|
||||||
from astrbot.core.utils.astrbot_path import get_astrbot_path
|
from astrbot.core.utils.astrbot_path import get_astrbot_data_path, get_astrbot_path
|
||||||
|
|
||||||
|
|
||||||
class TemplateManager:
|
class TemplateManager:
|
||||||
"""
|
"""
|
||||||
负责管理 t2i HTML 模板的 CRUD 和重置操作。
|
负责管理 t2i HTML 模板的 CRUD 和重置操作。
|
||||||
|
采用“用户覆盖内置”策略:用户模板存储在 data 目录中,并优先于内置模板加载。
|
||||||
|
所有创建、更新、删除操作仅影响用户目录,以确保更新框架时用户数据安全。
|
||||||
"""
|
"""
|
||||||
|
|
||||||
|
CORE_TEMPLATES = ["base.html", "astrbot_powershell.html"]
|
||||||
|
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
# 修正路径拼接,加入缺失的 'astrbot' 目录
|
self.builtin_template_dir = os.path.join(
|
||||||
self.template_dir = os.path.join(
|
|
||||||
get_astrbot_path(), "astrbot", "core", "utils", "t2i", "template"
|
get_astrbot_path(), "astrbot", "core", "utils", "t2i", "template"
|
||||||
)
|
)
|
||||||
self.backup_template_path = os.path.join(
|
self.user_template_dir = os.path.join(get_astrbot_data_path(), "t2i_templates")
|
||||||
self.template_dir, "default_template.html.bak"
|
|
||||||
)
|
|
||||||
# 确保模板目录存在
|
|
||||||
os.makedirs(self.template_dir, exist_ok=True)
|
|
||||||
|
|
||||||
# 检查模板目录中是否有 .html 文件
|
os.makedirs(self.user_template_dir, exist_ok=True)
|
||||||
html_files = [f for f in os.listdir(self.template_dir) if f.endswith(".html")]
|
self._initialize_user_templates()
|
||||||
if not html_files and os.path.exists(self.backup_template_path):
|
|
||||||
self.reset_default_template()
|
|
||||||
|
|
||||||
def _get_template_path(self, name: str) -> str:
|
def _copy_core_templates(self, overwrite: bool = False):
|
||||||
"""获取模板的完整路径,防止路径遍历漏洞。"""
|
"""从内置目录复制核心模板到用户目录。"""
|
||||||
|
for filename in self.CORE_TEMPLATES:
|
||||||
|
src = os.path.join(self.builtin_template_dir, filename)
|
||||||
|
dst = os.path.join(self.user_template_dir, filename)
|
||||||
|
if os.path.exists(src) and (overwrite or not os.path.exists(dst)):
|
||||||
|
shutil.copyfile(src, dst)
|
||||||
|
|
||||||
|
def _initialize_user_templates(self):
|
||||||
|
"""如果用户目录下缺少核心模板,则进行复制。"""
|
||||||
|
self._copy_core_templates(overwrite=False)
|
||||||
|
|
||||||
|
def _get_user_template_path(self, name: str) -> str:
|
||||||
|
"""获取用户模板的完整路径,防止路径遍历漏洞。"""
|
||||||
if ".." in name or "/" in name or "\\" in name:
|
if ".." in name or "/" in name or "\\" in name:
|
||||||
raise ValueError("模板名称包含非法字符。")
|
raise ValueError("模板名称包含非法字符。")
|
||||||
return os.path.join(self.template_dir, f"{name}.html")
|
return os.path.join(self.user_template_dir, f"{name}.html")
|
||||||
|
|
||||||
def list_templates(self) -> list[dict]:
|
def _read_file(self, path: str) -> str:
|
||||||
"""列出所有可用的模板。"""
|
"""读取文件内容。"""
|
||||||
templates = []
|
|
||||||
for filename in os.listdir(self.template_dir):
|
|
||||||
if filename.endswith(".html"):
|
|
||||||
templates.append(
|
|
||||||
{
|
|
||||||
"name": os.path.splitext(filename)[0],
|
|
||||||
"is_default": filename == "base.html",
|
|
||||||
}
|
|
||||||
)
|
|
||||||
return templates
|
|
||||||
|
|
||||||
def get_template(self, name: str) -> str:
|
|
||||||
"""获取指定模板的内容。"""
|
|
||||||
path = self._get_template_path(name)
|
|
||||||
if not os.path.exists(path):
|
|
||||||
raise FileNotFoundError("模板不存在。")
|
|
||||||
with open(path, "r", encoding="utf-8") as f:
|
with open(path, "r", encoding="utf-8") as f:
|
||||||
return f.read()
|
return f.read()
|
||||||
|
|
||||||
|
def list_templates(self) -> list[dict]:
|
||||||
|
"""
|
||||||
|
列出所有可用模板。
|
||||||
|
该列表是内置模板和用户模板的合并视图,用户模板将覆盖同名的内置模板。
|
||||||
|
"""
|
||||||
|
dirs_to_scan = [self.builtin_template_dir, self.user_template_dir]
|
||||||
|
all_names = {
|
||||||
|
os.path.splitext(f)[0]
|
||||||
|
for d in dirs_to_scan
|
||||||
|
for f in os.listdir(d)
|
||||||
|
if f.endswith(".html")
|
||||||
|
}
|
||||||
|
return [
|
||||||
|
{"name": name, "is_default": name == "base"} for name in sorted(all_names)
|
||||||
|
]
|
||||||
|
|
||||||
|
def get_template(self, name: str) -> str:
|
||||||
|
"""
|
||||||
|
获取指定模板的内容。
|
||||||
|
优先从用户目录加载,如果不存在则回退到内置目录。
|
||||||
|
"""
|
||||||
|
user_path = self._get_user_template_path(name)
|
||||||
|
if os.path.exists(user_path):
|
||||||
|
return self._read_file(user_path)
|
||||||
|
|
||||||
|
builtin_path = os.path.join(self.builtin_template_dir, f"{name}.html")
|
||||||
|
if os.path.exists(builtin_path):
|
||||||
|
return self._read_file(builtin_path)
|
||||||
|
|
||||||
|
raise FileNotFoundError("模板不存在。")
|
||||||
|
|
||||||
def create_template(self, name: str, content: str):
|
def create_template(self, name: str, content: str):
|
||||||
"""创建一个新的模板文件。"""
|
"""在用户目录中创建一个新的模板文件。"""
|
||||||
path = self._get_template_path(name)
|
path = self._get_user_template_path(name)
|
||||||
if os.path.exists(path):
|
if os.path.exists(path):
|
||||||
raise FileExistsError("同名模板已存在。")
|
raise FileExistsError("同名模板已存在。")
|
||||||
with open(path, "w", encoding="utf-8") as f:
|
with open(path, "w", encoding="utf-8") as f:
|
||||||
f.write(content)
|
f.write(content)
|
||||||
|
|
||||||
def update_template(self, name: str, content: str):
|
def update_template(self, name: str, content: str):
|
||||||
"""更新一个已存在的模板文件。"""
|
"""
|
||||||
path = self._get_template_path(name)
|
更新一个模板。此操作始终写入用户目录。
|
||||||
if not os.path.exists(path):
|
如果更新的是一个内置模板,此操作实际上会在用户目录中创建一个修改后的副本,
|
||||||
raise FileNotFoundError("模板不存在。")
|
从而实现对内置模板的“覆盖”。
|
||||||
|
"""
|
||||||
|
path = self._get_user_template_path(name)
|
||||||
with open(path, "w", encoding="utf-8") as f:
|
with open(path, "w", encoding="utf-8") as f:
|
||||||
f.write(content)
|
f.write(content)
|
||||||
|
|
||||||
def delete_template(self, name: str):
|
def delete_template(self, name: str):
|
||||||
"""删除一个模板文件。"""
|
"""
|
||||||
if name == "base":
|
仅删除用户目录中的模板文件。
|
||||||
raise ValueError("不能删除默认的 base 模板。")
|
如果删除的是一个覆盖了内置模板的用户模板,这将有效地“恢复”到内置版本。
|
||||||
path = self._get_template_path(name)
|
"""
|
||||||
|
path = self._get_user_template_path(name)
|
||||||
if not os.path.exists(path):
|
if not os.path.exists(path):
|
||||||
raise FileNotFoundError("模板不存在。")
|
raise FileNotFoundError("用户模板不存在,无法删除。")
|
||||||
os.remove(path)
|
os.remove(path)
|
||||||
|
|
||||||
def backup_default_template_if_not_exist(self):
|
|
||||||
"""如果备份不存在,则创建默认模板的备份。"""
|
|
||||||
default_path = os.path.join(self.template_dir, "base.html")
|
|
||||||
if not os.path.exists(self.backup_template_path) and os.path.exists(
|
|
||||||
default_path
|
|
||||||
):
|
|
||||||
shutil.copyfile(default_path, self.backup_template_path)
|
|
||||||
|
|
||||||
def reset_default_template(self):
|
def reset_default_template(self):
|
||||||
"""重置默认模板。"""
|
"""
|
||||||
if not os.path.exists(self.backup_template_path):
|
将核心模板从内置目录强制重置到用户目录。
|
||||||
raise FileNotFoundError("默认模板的备份文件不存在,无法重置。")
|
"""
|
||||||
|
self._copy_core_templates(overwrite=True)
|
||||||
default_path = os.path.join(self.template_dir, "base.html")
|
|
||||||
shutil.copyfile(self.backup_template_path, default_path)
|
|
||||||
|
|||||||
@@ -1,17 +1,27 @@
|
|||||||
import uuid
|
import uuid
|
||||||
import json
|
import json
|
||||||
import os
|
import os
|
||||||
|
import asyncio
|
||||||
|
from contextlib import asynccontextmanager
|
||||||
from .route import Route, Response, RouteContext
|
from .route import Route, Response, RouteContext
|
||||||
from astrbot.core.platform.sources.webchat.webchat_queue_mgr import webchat_queue_mgr
|
from astrbot.core.platform.sources.webchat.webchat_queue_mgr import webchat_queue_mgr
|
||||||
from quart import request, Response as QuartResponse, g, make_response
|
from quart import request, Response as QuartResponse, g, make_response
|
||||||
from astrbot.core.db import BaseDatabase
|
from astrbot.core.db import BaseDatabase
|
||||||
import asyncio
|
|
||||||
from astrbot.core import logger
|
from astrbot.core import logger
|
||||||
from astrbot.core.core_lifecycle import AstrBotCoreLifecycle
|
from astrbot.core.core_lifecycle import AstrBotCoreLifecycle
|
||||||
from astrbot.core.utils.astrbot_path import get_astrbot_data_path
|
from astrbot.core.utils.astrbot_path import get_astrbot_data_path
|
||||||
from astrbot.core.platform.astr_message_event import MessageSession
|
from astrbot.core.platform.astr_message_event import MessageSession
|
||||||
|
|
||||||
|
|
||||||
|
@asynccontextmanager
|
||||||
|
async def track_conversation(convs: dict, conv_id: str):
|
||||||
|
convs[conv_id] = True
|
||||||
|
try:
|
||||||
|
yield
|
||||||
|
finally:
|
||||||
|
convs.pop(conv_id, None)
|
||||||
|
|
||||||
|
|
||||||
class ChatRoute(Route):
|
class ChatRoute(Route):
|
||||||
def __init__(
|
def __init__(
|
||||||
self,
|
self,
|
||||||
@@ -40,6 +50,8 @@ class ChatRoute(Route):
|
|||||||
self.conv_mgr = core_lifecycle.conversation_manager
|
self.conv_mgr = core_lifecycle.conversation_manager
|
||||||
self.platform_history_mgr = core_lifecycle.platform_message_history_manager
|
self.platform_history_mgr = core_lifecycle.platform_message_history_manager
|
||||||
|
|
||||||
|
self.running_convs: dict[str, bool] = {}
|
||||||
|
|
||||||
async def get_file(self):
|
async def get_file(self):
|
||||||
filename = request.args.get("filename")
|
filename = request.args.get("filename")
|
||||||
if not filename:
|
if not filename:
|
||||||
@@ -139,12 +151,20 @@ class ChatRoute(Route):
|
|||||||
)
|
)
|
||||||
|
|
||||||
async def stream():
|
async def stream():
|
||||||
|
client_disconnected = False
|
||||||
|
|
||||||
try:
|
try:
|
||||||
|
async with track_conversation(self.running_convs, webchat_conv_id):
|
||||||
while True:
|
while True:
|
||||||
try:
|
try:
|
||||||
result = await asyncio.wait_for(back_queue.get(), timeout=10)
|
result = await asyncio.wait_for(back_queue.get(), timeout=1)
|
||||||
except asyncio.TimeoutError:
|
except asyncio.TimeoutError:
|
||||||
continue
|
continue
|
||||||
|
except asyncio.CancelledError:
|
||||||
|
logger.debug(f"[WebChat] 用户 {username} 断开聊天长连接。")
|
||||||
|
client_disconnected = True
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"WebChat stream error: {e}")
|
||||||
|
|
||||||
if not result:
|
if not result:
|
||||||
continue
|
continue
|
||||||
@@ -152,12 +172,31 @@ class ChatRoute(Route):
|
|||||||
result_text = result["data"]
|
result_text = result["data"]
|
||||||
type = result.get("type")
|
type = result.get("type")
|
||||||
streaming = result.get("streaming", False)
|
streaming = result.get("streaming", False)
|
||||||
|
|
||||||
|
try:
|
||||||
|
if not client_disconnected:
|
||||||
yield f"data: {json.dumps(result, ensure_ascii=False)}\n\n"
|
yield f"data: {json.dumps(result, ensure_ascii=False)}\n\n"
|
||||||
|
except Exception as e:
|
||||||
|
if not client_disconnected:
|
||||||
|
logger.debug(
|
||||||
|
f"[WebChat] 用户 {username} 断开聊天长连接。 {e}"
|
||||||
|
)
|
||||||
|
client_disconnected = True
|
||||||
|
|
||||||
|
try:
|
||||||
|
if not client_disconnected:
|
||||||
await asyncio.sleep(0.05)
|
await asyncio.sleep(0.05)
|
||||||
|
except asyncio.CancelledError:
|
||||||
|
logger.debug(f"[WebChat] 用户 {username} 断开聊天长连接。")
|
||||||
|
client_disconnected = True
|
||||||
|
|
||||||
if type == "end":
|
if type == "end":
|
||||||
break
|
break
|
||||||
elif (streaming and type == "complete") or not streaming:
|
elif (
|
||||||
|
(streaming and type == "complete")
|
||||||
|
or not streaming
|
||||||
|
or type == "break"
|
||||||
|
):
|
||||||
# append bot message
|
# append bot message
|
||||||
new_his = {"type": "bot", "message": result_text}
|
new_his = {"type": "bot", "message": result_text}
|
||||||
await self.platform_history_mgr.insert(
|
await self.platform_history_mgr.insert(
|
||||||
@@ -167,10 +206,8 @@ class ChatRoute(Route):
|
|||||||
sender_id="bot",
|
sender_id="bot",
|
||||||
sender_name="bot",
|
sender_name="bot",
|
||||||
)
|
)
|
||||||
|
except BaseException as e:
|
||||||
except BaseException as _:
|
logger.exception(f"WebChat stream unexpected error: {e}", exc_info=True)
|
||||||
logger.debug(f"用户 {username} 断开聊天长连接。")
|
|
||||||
return
|
|
||||||
|
|
||||||
# Put message to conversation-specific queue
|
# Put message to conversation-specific queue
|
||||||
chat_queue = webchat_queue_mgr.get_or_create_queue(webchat_conv_id)
|
chat_queue = webchat_queue_mgr.get_or_create_queue(webchat_conv_id)
|
||||||
@@ -197,6 +234,7 @@ class ChatRoute(Route):
|
|||||||
"Connection": "keep-alive",
|
"Connection": "keep-alive",
|
||||||
},
|
},
|
||||||
)
|
)
|
||||||
|
response.timeout = None # fix SSE auto disconnect issue
|
||||||
return response
|
return response
|
||||||
|
|
||||||
async def _get_webchat_conv_id_from_conv_id(self, conversation_id: str) -> str:
|
async def _get_webchat_conv_id_from_conv_id(self, conversation_id: str) -> str:
|
||||||
@@ -286,6 +324,7 @@ class ChatRoute(Route):
|
|||||||
.ok(
|
.ok(
|
||||||
data={
|
data={
|
||||||
"history": history_res,
|
"history": history_res,
|
||||||
|
"is_running": self.running_convs.get(webchat_conv_id, False),
|
||||||
}
|
}
|
||||||
)
|
)
|
||||||
.__dict__
|
.__dict__
|
||||||
|
|||||||
@@ -1,7 +1,7 @@
|
|||||||
import typing
|
import typing
|
||||||
import traceback
|
import traceback
|
||||||
import os
|
import os
|
||||||
import copy
|
import inspect
|
||||||
from .route import Route, Response, RouteContext
|
from .route import Route, Response, RouteContext
|
||||||
from astrbot.core.provider.entities import ProviderType
|
from astrbot.core.provider.entities import ProviderType
|
||||||
from quart import request
|
from quart import request
|
||||||
@@ -14,10 +14,10 @@ from astrbot.core.config.default import (
|
|||||||
from astrbot.core.utils.astrbot_path import get_astrbot_path
|
from astrbot.core.utils.astrbot_path import get_astrbot_path
|
||||||
from astrbot.core.config.astrbot_config import AstrBotConfig
|
from astrbot.core.config.astrbot_config import AstrBotConfig
|
||||||
from astrbot.core.core_lifecycle import AstrBotCoreLifecycle
|
from astrbot.core.core_lifecycle import AstrBotCoreLifecycle
|
||||||
from astrbot.core.platform.register import platform_registry
|
from astrbot.core.platform.register import platform_registry, platform_cls_map
|
||||||
from astrbot.core.provider.register import provider_registry
|
from astrbot.core.provider.register import provider_registry
|
||||||
from astrbot.core.star.star import star_registry
|
from astrbot.core.star.star import star_registry
|
||||||
from astrbot.core import logger
|
from astrbot.core import logger, file_token_service
|
||||||
from astrbot.core.provider import Provider
|
from astrbot.core.provider import Provider
|
||||||
from astrbot.core.provider.provider import RerankProvider
|
from astrbot.core.provider.provider import RerankProvider
|
||||||
import asyncio
|
import asyncio
|
||||||
@@ -52,24 +52,6 @@ def validate_config(
|
|||||||
def validate(data: dict, metadata: dict = schema, path=""):
|
def validate(data: dict, metadata: dict = schema, path=""):
|
||||||
for key, value in data.items():
|
for key, value in data.items():
|
||||||
if key not in metadata:
|
if key not in metadata:
|
||||||
# 无 schema 的配置项,执行类型猜测
|
|
||||||
if isinstance(value, str):
|
|
||||||
try:
|
|
||||||
data[key] = int(value)
|
|
||||||
continue
|
|
||||||
except ValueError:
|
|
||||||
pass
|
|
||||||
|
|
||||||
try:
|
|
||||||
data[key] = float(value)
|
|
||||||
continue
|
|
||||||
except ValueError:
|
|
||||||
pass
|
|
||||||
|
|
||||||
if value.lower() == "true":
|
|
||||||
data[key] = True
|
|
||||||
elif value.lower() == "false":
|
|
||||||
data[key] = False
|
|
||||||
continue
|
continue
|
||||||
meta = metadata[key]
|
meta = metadata[key]
|
||||||
if "type" not in meta:
|
if "type" not in meta:
|
||||||
@@ -128,12 +110,12 @@ def validate_config(
|
|||||||
)
|
)
|
||||||
|
|
||||||
if is_core:
|
if is_core:
|
||||||
for key, group in schema.items():
|
meta_all = {
|
||||||
group_meta = group.get("metadata")
|
**schema["platform_group"]["metadata"],
|
||||||
if not group_meta:
|
**schema["provider_group"]["metadata"],
|
||||||
continue
|
**schema["misc_config_group"]["metadata"],
|
||||||
# logger.info(f"验证配置: 组 {key} ...")
|
}
|
||||||
validate(data, group_meta, path=f"{key}.")
|
validate(data, meta_all)
|
||||||
else:
|
else:
|
||||||
validate(data, schema)
|
validate(data, schema)
|
||||||
|
|
||||||
@@ -143,6 +125,7 @@ def validate_config(
|
|||||||
def save_config(post_config: dict, config: AstrBotConfig, is_core: bool = False):
|
def save_config(post_config: dict, config: AstrBotConfig, is_core: bool = False):
|
||||||
"""验证并保存配置"""
|
"""验证并保存配置"""
|
||||||
errors = None
|
errors = None
|
||||||
|
logger.info(f"Saving config, is_core={is_core}")
|
||||||
try:
|
try:
|
||||||
if is_core:
|
if is_core:
|
||||||
errors, post_config = validate_config(
|
errors, post_config = validate_config(
|
||||||
@@ -167,6 +150,7 @@ class ConfigRoute(Route):
|
|||||||
super().__init__(context)
|
super().__init__(context)
|
||||||
self.core_lifecycle = core_lifecycle
|
self.core_lifecycle = core_lifecycle
|
||||||
self.config: AstrBotConfig = core_lifecycle.astrbot_config
|
self.config: AstrBotConfig = core_lifecycle.astrbot_config
|
||||||
|
self._logo_token_cache = {} # 缓存logo token,避免重复注册
|
||||||
self.acm = core_lifecycle.astrbot_config_mgr
|
self.acm = core_lifecycle.astrbot_config_mgr
|
||||||
self.routes = {
|
self.routes = {
|
||||||
"/config/abconf/new": ("POST", self.create_abconf),
|
"/config/abconf/new": ("POST", self.create_abconf),
|
||||||
@@ -673,6 +657,78 @@ class ConfigRoute(Route):
|
|||||||
return Response().error(str(e)).__dict__
|
return Response().error(str(e)).__dict__
|
||||||
return Response().ok(None, "删除成功,已经实时生效~").__dict__
|
return Response().ok(None, "删除成功,已经实时生效~").__dict__
|
||||||
|
|
||||||
|
async def get_llm_tools(self):
|
||||||
|
"""获取函数调用工具。包含了本地加载的以及 MCP 服务的工具"""
|
||||||
|
tool_mgr = self.core_lifecycle.provider_manager.llm_tools
|
||||||
|
tools = tool_mgr.get_func_desc_openai_style()
|
||||||
|
return Response().ok(tools).__dict__
|
||||||
|
|
||||||
|
async def _register_platform_logo(self, platform, platform_default_tmpl):
|
||||||
|
"""注册平台logo文件并生成访问令牌"""
|
||||||
|
if not platform.logo_path:
|
||||||
|
return
|
||||||
|
|
||||||
|
try:
|
||||||
|
# 检查缓存
|
||||||
|
cache_key = f"{platform.name}:{platform.logo_path}"
|
||||||
|
if cache_key in self._logo_token_cache:
|
||||||
|
cached_token = self._logo_token_cache[cache_key]
|
||||||
|
# 确保platform_default_tmpl[platform.name]存在且为字典
|
||||||
|
if platform.name not in platform_default_tmpl:
|
||||||
|
platform_default_tmpl[platform.name] = {}
|
||||||
|
elif not isinstance(platform_default_tmpl[platform.name], dict):
|
||||||
|
platform_default_tmpl[platform.name] = {}
|
||||||
|
platform_default_tmpl[platform.name]["logo_token"] = cached_token
|
||||||
|
logger.debug(f"Using cached logo token for platform {platform.name}")
|
||||||
|
return
|
||||||
|
|
||||||
|
# 获取平台适配器类
|
||||||
|
platform_cls = platform_cls_map.get(platform.name)
|
||||||
|
if not platform_cls:
|
||||||
|
logger.warning(f"Platform class not found for {platform.name}")
|
||||||
|
return
|
||||||
|
|
||||||
|
# 获取插件目录路径
|
||||||
|
module_file = inspect.getfile(platform_cls)
|
||||||
|
plugin_dir = os.path.dirname(module_file)
|
||||||
|
|
||||||
|
# 解析logo文件路径
|
||||||
|
logo_file_path = os.path.join(plugin_dir, platform.logo_path)
|
||||||
|
|
||||||
|
# 检查文件是否存在并注册令牌
|
||||||
|
if os.path.exists(logo_file_path):
|
||||||
|
logo_token = await file_token_service.register_file(
|
||||||
|
logo_file_path, timeout=3600
|
||||||
|
)
|
||||||
|
|
||||||
|
# 确保platform_default_tmpl[platform.name]存在且为字典
|
||||||
|
if platform.name not in platform_default_tmpl:
|
||||||
|
platform_default_tmpl[platform.name] = {}
|
||||||
|
elif not isinstance(platform_default_tmpl[platform.name], dict):
|
||||||
|
platform_default_tmpl[platform.name] = {}
|
||||||
|
|
||||||
|
platform_default_tmpl[platform.name]["logo_token"] = logo_token
|
||||||
|
|
||||||
|
# 缓存token
|
||||||
|
self._logo_token_cache[cache_key] = logo_token
|
||||||
|
|
||||||
|
logger.debug(f"Logo token registered for platform {platform.name}")
|
||||||
|
else:
|
||||||
|
logger.warning(
|
||||||
|
f"Platform {platform.name} logo file not found: {logo_file_path}"
|
||||||
|
)
|
||||||
|
|
||||||
|
except (ImportError, AttributeError) as e:
|
||||||
|
logger.warning(
|
||||||
|
f"Failed to import required modules for platform {platform.name}: {e}"
|
||||||
|
)
|
||||||
|
except (OSError, IOError) as e:
|
||||||
|
logger.warning(f"File system error for platform {platform.name} logo: {e}")
|
||||||
|
except Exception as e:
|
||||||
|
logger.warning(
|
||||||
|
f"Unexpected error registering logo for platform {platform.name}: {e}"
|
||||||
|
)
|
||||||
|
|
||||||
async def _get_astrbot_config(self):
|
async def _get_astrbot_config(self):
|
||||||
config = self.config
|
config = self.config
|
||||||
|
|
||||||
@@ -680,9 +736,21 @@ class ConfigRoute(Route):
|
|||||||
platform_default_tmpl = CONFIG_METADATA_2["platform_group"]["metadata"][
|
platform_default_tmpl = CONFIG_METADATA_2["platform_group"]["metadata"][
|
||||||
"platform"
|
"platform"
|
||||||
]["config_template"]
|
]["config_template"]
|
||||||
|
|
||||||
|
# 收集需要注册logo的平台
|
||||||
|
logo_registration_tasks = []
|
||||||
for platform in platform_registry:
|
for platform in platform_registry:
|
||||||
if platform.default_config_tmpl:
|
if platform.default_config_tmpl:
|
||||||
platform_default_tmpl[platform.name] = platform.default_config_tmpl
|
platform_default_tmpl[platform.name] = platform.default_config_tmpl
|
||||||
|
# 收集logo注册任务
|
||||||
|
if platform.logo_path:
|
||||||
|
logo_registration_tasks.append(
|
||||||
|
self._register_platform_logo(platform, platform_default_tmpl)
|
||||||
|
)
|
||||||
|
|
||||||
|
# 并行执行logo注册
|
||||||
|
if logo_registration_tasks:
|
||||||
|
await asyncio.gather(*logo_registration_tasks, return_exceptions=True)
|
||||||
|
|
||||||
# 服务提供商的默认配置模板注入
|
# 服务提供商的默认配置模板注入
|
||||||
provider_default_tmpl = CONFIG_METADATA_2["provider_group"]["metadata"][
|
provider_default_tmpl = CONFIG_METADATA_2["provider_group"]["metadata"][
|
||||||
|
|||||||
@@ -169,11 +169,61 @@ class ConversationRoute(Route):
|
|||||||
"""删除对话"""
|
"""删除对话"""
|
||||||
try:
|
try:
|
||||||
data = await request.get_json()
|
data = await request.get_json()
|
||||||
|
|
||||||
|
# 检查是否是批量删除
|
||||||
|
if "conversations" in data:
|
||||||
|
# 批量删除
|
||||||
|
conversations = data.get("conversations", [])
|
||||||
|
if not conversations:
|
||||||
|
return (
|
||||||
|
Response().error("批量删除时conversations参数不能为空").__dict__
|
||||||
|
)
|
||||||
|
|
||||||
|
deleted_count = 0
|
||||||
|
failed_items = []
|
||||||
|
|
||||||
|
for conv in conversations:
|
||||||
|
user_id = conv.get("user_id")
|
||||||
|
cid = conv.get("cid")
|
||||||
|
|
||||||
|
if not user_id or not cid:
|
||||||
|
failed_items.append(
|
||||||
|
f"user_id:{user_id}, cid:{cid} - 缺少必要参数"
|
||||||
|
)
|
||||||
|
continue
|
||||||
|
|
||||||
|
try:
|
||||||
|
await self.core_lifecycle.conversation_manager.delete_conversation(
|
||||||
|
unified_msg_origin=user_id, conversation_id=cid
|
||||||
|
)
|
||||||
|
deleted_count += 1
|
||||||
|
except Exception as e:
|
||||||
|
failed_items.append(f"user_id:{user_id}, cid:{cid} - {str(e)}")
|
||||||
|
|
||||||
|
message = f"成功删除 {deleted_count} 个对话"
|
||||||
|
if failed_items:
|
||||||
|
message += f",失败 {len(failed_items)} 个"
|
||||||
|
|
||||||
|
return (
|
||||||
|
Response()
|
||||||
|
.ok(
|
||||||
|
{
|
||||||
|
"message": message,
|
||||||
|
"deleted_count": deleted_count,
|
||||||
|
"failed_count": len(failed_items),
|
||||||
|
"failed_items": failed_items,
|
||||||
|
}
|
||||||
|
)
|
||||||
|
.__dict__
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
# 单个删除
|
||||||
user_id = data.get("user_id")
|
user_id = data.get("user_id")
|
||||||
cid = data.get("cid")
|
cid = data.get("cid")
|
||||||
|
|
||||||
if not user_id or not cid:
|
if not user_id or not cid:
|
||||||
return Response().error("缺少必要参数: user_id 和 cid").__dict__
|
return Response().error("缺少必要参数: user_id 和 cid").__dict__
|
||||||
|
|
||||||
await self.core_lifecycle.conversation_manager.delete_conversation(
|
await self.core_lifecycle.conversation_manager.delete_conversation(
|
||||||
unified_msg_origin=user_id, conversation_id=cid
|
unified_msg_origin=user_id, conversation_id=cid
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -10,7 +10,9 @@ class LogRoute(Route):
|
|||||||
super().__init__(context)
|
super().__init__(context)
|
||||||
self.log_broker = log_broker
|
self.log_broker = log_broker
|
||||||
self.app.add_url_rule("/api/live-log", view_func=self.log, methods=["GET"])
|
self.app.add_url_rule("/api/live-log", view_func=self.log, methods=["GET"])
|
||||||
self.app.add_url_rule("/api/log-history", view_func=self.log_history, methods=["GET"])
|
self.app.add_url_rule(
|
||||||
|
"/api/log-history", view_func=self.log_history, methods=["GET"]
|
||||||
|
)
|
||||||
|
|
||||||
async def log(self):
|
async def log(self):
|
||||||
async def stream():
|
async def stream():
|
||||||
@@ -48,9 +50,15 @@ class LogRoute(Route):
|
|||||||
"""获取日志历史"""
|
"""获取日志历史"""
|
||||||
try:
|
try:
|
||||||
logs = list(self.log_broker.log_cache)
|
logs = list(self.log_broker.log_cache)
|
||||||
return Response().ok(data={
|
return (
|
||||||
|
Response()
|
||||||
|
.ok(
|
||||||
|
data={
|
||||||
"logs": logs,
|
"logs": logs,
|
||||||
}).__dict__
|
}
|
||||||
|
)
|
||||||
|
.__dict__
|
||||||
|
)
|
||||||
except BaseException as e:
|
except BaseException as e:
|
||||||
logger.error(f"获取日志历史失败: {e}")
|
logger.error(f"获取日志历史失败: {e}")
|
||||||
return Response().error(f"获取日志历史失败: {e}").__dict__
|
return Response().error(f"获取日志历史失败: {e}").__dict__
|
||||||
|
|||||||
@@ -1,4 +1,3 @@
|
|||||||
from astrbot.core import logger
|
|
||||||
from astrbot.core.config.astrbot_config import AstrBotConfig
|
from astrbot.core.config.astrbot_config import AstrBotConfig
|
||||||
from dataclasses import dataclass
|
from dataclasses import dataclass
|
||||||
from quart import Quart
|
from quart import Quart
|
||||||
|
|||||||
Some files were not shown because too many files have changed in this diff Show More
Reference in New Issue
Block a user