chore: ruff的修改
This commit is contained in:
20
bot.py
20
bot.py
@@ -105,13 +105,11 @@ def load_logger():
|
||||
current_env = os.getenv("ENV", "dev")
|
||||
|
||||
# 公共配置参数
|
||||
log_level = os.getenv(
|
||||
"LOG_LEVEL",
|
||||
"INFO" if current_env == "prod" else "DEBUG"
|
||||
)
|
||||
log_level = os.getenv("LOG_LEVEL", "INFO" if current_env == "prod" else "DEBUG")
|
||||
log_filter = lambda record: (
|
||||
("nonebot" not in record["name"] or record["level"].no >= logger.level("ERROR").no
|
||||
) if current_env == "prod" else True
|
||||
("nonebot" not in record["name"] or record["level"].no >= logger.level("ERROR").no)
|
||||
if current_env == "prod"
|
||||
else True
|
||||
)
|
||||
log_format = (
|
||||
"<green>{time:YYYY-MM-DD HH:mm:ss.SSS}</green> "
|
||||
@@ -129,17 +127,11 @@ def load_logger():
|
||||
colorize=False,
|
||||
level=log_level,
|
||||
filter=log_filter,
|
||||
encoding="utf-8"
|
||||
encoding="utf-8",
|
||||
)
|
||||
|
||||
# 终端输出
|
||||
logger.add(
|
||||
sys.stderr,
|
||||
format=log_format,
|
||||
colorize=True,
|
||||
level=log_level,
|
||||
filter=log_filter
|
||||
)
|
||||
logger.add(sys.stderr, format=log_format, colorize=True, level=log_level, filter=log_filter)
|
||||
|
||||
|
||||
def scan_provider(env_config: dict):
|
||||
|
||||
@@ -22,9 +22,7 @@ def __create_database_instance():
|
||||
|
||||
if username and password:
|
||||
# 如果有用户名和密码,使用认证连接
|
||||
return MongoClient(
|
||||
host, port, username=username, password=password, authSource=auth_source
|
||||
)
|
||||
return MongoClient(host, port, username=username, password=password, authSource=auth_source)
|
||||
|
||||
# 否则使用无认证连接
|
||||
return MongoClient(host, port)
|
||||
|
||||
@@ -99,15 +99,14 @@ async def _(bot: Bot, event: MessageEvent, state: T_State):
|
||||
@scheduler.scheduled_job("interval", seconds=global_config.build_memory_interval, id="build_memory")
|
||||
async def build_memory_task():
|
||||
"""每build_memory_interval秒执行一次记忆构建"""
|
||||
logger.debug(
|
||||
"[记忆构建]"
|
||||
"------------------------------------开始构建记忆--------------------------------------")
|
||||
logger.debug("[记忆构建]------------------------------------开始构建记忆--------------------------------------")
|
||||
start_time = time.time()
|
||||
await hippocampus.operation_build_memory(chat_size=20)
|
||||
end_time = time.time()
|
||||
logger.success(
|
||||
f"[记忆构建]--------------------------记忆构建完成:耗时: {end_time - start_time:.2f} "
|
||||
"秒-------------------------------------------")
|
||||
"秒-------------------------------------------"
|
||||
)
|
||||
|
||||
|
||||
@scheduler.scheduled_job("interval", seconds=global_config.forget_memory_interval, id="forget_memory")
|
||||
|
||||
Reference in New Issue
Block a user