mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-06-05 05:16:03 +00:00
Compare commits
35 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
e775499080 | ||
|
|
735aad5a91 | ||
|
|
fb4e106f69 | ||
|
|
e5659db535 | ||
|
|
5381e09a6c | ||
|
|
21f16ecd68 | ||
|
|
12fc76b326 | ||
|
|
d7f87dd269 | ||
|
|
56227f3713 | ||
|
|
f492fee486 | ||
|
|
41a7814615 | ||
|
|
8644f2c166 | ||
|
|
e4a9365caf | ||
|
|
9fc7af1295 | ||
|
|
d0eeb2b304 | ||
|
|
e4518ebcf1 | ||
|
|
7604cefd0f | ||
|
|
71729d4784 | ||
|
|
1d16bc4968 | ||
|
|
de2bf79004 | ||
|
|
83ed7a9f38 | ||
|
|
c326e72758 | ||
|
|
ac9cef82cc | ||
|
|
ea254d57d2 | ||
|
|
a661f24ae0 | ||
|
|
afabf9256b | ||
|
|
74a8f9c9e2 | ||
|
|
1d11e448f9 | ||
|
|
e3e23cbccb | ||
|
|
79132aa11d | ||
|
|
7bb9e6e951 | ||
|
|
37dc5b4135 | ||
|
|
d588faf470 | ||
|
|
8b51a81158 | ||
|
|
9f125974bf |
6
.github/ISSUE_TEMPLATE/bug-report.yml
vendored
6
.github/ISSUE_TEMPLATE/bug-report.yml
vendored
@@ -12,6 +12,8 @@ body:
|
|||||||
- Nakuru(go-cqhttp)
|
- Nakuru(go-cqhttp)
|
||||||
- aiocqhttp(使用 OneBot 协议接入的)
|
- aiocqhttp(使用 OneBot 协议接入的)
|
||||||
- qq-botpy(QQ官方API)
|
- qq-botpy(QQ官方API)
|
||||||
|
- lark(飞书)
|
||||||
|
- wecom(企业微信)
|
||||||
validations:
|
validations:
|
||||||
required: true
|
required: true
|
||||||
- type: input
|
- type: input
|
||||||
@@ -30,9 +32,9 @@ body:
|
|||||||
- type: textarea
|
- type: textarea
|
||||||
attributes:
|
attributes:
|
||||||
label: 复现步骤
|
label: 复现步骤
|
||||||
description: 如何重现这个问题,越详细越好
|
description: 如何重现这个问题,越详细越好;请贴上所有相关的配置文件和元数据文件(注意隐去敏感信息)
|
||||||
validations:
|
validations:
|
||||||
required: false
|
required: true
|
||||||
- type: textarea
|
- type: textarea
|
||||||
attributes:
|
attributes:
|
||||||
label: 启用的插件
|
label: 启用的插件
|
||||||
|
|||||||
33
README.md
33
README.md
@@ -15,21 +15,14 @@
|
|||||||
<a href="https://docs.langbot.app/plugin/plugin-intro.html">插件介绍</a> |
|
<a href="https://docs.langbot.app/plugin/plugin-intro.html">插件介绍</a> |
|
||||||
<a href="https://github.com/RockChinQ/LangBot/issues/new?assignees=&labels=%E7%8B%AC%E7%AB%8B%E6%8F%92%E4%BB%B6&projects=&template=submit-plugin.yml&title=%5BPlugin%5D%3A+%E8%AF%B7%E6%B1%82%E7%99%BB%E8%AE%B0%E6%96%B0%E6%8F%92%E4%BB%B6">提交插件</a>
|
<a href="https://github.com/RockChinQ/LangBot/issues/new?assignees=&labels=%E7%8B%AC%E7%AB%8B%E6%8F%92%E4%BB%B6&projects=&template=submit-plugin.yml&title=%5BPlugin%5D%3A+%E8%AF%B7%E6%B1%82%E7%99%BB%E8%AE%B0%E6%96%B0%E6%8F%92%E4%BB%B6">提交插件</a>
|
||||||
|
|
||||||
|
|
||||||
<div align="center">
|
<div align="center">
|
||||||
😎高稳定、🧩支持扩展、🦄多模态 - 基于大语言模型的即时通讯机器人平台🤖
|
😎高稳定、🧩支持扩展、🦄多模态 - 大模型原生即时通信机器人平台🤖
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
<br/>
|
<br/>
|
||||||
|
|
||||||
<a href="http://qm.qq.com/cgi-bin/qm/qr?_wv=1027&k=66-aWvn8cbP4c1ut_1YYkvvGVeEtyTH8&authKey=pTaKBK5C%2B8dFzQ4XlENf6MHTCLaHnlKcCRx7c14EeVVlpX2nRSaS8lJm8YeM4mCU&noverify=0&group_code=195992197">
|
|
||||||
<img alt="Static Badge" src="https://img.shields.io/badge/%E5%AE%98%E6%96%B9%E7%BE%A4-195992197-green">
|
|
||||||
</a>
|
|
||||||
<a href="https://qm.qq.com/q/PClALFK242">
|
|
||||||
<img alt="Static Badge" src="https://img.shields.io/badge/%E7%A4%BE%E5%8C%BA%E7%BE%A4-619154800-green">
|
|
||||||
</a>
|
|
||||||
<br/>
|
|
||||||
|
|
||||||
|
[](https://qm.qq.com/q/PF9OuQCCcM)
|
||||||
[](https://github.com/RockChinQ/LangBot/releases/latest)
|
[](https://github.com/RockChinQ/LangBot/releases/latest)
|
||||||

|

|
||||||
<img src="https://img.shields.io/badge/python-3.10 | 3.11 | 3.12-blue.svg" alt="python">
|
<img src="https://img.shields.io/badge/python-3.10 | 3.11 | 3.12-blue.svg" alt="python">
|
||||||
@@ -39,7 +32,7 @@
|
|||||||
|
|
||||||
## ✨ Features
|
## ✨ Features
|
||||||
|
|
||||||
- 💬 大模型对话、Agent:支持多种大模型,适配群聊和私聊;具有多轮对话、工具调用、多模态能力,并深度适配 [Dify](https://dify.ai)。目前支持 QQ、QQ频道,后续还将支持微信、WhatsApp、Discord等平台。
|
- 💬 大模型对话、Agent:支持多种大模型,适配群聊和私聊;具有多轮对话、工具调用、多模态能力,并深度适配 [Dify](https://dify.ai)。目前支持 QQ、QQ频道、企业微信、飞书,后续还将支持微信、WhatsApp、Discord等平台。
|
||||||
- 🛠️ 高稳定性、功能完备:原生支持访问控制、限速、敏感词过滤等机制;配置简单,支持多种部署方式。
|
- 🛠️ 高稳定性、功能完备:原生支持访问控制、限速、敏感词过滤等机制;配置简单,支持多种部署方式。
|
||||||
- 🧩 插件扩展、活跃社区:支持事件驱动、组件扩展等插件机制;丰富生态,目前已有数十个[插件](https://docs.langbot.app/plugin/plugin-intro.html)
|
- 🧩 插件扩展、活跃社区:支持事件驱动、组件扩展等插件机制;丰富生态,目前已有数十个[插件](https://docs.langbot.app/plugin/plugin-intro.html)
|
||||||
- 😻 [New] Web 管理面板:支持通过浏览器管理 LangBot 实例,具体支持功能,查看[文档](https://docs.langbot.app/webui/intro.html)
|
- 😻 [New] Web 管理面板:支持通过浏览器管理 LangBot 实例,具体支持功能,查看[文档](https://docs.langbot.app/webui/intro.html)
|
||||||
@@ -87,8 +80,12 @@
|
|||||||
| 平台 | 状态 | 备注 |
|
| 平台 | 状态 | 备注 |
|
||||||
| --- | --- | --- |
|
| --- | --- | --- |
|
||||||
| QQ 个人号 | ✅ | QQ 个人号私聊、群聊 |
|
| QQ 个人号 | ✅ | QQ 个人号私聊、群聊 |
|
||||||
| QQ 官方机器人 | ✅ | QQ 频道机器人,支持频道、私聊、群聊 |
|
| QQ 官方机器人 | ✅ | QQ 官方机器人,支持频道、私聊、群聊 |
|
||||||
| 企业微信 | ✅ | |
|
| 企业微信 | ✅ | |
|
||||||
|
| 飞书 | ✅ | |
|
||||||
|
| Discord | ✅ | |
|
||||||
|
| 个人微信 | 🚧 | |
|
||||||
|
| WhatsApp | 🚧 | |
|
||||||
| 钉钉 | 🚧 | |
|
| 钉钉 | 🚧 | |
|
||||||
|
|
||||||
🚧: 正在开发中
|
🚧: 正在开发中
|
||||||
@@ -104,5 +101,17 @@
|
|||||||
| [xAI](https://x.ai/) | ✅ | |
|
| [xAI](https://x.ai/) | ✅ | |
|
||||||
| [智谱AI](https://open.bigmodel.cn/) | ✅ | |
|
| [智谱AI](https://open.bigmodel.cn/) | ✅ | |
|
||||||
| [Dify](https://dify.ai) | ✅ | LLMOps 平台 |
|
| [Dify](https://dify.ai) | ✅ | LLMOps 平台 |
|
||||||
| [Ollama](https://ollama.com/) | ✅ | 本地大模型管理平台 |
|
| [Ollama](https://ollama.com/) | ✅ | 本地大模型运行平台 |
|
||||||
|
| [LMStudio](https://lmstudio.ai/) | ✅ | 本地大模型运行平台 |
|
||||||
| [GiteeAI](https://ai.gitee.com/) | ✅ | 大模型接口聚合平台 |
|
| [GiteeAI](https://ai.gitee.com/) | ✅ | 大模型接口聚合平台 |
|
||||||
|
| [SiliconFlow](https://siliconflow.cn/) | ✅ | 大模型聚合平台 |
|
||||||
|
| [阿里云百炼](https://bailian.console.aliyun.com/) | ✅ | 大模型聚合平台 |
|
||||||
|
|
||||||
|
## 😘 社区贡献
|
||||||
|
|
||||||
|
LangBot 离不开以下贡献者和社区内所有人的贡献,我们欢迎任何形式的贡献和反馈。
|
||||||
|
|
||||||
|
|
||||||
|
<a href="https://github.com/RockChinQ/LangBot/graphs/contributors">
|
||||||
|
<img src="https://contrib.rocks/image?repo=RockChinQ/LangBot" />
|
||||||
|
</a>
|
||||||
|
|||||||
@@ -110,8 +110,17 @@ class WecomClient():
|
|||||||
"enable_duplicate_check": 0,
|
"enable_duplicate_check": 0,
|
||||||
"duplicate_check_interval": 1800
|
"duplicate_check_interval": 1800
|
||||||
}
|
}
|
||||||
response = await client.post(url,json=params)
|
try:
|
||||||
data = response.json()
|
response = await client.post(url,json=params)
|
||||||
|
data = response.json()
|
||||||
|
except Exception as e:
|
||||||
|
raise Exception("Failed to send image: "+str(e))
|
||||||
|
|
||||||
|
# 企业微信错误码40014和42001,代表accesstoken问题
|
||||||
|
if data['errcode'] == 40014 or data['errcode'] == 42001:
|
||||||
|
self.access_token = await self.get_access_token(self.secret)
|
||||||
|
return await self.send_image(user_id,agent_id,media_id)
|
||||||
|
|
||||||
if data['errcode'] != 0:
|
if data['errcode'] != 0:
|
||||||
raise Exception("Failed to send image: "+str(data))
|
raise Exception("Failed to send image: "+str(data))
|
||||||
|
|
||||||
@@ -136,7 +145,9 @@ class WecomClient():
|
|||||||
}
|
}
|
||||||
response = await client.post(url,json=params)
|
response = await client.post(url,json=params)
|
||||||
data = response.json()
|
data = response.json()
|
||||||
|
if data['errcode'] == 40014 or data['errcode'] == 42001:
|
||||||
|
self.access_token = await self.get_access_token(self.secret)
|
||||||
|
return await self.send_private_msg(user_id,agent_id,content)
|
||||||
if data['errcode'] != 0:
|
if data['errcode'] != 0:
|
||||||
raise Exception("Failed to send message: "+str(data))
|
raise Exception("Failed to send message: "+str(data))
|
||||||
|
|
||||||
@@ -286,11 +297,14 @@ class WecomClient():
|
|||||||
async with httpx.AsyncClient() as client:
|
async with httpx.AsyncClient() as client:
|
||||||
response = await client.post(url, headers=headers, content=body)
|
response = await client.post(url, headers=headers, content=body)
|
||||||
data = response.json()
|
data = response.json()
|
||||||
|
if data['errcode'] == 40014 or data['errcode'] == 42001:
|
||||||
|
self.access_token = await self.get_access_token(self.secret)
|
||||||
|
media_id = await self.upload_to_work(image)
|
||||||
if data.get('errcode', 0) != 0:
|
if data.get('errcode', 0) != 0:
|
||||||
raise Exception("failed to upload file")
|
raise Exception("failed to upload file")
|
||||||
|
|
||||||
return data.get('media_id')
|
media_id = data.get('media_id')
|
||||||
|
return media_id
|
||||||
|
|
||||||
async def download_image_to_bytes(self,url:str) -> bytes:
|
async def download_image_to_bytes(self,url:str) -> bytes:
|
||||||
async with httpx.AsyncClient() as client:
|
async with httpx.AsyncClient() as client:
|
||||||
|
|||||||
@@ -1,5 +1,7 @@
|
|||||||
import pip
|
import pip
|
||||||
|
|
||||||
|
# 检查依赖,防止用户未安装
|
||||||
|
# 左边为引入名称,右边为依赖名称
|
||||||
required_deps = {
|
required_deps = {
|
||||||
"requests": "requests",
|
"requests": "requests",
|
||||||
"openai": "openai",
|
"openai": "openai",
|
||||||
@@ -23,6 +25,9 @@ required_deps = {
|
|||||||
"aioshutil": "aioshutil",
|
"aioshutil": "aioshutil",
|
||||||
"argon2": "argon2-cffi",
|
"argon2": "argon2-cffi",
|
||||||
"jwt": "pyjwt",
|
"jwt": "pyjwt",
|
||||||
|
"Crypto": "pycryptodome",
|
||||||
|
"lark_oapi": "lark-oapi",
|
||||||
|
"discord": "discord.py"
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
29
pkg/core/migrations/m021_lark_config.py
Normal file
29
pkg/core/migrations/m021_lark_config.py
Normal file
@@ -0,0 +1,29 @@
|
|||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
from .. import migration
|
||||||
|
|
||||||
|
|
||||||
|
@migration.migration_class("lark-config", 21)
|
||||||
|
class LarkConfigMigration(migration.Migration):
|
||||||
|
"""迁移"""
|
||||||
|
|
||||||
|
async def need_migrate(self) -> bool:
|
||||||
|
"""判断当前环境是否需要运行此迁移"""
|
||||||
|
|
||||||
|
for adapter in self.ap.platform_cfg.data['platform-adapters']:
|
||||||
|
if adapter['adapter'] == 'lark':
|
||||||
|
return False
|
||||||
|
|
||||||
|
return True
|
||||||
|
|
||||||
|
async def run(self):
|
||||||
|
"""执行迁移"""
|
||||||
|
self.ap.platform_cfg.data['platform-adapters'].append({
|
||||||
|
"adapter": "lark",
|
||||||
|
"enable": False,
|
||||||
|
"app_id": "cli_abcdefgh",
|
||||||
|
"app_secret": "XXXXXXXXXX",
|
||||||
|
"bot_name": "LangBot"
|
||||||
|
})
|
||||||
|
|
||||||
|
await self.ap.platform_cfg.dump_config()
|
||||||
23
pkg/core/migrations/m022_lmstudio_config.py
Normal file
23
pkg/core/migrations/m022_lmstudio_config.py
Normal file
@@ -0,0 +1,23 @@
|
|||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
from .. import migration
|
||||||
|
|
||||||
|
|
||||||
|
@migration.migration_class("lmstudio-config", 22)
|
||||||
|
class LmStudioConfigMigration(migration.Migration):
|
||||||
|
"""迁移"""
|
||||||
|
|
||||||
|
async def need_migrate(self) -> bool:
|
||||||
|
"""判断当前环境是否需要运行此迁移"""
|
||||||
|
|
||||||
|
return 'lmstudio-chat-completions' not in self.ap.provider_cfg.data['requester']
|
||||||
|
|
||||||
|
async def run(self):
|
||||||
|
"""执行迁移"""
|
||||||
|
self.ap.provider_cfg.data['requester']['lmstudio-chat-completions'] = {
|
||||||
|
"base-url": "http://127.0.0.1:1234/v1",
|
||||||
|
"args": {},
|
||||||
|
"timeout": 120
|
||||||
|
}
|
||||||
|
|
||||||
|
await self.ap.provider_cfg.dump_config()
|
||||||
27
pkg/core/migrations/m023_siliconflow_config.py
Normal file
27
pkg/core/migrations/m023_siliconflow_config.py
Normal file
@@ -0,0 +1,27 @@
|
|||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
from .. import migration
|
||||||
|
|
||||||
|
|
||||||
|
@migration.migration_class("siliconflow-config", 23)
|
||||||
|
class SiliconFlowConfigMigration(migration.Migration):
|
||||||
|
"""迁移"""
|
||||||
|
|
||||||
|
async def need_migrate(self) -> bool:
|
||||||
|
"""判断当前环境是否需要运行此迁移"""
|
||||||
|
|
||||||
|
return 'siliconflow-chat-completions' not in self.ap.provider_cfg.data['requester']
|
||||||
|
|
||||||
|
async def run(self):
|
||||||
|
"""执行迁移"""
|
||||||
|
self.ap.provider_cfg.data['keys']['siliconflow'] = [
|
||||||
|
"xxxxxxx"
|
||||||
|
]
|
||||||
|
|
||||||
|
self.ap.provider_cfg.data['requester']['siliconflow-chat-completions'] = {
|
||||||
|
"base-url": "https://api.siliconflow.cn/v1",
|
||||||
|
"args": {},
|
||||||
|
"timeout": 120
|
||||||
|
}
|
||||||
|
|
||||||
|
await self.ap.provider_cfg.dump_config()
|
||||||
28
pkg/core/migrations/m024_discord_config.py
Normal file
28
pkg/core/migrations/m024_discord_config.py
Normal file
@@ -0,0 +1,28 @@
|
|||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
from .. import migration
|
||||||
|
|
||||||
|
|
||||||
|
@migration.migration_class("discord-config", 24)
|
||||||
|
class DiscordConfigMigration(migration.Migration):
|
||||||
|
"""迁移"""
|
||||||
|
|
||||||
|
async def need_migrate(self) -> bool:
|
||||||
|
"""判断当前环境是否需要运行此迁移"""
|
||||||
|
|
||||||
|
for adapter in self.ap.platform_cfg.data['platform-adapters']:
|
||||||
|
if adapter['adapter'] == 'discord':
|
||||||
|
return False
|
||||||
|
|
||||||
|
return True
|
||||||
|
|
||||||
|
async def run(self):
|
||||||
|
"""执行迁移"""
|
||||||
|
self.ap.platform_cfg.data['platform-adapters'].append({
|
||||||
|
"adapter": "discord",
|
||||||
|
"enable": False,
|
||||||
|
"client_id": "1234567890",
|
||||||
|
"token": "XXXXXXXXXX"
|
||||||
|
})
|
||||||
|
|
||||||
|
await self.ap.platform_cfg.dump_config()
|
||||||
@@ -8,7 +8,7 @@ from ..migrations import m001_sensitive_word_migration, m002_openai_config_migra
|
|||||||
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, m018_xai_config, m019_zhipuai_config
|
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
|
from ..migrations import m020_wecom_config, m021_lark_config, m022_lmstudio_config, m023_siliconflow_config, m024_discord_config
|
||||||
|
|
||||||
|
|
||||||
@stage.stage_class("MigrationStage")
|
@stage.stage_class("MigrationStage")
|
||||||
|
|||||||
@@ -37,7 +37,7 @@ class PlatformManager:
|
|||||||
|
|
||||||
async def initialize(self):
|
async def initialize(self):
|
||||||
|
|
||||||
from .sources import nakuru, aiocqhttp, qqbotpy,wecom
|
from .sources import nakuru, aiocqhttp, qqbotpy, wecom, lark, discord
|
||||||
|
|
||||||
async def on_friend_message(event: platform_events.FriendMessage, adapter: msadapter.MessageSourceAdapter):
|
async def on_friend_message(event: platform_events.FriendMessage, adapter: msadapter.MessageSourceAdapter):
|
||||||
|
|
||||||
|
|||||||
264
pkg/platform/sources/discord.py
Normal file
264
pkg/platform/sources/discord.py
Normal file
@@ -0,0 +1,264 @@
|
|||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
import discord
|
||||||
|
|
||||||
|
import typing
|
||||||
|
import asyncio
|
||||||
|
import traceback
|
||||||
|
import time
|
||||||
|
import re
|
||||||
|
import base64
|
||||||
|
import uuid
|
||||||
|
import json
|
||||||
|
import os
|
||||||
|
import datetime
|
||||||
|
|
||||||
|
import aiohttp
|
||||||
|
|
||||||
|
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 ...utils import image
|
||||||
|
|
||||||
|
|
||||||
|
class DiscordMessageConverter(adapter.MessageConverter):
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def yiri2target(
|
||||||
|
message_chain: platform_message.MessageChain
|
||||||
|
) -> typing.Tuple[str, typing.List[discord.File]]:
|
||||||
|
for ele in message_chain:
|
||||||
|
if isinstance(ele, platform_message.At):
|
||||||
|
message_chain.remove(ele)
|
||||||
|
break
|
||||||
|
|
||||||
|
text_string = ""
|
||||||
|
image_files = []
|
||||||
|
|
||||||
|
for ele in message_chain:
|
||||||
|
if isinstance(ele, platform_message.Image):
|
||||||
|
image_bytes = None
|
||||||
|
|
||||||
|
if ele.base64:
|
||||||
|
image_bytes = base64.b64decode(ele.base64)
|
||||||
|
elif ele.url:
|
||||||
|
async with aiohttp.ClientSession() as session:
|
||||||
|
async with session.get(ele.url) as response:
|
||||||
|
image_bytes = await response.read()
|
||||||
|
elif ele.path:
|
||||||
|
with open(ele.path, "rb") as f:
|
||||||
|
image_bytes = f.read()
|
||||||
|
|
||||||
|
image_files.append(discord.File(fp=image_bytes, filename=f"{uuid.uuid4()}.png"))
|
||||||
|
elif isinstance(ele, platform_message.Plain):
|
||||||
|
text_string += ele.text
|
||||||
|
|
||||||
|
return text_string, image_files
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def target2yiri(
|
||||||
|
message: discord.Message
|
||||||
|
) -> platform_message.MessageChain:
|
||||||
|
lb_msg_list = []
|
||||||
|
|
||||||
|
msg_create_time = datetime.datetime.fromtimestamp(
|
||||||
|
int(message.created_at.timestamp())
|
||||||
|
)
|
||||||
|
|
||||||
|
lb_msg_list.append(
|
||||||
|
platform_message.Source(id=message.id, time=msg_create_time)
|
||||||
|
)
|
||||||
|
|
||||||
|
element_list = []
|
||||||
|
|
||||||
|
def text_element_recur(text_ele: str) -> list[platform_message.MessageComponent]:
|
||||||
|
if text_ele == "":
|
||||||
|
return []
|
||||||
|
|
||||||
|
# <@1234567890>
|
||||||
|
# @everyone
|
||||||
|
# @here
|
||||||
|
at_pattern = re.compile(r"(@everyone|@here|<@[\d]+>)")
|
||||||
|
at_matches = at_pattern.findall(text_ele)
|
||||||
|
|
||||||
|
if len(at_matches) > 0:
|
||||||
|
mid_at = at_matches[0]
|
||||||
|
|
||||||
|
text_split = text_ele.split(mid_at)
|
||||||
|
|
||||||
|
mid_at_component = []
|
||||||
|
|
||||||
|
if mid_at == "@everyone" or mid_at == "@here":
|
||||||
|
mid_at_component.append(platform_message.AtAll())
|
||||||
|
else:
|
||||||
|
mid_at_component.append(platform_message.At(target=mid_at[2:-1]))
|
||||||
|
|
||||||
|
return text_element_recur(text_split[0]) + \
|
||||||
|
mid_at_component + \
|
||||||
|
text_element_recur(text_split[1])
|
||||||
|
else:
|
||||||
|
return [platform_message.Plain(text=text_ele)]
|
||||||
|
|
||||||
|
|
||||||
|
element_list.extend(text_element_recur(message.content))
|
||||||
|
|
||||||
|
# attachments
|
||||||
|
for attachment in message.attachments:
|
||||||
|
async with aiohttp.ClientSession(trust_env=True) as session:
|
||||||
|
async with session.get(attachment.url) as response:
|
||||||
|
image_data = await response.read()
|
||||||
|
image_base64 = base64.b64encode(image_data).decode("utf-8")
|
||||||
|
image_format = response.headers["Content-Type"]
|
||||||
|
element_list.append(platform_message.Image(base64=f"data:{image_format};base64,{image_base64}"))
|
||||||
|
|
||||||
|
return platform_message.MessageChain(element_list)
|
||||||
|
|
||||||
|
|
||||||
|
class DiscordEventConverter(adapter.EventConverter):
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def yiri2target(
|
||||||
|
event: platform_events.Event
|
||||||
|
) -> discord.Message:
|
||||||
|
pass
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def target2yiri(
|
||||||
|
event: discord.Message
|
||||||
|
) -> platform_events.Event:
|
||||||
|
message_chain = await DiscordMessageConverter.target2yiri(event)
|
||||||
|
|
||||||
|
if type(event.channel) == discord.DMChannel:
|
||||||
|
return platform_events.FriendMessage(
|
||||||
|
sender=platform_entities.Friend(
|
||||||
|
id=event.author.id,
|
||||||
|
nickname=event.author.name,
|
||||||
|
remark=event.channel.id,
|
||||||
|
),
|
||||||
|
message_chain=message_chain,
|
||||||
|
time=event.created_at.timestamp(),
|
||||||
|
source_platform_object=event,
|
||||||
|
)
|
||||||
|
elif type(event.channel) == discord.TextChannel:
|
||||||
|
return platform_events.GroupMessage(
|
||||||
|
sender=platform_entities.GroupMember(
|
||||||
|
id=event.author.id,
|
||||||
|
member_name=event.author.name,
|
||||||
|
permission=platform_entities.Permission.Member,
|
||||||
|
group=platform_entities.Group(
|
||||||
|
id=event.channel.id,
|
||||||
|
name=event.channel.name,
|
||||||
|
permission=platform_entities.Permission.Member,
|
||||||
|
),
|
||||||
|
special_title="",
|
||||||
|
join_timestamp=0,
|
||||||
|
last_speak_timestamp=0,
|
||||||
|
mute_time_remaining=0,
|
||||||
|
),
|
||||||
|
message_chain=message_chain,
|
||||||
|
time=event.created_at.timestamp(),
|
||||||
|
source_platform_object=event,
|
||||||
|
)
|
||||||
|
|
||||||
|
@adapter.adapter_class("discord")
|
||||||
|
class DiscordMessageSourceAdapter(adapter.MessageSourceAdapter):
|
||||||
|
|
||||||
|
bot: discord.Client
|
||||||
|
|
||||||
|
bot_account_id: str # 用于在流水线中识别at是否是本bot,直接以bot_name作为标识
|
||||||
|
|
||||||
|
config: dict
|
||||||
|
|
||||||
|
ap: app.Application
|
||||||
|
|
||||||
|
message_converter: DiscordMessageConverter = DiscordMessageConverter()
|
||||||
|
event_converter: DiscordEventConverter = DiscordEventConverter()
|
||||||
|
|
||||||
|
listeners: typing.Dict[
|
||||||
|
typing.Type[platform_events.Event],
|
||||||
|
typing.Callable[[platform_events.Event, adapter.MessageSourceAdapter], None],
|
||||||
|
] = {}
|
||||||
|
|
||||||
|
def __init__(self, config: dict, ap: app.Application):
|
||||||
|
self.config = config
|
||||||
|
self.ap = ap
|
||||||
|
|
||||||
|
self.bot_account_id = self.config["client_id"]
|
||||||
|
|
||||||
|
adapter_self = self
|
||||||
|
|
||||||
|
class MyClient(discord.Client):
|
||||||
|
|
||||||
|
async def on_message(self: discord.Client, message: discord.Message):
|
||||||
|
if message.author.id == self.user.id or message.author.bot:
|
||||||
|
return
|
||||||
|
|
||||||
|
lb_event = await adapter_self.event_converter.target2yiri(message)
|
||||||
|
await adapter_self.listeners[type(lb_event)](lb_event, adapter_self)
|
||||||
|
|
||||||
|
intents = discord.Intents.default()
|
||||||
|
intents.message_content = True
|
||||||
|
|
||||||
|
args = {}
|
||||||
|
|
||||||
|
if os.getenv("http_proxy"):
|
||||||
|
args["proxy"] = os.getenv("http_proxy")
|
||||||
|
|
||||||
|
self.bot = MyClient(intents=intents, **args)
|
||||||
|
|
||||||
|
async def send_message(
|
||||||
|
self, target_type: str, target_id: str, message: platform_message.MessageChain
|
||||||
|
):
|
||||||
|
pass
|
||||||
|
|
||||||
|
async def reply_message(
|
||||||
|
self,
|
||||||
|
message_source: platform_events.MessageEvent,
|
||||||
|
message: platform_message.MessageChain,
|
||||||
|
quote_origin: bool = False,
|
||||||
|
):
|
||||||
|
msg_to_send, image_files = await self.message_converter.yiri2target(message)
|
||||||
|
assert isinstance(message_source.source_platform_object, discord.Message)
|
||||||
|
|
||||||
|
args = {
|
||||||
|
"content": msg_to_send,
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(image_files) > 0:
|
||||||
|
args["files"] = image_files
|
||||||
|
|
||||||
|
if quote_origin:
|
||||||
|
args["reference"] = message_source.source_platform_object
|
||||||
|
|
||||||
|
if message.has(platform_message.At):
|
||||||
|
args["mention_author"] = True
|
||||||
|
|
||||||
|
await message_source.source_platform_object.channel.send(**args)
|
||||||
|
|
||||||
|
async def is_muted(self, group_id: int) -> bool:
|
||||||
|
return False
|
||||||
|
|
||||||
|
def register_listener(
|
||||||
|
self,
|
||||||
|
event_type: typing.Type[platform_events.Event],
|
||||||
|
callback: typing.Callable[[platform_events.Event, adapter.MessageSourceAdapter], None],
|
||||||
|
):
|
||||||
|
self.listeners[event_type] = callback
|
||||||
|
|
||||||
|
def unregister_listener(
|
||||||
|
self,
|
||||||
|
event_type: typing.Type[platform_events.Event],
|
||||||
|
callback: typing.Callable[[platform_events.Event, adapter.MessageSourceAdapter], None],
|
||||||
|
):
|
||||||
|
self.listeners.pop(event_type)
|
||||||
|
|
||||||
|
async def run_async(self):
|
||||||
|
async with self.bot:
|
||||||
|
await self.bot.start(self.config["token"], reconnect=True)
|
||||||
|
|
||||||
|
async def kill(self) -> bool:
|
||||||
|
await self.bot.close()
|
||||||
|
return True
|
||||||
404
pkg/platform/sources/lark.py
Normal file
404
pkg/platform/sources/lark.py
Normal file
@@ -0,0 +1,404 @@
|
|||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
import lark_oapi
|
||||||
|
|
||||||
|
import typing
|
||||||
|
import asyncio
|
||||||
|
import traceback
|
||||||
|
import time
|
||||||
|
import re
|
||||||
|
import base64
|
||||||
|
import uuid
|
||||||
|
import json
|
||||||
|
import datetime
|
||||||
|
|
||||||
|
import aiohttp
|
||||||
|
import lark_oapi.ws.exception
|
||||||
|
from lark_oapi.api.im.v1 import *
|
||||||
|
|
||||||
|
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 ...utils import image
|
||||||
|
|
||||||
|
|
||||||
|
class LarkMessageConverter(adapter.MessageConverter):
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def yiri2target(
|
||||||
|
message_chain: platform_message.MessageChain, api_client: lark_oapi.Client
|
||||||
|
) -> typing.Tuple[list]:
|
||||||
|
message_elements = []
|
||||||
|
|
||||||
|
pending_paragraph = []
|
||||||
|
|
||||||
|
for msg in message_chain:
|
||||||
|
if isinstance(msg, platform_message.Plain):
|
||||||
|
pending_paragraph.append({"tag": "md", "text": msg.text})
|
||||||
|
elif isinstance(msg, platform_message.At):
|
||||||
|
pending_paragraph.append(
|
||||||
|
{"tag": "at", "user_id": msg.target, "style": []}
|
||||||
|
)
|
||||||
|
elif isinstance(msg, platform_message.AtAll):
|
||||||
|
pending_paragraph.append({"tag": "at", "user_id": "all", "style": []})
|
||||||
|
elif isinstance(msg, platform_message.Image):
|
||||||
|
|
||||||
|
image_bytes = None
|
||||||
|
|
||||||
|
if msg.base64:
|
||||||
|
image_bytes = base64.b64decode(msg.base64)
|
||||||
|
elif msg.url:
|
||||||
|
async with aiohttp.ClientSession() as session:
|
||||||
|
async with session.get(msg.url) as response:
|
||||||
|
image_bytes = await response.read()
|
||||||
|
elif msg.path:
|
||||||
|
with open(msg.path, "rb") as f:
|
||||||
|
image_bytes = f.read()
|
||||||
|
|
||||||
|
request: CreateImageRequest = (
|
||||||
|
CreateImageRequest.builder()
|
||||||
|
.request_body(
|
||||||
|
CreateImageRequestBody.builder()
|
||||||
|
.image_type("message")
|
||||||
|
.image(image_bytes)
|
||||||
|
.build()
|
||||||
|
)
|
||||||
|
.build()
|
||||||
|
)
|
||||||
|
|
||||||
|
response: CreateImageResponse = await api_client.im.v1.image.acreate(
|
||||||
|
request
|
||||||
|
)
|
||||||
|
|
||||||
|
if not response.success():
|
||||||
|
raise Exception(
|
||||||
|
f"client.im.v1.image.create failed, code: {response.code}, msg: {response.msg}, log_id: {response.get_log_id()}, resp: \n{json.dumps(json.loads(response.raw.content), indent=4, ensure_ascii=False)}"
|
||||||
|
)
|
||||||
|
|
||||||
|
image_key = response.data.image_key
|
||||||
|
|
||||||
|
message_elements.append(pending_paragraph)
|
||||||
|
message_elements.append(
|
||||||
|
[
|
||||||
|
{
|
||||||
|
"tag": "img",
|
||||||
|
"image_key": image_key,
|
||||||
|
}
|
||||||
|
]
|
||||||
|
)
|
||||||
|
pending_paragraph = []
|
||||||
|
|
||||||
|
if pending_paragraph:
|
||||||
|
message_elements.append(pending_paragraph)
|
||||||
|
|
||||||
|
return message_elements
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def target2yiri(
|
||||||
|
message: lark_oapi.api.im.v1.model.event_message.EventMessage,
|
||||||
|
api_client: lark_oapi.Client,
|
||||||
|
) -> platform_message.MessageChain:
|
||||||
|
message_content = json.loads(message.content)
|
||||||
|
|
||||||
|
lb_msg_list = []
|
||||||
|
|
||||||
|
msg_create_time = datetime.datetime.fromtimestamp(
|
||||||
|
int(message.create_time) / 1000
|
||||||
|
)
|
||||||
|
|
||||||
|
lb_msg_list.append(
|
||||||
|
platform_message.Source(id=message.message_id, time=msg_create_time)
|
||||||
|
)
|
||||||
|
|
||||||
|
if message.message_type == "text":
|
||||||
|
element_list = []
|
||||||
|
|
||||||
|
def text_element_recur(text_ele: dict) -> list[dict]:
|
||||||
|
if text_ele["text"] == "":
|
||||||
|
return []
|
||||||
|
|
||||||
|
at_pattern = re.compile(r"@_user_[\d]+")
|
||||||
|
at_matches = at_pattern.findall(text_ele["text"])
|
||||||
|
|
||||||
|
name_mapping = {}
|
||||||
|
for mathc in at_matches:
|
||||||
|
for mention in message.mentions:
|
||||||
|
if mention.key == mathc:
|
||||||
|
name_mapping[mathc] = mention.name
|
||||||
|
break
|
||||||
|
|
||||||
|
if len(name_mapping.keys()) == 0:
|
||||||
|
return [text_ele]
|
||||||
|
|
||||||
|
# 只处理第一个,剩下的递归处理
|
||||||
|
text_split = text_ele["text"].split(list(name_mapping.keys())[0])
|
||||||
|
|
||||||
|
new_list = []
|
||||||
|
|
||||||
|
left_text = text_split[0]
|
||||||
|
right_text = text_split[1]
|
||||||
|
|
||||||
|
new_list.extend(
|
||||||
|
text_element_recur({"tag": "text", "text": left_text, "style": []})
|
||||||
|
)
|
||||||
|
|
||||||
|
new_list.append(
|
||||||
|
{
|
||||||
|
"tag": "at",
|
||||||
|
"user_id": list(name_mapping.keys())[0],
|
||||||
|
"user_name": name_mapping[list(name_mapping.keys())[0]],
|
||||||
|
"style": [],
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
new_list.extend(
|
||||||
|
text_element_recur({"tag": "text", "text": right_text, "style": []})
|
||||||
|
)
|
||||||
|
|
||||||
|
return new_list
|
||||||
|
|
||||||
|
element_list = text_element_recur(
|
||||||
|
{"tag": "text", "text": message_content["text"], "style": []}
|
||||||
|
)
|
||||||
|
|
||||||
|
message_content = {"title": "", "content": element_list}
|
||||||
|
|
||||||
|
elif message.message_type == "post":
|
||||||
|
new_list = []
|
||||||
|
|
||||||
|
for ele in message_content["content"]:
|
||||||
|
if type(ele) is dict:
|
||||||
|
new_list.append(ele)
|
||||||
|
elif type(ele) is list:
|
||||||
|
new_list.extend(ele)
|
||||||
|
|
||||||
|
message_content["content"] = new_list
|
||||||
|
elif message.message_type == "image":
|
||||||
|
message_content["content"] = [
|
||||||
|
{"tag": "img", "image_key": message_content["image_key"], "style": []}
|
||||||
|
]
|
||||||
|
|
||||||
|
for ele in message_content["content"]:
|
||||||
|
if ele["tag"] == "text":
|
||||||
|
lb_msg_list.append(platform_message.Plain(text=ele["text"]))
|
||||||
|
elif ele["tag"] == "at":
|
||||||
|
lb_msg_list.append(platform_message.At(target=ele["user_name"]))
|
||||||
|
elif ele["tag"] == "img":
|
||||||
|
image_key = ele["image_key"]
|
||||||
|
|
||||||
|
request: GetMessageResourceRequest = (
|
||||||
|
GetMessageResourceRequest.builder()
|
||||||
|
.message_id(message.message_id)
|
||||||
|
.file_key(image_key)
|
||||||
|
.type("image")
|
||||||
|
.build()
|
||||||
|
)
|
||||||
|
|
||||||
|
response: GetMessageResourceResponse = (
|
||||||
|
await api_client.im.v1.message_resource.aget(request)
|
||||||
|
)
|
||||||
|
|
||||||
|
if not response.success():
|
||||||
|
raise Exception(
|
||||||
|
f"client.im.v1.message_resource.get failed, code: {response.code}, msg: {response.msg}, log_id: {response.get_log_id()}, resp: \n{json.dumps(json.loads(response.raw.content), indent=4, ensure_ascii=False)}"
|
||||||
|
)
|
||||||
|
|
||||||
|
image_bytes = response.file.read()
|
||||||
|
image_base64 = base64.b64encode(image_bytes).decode()
|
||||||
|
|
||||||
|
image_format = response.raw.headers["content-type"]
|
||||||
|
|
||||||
|
lb_msg_list.append(
|
||||||
|
platform_message.Image(
|
||||||
|
base64=f"data:{image_format};base64,{image_base64}"
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
return platform_message.MessageChain(lb_msg_list)
|
||||||
|
|
||||||
|
|
||||||
|
class LarkEventConverter(adapter.EventConverter):
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def yiri2target(
|
||||||
|
event: platform_events.MessageEvent,
|
||||||
|
) -> lark_oapi.im.v1.P2ImMessageReceiveV1:
|
||||||
|
pass
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def target2yiri(
|
||||||
|
event: lark_oapi.im.v1.P2ImMessageReceiveV1, api_client: lark_oapi.Client
|
||||||
|
) -> platform_events.Event:
|
||||||
|
message_chain = await LarkMessageConverter.target2yiri(
|
||||||
|
event.event.message, api_client
|
||||||
|
)
|
||||||
|
|
||||||
|
if event.event.message.chat_type == "p2p":
|
||||||
|
return platform_events.FriendMessage(
|
||||||
|
sender=platform_entities.Friend(
|
||||||
|
id=event.event.sender.sender_id.open_id,
|
||||||
|
nickname=event.event.sender.sender_id.union_id,
|
||||||
|
remark="",
|
||||||
|
),
|
||||||
|
message_chain=message_chain,
|
||||||
|
time=event.event.message.create_time,
|
||||||
|
)
|
||||||
|
elif event.event.message.chat_type == "group":
|
||||||
|
return platform_events.GroupMessage(
|
||||||
|
sender=platform_entities.GroupMember(
|
||||||
|
id=event.event.sender.sender_id.open_id,
|
||||||
|
member_name=event.event.sender.sender_id.union_id,
|
||||||
|
permission=platform_entities.Permission.Member,
|
||||||
|
group=platform_entities.Group(
|
||||||
|
id=event.event.message.chat_id,
|
||||||
|
name="",
|
||||||
|
permission=platform_entities.Permission.Member,
|
||||||
|
),
|
||||||
|
special_title="",
|
||||||
|
join_timestamp=0,
|
||||||
|
last_speak_timestamp=0,
|
||||||
|
mute_time_remaining=0,
|
||||||
|
),
|
||||||
|
message_chain=message_chain,
|
||||||
|
time=event.event.message.create_time,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@adapter.adapter_class("lark")
|
||||||
|
class LarkMessageSourceAdapter(adapter.MessageSourceAdapter):
|
||||||
|
|
||||||
|
bot: lark_oapi.ws.Client
|
||||||
|
api_client: lark_oapi.Client
|
||||||
|
|
||||||
|
bot_account_id: str # 用于在流水线中识别at是否是本bot,直接以bot_name作为标识
|
||||||
|
lark_tenant_key: str # 飞书企业key
|
||||||
|
|
||||||
|
message_converter: LarkMessageConverter = LarkMessageConverter()
|
||||||
|
event_converter: LarkEventConverter = LarkEventConverter()
|
||||||
|
|
||||||
|
listeners: typing.Dict[
|
||||||
|
typing.Type[platform_events.Event],
|
||||||
|
typing.Callable[[platform_events.Event, adapter.MessageSourceAdapter], None],
|
||||||
|
] = {}
|
||||||
|
|
||||||
|
config: dict
|
||||||
|
|
||||||
|
ap: app.Application
|
||||||
|
|
||||||
|
def __init__(self, config: dict, ap: app.Application):
|
||||||
|
self.config = config
|
||||||
|
self.ap = ap
|
||||||
|
|
||||||
|
async def on_message(event: lark_oapi.im.v1.P2ImMessageReceiveV1):
|
||||||
|
|
||||||
|
lb_event = await self.event_converter.target2yiri(event, self.api_client)
|
||||||
|
|
||||||
|
await self.listeners[type(lb_event)](lb_event, self)
|
||||||
|
|
||||||
|
def sync_on_message(event: lark_oapi.im.v1.P2ImMessageReceiveV1):
|
||||||
|
asyncio.create_task(on_message(event))
|
||||||
|
|
||||||
|
event_handler = (
|
||||||
|
lark_oapi.EventDispatcherHandler.builder("", "")
|
||||||
|
.register_p2_im_message_receive_v1(sync_on_message)
|
||||||
|
.build()
|
||||||
|
)
|
||||||
|
|
||||||
|
self.bot_account_id = config["bot_name"]
|
||||||
|
|
||||||
|
self.bot = lark_oapi.ws.Client(
|
||||||
|
config["app_id"], config["app_secret"], event_handler=event_handler
|
||||||
|
)
|
||||||
|
self.api_client = (
|
||||||
|
lark_oapi.Client.builder()
|
||||||
|
.app_id(config["app_id"])
|
||||||
|
.app_secret(config["app_secret"])
|
||||||
|
.build()
|
||||||
|
)
|
||||||
|
|
||||||
|
async def send_message(
|
||||||
|
self, target_type: str, target_id: str, message: platform_message.MessageChain
|
||||||
|
):
|
||||||
|
pass
|
||||||
|
|
||||||
|
async def reply_message(
|
||||||
|
self,
|
||||||
|
message_source: platform_events.MessageEvent,
|
||||||
|
message: platform_message.MessageChain,
|
||||||
|
quote_origin: bool = False,
|
||||||
|
):
|
||||||
|
|
||||||
|
# 不再需要了,因为message_id已经被包含到message_chain中
|
||||||
|
# lark_event = await self.event_converter.yiri2target(message_source)
|
||||||
|
lark_message = await self.message_converter.yiri2target(
|
||||||
|
message, self.api_client
|
||||||
|
)
|
||||||
|
|
||||||
|
final_content = {
|
||||||
|
"zh_cn": {
|
||||||
|
"title": "",
|
||||||
|
"content": lark_message,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
request: ReplyMessageRequest = (
|
||||||
|
ReplyMessageRequest.builder()
|
||||||
|
.message_id(message_source.message_chain.message_id)
|
||||||
|
.request_body(
|
||||||
|
ReplyMessageRequestBody.builder()
|
||||||
|
.content(json.dumps(final_content))
|
||||||
|
.msg_type("post")
|
||||||
|
.reply_in_thread(False)
|
||||||
|
.uuid(str(uuid.uuid4()))
|
||||||
|
.build()
|
||||||
|
)
|
||||||
|
.build()
|
||||||
|
)
|
||||||
|
|
||||||
|
response: ReplyMessageResponse = await self.api_client.im.v1.message.areply(
|
||||||
|
request
|
||||||
|
)
|
||||||
|
|
||||||
|
if not response.success():
|
||||||
|
raise Exception(
|
||||||
|
f"client.im.v1.message.reply failed, code: {response.code}, msg: {response.msg}, log_id: {response.get_log_id()}, resp: \n{json.dumps(json.loads(response.raw.content), indent=4, ensure_ascii=False)}"
|
||||||
|
)
|
||||||
|
|
||||||
|
async def is_muted(self, group_id: int) -> bool:
|
||||||
|
return False
|
||||||
|
|
||||||
|
def register_listener(
|
||||||
|
self,
|
||||||
|
event_type: typing.Type[platform_events.Event],
|
||||||
|
callback: typing.Callable[
|
||||||
|
[platform_events.Event, adapter.MessageSourceAdapter], None
|
||||||
|
],
|
||||||
|
):
|
||||||
|
self.listeners[event_type] = callback
|
||||||
|
|
||||||
|
def unregister_listener(
|
||||||
|
self,
|
||||||
|
event_type: typing.Type[platform_events.Event],
|
||||||
|
callback: typing.Callable[
|
||||||
|
[platform_events.Event, adapter.MessageSourceAdapter], None
|
||||||
|
],
|
||||||
|
):
|
||||||
|
self.listeners.pop(event_type)
|
||||||
|
|
||||||
|
async def run_async(self):
|
||||||
|
try:
|
||||||
|
await self.bot._connect()
|
||||||
|
except lark_oapi.ws.exception.ClientException as e:
|
||||||
|
raise e
|
||||||
|
except Exception as e:
|
||||||
|
await self.bot._disconnect()
|
||||||
|
if self.bot._auto_reconnect:
|
||||||
|
await self.bot._reconnect()
|
||||||
|
else:
|
||||||
|
raise e
|
||||||
|
|
||||||
|
async def kill(self) -> bool:
|
||||||
|
return False
|
||||||
@@ -72,6 +72,11 @@ class MessageEvent(Event):
|
|||||||
message_chain: platform_message.MessageChain
|
message_chain: platform_message.MessageChain
|
||||||
"""消息内容。"""
|
"""消息内容。"""
|
||||||
|
|
||||||
|
source_platform_object: typing.Optional[typing.Any] = None
|
||||||
|
"""原消息平台对象。
|
||||||
|
供消息平台适配器开发者使用,如果回复用户时需要使用原消息事件对象的信息,
|
||||||
|
那么可以将其存到这个字段以供之后取出使用。"""
|
||||||
|
|
||||||
|
|
||||||
class FriendMessage(MessageEvent):
|
class FriendMessage(MessageEvent):
|
||||||
"""好友消息。
|
"""好友消息。
|
||||||
|
|||||||
@@ -460,7 +460,7 @@ class Source(MessageComponent):
|
|||||||
"""源。包含消息的基本信息。"""
|
"""源。包含消息的基本信息。"""
|
||||||
type: str = "Source"
|
type: str = "Source"
|
||||||
"""消息组件类型。"""
|
"""消息组件类型。"""
|
||||||
id: int
|
id: typing.Union[int, str]
|
||||||
"""消息的识别号,用于引用回复(Source 类型永远为 MessageChain 的第一个元素)。"""
|
"""消息的识别号,用于引用回复(Source 类型永远为 MessageChain 的第一个元素)。"""
|
||||||
time: datetime
|
time: datetime
|
||||||
"""消息时间。"""
|
"""消息时间。"""
|
||||||
@@ -503,7 +503,7 @@ class At(MessageComponent):
|
|||||||
"""At某人。"""
|
"""At某人。"""
|
||||||
type: str = "At"
|
type: str = "At"
|
||||||
"""消息组件类型。"""
|
"""消息组件类型。"""
|
||||||
target: int
|
target: typing.Union[int, str]
|
||||||
"""群员 QQ 号。"""
|
"""群员 QQ 号。"""
|
||||||
display: typing.Optional[str] = None
|
display: typing.Optional[str] = None
|
||||||
"""At时显示的文字,发送消息时无效,自动使用群名片。"""
|
"""At时显示的文字,发送消息时无效,自动使用群名片。"""
|
||||||
|
|||||||
@@ -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, xaichatcmpl, zhipuaichatcmpl
|
from .requesters import chatcmpl, anthropicmsgs, moonshotchatcmpl, deepseekchatcmpl, ollamachat, giteeaichatcmpl, xaichatcmpl, zhipuaichatcmpl, lmstudiochatcmpl, siliconflowchatcmpl
|
||||||
|
|
||||||
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"
|
||||||
|
|
||||||
@@ -109,4 +109,4 @@ class ModelManager:
|
|||||||
self.model_list.append(model_info)
|
self.model_list.append(model_info)
|
||||||
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
self.ap.logger.error(f"初始化模型 {model['name']} 失败: {e} ,请检查配置文件")
|
self.ap.logger.error(f"初始化模型 {model['name']} 失败: {type(e)} {e} ,请检查配置文件")
|
||||||
|
|||||||
@@ -30,7 +30,7 @@ class AnthropicMessages(requester.LLMAPIRequester):
|
|||||||
timeout=typing.cast(httpx.Timeout, self.ap.provider_cfg.data['requester']['anthropic-messages']['timeout']),
|
timeout=typing.cast(httpx.Timeout, self.ap.provider_cfg.data['requester']['anthropic-messages']['timeout']),
|
||||||
limits=anthropic._constants.DEFAULT_CONNECTION_LIMITS,
|
limits=anthropic._constants.DEFAULT_CONNECTION_LIMITS,
|
||||||
follow_redirects=True,
|
follow_redirects=True,
|
||||||
proxies=self.ap.proxy_mgr.get_forward_proxies()
|
trust_env=True,
|
||||||
)
|
)
|
||||||
|
|
||||||
self.client = anthropic.AsyncAnthropic(
|
self.client = anthropic.AsyncAnthropic(
|
||||||
|
|||||||
@@ -39,7 +39,7 @@ class OpenAIChatCompletions(requester.LLMAPIRequester):
|
|||||||
base_url=self.requester_cfg['base-url'],
|
base_url=self.requester_cfg['base-url'],
|
||||||
timeout=self.requester_cfg['timeout'],
|
timeout=self.requester_cfg['timeout'],
|
||||||
http_client=httpx.AsyncClient(
|
http_client=httpx.AsyncClient(
|
||||||
proxies=self.ap.proxy_mgr.get_forward_proxies()
|
trust_env=True,
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
21
pkg/provider/modelmgr/requesters/lmstudiochatcmpl.py
Normal file
21
pkg/provider/modelmgr/requesters/lmstudiochatcmpl.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("lmstudio-chat-completions")
|
||||||
|
class LmStudioChatCompletions(chatcmpl.OpenAIChatCompletions):
|
||||||
|
"""LMStudio 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']['lmstudio-chat-completions']
|
||||||
21
pkg/provider/modelmgr/requesters/siliconflowchatcmpl.py
Normal file
21
pkg/provider/modelmgr/requesters/siliconflowchatcmpl.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("siliconflow-chat-completions")
|
||||||
|
class SiliconFlowChatCompletions(chatcmpl.OpenAIChatCompletions):
|
||||||
|
"""SiliconFlow 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']['siliconflow-chat-completions']
|
||||||
@@ -1,4 +1,4 @@
|
|||||||
semantic_version = "v3.4.3"
|
semantic_version = "v3.4.5"
|
||||||
|
|
||||||
debug_mode = False
|
debug_mode = False
|
||||||
|
|
||||||
|
|||||||
@@ -25,6 +25,8 @@ aioshutil
|
|||||||
argon2-cffi
|
argon2-cffi
|
||||||
pyjwt
|
pyjwt
|
||||||
pycryptodome
|
pycryptodome
|
||||||
|
lark-oapi
|
||||||
|
discord.py
|
||||||
|
|
||||||
# indirect
|
# indirect
|
||||||
taskgroup==0.0.0a4
|
taskgroup==0.0.0a4
|
||||||
@@ -116,6 +116,11 @@
|
|||||||
"requester": "deepseek-chat-completions",
|
"requester": "deepseek-chat-completions",
|
||||||
"token_mgr": "deepseek"
|
"token_mgr": "deepseek"
|
||||||
},
|
},
|
||||||
|
{
|
||||||
|
"name": "deepseek-reasoner",
|
||||||
|
"requester": "deepseek-chat-completions",
|
||||||
|
"token_mgr": "deepseek"
|
||||||
|
},
|
||||||
{
|
{
|
||||||
"name": "grok-2-latest",
|
"name": "grok-2-latest",
|
||||||
"requester": "xai-chat-completions",
|
"requester": "xai-chat-completions",
|
||||||
|
|||||||
@@ -35,6 +35,19 @@
|
|||||||
"token": "",
|
"token": "",
|
||||||
"EncodingAESKey": "",
|
"EncodingAESKey": "",
|
||||||
"contacts_secret": ""
|
"contacts_secret": ""
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"adapter": "lark",
|
||||||
|
"enable": false,
|
||||||
|
"app_id": "cli_abcdefgh",
|
||||||
|
"app_secret": "XXXXXXXXXX",
|
||||||
|
"bot_name": "LangBot"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"adapter": "discord",
|
||||||
|
"enable": true,
|
||||||
|
"client_id": "1234567890",
|
||||||
|
"token": "XXXXXXXXXX"
|
||||||
}
|
}
|
||||||
],
|
],
|
||||||
"track-function-calls": true,
|
"track-function-calls": true,
|
||||||
|
|||||||
@@ -22,6 +22,9 @@
|
|||||||
],
|
],
|
||||||
"zhipuai": [
|
"zhipuai": [
|
||||||
"xxxxxxx"
|
"xxxxxxx"
|
||||||
|
],
|
||||||
|
"siliconflow": [
|
||||||
|
"xxxxxxx"
|
||||||
]
|
]
|
||||||
},
|
},
|
||||||
"requester": {
|
"requester": {
|
||||||
@@ -66,6 +69,16 @@
|
|||||||
"base-url": "https://open.bigmodel.cn/api/paas/v4",
|
"base-url": "https://open.bigmodel.cn/api/paas/v4",
|
||||||
"args": {},
|
"args": {},
|
||||||
"timeout": 120
|
"timeout": 120
|
||||||
|
},
|
||||||
|
"lmstudio-chat-completions": {
|
||||||
|
"base-url": "http://127.0.0.1:1234/v1",
|
||||||
|
"args": {},
|
||||||
|
"timeout": 120
|
||||||
|
},
|
||||||
|
"siliconflow-chat-completions": {
|
||||||
|
"base-url": "https://api.siliconflow.cn/v1",
|
||||||
|
"args": {},
|
||||||
|
"timeout": 120
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
"model": "gpt-4o",
|
"model": "gpt-4o",
|
||||||
|
|||||||
@@ -121,6 +121,129 @@
|
|||||||
]
|
]
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"title": "企业微信适配器",
|
||||||
|
"description": "用于接入企业微信",
|
||||||
|
"properties": {
|
||||||
|
"adapter": {
|
||||||
|
"type": "string",
|
||||||
|
"const": "wecom"
|
||||||
|
},
|
||||||
|
"enable": {
|
||||||
|
"type": "boolean",
|
||||||
|
"default": false,
|
||||||
|
"description": "是否启用此适配器",
|
||||||
|
"layout": {
|
||||||
|
"comp": "switch",
|
||||||
|
"props": {
|
||||||
|
"color": "primary"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"host": {
|
||||||
|
"type": "string",
|
||||||
|
"default": "0.0.0.0",
|
||||||
|
"description": "监听的IP地址"
|
||||||
|
},
|
||||||
|
"port": {
|
||||||
|
"type": "integer",
|
||||||
|
"default": 2290,
|
||||||
|
"description": "监听的端口"
|
||||||
|
},
|
||||||
|
"corpid": {
|
||||||
|
"type": "string",
|
||||||
|
"default": "",
|
||||||
|
"description": "企业微信的corpid"
|
||||||
|
},
|
||||||
|
"secret": {
|
||||||
|
"type": "string",
|
||||||
|
"default": "",
|
||||||
|
"description": "企业微信的secret"
|
||||||
|
},
|
||||||
|
"token": {
|
||||||
|
"type": "string",
|
||||||
|
"default": "",
|
||||||
|
"description": "企业微信的token"
|
||||||
|
},
|
||||||
|
"EncodingAESKey": {
|
||||||
|
"type": "string",
|
||||||
|
"default": "",
|
||||||
|
"description": "企业微信的EncodingAESKey"
|
||||||
|
},
|
||||||
|
"contacts_secret": {
|
||||||
|
"type": "string",
|
||||||
|
"default": "",
|
||||||
|
"description": "企业微信的contacts_secret"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"title": "飞书适配器",
|
||||||
|
"description": "用于接入飞书",
|
||||||
|
"properties": {
|
||||||
|
"adapter": {
|
||||||
|
"type": "string",
|
||||||
|
"const": "lark"
|
||||||
|
},
|
||||||
|
"enable": {
|
||||||
|
"type": "boolean",
|
||||||
|
"default": false,
|
||||||
|
"description": "是否启用此适配器",
|
||||||
|
"layout": {
|
||||||
|
"comp": "switch",
|
||||||
|
"props": {
|
||||||
|
"color": "primary"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"app_id": {
|
||||||
|
"type": "string",
|
||||||
|
"default": "",
|
||||||
|
"description": "飞书的app_id"
|
||||||
|
},
|
||||||
|
"app_secret": {
|
||||||
|
"type": "string",
|
||||||
|
"default": "",
|
||||||
|
"description": "飞书的app_secret"
|
||||||
|
},
|
||||||
|
"bot_name": {
|
||||||
|
"type": "string",
|
||||||
|
"default": "",
|
||||||
|
"description": "飞书的bot_name"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"title": "Discord 适配器",
|
||||||
|
"description": "用于接入 Discord",
|
||||||
|
"properties": {
|
||||||
|
"adapter": {
|
||||||
|
"type": "string",
|
||||||
|
"const": "discord"
|
||||||
|
},
|
||||||
|
"enable": {
|
||||||
|
"type": "boolean",
|
||||||
|
"default": false,
|
||||||
|
"description": "是否启用此适配器",
|
||||||
|
"layout": {
|
||||||
|
"comp": "switch",
|
||||||
|
"props": {
|
||||||
|
"color": "primary"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"client_id": {
|
||||||
|
"type": "string",
|
||||||
|
"default": "",
|
||||||
|
"description": "Discord 的 client_id"
|
||||||
|
},
|
||||||
|
"token": {
|
||||||
|
"type": "string",
|
||||||
|
"default": "",
|
||||||
|
"description": "Discord 的 token"
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
]
|
]
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -74,6 +74,14 @@
|
|||||||
"type": "string"
|
"type": "string"
|
||||||
},
|
},
|
||||||
"default": []
|
"default": []
|
||||||
|
},
|
||||||
|
"siliconflow": {
|
||||||
|
"type": "array",
|
||||||
|
"title": "SiliconFlow API 密钥",
|
||||||
|
"items": {
|
||||||
|
"type": "string"
|
||||||
|
},
|
||||||
|
"default": []
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
@@ -172,7 +180,8 @@
|
|||||||
"title": "API URL"
|
"title": "API URL"
|
||||||
},
|
},
|
||||||
"args": {
|
"args": {
|
||||||
"type": "object"
|
"type": "object",
|
||||||
|
"default": {}
|
||||||
},
|
},
|
||||||
"timeout": {
|
"timeout": {
|
||||||
"type": "number",
|
"type": "number",
|
||||||
@@ -191,7 +200,8 @@
|
|||||||
"title": "API URL"
|
"title": "API URL"
|
||||||
},
|
},
|
||||||
"args": {
|
"args": {
|
||||||
"type": "object"
|
"type": "object",
|
||||||
|
"default": {}
|
||||||
},
|
},
|
||||||
"timeout": {
|
"timeout": {
|
||||||
"type": "number",
|
"type": "number",
|
||||||
@@ -210,7 +220,8 @@
|
|||||||
"title": "API URL"
|
"title": "API URL"
|
||||||
},
|
},
|
||||||
"args": {
|
"args": {
|
||||||
"type": "object"
|
"type": "object",
|
||||||
|
"default": {}
|
||||||
},
|
},
|
||||||
"timeout": {
|
"timeout": {
|
||||||
"type": "number",
|
"type": "number",
|
||||||
@@ -229,10 +240,52 @@
|
|||||||
"title": "API URL"
|
"title": "API URL"
|
||||||
},
|
},
|
||||||
"args": {
|
"args": {
|
||||||
"type": "object"
|
"type": "object",
|
||||||
|
"default": {}
|
||||||
},
|
},
|
||||||
"timeout": {
|
"timeout": {
|
||||||
"type": "number"
|
"type": "number",
|
||||||
|
"default": 120
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"lmstudio-chat-completions": {
|
||||||
|
"type": "object",
|
||||||
|
"title": "LMStudio API 请求配置",
|
||||||
|
"description": "仅可编辑 URL 和 超时时间,额外请求参数不支持可视化编辑,请到编辑器编辑",
|
||||||
|
"properties": {
|
||||||
|
"base-url": {
|
||||||
|
"type": "string",
|
||||||
|
"title": "API URL"
|
||||||
|
},
|
||||||
|
"args": {
|
||||||
|
"type": "object",
|
||||||
|
"default": {}
|
||||||
|
},
|
||||||
|
"timeout": {
|
||||||
|
"type": "number",
|
||||||
|
"title": "API 请求超时时间",
|
||||||
|
"default": 120
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"siliconflow-chat-completions": {
|
||||||
|
"type": "object",
|
||||||
|
"title": "SiliconFlow API 请求配置",
|
||||||
|
"description": "仅可编辑 URL 和 超时时间,额外请求参数不支持可视化编辑,请到编辑器编辑",
|
||||||
|
"properties": {
|
||||||
|
"base-url": {
|
||||||
|
"type": "string",
|
||||||
|
"title": "API URL"
|
||||||
|
},
|
||||||
|
"args": {
|
||||||
|
"type": "object",
|
||||||
|
"default": {}
|
||||||
|
},
|
||||||
|
"timeout": {
|
||||||
|
"type": "number",
|
||||||
|
"title": "API 请求超时时间",
|
||||||
|
"default": 120
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -292,7 +345,7 @@
|
|||||||
"type": "string",
|
"type": "string",
|
||||||
"title": "应用类型",
|
"title": "应用类型",
|
||||||
"description": "支持 chat 和 workflow,chat:聊天助手(含高级编排)和 Agent;workflow:工作流;请填写下方对应的应用类型 API 参数",
|
"description": "支持 chat 和 workflow,chat:聊天助手(含高级编排)和 Agent;workflow:工作流;请填写下方对应的应用类型 API 参数",
|
||||||
"enum": ["chat", "workflow"],
|
"enum": ["chat", "workflow", "agent"],
|
||||||
"default": "chat"
|
"default": "chat"
|
||||||
},
|
},
|
||||||
"chat": {
|
"chat": {
|
||||||
|
|||||||
Reference in New Issue
Block a user