mirror of
https://github.com/langbot-app/LangBot.git
synced 2025-11-25 19:37:36 +08:00
Compare commits
31 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
d0aed48ca9 | ||
|
|
bf548df6ae | ||
|
|
a3fe105f8e | ||
|
|
5add1d71bc | ||
|
|
7a01cff0c8 | ||
|
|
e8602f7134 | ||
|
|
e9aad2c8d7 | ||
|
|
60d4f3d77c | ||
|
|
9b8c5a3499 | ||
|
|
53dde0607d | ||
|
|
7f034b4ffa | ||
|
|
599ab83100 | ||
|
|
f4a3508ec2 | ||
|
|
44b92909eb | ||
|
|
8ed07b8d1a | ||
|
|
2ff9ced15e | ||
|
|
641b8d71ed | ||
|
|
a31b450f54 | ||
|
|
97bb24c5b9 | ||
|
|
5e5a3639d1 | ||
|
|
0a68a77e28 | ||
|
|
11a0c4142e | ||
|
|
d214d80579 | ||
|
|
ed719fd44e | ||
|
|
5dc6bed0d1 | ||
|
|
b1244a4d4e | ||
|
|
6aa325a4b1 | ||
|
|
88a11561f9 | ||
|
|
fd30022065 | ||
|
|
9486312737 | ||
|
|
e37070a985 |
5
.github/workflows/build-dev-image.yaml
vendored
5
.github/workflows/build-dev-image.yaml
vendored
@@ -7,9 +7,14 @@ on:
|
|||||||
jobs:
|
jobs:
|
||||||
build-dev-image:
|
build-dev-image:
|
||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
|
# 如果是tag则跳过
|
||||||
|
if: ${{ !startsWith(github.ref, 'refs/tags/') }}
|
||||||
steps:
|
steps:
|
||||||
- name: Checkout
|
- name: Checkout
|
||||||
uses: actions/checkout@v2
|
uses: actions/checkout@v2
|
||||||
|
with:
|
||||||
|
persist-credentials: false
|
||||||
|
|
||||||
- name: Generate Tag
|
- name: Generate Tag
|
||||||
id: generate_tag
|
id: generate_tag
|
||||||
run: |
|
run: |
|
||||||
|
|||||||
3
.github/workflows/build-docker-image.yml
vendored
3
.github/workflows/build-docker-image.yml
vendored
@@ -13,6 +13,9 @@ jobs:
|
|||||||
steps:
|
steps:
|
||||||
- name: Checkout
|
- name: Checkout
|
||||||
uses: actions/checkout@v2
|
uses: actions/checkout@v2
|
||||||
|
with:
|
||||||
|
persist-credentials: false
|
||||||
|
|
||||||
- name: judge has env GITHUB_REF # 如果没有GITHUB_REF环境变量,则把github.ref变量赋值给GITHUB_REF
|
- name: judge has env GITHUB_REF # 如果没有GITHUB_REF环境变量,则把github.ref变量赋值给GITHUB_REF
|
||||||
run: |
|
run: |
|
||||||
if [ -z "$GITHUB_REF" ]; then
|
if [ -z "$GITHUB_REF" ]; then
|
||||||
|
|||||||
10
.github/workflows/build-release-artifacts.yaml
vendored
10
.github/workflows/build-release-artifacts.yaml
vendored
@@ -12,6 +12,8 @@ jobs:
|
|||||||
steps:
|
steps:
|
||||||
- name: Checkout
|
- name: Checkout
|
||||||
uses: actions/checkout@v2
|
uses: actions/checkout@v2
|
||||||
|
with:
|
||||||
|
persist-credentials: false
|
||||||
|
|
||||||
- name: Check version
|
- name: Check version
|
||||||
id: check_version
|
id: check_version
|
||||||
@@ -50,3 +52,11 @@ jobs:
|
|||||||
with:
|
with:
|
||||||
name: langbot-${{ steps.check_version.outputs.version }}-all
|
name: langbot-${{ steps.check_version.outputs.version }}-all
|
||||||
path: .
|
path: .
|
||||||
|
|
||||||
|
- name: Upload To Release
|
||||||
|
env:
|
||||||
|
GH_TOKEN: ${{ secrets.RELEASE_UPLOAD_GITHUB_TOKEN }}
|
||||||
|
run: |
|
||||||
|
# 本目录下所有文件打包成zip
|
||||||
|
zip -r langbot-${{ steps.check_version.outputs.version }}-all.zip .
|
||||||
|
gh release upload ${{ github.event.release.tag_name }} langbot-${{ steps.check_version.outputs.version }}-all.zip
|
||||||
|
|||||||
80
.github/workflows/test-pr.yml
vendored
80
.github/workflows/test-pr.yml
vendored
@@ -1,80 +0,0 @@
|
|||||||
name: Test Pull Request
|
|
||||||
|
|
||||||
on:
|
|
||||||
pull_request:
|
|
||||||
types: [ready_for_review]
|
|
||||||
paths:
|
|
||||||
# 任何py文件改动都会触发
|
|
||||||
- '**.py'
|
|
||||||
pull_request_review:
|
|
||||||
types: [submitted]
|
|
||||||
issue_comment:
|
|
||||||
types: [created]
|
|
||||||
# 允许手动触发
|
|
||||||
workflow_dispatch:
|
|
||||||
|
|
||||||
jobs:
|
|
||||||
perform-test:
|
|
||||||
runs-on: ubuntu-latest
|
|
||||||
# 如果事件为pull_request_review且review状态为approved,则执行
|
|
||||||
if: >
|
|
||||||
github.event_name == 'pull_request' ||
|
|
||||||
(github.event_name == 'pull_request_review' && github.event.review.state == 'APPROVED') ||
|
|
||||||
github.event_name == 'workflow_dispatch' ||
|
|
||||||
(github.event_name == 'issue_comment' && github.event.issue.pull_request != '' && contains(github.event.comment.body, '/test') && github.event.comment.user.login == 'RockChinQ')
|
|
||||||
steps:
|
|
||||||
# 签出测试工程仓库代码
|
|
||||||
- name: Checkout
|
|
||||||
uses: actions/checkout@v2
|
|
||||||
with:
|
|
||||||
# 仓库地址
|
|
||||||
repository: RockChinQ/qcg-tester
|
|
||||||
# 仓库路径
|
|
||||||
path: qcg-tester
|
|
||||||
- name: Setup Python
|
|
||||||
uses: actions/setup-python@v2
|
|
||||||
with:
|
|
||||||
python-version: '3.10'
|
|
||||||
|
|
||||||
- name: Install dependencies
|
|
||||||
run: |
|
|
||||||
cd qcg-tester
|
|
||||||
python -m pip install --upgrade pip
|
|
||||||
pip install -r requirements.txt
|
|
||||||
|
|
||||||
- name: Get PR details
|
|
||||||
id: get-pr
|
|
||||||
if: github.event_name == 'issue_comment'
|
|
||||||
uses: octokit/request-action@v2.x
|
|
||||||
with:
|
|
||||||
route: GET /repos/${{ github.repository }}/pulls/${{ github.event.issue.number }}
|
|
||||||
env:
|
|
||||||
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
|
||||||
|
|
||||||
- name: Set PR source branch as env variable
|
|
||||||
if: github.event_name == 'issue_comment'
|
|
||||||
run: |
|
|
||||||
PR_SOURCE_BRANCH=$(echo '${{ steps.get-pr.outputs.data }}' | jq -r '.head.ref')
|
|
||||||
echo "BRANCH=$PR_SOURCE_BRANCH" >> $GITHUB_ENV
|
|
||||||
|
|
||||||
- name: Set PR Branch as bash env
|
|
||||||
if: github.event_name != 'issue_comment'
|
|
||||||
run: |
|
|
||||||
echo "BRANCH=${{ github.head_ref }}" >> $GITHUB_ENV
|
|
||||||
- name: Set OpenAI API Key from Secrets
|
|
||||||
run: |
|
|
||||||
echo "OPENAI_API_KEY=${{ secrets.OPENAI_API_KEY }}" >> $GITHUB_ENV
|
|
||||||
- name: Set OpenAI Reverse Proxy URL from Secrets
|
|
||||||
run: |
|
|
||||||
echo "OPENAI_REVERSE_PROXY=${{ secrets.OPENAI_REVERSE_PROXY }}" >> $GITHUB_ENV
|
|
||||||
- name: Run test
|
|
||||||
run: |
|
|
||||||
cd qcg-tester
|
|
||||||
python main.py
|
|
||||||
|
|
||||||
- name: Upload coverage reports to Codecov
|
|
||||||
run: |
|
|
||||||
cd qcg-tester/resource/QChatGPT
|
|
||||||
curl -Os https://uploader.codecov.io/latest/linux/codecov
|
|
||||||
chmod +x codecov
|
|
||||||
./codecov -t ${{ secrets.CODECOV_TOKEN }}
|
|
||||||
11
.gitignore
vendored
11
.gitignore
vendored
@@ -2,7 +2,6 @@
|
|||||||
.idea/
|
.idea/
|
||||||
__pycache__/
|
__pycache__/
|
||||||
database.db
|
database.db
|
||||||
qchatgpt.log
|
|
||||||
langbot.log
|
langbot.log
|
||||||
/banlist.py
|
/banlist.py
|
||||||
/plugins/
|
/plugins/
|
||||||
@@ -17,8 +16,7 @@ scenario/
|
|||||||
!scenario/default-template.json
|
!scenario/default-template.json
|
||||||
override.json
|
override.json
|
||||||
cookies.json
|
cookies.json
|
||||||
res/announcement_saved
|
data/labels/announcement_saved.json
|
||||||
res/announcement_saved.json
|
|
||||||
cmdpriv.json
|
cmdpriv.json
|
||||||
tips.py
|
tips.py
|
||||||
.venv
|
.venv
|
||||||
@@ -32,8 +30,11 @@ claude.json
|
|||||||
bard.json
|
bard.json
|
||||||
/*yaml
|
/*yaml
|
||||||
!/docker-compose.yaml
|
!/docker-compose.yaml
|
||||||
res/instance_id.json
|
data/labels/instance_id.json
|
||||||
.DS_Store
|
.DS_Store
|
||||||
/data
|
/data
|
||||||
botpy.log*
|
botpy.log*
|
||||||
/poc
|
/poc
|
||||||
|
/libs/wecom_api/test.py
|
||||||
|
/venv
|
||||||
|
|
||||||
|
|||||||
34
README.md
34
README.md
@@ -1,11 +1,8 @@
|
|||||||
> [!IMPORTANT]
|
|
||||||
> 我们被人在 X.com 和 pump.fun 上冒充了,以下两个账号利用本项目和作者信息在 X.com 上发布数字货币营销信息,请勿相信!我们已向 X 官方举报!我们从未以 LangBot 名义创建任何社交媒体账号或者数字货币。
|
|
||||||
> We have been impersonated on X.com and pump.fun . The following two accounts are using this project and author information to post digital currency marketing information on X.com. Please do not believe that! We have reported to X official! We have never created any social media account or digital currency under the name LangBot.
|
|
||||||
> 1. https://x.com/RockChinQ
|
|
||||||
> 2. https://x.com/LangBotAI
|
|
||||||
|
|
||||||
<p align="center">
|
<p align="center">
|
||||||
|
<a href="https://langbot.app">
|
||||||
<img src="https://docs.langbot.app/social.png" alt="LangBot"/>
|
<img src="https://docs.langbot.app/social.png" alt="LangBot"/>
|
||||||
|
</a>
|
||||||
|
|
||||||
<div align="center">
|
<div align="center">
|
||||||
|
|
||||||
@@ -82,3 +79,30 @@
|
|||||||
- WebUI Demo: https://demo.langbot.dev/
|
- WebUI Demo: https://demo.langbot.dev/
|
||||||
- 登录信息:邮箱:`demo@langbot.app` 密码:`langbot123456`
|
- 登录信息:邮箱:`demo@langbot.app` 密码:`langbot123456`
|
||||||
- 注意:仅展示webui效果,公开环境,请不要在其中填入您的任何敏感信息。
|
- 注意:仅展示webui效果,公开环境,请不要在其中填入您的任何敏感信息。
|
||||||
|
|
||||||
|
## 🔌 组件兼容性
|
||||||
|
|
||||||
|
### 消息平台
|
||||||
|
|
||||||
|
| 平台 | 状态 | 备注 |
|
||||||
|
| --- | --- | --- |
|
||||||
|
| QQ 个人号 | ✅ | QQ 个人号私聊、群聊 |
|
||||||
|
| QQ 官方机器人 | ✅ | QQ 频道机器人,支持频道、私聊、群聊 |
|
||||||
|
| 企业微信 | ✅ | |
|
||||||
|
| 钉钉 | 🚧 | |
|
||||||
|
|
||||||
|
🚧: 正在开发中
|
||||||
|
|
||||||
|
### 大模型
|
||||||
|
|
||||||
|
| 模型 | 状态 | 备注 |
|
||||||
|
| --- | --- | --- |
|
||||||
|
| [OpenAI](https://platform.openai.com/) | ✅ | 可接入任何 OpenAI 接口格式模型 |
|
||||||
|
| [DeepSeek](https://www.deepseek.com/) | ✅ | |
|
||||||
|
| [Moonshot](https://www.moonshot.cn/) | ✅ | |
|
||||||
|
| [Anthropic](https://www.anthropic.com/) | ✅ | |
|
||||||
|
| [xAI](https://x.ai/) | ✅ | |
|
||||||
|
| [智谱AI](https://open.bigmodel.cn/) | ✅ | |
|
||||||
|
| [Dify](https://dify.ai) | ✅ | LLMOps 平台 |
|
||||||
|
| [Ollama](https://ollama.com/) | ✅ | 本地大模型管理平台 |
|
||||||
|
| [GiteeAI](https://ai.gitee.com/) | ✅ | 大模型接口聚合平台 |
|
||||||
|
|||||||
279
libs/wecom_api/WXBizMsgCrypt3.py
Normal file
279
libs/wecom_api/WXBizMsgCrypt3.py
Normal file
@@ -0,0 +1,279 @@
|
|||||||
|
#!/usr/bin/env python
|
||||||
|
# -*- encoding:utf-8 -*-
|
||||||
|
|
||||||
|
""" 对企业微信发送给企业后台的消息加解密示例代码.
|
||||||
|
@copyright: Copyright (c) 1998-2014 Tencent Inc.
|
||||||
|
|
||||||
|
"""
|
||||||
|
# ------------------------------------------------------------------------
|
||||||
|
import logging
|
||||||
|
import base64
|
||||||
|
import random
|
||||||
|
import hashlib
|
||||||
|
import time
|
||||||
|
import struct
|
||||||
|
from Crypto.Cipher import AES
|
||||||
|
import xml.etree.cElementTree as ET
|
||||||
|
import socket
|
||||||
|
|
||||||
|
from . import ierror
|
||||||
|
|
||||||
|
|
||||||
|
"""
|
||||||
|
Crypto.Cipher包已不再维护,开发者可以通过以下命令下载安装最新版的加解密工具包
|
||||||
|
pip install pycryptodome
|
||||||
|
"""
|
||||||
|
|
||||||
|
|
||||||
|
class FormatException(Exception):
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
def throw_exception(message, exception_class=FormatException):
|
||||||
|
"""my define raise exception function"""
|
||||||
|
raise exception_class(message)
|
||||||
|
|
||||||
|
|
||||||
|
class SHA1:
|
||||||
|
"""计算企业微信的消息签名接口"""
|
||||||
|
|
||||||
|
def getSHA1(self, token, timestamp, nonce, encrypt):
|
||||||
|
"""用SHA1算法生成安全签名
|
||||||
|
@param token: 票据
|
||||||
|
@param timestamp: 时间戳
|
||||||
|
@param encrypt: 密文
|
||||||
|
@param nonce: 随机字符串
|
||||||
|
@return: 安全签名
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
sortlist = [token, timestamp, nonce, encrypt]
|
||||||
|
sortlist.sort()
|
||||||
|
sha = hashlib.sha1()
|
||||||
|
sha.update("".join(sortlist).encode())
|
||||||
|
return ierror.WXBizMsgCrypt_OK, sha.hexdigest()
|
||||||
|
except Exception as e:
|
||||||
|
logger = logging.getLogger()
|
||||||
|
logger.error(e)
|
||||||
|
return ierror.WXBizMsgCrypt_ComputeSignature_Error, None
|
||||||
|
|
||||||
|
|
||||||
|
class XMLParse:
|
||||||
|
"""提供提取消息格式中的密文及生成回复消息格式的接口"""
|
||||||
|
|
||||||
|
# xml消息模板
|
||||||
|
AES_TEXT_RESPONSE_TEMPLATE = """<xml>
|
||||||
|
<Encrypt><![CDATA[%(msg_encrypt)s]]></Encrypt>
|
||||||
|
<MsgSignature><![CDATA[%(msg_signaturet)s]]></MsgSignature>
|
||||||
|
<TimeStamp>%(timestamp)s</TimeStamp>
|
||||||
|
<Nonce><![CDATA[%(nonce)s]]></Nonce>
|
||||||
|
</xml>"""
|
||||||
|
|
||||||
|
def extract(self, xmltext):
|
||||||
|
"""提取出xml数据包中的加密消息
|
||||||
|
@param xmltext: 待提取的xml字符串
|
||||||
|
@return: 提取出的加密消息字符串
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
xml_tree = ET.fromstring(xmltext)
|
||||||
|
encrypt = xml_tree.find("Encrypt")
|
||||||
|
return ierror.WXBizMsgCrypt_OK, encrypt.text
|
||||||
|
except Exception as e:
|
||||||
|
logger = logging.getLogger()
|
||||||
|
logger.error(e)
|
||||||
|
return ierror.WXBizMsgCrypt_ParseXml_Error, None
|
||||||
|
|
||||||
|
def generate(self, encrypt, signature, timestamp, nonce):
|
||||||
|
"""生成xml消息
|
||||||
|
@param encrypt: 加密后的消息密文
|
||||||
|
@param signature: 安全签名
|
||||||
|
@param timestamp: 时间戳
|
||||||
|
@param nonce: 随机字符串
|
||||||
|
@return: 生成的xml字符串
|
||||||
|
"""
|
||||||
|
resp_dict = {
|
||||||
|
'msg_encrypt': encrypt,
|
||||||
|
'msg_signaturet': signature,
|
||||||
|
'timestamp': timestamp,
|
||||||
|
'nonce': nonce,
|
||||||
|
}
|
||||||
|
resp_xml = self.AES_TEXT_RESPONSE_TEMPLATE % resp_dict
|
||||||
|
return resp_xml
|
||||||
|
|
||||||
|
|
||||||
|
class PKCS7Encoder():
|
||||||
|
"""提供基于PKCS7算法的加解密接口"""
|
||||||
|
|
||||||
|
block_size = 32
|
||||||
|
|
||||||
|
def encode(self, text):
|
||||||
|
""" 对需要加密的明文进行填充补位
|
||||||
|
@param text: 需要进行填充补位操作的明文
|
||||||
|
@return: 补齐明文字符串
|
||||||
|
"""
|
||||||
|
text_length = len(text)
|
||||||
|
# 计算需要填充的位数
|
||||||
|
amount_to_pad = self.block_size - (text_length % self.block_size)
|
||||||
|
if amount_to_pad == 0:
|
||||||
|
amount_to_pad = self.block_size
|
||||||
|
# 获得补位所用的字符
|
||||||
|
pad = chr(amount_to_pad)
|
||||||
|
return text + (pad * amount_to_pad).encode()
|
||||||
|
|
||||||
|
def decode(self, decrypted):
|
||||||
|
"""删除解密后明文的补位字符
|
||||||
|
@param decrypted: 解密后的明文
|
||||||
|
@return: 删除补位字符后的明文
|
||||||
|
"""
|
||||||
|
pad = ord(decrypted[-1])
|
||||||
|
if pad < 1 or pad > 32:
|
||||||
|
pad = 0
|
||||||
|
return decrypted[:-pad]
|
||||||
|
|
||||||
|
|
||||||
|
class Prpcrypt(object):
|
||||||
|
"""提供接收和推送给企业微信消息的加解密接口"""
|
||||||
|
|
||||||
|
def __init__(self, key):
|
||||||
|
|
||||||
|
# self.key = base64.b64decode(key+"=")
|
||||||
|
self.key = key
|
||||||
|
# 设置加解密模式为AES的CBC模式
|
||||||
|
self.mode = AES.MODE_CBC
|
||||||
|
|
||||||
|
def encrypt(self, text, receiveid):
|
||||||
|
"""对明文进行加密
|
||||||
|
@param text: 需要加密的明文
|
||||||
|
@return: 加密得到的字符串
|
||||||
|
"""
|
||||||
|
# 16位随机字符串添加到明文开头
|
||||||
|
text = text.encode()
|
||||||
|
text = self.get_random_str() + struct.pack("I", socket.htonl(len(text))) + text + receiveid.encode()
|
||||||
|
|
||||||
|
# 使用自定义的填充方式对明文进行补位填充
|
||||||
|
pkcs7 = PKCS7Encoder()
|
||||||
|
text = pkcs7.encode(text)
|
||||||
|
# 加密
|
||||||
|
cryptor = AES.new(self.key, self.mode, self.key[:16])
|
||||||
|
try:
|
||||||
|
ciphertext = cryptor.encrypt(text)
|
||||||
|
# 使用BASE64对加密后的字符串进行编码
|
||||||
|
return ierror.WXBizMsgCrypt_OK, base64.b64encode(ciphertext)
|
||||||
|
except Exception as e:
|
||||||
|
logger = logging.getLogger()
|
||||||
|
logger.error(e)
|
||||||
|
return ierror.WXBizMsgCrypt_EncryptAES_Error, None
|
||||||
|
|
||||||
|
def decrypt(self, text, receiveid):
|
||||||
|
"""对解密后的明文进行补位删除
|
||||||
|
@param text: 密文
|
||||||
|
@return: 删除填充补位后的明文
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
cryptor = AES.new(self.key, self.mode, self.key[:16])
|
||||||
|
# 使用BASE64对密文进行解码,然后AES-CBC解密
|
||||||
|
plain_text = cryptor.decrypt(base64.b64decode(text))
|
||||||
|
except Exception as e:
|
||||||
|
logger = logging.getLogger()
|
||||||
|
logger.error(e)
|
||||||
|
return ierror.WXBizMsgCrypt_DecryptAES_Error, None
|
||||||
|
try:
|
||||||
|
pad = plain_text[-1]
|
||||||
|
# 去掉补位字符串
|
||||||
|
# pkcs7 = PKCS7Encoder()
|
||||||
|
# plain_text = pkcs7.encode(plain_text)
|
||||||
|
# 去除16位随机字符串
|
||||||
|
content = plain_text[16:-pad]
|
||||||
|
xml_len = socket.ntohl(struct.unpack("I", content[: 4])[0])
|
||||||
|
xml_content = content[4: xml_len + 4]
|
||||||
|
from_receiveid = content[xml_len + 4:]
|
||||||
|
except Exception as e:
|
||||||
|
logger = logging.getLogger()
|
||||||
|
logger.error(e)
|
||||||
|
return ierror.WXBizMsgCrypt_IllegalBuffer, None
|
||||||
|
|
||||||
|
if from_receiveid.decode('utf8') != receiveid:
|
||||||
|
return ierror.WXBizMsgCrypt_ValidateCorpid_Error, None
|
||||||
|
return 0, xml_content
|
||||||
|
|
||||||
|
def get_random_str(self):
|
||||||
|
""" 随机生成16位字符串
|
||||||
|
@return: 16位字符串
|
||||||
|
"""
|
||||||
|
return str(random.randint(1000000000000000, 9999999999999999)).encode()
|
||||||
|
|
||||||
|
|
||||||
|
class WXBizMsgCrypt(object):
|
||||||
|
# 构造函数
|
||||||
|
def __init__(self, sToken, sEncodingAESKey, sReceiveId):
|
||||||
|
try:
|
||||||
|
self.key = base64.b64decode(sEncodingAESKey + "=")
|
||||||
|
assert len(self.key) == 32
|
||||||
|
except:
|
||||||
|
throw_exception("[error]: EncodingAESKey unvalid !", FormatException)
|
||||||
|
# return ierror.WXBizMsgCrypt_IllegalAesKey,None
|
||||||
|
self.m_sToken = sToken
|
||||||
|
self.m_sReceiveId = sReceiveId
|
||||||
|
|
||||||
|
# 验证URL
|
||||||
|
# @param sMsgSignature: 签名串,对应URL参数的msg_signature
|
||||||
|
# @param sTimeStamp: 时间戳,对应URL参数的timestamp
|
||||||
|
# @param sNonce: 随机串,对应URL参数的nonce
|
||||||
|
# @param sEchoStr: 随机串,对应URL参数的echostr
|
||||||
|
# @param sReplyEchoStr: 解密之后的echostr,当return返回0时有效
|
||||||
|
# @return:成功0,失败返回对应的错误码
|
||||||
|
|
||||||
|
def VerifyURL(self, sMsgSignature, sTimeStamp, sNonce, sEchoStr):
|
||||||
|
sha1 = SHA1()
|
||||||
|
ret, signature = sha1.getSHA1(self.m_sToken, sTimeStamp, sNonce, sEchoStr)
|
||||||
|
if ret != 0:
|
||||||
|
return ret, None
|
||||||
|
if not signature == sMsgSignature:
|
||||||
|
return ierror.WXBizMsgCrypt_ValidateSignature_Error, None
|
||||||
|
pc = Prpcrypt(self.key)
|
||||||
|
ret, sReplyEchoStr = pc.decrypt(sEchoStr, self.m_sReceiveId)
|
||||||
|
return ret, sReplyEchoStr
|
||||||
|
|
||||||
|
def EncryptMsg(self, sReplyMsg, sNonce, timestamp=None):
|
||||||
|
# 将企业回复用户的消息加密打包
|
||||||
|
# @param sReplyMsg: 企业号待回复用户的消息,xml格式的字符串
|
||||||
|
# @param sTimeStamp: 时间戳,可以自己生成,也可以用URL参数的timestamp,如为None则自动用当前时间
|
||||||
|
# @param sNonce: 随机串,可以自己生成,也可以用URL参数的nonce
|
||||||
|
# sEncryptMsg: 加密后的可以直接回复用户的密文,包括msg_signature, timestamp, nonce, encrypt的xml格式的字符串,
|
||||||
|
# return:成功0,sEncryptMsg,失败返回对应的错误码None
|
||||||
|
pc = Prpcrypt(self.key)
|
||||||
|
ret, encrypt = pc.encrypt(sReplyMsg, self.m_sReceiveId)
|
||||||
|
encrypt = encrypt.decode('utf8')
|
||||||
|
if ret != 0:
|
||||||
|
return ret, None
|
||||||
|
if timestamp is None:
|
||||||
|
timestamp = str(int(time.time()))
|
||||||
|
# 生成安全签名
|
||||||
|
sha1 = SHA1()
|
||||||
|
ret, signature = sha1.getSHA1(self.m_sToken, timestamp, sNonce, encrypt)
|
||||||
|
if ret != 0:
|
||||||
|
return ret, None
|
||||||
|
xmlParse = XMLParse()
|
||||||
|
return ret, xmlParse.generate(encrypt, signature, timestamp, sNonce)
|
||||||
|
|
||||||
|
def DecryptMsg(self, sPostData, sMsgSignature, sTimeStamp, sNonce):
|
||||||
|
# 检验消息的真实性,并且获取解密后的明文
|
||||||
|
# @param sMsgSignature: 签名串,对应URL参数的msg_signature
|
||||||
|
# @param sTimeStamp: 时间戳,对应URL参数的timestamp
|
||||||
|
# @param sNonce: 随机串,对应URL参数的nonce
|
||||||
|
# @param sPostData: 密文,对应POST请求的数据
|
||||||
|
# xml_content: 解密后的原文,当return返回0时有效
|
||||||
|
# @return: 成功0,失败返回对应的错误码
|
||||||
|
# 验证安全签名
|
||||||
|
xmlParse = XMLParse()
|
||||||
|
ret, encrypt = xmlParse.extract(sPostData)
|
||||||
|
if ret != 0:
|
||||||
|
return ret, None
|
||||||
|
sha1 = SHA1()
|
||||||
|
ret, signature = sha1.getSHA1(self.m_sToken, sTimeStamp, sNonce, encrypt)
|
||||||
|
if ret != 0:
|
||||||
|
return ret, None
|
||||||
|
if not signature == sMsgSignature:
|
||||||
|
return ierror.WXBizMsgCrypt_ValidateSignature_Error, None
|
||||||
|
pc = Prpcrypt(self.key)
|
||||||
|
ret, xml_content = pc.decrypt(encrypt, self.m_sReceiveId)
|
||||||
|
return ret, xml_content
|
||||||
0
libs/wecom_api/__init__.py
Normal file
0
libs/wecom_api/__init__.py
Normal file
305
libs/wecom_api/api.py
Normal file
305
libs/wecom_api/api.py
Normal file
@@ -0,0 +1,305 @@
|
|||||||
|
from quart import request
|
||||||
|
from .WXBizMsgCrypt3 import WXBizMsgCrypt
|
||||||
|
import base64
|
||||||
|
import binascii
|
||||||
|
import httpx
|
||||||
|
from quart import Quart
|
||||||
|
import xml.etree.ElementTree as ET
|
||||||
|
from typing import Callable, Dict, Any
|
||||||
|
from .wecomevent import WecomEvent
|
||||||
|
from pkg.platform.types import events as platform_events, message as platform_message
|
||||||
|
import aiofiles
|
||||||
|
|
||||||
|
|
||||||
|
class WecomClient():
|
||||||
|
def __init__(self,corpid:str,secret:str,token:str,EncodingAESKey:str,contacts_secret:str):
|
||||||
|
self.corpid = corpid
|
||||||
|
self.secret = secret
|
||||||
|
self.access_token_for_contacts =''
|
||||||
|
self.token = token
|
||||||
|
self.aes = EncodingAESKey
|
||||||
|
self.base_url = 'https://qyapi.weixin.qq.com/cgi-bin'
|
||||||
|
self.access_token = ''
|
||||||
|
self.secret_for_contacts = contacts_secret
|
||||||
|
self.app = Quart(__name__)
|
||||||
|
self.wxcpt = WXBizMsgCrypt(self.token, self.aes, self.corpid)
|
||||||
|
self.app.add_url_rule('/callback/command', 'handle_callback', self.handle_callback_request, methods=['GET', 'POST'])
|
||||||
|
self._message_handlers = {
|
||||||
|
"example":[],
|
||||||
|
}
|
||||||
|
|
||||||
|
#access——token操作
|
||||||
|
async def check_access_token(self):
|
||||||
|
return bool(self.access_token and self.access_token.strip())
|
||||||
|
|
||||||
|
async def check_access_token_for_contacts(self):
|
||||||
|
return bool(self.access_token_for_contacts and self.access_token_for_contacts.strip())
|
||||||
|
|
||||||
|
async def get_access_token(self,secret):
|
||||||
|
url = f'https://qyapi.weixin.qq.com/cgi-bin/gettoken?corpid={self.corpid}&corpsecret={secret}'
|
||||||
|
async with httpx.AsyncClient() as client:
|
||||||
|
response = await client.get(url)
|
||||||
|
data = response.json()
|
||||||
|
if 'access_token' in data:
|
||||||
|
return data['access_token']
|
||||||
|
else:
|
||||||
|
raise Exception(f"未获取access token: {data}")
|
||||||
|
|
||||||
|
async def get_users(self):
|
||||||
|
if not self.check_access_token_for_contacts():
|
||||||
|
self.access_token_for_contacts = await self.get_access_token(self.secret_for_contacts)
|
||||||
|
|
||||||
|
url = self.base_url+'/user/list_id?access_token='+self.access_token_for_contacts
|
||||||
|
async with httpx.AsyncClient() as client:
|
||||||
|
params = {
|
||||||
|
"cursor":"",
|
||||||
|
"limit":10000,
|
||||||
|
}
|
||||||
|
response = await client.post(url,json=params)
|
||||||
|
data = response.json()
|
||||||
|
if data['errcode'] == 0:
|
||||||
|
dept_users = data['dept_user']
|
||||||
|
userid = []
|
||||||
|
for user in dept_users:
|
||||||
|
userid.append(user["userid"])
|
||||||
|
return userid
|
||||||
|
else:
|
||||||
|
raise Exception("未获取用户")
|
||||||
|
|
||||||
|
async def send_to_all(self,content:str):
|
||||||
|
if not self.check_access_token_for_contacts():
|
||||||
|
self.access_token_for_contacts = await self.get_access_token(self.secret_for_contacts)
|
||||||
|
|
||||||
|
url = self.base_url+'/message/send?access_token='+self.access_token_for_contacts
|
||||||
|
user_ids = await self.get_users()
|
||||||
|
user_ids_string = "|".join(user_ids)
|
||||||
|
async with httpx.AsyncClient() as client:
|
||||||
|
params = {
|
||||||
|
"touser" : user_ids_string,
|
||||||
|
"msgtype" : "text",
|
||||||
|
"agentid" : 1000002,
|
||||||
|
"text" : {
|
||||||
|
"content" : content,
|
||||||
|
},
|
||||||
|
"safe":0,
|
||||||
|
"enable_id_trans": 0,
|
||||||
|
"enable_duplicate_check": 0,
|
||||||
|
"duplicate_check_interval": 1800
|
||||||
|
}
|
||||||
|
response = await client.post(url,json=params)
|
||||||
|
data = response.json()
|
||||||
|
if data['errcode'] != 0:
|
||||||
|
raise Exception("Failed to send message: "+str(data))
|
||||||
|
|
||||||
|
async def send_image(self,user_id:str,agent_id:int,media_id:str):
|
||||||
|
if not await self.check_access_token():
|
||||||
|
self.access_token = await self.get_access_token(self.secret)
|
||||||
|
url = self.base_url+'/media/upload?access_token='+self.access_token
|
||||||
|
async with httpx.AsyncClient() as client:
|
||||||
|
params = {
|
||||||
|
"touser" : user_id,
|
||||||
|
"toparty" : "",
|
||||||
|
"totag":"",
|
||||||
|
"agentid" : agent_id,
|
||||||
|
"msgtype" : "image",
|
||||||
|
"image" : {
|
||||||
|
"media_id" : media_id,
|
||||||
|
},
|
||||||
|
"safe":0,
|
||||||
|
"enable_id_trans": 0,
|
||||||
|
"enable_duplicate_check": 0,
|
||||||
|
"duplicate_check_interval": 1800
|
||||||
|
}
|
||||||
|
response = await client.post(url,json=params)
|
||||||
|
data = response.json()
|
||||||
|
if data['errcode'] != 0:
|
||||||
|
raise Exception("Failed to send image: "+str(data))
|
||||||
|
|
||||||
|
async def send_private_msg(self,user_id:str, agent_id:int,content:str):
|
||||||
|
if not await self.check_access_token():
|
||||||
|
self.access_token = await self.get_access_token(self.secret)
|
||||||
|
|
||||||
|
url = self.base_url+'/message/send?access_token='+self.access_token
|
||||||
|
|
||||||
|
async with httpx.AsyncClient() as client:
|
||||||
|
params={
|
||||||
|
"touser" : user_id,
|
||||||
|
"msgtype" : "text",
|
||||||
|
"agentid" : agent_id,
|
||||||
|
"text" : {
|
||||||
|
"content" : content,
|
||||||
|
},
|
||||||
|
"safe":0,
|
||||||
|
"enable_id_trans": 0,
|
||||||
|
"enable_duplicate_check": 0,
|
||||||
|
"duplicate_check_interval": 1800
|
||||||
|
}
|
||||||
|
response = await client.post(url,json=params)
|
||||||
|
data = response.json()
|
||||||
|
|
||||||
|
if data['errcode'] != 0:
|
||||||
|
raise Exception("Failed to send message: "+str(data))
|
||||||
|
|
||||||
|
async def handle_callback_request(self):
|
||||||
|
"""
|
||||||
|
处理回调请求,包括 GET 验证和 POST 消息接收。
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
|
||||||
|
msg_signature = request.args.get("msg_signature")
|
||||||
|
timestamp = request.args.get("timestamp")
|
||||||
|
nonce = request.args.get("nonce")
|
||||||
|
|
||||||
|
if request.method == "GET":
|
||||||
|
echostr = request.args.get("echostr")
|
||||||
|
ret, reply_echo_str = self.wxcpt.VerifyURL(msg_signature, timestamp, nonce, echostr)
|
||||||
|
if ret != 0:
|
||||||
|
raise Exception(f"验证失败,错误码: {ret}")
|
||||||
|
return reply_echo_str
|
||||||
|
|
||||||
|
elif request.method == "POST":
|
||||||
|
encrypt_msg = await request.data
|
||||||
|
ret, xml_msg = self.wxcpt.DecryptMsg(encrypt_msg, msg_signature, timestamp, nonce)
|
||||||
|
if ret != 0:
|
||||||
|
raise Exception(f"消息解密失败,错误码: {ret}")
|
||||||
|
|
||||||
|
# 解析消息并处理
|
||||||
|
message_data = await self.get_message(xml_msg)
|
||||||
|
if message_data:
|
||||||
|
event = WecomEvent.from_payload(message_data) # 转换为 WecomEvent 对象
|
||||||
|
if event:
|
||||||
|
await self._handle_message(event)
|
||||||
|
|
||||||
|
return "success"
|
||||||
|
except Exception as e:
|
||||||
|
return f"Error processing request: {str(e)}", 400
|
||||||
|
|
||||||
|
async def run_task(self, host: str, port: int, *args, **kwargs):
|
||||||
|
"""
|
||||||
|
启动 Quart 应用。
|
||||||
|
"""
|
||||||
|
await self.app.run_task(host=host, port=port, *args, **kwargs)
|
||||||
|
|
||||||
|
def on_message(self, msg_type: str):
|
||||||
|
"""
|
||||||
|
注册消息类型处理器。
|
||||||
|
"""
|
||||||
|
def decorator(func: Callable[[WecomEvent], None]):
|
||||||
|
if msg_type not in self._message_handlers:
|
||||||
|
self._message_handlers[msg_type] = []
|
||||||
|
self._message_handlers[msg_type].append(func)
|
||||||
|
return func
|
||||||
|
return decorator
|
||||||
|
|
||||||
|
async def _handle_message(self, event: WecomEvent):
|
||||||
|
"""
|
||||||
|
处理消息事件。
|
||||||
|
"""
|
||||||
|
msg_type = event.type
|
||||||
|
if msg_type in self._message_handlers:
|
||||||
|
for handler in self._message_handlers[msg_type]:
|
||||||
|
await handler(event)
|
||||||
|
|
||||||
|
async def get_message(self, xml_msg: str) -> Dict[str, Any]:
|
||||||
|
"""
|
||||||
|
解析微信返回的 XML 消息并转换为字典。
|
||||||
|
"""
|
||||||
|
root = ET.fromstring(xml_msg)
|
||||||
|
message_data = {
|
||||||
|
"ToUserName": root.find("ToUserName").text,
|
||||||
|
"FromUserName": root.find("FromUserName").text,
|
||||||
|
"CreateTime": int(root.find("CreateTime").text),
|
||||||
|
"MsgType": root.find("MsgType").text,
|
||||||
|
"Content": root.find("Content").text if root.find("Content") is not None else None,
|
||||||
|
"MsgId": int(root.find("MsgId").text) if root.find("MsgId") is not None else None,
|
||||||
|
"AgentID": int(root.find("AgentID").text) if root.find("AgentID") is not None else None,
|
||||||
|
}
|
||||||
|
if message_data["MsgType"] == "image":
|
||||||
|
message_data["MediaId"] = root.find("MediaId").text if root.find("MediaId") is not None else None
|
||||||
|
message_data["PicUrl"] = root.find("PicUrl").text if root.find("PicUrl") is not None else None
|
||||||
|
|
||||||
|
return message_data
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def get_image_type(image_bytes: bytes) -> str:
|
||||||
|
"""
|
||||||
|
通过图片的magic numbers判断图片类型
|
||||||
|
"""
|
||||||
|
magic_numbers = {
|
||||||
|
b'\xFF\xD8\xFF': 'jpg',
|
||||||
|
b'\x89\x50\x4E\x47': 'png',
|
||||||
|
b'\x47\x49\x46': 'gif',
|
||||||
|
b'\x42\x4D': 'bmp',
|
||||||
|
b'\x00\x00\x01\x00': 'ico'
|
||||||
|
}
|
||||||
|
|
||||||
|
for magic, ext in magic_numbers.items():
|
||||||
|
if image_bytes.startswith(magic):
|
||||||
|
return ext
|
||||||
|
return 'jpg' # 默认返回jpg
|
||||||
|
|
||||||
|
|
||||||
|
async def upload_to_work(self, image: platform_message.Image):
|
||||||
|
"""
|
||||||
|
获取 media_id
|
||||||
|
"""
|
||||||
|
if not await self.check_access_token():
|
||||||
|
self.access_token = await self.get_access_token(self.secret)
|
||||||
|
|
||||||
|
url = self.base_url + '/media/upload?access_token=' + self.access_token + '&type=file'
|
||||||
|
file_bytes = None
|
||||||
|
file_name = "uploaded_file.txt"
|
||||||
|
|
||||||
|
# 获取文件的二进制数据
|
||||||
|
if image.path:
|
||||||
|
async with aiofiles.open(image.path, 'rb') as f:
|
||||||
|
file_bytes = await f.read()
|
||||||
|
file_name = image.path.split('/')[-1]
|
||||||
|
elif image.url:
|
||||||
|
file_bytes = await self.download_image_to_bytes(image.url)
|
||||||
|
file_name = image.url.split('/')[-1]
|
||||||
|
elif image.base64:
|
||||||
|
try:
|
||||||
|
base64_data = image.base64
|
||||||
|
if ',' in base64_data:
|
||||||
|
base64_data = base64_data.split(',', 1)[1]
|
||||||
|
padding = 4 - (len(base64_data) % 4) if len(base64_data) % 4 else 0
|
||||||
|
padded_base64 = base64_data + '=' * padding
|
||||||
|
file_bytes = base64.b64decode(padded_base64)
|
||||||
|
except binascii.Error as e:
|
||||||
|
raise ValueError(f"Invalid base64 string: {str(e)}")
|
||||||
|
else:
|
||||||
|
raise ValueError("image对象出错")
|
||||||
|
|
||||||
|
# 设置 multipart/form-data 格式的文件
|
||||||
|
boundary = "-------------------------acebdf13572468"
|
||||||
|
headers = {
|
||||||
|
'Content-Type': f'multipart/form-data; boundary={boundary}'
|
||||||
|
}
|
||||||
|
body = (
|
||||||
|
f"--{boundary}\r\n"
|
||||||
|
f"Content-Disposition: form-data; name=\"media\"; filename=\"{file_name}\"; filelength={len(file_bytes)}\r\n"
|
||||||
|
f"Content-Type: application/octet-stream\r\n\r\n"
|
||||||
|
).encode('utf-8') + file_bytes + f"\r\n--{boundary}--\r\n".encode('utf-8')
|
||||||
|
|
||||||
|
# 上传文件
|
||||||
|
async with httpx.AsyncClient() as client:
|
||||||
|
response = await client.post(url, headers=headers, content=body)
|
||||||
|
data = response.json()
|
||||||
|
if data.get('errcode', 0) != 0:
|
||||||
|
raise Exception("failed to upload file")
|
||||||
|
|
||||||
|
return data.get('media_id')
|
||||||
|
|
||||||
|
|
||||||
|
async def download_image_to_bytes(self,url:str) -> bytes:
|
||||||
|
async with httpx.AsyncClient() as client:
|
||||||
|
response = await client.get(url)
|
||||||
|
response.raise_for_status()
|
||||||
|
return response.content
|
||||||
|
|
||||||
|
#进行media_id的获取
|
||||||
|
async def get_media_id(self, image: platform_message.Image):
|
||||||
|
|
||||||
|
media_id = await self.upload_to_work(image=image)
|
||||||
|
return media_id
|
||||||
20
libs/wecom_api/ierror.py
Normal file
20
libs/wecom_api/ierror.py
Normal file
@@ -0,0 +1,20 @@
|
|||||||
|
#!/usr/bin/env python
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
#########################################################################
|
||||||
|
# Author: jonyqin
|
||||||
|
# Created Time: Thu 11 Sep 2014 01:53:58 PM CST
|
||||||
|
# File Name: ierror.py
|
||||||
|
# Description:定义错误码含义
|
||||||
|
#########################################################################
|
||||||
|
WXBizMsgCrypt_OK = 0
|
||||||
|
WXBizMsgCrypt_ValidateSignature_Error = -40001
|
||||||
|
WXBizMsgCrypt_ParseXml_Error = -40002
|
||||||
|
WXBizMsgCrypt_ComputeSignature_Error = -40003
|
||||||
|
WXBizMsgCrypt_IllegalAesKey = -40004
|
||||||
|
WXBizMsgCrypt_ValidateCorpid_Error = -40005
|
||||||
|
WXBizMsgCrypt_EncryptAES_Error = -40006
|
||||||
|
WXBizMsgCrypt_DecryptAES_Error = -40007
|
||||||
|
WXBizMsgCrypt_IllegalBuffer = -40008
|
||||||
|
WXBizMsgCrypt_EncodeBase64_Error = -40009
|
||||||
|
WXBizMsgCrypt_DecodeBase64_Error = -40010
|
||||||
|
WXBizMsgCrypt_GenReturnXml_Error = -40011
|
||||||
179
libs/wecom_api/wecomevent.py
Normal file
179
libs/wecom_api/wecomevent.py
Normal file
@@ -0,0 +1,179 @@
|
|||||||
|
from typing import Dict, Any, Optional
|
||||||
|
|
||||||
|
|
||||||
|
class WecomEvent(dict):
|
||||||
|
"""
|
||||||
|
封装从企业微信收到的事件数据对象(字典),提供属性以获取其中的字段。
|
||||||
|
|
||||||
|
除 `type` 和 `detail_type` 属性对于任何事件都有效外,其它属性是否存在(若不存在则返回 `None`)依事件类型不同而不同。
|
||||||
|
"""
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def from_payload(payload: Dict[str, Any]) -> Optional["WecomEvent"]:
|
||||||
|
"""
|
||||||
|
从企业微信事件数据构造 `WecomEvent` 对象。
|
||||||
|
|
||||||
|
Args:
|
||||||
|
payload (Dict[str, Any]): 解密后的企业微信事件数据。
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Optional[WecomEvent]: 如果事件数据合法,则返回 WecomEvent 对象;否则返回 None。
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
event = WecomEvent(payload)
|
||||||
|
_ = event.type, event.detail_type # 确保必须字段存在
|
||||||
|
return event
|
||||||
|
except KeyError:
|
||||||
|
return None
|
||||||
|
|
||||||
|
@property
|
||||||
|
def type(self) -> str:
|
||||||
|
"""
|
||||||
|
事件类型,例如 "message"、"event"、"text" 等。
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
str: 事件类型。
|
||||||
|
"""
|
||||||
|
return self.get("MsgType", "")
|
||||||
|
|
||||||
|
@property
|
||||||
|
def picurl(self) -> str:
|
||||||
|
"""
|
||||||
|
图片链接
|
||||||
|
"""
|
||||||
|
return self.get("PicUrl")
|
||||||
|
|
||||||
|
@property
|
||||||
|
def detail_type(self) -> str:
|
||||||
|
"""
|
||||||
|
事件详细类型,依 `type` 的不同而不同。例如:
|
||||||
|
- 消息事件: "text", "image", "voice", 等
|
||||||
|
- 事件通知: "subscribe", "unsubscribe", "click", 等
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
str: 事件详细类型。
|
||||||
|
"""
|
||||||
|
if self.type == "event":
|
||||||
|
return self.get("Event", "")
|
||||||
|
return self.type
|
||||||
|
|
||||||
|
@property
|
||||||
|
def name(self) -> str:
|
||||||
|
"""
|
||||||
|
事件名,对于消息事件是 `type.detail_type`,对于其他事件是 `event_type`。
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
str: 事件名。
|
||||||
|
"""
|
||||||
|
return f"{self.type}.{self.detail_type}"
|
||||||
|
|
||||||
|
@property
|
||||||
|
def user_id(self) -> Optional[str]:
|
||||||
|
"""
|
||||||
|
用户 ID,例如消息的发送者或事件的触发者。
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Optional[str]: 用户 ID。
|
||||||
|
"""
|
||||||
|
return self.get("FromUserName")
|
||||||
|
|
||||||
|
@property
|
||||||
|
def agent_id(self) -> Optional[int]:
|
||||||
|
"""
|
||||||
|
机器人 ID,仅在消息类型事件中存在。
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Optional[int]: 机器人 ID。
|
||||||
|
"""
|
||||||
|
return self.get("AgentID")
|
||||||
|
|
||||||
|
@property
|
||||||
|
def receiver_id(self) -> Optional[str]:
|
||||||
|
"""
|
||||||
|
接收者 ID,例如机器人自身的企业微信 ID。
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Optional[str]: 接收者 ID。
|
||||||
|
"""
|
||||||
|
return self.get("ToUserName")
|
||||||
|
|
||||||
|
@property
|
||||||
|
def message_id(self) -> Optional[str]:
|
||||||
|
"""
|
||||||
|
消息 ID,仅在消息类型事件中存在。
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Optional[str]: 消息 ID。
|
||||||
|
"""
|
||||||
|
return self.get("MsgId")
|
||||||
|
|
||||||
|
@property
|
||||||
|
def message(self) -> Optional[str]:
|
||||||
|
"""
|
||||||
|
消息内容,仅在消息类型事件中存在。
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Optional[str]: 消息内容。
|
||||||
|
"""
|
||||||
|
return self.get("Content")
|
||||||
|
|
||||||
|
@property
|
||||||
|
def media_id(self) -> Optional[str]:
|
||||||
|
"""
|
||||||
|
媒体文件 ID,仅在图片、语音等消息类型中存在。
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Optional[str]: 媒体文件 ID。
|
||||||
|
"""
|
||||||
|
return self.get("MediaId")
|
||||||
|
|
||||||
|
@property
|
||||||
|
def timestamp(self) -> Optional[int]:
|
||||||
|
"""
|
||||||
|
事件发生的时间戳。
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Optional[int]: 时间戳。
|
||||||
|
"""
|
||||||
|
return self.get("CreateTime")
|
||||||
|
|
||||||
|
@property
|
||||||
|
def event_key(self) -> Optional[str]:
|
||||||
|
"""
|
||||||
|
事件的 Key 值,例如点击菜单时的 `EventKey`。
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Optional[str]: 事件 Key。
|
||||||
|
"""
|
||||||
|
return self.get("EventKey")
|
||||||
|
|
||||||
|
def __getattr__(self, key: str) -> Optional[Any]:
|
||||||
|
"""
|
||||||
|
允许通过属性访问数据中的任意字段。
|
||||||
|
|
||||||
|
Args:
|
||||||
|
key (str): 字段名。
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Optional[Any]: 字段值。
|
||||||
|
"""
|
||||||
|
return self.get(key)
|
||||||
|
|
||||||
|
def __setattr__(self, key: str, value: Any) -> None:
|
||||||
|
"""
|
||||||
|
允许通过属性设置数据中的任意字段。
|
||||||
|
|
||||||
|
Args:
|
||||||
|
key (str): 字段名。
|
||||||
|
value (Any): 字段值。
|
||||||
|
"""
|
||||||
|
self[key] = value
|
||||||
|
|
||||||
|
def __repr__(self) -> str:
|
||||||
|
"""
|
||||||
|
生成事件对象的字符串表示。
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
str: 字符串表示。
|
||||||
|
"""
|
||||||
|
return f"<WecomEvent {super().__repr__()}>"
|
||||||
4
main.py
4
main.py
@@ -49,12 +49,10 @@ async def main_entry(loop: asyncio.AbstractEventLoop):
|
|||||||
generated_files = await files.generate_files()
|
generated_files = await files.generate_files()
|
||||||
|
|
||||||
if generated_files:
|
if generated_files:
|
||||||
print("以下文件不存在,已自动生成,请按需修改配置文件后重启:")
|
print("以下文件不存在,已自动生成:")
|
||||||
for file in generated_files:
|
for file in generated_files:
|
||||||
print("-", file)
|
print("-", file)
|
||||||
|
|
||||||
sys.exit(0)
|
|
||||||
|
|
||||||
from pkg.core import boot
|
from pkg.core import boot
|
||||||
await boot.main(loop)
|
await boot.main(loop)
|
||||||
|
|
||||||
|
|||||||
@@ -12,7 +12,7 @@ from .. import group
|
|||||||
class LogsRouterGroup(group.RouterGroup):
|
class LogsRouterGroup(group.RouterGroup):
|
||||||
|
|
||||||
async def initialize(self) -> None:
|
async def initialize(self) -> None:
|
||||||
@self.route('', methods=['GET'])
|
@self.route('', methods=['GET'], auth_type=group.AuthType.USER_TOKEN)
|
||||||
async def _() -> str:
|
async def _() -> str:
|
||||||
|
|
||||||
start_page_number = int(quart.request.args.get('start_page_number', 0))
|
start_page_number = int(quart.request.args.get('start_page_number', 0))
|
||||||
|
|||||||
@@ -13,7 +13,7 @@ from .. import group
|
|||||||
class PluginsRouterGroup(group.RouterGroup):
|
class PluginsRouterGroup(group.RouterGroup):
|
||||||
|
|
||||||
async def initialize(self) -> None:
|
async def initialize(self) -> None:
|
||||||
@self.route('', methods=['GET'])
|
@self.route('', methods=['GET'], auth_type=group.AuthType.USER_TOKEN)
|
||||||
async def _() -> str:
|
async def _() -> str:
|
||||||
plugins = self.ap.plugin_mgr.plugins()
|
plugins = self.ap.plugin_mgr.plugins()
|
||||||
|
|
||||||
@@ -23,14 +23,14 @@ class PluginsRouterGroup(group.RouterGroup):
|
|||||||
'plugins': plugins_data
|
'plugins': plugins_data
|
||||||
})
|
})
|
||||||
|
|
||||||
@self.route('/<author>/<plugin_name>/toggle', methods=['PUT'])
|
@self.route('/<author>/<plugin_name>/toggle', methods=['PUT'], auth_type=group.AuthType.USER_TOKEN)
|
||||||
async def _(author: str, plugin_name: str) -> str:
|
async def _(author: str, plugin_name: str) -> str:
|
||||||
data = await quart.request.json
|
data = await quart.request.json
|
||||||
target_enabled = data.get('target_enabled')
|
target_enabled = data.get('target_enabled')
|
||||||
await self.ap.plugin_mgr.update_plugin_switch(plugin_name, target_enabled)
|
await self.ap.plugin_mgr.update_plugin_switch(plugin_name, target_enabled)
|
||||||
return self.success()
|
return self.success()
|
||||||
|
|
||||||
@self.route('/<author>/<plugin_name>/update', methods=['POST'])
|
@self.route('/<author>/<plugin_name>/update', methods=['POST'], auth_type=group.AuthType.USER_TOKEN)
|
||||||
async def _(author: str, plugin_name: str) -> str:
|
async def _(author: str, plugin_name: str) -> str:
|
||||||
ctx = taskmgr.TaskContext.new()
|
ctx = taskmgr.TaskContext.new()
|
||||||
wrapper = self.ap.task_mgr.create_user_task(
|
wrapper = self.ap.task_mgr.create_user_task(
|
||||||
@@ -44,7 +44,7 @@ class PluginsRouterGroup(group.RouterGroup):
|
|||||||
'task_id': wrapper.id
|
'task_id': wrapper.id
|
||||||
})
|
})
|
||||||
|
|
||||||
@self.route('/<author>/<plugin_name>', methods=['DELETE'])
|
@self.route('/<author>/<plugin_name>', methods=['DELETE'], auth_type=group.AuthType.USER_TOKEN)
|
||||||
async def _(author: str, plugin_name: str) -> str:
|
async def _(author: str, plugin_name: str) -> str:
|
||||||
ctx = taskmgr.TaskContext.new()
|
ctx = taskmgr.TaskContext.new()
|
||||||
wrapper = self.ap.task_mgr.create_user_task(
|
wrapper = self.ap.task_mgr.create_user_task(
|
||||||
@@ -59,13 +59,13 @@ class PluginsRouterGroup(group.RouterGroup):
|
|||||||
'task_id': wrapper.id
|
'task_id': wrapper.id
|
||||||
})
|
})
|
||||||
|
|
||||||
@self.route('/reorder', methods=['PUT'])
|
@self.route('/reorder', methods=['PUT'], auth_type=group.AuthType.USER_TOKEN)
|
||||||
async def _() -> str:
|
async def _() -> str:
|
||||||
data = await quart.request.json
|
data = await quart.request.json
|
||||||
await self.ap.plugin_mgr.reorder_plugins(data.get('plugins'))
|
await self.ap.plugin_mgr.reorder_plugins(data.get('plugins'))
|
||||||
return self.success()
|
return self.success()
|
||||||
|
|
||||||
@self.route('/install/github', methods=['POST'])
|
@self.route('/install/github', methods=['POST'], auth_type=group.AuthType.USER_TOKEN)
|
||||||
async def _() -> str:
|
async def _() -> str:
|
||||||
data = await quart.request.json
|
data = await quart.request.json
|
||||||
|
|
||||||
|
|||||||
@@ -9,7 +9,7 @@ class SettingsRouterGroup(group.RouterGroup):
|
|||||||
|
|
||||||
async def initialize(self) -> None:
|
async def initialize(self) -> None:
|
||||||
|
|
||||||
@self.route('', methods=['GET'])
|
@self.route('', methods=['GET'], auth_type=group.AuthType.USER_TOKEN)
|
||||||
async def _() -> str:
|
async def _() -> str:
|
||||||
return self.success(
|
return self.success(
|
||||||
data={
|
data={
|
||||||
@@ -23,7 +23,7 @@ class SettingsRouterGroup(group.RouterGroup):
|
|||||||
}
|
}
|
||||||
)
|
)
|
||||||
|
|
||||||
@self.route('/<manager_name>', methods=['GET'])
|
@self.route('/<manager_name>', methods=['GET'], auth_type=group.AuthType.USER_TOKEN)
|
||||||
async def _(manager_name: str) -> str:
|
async def _(manager_name: str) -> str:
|
||||||
|
|
||||||
manager = self.ap.settings_mgr.get_manager(manager_name)
|
manager = self.ap.settings_mgr.get_manager(manager_name)
|
||||||
@@ -44,7 +44,7 @@ class SettingsRouterGroup(group.RouterGroup):
|
|||||||
}
|
}
|
||||||
)
|
)
|
||||||
|
|
||||||
@self.route('/<manager_name>/data', methods=['PUT'])
|
@self.route('/<manager_name>/data', methods=['PUT'], auth_type=group.AuthType.USER_TOKEN)
|
||||||
async def _(manager_name: str) -> str:
|
async def _(manager_name: str) -> str:
|
||||||
data = await quart.request.json
|
data = await quart.request.json
|
||||||
manager = self.ap.settings_mgr.get_manager(manager_name)
|
manager = self.ap.settings_mgr.get_manager(manager_name)
|
||||||
|
|||||||
@@ -9,7 +9,7 @@ from .. import group
|
|||||||
class StatsRouterGroup(group.RouterGroup):
|
class StatsRouterGroup(group.RouterGroup):
|
||||||
|
|
||||||
async def initialize(self) -> None:
|
async def initialize(self) -> None:
|
||||||
@self.route('/basic', methods=['GET'])
|
@self.route('/basic', methods=['GET'], auth_type=group.AuthType.USER_TOKEN)
|
||||||
async def _() -> str:
|
async def _() -> str:
|
||||||
|
|
||||||
conv_count = 0
|
conv_count = 0
|
||||||
|
|||||||
@@ -20,7 +20,7 @@ class SystemRouterGroup(group.RouterGroup):
|
|||||||
}
|
}
|
||||||
)
|
)
|
||||||
|
|
||||||
@self.route('/tasks', methods=['GET'])
|
@self.route('/tasks', methods=['GET'], auth_type=group.AuthType.USER_TOKEN)
|
||||||
async def _() -> str:
|
async def _() -> str:
|
||||||
task_type = quart.request.args.get("type")
|
task_type = quart.request.args.get("type")
|
||||||
|
|
||||||
@@ -31,7 +31,7 @@ class SystemRouterGroup(group.RouterGroup):
|
|||||||
data=self.ap.task_mgr.get_tasks_dict(task_type)
|
data=self.ap.task_mgr.get_tasks_dict(task_type)
|
||||||
)
|
)
|
||||||
|
|
||||||
@self.route('/tasks/<task_id>', methods=['GET'])
|
@self.route('/tasks/<task_id>', methods=['GET'], auth_type=group.AuthType.USER_TOKEN)
|
||||||
async def _(task_id: str) -> str:
|
async def _(task_id: str) -> str:
|
||||||
task = self.ap.task_mgr.get_task_by_id(int(task_id))
|
task = self.ap.task_mgr.get_task_by_id(int(task_id))
|
||||||
|
|
||||||
@@ -40,7 +40,7 @@ class SystemRouterGroup(group.RouterGroup):
|
|||||||
|
|
||||||
return self.success(data=task.to_dict())
|
return self.success(data=task.to_dict())
|
||||||
|
|
||||||
@self.route('/reload', methods=['POST'])
|
@self.route('/reload', methods=['POST'], auth_type=group.AuthType.USER_TOKEN)
|
||||||
async def _() -> str:
|
async def _() -> str:
|
||||||
json_data = await quart.request.json
|
json_data = await quart.request.json
|
||||||
|
|
||||||
@@ -51,7 +51,7 @@ class SystemRouterGroup(group.RouterGroup):
|
|||||||
)
|
)
|
||||||
return self.success()
|
return self.success()
|
||||||
|
|
||||||
@self.route('/_debug/exec', methods=['POST'])
|
@self.route('/_debug/exec', methods=['POST'], auth_type=group.AuthType.USER_TOKEN)
|
||||||
async def _() -> str:
|
async def _() -> str:
|
||||||
if not constants.debug_mode:
|
if not constants.debug_mode:
|
||||||
return self.http_status(403, 403, "Forbidden")
|
return self.http_status(403, 403, "Forbidden")
|
||||||
|
|||||||
@@ -13,14 +13,14 @@ identifier = {
|
|||||||
'instance_create_ts': 0,
|
'instance_create_ts': 0,
|
||||||
}
|
}
|
||||||
|
|
||||||
HOST_ID_FILE = os.path.expanduser('~/.qchatgpt/host_id.json')
|
HOST_ID_FILE = os.path.expanduser('~/.langbot/host_id.json')
|
||||||
INSTANCE_ID_FILE = 'res/instance_id.json'
|
INSTANCE_ID_FILE = 'data/labels/instance_id.json'
|
||||||
|
|
||||||
def init():
|
def init():
|
||||||
global identifier
|
global identifier
|
||||||
|
|
||||||
if not os.path.exists(os.path.expanduser('~/.qchatgpt')):
|
if not os.path.exists(os.path.expanduser('~/.langbot')):
|
||||||
os.mkdir(os.path.expanduser('~/.qchatgpt'))
|
os.mkdir(os.path.expanduser('~/.langbot'))
|
||||||
|
|
||||||
if not os.path.exists(HOST_ID_FILE):
|
if not os.path.exists(HOST_ID_FILE):
|
||||||
new_host_id = 'host_'+str(uuid.uuid4())
|
new_host_id = 'host_'+str(uuid.uuid4())
|
||||||
|
|||||||
@@ -197,5 +197,27 @@ class Application:
|
|||||||
|
|
||||||
await self.plugin_mgr.load_plugins()
|
await self.plugin_mgr.load_plugins()
|
||||||
await self.plugin_mgr.initialize_plugins()
|
await self.plugin_mgr.initialize_plugins()
|
||||||
|
case core_entities.LifecycleControlScope.PROVIDER.value:
|
||||||
|
self.logger.info("执行热重载 scope="+scope)
|
||||||
|
|
||||||
|
llm_model_mgr_inst = llm_model_mgr.ModelManager(self)
|
||||||
|
await llm_model_mgr_inst.initialize()
|
||||||
|
self.model_mgr = llm_model_mgr_inst
|
||||||
|
|
||||||
|
llm_session_mgr_inst = llm_session_mgr.SessionManager(self)
|
||||||
|
await llm_session_mgr_inst.initialize()
|
||||||
|
self.sess_mgr = llm_session_mgr_inst
|
||||||
|
|
||||||
|
llm_prompt_mgr_inst = llm_prompt_mgr.PromptManager(self)
|
||||||
|
await llm_prompt_mgr_inst.initialize()
|
||||||
|
self.prompt_mgr = llm_prompt_mgr_inst
|
||||||
|
|
||||||
|
llm_tool_mgr_inst = llm_tool_mgr.ToolManager(self)
|
||||||
|
await llm_tool_mgr_inst.initialize()
|
||||||
|
self.tool_mgr = llm_tool_mgr_inst
|
||||||
|
|
||||||
|
runner_mgr_inst = runnermgr.RunnerManager(self)
|
||||||
|
await runner_mgr_inst.initialize()
|
||||||
|
self.runner_mgr = runner_mgr_inst
|
||||||
case _:
|
case _:
|
||||||
pass
|
pass
|
||||||
@@ -24,6 +24,7 @@ required_paths = [
|
|||||||
"data/scenario",
|
"data/scenario",
|
||||||
"data/logs",
|
"data/logs",
|
||||||
"data/config",
|
"data/config",
|
||||||
|
"data/labels",
|
||||||
"plugins"
|
"plugins"
|
||||||
]
|
]
|
||||||
|
|
||||||
|
|||||||
@@ -23,6 +23,7 @@ class LifecycleControlScope(enum.Enum):
|
|||||||
APPLICATION = "application"
|
APPLICATION = "application"
|
||||||
PLATFORM = "platform"
|
PLATFORM = "platform"
|
||||||
PLUGIN = "plugin"
|
PLUGIN = "plugin"
|
||||||
|
PROVIDER = "provider"
|
||||||
|
|
||||||
|
|
||||||
class LauncherTypes(enum.Enum):
|
class LauncherTypes(enum.Enum):
|
||||||
@@ -44,10 +45,10 @@ class Query(pydantic.BaseModel):
|
|||||||
launcher_type: LauncherTypes
|
launcher_type: LauncherTypes
|
||||||
"""会话类型,platform处理阶段设置"""
|
"""会话类型,platform处理阶段设置"""
|
||||||
|
|
||||||
launcher_id: int
|
launcher_id: typing.Union[int, str]
|
||||||
"""会话ID,platform处理阶段设置"""
|
"""会话ID,platform处理阶段设置"""
|
||||||
|
|
||||||
sender_id: int
|
sender_id: typing.Union[int, str]
|
||||||
"""发送者ID,platform处理阶段设置"""
|
"""发送者ID,platform处理阶段设置"""
|
||||||
|
|
||||||
message_event: platform_events.MessageEvent
|
message_event: platform_events.MessageEvent
|
||||||
@@ -113,9 +114,9 @@ class Session(pydantic.BaseModel):
|
|||||||
"""会话,一个 Session 对应一个 {launcher_type.value}_{launcher_id}"""
|
"""会话,一个 Session 对应一个 {launcher_type.value}_{launcher_id}"""
|
||||||
launcher_type: LauncherTypes
|
launcher_type: LauncherTypes
|
||||||
|
|
||||||
launcher_id: int
|
launcher_id: typing.Union[int, str]
|
||||||
|
|
||||||
sender_id: typing.Optional[int] = 0
|
sender_id: typing.Optional[typing.Union[int, str]] = 0
|
||||||
|
|
||||||
use_prompt_name: typing.Optional[str] = 'default'
|
use_prompt_name: typing.Optional[str] = 'default'
|
||||||
|
|
||||||
|
|||||||
25
pkg/core/migrations/m018_xai_config.py
Normal file
25
pkg/core/migrations/m018_xai_config.py
Normal file
@@ -0,0 +1,25 @@
|
|||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
from .. import migration
|
||||||
|
|
||||||
|
|
||||||
|
@migration.migration_class("xai-config", 18)
|
||||||
|
class XaiConfigMigration(migration.Migration):
|
||||||
|
"""迁移"""
|
||||||
|
|
||||||
|
async def need_migrate(self) -> bool:
|
||||||
|
"""判断当前环境是否需要运行此迁移"""
|
||||||
|
return 'xai-chat-completions' not in self.ap.provider_cfg.data['requester']
|
||||||
|
|
||||||
|
async def run(self):
|
||||||
|
"""执行迁移"""
|
||||||
|
self.ap.provider_cfg.data['requester']['xai-chat-completions'] = {
|
||||||
|
"base-url": "https://api.x.ai/v1",
|
||||||
|
"args": {},
|
||||||
|
"timeout": 120
|
||||||
|
}
|
||||||
|
self.ap.provider_cfg.data['keys']['xai'] = [
|
||||||
|
"xai-1234567890"
|
||||||
|
]
|
||||||
|
|
||||||
|
await self.ap.provider_cfg.dump_config()
|
||||||
25
pkg/core/migrations/m019_zhipuai_config.py
Normal file
25
pkg/core/migrations/m019_zhipuai_config.py
Normal file
@@ -0,0 +1,25 @@
|
|||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
from .. import migration
|
||||||
|
|
||||||
|
|
||||||
|
@migration.migration_class("zhipuai-config", 19)
|
||||||
|
class ZhipuaiConfigMigration(migration.Migration):
|
||||||
|
"""迁移"""
|
||||||
|
|
||||||
|
async def need_migrate(self) -> bool:
|
||||||
|
"""判断当前环境是否需要运行此迁移"""
|
||||||
|
return 'zhipuai-chat-completions' not in self.ap.provider_cfg.data['requester']
|
||||||
|
|
||||||
|
async def run(self):
|
||||||
|
"""执行迁移"""
|
||||||
|
self.ap.provider_cfg.data['requester']['zhipuai-chat-completions'] = {
|
||||||
|
"base-url": "https://open.bigmodel.cn/api/paas/v4",
|
||||||
|
"args": {},
|
||||||
|
"timeout": 120
|
||||||
|
}
|
||||||
|
self.ap.provider_cfg.data['keys']['zhipuai'] = [
|
||||||
|
"xxxxxxx"
|
||||||
|
]
|
||||||
|
|
||||||
|
await self.ap.provider_cfg.dump_config()
|
||||||
33
pkg/core/migrations/m020_wecom_config.py
Normal file
33
pkg/core/migrations/m020_wecom_config.py
Normal file
@@ -0,0 +1,33 @@
|
|||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
from .. import migration
|
||||||
|
|
||||||
|
|
||||||
|
@migration.migration_class("wecom-config", 20)
|
||||||
|
class WecomConfigMigration(migration.Migration):
|
||||||
|
"""迁移"""
|
||||||
|
|
||||||
|
async def need_migrate(self) -> bool:
|
||||||
|
"""判断当前环境是否需要运行此迁移"""
|
||||||
|
|
||||||
|
for adapter in self.ap.platform_cfg.data['platform-adapters']:
|
||||||
|
if adapter['adapter'] == 'wecom':
|
||||||
|
return False
|
||||||
|
|
||||||
|
return True
|
||||||
|
|
||||||
|
async def run(self):
|
||||||
|
"""执行迁移"""
|
||||||
|
self.ap.platform_cfg.data['platform-adapters'].append({
|
||||||
|
"adapter": "wecom",
|
||||||
|
"enable": False,
|
||||||
|
"host": "0.0.0.0",
|
||||||
|
"port": 2290,
|
||||||
|
"corpid": "",
|
||||||
|
"secret": "",
|
||||||
|
"token": "",
|
||||||
|
"EncodingAESKey": "",
|
||||||
|
"contacts_secret": ""
|
||||||
|
})
|
||||||
|
|
||||||
|
await self.ap.platform_cfg.dump_config()
|
||||||
@@ -7,7 +7,8 @@ from .. import migration
|
|||||||
from ..migrations import m001_sensitive_word_migration, m002_openai_config_migration, m003_anthropic_requester_cfg_completion, m004_moonshot_cfg_completion
|
from ..migrations import m001_sensitive_word_migration, m002_openai_config_migration, m003_anthropic_requester_cfg_completion, m004_moonshot_cfg_completion
|
||||||
from ..migrations import m005_deepseek_cfg_completion, m006_vision_config, m007_qcg_center_url, m008_ad_fixwin_config_migrate, m009_msg_truncator_cfg
|
from ..migrations import m005_deepseek_cfg_completion, m006_vision_config, m007_qcg_center_url, m008_ad_fixwin_config_migrate, m009_msg_truncator_cfg
|
||||||
from ..migrations import m010_ollama_requester_config, m011_command_prefix_config, m012_runner_config, m013_http_api_config, m014_force_delay_config
|
from ..migrations import m010_ollama_requester_config, m011_command_prefix_config, m012_runner_config, m013_http_api_config, m014_force_delay_config
|
||||||
from ..migrations import m015_gitee_ai_config, m016_dify_service_api, m017_dify_api_timeout_params
|
from ..migrations import m015_gitee_ai_config, m016_dify_service_api, m017_dify_api_timeout_params, m018_xai_config, m019_zhipuai_config
|
||||||
|
from ..migrations import m020_wecom_config
|
||||||
|
|
||||||
|
|
||||||
@stage.stage_class("MigrationStage")
|
@stage.stage_class("MigrationStage")
|
||||||
|
|||||||
@@ -68,7 +68,7 @@ class Controller:
|
|||||||
except Exception as e:
|
except Exception as e:
|
||||||
# traceback.print_exc()
|
# traceback.print_exc()
|
||||||
self.ap.logger.error(f"控制器循环出错: {e}")
|
self.ap.logger.error(f"控制器循环出错: {e}")
|
||||||
self.ap.logger.debug(f"Traceback: {traceback.format_exc()}")
|
self.ap.logger.error(f"Traceback: {traceback.format_exc()}")
|
||||||
|
|
||||||
async def _check_output(self, query: entities.Query, result: pipeline_entities.StageProcessResult):
|
async def _check_output(self, query: entities.Query, result: pipeline_entities.StageProcessResult):
|
||||||
"""检查输出
|
"""检查输出
|
||||||
@@ -163,29 +163,30 @@ class Controller:
|
|||||||
async def process_query(self, query: entities.Query):
|
async def process_query(self, query: entities.Query):
|
||||||
"""处理请求
|
"""处理请求
|
||||||
"""
|
"""
|
||||||
|
|
||||||
# ======== 触发 MessageReceived 事件 ========
|
|
||||||
event_type = events.PersonMessageReceived if query.launcher_type == entities.LauncherTypes.PERSON else events.GroupMessageReceived
|
|
||||||
|
|
||||||
event_ctx = await self.ap.plugin_mgr.emit_event(
|
|
||||||
event=event_type(
|
|
||||||
launcher_type=query.launcher_type.value,
|
|
||||||
launcher_id=query.launcher_id,
|
|
||||||
sender_id=query.sender_id,
|
|
||||||
message_chain=query.message_chain,
|
|
||||||
query=query
|
|
||||||
)
|
|
||||||
)
|
|
||||||
|
|
||||||
if event_ctx.is_prevented_default():
|
|
||||||
return
|
|
||||||
|
|
||||||
self.ap.logger.debug(f"Processing query {query}")
|
|
||||||
|
|
||||||
try:
|
try:
|
||||||
|
|
||||||
|
# ======== 触发 MessageReceived 事件 ========
|
||||||
|
event_type = events.PersonMessageReceived if query.launcher_type == entities.LauncherTypes.PERSON else events.GroupMessageReceived
|
||||||
|
|
||||||
|
event_ctx = await self.ap.plugin_mgr.emit_event(
|
||||||
|
event=event_type(
|
||||||
|
launcher_type=query.launcher_type.value,
|
||||||
|
launcher_id=query.launcher_id,
|
||||||
|
sender_id=query.sender_id,
|
||||||
|
message_chain=query.message_chain,
|
||||||
|
query=query
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
if event_ctx.is_prevented_default():
|
||||||
|
return
|
||||||
|
|
||||||
|
self.ap.logger.debug(f"Processing query {query}")
|
||||||
|
|
||||||
await self._execute_from_stage(0, query)
|
await self._execute_from_stage(0, query)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
self.ap.logger.error(f"处理请求时出错 query_id={query.query_id} stage={query.current_stage.inst_name} : {e}")
|
inst_name = query.current_stage.inst_name if query.current_stage else 'unknown'
|
||||||
|
self.ap.logger.error(f"处理请求时出错 query_id={query.query_id} stage={inst_name} : {e}")
|
||||||
self.ap.logger.debug(f"Traceback: {traceback.format_exc()}")
|
self.ap.logger.debug(f"Traceback: {traceback.format_exc()}")
|
||||||
finally:
|
finally:
|
||||||
self.ap.logger.debug(f"Query {query} processed")
|
self.ap.logger.debug(f"Query {query} processed")
|
||||||
|
|||||||
@@ -1,7 +1,7 @@
|
|||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
|
|
||||||
import asyncio
|
import asyncio
|
||||||
|
import typing
|
||||||
|
|
||||||
from ..core import entities
|
from ..core import entities
|
||||||
from ..platform import adapter as msadapter
|
from ..platform import adapter as msadapter
|
||||||
@@ -29,8 +29,8 @@ class QueryPool:
|
|||||||
async def add_query(
|
async def add_query(
|
||||||
self,
|
self,
|
||||||
launcher_type: entities.LauncherTypes,
|
launcher_type: entities.LauncherTypes,
|
||||||
launcher_id: int,
|
launcher_id: typing.Union[int, str],
|
||||||
sender_id: int,
|
sender_id: typing.Union[int, str],
|
||||||
message_event: platform_events.MessageEvent,
|
message_event: platform_events.MessageEvent,
|
||||||
message_chain: platform_message.MessageChain,
|
message_chain: platform_message.MessageChain,
|
||||||
adapter: msadapter.MessageSourceAdapter
|
adapter: msadapter.MessageSourceAdapter
|
||||||
|
|||||||
@@ -105,7 +105,7 @@ class ChatMessageHandler(handler.MessageHandler):
|
|||||||
await self.ap.ctr_mgr.usage.post_query_record(
|
await self.ap.ctr_mgr.usage.post_query_record(
|
||||||
session_type=query.session.launcher_type.value,
|
session_type=query.session.launcher_type.value,
|
||||||
session_id=str(query.session.launcher_id),
|
session_id=str(query.session.launcher_id),
|
||||||
query_ability_provider="QChatGPT.Chat",
|
query_ability_provider="LangBot.Chat",
|
||||||
usage=text_length,
|
usage=text_length,
|
||||||
model_name=query.use_model.name,
|
model_name=query.use_model.name,
|
||||||
response_seconds=int(time.time() - start_time),
|
response_seconds=int(time.time() - start_time),
|
||||||
|
|||||||
@@ -31,7 +31,7 @@ class ReteLimitAlgo(metaclass=abc.ABCMeta):
|
|||||||
pass
|
pass
|
||||||
|
|
||||||
@abc.abstractmethod
|
@abc.abstractmethod
|
||||||
async def require_access(self, launcher_type: str, launcher_id: int) -> bool:
|
async def require_access(self, launcher_type: str, launcher_id: typing.Union[int, str]) -> bool:
|
||||||
"""进入处理流程
|
"""进入处理流程
|
||||||
|
|
||||||
这个方法对等待是友好的,意味着算法可以实现在这里等待一段时间以控制速率。
|
这个方法对等待是友好的,意味着算法可以实现在这里等待一段时间以控制速率。
|
||||||
@@ -46,7 +46,7 @@ class ReteLimitAlgo(metaclass=abc.ABCMeta):
|
|||||||
raise NotImplementedError
|
raise NotImplementedError
|
||||||
|
|
||||||
@abc.abstractmethod
|
@abc.abstractmethod
|
||||||
async def release_access(self, launcher_type: str, launcher_id: int):
|
async def release_access(self, launcher_type: str, launcher_id: typing.Union[int, str]):
|
||||||
"""退出处理流程
|
"""退出处理流程
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
|
|||||||
@@ -1,6 +1,7 @@
|
|||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
import asyncio
|
import asyncio
|
||||||
import time
|
import time
|
||||||
|
import typing
|
||||||
from .. import algo
|
from .. import algo
|
||||||
|
|
||||||
# 固定窗口算法
|
# 固定窗口算法
|
||||||
@@ -29,7 +30,7 @@ class FixedWindowAlgo(algo.ReteLimitAlgo):
|
|||||||
self.containers_lock = asyncio.Lock()
|
self.containers_lock = asyncio.Lock()
|
||||||
self.containers = {}
|
self.containers = {}
|
||||||
|
|
||||||
async def require_access(self, launcher_type: str, launcher_id: int) -> bool:
|
async def require_access(self, launcher_type: str, launcher_id: typing.Union[int, str]) -> bool:
|
||||||
# 加锁,找容器
|
# 加锁,找容器
|
||||||
container: SessionContainer = None
|
container: SessionContainer = None
|
||||||
|
|
||||||
@@ -83,5 +84,5 @@ class FixedWindowAlgo(algo.ReteLimitAlgo):
|
|||||||
# 返回True
|
# 返回True
|
||||||
return True
|
return True
|
||||||
|
|
||||||
async def release_access(self, launcher_type: str, launcher_id: int):
|
async def release_access(self, launcher_type: str, launcher_id: typing.Union[int, str]):
|
||||||
pass
|
pass
|
||||||
|
|||||||
@@ -52,7 +52,7 @@ class MessageSourceAdapter(metaclass=abc.ABCMeta):
|
|||||||
self.config = config
|
self.config = config
|
||||||
self.ap = ap
|
self.ap = ap
|
||||||
|
|
||||||
async def send_message(
|
async def send_message(
|
||||||
self,
|
self,
|
||||||
target_type: str,
|
target_type: str,
|
||||||
target_id: str,
|
target_id: str,
|
||||||
|
|||||||
@@ -37,7 +37,7 @@ class PlatformManager:
|
|||||||
|
|
||||||
async def initialize(self):
|
async def initialize(self):
|
||||||
|
|
||||||
from .sources import nakuru, aiocqhttp, qqbotpy
|
from .sources import nakuru, aiocqhttp, qqbotpy,wecom
|
||||||
|
|
||||||
async def on_friend_message(event: platform_events.FriendMessage, adapter: msadapter.MessageSourceAdapter):
|
async def on_friend_message(event: platform_events.FriendMessage, adapter: msadapter.MessageSourceAdapter):
|
||||||
|
|
||||||
@@ -50,17 +50,6 @@ class PlatformManager:
|
|||||||
adapter=adapter
|
adapter=adapter
|
||||||
)
|
)
|
||||||
|
|
||||||
async def on_stranger_message(event: platform_events.StrangerMessage, adapter: msadapter.MessageSourceAdapter):
|
|
||||||
|
|
||||||
await self.ap.query_pool.add_query(
|
|
||||||
launcher_type=core_entities.LauncherTypes.PERSON,
|
|
||||||
launcher_id=event.sender.id,
|
|
||||||
sender_id=event.sender.id,
|
|
||||||
message_event=event,
|
|
||||||
message_chain=event.message_chain,
|
|
||||||
adapter=adapter
|
|
||||||
)
|
|
||||||
|
|
||||||
async def on_group_message(event: platform_events.GroupMessage, adapter: msadapter.MessageSourceAdapter):
|
async def on_group_message(event: platform_events.GroupMessage, adapter: msadapter.MessageSourceAdapter):
|
||||||
|
|
||||||
await self.ap.query_pool.add_query(
|
await self.ap.query_pool.add_query(
|
||||||
@@ -96,12 +85,6 @@ class PlatformManager:
|
|||||||
)
|
)
|
||||||
self.adapters.append(adapter_inst)
|
self.adapters.append(adapter_inst)
|
||||||
|
|
||||||
if adapter_name == 'yiri-mirai':
|
|
||||||
adapter_inst.register_listener(
|
|
||||||
platform_events.StrangerMessage,
|
|
||||||
on_stranger_message
|
|
||||||
)
|
|
||||||
|
|
||||||
adapter_inst.register_listener(
|
adapter_inst.register_listener(
|
||||||
platform_events.FriendMessage,
|
platform_events.FriendMessage,
|
||||||
on_friend_message
|
on_friend_message
|
||||||
|
|||||||
258
pkg/platform/sources/wecom.py
Normal file
258
pkg/platform/sources/wecom.py
Normal file
@@ -0,0 +1,258 @@
|
|||||||
|
from __future__ import annotations
|
||||||
|
import typing
|
||||||
|
import asyncio
|
||||||
|
import traceback
|
||||||
|
import time
|
||||||
|
import datetime
|
||||||
|
|
||||||
|
import aiocqhttp
|
||||||
|
import aiohttp
|
||||||
|
from libs.wecom_api.api import WecomClient
|
||||||
|
from pkg.platform.adapter import MessageSourceAdapter
|
||||||
|
from pkg.platform.types import events as platform_events, message as platform_message
|
||||||
|
from libs.wecom_api.wecomevent import WecomEvent
|
||||||
|
from pkg.core import app
|
||||||
|
from .. import adapter
|
||||||
|
from ...pipeline.longtext.strategies import forward
|
||||||
|
from ...core import app
|
||||||
|
from ..types import message as platform_message
|
||||||
|
from ..types import events as platform_events
|
||||||
|
from ..types import entities as platform_entities
|
||||||
|
from ...command.errors import ParamNotEnoughError
|
||||||
|
from ...utils import image
|
||||||
|
|
||||||
|
class WecomMessageConverter(adapter.MessageConverter):
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def yiri2target(
|
||||||
|
message_chain: platform_message.MessageChain, bot: WecomClient
|
||||||
|
):
|
||||||
|
content_list = []
|
||||||
|
|
||||||
|
[
|
||||||
|
{
|
||||||
|
"type": "text",
|
||||||
|
"content": "text",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"type": "image",
|
||||||
|
"media_id": "media_id",
|
||||||
|
}
|
||||||
|
]
|
||||||
|
|
||||||
|
for msg in message_chain:
|
||||||
|
if type(msg) is platform_message.Plain:
|
||||||
|
content_list.append({
|
||||||
|
"type": "text",
|
||||||
|
"content": msg.text,
|
||||||
|
})
|
||||||
|
elif type(msg) is platform_message.Image:
|
||||||
|
content_list.append({
|
||||||
|
"type": "image",
|
||||||
|
"media_id": await bot.get_media_id(msg),
|
||||||
|
})
|
||||||
|
elif type(msg) is platform_message.Forward:
|
||||||
|
for node in msg.node_list:
|
||||||
|
content_list.extend((await WecomMessageConverter.yiri2target(node.message_chain, bot)))
|
||||||
|
else:
|
||||||
|
content_list.append({
|
||||||
|
"type": "text",
|
||||||
|
"content": str(msg),
|
||||||
|
})
|
||||||
|
|
||||||
|
return content_list
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def target2yiri(message: str, message_id: int = -1):
|
||||||
|
yiri_msg_list = []
|
||||||
|
yiri_msg_list.append(
|
||||||
|
platform_message.Source(id=message_id, time=datetime.datetime.now())
|
||||||
|
)
|
||||||
|
|
||||||
|
yiri_msg_list.append(platform_message.Plain(text=message))
|
||||||
|
chain = platform_message.MessageChain(yiri_msg_list)
|
||||||
|
|
||||||
|
return chain
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def target2yiri_image(picurl: str, message_id: int = -1):
|
||||||
|
yiri_msg_list = []
|
||||||
|
yiri_msg_list.append(
|
||||||
|
platform_message.Source(id=message_id, time=datetime.datetime.now())
|
||||||
|
)
|
||||||
|
image_base64, image_format = await image.get_wecom_image_base64(pic_url=picurl)
|
||||||
|
yiri_msg_list.append(platform_message.Image(base64=f"data:image/{image_format};base64,{image_base64}"))
|
||||||
|
chain = platform_message.MessageChain(yiri_msg_list)
|
||||||
|
|
||||||
|
return chain
|
||||||
|
|
||||||
|
|
||||||
|
class WecomEventConverter:
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def yiri2target(
|
||||||
|
event: platform_events.Event, bot_account_id: int, bot: WecomClient
|
||||||
|
) -> WecomEvent:
|
||||||
|
# only for extracting user information
|
||||||
|
|
||||||
|
if type(event) is platform_events.GroupMessage:
|
||||||
|
pass
|
||||||
|
|
||||||
|
if type(event) is platform_events.FriendMessage:
|
||||||
|
|
||||||
|
payload = {
|
||||||
|
"MsgType": "text",
|
||||||
|
"Content": '',
|
||||||
|
"FromUserName": event.sender.id,
|
||||||
|
"ToUserName": bot_account_id,
|
||||||
|
"CreateTime": int(datetime.datetime.now().timestamp()),
|
||||||
|
"AgentID": event.sender.nickname,
|
||||||
|
}
|
||||||
|
wecom_event = WecomEvent.from_payload(payload=payload)
|
||||||
|
if not wecom_event:
|
||||||
|
raise ValueError("无法从 message_data 构造 WecomEvent 对象")
|
||||||
|
|
||||||
|
return wecom_event
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def target2yiri(event: WecomEvent):
|
||||||
|
"""
|
||||||
|
将 WecomEvent 转换为平台的 FriendMessage 对象。
|
||||||
|
|
||||||
|
Args:
|
||||||
|
event (WecomEvent): 企业微信事件。
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
platform_events.FriendMessage: 转换后的 FriendMessage 对象。
|
||||||
|
"""
|
||||||
|
# 转换消息链
|
||||||
|
if event.type == "text":
|
||||||
|
yiri_chain = await WecomMessageConverter.target2yiri(
|
||||||
|
event.message, event.message_id
|
||||||
|
)
|
||||||
|
|
||||||
|
friend = platform_entities.Friend(
|
||||||
|
id=event.user_id,
|
||||||
|
nickname=str(event.agent_id),
|
||||||
|
remark="",
|
||||||
|
)
|
||||||
|
|
||||||
|
return platform_events.FriendMessage(
|
||||||
|
sender=friend, message_chain=yiri_chain, time=event.timestamp
|
||||||
|
)
|
||||||
|
elif event.type == "image":
|
||||||
|
friend = platform_entities.Friend(
|
||||||
|
id=event.user_id,
|
||||||
|
nickname=str(event.agent_id),
|
||||||
|
remark="",
|
||||||
|
)
|
||||||
|
|
||||||
|
yiri_chain = await WecomMessageConverter.target2yiri_image(
|
||||||
|
picurl=event.picurl, message_id=event.message_id
|
||||||
|
)
|
||||||
|
|
||||||
|
return platform_events.FriendMessage(
|
||||||
|
sender=friend, message_chain=yiri_chain, time=event.timestamp
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@adapter.adapter_class("wecom")
|
||||||
|
class WecomeAdapter(adapter.MessageSourceAdapter):
|
||||||
|
|
||||||
|
bot: WecomClient
|
||||||
|
ap: app.Application
|
||||||
|
bot_account_id: str
|
||||||
|
message_converter: WecomMessageConverter = WecomMessageConverter()
|
||||||
|
event_converter: WecomEventConverter = WecomEventConverter()
|
||||||
|
config: dict
|
||||||
|
ap: app.Application
|
||||||
|
|
||||||
|
def __init__(self, config: dict, ap: app.Application):
|
||||||
|
self.config = config
|
||||||
|
|
||||||
|
self.ap = ap
|
||||||
|
|
||||||
|
required_keys = [
|
||||||
|
"corpid",
|
||||||
|
"secret",
|
||||||
|
"token",
|
||||||
|
"EncodingAESKey",
|
||||||
|
"contacts_secret",
|
||||||
|
]
|
||||||
|
missing_keys = [key for key in required_keys if key not in config]
|
||||||
|
if missing_keys:
|
||||||
|
raise ParamNotEnoughError("企业微信缺少相关配置项,请查看文档或联系管理员")
|
||||||
|
|
||||||
|
self.bot = WecomClient(
|
||||||
|
corpid=config["corpid"],
|
||||||
|
secret=config["secret"],
|
||||||
|
token=config["token"],
|
||||||
|
EncodingAESKey=config["EncodingAESKey"],
|
||||||
|
contacts_secret=config["contacts_secret"],
|
||||||
|
)
|
||||||
|
|
||||||
|
async def reply_message(
|
||||||
|
self,
|
||||||
|
message_source: platform_events.MessageEvent,
|
||||||
|
message: platform_message.MessageChain,
|
||||||
|
quote_origin: bool = False,
|
||||||
|
):
|
||||||
|
|
||||||
|
Wecom_event = await WecomEventConverter.yiri2target(
|
||||||
|
message_source, self.bot_account_id, self.bot
|
||||||
|
)
|
||||||
|
content_list = await WecomMessageConverter.yiri2target(message, self.bot)
|
||||||
|
|
||||||
|
for content in content_list:
|
||||||
|
if content["type"] == "text":
|
||||||
|
await self.bot.send_private_msg(Wecom_event.user_id, Wecom_event.agent_id, content["content"])
|
||||||
|
elif content["type"] == "image":
|
||||||
|
await self.bot.send_image(Wecom_event.user_id, Wecom_event.agent_id, content["media_id"])
|
||||||
|
|
||||||
|
async def send_message(
|
||||||
|
self, target_type: str, target_id: str, message: platform_message.MessageChain
|
||||||
|
):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def register_listener(
|
||||||
|
self,
|
||||||
|
event_type: typing.Type[platform_events.Event],
|
||||||
|
callback: typing.Callable[
|
||||||
|
[platform_events.Event, adapter.MessageSourceAdapter], None
|
||||||
|
],
|
||||||
|
):
|
||||||
|
async def on_message(event: WecomEvent):
|
||||||
|
self.bot_account_id = event.receiver_id
|
||||||
|
try:
|
||||||
|
return await callback(
|
||||||
|
await self.event_converter.target2yiri(event), self
|
||||||
|
)
|
||||||
|
except:
|
||||||
|
traceback.print_exc()
|
||||||
|
|
||||||
|
if event_type == platform_events.FriendMessage:
|
||||||
|
self.bot.on_message("text")(on_message)
|
||||||
|
self.bot.on_message("image")(on_message)
|
||||||
|
elif event_type == platform_events.GroupMessage:
|
||||||
|
pass
|
||||||
|
|
||||||
|
async def run_async(self):
|
||||||
|
async def shutdown_trigger_placeholder():
|
||||||
|
while True:
|
||||||
|
await asyncio.sleep(1)
|
||||||
|
|
||||||
|
await self.bot.run_task(
|
||||||
|
host=self.config["host"],
|
||||||
|
port=self.config["port"],
|
||||||
|
shutdown_trigger=shutdown_trigger_placeholder,
|
||||||
|
)
|
||||||
|
|
||||||
|
async def kill(self) -> bool:
|
||||||
|
return False
|
||||||
|
|
||||||
|
async def unregister_listener(
|
||||||
|
self,
|
||||||
|
event_type: type,
|
||||||
|
callback: typing.Callable[[platform_events.Event, MessageSourceAdapter], None],
|
||||||
|
):
|
||||||
|
return super().unregister_listener(event_type, callback)
|
||||||
@@ -25,7 +25,7 @@ class Entity(pydantic.BaseModel):
|
|||||||
|
|
||||||
class Friend(Entity):
|
class Friend(Entity):
|
||||||
"""好友。"""
|
"""好友。"""
|
||||||
id: int
|
id: typing.Union[int, str]
|
||||||
"""QQ 号。"""
|
"""QQ 号。"""
|
||||||
nickname: typing.Optional[str]
|
nickname: typing.Optional[str]
|
||||||
"""昵称。"""
|
"""昵称。"""
|
||||||
@@ -52,7 +52,7 @@ class Permission(str, Enum):
|
|||||||
|
|
||||||
class Group(Entity):
|
class Group(Entity):
|
||||||
"""群。"""
|
"""群。"""
|
||||||
id: int
|
id: typing.Union[int, str]
|
||||||
"""群号。"""
|
"""群号。"""
|
||||||
name: str
|
name: str
|
||||||
"""群名称。"""
|
"""群名称。"""
|
||||||
@@ -67,7 +67,7 @@ class Group(Entity):
|
|||||||
|
|
||||||
class GroupMember(Entity):
|
class GroupMember(Entity):
|
||||||
"""群成员。"""
|
"""群成员。"""
|
||||||
id: int
|
id: typing.Union[int, str]
|
||||||
"""QQ 号。"""
|
"""QQ 号。"""
|
||||||
member_name: str
|
member_name: str
|
||||||
"""群成员名称。"""
|
"""群成员名称。"""
|
||||||
@@ -92,7 +92,7 @@ class GroupMember(Entity):
|
|||||||
|
|
||||||
class Client(Entity):
|
class Client(Entity):
|
||||||
"""来自其他客户端的用户。"""
|
"""来自其他客户端的用户。"""
|
||||||
id: int
|
id: typing.Union[int, str]
|
||||||
"""识别 id。"""
|
"""识别 id。"""
|
||||||
platform: str
|
platform: str
|
||||||
"""来源平台。"""
|
"""来源平台。"""
|
||||||
@@ -105,7 +105,7 @@ class Client(Entity):
|
|||||||
|
|
||||||
class Subject(pydantic.BaseModel):
|
class Subject(pydantic.BaseModel):
|
||||||
"""另一种实体类型表示。"""
|
"""另一种实体类型表示。"""
|
||||||
id: int
|
id: typing.Union[int, str]
|
||||||
"""QQ 号或群号。"""
|
"""QQ 号或群号。"""
|
||||||
kind: typing.Literal['Friend', 'Group', 'Stranger']
|
kind: typing.Literal['Friend', 'Group', 'Stranger']
|
||||||
"""类型。"""
|
"""类型。"""
|
||||||
|
|||||||
@@ -485,11 +485,11 @@ class Quote(MessageComponent):
|
|||||||
"""消息组件类型。"""
|
"""消息组件类型。"""
|
||||||
id: typing.Optional[int] = None
|
id: typing.Optional[int] = None
|
||||||
"""被引用回复的原消息的 message_id。"""
|
"""被引用回复的原消息的 message_id。"""
|
||||||
group_id: typing.Optional[int] = None
|
group_id: typing.Optional[typing.Union[int, str]] = None
|
||||||
"""被引用回复的原消息所接收的群号,当为好友消息时为0。"""
|
"""被引用回复的原消息所接收的群号,当为好友消息时为0。"""
|
||||||
sender_id: typing.Optional[int] = None
|
sender_id: typing.Optional[typing.Union[int, str]] = None
|
||||||
"""被引用回复的原消息的发送者的QQ号。"""
|
"""被引用回复的原消息的发送者的QQ号。"""
|
||||||
target_id: typing.Optional[int] = None
|
target_id: typing.Optional[typing.Union[int, str]] = None
|
||||||
"""被引用回复的原消息的接收者者的QQ号(或群号)。"""
|
"""被引用回复的原消息的接收者者的QQ号(或群号)。"""
|
||||||
origin: MessageChain
|
origin: MessageChain
|
||||||
"""被引用回复的原消息的消息链对象。"""
|
"""被引用回复的原消息的消息链对象。"""
|
||||||
@@ -749,7 +749,7 @@ class Voice(MessageComponent):
|
|||||||
|
|
||||||
class ForwardMessageNode(pydantic.BaseModel):
|
class ForwardMessageNode(pydantic.BaseModel):
|
||||||
"""合并转发中的一条消息。"""
|
"""合并转发中的一条消息。"""
|
||||||
sender_id: typing.Optional[int] = None
|
sender_id: typing.Optional[typing.Union[int, str]] = None
|
||||||
"""发送人QQ号。"""
|
"""发送人QQ号。"""
|
||||||
sender_name: typing.Optional[str] = None
|
sender_name: typing.Optional[str] = None
|
||||||
"""显示名称。"""
|
"""显示名称。"""
|
||||||
|
|||||||
@@ -9,6 +9,7 @@ from . import events
|
|||||||
from ..provider.tools import entities as tools_entities
|
from ..provider.tools import entities as tools_entities
|
||||||
from ..core import app
|
from ..core import app
|
||||||
from ..platform.types import message as platform_message
|
from ..platform.types import message as platform_message
|
||||||
|
from ..platform import adapter as platform_adapter
|
||||||
|
|
||||||
|
|
||||||
def register(
|
def register(
|
||||||
@@ -113,6 +114,37 @@ class APIHost:
|
|||||||
async def initialize(self):
|
async def initialize(self):
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
# ========== 插件可调用的 API(主程序API) ==========
|
||||||
|
|
||||||
|
def get_platform_adapters(self) -> list[platform_adapter.MessageSourceAdapter]:
|
||||||
|
"""获取已启用的消息平台适配器列表
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
list[platform.adapter.MessageSourceAdapter]: 已启用的消息平台适配器列表
|
||||||
|
"""
|
||||||
|
return self.ap.platform_mgr.adapters
|
||||||
|
|
||||||
|
async def send_active_message(
|
||||||
|
self,
|
||||||
|
adapter: platform_adapter.MessageSourceAdapter,
|
||||||
|
target_type: str,
|
||||||
|
target_id: str,
|
||||||
|
message: platform_message.MessageChain,
|
||||||
|
):
|
||||||
|
"""发送主动消息
|
||||||
|
|
||||||
|
Args:
|
||||||
|
adapter (platform.adapter.MessageSourceAdapter): 消息平台适配器对象,调用 host.get_platform_adapters() 获取并取用其中某个
|
||||||
|
target_type (str): 目标类型,`person`或`group`
|
||||||
|
target_id (str): 目标ID
|
||||||
|
message (platform.types.MessageChain): 消息链
|
||||||
|
"""
|
||||||
|
await adapter.send_message(
|
||||||
|
target_type=target_type,
|
||||||
|
target_id=target_id,
|
||||||
|
message=message,
|
||||||
|
)
|
||||||
|
|
||||||
def require_ver(
|
def require_ver(
|
||||||
self,
|
self,
|
||||||
ge: str,
|
ge: str,
|
||||||
@@ -127,16 +159,16 @@ class APIHost:
|
|||||||
Returns:
|
Returns:
|
||||||
bool: 是否满足要求, False时为无法获取版本号,True时为满足要求,报错为不满足要求
|
bool: 是否满足要求, False时为无法获取版本号,True时为满足要求,报错为不满足要求
|
||||||
"""
|
"""
|
||||||
qchatgpt_version = ""
|
langbot_version = ""
|
||||||
|
|
||||||
try:
|
try:
|
||||||
qchatgpt_version = self.ap.ver_mgr.get_current_version() # 从updater模块获取版本号
|
langbot_version = self.ap.ver_mgr.get_current_version() # 从updater模块获取版本号
|
||||||
except:
|
except:
|
||||||
return False
|
return False
|
||||||
|
|
||||||
if self.ap.ver_mgr.compare_version_str(qchatgpt_version, ge) < 0 or \
|
if self.ap.ver_mgr.compare_version_str(langbot_version, ge) < 0 or \
|
||||||
(self.ap.ver_mgr.compare_version_str(qchatgpt_version, le) > 0):
|
(self.ap.ver_mgr.compare_version_str(langbot_version, le) > 0):
|
||||||
raise Exception("LangBot 版本不满足要求,某些功能(可能是由插件提供的)无法正常使用。(要求版本:{}-{},但当前版本:{})".format(ge, le, qchatgpt_version))
|
raise Exception("LangBot 版本不满足要求,某些功能(可能是由插件提供的)无法正常使用。(要求版本:{}-{},但当前版本:{})".format(ge, le, langbot_version))
|
||||||
|
|
||||||
return True
|
return True
|
||||||
|
|
||||||
|
|||||||
@@ -25,10 +25,10 @@ class PersonMessageReceived(BaseEventModel):
|
|||||||
launcher_type: str
|
launcher_type: str
|
||||||
"""发起对象类型(group/person)"""
|
"""发起对象类型(group/person)"""
|
||||||
|
|
||||||
launcher_id: int
|
launcher_id: typing.Union[int, str]
|
||||||
"""发起对象ID(群号/QQ号)"""
|
"""发起对象ID(群号/QQ号)"""
|
||||||
|
|
||||||
sender_id: int
|
sender_id: typing.Union[int, str]
|
||||||
"""发送者ID(QQ号)"""
|
"""发送者ID(QQ号)"""
|
||||||
|
|
||||||
message_chain: platform_message.MessageChain
|
message_chain: platform_message.MessageChain
|
||||||
@@ -39,9 +39,9 @@ class GroupMessageReceived(BaseEventModel):
|
|||||||
|
|
||||||
launcher_type: str
|
launcher_type: str
|
||||||
|
|
||||||
launcher_id: int
|
launcher_id: typing.Union[int, str]
|
||||||
|
|
||||||
sender_id: int
|
sender_id: typing.Union[int, str]
|
||||||
|
|
||||||
message_chain: platform_message.MessageChain
|
message_chain: platform_message.MessageChain
|
||||||
|
|
||||||
@@ -51,9 +51,9 @@ class PersonNormalMessageReceived(BaseEventModel):
|
|||||||
|
|
||||||
launcher_type: str
|
launcher_type: str
|
||||||
|
|
||||||
launcher_id: int
|
launcher_id: typing.Union[int, str]
|
||||||
|
|
||||||
sender_id: int
|
sender_id: typing.Union[int, str]
|
||||||
|
|
||||||
text_message: str
|
text_message: str
|
||||||
|
|
||||||
@@ -69,9 +69,9 @@ class PersonCommandSent(BaseEventModel):
|
|||||||
|
|
||||||
launcher_type: str
|
launcher_type: str
|
||||||
|
|
||||||
launcher_id: int
|
launcher_id: typing.Union[int, str]
|
||||||
|
|
||||||
sender_id: int
|
sender_id: typing.Union[int, str]
|
||||||
|
|
||||||
command: str
|
command: str
|
||||||
|
|
||||||
@@ -93,9 +93,9 @@ class GroupNormalMessageReceived(BaseEventModel):
|
|||||||
|
|
||||||
launcher_type: str
|
launcher_type: str
|
||||||
|
|
||||||
launcher_id: int
|
launcher_id: typing.Union[int, str]
|
||||||
|
|
||||||
sender_id: int
|
sender_id: typing.Union[int, str]
|
||||||
|
|
||||||
text_message: str
|
text_message: str
|
||||||
|
|
||||||
@@ -111,9 +111,9 @@ class GroupCommandSent(BaseEventModel):
|
|||||||
|
|
||||||
launcher_type: str
|
launcher_type: str
|
||||||
|
|
||||||
launcher_id: int
|
launcher_id: typing.Union[int, str]
|
||||||
|
|
||||||
sender_id: int
|
sender_id: typing.Union[int, str]
|
||||||
|
|
||||||
command: str
|
command: str
|
||||||
|
|
||||||
@@ -135,9 +135,9 @@ class NormalMessageResponded(BaseEventModel):
|
|||||||
|
|
||||||
launcher_type: str
|
launcher_type: str
|
||||||
|
|
||||||
launcher_id: int
|
launcher_id: typing.Union[int, str]
|
||||||
|
|
||||||
sender_id: int
|
sender_id: typing.Union[int, str]
|
||||||
|
|
||||||
session: core_entities.Session
|
session: core_entities.Session
|
||||||
"""会话对象"""
|
"""会话对象"""
|
||||||
|
|||||||
@@ -6,7 +6,7 @@ from . import entities, requester
|
|||||||
from ...core import app
|
from ...core import app
|
||||||
|
|
||||||
from . import token
|
from . import token
|
||||||
from .requesters import chatcmpl, anthropicmsgs, moonshotchatcmpl, deepseekchatcmpl, ollamachat, giteeaichatcmpl
|
from .requesters import chatcmpl, anthropicmsgs, moonshotchatcmpl, deepseekchatcmpl, ollamachat, giteeaichatcmpl, xaichatcmpl, zhipuaichatcmpl
|
||||||
|
|
||||||
FETCH_MODEL_LIST_URL = "https://api.qchatgpt.rockchin.top/api/v2/fetch/model_list"
|
FETCH_MODEL_LIST_URL = "https://api.qchatgpt.rockchin.top/api/v2/fetch/model_list"
|
||||||
|
|
||||||
|
|||||||
@@ -124,7 +124,7 @@ class OpenAIChatCompletions(requester.LLMAPIRequester):
|
|||||||
req_messages.append(msg_dict)
|
req_messages.append(msg_dict)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
return await self._closure(query, req_messages, model, funcs)
|
return await self._closure(query=query, req_messages=req_messages, use_model=model, use_funcs=funcs)
|
||||||
except asyncio.TimeoutError:
|
except asyncio.TimeoutError:
|
||||||
raise errors.RequesterError('请求超时')
|
raise errors.RequesterError('请求超时')
|
||||||
except openai.BadRequestError as e:
|
except openai.BadRequestError as e:
|
||||||
|
|||||||
@@ -1,7 +1,5 @@
|
|||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
|
|
||||||
from ....core import app
|
|
||||||
|
|
||||||
from . import chatcmpl
|
from . import chatcmpl
|
||||||
from .. import entities, errors, requester
|
from .. import entities, errors, requester
|
||||||
from ....core import entities as core_entities, app
|
from ....core import entities as core_entities, app
|
||||||
@@ -19,6 +17,7 @@ class DeepseekChatCompletions(chatcmpl.OpenAIChatCompletions):
|
|||||||
|
|
||||||
async def _closure(
|
async def _closure(
|
||||||
self,
|
self,
|
||||||
|
query: core_entities.Query,
|
||||||
req_messages: list[dict],
|
req_messages: list[dict],
|
||||||
use_model: entities.LLMModelInfo,
|
use_model: entities.LLMModelInfo,
|
||||||
use_funcs: list[tools_entities.LLMFunction] = None,
|
use_funcs: list[tools_entities.LLMFunction] = None,
|
||||||
|
|||||||
@@ -8,7 +8,7 @@ import typing
|
|||||||
|
|
||||||
from . import chatcmpl
|
from . import chatcmpl
|
||||||
from .. import entities, errors, requester
|
from .. import entities, errors, requester
|
||||||
from ....core import app
|
from ....core import app, entities as core_entities
|
||||||
from ... import entities as llm_entities
|
from ... import entities as llm_entities
|
||||||
from ...tools import entities as tools_entities
|
from ...tools import entities as tools_entities
|
||||||
from .. import entities as modelmgr_entities
|
from .. import entities as modelmgr_entities
|
||||||
@@ -24,6 +24,7 @@ class GiteeAIChatCompletions(chatcmpl.OpenAIChatCompletions):
|
|||||||
|
|
||||||
async def _closure(
|
async def _closure(
|
||||||
self,
|
self,
|
||||||
|
query: core_entities.Query,
|
||||||
req_messages: list[dict],
|
req_messages: list[dict],
|
||||||
use_model: entities.LLMModelInfo,
|
use_model: entities.LLMModelInfo,
|
||||||
use_funcs: list[tools_entities.LLMFunction] = None,
|
use_funcs: list[tools_entities.LLMFunction] = None,
|
||||||
|
|||||||
@@ -19,6 +19,7 @@ class MoonshotChatCompletions(chatcmpl.OpenAIChatCompletions):
|
|||||||
|
|
||||||
async def _closure(
|
async def _closure(
|
||||||
self,
|
self,
|
||||||
|
query: core_entities.Query,
|
||||||
req_messages: list[dict],
|
req_messages: list[dict],
|
||||||
use_model: entities.LLMModelInfo,
|
use_model: entities.LLMModelInfo,
|
||||||
use_funcs: list[tools_entities.LLMFunction] = None,
|
use_funcs: list[tools_entities.LLMFunction] = None,
|
||||||
|
|||||||
21
pkg/provider/modelmgr/requesters/xaichatcmpl.py
Normal file
21
pkg/provider/modelmgr/requesters/xaichatcmpl.py
Normal file
@@ -0,0 +1,21 @@
|
|||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
import openai
|
||||||
|
|
||||||
|
from . import chatcmpl
|
||||||
|
from .. import requester
|
||||||
|
from ....core import app
|
||||||
|
|
||||||
|
|
||||||
|
@requester.requester_class("xai-chat-completions")
|
||||||
|
class XaiChatCompletions(chatcmpl.OpenAIChatCompletions):
|
||||||
|
"""xAI ChatCompletion API 请求器"""
|
||||||
|
|
||||||
|
client: openai.AsyncClient
|
||||||
|
|
||||||
|
requester_cfg: dict
|
||||||
|
|
||||||
|
def __init__(self, ap: app.Application):
|
||||||
|
self.ap = ap
|
||||||
|
|
||||||
|
self.requester_cfg = self.ap.provider_cfg.data['requester']['xai-chat-completions']
|
||||||
21
pkg/provider/modelmgr/requesters/zhipuaichatcmpl.py
Normal file
21
pkg/provider/modelmgr/requesters/zhipuaichatcmpl.py
Normal file
@@ -0,0 +1,21 @@
|
|||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
import openai
|
||||||
|
|
||||||
|
from ....core import app
|
||||||
|
from . import chatcmpl
|
||||||
|
from .. import requester
|
||||||
|
|
||||||
|
|
||||||
|
@requester.requester_class("zhipuai-chat-completions")
|
||||||
|
class ZhipuAIChatCompletions(chatcmpl.OpenAIChatCompletions):
|
||||||
|
"""智谱AI ChatCompletion API 请求器"""
|
||||||
|
|
||||||
|
client: openai.AsyncClient
|
||||||
|
|
||||||
|
requester_cfg: dict
|
||||||
|
|
||||||
|
def __init__(self, ap: app.Application):
|
||||||
|
self.ap = ap
|
||||||
|
|
||||||
|
self.requester_cfg = self.ap.provider_cfg.data['requester']['zhipuai-chat-completions']
|
||||||
@@ -5,6 +5,8 @@ import json
|
|||||||
import uuid
|
import uuid
|
||||||
import base64
|
import base64
|
||||||
|
|
||||||
|
import aiohttp
|
||||||
|
|
||||||
from .. import runner
|
from .. import runner
|
||||||
from ...core import entities as core_entities
|
from ...core import entities as core_entities
|
||||||
from .. import entities as llm_entities
|
from .. import entities as llm_entities
|
||||||
@@ -97,7 +99,7 @@ class DifyServiceAPIRunner(runner.RequestRunner):
|
|||||||
files=files,
|
files=files,
|
||||||
timeout=self.ap.provider_cfg.data["dify-service-api"]["chat"]["timeout"],
|
timeout=self.ap.provider_cfg.data["dify-service-api"]["chat"]["timeout"],
|
||||||
):
|
):
|
||||||
self.ap.logger.debug("dify-chat-chunk: ", chunk)
|
self.ap.logger.debug("dify-chat-chunk: " + str(chunk))
|
||||||
|
|
||||||
if chunk['event'] == 'workflow_started':
|
if chunk['event'] == 'workflow_started':
|
||||||
mode = "workflow"
|
mode = "workflow"
|
||||||
@@ -149,7 +151,8 @@ class DifyServiceAPIRunner(runner.RequestRunner):
|
|||||||
files=files,
|
files=files,
|
||||||
timeout=self.ap.provider_cfg.data["dify-service-api"]["chat"]["timeout"],
|
timeout=self.ap.provider_cfg.data["dify-service-api"]["chat"]["timeout"],
|
||||||
):
|
):
|
||||||
self.ap.logger.debug("dify-agent-chunk: ", chunk)
|
self.ap.logger.debug("dify-agent-chunk: " + str(chunk))
|
||||||
|
|
||||||
if chunk["event"] in ignored_events:
|
if chunk["event"] in ignored_events:
|
||||||
continue
|
continue
|
||||||
if chunk["event"] == "agent_thought":
|
if chunk["event"] == "agent_thought":
|
||||||
@@ -179,6 +182,21 @@ class DifyServiceAPIRunner(runner.RequestRunner):
|
|||||||
],
|
],
|
||||||
)
|
)
|
||||||
yield msg
|
yield msg
|
||||||
|
if chunk['event'] == 'message_file':
|
||||||
|
|
||||||
|
if chunk['type'] == 'image' and chunk['belongs_to'] == 'assistant':
|
||||||
|
|
||||||
|
base_url = self.dify_client.base_url
|
||||||
|
|
||||||
|
if base_url.endswith('/v1'):
|
||||||
|
base_url = base_url[:-3]
|
||||||
|
|
||||||
|
image_url = base_url + chunk['url']
|
||||||
|
|
||||||
|
yield llm_entities.Message(
|
||||||
|
role="assistant",
|
||||||
|
content=[llm_entities.ContentElement.from_image_url(image_url)],
|
||||||
|
)
|
||||||
|
|
||||||
query.session.using_conversation.uuid = chunk["conversation_id"]
|
query.session.using_conversation.uuid = chunk["conversation_id"]
|
||||||
|
|
||||||
@@ -215,7 +233,7 @@ class DifyServiceAPIRunner(runner.RequestRunner):
|
|||||||
files=files,
|
files=files,
|
||||||
timeout=self.ap.provider_cfg.data["dify-service-api"]["workflow"]["timeout"],
|
timeout=self.ap.provider_cfg.data["dify-service-api"]["workflow"]["timeout"],
|
||||||
):
|
):
|
||||||
self.ap.logger.debug("dify-workflow-chunk: ", chunk)
|
self.ap.logger.debug("dify-workflow-chunk: " + str(chunk))
|
||||||
if chunk["event"] in ignored_events:
|
if chunk["event"] in ignored_events:
|
||||||
continue
|
continue
|
||||||
|
|
||||||
|
|||||||
@@ -62,11 +62,11 @@ class AnnouncementManager:
|
|||||||
async def fetch_saved(
|
async def fetch_saved(
|
||||||
self
|
self
|
||||||
) -> list[Announcement]:
|
) -> list[Announcement]:
|
||||||
if not os.path.exists("res/announcement_saved.json"):
|
if not os.path.exists("data/labels/announcement_saved.json"):
|
||||||
with open("res/announcement_saved.json", "w", encoding="utf-8") as f:
|
with open("data/labels/announcement_saved.json", "w", encoding="utf-8") as f:
|
||||||
f.write("[]")
|
f.write("[]")
|
||||||
|
|
||||||
with open("res/announcement_saved.json", "r", encoding="utf-8") as f:
|
with open("data/labels/announcement_saved.json", "r", encoding="utf-8") as f:
|
||||||
content = f.read()
|
content = f.read()
|
||||||
|
|
||||||
if not content:
|
if not content:
|
||||||
@@ -79,7 +79,7 @@ class AnnouncementManager:
|
|||||||
content: list[Announcement]
|
content: list[Announcement]
|
||||||
):
|
):
|
||||||
|
|
||||||
with open("res/announcement_saved.json", "w", encoding="utf-8") as f:
|
with open("data/labels/announcement_saved.json", "w", encoding="utf-8") as f:
|
||||||
f.write(json.dumps([
|
f.write(json.dumps([
|
||||||
item.to_dict() for item in content
|
item.to_dict() for item in content
|
||||||
], indent=4, ensure_ascii=False))
|
], indent=4, ensure_ascii=False))
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
semantic_version = "v3.4.1.4"
|
semantic_version = "v3.4.3"
|
||||||
|
|
||||||
debug_mode = False
|
debug_mode = False
|
||||||
|
|
||||||
|
|||||||
@@ -7,6 +7,31 @@ import ssl
|
|||||||
import aiohttp
|
import aiohttp
|
||||||
import PIL.Image
|
import PIL.Image
|
||||||
|
|
||||||
|
async def get_wecom_image_base64(pic_url: str) -> tuple[str, str]:
|
||||||
|
"""
|
||||||
|
下载企业微信图片并转换为 base64
|
||||||
|
:param pic_url: 企业微信图片URL
|
||||||
|
:return: (base64_str, image_format)
|
||||||
|
"""
|
||||||
|
async with aiohttp.ClientSession() as session:
|
||||||
|
async with session.get(pic_url) as response:
|
||||||
|
if response.status != 200:
|
||||||
|
raise Exception(f"Failed to download image: {response.status}")
|
||||||
|
|
||||||
|
# 读取图片数据
|
||||||
|
image_data = await response.read()
|
||||||
|
|
||||||
|
# 获取图片格式
|
||||||
|
content_type = response.headers.get('Content-Type', '')
|
||||||
|
image_format = content_type.split('/')[-1] # 例如 'image/jpeg' -> 'jpeg'
|
||||||
|
|
||||||
|
# 转换为 base64
|
||||||
|
import base64
|
||||||
|
image_base64 = base64.b64encode(image_data).decode('utf-8')
|
||||||
|
|
||||||
|
return image_base64, image_format
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
def get_qq_image_downloadable_url(image_url: str) -> tuple[str, dict]:
|
def get_qq_image_downloadable_url(image_url: str) -> tuple[str, dict]:
|
||||||
"""获取QQ图片的下载链接"""
|
"""获取QQ图片的下载链接"""
|
||||||
|
|||||||
@@ -24,6 +24,7 @@ aiofiles
|
|||||||
aioshutil
|
aioshutil
|
||||||
argon2-cffi
|
argon2-cffi
|
||||||
pyjwt
|
pyjwt
|
||||||
|
pycryptodome
|
||||||
|
|
||||||
# indirect
|
# indirect
|
||||||
taskgroup==0.0.0a4
|
taskgroup==0.0.0a4
|
||||||
@@ -115,6 +115,97 @@
|
|||||||
"name": "deepseek-coder",
|
"name": "deepseek-coder",
|
||||||
"requester": "deepseek-chat-completions",
|
"requester": "deepseek-chat-completions",
|
||||||
"token_mgr": "deepseek"
|
"token_mgr": "deepseek"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "grok-2-latest",
|
||||||
|
"requester": "xai-chat-completions",
|
||||||
|
"token_mgr": "xai"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "grok-2",
|
||||||
|
"requester": "xai-chat-completions",
|
||||||
|
"token_mgr": "xai"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "grok-2-vision-1212",
|
||||||
|
"requester": "xai-chat-completions",
|
||||||
|
"token_mgr": "xai",
|
||||||
|
"vision_supported": true
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "grok-2-1212",
|
||||||
|
"requester": "xai-chat-completions",
|
||||||
|
"token_mgr": "xai"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "grok-vision-beta",
|
||||||
|
"requester": "xai-chat-completions",
|
||||||
|
"token_mgr": "xai",
|
||||||
|
"vision_supported": true
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "grok-beta",
|
||||||
|
"requester": "xai-chat-completions",
|
||||||
|
"token_mgr": "xai"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "glm-4-plus",
|
||||||
|
"requester": "zhipuai-chat-completions",
|
||||||
|
"token_mgr": "zhipuai"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "glm-4-0520",
|
||||||
|
"requester": "zhipuai-chat-completions",
|
||||||
|
"token_mgr": "zhipuai"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "glm-4-air",
|
||||||
|
"requester": "zhipuai-chat-completions",
|
||||||
|
"token_mgr": "zhipuai"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "glm-4-airx",
|
||||||
|
"requester": "zhipuai-chat-completions",
|
||||||
|
"token_mgr": "zhipuai"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "glm-4-long",
|
||||||
|
"requester": "zhipuai-chat-completions",
|
||||||
|
"token_mgr": "zhipuai"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "glm-4-flashx",
|
||||||
|
"requester": "zhipuai-chat-completions",
|
||||||
|
"token_mgr": "zhipuai"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "glm-4-flash",
|
||||||
|
"requester": "zhipuai-chat-completions",
|
||||||
|
"token_mgr": "zhipuai"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "glm-4v-plus",
|
||||||
|
"requester": "zhipuai-chat-completions",
|
||||||
|
"token_mgr": "zhipuai",
|
||||||
|
"vision_supported": true
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "glm-4v",
|
||||||
|
"requester": "zhipuai-chat-completions",
|
||||||
|
"token_mgr": "zhipuai",
|
||||||
|
"vision_supported": true
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "glm-4v-flash",
|
||||||
|
"requester": "zhipuai-chat-completions",
|
||||||
|
"token_mgr": "zhipuai",
|
||||||
|
"vision_supported": true
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "glm-zero-preview",
|
||||||
|
"requester": "zhipuai-chat-completions",
|
||||||
|
"token_mgr": "zhipuai",
|
||||||
|
"vision_supported": true
|
||||||
}
|
}
|
||||||
]
|
]
|
||||||
}
|
}
|
||||||
@@ -24,6 +24,17 @@
|
|||||||
"public_guild_messages",
|
"public_guild_messages",
|
||||||
"direct_message"
|
"direct_message"
|
||||||
]
|
]
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"adapter": "wecom",
|
||||||
|
"enable": false,
|
||||||
|
"host": "0.0.0.0",
|
||||||
|
"port": 2290,
|
||||||
|
"corpid": "",
|
||||||
|
"secret": "",
|
||||||
|
"token": "",
|
||||||
|
"EncodingAESKey": "",
|
||||||
|
"contacts_secret": ""
|
||||||
}
|
}
|
||||||
],
|
],
|
||||||
"track-function-calls": true,
|
"track-function-calls": true,
|
||||||
|
|||||||
@@ -16,6 +16,12 @@
|
|||||||
],
|
],
|
||||||
"gitee-ai": [
|
"gitee-ai": [
|
||||||
"XXXXX"
|
"XXXXX"
|
||||||
|
],
|
||||||
|
"xai": [
|
||||||
|
"xai-1234567890"
|
||||||
|
],
|
||||||
|
"zhipuai": [
|
||||||
|
"xxxxxxx"
|
||||||
]
|
]
|
||||||
},
|
},
|
||||||
"requester": {
|
"requester": {
|
||||||
@@ -50,6 +56,16 @@
|
|||||||
"base-url": "https://ai.gitee.com/v1",
|
"base-url": "https://ai.gitee.com/v1",
|
||||||
"args": {},
|
"args": {},
|
||||||
"timeout": 120
|
"timeout": 120
|
||||||
|
},
|
||||||
|
"xai-chat-completions": {
|
||||||
|
"base-url": "https://api.x.ai/v1",
|
||||||
|
"args": {},
|
||||||
|
"timeout": 120
|
||||||
|
},
|
||||||
|
"zhipuai-chat-completions": {
|
||||||
|
"base-url": "https://open.bigmodel.cn/api/paas/v4",
|
||||||
|
"args": {},
|
||||||
|
"timeout": 120
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
"model": "gpt-4o",
|
"model": "gpt-4o",
|
||||||
|
|||||||
@@ -22,7 +22,6 @@
|
|||||||
"openai": {
|
"openai": {
|
||||||
"type": "array",
|
"type": "array",
|
||||||
"title": "OpenAI API 密钥",
|
"title": "OpenAI API 密钥",
|
||||||
"description": "OpenAI API 密钥",
|
|
||||||
"items": {
|
"items": {
|
||||||
"type": "string"
|
"type": "string"
|
||||||
},
|
},
|
||||||
@@ -31,7 +30,6 @@
|
|||||||
"anthropic": {
|
"anthropic": {
|
||||||
"type": "array",
|
"type": "array",
|
||||||
"title": "Anthropic API 密钥",
|
"title": "Anthropic API 密钥",
|
||||||
"description": "Anthropic API 密钥",
|
|
||||||
"items": {
|
"items": {
|
||||||
"type": "string"
|
"type": "string"
|
||||||
},
|
},
|
||||||
@@ -40,7 +38,6 @@
|
|||||||
"moonshot": {
|
"moonshot": {
|
||||||
"type": "array",
|
"type": "array",
|
||||||
"title": "Moonshot API 密钥",
|
"title": "Moonshot API 密钥",
|
||||||
"description": "Moonshot API 密钥",
|
|
||||||
"items": {
|
"items": {
|
||||||
"type": "string"
|
"type": "string"
|
||||||
},
|
},
|
||||||
@@ -49,7 +46,6 @@
|
|||||||
"deepseek": {
|
"deepseek": {
|
||||||
"type": "array",
|
"type": "array",
|
||||||
"title": "DeepSeek API 密钥",
|
"title": "DeepSeek API 密钥",
|
||||||
"description": "DeepSeek API 密钥",
|
|
||||||
"items": {
|
"items": {
|
||||||
"type": "string"
|
"type": "string"
|
||||||
},
|
},
|
||||||
@@ -57,8 +53,23 @@
|
|||||||
},
|
},
|
||||||
"gitee": {
|
"gitee": {
|
||||||
"type": "array",
|
"type": "array",
|
||||||
"title": "Gitee API 密钥",
|
"title": "Gitee AI API 密钥",
|
||||||
"description": "Gitee API 密钥",
|
"items": {
|
||||||
|
"type": "string"
|
||||||
|
},
|
||||||
|
"default": []
|
||||||
|
},
|
||||||
|
"xai": {
|
||||||
|
"type": "array",
|
||||||
|
"title": "xAI API 密钥",
|
||||||
|
"items": {
|
||||||
|
"type": "string"
|
||||||
|
},
|
||||||
|
"default": []
|
||||||
|
},
|
||||||
|
"zhipuai": {
|
||||||
|
"type": "array",
|
||||||
|
"title": "智谱AI API 密钥",
|
||||||
"items": {
|
"items": {
|
||||||
"type": "string"
|
"type": "string"
|
||||||
},
|
},
|
||||||
@@ -188,6 +199,42 @@
|
|||||||
"default": 120
|
"default": 120
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
},
|
||||||
|
"xai-chat-completions": {
|
||||||
|
"type": "object",
|
||||||
|
"title": "xAI API 请求配置",
|
||||||
|
"description": "仅可编辑 URL 和 超时时间,额外请求参数不支持可视化编辑,请到编辑器编辑",
|
||||||
|
"properties": {
|
||||||
|
"base-url": {
|
||||||
|
"type": "string",
|
||||||
|
"title": "API URL"
|
||||||
|
},
|
||||||
|
"args": {
|
||||||
|
"type": "object"
|
||||||
|
},
|
||||||
|
"timeout": {
|
||||||
|
"type": "number",
|
||||||
|
"title": "API 请求超时时间",
|
||||||
|
"default": 120
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"zhipuai-chat-completions": {
|
||||||
|
"type": "object",
|
||||||
|
"title": "智谱AI API 请求配置",
|
||||||
|
"description": "仅可编辑 URL 和 超时时间,额外请求参数不支持可视化编辑,请到编辑器编辑",
|
||||||
|
"properties": {
|
||||||
|
"base-url": {
|
||||||
|
"type": "string",
|
||||||
|
"title": "API URL"
|
||||||
|
},
|
||||||
|
"args": {
|
||||||
|
"type": "object"
|
||||||
|
},
|
||||||
|
"timeout": {
|
||||||
|
"type": "number"
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
|||||||
@@ -79,6 +79,12 @@
|
|||||||
重载插件
|
重载插件
|
||||||
</v-list-item-title>
|
</v-list-item-title>
|
||||||
</v-list-item>
|
</v-list-item>
|
||||||
|
|
||||||
|
<v-list-item @click="reload('provider')">
|
||||||
|
<v-list-item-title>
|
||||||
|
重载 LLM 管理器
|
||||||
|
</v-list-item-title>
|
||||||
|
</v-list-item>
|
||||||
</v-list>
|
</v-list>
|
||||||
</v-menu>
|
</v-menu>
|
||||||
</v-list-item>
|
</v-list-item>
|
||||||
@@ -169,7 +175,8 @@ function openDocs() {
|
|||||||
|
|
||||||
const reloadScopeLabel = {
|
const reloadScopeLabel = {
|
||||||
'platform': "消息平台",
|
'platform': "消息平台",
|
||||||
'plugin': "插件"
|
'plugin': "插件",
|
||||||
|
'provider': "LLM 管理器"
|
||||||
}
|
}
|
||||||
|
|
||||||
function reload(scope) {
|
function reload(scope) {
|
||||||
|
|||||||
Reference in New Issue
Block a user