Soulter
93 lines
2.8 KiB
Python
93 lines
2.8 KiB
Python
import time
|
|
import asyncio
|
|
import requests
|
|
import json
|
|
import sys
|
|
import psutil
|
|
|
|
from type.types import Context
|
|
from SparkleLogging.utils.core import LogManager
|
|
from logging import Logger
|
|
|
|
logger: Logger = LogManager.GetLogger(log_name='astrbot')
|
|
|
|
async def upload_metrics(_global_object: Context):
|
|
'''
|
|
上传相关非敏感统计数据
|
|
'''
|
|
await asyncio.sleep(10)
|
|
while True:
|
|
platform_stats = {}
|
|
llm_stats = {}
|
|
plugin_stats = {}
|
|
for platform in _global_object.platforms:
|
|
platform_stats[platform.platform_name] = {
|
|
"cnt_receive": 0,
|
|
"cnt_reply": 0
|
|
}
|
|
|
|
for llm in _global_object.llms:
|
|
stat = llm.llm_instance.model_stat
|
|
for k in stat:
|
|
llm_stats[llm.llm_name + "#" + k] = stat[k]
|
|
llm.llm_instance.reset_model_stat()
|
|
|
|
for plugin in _global_object.cached_plugins:
|
|
plugin_stats[plugin.metadata.plugin_name] = {
|
|
"metadata": plugin.metadata,
|
|
"trig_cnt": plugin.trig_cnt
|
|
}
|
|
plugin.reset_trig_cnt()
|
|
|
|
try:
|
|
res = {
|
|
"stat_version": "moon",
|
|
"version": _global_object.version, # 版本号
|
|
"platform_stats": platform_stats, # 过去 30 分钟各消息平台交互消息数
|
|
"llm_stats": llm_stats,
|
|
"plugin_stats": plugin_stats,
|
|
"sys": sys.platform, # 系统版本
|
|
}
|
|
resp = requests.post(
|
|
'https://api.soulter.top/upload', data=json.dumps(res), timeout=5)
|
|
if resp.status_code == 200:
|
|
ok = resp.json()
|
|
if ok['status'] == 'ok':
|
|
_global_object.cnt_total = 0
|
|
except BaseException as e:
|
|
pass
|
|
await asyncio.sleep(30*60)
|
|
|
|
def retry(n: int = 3):
|
|
'''
|
|
重试装饰器
|
|
'''
|
|
def decorator(func):
|
|
def wrapper(*args, **kwargs):
|
|
for i in range(n):
|
|
try:
|
|
return func(*args, **kwargs)
|
|
except Exception as e:
|
|
if i == n-1: raise e
|
|
logger.warning(f"函数 {func.__name__} 第 {i+1} 次重试... {e}")
|
|
return wrapper
|
|
return decorator
|
|
|
|
def run_monitor(global_object: Context):
|
|
'''
|
|
监测机器性能
|
|
- Bot 内存使用量
|
|
- CPU 占用率
|
|
'''
|
|
start_time = time.time()
|
|
while True:
|
|
stat = global_object.dashboard_data.stats
|
|
# 程序占用的内存大小
|
|
mem = psutil.Process().memory_info().rss / 1024 / 1024 # MB
|
|
stat['sys_perf'] = {
|
|
'memory': mem,
|
|
'cpu': psutil.cpu_percent()
|
|
}
|
|
stat['sys_start_time'] = start_time
|
|
time.sleep(30)
|