mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-06-04 21:06:03 +00:00
Compare commits
19 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
9fccf84987 | ||
|
|
0f59788184 | ||
|
|
0ad52bcd3f | ||
|
|
d7d710ec07 | ||
|
|
75a9a3e9af | ||
|
|
70503bedb7 | ||
|
|
7890eac3f8 | ||
|
|
e15f3595b3 | ||
|
|
eebd6a6ba3 | ||
|
|
0407f3e4ac | ||
|
|
5abca84437 | ||
|
|
d2776cc1e6 | ||
|
|
9fe0ee2b77 | ||
|
|
b68daac323 | ||
|
|
665de5dc43 | ||
|
|
e3b280758c | ||
|
|
374ae25d9c | ||
|
|
c86529ac99 | ||
|
|
6309f1fb78 |
2
.github/dependabot.yml
vendored
2
.github/dependabot.yml
vendored
@@ -10,6 +10,6 @@ updates:
|
|||||||
schedule:
|
schedule:
|
||||||
interval: "weekly"
|
interval: "weekly"
|
||||||
allow:
|
allow:
|
||||||
- dependency-name: "yiri-mirai"
|
- dependency-name: "yiri-mirai-rc"
|
||||||
- dependency-name: "dulwich"
|
- dependency-name: "dulwich"
|
||||||
- dependency-name: "openai"
|
- dependency-name: "openai"
|
||||||
|
|||||||
@@ -25,7 +25,7 @@ jobs:
|
|||||||
|
|
||||||
- name: Install dependencies
|
- name: Install dependencies
|
||||||
run: |
|
run: |
|
||||||
python -m pip install --upgrade yiri-mirai openai>=1.0.0 colorlog func_timeout dulwich Pillow CallingGPT tiktoken
|
python -m pip install --upgrade yiri-mirai-rc openai>=1.0.0 colorlog func_timeout dulwich Pillow CallingGPT tiktoken
|
||||||
python -m pip install -U openai>=1.0.0
|
python -m pip install -U openai>=1.0.0
|
||||||
|
|
||||||
- name: Copy Scripts
|
- name: Copy Scripts
|
||||||
|
|||||||
@@ -278,7 +278,7 @@ cd QChatGPT
|
|||||||
2. 安装依赖
|
2. 安装依赖
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
pip3 install requests yiri-mirai openai colorlog func_timeout dulwich Pillow nakuru-project-idk CallingGPT tiktoken
|
pip3 install requests -r requirements.txt
|
||||||
```
|
```
|
||||||
|
|
||||||
3. 运行一次主程序,生成配置文件
|
3. 运行一次主程序,生成配置文件
|
||||||
|
|||||||
@@ -141,7 +141,7 @@ cd QChatGPT
|
|||||||
2. Install dependencies
|
2. Install dependencies
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
pip3 install requests yiri-mirai openai colorlog func_timeout dulwich Pillow nakuru-project-idk
|
pip3 install requests yiri-mirai-rc openai colorlog func_timeout dulwich Pillow nakuru-project-idk
|
||||||
```
|
```
|
||||||
|
|
||||||
3. Generate `config.py`
|
3. Generate `config.py`
|
||||||
|
|||||||
@@ -114,7 +114,7 @@ admin_qq = 0
|
|||||||
#
|
#
|
||||||
# 还可以加载文件中的预设文字,使用方法请查看:https://github.com/RockChinQ/QChatGPT/wiki/%E5%8A%9F%E8%83%BD%E4%BD%BF%E7%94%A8#%E9%A2%84%E8%AE%BE%E6%96%87%E5%AD%97
|
# 还可以加载文件中的预设文字,使用方法请查看:https://github.com/RockChinQ/QChatGPT/wiki/%E5%8A%9F%E8%83%BD%E4%BD%BF%E7%94%A8#%E9%A2%84%E8%AE%BE%E6%96%87%E5%AD%97
|
||||||
default_prompt = {
|
default_prompt = {
|
||||||
"default": "如果我之后想获取帮助,请你说“输入!help获取帮助”",
|
"default": "如果用户之后想获取帮助,请你说“输入!help获取帮助”。",
|
||||||
}
|
}
|
||||||
|
|
||||||
# 情景预设格式
|
# 情景预设格式
|
||||||
@@ -291,7 +291,7 @@ show_prefix = False
|
|||||||
# 当此次消息处理时间低于此秒数时,将会强制延迟至此秒数
|
# 当此次消息处理时间低于此秒数时,将会强制延迟至此秒数
|
||||||
# 例如:[1.5, 3],则每次处理时会随机取一个1.5-3秒的随机数,若处理时间低于此随机数,则强制延迟至此随机秒数
|
# 例如:[1.5, 3],则每次处理时会随机取一个1.5-3秒的随机数,若处理时间低于此随机数,则强制延迟至此随机秒数
|
||||||
# 若您不需要此功能,请将force_delay_range设置为[0, 0]
|
# 若您不需要此功能,请将force_delay_range设置为[0, 0]
|
||||||
force_delay_range = [1.5, 3]
|
force_delay_range = [0, 0]
|
||||||
|
|
||||||
# 应用长消息处理策略的阈值
|
# 应用长消息处理策略的阈值
|
||||||
# 当回复消息长度超过此值时,将使用长消息处理策略
|
# 当回复消息长度超过此值时,将使用长消息处理策略
|
||||||
|
|||||||
85
main.py
85
main.py
@@ -13,6 +13,48 @@ sys.path.append(".")
|
|||||||
|
|
||||||
from pkg.utils.log import init_runtime_log_file, reset_logging
|
from pkg.utils.log import init_runtime_log_file, reset_logging
|
||||||
|
|
||||||
|
|
||||||
|
def check_file():
|
||||||
|
# 检查是否有banlist.py,如果没有就把banlist-template.py复制一份
|
||||||
|
if not os.path.exists('banlist.py'):
|
||||||
|
shutil.copy('res/templates/banlist-template.py', 'banlist.py')
|
||||||
|
|
||||||
|
# 检查是否有sensitive.json
|
||||||
|
if not os.path.exists("sensitive.json"):
|
||||||
|
shutil.copy("res/templates/sensitive-template.json", "sensitive.json")
|
||||||
|
|
||||||
|
# 检查是否有scenario/default.json
|
||||||
|
if not os.path.exists("scenario/default.json"):
|
||||||
|
shutil.copy("scenario/default-template.json", "scenario/default.json")
|
||||||
|
|
||||||
|
# 检查cmdpriv.json
|
||||||
|
if not os.path.exists("cmdpriv.json"):
|
||||||
|
shutil.copy("res/templates/cmdpriv-template.json", "cmdpriv.json")
|
||||||
|
|
||||||
|
# 检查tips_custom
|
||||||
|
if not os.path.exists("tips.py"):
|
||||||
|
shutil.copy("tips-custom-template.py", "tips.py")
|
||||||
|
|
||||||
|
# 检查temp目录
|
||||||
|
if not os.path.exists("temp/"):
|
||||||
|
os.mkdir("temp/")
|
||||||
|
|
||||||
|
# 检查并创建plugins、prompts目录
|
||||||
|
check_path = ["plugins", "prompts"]
|
||||||
|
for path in check_path:
|
||||||
|
if not os.path.exists(path):
|
||||||
|
os.mkdir(path)
|
||||||
|
|
||||||
|
# 配置文件存在性校验
|
||||||
|
if not os.path.exists('config.py'):
|
||||||
|
shutil.copy('config-template.py', 'config.py')
|
||||||
|
print('请先在config.py中填写配置')
|
||||||
|
sys.exit(0)
|
||||||
|
|
||||||
|
|
||||||
|
# 初始化相关文件
|
||||||
|
check_file()
|
||||||
|
|
||||||
try:
|
try:
|
||||||
import colorlog
|
import colorlog
|
||||||
except ImportError:
|
except ImportError:
|
||||||
@@ -20,7 +62,6 @@ except ImportError:
|
|||||||
import pkg.utils.pkgmgr as pkgmgr
|
import pkg.utils.pkgmgr as pkgmgr
|
||||||
try:
|
try:
|
||||||
pkgmgr.install_requirements("requirements.txt")
|
pkgmgr.install_requirements("requirements.txt")
|
||||||
pkgmgr.install_upgrade("websockets")
|
|
||||||
import colorlog
|
import colorlog
|
||||||
except ImportError:
|
except ImportError:
|
||||||
print("依赖不满足,请查看 https://github.com/RockChinQ/qcg-installer/issues/15")
|
print("依赖不满足,请查看 https://github.com/RockChinQ/qcg-installer/issues/15")
|
||||||
@@ -200,6 +241,7 @@ def start(first_time_init=False):
|
|||||||
|
|
||||||
# 配置openai api_base
|
# 配置openai api_base
|
||||||
if "reverse_proxy" in config.openai_config and config.openai_config["reverse_proxy"] is not None:
|
if "reverse_proxy" in config.openai_config and config.openai_config["reverse_proxy"] is not None:
|
||||||
|
logging.debug("设置反向代理: "+config.openai_config['reverse_proxy'])
|
||||||
openai.base_url = config.openai_config["reverse_proxy"]
|
openai.base_url = config.openai_config["reverse_proxy"]
|
||||||
|
|
||||||
# 主启动流程
|
# 主启动流程
|
||||||
@@ -368,53 +410,12 @@ def stop():
|
|||||||
raise e
|
raise e
|
||||||
|
|
||||||
|
|
||||||
def check_file():
|
|
||||||
# 检查是否有banlist.py,如果没有就把banlist-template.py复制一份
|
|
||||||
if not os.path.exists('banlist.py'):
|
|
||||||
shutil.copy('res/templates/banlist-template.py', 'banlist.py')
|
|
||||||
|
|
||||||
# 检查是否有sensitive.json
|
|
||||||
if not os.path.exists("sensitive.json"):
|
|
||||||
shutil.copy("res/templates/sensitive-template.json", "sensitive.json")
|
|
||||||
|
|
||||||
# 检查是否有scenario/default.json
|
|
||||||
if not os.path.exists("scenario/default.json"):
|
|
||||||
shutil.copy("scenario/default-template.json", "scenario/default.json")
|
|
||||||
|
|
||||||
# 检查cmdpriv.json
|
|
||||||
if not os.path.exists("cmdpriv.json"):
|
|
||||||
shutil.copy("res/templates/cmdpriv-template.json", "cmdpriv.json")
|
|
||||||
|
|
||||||
# 检查tips_custom
|
|
||||||
if not os.path.exists("tips.py"):
|
|
||||||
shutil.copy("tips-custom-template.py", "tips.py")
|
|
||||||
|
|
||||||
# 检查temp目录
|
|
||||||
if not os.path.exists("temp/"):
|
|
||||||
os.mkdir("temp/")
|
|
||||||
|
|
||||||
# 检查并创建plugins、prompts目录
|
|
||||||
check_path = ["plugins", "prompts"]
|
|
||||||
for path in check_path:
|
|
||||||
if not os.path.exists(path):
|
|
||||||
os.mkdir(path)
|
|
||||||
|
|
||||||
# 配置文件存在性校验
|
|
||||||
if not os.path.exists('config.py'):
|
|
||||||
shutil.copy('config-template.py', 'config.py')
|
|
||||||
print('请先在config.py中填写配置')
|
|
||||||
sys.exit(0)
|
|
||||||
|
|
||||||
|
|
||||||
def main():
|
def main():
|
||||||
global use_override
|
global use_override
|
||||||
# 检查是否携带了 --override 或 -r 参数
|
# 检查是否携带了 --override 或 -r 参数
|
||||||
if '--override' in sys.argv or '-r' in sys.argv:
|
if '--override' in sys.argv or '-r' in sys.argv:
|
||||||
use_override = True
|
use_override = True
|
||||||
|
|
||||||
# 初始化相关文件
|
|
||||||
check_file()
|
|
||||||
|
|
||||||
# 初始化logging
|
# 初始化logging
|
||||||
init_runtime_log_file()
|
init_runtime_log_file()
|
||||||
pkg.utils.context.context['logger_handler'] = reset_logging()
|
pkg.utils.context.context['logger_handler'] = reset_logging()
|
||||||
|
|||||||
@@ -24,7 +24,7 @@
|
|||||||
"switch_strategy": "active",
|
"switch_strategy": "active",
|
||||||
"admin_qq": 0,
|
"admin_qq": 0,
|
||||||
"default_prompt": {
|
"default_prompt": {
|
||||||
"default": "如果我之后想获取帮助,请你说“输入!help获取帮助”"
|
"default": "如果用户之后想获取帮助,请你说“输入!help获取帮助”。"
|
||||||
},
|
},
|
||||||
"preset_mode": "normal",
|
"preset_mode": "normal",
|
||||||
"response_rules": {
|
"response_rules": {
|
||||||
@@ -69,8 +69,8 @@
|
|||||||
"process_message_timeout": 120,
|
"process_message_timeout": 120,
|
||||||
"show_prefix": false,
|
"show_prefix": false,
|
||||||
"force_delay_range": [
|
"force_delay_range": [
|
||||||
1.5,
|
0,
|
||||||
3
|
0
|
||||||
],
|
],
|
||||||
"blob_message_threshold": 256,
|
"blob_message_threshold": 256,
|
||||||
"blob_message_strategy": "forward",
|
"blob_message_strategy": "forward",
|
||||||
|
|||||||
@@ -9,8 +9,8 @@ import threading
|
|||||||
|
|
||||||
import requests
|
import requests
|
||||||
|
|
||||||
import pkg.utils.context
|
from ..utils import context
|
||||||
import pkg.utils.updater
|
from ..utils import updater
|
||||||
|
|
||||||
|
|
||||||
class DataGatherer:
|
class DataGatherer:
|
||||||
@@ -33,7 +33,7 @@ class DataGatherer:
|
|||||||
def __init__(self):
|
def __init__(self):
|
||||||
self.load_from_db()
|
self.load_from_db()
|
||||||
try:
|
try:
|
||||||
self.version_str = pkg.utils.updater.get_current_tag() # 从updater模块获取版本号
|
self.version_str = updater.get_current_tag() # 从updater模块获取版本号
|
||||||
except:
|
except:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
@@ -47,7 +47,7 @@ class DataGatherer:
|
|||||||
def thread_func():
|
def thread_func():
|
||||||
|
|
||||||
try:
|
try:
|
||||||
config = pkg.utils.context.get_config()
|
config = context.get_config()
|
||||||
if not config.report_usage:
|
if not config.report_usage:
|
||||||
return
|
return
|
||||||
res = requests.get("http://reports.rockchin.top:18989/usage?service_name=qchatgpt.{}&version={}&count={}&msg_source={}".format(subservice_name, self.version_str, count, config.msg_source_adapter))
|
res = requests.get("http://reports.rockchin.top:18989/usage?service_name=qchatgpt.{}&version={}&count={}&msg_source={}".format(subservice_name, self.version_str, count, config.msg_source_adapter))
|
||||||
@@ -64,7 +64,7 @@ class DataGatherer:
|
|||||||
def report_text_model_usage(self, model, total_tokens):
|
def report_text_model_usage(self, model, total_tokens):
|
||||||
"""调用方报告文字模型请求文字使用量"""
|
"""调用方报告文字模型请求文字使用量"""
|
||||||
|
|
||||||
key_md5 = pkg.utils.context.get_openai_manager().key_mgr.get_using_key_md5() # 以key的md5进行储存
|
key_md5 = context.get_openai_manager().key_mgr.get_using_key_md5() # 以key的md5进行储存
|
||||||
|
|
||||||
if key_md5 not in self.usage:
|
if key_md5 not in self.usage:
|
||||||
self.usage[key_md5] = {}
|
self.usage[key_md5] = {}
|
||||||
@@ -84,7 +84,7 @@ class DataGatherer:
|
|||||||
def report_image_model_usage(self, size):
|
def report_image_model_usage(self, size):
|
||||||
"""调用方报告图片模型请求图片使用量"""
|
"""调用方报告图片模型请求图片使用量"""
|
||||||
|
|
||||||
key_md5 = pkg.utils.context.get_openai_manager().key_mgr.get_using_key_md5()
|
key_md5 = context.get_openai_manager().key_mgr.get_using_key_md5()
|
||||||
|
|
||||||
if key_md5 not in self.usage:
|
if key_md5 not in self.usage:
|
||||||
self.usage[key_md5] = {}
|
self.usage[key_md5] = {}
|
||||||
@@ -131,9 +131,9 @@ class DataGatherer:
|
|||||||
return total
|
return total
|
||||||
|
|
||||||
def dump_to_db(self):
|
def dump_to_db(self):
|
||||||
pkg.utils.context.get_database_manager().dump_usage_json(self.usage)
|
context.get_database_manager().dump_usage_json(self.usage)
|
||||||
|
|
||||||
def load_from_db(self):
|
def load_from_db(self):
|
||||||
json_str = pkg.utils.context.get_database_manager().load_usage_json()
|
json_str = context.get_database_manager().load_usage_json()
|
||||||
if json_str is not None:
|
if json_str is not None:
|
||||||
self.usage = json.loads(json_str)
|
self.usage = json.loads(json_str)
|
||||||
|
|||||||
@@ -5,11 +5,10 @@ import hashlib
|
|||||||
import json
|
import json
|
||||||
import logging
|
import logging
|
||||||
import time
|
import time
|
||||||
from sqlite3 import Cursor
|
|
||||||
|
|
||||||
import sqlite3
|
import sqlite3
|
||||||
|
|
||||||
import pkg.utils.context
|
from ..utils import context
|
||||||
|
|
||||||
|
|
||||||
class DatabaseManager:
|
class DatabaseManager:
|
||||||
@@ -22,7 +21,7 @@ class DatabaseManager:
|
|||||||
|
|
||||||
self.reconnect()
|
self.reconnect()
|
||||||
|
|
||||||
pkg.utils.context.set_database_manager(self)
|
context.set_database_manager(self)
|
||||||
|
|
||||||
# 连接到数据库文件
|
# 连接到数据库文件
|
||||||
def reconnect(self):
|
def reconnect(self):
|
||||||
@@ -33,7 +32,7 @@ class DatabaseManager:
|
|||||||
def close(self):
|
def close(self):
|
||||||
self.conn.close()
|
self.conn.close()
|
||||||
|
|
||||||
def __execute__(self, *args, **kwargs) -> Cursor:
|
def __execute__(self, *args, **kwargs) -> sqlite3.Cursor:
|
||||||
# logging.debug('SQL: {}'.format(sql))
|
# logging.debug('SQL: {}'.format(sql))
|
||||||
logging.debug('SQL: {}'.format(args))
|
logging.debug('SQL: {}'.format(args))
|
||||||
c = self.cursor.execute(*args, **kwargs)
|
c = self.cursor.execute(*args, **kwargs)
|
||||||
@@ -145,7 +144,7 @@ class DatabaseManager:
|
|||||||
# 从数据库加载还没过期的session数据
|
# 从数据库加载还没过期的session数据
|
||||||
def load_valid_sessions(self) -> dict:
|
def load_valid_sessions(self) -> dict:
|
||||||
# 从数据库中加载所有还没过期的session
|
# 从数据库中加载所有还没过期的session
|
||||||
config = pkg.utils.context.get_config()
|
config = context.get_config()
|
||||||
self.__execute__("""
|
self.__execute__("""
|
||||||
select `name`, `type`, `number`, `create_timestamp`, `last_interact_timestamp`, `prompt`, `status`, `default_prompt`, `token_counts`
|
select `name`, `type`, `number`, `create_timestamp`, `last_interact_timestamp`, `prompt`, `status`, `default_prompt`, `token_counts`
|
||||||
from `sessions` where `last_interact_timestamp` > {}
|
from `sessions` where `last_interact_timestamp` > {}
|
||||||
|
|||||||
@@ -1,11 +1,11 @@
|
|||||||
import openai
|
|
||||||
from openai.types.chat import chat_completion_message
|
|
||||||
import json
|
import json
|
||||||
import logging
|
import logging
|
||||||
|
|
||||||
from .model import RequestBase
|
import openai
|
||||||
|
from openai.types.chat import chat_completion_message
|
||||||
|
|
||||||
from ..funcmgr import get_func_schema_list, execute_function, get_func, get_func_schema, ContentFunctionNotFoundError
|
from .model import RequestBase
|
||||||
|
from .. import funcmgr
|
||||||
|
|
||||||
|
|
||||||
class ChatCompletionRequest(RequestBase):
|
class ChatCompletionRequest(RequestBase):
|
||||||
@@ -81,7 +81,7 @@ class ChatCompletionRequest(RequestBase):
|
|||||||
"messages": self.messages,
|
"messages": self.messages,
|
||||||
}
|
}
|
||||||
|
|
||||||
funcs = get_func_schema_list()
|
funcs = funcmgr.get_func_schema_list()
|
||||||
|
|
||||||
if len(funcs) > 0:
|
if len(funcs) > 0:
|
||||||
args['functions'] = funcs
|
args['functions'] = funcs
|
||||||
@@ -171,7 +171,7 @@ class ChatCompletionRequest(RequestBase):
|
|||||||
# 若不是json格式的异常处理
|
# 若不是json格式的异常处理
|
||||||
except json.decoder.JSONDecodeError:
|
except json.decoder.JSONDecodeError:
|
||||||
# 获取函数的参数列表
|
# 获取函数的参数列表
|
||||||
func_schema = get_func_schema(func_name)
|
func_schema = funcmgr.get_func_schema(func_name)
|
||||||
|
|
||||||
arguments = {
|
arguments = {
|
||||||
func_schema['parameters']['required'][0]: cp_pending_func_call.arguments
|
func_schema['parameters']['required'][0]: cp_pending_func_call.arguments
|
||||||
@@ -182,7 +182,7 @@ class ChatCompletionRequest(RequestBase):
|
|||||||
# 执行函数调用
|
# 执行函数调用
|
||||||
ret = ""
|
ret = ""
|
||||||
try:
|
try:
|
||||||
ret = execute_function(func_name, arguments)
|
ret = funcmgr.execute_function(func_name, arguments)
|
||||||
|
|
||||||
logging.info("函数执行完成。")
|
logging.info("函数执行完成。")
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
@@ -216,6 +216,5 @@ class ChatCompletionRequest(RequestBase):
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
except ContentFunctionNotFoundError:
|
except funcmgr.ContentFunctionNotFoundError:
|
||||||
raise Exception("没有找到函数: {}".format(func_name))
|
raise Exception("没有找到函数: {}".format(func_name))
|
||||||
|
|
||||||
|
|||||||
@@ -1,10 +1,10 @@
|
|||||||
import openai
|
import openai
|
||||||
from openai.types import completion, completion_choice
|
from openai.types import completion, completion_choice
|
||||||
|
|
||||||
from .model import RequestBase
|
from . import model
|
||||||
|
|
||||||
|
|
||||||
class CompletionRequest(RequestBase):
|
class CompletionRequest(model.RequestBase):
|
||||||
"""调用Completion接口的请求类。
|
"""调用Completion接口的请求类。
|
||||||
|
|
||||||
调用方可以一直next completion直到finish_reason为stop。
|
调用方可以一直next completion直到finish_reason为stop。
|
||||||
|
|||||||
@@ -1,6 +1,4 @@
|
|||||||
# 定义不同接口请求的模型
|
# 定义不同接口请求的模型
|
||||||
import threading
|
|
||||||
import asyncio
|
|
||||||
import logging
|
import logging
|
||||||
|
|
||||||
import openai
|
import openai
|
||||||
@@ -23,6 +21,7 @@ class RequestBase:
|
|||||||
|
|
||||||
def _req(self, **kwargs):
|
def _req(self, **kwargs):
|
||||||
"""处理代理问题"""
|
"""处理代理问题"""
|
||||||
|
logging.debug("请求接口参数: %s", str(kwargs))
|
||||||
import config
|
import config
|
||||||
|
|
||||||
ret = self.req_func(**kwargs)
|
ret = self.req_func(**kwargs)
|
||||||
|
|||||||
@@ -16,10 +16,6 @@ import os
|
|||||||
# __scenario_from_files__ = {}
|
# __scenario_from_files__ = {}
|
||||||
|
|
||||||
|
|
||||||
__universal_first_reply__ = "ok, I'll follow your commands."
|
|
||||||
"""通用首次回复"""
|
|
||||||
|
|
||||||
|
|
||||||
class ScenarioMode:
|
class ScenarioMode:
|
||||||
"""情景预设模式抽象类"""
|
"""情景预设模式抽象类"""
|
||||||
|
|
||||||
@@ -66,17 +62,13 @@ class NormalScenarioMode(ScenarioMode):
|
|||||||
"""普通情景预设模式"""
|
"""普通情景预设模式"""
|
||||||
|
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
global __universal_first_reply__
|
|
||||||
# 加载config中的default_prompt值
|
# 加载config中的default_prompt值
|
||||||
if type(config.default_prompt) == str:
|
if type(config.default_prompt) == str:
|
||||||
self.using_prompt_name = "default"
|
self.using_prompt_name = "default"
|
||||||
self.prompts = {"default": [
|
self.prompts = {"default": [
|
||||||
{
|
{
|
||||||
"role": "user",
|
"role": "system",
|
||||||
"content": config.default_prompt
|
"content": config.default_prompt
|
||||||
},{
|
|
||||||
"role": "assistant",
|
|
||||||
"content": __universal_first_reply__
|
|
||||||
}
|
}
|
||||||
]}
|
]}
|
||||||
|
|
||||||
@@ -84,11 +76,8 @@ class NormalScenarioMode(ScenarioMode):
|
|||||||
for key in config.default_prompt:
|
for key in config.default_prompt:
|
||||||
self.prompts[key] = [
|
self.prompts[key] = [
|
||||||
{
|
{
|
||||||
"role": "user",
|
"role": "system",
|
||||||
"content": config.default_prompt[key]
|
"content": config.default_prompt[key]
|
||||||
},{
|
|
||||||
"role": "assistant",
|
|
||||||
"content": __universal_first_reply__
|
|
||||||
}
|
}
|
||||||
]
|
]
|
||||||
|
|
||||||
@@ -98,11 +87,8 @@ class NormalScenarioMode(ScenarioMode):
|
|||||||
with open(os.path.join("prompts", file), encoding="utf-8") as f:
|
with open(os.path.join("prompts", file), encoding="utf-8") as f:
|
||||||
self.prompts[file] = [
|
self.prompts[file] = [
|
||||||
{
|
{
|
||||||
"role": "user",
|
"role": "system",
|
||||||
"content": f.read()
|
"content": f.read()
|
||||||
},{
|
|
||||||
"role": "assistant",
|
|
||||||
"content": __universal_first_reply__
|
|
||||||
}
|
}
|
||||||
]
|
]
|
||||||
|
|
||||||
|
|||||||
@@ -1,8 +1,7 @@
|
|||||||
# 封装了function calling的一些支持函数
|
# 封装了function calling的一些支持函数
|
||||||
import logging
|
import logging
|
||||||
|
|
||||||
|
from ..plugin import host
|
||||||
from pkg.plugin import host
|
|
||||||
|
|
||||||
|
|
||||||
class ContentFunctionNotFoundError(Exception):
|
class ContentFunctionNotFoundError(Exception):
|
||||||
|
|||||||
@@ -2,8 +2,8 @@
|
|||||||
import hashlib
|
import hashlib
|
||||||
import logging
|
import logging
|
||||||
|
|
||||||
import pkg.plugin.host as plugin_host
|
from ..plugin import host as plugin_host
|
||||||
import pkg.plugin.models as plugin_models
|
from ..plugin import models as plugin_models
|
||||||
|
|
||||||
|
|
||||||
class KeysManager:
|
class KeysManager:
|
||||||
|
|||||||
@@ -2,12 +2,11 @@ import logging
|
|||||||
|
|
||||||
import openai
|
import openai
|
||||||
|
|
||||||
import pkg.openai.keymgr
|
from ..openai import keymgr
|
||||||
import pkg.utils.context
|
from ..utils import context
|
||||||
import pkg.audit.gatherer
|
from ..audit import gatherer
|
||||||
from pkg.openai.modelmgr import select_request_cls
|
from ..openai import modelmgr
|
||||||
|
from ..openai.api import model as api_model
|
||||||
from pkg.openai.api.model import RequestBase
|
|
||||||
|
|
||||||
|
|
||||||
class OpenAIInteract:
|
class OpenAIInteract:
|
||||||
@@ -16,9 +15,9 @@ class OpenAIInteract:
|
|||||||
将文字接口和图片接口封装供调用方使用
|
将文字接口和图片接口封装供调用方使用
|
||||||
"""
|
"""
|
||||||
|
|
||||||
key_mgr: pkg.openai.keymgr.KeysManager = None
|
key_mgr: keymgr.KeysManager = None
|
||||||
|
|
||||||
audit_mgr: pkg.audit.gatherer.DataGatherer = None
|
audit_mgr: gatherer.DataGatherer = None
|
||||||
|
|
||||||
default_image_api_params = {
|
default_image_api_params = {
|
||||||
"size": "256x256",
|
"size": "256x256",
|
||||||
@@ -28,31 +27,32 @@ class OpenAIInteract:
|
|||||||
|
|
||||||
def __init__(self, api_key: str):
|
def __init__(self, api_key: str):
|
||||||
|
|
||||||
self.key_mgr = pkg.openai.keymgr.KeysManager(api_key)
|
self.key_mgr = keymgr.KeysManager(api_key)
|
||||||
self.audit_mgr = pkg.audit.gatherer.DataGatherer()
|
self.audit_mgr = gatherer.DataGatherer()
|
||||||
|
|
||||||
# logging.info("文字总使用量:%d", self.audit_mgr.get_total_text_length())
|
# logging.info("文字总使用量:%d", self.audit_mgr.get_total_text_length())
|
||||||
|
|
||||||
self.client = openai.Client(
|
self.client = openai.Client(
|
||||||
api_key=self.key_mgr.get_using_key()
|
api_key=self.key_mgr.get_using_key(),
|
||||||
|
base_url=openai.base_url
|
||||||
)
|
)
|
||||||
|
|
||||||
pkg.utils.context.set_openai_manager(self)
|
context.set_openai_manager(self)
|
||||||
|
|
||||||
def request_completion(self, messages: list):
|
def request_completion(self, messages: list):
|
||||||
"""请求补全接口回复=
|
"""请求补全接口回复=
|
||||||
"""
|
"""
|
||||||
# 选择接口请求类
|
# 选择接口请求类
|
||||||
config = pkg.utils.context.get_config()
|
config = context.get_config()
|
||||||
|
|
||||||
request: RequestBase
|
request: api_model.RequestBase
|
||||||
|
|
||||||
model: str = config.completion_api_params['model']
|
model: str = config.completion_api_params['model']
|
||||||
|
|
||||||
cp_parmas = config.completion_api_params.copy()
|
cp_parmas = config.completion_api_params.copy()
|
||||||
del cp_parmas['model']
|
del cp_parmas['model']
|
||||||
|
|
||||||
request = select_request_cls(self.client, model, messages, cp_parmas)
|
request = modelmgr.select_request_cls(self.client, model, messages, cp_parmas)
|
||||||
|
|
||||||
# 请求接口
|
# 请求接口
|
||||||
for resp in request:
|
for resp in request:
|
||||||
@@ -74,7 +74,7 @@ class OpenAIInteract:
|
|||||||
Returns:
|
Returns:
|
||||||
dict: 响应
|
dict: 响应
|
||||||
"""
|
"""
|
||||||
config = pkg.utils.context.get_config()
|
config = context.get_config()
|
||||||
params = config.image_api_params
|
params = config.image_api_params
|
||||||
|
|
||||||
response = openai.Image.create(
|
response = openai.Image.create(
|
||||||
|
|||||||
@@ -8,9 +8,9 @@ Completion - text-davinci-003 等模型
|
|||||||
import tiktoken
|
import tiktoken
|
||||||
import openai
|
import openai
|
||||||
|
|
||||||
from pkg.openai.api.model import RequestBase
|
from ..openai.api import model as api_model
|
||||||
from pkg.openai.api.completion import CompletionRequest
|
from ..openai.api import completion as api_completion
|
||||||
from pkg.openai.api.chat_completion import ChatCompletionRequest
|
from ..openai.api import chat_completion as api_chat_completion
|
||||||
|
|
||||||
COMPLETION_MODELS = {
|
COMPLETION_MODELS = {
|
||||||
"text-davinci-003", # legacy
|
"text-davinci-003", # legacy
|
||||||
@@ -60,11 +60,11 @@ IMAGE_MODELS = {
|
|||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
def select_request_cls(client: openai.Client, model_name: str, messages: list, args: dict) -> RequestBase:
|
def select_request_cls(client: openai.Client, model_name: str, messages: list, args: dict) -> api_model.RequestBase:
|
||||||
if model_name in CHAT_COMPLETION_MODELS:
|
if model_name in CHAT_COMPLETION_MODELS:
|
||||||
return ChatCompletionRequest(client, model_name, messages, **args)
|
return api_chat_completion.ChatCompletionRequest(client, model_name, messages, **args)
|
||||||
elif model_name in COMPLETION_MODELS:
|
elif model_name in COMPLETION_MODELS:
|
||||||
return CompletionRequest(client, model_name, messages, **args)
|
return api_completion.CompletionRequest(client, model_name, messages, **args)
|
||||||
raise ValueError("不支持模型[{}],请检查配置文件".format(model_name))
|
raise ValueError("不支持模型[{}],请检查配置文件".format(model_name))
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -8,15 +8,13 @@ import threading
|
|||||||
import time
|
import time
|
||||||
import json
|
import json
|
||||||
|
|
||||||
import pkg.openai.manager
|
from ..openai import manager as openai_manager
|
||||||
import pkg.openai.modelmgr
|
from ..openai import modelmgr as openai_modelmgr
|
||||||
import pkg.database.manager
|
from ..database import manager as database_manager
|
||||||
import pkg.utils.context
|
from ..utils import context as context
|
||||||
|
|
||||||
import pkg.plugin.host as plugin_host
|
from ..plugin import host as plugin_host
|
||||||
import pkg.plugin.models as plugin_models
|
from ..plugin import models as plugin_models
|
||||||
|
|
||||||
from pkg.openai.modelmgr import count_tokens
|
|
||||||
|
|
||||||
# 运行时保存的所有session
|
# 运行时保存的所有session
|
||||||
sessions = {}
|
sessions = {}
|
||||||
@@ -38,7 +36,7 @@ def reset_session_prompt(session_name, prompt):
|
|||||||
f.write(prompt)
|
f.write(prompt)
|
||||||
f.close()
|
f.close()
|
||||||
# 生成新数据
|
# 生成新数据
|
||||||
config = pkg.utils.context.get_config()
|
config = context.get_config()
|
||||||
prompt = [
|
prompt = [
|
||||||
{
|
{
|
||||||
'role': 'system',
|
'role': 'system',
|
||||||
@@ -61,7 +59,7 @@ def load_sessions():
|
|||||||
|
|
||||||
global sessions
|
global sessions
|
||||||
|
|
||||||
db_inst = pkg.utils.context.get_database_manager()
|
db_inst = context.get_database_manager()
|
||||||
|
|
||||||
session_data = db_inst.load_valid_sessions()
|
session_data = db_inst.load_valid_sessions()
|
||||||
|
|
||||||
@@ -172,7 +170,7 @@ class Session:
|
|||||||
if self.create_timestamp != create_timestamp or self not in sessions.values():
|
if self.create_timestamp != create_timestamp or self not in sessions.values():
|
||||||
return
|
return
|
||||||
|
|
||||||
config = pkg.utils.context.get_config()
|
config = context.get_config()
|
||||||
if int(time.time()) - self.last_interact_timestamp > config.session_expire_time:
|
if int(time.time()) - self.last_interact_timestamp > config.session_expire_time:
|
||||||
logging.info('session {} 已过期'.format(self.name))
|
logging.info('session {} 已过期'.format(self.name))
|
||||||
|
|
||||||
@@ -182,7 +180,7 @@ class Session:
|
|||||||
'session': self,
|
'session': self,
|
||||||
'session_expire_time': config.session_expire_time
|
'session_expire_time': config.session_expire_time
|
||||||
}
|
}
|
||||||
event = pkg.plugin.host.emit(plugin_models.SessionExpired, **args)
|
event = plugin_host.emit(plugin_models.SessionExpired, **args)
|
||||||
if event.is_prevented_default():
|
if event.is_prevented_default():
|
||||||
return
|
return
|
||||||
|
|
||||||
@@ -214,11 +212,11 @@ class Session:
|
|||||||
'default_prompt': self.default_prompt,
|
'default_prompt': self.default_prompt,
|
||||||
}
|
}
|
||||||
|
|
||||||
event = pkg.plugin.host.emit(plugin_models.SessionFirstMessageReceived, **args)
|
event = plugin_host.emit(plugin_models.SessionFirstMessageReceived, **args)
|
||||||
if event.is_prevented_default():
|
if event.is_prevented_default():
|
||||||
return None, None, None
|
return None, None, None
|
||||||
|
|
||||||
config = pkg.utils.context.get_config()
|
config = context.get_config()
|
||||||
max_length = config.prompt_submit_length
|
max_length = config.prompt_submit_length
|
||||||
|
|
||||||
local_default_prompt = self.default_prompt.copy()
|
local_default_prompt = self.default_prompt.copy()
|
||||||
@@ -232,7 +230,7 @@ class Session:
|
|||||||
'text_message': text,
|
'text_message': text,
|
||||||
}
|
}
|
||||||
|
|
||||||
event = pkg.plugin.host.emit(plugin_models.PromptPreProcessing, **args)
|
event = plugin_host.emit(plugin_models.PromptPreProcessing, **args)
|
||||||
|
|
||||||
if event.get_return_value('default_prompt') is not None:
|
if event.get_return_value('default_prompt') is not None:
|
||||||
local_default_prompt = event.get_return_value('default_prompt')
|
local_default_prompt = event.get_return_value('default_prompt')
|
||||||
@@ -243,6 +241,7 @@ class Session:
|
|||||||
if event.get_return_value('text_message') is not None:
|
if event.get_return_value('text_message') is not None:
|
||||||
text = event.get_return_value('text_message')
|
text = event.get_return_value('text_message')
|
||||||
|
|
||||||
|
# 裁剪messages到合适长度
|
||||||
prompts, _ = self.cut_out(text, max_length, local_default_prompt, local_prompt)
|
prompts, _ = self.cut_out(text, max_length, local_default_prompt, local_prompt)
|
||||||
|
|
||||||
res_text = ""
|
res_text = ""
|
||||||
@@ -256,14 +255,14 @@ class Session:
|
|||||||
funcs = []
|
funcs = []
|
||||||
|
|
||||||
trace_func_calls = config.trace_function_calls
|
trace_func_calls = config.trace_function_calls
|
||||||
botmgr = pkg.utils.context.get_qqbot_manager()
|
botmgr = context.get_qqbot_manager()
|
||||||
|
|
||||||
session_name_spt: list[str] = self.name.split("_")
|
session_name_spt: list[str] = self.name.split("_")
|
||||||
|
|
||||||
pending_res_text = ""
|
pending_res_text = ""
|
||||||
|
|
||||||
# TODO 对不起,我知道这样非常非常屎山,但我之后会重构的
|
# TODO 对不起,我知道这样非常非常屎山,但我之后会重构的
|
||||||
for resp in pkg.utils.context.get_openai_manager().request_completion(prompts):
|
for resp in context.get_openai_manager().request_completion(prompts):
|
||||||
|
|
||||||
if pending_res_text != "":
|
if pending_res_text != "":
|
||||||
botmgr.adapter.send_message(
|
botmgr.adapter.send_message(
|
||||||
@@ -325,7 +324,6 @@ class Session:
|
|||||||
)
|
)
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
||||||
# 向API请求补全
|
# 向API请求补全
|
||||||
# message, total_token = pkg.utils.context.get_openai_manager().request_completion(
|
# message, total_token = pkg.utils.context.get_openai_manager().request_completion(
|
||||||
# prompts,
|
# prompts,
|
||||||
@@ -383,13 +381,13 @@ class Session:
|
|||||||
# 包装目前的对话回合内容
|
# 包装目前的对话回合内容
|
||||||
changable_prompts = []
|
changable_prompts = []
|
||||||
|
|
||||||
use_model = pkg.utils.context.get_config().completion_api_params['model']
|
use_model = context.get_config().completion_api_params['model']
|
||||||
|
|
||||||
ptr = len(prompt) - 1
|
ptr = len(prompt) - 1
|
||||||
|
|
||||||
# 直接从后向前扫描拼接,不管是否是整回合
|
# 直接从后向前扫描拼接,不管是否是整回合
|
||||||
while ptr >= 0:
|
while ptr >= 0:
|
||||||
if count_tokens(prompt[ptr:ptr+1]+changable_prompts, use_model) > max_tokens:
|
if openai_modelmgr.count_tokens(prompt[ptr:ptr+1]+changable_prompts, use_model) > max_tokens:
|
||||||
break
|
break
|
||||||
|
|
||||||
changable_prompts.insert(0, prompt[ptr])
|
changable_prompts.insert(0, prompt[ptr])
|
||||||
@@ -410,14 +408,14 @@ class Session:
|
|||||||
|
|
||||||
logging.debug("cut_out: {}".format(json.dumps(result_prompt, ensure_ascii=False, indent=4)))
|
logging.debug("cut_out: {}".format(json.dumps(result_prompt, ensure_ascii=False, indent=4)))
|
||||||
|
|
||||||
return result_prompt, count_tokens(changable_prompts, use_model)
|
return result_prompt, openai_modelmgr.count_tokens(changable_prompts, use_model)
|
||||||
|
|
||||||
# 持久化session
|
# 持久化session
|
||||||
def persistence(self):
|
def persistence(self):
|
||||||
if self.prompt == self.get_default_prompt():
|
if self.prompt == self.get_default_prompt():
|
||||||
return
|
return
|
||||||
|
|
||||||
db_inst = pkg.utils.context.get_database_manager()
|
db_inst = context.get_database_manager()
|
||||||
|
|
||||||
name_spt = self.name.split('_')
|
name_spt = self.name.split('_')
|
||||||
|
|
||||||
@@ -439,12 +437,12 @@ class Session:
|
|||||||
}
|
}
|
||||||
|
|
||||||
# 此事件不支持阻止默认行为
|
# 此事件不支持阻止默认行为
|
||||||
_ = pkg.plugin.host.emit(plugin_models.SessionExplicitReset, **args)
|
_ = plugin_host.emit(plugin_models.SessionExplicitReset, **args)
|
||||||
|
|
||||||
pkg.utils.context.get_database_manager().explicit_close_session(self.name, self.create_timestamp)
|
context.get_database_manager().explicit_close_session(self.name, self.create_timestamp)
|
||||||
|
|
||||||
if expired:
|
if expired:
|
||||||
pkg.utils.context.get_database_manager().set_session_expired(self.name, self.create_timestamp)
|
context.get_database_manager().set_session_expired(self.name, self.create_timestamp)
|
||||||
|
|
||||||
if not persist: # 不要求保持default prompt
|
if not persist: # 不要求保持default prompt
|
||||||
self.default_prompt = self.get_default_prompt(use_prompt)
|
self.default_prompt = self.get_default_prompt(use_prompt)
|
||||||
@@ -461,11 +459,11 @@ class Session:
|
|||||||
|
|
||||||
# 将本session的数据库状态设置为on_going
|
# 将本session的数据库状态设置为on_going
|
||||||
def set_ongoing(self):
|
def set_ongoing(self):
|
||||||
pkg.utils.context.get_database_manager().set_session_ongoing(self.name, self.create_timestamp)
|
context.get_database_manager().set_session_ongoing(self.name, self.create_timestamp)
|
||||||
|
|
||||||
# 切换到上一个session
|
# 切换到上一个session
|
||||||
def last_session(self):
|
def last_session(self):
|
||||||
last_one = pkg.utils.context.get_database_manager().last_session(self.name, self.last_interact_timestamp)
|
last_one = context.get_database_manager().last_session(self.name, self.last_interact_timestamp)
|
||||||
if last_one is None:
|
if last_one is None:
|
||||||
return None
|
return None
|
||||||
else:
|
else:
|
||||||
@@ -486,7 +484,7 @@ class Session:
|
|||||||
|
|
||||||
# 切换到下一个session
|
# 切换到下一个session
|
||||||
def next_session(self):
|
def next_session(self):
|
||||||
next_one = pkg.utils.context.get_database_manager().next_session(self.name, self.last_interact_timestamp)
|
next_one = context.get_database_manager().next_session(self.name, self.last_interact_timestamp)
|
||||||
if next_one is None:
|
if next_one is None:
|
||||||
return None
|
return None
|
||||||
else:
|
else:
|
||||||
@@ -506,13 +504,13 @@ class Session:
|
|||||||
return self
|
return self
|
||||||
|
|
||||||
def list_history(self, capacity: int = 10, page: int = 0):
|
def list_history(self, capacity: int = 10, page: int = 0):
|
||||||
return pkg.utils.context.get_database_manager().list_history(self.name, capacity, page)
|
return context.get_database_manager().list_history(self.name, capacity, page)
|
||||||
|
|
||||||
def delete_history(self, index: int) -> bool:
|
def delete_history(self, index: int) -> bool:
|
||||||
return pkg.utils.context.get_database_manager().delete_history(self.name, index)
|
return context.get_database_manager().delete_history(self.name, index)
|
||||||
|
|
||||||
def delete_all_history(self) -> bool:
|
def delete_all_history(self) -> bool:
|
||||||
return pkg.utils.context.get_database_manager().delete_all_history(self.name)
|
return context.get_database_manager().delete_all_history(self.name)
|
||||||
|
|
||||||
def draw_image(self, prompt: str):
|
def draw_image(self, prompt: str):
|
||||||
return pkg.utils.context.get_openai_manager().request_image(prompt)
|
return context.get_openai_manager().request_image(prompt)
|
||||||
|
|||||||
@@ -10,13 +10,13 @@ import traceback
|
|||||||
import time
|
import time
|
||||||
import re
|
import re
|
||||||
|
|
||||||
import pkg.utils.updater as updater
|
from ..utils import updater as updater
|
||||||
import pkg.utils.context as context
|
from ..utils import network as network
|
||||||
import pkg.plugin.switch as switch
|
from ..utils import context as context
|
||||||
import pkg.plugin.settings as settings
|
from ..plugin import switch as switch
|
||||||
import pkg.qqbot.adapter as msadapter
|
from ..plugin import settings as settings
|
||||||
import pkg.utils.network as network
|
from ..qqbot import adapter as msadapter
|
||||||
import pkg.plugin.metadata as metadata
|
from ..plugin import metadata as metadata
|
||||||
|
|
||||||
from mirai import Mirai
|
from mirai import Mirai
|
||||||
import requests
|
import requests
|
||||||
@@ -147,6 +147,7 @@ def initialize_plugins():
|
|||||||
successfully_initialized_plugins.append(plugin['name'])
|
successfully_initialized_plugins.append(plugin['name'])
|
||||||
except:
|
except:
|
||||||
logging.error("插件{}初始化时发生错误: {}".format(plugin['name'], sys.exc_info()))
|
logging.error("插件{}初始化时发生错误: {}".format(plugin['name'], sys.exc_info()))
|
||||||
|
logging.debug(traceback.format_exc())
|
||||||
|
|
||||||
logging.info("以下插件已初始化: {}".format(", ".join(successfully_initialized_plugins)))
|
logging.info("以下插件已初始化: {}".format(", ".join(successfully_initialized_plugins)))
|
||||||
|
|
||||||
|
|||||||
@@ -1,7 +1,7 @@
|
|||||||
import logging
|
import logging
|
||||||
|
|
||||||
import pkg.plugin.host as host
|
from ..plugin import host
|
||||||
import pkg.utils.context
|
from ..utils import context
|
||||||
|
|
||||||
PersonMessageReceived = "person_message_received"
|
PersonMessageReceived = "person_message_received"
|
||||||
"""收到私聊消息时,在判断是否应该响应前触发
|
"""收到私聊消息时,在判断是否应该响应前触发
|
||||||
@@ -285,7 +285,7 @@ def register(name: str, description: str, version: str, author: str):
|
|||||||
cls.description = description
|
cls.description = description
|
||||||
cls.version = version
|
cls.version = version
|
||||||
cls.author = author
|
cls.author = author
|
||||||
cls.host = pkg.utils.context.get_plugin_host()
|
cls.host = context.get_plugin_host()
|
||||||
cls.enabled = True
|
cls.enabled = True
|
||||||
cls.path = host.__current_module_path__
|
cls.path = host.__current_module_path__
|
||||||
|
|
||||||
|
|||||||
@@ -1,9 +1,9 @@
|
|||||||
import json
|
import json
|
||||||
import os
|
import os
|
||||||
|
|
||||||
import pkg.plugin.host as host
|
|
||||||
import logging
|
import logging
|
||||||
|
|
||||||
|
from ..plugin import host
|
||||||
|
|
||||||
def wrapper_dict_from_runtime_context() -> dict:
|
def wrapper_dict_from_runtime_context() -> dict:
|
||||||
"""从变量中包装settings.json的数据字典"""
|
"""从变量中包装settings.json的数据字典"""
|
||||||
|
|||||||
@@ -3,7 +3,7 @@ import json
|
|||||||
import logging
|
import logging
|
||||||
import os
|
import os
|
||||||
|
|
||||||
import pkg.plugin.host as host
|
from ..plugin import host
|
||||||
|
|
||||||
|
|
||||||
def wrapper_dict_from_plugin_list() -> dict:
|
def wrapper_dict_from_plugin_list() -> dict:
|
||||||
|
|||||||
@@ -1,18 +1,18 @@
|
|||||||
import pkg.utils.context
|
from ..utils import context
|
||||||
|
|
||||||
|
|
||||||
def is_banned(launcher_type: str, launcher_id: int, sender_id: int) -> bool:
|
def is_banned(launcher_type: str, launcher_id: int, sender_id: int) -> bool:
|
||||||
if not pkg.utils.context.get_qqbot_manager().enable_banlist:
|
if not context.get_qqbot_manager().enable_banlist:
|
||||||
return False
|
return False
|
||||||
|
|
||||||
result = False
|
result = False
|
||||||
|
|
||||||
if launcher_type == 'group':
|
if launcher_type == 'group':
|
||||||
# 检查是否显式声明发起人QQ要被person忽略
|
# 检查是否显式声明发起人QQ要被person忽略
|
||||||
if sender_id in pkg.utils.context.get_qqbot_manager().ban_person:
|
if sender_id in context.get_qqbot_manager().ban_person:
|
||||||
result = True
|
result = True
|
||||||
else:
|
else:
|
||||||
for group_rule in pkg.utils.context.get_qqbot_manager().ban_group:
|
for group_rule in context.get_qqbot_manager().ban_group:
|
||||||
if type(group_rule) == int:
|
if type(group_rule) == int:
|
||||||
if group_rule == launcher_id: # 此群群号被禁用
|
if group_rule == launcher_id: # 此群群号被禁用
|
||||||
result = True
|
result = True
|
||||||
@@ -32,7 +32,7 @@ def is_banned(launcher_type: str, launcher_id: int, sender_id: int) -> bool:
|
|||||||
|
|
||||||
else:
|
else:
|
||||||
# ban_person, 与群规则相同
|
# ban_person, 与群规则相同
|
||||||
for person_rule in pkg.utils.context.get_qqbot_manager().ban_person:
|
for person_rule in context.get_qqbot_manager().ban_person:
|
||||||
if type(person_rule) == int:
|
if type(person_rule) == int:
|
||||||
if person_rule == launcher_id:
|
if person_rule == launcher_id:
|
||||||
result = True
|
result = True
|
||||||
|
|||||||
@@ -2,21 +2,21 @@
|
|||||||
import os
|
import os
|
||||||
import time
|
import time
|
||||||
import base64
|
import base64
|
||||||
|
import typing
|
||||||
|
|
||||||
import config
|
|
||||||
from mirai.models.message import MessageComponent, MessageChain, Image
|
from mirai.models.message import MessageComponent, MessageChain, Image
|
||||||
from mirai.models.message import ForwardMessageNode
|
from mirai.models.message import ForwardMessageNode
|
||||||
from mirai.models.base import MiraiBaseModel
|
from mirai.models.base import MiraiBaseModel
|
||||||
from typing import List
|
|
||||||
import pkg.utils.context as context
|
from ..utils import text2img
|
||||||
import pkg.utils.text2img as text2img
|
import config
|
||||||
|
|
||||||
|
|
||||||
class ForwardMessageDiaplay(MiraiBaseModel):
|
class ForwardMessageDiaplay(MiraiBaseModel):
|
||||||
title: str = "群聊的聊天记录"
|
title: str = "群聊的聊天记录"
|
||||||
brief: str = "[聊天记录]"
|
brief: str = "[聊天记录]"
|
||||||
source: str = "聊天记录"
|
source: str = "聊天记录"
|
||||||
preview: List[str] = []
|
preview: typing.List[str] = []
|
||||||
summary: str = "查看x条转发消息"
|
summary: str = "查看x条转发消息"
|
||||||
|
|
||||||
|
|
||||||
@@ -26,7 +26,7 @@ class Forward(MessageComponent):
|
|||||||
"""消息组件类型。"""
|
"""消息组件类型。"""
|
||||||
display: ForwardMessageDiaplay
|
display: ForwardMessageDiaplay
|
||||||
"""显示信息"""
|
"""显示信息"""
|
||||||
node_list: List[ForwardMessageNode]
|
node_list: typing.List[ForwardMessageNode]
|
||||||
"""转发消息节点列表。"""
|
"""转发消息节点列表。"""
|
||||||
def __init__(self, *args, **kwargs):
|
def __init__(self, *args, **kwargs):
|
||||||
if len(args) == 1:
|
if len(args) == 1:
|
||||||
|
|||||||
@@ -1,10 +1,7 @@
|
|||||||
import importlib
|
|
||||||
import inspect
|
|
||||||
import logging
|
import logging
|
||||||
import copy
|
import copy
|
||||||
import pkgutil
|
import pkgutil
|
||||||
import traceback
|
import traceback
|
||||||
import types
|
|
||||||
import json
|
import json
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -1,11 +1,12 @@
|
|||||||
from ..aamgr import AbstractCommandNode, Context
|
|
||||||
import logging
|
import logging
|
||||||
|
|
||||||
from mirai import Image
|
import mirai
|
||||||
|
|
||||||
|
from .. import aamgr
|
||||||
import config
|
import config
|
||||||
|
|
||||||
|
|
||||||
@AbstractCommandNode.register(
|
@aamgr.AbstractCommandNode.register(
|
||||||
parent=None,
|
parent=None,
|
||||||
name="draw",
|
name="draw",
|
||||||
description="使用DALL·E生成图片",
|
description="使用DALL·E生成图片",
|
||||||
@@ -13,9 +14,9 @@ import config
|
|||||||
aliases=[],
|
aliases=[],
|
||||||
privilege=1
|
privilege=1
|
||||||
)
|
)
|
||||||
class DrawCommand(AbstractCommandNode):
|
class DrawCommand(aamgr.AbstractCommandNode):
|
||||||
@classmethod
|
@classmethod
|
||||||
def process(cls, ctx: Context) -> tuple[bool, list]:
|
def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
|
||||||
import pkg.openai.session
|
import pkg.openai.session
|
||||||
|
|
||||||
reply = []
|
reply = []
|
||||||
@@ -28,7 +29,7 @@ class DrawCommand(AbstractCommandNode):
|
|||||||
res = session.draw_image(" ".join(ctx.params))
|
res = session.draw_image(" ".join(ctx.params))
|
||||||
|
|
||||||
logging.debug("draw_image result:{}".format(res))
|
logging.debug("draw_image result:{}".format(res))
|
||||||
reply = [Image(url=res['data'][0]['url'])]
|
reply = [mirai.Image(url=res['data'][0]['url'])]
|
||||||
if not (hasattr(config, 'include_image_description')
|
if not (hasattr(config, 'include_image_description')
|
||||||
and not config.include_image_description):
|
and not config.include_image_description):
|
||||||
reply.append(" ".join(ctx.params))
|
reply.append(" ".join(ctx.params))
|
||||||
|
|||||||
@@ -1,10 +1,9 @@
|
|||||||
from ..aamgr import AbstractCommandNode, Context
|
|
||||||
import logging
|
import logging
|
||||||
|
|
||||||
import json
|
import json
|
||||||
|
|
||||||
|
from .. import aamgr
|
||||||
|
|
||||||
@AbstractCommandNode.register(
|
@aamgr.AbstractCommandNode.register(
|
||||||
parent=None,
|
parent=None,
|
||||||
name="func",
|
name="func",
|
||||||
description="管理内容函数",
|
description="管理内容函数",
|
||||||
@@ -12,9 +11,9 @@ import json
|
|||||||
aliases=[],
|
aliases=[],
|
||||||
privilege=1
|
privilege=1
|
||||||
)
|
)
|
||||||
class FuncCommand(AbstractCommandNode):
|
class FuncCommand(aamgr.AbstractCommandNode):
|
||||||
@classmethod
|
@classmethod
|
||||||
def process(cls, ctx: Context) -> tuple[bool, list]:
|
def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
|
||||||
from pkg.plugin.models import host
|
from pkg.plugin.models import host
|
||||||
|
|
||||||
reply = []
|
reply = []
|
||||||
|
|||||||
@@ -1,12 +1,9 @@
|
|||||||
from ..aamgr import AbstractCommandNode, Context
|
from ....plugin import host as plugin_host
|
||||||
|
from ....utils import updater
|
||||||
import os
|
from .. import aamgr
|
||||||
|
|
||||||
import pkg.plugin.host as plugin_host
|
|
||||||
import pkg.utils.updater as updater
|
|
||||||
|
|
||||||
|
|
||||||
@AbstractCommandNode.register(
|
@aamgr.AbstractCommandNode.register(
|
||||||
parent=None,
|
parent=None,
|
||||||
name="plugin",
|
name="plugin",
|
||||||
description="插件管理",
|
description="插件管理",
|
||||||
@@ -14,9 +11,9 @@ import pkg.utils.updater as updater
|
|||||||
aliases=[],
|
aliases=[],
|
||||||
privilege=1
|
privilege=1
|
||||||
)
|
)
|
||||||
class PluginCommand(AbstractCommandNode):
|
class PluginCommand(aamgr.AbstractCommandNode):
|
||||||
@classmethod
|
@classmethod
|
||||||
def process(cls, ctx: Context) -> tuple[bool, list]:
|
def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
|
||||||
reply = []
|
reply = []
|
||||||
plugin_list = plugin_host.__plugins__
|
plugin_list = plugin_host.__plugins__
|
||||||
if len(ctx.params) == 0:
|
if len(ctx.params) == 0:
|
||||||
@@ -48,7 +45,7 @@ class PluginCommand(AbstractCommandNode):
|
|||||||
return False, []
|
return False, []
|
||||||
|
|
||||||
|
|
||||||
@AbstractCommandNode.register(
|
@aamgr.AbstractCommandNode.register(
|
||||||
parent=PluginCommand,
|
parent=PluginCommand,
|
||||||
name="get",
|
name="get",
|
||||||
description="安装插件",
|
description="安装插件",
|
||||||
@@ -56,9 +53,9 @@ class PluginCommand(AbstractCommandNode):
|
|||||||
aliases=[],
|
aliases=[],
|
||||||
privilege=2
|
privilege=2
|
||||||
)
|
)
|
||||||
class PluginGetCommand(AbstractCommandNode):
|
class PluginGetCommand(aamgr.AbstractCommandNode):
|
||||||
@classmethod
|
@classmethod
|
||||||
def process(cls, ctx: Context) -> tuple[bool, list]:
|
def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
|
||||||
import threading
|
import threading
|
||||||
import logging
|
import logging
|
||||||
import pkg.utils.context
|
import pkg.utils.context
|
||||||
@@ -81,7 +78,7 @@ class PluginGetCommand(AbstractCommandNode):
|
|||||||
return True, reply
|
return True, reply
|
||||||
|
|
||||||
|
|
||||||
@AbstractCommandNode.register(
|
@aamgr.AbstractCommandNode.register(
|
||||||
parent=PluginCommand,
|
parent=PluginCommand,
|
||||||
name="update",
|
name="update",
|
||||||
description="更新指定插件或全部插件",
|
description="更新指定插件或全部插件",
|
||||||
@@ -89,9 +86,9 @@ class PluginGetCommand(AbstractCommandNode):
|
|||||||
aliases=[],
|
aliases=[],
|
||||||
privilege=2
|
privilege=2
|
||||||
)
|
)
|
||||||
class PluginUpdateCommand(AbstractCommandNode):
|
class PluginUpdateCommand(aamgr.AbstractCommandNode):
|
||||||
@classmethod
|
@classmethod
|
||||||
def process(cls, ctx: Context) -> tuple[bool, list]:
|
def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
|
||||||
import threading
|
import threading
|
||||||
import logging
|
import logging
|
||||||
plugin_list = plugin_host.__plugins__
|
plugin_list = plugin_host.__plugins__
|
||||||
@@ -130,7 +127,7 @@ class PluginUpdateCommand(AbstractCommandNode):
|
|||||||
return True, reply
|
return True, reply
|
||||||
|
|
||||||
|
|
||||||
@AbstractCommandNode.register(
|
@aamgr.AbstractCommandNode.register(
|
||||||
parent=PluginCommand,
|
parent=PluginCommand,
|
||||||
name="del",
|
name="del",
|
||||||
description="删除插件",
|
description="删除插件",
|
||||||
@@ -138,9 +135,9 @@ class PluginUpdateCommand(AbstractCommandNode):
|
|||||||
aliases=[],
|
aliases=[],
|
||||||
privilege=2
|
privilege=2
|
||||||
)
|
)
|
||||||
class PluginDelCommand(AbstractCommandNode):
|
class PluginDelCommand(aamgr.AbstractCommandNode):
|
||||||
@classmethod
|
@classmethod
|
||||||
def process(cls, ctx: Context) -> tuple[bool, list]:
|
def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
|
||||||
plugin_list = plugin_host.__plugins__
|
plugin_list = plugin_host.__plugins__
|
||||||
reply = []
|
reply = []
|
||||||
|
|
||||||
@@ -157,7 +154,7 @@ class PluginDelCommand(AbstractCommandNode):
|
|||||||
return True, reply
|
return True, reply
|
||||||
|
|
||||||
|
|
||||||
@AbstractCommandNode.register(
|
@aamgr.AbstractCommandNode.register(
|
||||||
parent=PluginCommand,
|
parent=PluginCommand,
|
||||||
name="on",
|
name="on",
|
||||||
description="启用指定插件",
|
description="启用指定插件",
|
||||||
@@ -165,7 +162,7 @@ class PluginDelCommand(AbstractCommandNode):
|
|||||||
aliases=[],
|
aliases=[],
|
||||||
privilege=2
|
privilege=2
|
||||||
)
|
)
|
||||||
@AbstractCommandNode.register(
|
@aamgr.AbstractCommandNode.register(
|
||||||
parent=PluginCommand,
|
parent=PluginCommand,
|
||||||
name="off",
|
name="off",
|
||||||
description="禁用指定插件",
|
description="禁用指定插件",
|
||||||
@@ -173,9 +170,9 @@ class PluginDelCommand(AbstractCommandNode):
|
|||||||
aliases=[],
|
aliases=[],
|
||||||
privilege=2
|
privilege=2
|
||||||
)
|
)
|
||||||
class PluginOnOffCommand(AbstractCommandNode):
|
class PluginOnOffCommand(aamgr.AbstractCommandNode):
|
||||||
@classmethod
|
@classmethod
|
||||||
def process(cls, ctx: Context) -> tuple[bool, list]:
|
def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
|
||||||
import pkg.plugin.switch as plugin_switch
|
import pkg.plugin.switch as plugin_switch
|
||||||
|
|
||||||
plugin_list = plugin_host.__plugins__
|
plugin_list = plugin_host.__plugins__
|
||||||
|
|||||||
@@ -1,7 +1,6 @@
|
|||||||
from ..aamgr import AbstractCommandNode, Context
|
from .. import aamgr
|
||||||
|
|
||||||
|
@aamgr.AbstractCommandNode.register(
|
||||||
@AbstractCommandNode.register(
|
|
||||||
parent=None,
|
parent=None,
|
||||||
name="default",
|
name="default",
|
||||||
description="操作情景预设",
|
description="操作情景预设",
|
||||||
@@ -9,9 +8,9 @@ from ..aamgr import AbstractCommandNode, Context
|
|||||||
aliases=[],
|
aliases=[],
|
||||||
privilege=1
|
privilege=1
|
||||||
)
|
)
|
||||||
class DefaultCommand(AbstractCommandNode):
|
class DefaultCommand(aamgr.AbstractCommandNode):
|
||||||
@classmethod
|
@classmethod
|
||||||
def process(cls, ctx: Context) -> tuple[bool, list]:
|
def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
|
||||||
import pkg.openai.session
|
import pkg.openai.session
|
||||||
session_name = ctx.session_name
|
session_name = ctx.session_name
|
||||||
params = ctx.params
|
params = ctx.params
|
||||||
@@ -45,7 +44,7 @@ class DefaultCommand(AbstractCommandNode):
|
|||||||
return True, reply
|
return True, reply
|
||||||
|
|
||||||
|
|
||||||
@AbstractCommandNode.register(
|
@aamgr.AbstractCommandNode.register(
|
||||||
parent=DefaultCommand,
|
parent=DefaultCommand,
|
||||||
name="set",
|
name="set",
|
||||||
description="设置默认情景预设",
|
description="设置默认情景预设",
|
||||||
@@ -53,9 +52,9 @@ class DefaultCommand(AbstractCommandNode):
|
|||||||
aliases=[],
|
aliases=[],
|
||||||
privilege=2
|
privilege=2
|
||||||
)
|
)
|
||||||
class DefaultSetCommand(AbstractCommandNode):
|
class DefaultSetCommand(aamgr.AbstractCommandNode):
|
||||||
@classmethod
|
@classmethod
|
||||||
def process(cls, ctx: Context) -> tuple[bool, list]:
|
def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
|
||||||
reply = []
|
reply = []
|
||||||
|
|
||||||
if len(ctx.crt_params) == 0:
|
if len(ctx.crt_params) == 0:
|
||||||
|
|||||||
@@ -1,8 +1,7 @@
|
|||||||
from ..aamgr import AbstractCommandNode, Context
|
from .. import aamgr
|
||||||
import datetime
|
|
||||||
|
|
||||||
|
|
||||||
@AbstractCommandNode.register(
|
@aamgr.AbstractCommandNode.register(
|
||||||
parent=None,
|
parent=None,
|
||||||
name="del",
|
name="del",
|
||||||
description="删除当前会话的历史记录",
|
description="删除当前会话的历史记录",
|
||||||
@@ -10,9 +9,9 @@ import datetime
|
|||||||
aliases=[],
|
aliases=[],
|
||||||
privilege=1
|
privilege=1
|
||||||
)
|
)
|
||||||
class DelCommand(AbstractCommandNode):
|
class DelCommand(aamgr.AbstractCommandNode):
|
||||||
@classmethod
|
@classmethod
|
||||||
def process(cls, ctx: Context) -> tuple[bool, list]:
|
def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
|
||||||
import pkg.openai.session
|
import pkg.openai.session
|
||||||
session_name = ctx.session_name
|
session_name = ctx.session_name
|
||||||
params = ctx.params
|
params = ctx.params
|
||||||
@@ -33,7 +32,7 @@ class DelCommand(AbstractCommandNode):
|
|||||||
return True, reply
|
return True, reply
|
||||||
|
|
||||||
|
|
||||||
@AbstractCommandNode.register(
|
@aamgr.AbstractCommandNode.register(
|
||||||
parent=DelCommand,
|
parent=DelCommand,
|
||||||
name="all",
|
name="all",
|
||||||
description="删除当前会话的全部历史记录",
|
description="删除当前会话的全部历史记录",
|
||||||
@@ -41,9 +40,9 @@ class DelCommand(AbstractCommandNode):
|
|||||||
aliases=[],
|
aliases=[],
|
||||||
privilege=1
|
privilege=1
|
||||||
)
|
)
|
||||||
class DelAllCommand(AbstractCommandNode):
|
class DelAllCommand(aamgr.AbstractCommandNode):
|
||||||
@classmethod
|
@classmethod
|
||||||
def process(cls, ctx: Context) -> tuple[bool, list]:
|
def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
|
||||||
import pkg.openai.session
|
import pkg.openai.session
|
||||||
session_name = ctx.session_name
|
session_name = ctx.session_name
|
||||||
reply = []
|
reply = []
|
||||||
|
|||||||
@@ -1,7 +1,7 @@
|
|||||||
from ..aamgr import AbstractCommandNode, Context
|
from .. import aamgr
|
||||||
|
|
||||||
|
|
||||||
@AbstractCommandNode.register(
|
@aamgr.AbstractCommandNode.register(
|
||||||
parent=None,
|
parent=None,
|
||||||
name="delhst",
|
name="delhst",
|
||||||
description="删除指定会话的所有历史记录",
|
description="删除指定会话的所有历史记录",
|
||||||
@@ -9,9 +9,9 @@ from ..aamgr import AbstractCommandNode, Context
|
|||||||
aliases=[],
|
aliases=[],
|
||||||
privilege=2
|
privilege=2
|
||||||
)
|
)
|
||||||
class DelHistoryCommand(AbstractCommandNode):
|
class DelHistoryCommand(aamgr.AbstractCommandNode):
|
||||||
@classmethod
|
@classmethod
|
||||||
def process(cls, ctx: Context) -> tuple[bool, list]:
|
def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
|
||||||
import pkg.openai.session
|
import pkg.openai.session
|
||||||
import pkg.utils.context
|
import pkg.utils.context
|
||||||
params = ctx.params
|
params = ctx.params
|
||||||
@@ -31,7 +31,7 @@ class DelHistoryCommand(AbstractCommandNode):
|
|||||||
return True, reply
|
return True, reply
|
||||||
|
|
||||||
|
|
||||||
@AbstractCommandNode.register(
|
@aamgr.AbstractCommandNode.register(
|
||||||
parent=DelHistoryCommand,
|
parent=DelHistoryCommand,
|
||||||
name="all",
|
name="all",
|
||||||
description="删除所有会话的全部历史记录",
|
description="删除所有会话的全部历史记录",
|
||||||
@@ -39,9 +39,9 @@ class DelHistoryCommand(AbstractCommandNode):
|
|||||||
aliases=[],
|
aliases=[],
|
||||||
privilege=2
|
privilege=2
|
||||||
)
|
)
|
||||||
class DelAllHistoryCommand(AbstractCommandNode):
|
class DelAllHistoryCommand(aamgr.AbstractCommandNode):
|
||||||
@classmethod
|
@classmethod
|
||||||
def process(cls, ctx: Context) -> tuple[bool, list]:
|
def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
|
||||||
import pkg.utils.context
|
import pkg.utils.context
|
||||||
reply = []
|
reply = []
|
||||||
pkg.utils.context.get_database_manager().delete_all_session_history()
|
pkg.utils.context.get_database_manager().delete_all_session_history()
|
||||||
|
|||||||
@@ -1,8 +1,9 @@
|
|||||||
from ..aamgr import AbstractCommandNode, Context
|
|
||||||
import datetime
|
import datetime
|
||||||
|
|
||||||
|
from .. import aamgr
|
||||||
|
|
||||||
@AbstractCommandNode.register(
|
|
||||||
|
@aamgr.AbstractCommandNode.register(
|
||||||
parent=None,
|
parent=None,
|
||||||
name="last",
|
name="last",
|
||||||
description="切换前一次对话",
|
description="切换前一次对话",
|
||||||
@@ -10,9 +11,9 @@ import datetime
|
|||||||
aliases=[],
|
aliases=[],
|
||||||
privilege=1
|
privilege=1
|
||||||
)
|
)
|
||||||
class LastCommand(AbstractCommandNode):
|
class LastCommand(aamgr.AbstractCommandNode):
|
||||||
@classmethod
|
@classmethod
|
||||||
def process(cls, ctx: Context) -> tuple[bool, list]:
|
def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
|
||||||
import pkg.openai.session
|
import pkg.openai.session
|
||||||
session_name = ctx.session_name
|
session_name = ctx.session_name
|
||||||
|
|
||||||
|
|||||||
@@ -1,9 +1,10 @@
|
|||||||
from ..aamgr import AbstractCommandNode, Context
|
|
||||||
import datetime
|
import datetime
|
||||||
import json
|
import json
|
||||||
|
|
||||||
|
from .. import aamgr
|
||||||
|
|
||||||
@AbstractCommandNode.register(
|
|
||||||
|
@aamgr.AbstractCommandNode.register(
|
||||||
parent=None,
|
parent=None,
|
||||||
name='list',
|
name='list',
|
||||||
description='列出当前会话的所有历史记录',
|
description='列出当前会话的所有历史记录',
|
||||||
@@ -11,9 +12,9 @@ import json
|
|||||||
aliases=[],
|
aliases=[],
|
||||||
privilege=1
|
privilege=1
|
||||||
)
|
)
|
||||||
class ListCommand(AbstractCommandNode):
|
class ListCommand(aamgr.AbstractCommandNode):
|
||||||
@classmethod
|
@classmethod
|
||||||
def process(cls, ctx: Context) -> tuple[bool, list]:
|
def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
|
||||||
import pkg.openai.session
|
import pkg.openai.session
|
||||||
session_name = ctx.session_name
|
session_name = ctx.session_name
|
||||||
params = ctx.params
|
params = ctx.params
|
||||||
|
|||||||
@@ -1,8 +1,9 @@
|
|||||||
from ..aamgr import AbstractCommandNode, Context
|
|
||||||
import datetime
|
import datetime
|
||||||
|
|
||||||
|
from .. import aamgr
|
||||||
|
|
||||||
@AbstractCommandNode.register(
|
|
||||||
|
@aamgr.AbstractCommandNode.register(
|
||||||
parent=None,
|
parent=None,
|
||||||
name="next",
|
name="next",
|
||||||
description="切换后一次对话",
|
description="切换后一次对话",
|
||||||
@@ -10,9 +11,9 @@ import datetime
|
|||||||
aliases=[],
|
aliases=[],
|
||||||
privilege=1
|
privilege=1
|
||||||
)
|
)
|
||||||
class NextCommand(AbstractCommandNode):
|
class NextCommand(aamgr.AbstractCommandNode):
|
||||||
@classmethod
|
@classmethod
|
||||||
def process(cls, ctx: Context) -> tuple[bool, list]:
|
def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
|
||||||
import pkg.openai.session
|
import pkg.openai.session
|
||||||
session_name = ctx.session_name
|
session_name = ctx.session_name
|
||||||
reply = []
|
reply = []
|
||||||
|
|||||||
@@ -1,8 +1,7 @@
|
|||||||
from ..aamgr import AbstractCommandNode, Context
|
from .. import aamgr
|
||||||
import datetime
|
|
||||||
|
|
||||||
|
|
||||||
@AbstractCommandNode.register(
|
@aamgr.AbstractCommandNode.register(
|
||||||
parent=None,
|
parent=None,
|
||||||
name="prompt",
|
name="prompt",
|
||||||
description="获取当前会话的前文",
|
description="获取当前会话的前文",
|
||||||
@@ -10,9 +9,9 @@ import datetime
|
|||||||
aliases=[],
|
aliases=[],
|
||||||
privilege=1
|
privilege=1
|
||||||
)
|
)
|
||||||
class PromptCommand(AbstractCommandNode):
|
class PromptCommand(aamgr.AbstractCommandNode):
|
||||||
@classmethod
|
@classmethod
|
||||||
def process(cls, ctx: Context) -> tuple[bool, list]:
|
def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
|
||||||
import pkg.openai.session
|
import pkg.openai.session
|
||||||
session_name = ctx.session_name
|
session_name = ctx.session_name
|
||||||
params = ctx.params
|
params = ctx.params
|
||||||
|
|||||||
@@ -1,8 +1,7 @@
|
|||||||
from ..aamgr import AbstractCommandNode, Context
|
from .. import aamgr
|
||||||
import datetime
|
|
||||||
|
|
||||||
|
|
||||||
@AbstractCommandNode.register(
|
@aamgr.AbstractCommandNode.register(
|
||||||
parent=None,
|
parent=None,
|
||||||
name="resend",
|
name="resend",
|
||||||
description="重新获取上一次问题的回复",
|
description="重新获取上一次问题的回复",
|
||||||
@@ -10,20 +9,22 @@ import datetime
|
|||||||
aliases=[],
|
aliases=[],
|
||||||
privilege=1
|
privilege=1
|
||||||
)
|
)
|
||||||
class ResendCommand(AbstractCommandNode):
|
class ResendCommand(aamgr.AbstractCommandNode):
|
||||||
@classmethod
|
@classmethod
|
||||||
def process(cls, ctx: Context) -> tuple[bool, list]:
|
def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
|
||||||
import pkg.openai.session
|
from ....openai import session as openai_session
|
||||||
|
from ....utils import context
|
||||||
|
from ....qqbot import message
|
||||||
import config
|
import config
|
||||||
session_name = ctx.session_name
|
session_name = ctx.session_name
|
||||||
reply = []
|
reply = []
|
||||||
|
|
||||||
session = pkg.openai.session.get_session(session_name)
|
session = openai_session.get_session(session_name)
|
||||||
to_send = session.undo()
|
to_send = session.undo()
|
||||||
|
|
||||||
mgr = pkg.utils.context.get_qqbot_manager()
|
mgr = context.get_qqbot_manager()
|
||||||
|
|
||||||
reply = pkg.qqbot.message.process_normal_message(to_send, mgr, config,
|
reply = message.process_normal_message(to_send, mgr, config,
|
||||||
ctx.launcher_type, ctx.launcher_id,
|
ctx.launcher_type, ctx.launcher_id,
|
||||||
ctx.sender_id)
|
ctx.sender_id)
|
||||||
|
|
||||||
|
|||||||
@@ -1,11 +1,11 @@
|
|||||||
from ..aamgr import AbstractCommandNode, Context
|
|
||||||
import tips as tips_custom
|
import tips as tips_custom
|
||||||
|
|
||||||
import pkg.openai.session
|
from .. import aamgr
|
||||||
import pkg.utils.context
|
from ....openai import session
|
||||||
|
from ....utils import context
|
||||||
|
|
||||||
|
|
||||||
@AbstractCommandNode.register(
|
@aamgr.AbstractCommandNode.register(
|
||||||
parent=None,
|
parent=None,
|
||||||
name='reset',
|
name='reset',
|
||||||
description='重置当前会话',
|
description='重置当前会话',
|
||||||
@@ -13,21 +13,21 @@ import pkg.utils.context
|
|||||||
aliases=[],
|
aliases=[],
|
||||||
privilege=1
|
privilege=1
|
||||||
)
|
)
|
||||||
class ResetCommand(AbstractCommandNode):
|
class ResetCommand(aamgr.AbstractCommandNode):
|
||||||
@classmethod
|
@classmethod
|
||||||
def process(cls, ctx: Context) -> tuple[bool, list]:
|
def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
|
||||||
params = ctx.params
|
params = ctx.params
|
||||||
session_name = ctx.session_name
|
session_name = ctx.session_name
|
||||||
|
|
||||||
reply = ""
|
reply = ""
|
||||||
|
|
||||||
if len(params) == 0:
|
if len(params) == 0:
|
||||||
pkg.openai.session.get_session(session_name).reset(explicit=True)
|
session.get_session(session_name).reset(explicit=True)
|
||||||
reply = [tips_custom.command_reset_message]
|
reply = [tips_custom.command_reset_message]
|
||||||
else:
|
else:
|
||||||
try:
|
try:
|
||||||
import pkg.openai.dprompt as dprompt
|
import pkg.openai.dprompt as dprompt
|
||||||
pkg.openai.session.get_session(session_name).reset(explicit=True, use_prompt=params[0])
|
session.get_session(session_name).reset(explicit=True, use_prompt=params[0])
|
||||||
reply = [tips_custom.command_reset_name_message+"{}".format(dprompt.mode_inst().get_full_name(params[0]))]
|
reply = [tips_custom.command_reset_name_message+"{}".format(dprompt.mode_inst().get_full_name(params[0]))]
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
reply = ["[bot]会话重置失败:{}".format(e)]
|
reply = ["[bot]会话重置失败:{}".format(e)]
|
||||||
|
|||||||
@@ -1,6 +1,7 @@
|
|||||||
from ..aamgr import AbstractCommandNode, Context
|
|
||||||
import json
|
import json
|
||||||
|
|
||||||
|
from .. import aamgr
|
||||||
|
|
||||||
|
|
||||||
def config_operation(cmd, params):
|
def config_operation(cmd, params):
|
||||||
reply = []
|
reply = []
|
||||||
@@ -85,7 +86,7 @@ def config_operation(cmd, params):
|
|||||||
return reply
|
return reply
|
||||||
|
|
||||||
|
|
||||||
@AbstractCommandNode.register(
|
@aamgr.AbstractCommandNode.register(
|
||||||
parent=None,
|
parent=None,
|
||||||
name="cfg",
|
name="cfg",
|
||||||
description="配置项管理",
|
description="配置项管理",
|
||||||
@@ -93,8 +94,8 @@ def config_operation(cmd, params):
|
|||||||
aliases=[],
|
aliases=[],
|
||||||
privilege=2
|
privilege=2
|
||||||
)
|
)
|
||||||
class CfgCommand(AbstractCommandNode):
|
class CfgCommand(aamgr.AbstractCommandNode):
|
||||||
@classmethod
|
@classmethod
|
||||||
def process(cls, ctx: Context) -> tuple[bool, list]:
|
def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
|
||||||
return True, config_operation(ctx.command, ctx.params)
|
return True, config_operation(ctx.command, ctx.params)
|
||||||
|
|
||||||
@@ -1,7 +1,7 @@
|
|||||||
from ..aamgr import AbstractCommandNode, Context, __command_list__
|
from .. import aamgr
|
||||||
|
|
||||||
|
|
||||||
@AbstractCommandNode.register(
|
@aamgr.AbstractCommandNode.register(
|
||||||
parent=None,
|
parent=None,
|
||||||
name="cmd",
|
name="cmd",
|
||||||
description="显示指令列表",
|
description="显示指令列表",
|
||||||
@@ -9,10 +9,10 @@ from ..aamgr import AbstractCommandNode, Context, __command_list__
|
|||||||
aliases=[],
|
aliases=[],
|
||||||
privilege=1
|
privilege=1
|
||||||
)
|
)
|
||||||
class CmdCommand(AbstractCommandNode):
|
class CmdCommand(aamgr.AbstractCommandNode):
|
||||||
@classmethod
|
@classmethod
|
||||||
def process(cls, ctx: Context) -> tuple[bool, list]:
|
def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
|
||||||
command_list = __command_list__
|
command_list = aamgr.__command_list__
|
||||||
|
|
||||||
reply = []
|
reply = []
|
||||||
|
|
||||||
|
|||||||
@@ -1,7 +1,7 @@
|
|||||||
from ..aamgr import AbstractCommandNode, Context
|
from .. import aamgr
|
||||||
|
|
||||||
|
|
||||||
@AbstractCommandNode.register(
|
@aamgr.AbstractCommandNode.register(
|
||||||
parent=None,
|
parent=None,
|
||||||
name="help",
|
name="help",
|
||||||
description="显示自定义的帮助信息",
|
description="显示自定义的帮助信息",
|
||||||
@@ -9,9 +9,9 @@ from ..aamgr import AbstractCommandNode, Context
|
|||||||
aliases=[],
|
aliases=[],
|
||||||
privilege=1
|
privilege=1
|
||||||
)
|
)
|
||||||
class HelpCommand(AbstractCommandNode):
|
class HelpCommand(aamgr.AbstractCommandNode):
|
||||||
@classmethod
|
@classmethod
|
||||||
def process(cls, ctx: Context) -> tuple[bool, list]:
|
def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
|
||||||
import tips
|
import tips
|
||||||
reply = ["[bot] "+tips.help_message + "\n请输入 !cmd 查看指令列表"]
|
reply = ["[bot] "+tips.help_message + "\n请输入 !cmd 查看指令列表"]
|
||||||
|
|
||||||
|
|||||||
@@ -1,7 +1,9 @@
|
|||||||
from ..aamgr import AbstractCommandNode, Context
|
|
||||||
import threading
|
import threading
|
||||||
|
|
||||||
@AbstractCommandNode.register(
|
from .. import aamgr
|
||||||
|
|
||||||
|
|
||||||
|
@aamgr.AbstractCommandNode.register(
|
||||||
parent=None,
|
parent=None,
|
||||||
name="reload",
|
name="reload",
|
||||||
description="执行热重载",
|
description="执行热重载",
|
||||||
@@ -9,9 +11,9 @@ import threading
|
|||||||
aliases=[],
|
aliases=[],
|
||||||
privilege=2
|
privilege=2
|
||||||
)
|
)
|
||||||
class ReloadCommand(AbstractCommandNode):
|
class ReloadCommand(aamgr.AbstractCommandNode):
|
||||||
@classmethod
|
@classmethod
|
||||||
def process(cls, ctx: Context) -> tuple[bool, list]:
|
def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
|
||||||
reply = []
|
reply = []
|
||||||
|
|
||||||
import pkg.utils.reloader
|
import pkg.utils.reloader
|
||||||
|
|||||||
@@ -1,9 +1,10 @@
|
|||||||
from ..aamgr import AbstractCommandNode, Context
|
|
||||||
import threading
|
import threading
|
||||||
import traceback
|
import traceback
|
||||||
|
|
||||||
|
from .. import aamgr
|
||||||
|
|
||||||
@AbstractCommandNode.register(
|
|
||||||
|
@aamgr.AbstractCommandNode.register(
|
||||||
parent=None,
|
parent=None,
|
||||||
name="update",
|
name="update",
|
||||||
description="更新程序",
|
description="更新程序",
|
||||||
@@ -11,9 +12,9 @@ import traceback
|
|||||||
aliases=[],
|
aliases=[],
|
||||||
privilege=2
|
privilege=2
|
||||||
)
|
)
|
||||||
class UpdateCommand(AbstractCommandNode):
|
class UpdateCommand(aamgr.AbstractCommandNode):
|
||||||
@classmethod
|
@classmethod
|
||||||
def process(cls, ctx: Context) -> tuple[bool, list]:
|
def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
|
||||||
reply = []
|
reply = []
|
||||||
import pkg.utils.updater
|
import pkg.utils.updater
|
||||||
import pkg.utils.reloader
|
import pkg.utils.reloader
|
||||||
@@ -22,8 +23,7 @@ class UpdateCommand(AbstractCommandNode):
|
|||||||
def update_task():
|
def update_task():
|
||||||
try:
|
try:
|
||||||
if pkg.utils.updater.update_all():
|
if pkg.utils.updater.update_all():
|
||||||
pkg.utils.reloader.reload_all(notify=False)
|
pkg.utils.context.get_qqbot_manager().notify_admin("更新完成, 请手动重启程序。")
|
||||||
pkg.utils.context.get_qqbot_manager().notify_admin("更新完成")
|
|
||||||
else:
|
else:
|
||||||
pkg.utils.context.get_qqbot_manager().notify_admin("无新版本")
|
pkg.utils.context.get_qqbot_manager().notify_admin("无新版本")
|
||||||
except Exception as e0:
|
except Exception as e0:
|
||||||
|
|||||||
@@ -1,8 +1,7 @@
|
|||||||
from ..aamgr import AbstractCommandNode, Context
|
from .. import aamgr
|
||||||
import logging
|
|
||||||
|
|
||||||
|
|
||||||
@AbstractCommandNode.register(
|
@aamgr.AbstractCommandNode.register(
|
||||||
parent=None,
|
parent=None,
|
||||||
name="usage",
|
name="usage",
|
||||||
description="获取使用情况",
|
description="获取使用情况",
|
||||||
@@ -10,9 +9,9 @@ import logging
|
|||||||
aliases=[],
|
aliases=[],
|
||||||
privilege=1
|
privilege=1
|
||||||
)
|
)
|
||||||
class UsageCommand(AbstractCommandNode):
|
class UsageCommand(aamgr.AbstractCommandNode):
|
||||||
@classmethod
|
@classmethod
|
||||||
def process(cls, ctx: Context) -> tuple[bool, list]:
|
def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
|
||||||
import config
|
import config
|
||||||
import pkg.utils.credit as credit
|
import pkg.utils.credit as credit
|
||||||
import pkg.utils.context
|
import pkg.utils.context
|
||||||
|
|||||||
@@ -1,7 +1,7 @@
|
|||||||
from ..aamgr import AbstractCommandNode, Context
|
from .. import aamgr
|
||||||
|
|
||||||
|
|
||||||
@AbstractCommandNode.register(
|
@aamgr.AbstractCommandNode.register(
|
||||||
parent=None,
|
parent=None,
|
||||||
name="version",
|
name="version",
|
||||||
description="查看版本信息",
|
description="查看版本信息",
|
||||||
@@ -9,9 +9,9 @@ from ..aamgr import AbstractCommandNode, Context
|
|||||||
aliases=[],
|
aliases=[],
|
||||||
privilege=1
|
privilege=1
|
||||||
)
|
)
|
||||||
class VersionCommand(AbstractCommandNode):
|
class VersionCommand(aamgr.AbstractCommandNode):
|
||||||
@classmethod
|
@classmethod
|
||||||
def process(cls, ctx: Context) -> tuple[bool, list]:
|
def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
|
||||||
reply = []
|
reply = []
|
||||||
import pkg.utils.updater
|
import pkg.utils.updater
|
||||||
|
|
||||||
|
|||||||
@@ -1,23 +1,7 @@
|
|||||||
# 指令处理模块
|
# 指令处理模块
|
||||||
import logging
|
import logging
|
||||||
import json
|
|
||||||
import datetime
|
|
||||||
import os
|
|
||||||
import threading
|
|
||||||
import traceback
|
|
||||||
|
|
||||||
import pkg.openai.session
|
|
||||||
import pkg.openai.manager
|
|
||||||
import pkg.utils.reloader
|
|
||||||
import pkg.utils.updater
|
|
||||||
import pkg.utils.context
|
|
||||||
import pkg.qqbot.message
|
|
||||||
import pkg.utils.credit as credit
|
|
||||||
# import pkg.qqbot.cmds.model as cmdmodel
|
|
||||||
import pkg.qqbot.cmds.aamgr as cmdmgr
|
|
||||||
|
|
||||||
from mirai import Image
|
|
||||||
|
|
||||||
|
from ..qqbot.cmds import aamgr as cmdmgr
|
||||||
|
|
||||||
|
|
||||||
def process_command(session_name: str, text_message: str, mgr, config,
|
def process_command(session_name: str, text_message: str, mgr, config,
|
||||||
|
|||||||
@@ -1,32 +1,25 @@
|
|||||||
import asyncio
|
|
||||||
import json
|
import json
|
||||||
import os
|
import os
|
||||||
import threading
|
import logging
|
||||||
|
|
||||||
|
|
||||||
from mirai import At, GroupMessage, MessageEvent, Mirai, StrangerMessage, WebSocketAdapter, HTTPAdapter, \
|
from mirai import At, GroupMessage, MessageEvent, Mirai, StrangerMessage, WebSocketAdapter, HTTPAdapter, \
|
||||||
FriendMessage, Image, MessageChain, Plain
|
FriendMessage, Image, MessageChain, Plain
|
||||||
from func_timeout import func_set_timeout
|
import func_timeout
|
||||||
|
|
||||||
import pkg.openai.session
|
from ..openai import session as openai_session
|
||||||
import pkg.openai.manager
|
|
||||||
from func_timeout import FunctionTimedOut
|
|
||||||
import logging
|
|
||||||
|
|
||||||
import pkg.qqbot.filter
|
from ..qqbot import filter as qqbot_filter
|
||||||
import pkg.qqbot.process as processor
|
from ..qqbot import process as processor
|
||||||
import pkg.utils.context
|
from ..utils import context
|
||||||
|
from ..plugin import host as plugin_host
|
||||||
import pkg.plugin.host as plugin_host
|
from ..plugin import models as plugin_models
|
||||||
import pkg.plugin.models as plugin_models
|
|
||||||
import tips as tips_custom
|
import tips as tips_custom
|
||||||
|
from ..qqbot import adapter as msadapter
|
||||||
import pkg.qqbot.adapter as msadapter
|
|
||||||
|
|
||||||
|
|
||||||
# 检查消息是否符合泛响应匹配机制
|
# 检查消息是否符合泛响应匹配机制
|
||||||
def check_response_rule(group_id:int, text: str):
|
def check_response_rule(group_id:int, text: str):
|
||||||
config = pkg.utils.context.get_config()
|
config = context.get_config()
|
||||||
|
|
||||||
rules = config.response_rules
|
rules = config.response_rules
|
||||||
|
|
||||||
@@ -55,7 +48,7 @@ def check_response_rule(group_id:int, text: str):
|
|||||||
|
|
||||||
|
|
||||||
def response_at(group_id: int):
|
def response_at(group_id: int):
|
||||||
config = pkg.utils.context.get_config()
|
config = context.get_config()
|
||||||
|
|
||||||
use_response_rule = config.response_rules
|
use_response_rule = config.response_rules
|
||||||
|
|
||||||
@@ -73,7 +66,7 @@ def response_at(group_id: int):
|
|||||||
|
|
||||||
|
|
||||||
def random_responding(group_id):
|
def random_responding(group_id):
|
||||||
config = pkg.utils.context.get_config()
|
config = context.get_config()
|
||||||
|
|
||||||
use_response_rule = config.response_rules
|
use_response_rule = config.response_rules
|
||||||
|
|
||||||
@@ -130,10 +123,10 @@ class QQBotManager:
|
|||||||
self.adapter = NakuruProjectAdapter(config.nakuru_config)
|
self.adapter = NakuruProjectAdapter(config.nakuru_config)
|
||||||
self.bot_account_id = self.adapter.bot_account_id
|
self.bot_account_id = self.adapter.bot_account_id
|
||||||
else:
|
else:
|
||||||
self.adapter = pkg.utils.context.get_qqbot_manager().adapter
|
self.adapter = context.get_qqbot_manager().adapter
|
||||||
self.bot_account_id = pkg.utils.context.get_qqbot_manager().bot_account_id
|
self.bot_account_id = context.get_qqbot_manager().bot_account_id
|
||||||
|
|
||||||
pkg.utils.context.set_qqbot_manager(self)
|
context.set_qqbot_manager(self)
|
||||||
|
|
||||||
# 注册诸事件
|
# 注册诸事件
|
||||||
# Caution: 注册新的事件处理器之后,请务必在unsubscribe_all中编写相应的取消订阅代码
|
# Caution: 注册新的事件处理器之后,请务必在unsubscribe_all中编写相应的取消订阅代码
|
||||||
@@ -154,7 +147,7 @@ class QQBotManager:
|
|||||||
|
|
||||||
self.on_person_message(event)
|
self.on_person_message(event)
|
||||||
|
|
||||||
pkg.utils.context.get_thread_ctl().submit_user_task(
|
context.get_thread_ctl().submit_user_task(
|
||||||
friend_message_handler,
|
friend_message_handler,
|
||||||
)
|
)
|
||||||
self.adapter.register_listener(
|
self.adapter.register_listener(
|
||||||
@@ -179,7 +172,7 @@ class QQBotManager:
|
|||||||
|
|
||||||
self.on_person_message(event)
|
self.on_person_message(event)
|
||||||
|
|
||||||
pkg.utils.context.get_thread_ctl().submit_user_task(
|
context.get_thread_ctl().submit_user_task(
|
||||||
stranger_message_handler,
|
stranger_message_handler,
|
||||||
)
|
)
|
||||||
# nakuru不区分好友和陌生人,故仅为yirimirai注册陌生人事件
|
# nakuru不区分好友和陌生人,故仅为yirimirai注册陌生人事件
|
||||||
@@ -206,7 +199,7 @@ class QQBotManager:
|
|||||||
|
|
||||||
self.on_group_message(event)
|
self.on_group_message(event)
|
||||||
|
|
||||||
pkg.utils.context.get_thread_ctl().submit_user_task(
|
context.get_thread_ctl().submit_user_task(
|
||||||
group_message_handler,
|
group_message_handler,
|
||||||
event
|
event
|
||||||
)
|
)
|
||||||
@@ -250,22 +243,22 @@ class QQBotManager:
|
|||||||
if hasattr(banlist, "enable_group"):
|
if hasattr(banlist, "enable_group"):
|
||||||
self.enable_group = banlist.enable_group
|
self.enable_group = banlist.enable_group
|
||||||
|
|
||||||
config = pkg.utils.context.get_config()
|
config = context.get_config()
|
||||||
if os.path.exists("sensitive.json") \
|
if os.path.exists("sensitive.json") \
|
||||||
and config.sensitive_word_filter is not None \
|
and config.sensitive_word_filter is not None \
|
||||||
and config.sensitive_word_filter:
|
and config.sensitive_word_filter:
|
||||||
with open("sensitive.json", "r", encoding="utf-8") as f:
|
with open("sensitive.json", "r", encoding="utf-8") as f:
|
||||||
sensitive_json = json.load(f)
|
sensitive_json = json.load(f)
|
||||||
self.reply_filter = pkg.qqbot.filter.ReplyFilter(
|
self.reply_filter = qqbot_filter.ReplyFilter(
|
||||||
sensitive_words=sensitive_json['words'],
|
sensitive_words=sensitive_json['words'],
|
||||||
mask=sensitive_json['mask'] if 'mask' in sensitive_json else '*',
|
mask=sensitive_json['mask'] if 'mask' in sensitive_json else '*',
|
||||||
mask_word=sensitive_json['mask_word'] if 'mask_word' in sensitive_json else ''
|
mask_word=sensitive_json['mask_word'] if 'mask_word' in sensitive_json else ''
|
||||||
)
|
)
|
||||||
else:
|
else:
|
||||||
self.reply_filter = pkg.qqbot.filter.ReplyFilter([])
|
self.reply_filter = qqbot_filter.ReplyFilter([])
|
||||||
|
|
||||||
def send(self, event, msg, check_quote=True, check_at_sender=True):
|
def send(self, event, msg, check_quote=True, check_at_sender=True):
|
||||||
config = pkg.utils.context.get_config()
|
config = context.get_config()
|
||||||
|
|
||||||
if check_at_sender and config.at_sender:
|
if check_at_sender and config.at_sender:
|
||||||
msg.insert(
|
msg.insert(
|
||||||
@@ -306,7 +299,7 @@ class QQBotManager:
|
|||||||
for i in range(self.retry):
|
for i in range(self.retry):
|
||||||
try:
|
try:
|
||||||
|
|
||||||
@func_set_timeout(config.process_message_timeout)
|
@func_timeout.func_set_timeout(config.process_message_timeout)
|
||||||
def time_ctrl_wrapper():
|
def time_ctrl_wrapper():
|
||||||
reply = processor.process_message('person', event.sender.id, str(event.message_chain),
|
reply = processor.process_message('person', event.sender.id, str(event.message_chain),
|
||||||
event.message_chain,
|
event.message_chain,
|
||||||
@@ -315,16 +308,16 @@ class QQBotManager:
|
|||||||
|
|
||||||
reply = time_ctrl_wrapper()
|
reply = time_ctrl_wrapper()
|
||||||
break
|
break
|
||||||
except FunctionTimedOut:
|
except func_timeout.FunctionTimedOut:
|
||||||
logging.warning("person_{}: 超时,重试中({})".format(event.sender.id, i))
|
logging.warning("person_{}: 超时,重试中({})".format(event.sender.id, i))
|
||||||
pkg.openai.session.get_session('person_{}'.format(event.sender.id)).release_response_lock()
|
openai_session.get_session('person_{}'.format(event.sender.id)).release_response_lock()
|
||||||
if "person_{}".format(event.sender.id) in pkg.qqbot.process.processing:
|
if "person_{}".format(event.sender.id) in processor.processing:
|
||||||
pkg.qqbot.process.processing.remove('person_{}'.format(event.sender.id))
|
processor.processing.remove('person_{}'.format(event.sender.id))
|
||||||
failed += 1
|
failed += 1
|
||||||
continue
|
continue
|
||||||
|
|
||||||
if failed == self.retry:
|
if failed == self.retry:
|
||||||
pkg.openai.session.get_session('person_{}'.format(event.sender.id)).release_response_lock()
|
openai_session.get_session('person_{}'.format(event.sender.id)).release_response_lock()
|
||||||
self.notify_admin("{} 请求超时".format("person_{}".format(event.sender.id)))
|
self.notify_admin("{} 请求超时".format("person_{}".format(event.sender.id)))
|
||||||
reply = [tips_custom.reply_message]
|
reply = [tips_custom.reply_message]
|
||||||
|
|
||||||
@@ -344,7 +337,7 @@ class QQBotManager:
|
|||||||
failed = 0
|
failed = 0
|
||||||
for i in range(self.retry):
|
for i in range(self.retry):
|
||||||
try:
|
try:
|
||||||
@func_set_timeout(config.process_message_timeout)
|
@func_timeout.func_set_timeout(config.process_message_timeout)
|
||||||
def time_ctrl_wrapper():
|
def time_ctrl_wrapper():
|
||||||
replys = processor.process_message('group', event.group.id,
|
replys = processor.process_message('group', event.group.id,
|
||||||
str(event.message_chain).strip() if text is None else text,
|
str(event.message_chain).strip() if text is None else text,
|
||||||
@@ -354,16 +347,16 @@ class QQBotManager:
|
|||||||
|
|
||||||
replys = time_ctrl_wrapper()
|
replys = time_ctrl_wrapper()
|
||||||
break
|
break
|
||||||
except FunctionTimedOut:
|
except func_timeout.FunctionTimedOut:
|
||||||
logging.warning("group_{}: 超时,重试中({})".format(event.group.id, i))
|
logging.warning("group_{}: 超时,重试中({})".format(event.group.id, i))
|
||||||
pkg.openai.session.get_session('group_{}'.format(event.group.id)).release_response_lock()
|
openai_session.get_session('group_{}'.format(event.group.id)).release_response_lock()
|
||||||
if "group_{}".format(event.group.id) in pkg.qqbot.process.processing:
|
if "group_{}".format(event.group.id) in processor.processing:
|
||||||
pkg.qqbot.process.processing.remove('group_{}'.format(event.group.id))
|
processor.processing.remove('group_{}'.format(event.group.id))
|
||||||
failed += 1
|
failed += 1
|
||||||
continue
|
continue
|
||||||
|
|
||||||
if failed == self.retry:
|
if failed == self.retry:
|
||||||
pkg.openai.session.get_session('group_{}'.format(event.group.id)).release_response_lock()
|
openai_session.get_session('group_{}'.format(event.group.id)).release_response_lock()
|
||||||
self.notify_admin("{} 请求超时".format("group_{}".format(event.group.id)))
|
self.notify_admin("{} 请求超时".format("group_{}".format(event.group.id)))
|
||||||
replys = [tips_custom.replys_message]
|
replys = [tips_custom.replys_message]
|
||||||
|
|
||||||
@@ -392,7 +385,7 @@ class QQBotManager:
|
|||||||
|
|
||||||
# 通知系统管理员
|
# 通知系统管理员
|
||||||
def notify_admin(self, message: str):
|
def notify_admin(self, message: str):
|
||||||
config = pkg.utils.context.get_config()
|
config = context.get_config()
|
||||||
if config.admin_qq != 0 and config.admin_qq != []:
|
if config.admin_qq != 0 and config.admin_qq != []:
|
||||||
logging.info("通知管理员:{}".format(message))
|
logging.info("通知管理员:{}".format(message))
|
||||||
if type(config.admin_qq) == int:
|
if type(config.admin_qq) == int:
|
||||||
@@ -410,7 +403,7 @@ class QQBotManager:
|
|||||||
)
|
)
|
||||||
|
|
||||||
def notify_admin_message_chain(self, message):
|
def notify_admin_message_chain(self, message):
|
||||||
config = pkg.utils.context.get_config()
|
config = context.get_config()
|
||||||
if config.admin_qq != 0 and config.admin_qq != []:
|
if config.admin_qq != 0 and config.admin_qq != []:
|
||||||
logging.info("通知管理员:{}".format(message))
|
logging.info("通知管理员:{}".format(message))
|
||||||
if type(config.admin_qq) == int:
|
if type(config.admin_qq) == int:
|
||||||
|
|||||||
@@ -1,19 +1,20 @@
|
|||||||
# 普通消息处理模块
|
# 普通消息处理模块
|
||||||
import logging
|
import logging
|
||||||
import openai
|
|
||||||
import pkg.utils.context
|
|
||||||
import pkg.openai.session
|
|
||||||
|
|
||||||
import pkg.plugin.host as plugin_host
|
import openai
|
||||||
import pkg.plugin.models as plugin_models
|
|
||||||
import pkg.qqbot.blob as blob
|
from ..utils import context
|
||||||
|
from ..openai import session as openai_session
|
||||||
|
|
||||||
|
from ..plugin import host as plugin_host
|
||||||
|
from ..plugin import models as plugin_models
|
||||||
import tips as tips_custom
|
import tips as tips_custom
|
||||||
|
|
||||||
|
|
||||||
def handle_exception(notify_admin: str = "", set_reply: str = "") -> list:
|
def handle_exception(notify_admin: str = "", set_reply: str = "") -> list:
|
||||||
"""处理异常,当notify_admin不为空时,会通知管理员,返回通知用户的消息"""
|
"""处理异常,当notify_admin不为空时,会通知管理员,返回通知用户的消息"""
|
||||||
import config
|
import config
|
||||||
pkg.utils.context.get_qqbot_manager().notify_admin(notify_admin)
|
context.get_qqbot_manager().notify_admin(notify_admin)
|
||||||
if config.hide_exce_info_to_user:
|
if config.hide_exce_info_to_user:
|
||||||
return [tips_custom.alter_tip_message] if tips_custom.alter_tip_message else []
|
return [tips_custom.alter_tip_message] if tips_custom.alter_tip_message else []
|
||||||
else:
|
else:
|
||||||
@@ -26,7 +27,7 @@ def process_normal_message(text_message: str, mgr, config, launcher_type: str,
|
|||||||
logging.info("[{}]发送消息:{}".format(session_name, text_message[:min(20, len(text_message))] + (
|
logging.info("[{}]发送消息:{}".format(session_name, text_message[:min(20, len(text_message))] + (
|
||||||
"..." if len(text_message) > 20 else "")))
|
"..." if len(text_message) > 20 else "")))
|
||||||
|
|
||||||
session = pkg.openai.session.get_session(session_name)
|
session = openai_session.get_session(session_name)
|
||||||
|
|
||||||
unexpected_exception_times = 0
|
unexpected_exception_times = 0
|
||||||
|
|
||||||
@@ -54,7 +55,7 @@ def process_normal_message(text_message: str, mgr, config, launcher_type: str,
|
|||||||
"funcs_called": funcs,
|
"funcs_called": funcs,
|
||||||
}
|
}
|
||||||
|
|
||||||
event = pkg.plugin.host.emit(plugin_models.NormalMessageResponded, **args)
|
event = plugin_host.emit(plugin_models.NormalMessageResponded, **args)
|
||||||
|
|
||||||
if event.get_return_value("prefix") is not None:
|
if event.get_return_value("prefix") is not None:
|
||||||
prefix = event.get_return_value("prefix")
|
prefix = event.get_return_value("prefix")
|
||||||
@@ -78,29 +79,29 @@ def process_normal_message(text_message: str, mgr, config, launcher_type: str,
|
|||||||
|
|
||||||
if 'message' in e.error and e.error['message'].__contains__('You exceeded your current quota'):
|
if 'message' in e.error and e.error['message'].__contains__('You exceeded your current quota'):
|
||||||
# 尝试切换api-key
|
# 尝试切换api-key
|
||||||
current_key_name = pkg.utils.context.get_openai_manager().key_mgr.get_key_name(
|
current_key_name = context.get_openai_manager().key_mgr.get_key_name(
|
||||||
pkg.utils.context.get_openai_manager().key_mgr.using_key
|
context.get_openai_manager().key_mgr.using_key
|
||||||
)
|
)
|
||||||
pkg.utils.context.get_openai_manager().key_mgr.set_current_exceeded()
|
context.get_openai_manager().key_mgr.set_current_exceeded()
|
||||||
|
|
||||||
# 触发插件事件
|
# 触发插件事件
|
||||||
args = {
|
args = {
|
||||||
'key_name': current_key_name,
|
'key_name': current_key_name,
|
||||||
'usage': pkg.utils.context.get_openai_manager().audit_mgr
|
'usage': context.get_openai_manager().audit_mgr
|
||||||
.get_usage(pkg.utils.context.get_openai_manager().key_mgr.get_using_key_md5()),
|
.get_usage(context.get_openai_manager().key_mgr.get_using_key_md5()),
|
||||||
'exceeded_keys': pkg.utils.context.get_openai_manager().key_mgr.exceeded,
|
'exceeded_keys': context.get_openai_manager().key_mgr.exceeded,
|
||||||
}
|
}
|
||||||
event = plugin_host.emit(plugin_models.KeyExceeded, **args)
|
event = plugin_host.emit(plugin_models.KeyExceeded, **args)
|
||||||
|
|
||||||
if not event.is_prevented_default():
|
if not event.is_prevented_default():
|
||||||
switched, name = pkg.utils.context.get_openai_manager().key_mgr.auto_switch()
|
switched, name = context.get_openai_manager().key_mgr.auto_switch()
|
||||||
|
|
||||||
if not switched:
|
if not switched:
|
||||||
reply = handle_exception(
|
reply = handle_exception(
|
||||||
"api-key调用额度超限({}),无可用api_key,请向OpenAI账户充值或在config.py中更换api_key;如果你认为这是误判,请尝试重启程序。".format(
|
"api-key调用额度超限({}),无可用api_key,请向OpenAI账户充值或在config.py中更换api_key;如果你认为这是误判,请尝试重启程序。".format(
|
||||||
current_key_name), "[bot]err:API调用额度超额,请联系管理员,或等待修复")
|
current_key_name), "[bot]err:API调用额度超额,请联系管理员,或等待修复")
|
||||||
else:
|
else:
|
||||||
openai.api_key = pkg.utils.context.get_openai_manager().key_mgr.get_using_key()
|
openai.api_key = context.get_openai_manager().key_mgr.get_using_key()
|
||||||
mgr.notify_admin("api-key调用额度超限({}),接口报错,已切换到{}".format(current_key_name, name))
|
mgr.notify_admin("api-key调用额度超限({}),接口报错,已切换到{}".format(current_key_name, name))
|
||||||
reply = ["[bot]err:API调用额度超额,已自动切换,请重新发送消息"]
|
reply = ["[bot]err:API调用额度超额,已自动切换,请重新发送消息"]
|
||||||
continue
|
continue
|
||||||
|
|||||||
@@ -5,28 +5,22 @@ import time
|
|||||||
import mirai
|
import mirai
|
||||||
import logging
|
import logging
|
||||||
|
|
||||||
from mirai import MessageChain, Plain
|
|
||||||
|
|
||||||
# 这里不使用动态引入config
|
# 这里不使用动态引入config
|
||||||
# 因为在这里动态引入会卡死程序
|
# 因为在这里动态引入会卡死程序
|
||||||
# 而此模块静态引用config与动态引入的表现一致
|
# 而此模块静态引用config与动态引入的表现一致
|
||||||
# 已弃用,由于超时时间现已动态使用
|
# 已弃用,由于超时时间现已动态使用
|
||||||
# import config as config_init_import
|
# import config as config_init_import
|
||||||
|
|
||||||
import pkg.openai.session
|
from ..qqbot import ratelimit
|
||||||
import pkg.openai.manager
|
from ..qqbot import command, message
|
||||||
import pkg.utils.reloader
|
from ..openai import session as openai_session
|
||||||
import pkg.utils.updater
|
from ..utils import context
|
||||||
import pkg.utils.context
|
|
||||||
import pkg.qqbot.message
|
|
||||||
import pkg.qqbot.command
|
|
||||||
import pkg.qqbot.ratelimit as ratelimit
|
|
||||||
|
|
||||||
import pkg.plugin.host as plugin_host
|
from ..plugin import host as plugin_host
|
||||||
import pkg.plugin.models as plugin_models
|
from ..plugin import models as plugin_models
|
||||||
import pkg.qqbot.ignore as ignore
|
from ..qqbot import ignore
|
||||||
import pkg.qqbot.banlist as banlist
|
from ..qqbot import banlist
|
||||||
import pkg.qqbot.blob as blob
|
from ..qqbot import blob
|
||||||
import tips as tips_custom
|
import tips as tips_custom
|
||||||
|
|
||||||
processing = []
|
processing = []
|
||||||
@@ -41,11 +35,11 @@ def is_admin(qq: int) -> bool:
|
|||||||
return qq == config.admin_qq
|
return qq == config.admin_qq
|
||||||
|
|
||||||
|
|
||||||
def process_message(launcher_type: str, launcher_id: int, text_message: str, message_chain: MessageChain,
|
def process_message(launcher_type: str, launcher_id: int, text_message: str, message_chain: mirai.MessageChain,
|
||||||
sender_id: int) -> MessageChain:
|
sender_id: int) -> mirai.MessageChain:
|
||||||
global processing
|
global processing
|
||||||
|
|
||||||
mgr = pkg.utils.context.get_qqbot_manager()
|
mgr = context.get_qqbot_manager()
|
||||||
|
|
||||||
reply = []
|
reply = []
|
||||||
session_name = "{}_{}".format(launcher_type, launcher_id)
|
session_name = "{}_{}".format(launcher_type, launcher_id)
|
||||||
@@ -62,7 +56,7 @@ def process_message(launcher_type: str, launcher_id: int, text_message: str, mes
|
|||||||
import config
|
import config
|
||||||
|
|
||||||
if not config.wait_last_done and session_name in processing:
|
if not config.wait_last_done and session_name in processing:
|
||||||
return MessageChain([Plain(tips_custom.message_drop_tip)])
|
return mirai.MessageChain([mirai.Plain(tips_custom.message_drop_tip)])
|
||||||
|
|
||||||
# 检查是否被禁言
|
# 检查是否被禁言
|
||||||
if launcher_type == 'group':
|
if launcher_type == 'group':
|
||||||
@@ -74,9 +68,9 @@ def process_message(launcher_type: str, launcher_id: int, text_message: str, mes
|
|||||||
import config
|
import config
|
||||||
if config.income_msg_check:
|
if config.income_msg_check:
|
||||||
if mgr.reply_filter.is_illegal(text_message):
|
if mgr.reply_filter.is_illegal(text_message):
|
||||||
return MessageChain(Plain("[bot] 消息中存在不合适的内容, 请更换措辞"))
|
return mirai.MessageChain(mirai.Plain("[bot] 消息中存在不合适的内容, 请更换措辞"))
|
||||||
|
|
||||||
pkg.openai.session.get_session(session_name).acquire_response_lock()
|
openai_session.get_session(session_name).acquire_response_lock()
|
||||||
|
|
||||||
text_message = text_message.strip()
|
text_message = text_message.strip()
|
||||||
|
|
||||||
@@ -87,7 +81,7 @@ def process_message(launcher_type: str, launcher_id: int, text_message: str, mes
|
|||||||
# 处理消息
|
# 处理消息
|
||||||
try:
|
try:
|
||||||
|
|
||||||
config = pkg.utils.context.get_config()
|
config = context.get_config()
|
||||||
|
|
||||||
processing.append(session_name)
|
processing.append(session_name)
|
||||||
try:
|
try:
|
||||||
@@ -114,7 +108,7 @@ def process_message(launcher_type: str, launcher_id: int, text_message: str, mes
|
|||||||
reply = event.get_return_value("reply")
|
reply = event.get_return_value("reply")
|
||||||
|
|
||||||
if not event.is_prevented_default():
|
if not event.is_prevented_default():
|
||||||
reply = pkg.qqbot.command.process_command(session_name, text_message,
|
reply = command.process_command(session_name, text_message,
|
||||||
mgr, config, launcher_type, launcher_id, sender_id, is_admin(sender_id))
|
mgr, config, launcher_type, launcher_id, sender_id, is_admin(sender_id))
|
||||||
|
|
||||||
else: # 消息
|
else: # 消息
|
||||||
@@ -124,7 +118,7 @@ def process_message(launcher_type: str, launcher_id: int, text_message: str, mes
|
|||||||
if ratelimit.is_reach_limit(session_name):
|
if ratelimit.is_reach_limit(session_name):
|
||||||
logging.info("根据限速策略丢弃[{}]消息: {}".format(session_name, text_message))
|
logging.info("根据限速策略丢弃[{}]消息: {}".format(session_name, text_message))
|
||||||
|
|
||||||
return MessageChain(["[bot]"+tips_custom.rate_limit_drop_tip]) if tips_custom.rate_limit_drop_tip != "" else []
|
return mirai.MessageChain(["[bot]"+tips_custom.rate_limit_drop_tip]) if tips_custom.rate_limit_drop_tip != "" else []
|
||||||
|
|
||||||
before = time.time()
|
before = time.time()
|
||||||
# 触发插件事件
|
# 触发插件事件
|
||||||
@@ -146,7 +140,7 @@ def process_message(launcher_type: str, launcher_id: int, text_message: str, mes
|
|||||||
reply = event.get_return_value("reply")
|
reply = event.get_return_value("reply")
|
||||||
|
|
||||||
if not event.is_prevented_default():
|
if not event.is_prevented_default():
|
||||||
reply = pkg.qqbot.message.process_normal_message(text_message,
|
reply = message.process_normal_message(text_message,
|
||||||
mgr, config, launcher_type, launcher_id, sender_id)
|
mgr, config, launcher_type, launcher_id, sender_id)
|
||||||
|
|
||||||
# 限速等待时间
|
# 限速等待时间
|
||||||
@@ -170,7 +164,7 @@ def process_message(launcher_type: str, launcher_id: int, text_message: str, mes
|
|||||||
finally:
|
finally:
|
||||||
processing.remove(session_name)
|
processing.remove(session_name)
|
||||||
finally:
|
finally:
|
||||||
pkg.openai.session.get_session(session_name).release_response_lock()
|
openai_session.get_session(session_name).release_response_lock()
|
||||||
|
|
||||||
# 检查延迟时间
|
# 检查延迟时间
|
||||||
if config.force_delay_range[1] == 0:
|
if config.force_delay_range[1] == 0:
|
||||||
@@ -191,4 +185,4 @@ def process_message(launcher_type: str, launcher_id: int, text_message: str, mes
|
|||||||
logging.info("[风控] 强制延迟{:.2f}秒(如需关闭,请到config.py修改force_delay_range字段)".format(delay_time))
|
logging.info("[风控] 强制延迟{:.2f}秒(如需关闭,请到config.py修改force_delay_range字段)".format(delay_time))
|
||||||
time.sleep(delay_time)
|
time.sleep(delay_time)
|
||||||
|
|
||||||
return MessageChain(reply)
|
return mirai.MessageChain(reply)
|
||||||
|
|||||||
@@ -1,19 +1,18 @@
|
|||||||
import mirai
|
|
||||||
|
|
||||||
from ..adapter import MessageSourceAdapter, MessageConverter, EventConverter
|
|
||||||
import nakuru
|
|
||||||
import nakuru.entities.components as nkc
|
|
||||||
|
|
||||||
import asyncio
|
import asyncio
|
||||||
import typing
|
import typing
|
||||||
import traceback
|
import traceback
|
||||||
import logging
|
import logging
|
||||||
import json
|
|
||||||
|
|
||||||
from pkg.qqbot.blob import Forward, ForwardMessageNode, ForwardMessageDiaplay
|
import mirai
|
||||||
|
|
||||||
|
import nakuru
|
||||||
|
import nakuru.entities.components as nkc
|
||||||
|
|
||||||
|
from .. import adapter as adapter_model
|
||||||
|
from ...qqbot import blob
|
||||||
|
|
||||||
|
|
||||||
class NakuruProjectMessageConverter(MessageConverter):
|
class NakuruProjectMessageConverter(adapter_model.MessageConverter):
|
||||||
"""消息转换器"""
|
"""消息转换器"""
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def yiri2target(message_chain: mirai.MessageChain) -> list:
|
def yiri2target(message_chain: mirai.MessageChain) -> list:
|
||||||
@@ -49,7 +48,7 @@ class NakuruProjectMessageConverter(MessageConverter):
|
|||||||
nakuru_msg_list.append(nkc.Record.fromURL(component.url))
|
nakuru_msg_list.append(nkc.Record.fromURL(component.url))
|
||||||
elif component.path is not None:
|
elif component.path is not None:
|
||||||
nakuru_msg_list.append(nkc.Record.fromFileSystem(component.path))
|
nakuru_msg_list.append(nkc.Record.fromFileSystem(component.path))
|
||||||
elif type(component) is Forward:
|
elif type(component) is blob.Forward:
|
||||||
# 转发消息
|
# 转发消息
|
||||||
yiri_forward_node_list = component.node_list
|
yiri_forward_node_list = component.node_list
|
||||||
nakuru_forward_node_list = []
|
nakuru_forward_node_list = []
|
||||||
@@ -102,7 +101,7 @@ class NakuruProjectMessageConverter(MessageConverter):
|
|||||||
return chain
|
return chain
|
||||||
|
|
||||||
|
|
||||||
class NakuruProjectEventConverter(EventConverter):
|
class NakuruProjectEventConverter(adapter_model.EventConverter):
|
||||||
"""事件转换器"""
|
"""事件转换器"""
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def yiri2target(event: typing.Type[mirai.Event]):
|
def yiri2target(event: typing.Type[mirai.Event]):
|
||||||
@@ -157,7 +156,7 @@ class NakuruProjectEventConverter(EventConverter):
|
|||||||
raise Exception("未支持转换的事件类型: " + str(event))
|
raise Exception("未支持转换的事件类型: " + str(event))
|
||||||
|
|
||||||
|
|
||||||
class NakuruProjectAdapter(MessageSourceAdapter):
|
class NakuruProjectAdapter(adapter_model.MessageSourceAdapter):
|
||||||
"""nakuru-project适配器"""
|
"""nakuru-project适配器"""
|
||||||
bot: nakuru.CQHTTP
|
bot: nakuru.CQHTTP
|
||||||
bot_account_id: int
|
bot_account_id: int
|
||||||
|
|||||||
@@ -1,13 +1,14 @@
|
|||||||
from ..adapter import MessageSourceAdapter
|
import asyncio
|
||||||
|
import typing
|
||||||
|
|
||||||
import mirai
|
import mirai
|
||||||
import mirai.models.bus
|
import mirai.models.bus
|
||||||
from mirai.bot import MiraiRunner
|
from mirai.bot import MiraiRunner
|
||||||
|
|
||||||
import asyncio
|
from .. import adapter as adapter_model
|
||||||
import typing
|
|
||||||
|
|
||||||
|
|
||||||
class YiriMiraiAdapter(MessageSourceAdapter):
|
class YiriMiraiAdapter(adapter_model.MessageSourceAdapter):
|
||||||
"""YiriMirai适配器"""
|
"""YiriMirai适配器"""
|
||||||
bot: mirai.Mirai
|
bot: mirai.Mirai
|
||||||
|
|
||||||
|
|||||||
File diff suppressed because one or more lines are too long
@@ -1,5 +1,5 @@
|
|||||||
import threading
|
import threading
|
||||||
from pkg.utils import ThreadCtl
|
from . import threadctl
|
||||||
|
|
||||||
|
|
||||||
context = {
|
context = {
|
||||||
@@ -87,8 +87,8 @@ def set_thread_ctl(inst):
|
|||||||
context_lock.release()
|
context_lock.release()
|
||||||
|
|
||||||
|
|
||||||
def get_thread_ctl() -> ThreadCtl:
|
def get_thread_ctl() -> threadctl.ThreadCtl:
|
||||||
context_lock.acquire()
|
context_lock.acquire()
|
||||||
t: ThreadCtl = context['pool_ctl']
|
t: threadctl.ThreadCtl = context['pool_ctl']
|
||||||
context_lock.release()
|
context_lock.release()
|
||||||
return t
|
return t
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
from pip._internal import main as pipmain
|
from pip._internal import main as pipmain
|
||||||
|
|
||||||
import pkg.utils.log as log
|
from . import log
|
||||||
|
|
||||||
|
|
||||||
def install(package):
|
def install(package):
|
||||||
@@ -19,7 +19,7 @@ def run_pip(params: list):
|
|||||||
|
|
||||||
|
|
||||||
def install_requirements(file):
|
def install_requirements(file):
|
||||||
pipmain(['install', '-r', file, "--upgrade", "-i", "https://pypi.tuna.tsinghua.edu.cn/simple",
|
pipmain(['install', '-r', file, "-i", "https://pypi.tuna.tsinghua.edu.cn/simple",
|
||||||
"--trusted-host", "pypi.tuna.tsinghua.edu.cn"])
|
"--trusted-host", "pypi.tuna.tsinghua.edu.cn"])
|
||||||
log.reset_logging()
|
log.reset_logging()
|
||||||
|
|
||||||
|
|||||||
@@ -1,10 +1,9 @@
|
|||||||
import logging
|
import logging
|
||||||
import threading
|
|
||||||
|
|
||||||
import importlib
|
import importlib
|
||||||
import pkgutil
|
import pkgutil
|
||||||
import pkg.utils.context as context
|
|
||||||
import pkg.plugin.host
|
from . import context
|
||||||
|
from ..plugin import host as plugin_host
|
||||||
|
|
||||||
|
|
||||||
def walk(module, prefix='', path_prefix=''):
|
def walk(module, prefix='', path_prefix=''):
|
||||||
@@ -15,7 +14,7 @@ def walk(module, prefix='', path_prefix=''):
|
|||||||
walk(__import__(module.__name__ + '.' + item.name, fromlist=['']), prefix + item.name + '.', path_prefix + item.name + '/')
|
walk(__import__(module.__name__ + '.' + item.name, fromlist=['']), prefix + item.name + '.', path_prefix + item.name + '/')
|
||||||
else:
|
else:
|
||||||
logging.info('reload module: {}, path: {}'.format(prefix + item.name, path_prefix + item.name + '.py'))
|
logging.info('reload module: {}, path: {}'.format(prefix + item.name, path_prefix + item.name + '.py'))
|
||||||
pkg.plugin.host.__current_module_path__ = "plugins/" + path_prefix + item.name + '.py'
|
plugin_host.__current_module_path__ = "plugins/" + path_prefix + item.name + '.py'
|
||||||
importlib.reload(__import__(module.__name__ + '.' + item.name, fromlist=['']))
|
importlib.reload(__import__(module.__name__ + '.' + item.name, fromlist=['']))
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -1,11 +1,11 @@
|
|||||||
import logging
|
import logging
|
||||||
|
|
||||||
from PIL import Image, ImageDraw, ImageFont
|
|
||||||
import re
|
import re
|
||||||
import os
|
import os
|
||||||
import config
|
import config
|
||||||
import traceback
|
import traceback
|
||||||
|
|
||||||
|
from PIL import Image, ImageDraw, ImageFont
|
||||||
|
|
||||||
text_render_font: ImageFont = None
|
text_render_font: ImageFont = None
|
||||||
|
|
||||||
if config.blob_message_strategy == "image": # 仅在启用了image时才加载字体
|
if config.blob_message_strategy == "image": # 仅在启用了image时才加载字体
|
||||||
|
|||||||
@@ -3,10 +3,9 @@ import logging
|
|||||||
import os.path
|
import os.path
|
||||||
|
|
||||||
import requests
|
import requests
|
||||||
import json
|
|
||||||
|
|
||||||
import pkg.utils.constants
|
from . import constants
|
||||||
import pkg.utils.network as network
|
from . import network
|
||||||
|
|
||||||
|
|
||||||
def check_dulwich_closure():
|
def check_dulwich_closure():
|
||||||
@@ -70,7 +69,7 @@ def get_release_list() -> list:
|
|||||||
|
|
||||||
def get_current_tag() -> str:
|
def get_current_tag() -> str:
|
||||||
"""获取当前tag"""
|
"""获取当前tag"""
|
||||||
current_tag = pkg.utils.constants.semantic_version
|
current_tag = constants.semantic_version
|
||||||
if os.path.exists("current_tag"):
|
if os.path.exists("current_tag"):
|
||||||
with open("current_tag", "r") as f:
|
with open("current_tag", "r") as f:
|
||||||
current_tag = f.read()
|
current_tag = f.read()
|
||||||
@@ -204,9 +203,9 @@ def update_all(cli: bool = False) -> bool:
|
|||||||
# 通知管理员
|
# 通知管理员
|
||||||
if not cli:
|
if not cli:
|
||||||
import pkg.utils.context
|
import pkg.utils.context
|
||||||
pkg.utils.context.get_qqbot_manager().notify_admin("已更新到最新版本: {}\n更新日志:\n{}\n完整的更新日志请前往 https://github.com/RockChinQ/QChatGPT/releases 查看".format(current_tag, "\n".join(rls_notes[:-1])))
|
pkg.utils.context.get_qqbot_manager().notify_admin("已更新到最新版本: {}\n更新日志:\n{}\n完整的更新日志请前往 https://github.com/RockChinQ/QChatGPT/releases 查看。\n请手动重启程序以使用新版本。".format(current_tag, "\n".join(rls_notes[:-1])))
|
||||||
else:
|
else:
|
||||||
print("已更新到最新版本: {}\n更新日志:\n{}\n完整的更新日志请前往 https://github.com/RockChinQ/QChatGPT/releases 查看".format(current_tag, "\n".join(rls_notes[:-1])))
|
print("已更新到最新版本: {}\n更新日志:\n{}\n完整的更新日志请前往 https://github.com/RockChinQ/QChatGPT/releases 查看。请手动重启程序以使用新版本。".format(current_tag, "\n".join(rls_notes[:-1])))
|
||||||
return True
|
return True
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -2,7 +2,7 @@ requests
|
|||||||
openai
|
openai
|
||||||
dulwich~=0.21.6
|
dulwich~=0.21.6
|
||||||
colorlog~=6.6.0
|
colorlog~=6.6.0
|
||||||
yiri-mirai
|
yiri-mirai-rc
|
||||||
websockets
|
websockets
|
||||||
urllib3
|
urllib3
|
||||||
func_timeout~=4.3.5
|
func_timeout~=4.3.5
|
||||||
|
|||||||
@@ -10,5 +10,11 @@
|
|||||||
"time": "2023-11-10 12:20:09",
|
"time": "2023-11-10 12:20:09",
|
||||||
"timestamp": 1699590009,
|
"timestamp": 1699590009,
|
||||||
"content": "OpenAI 库1.0版本已发行,若出现 OpenAI 调用问题,请更新 QChatGPT 版本。详见项目主页:https://github.com/RockChinQ/QChatGPT"
|
"content": "OpenAI 库1.0版本已发行,若出现 OpenAI 调用问题,请更新 QChatGPT 版本。详见项目主页:https://github.com/RockChinQ/QChatGPT"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"id": 4,
|
||||||
|
"time": "2023-11-13 18:02:39",
|
||||||
|
"timestamp": 1699869759,
|
||||||
|
"content": "近期 OpenAI 接口改动频繁,正在积极适配并添加新功能,请尽快更新到最新版本,更新方式:https://github.com/RockChinQ/QChatGPT/discussions/595"
|
||||||
}
|
}
|
||||||
]
|
]
|
||||||
Reference in New Issue
Block a user