mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-09-29 00:32:06 +08:00
121 lines
3.0 KiB
Python
121 lines
3.0 KiB
Python
"""
|
|
Zhenxun Bot - 核心服务模块
|
|
|
|
主要服务包括：
|
|
- 数据库上下文 (db_context): 提供数据库模型基类和连接管理。
|
|
- 日志服务 (log): 提供增强的、带上下文的日志记录器。
|
|
- LLM服务 (llm): 提供与大语言模型交互的统一API。
|
|
- 插件生命周期管理 (plugin_init): 支持插件安装和卸载时的钩子函数。
|
|
- 定时任务调度器 (scheduler): 提供持久化的、可管理的定时任务服务。
|
|
"""
|
|
|
|
import asyncio
|
|
|
|
import nonebot
|
|
from nonebot import require
|
|
|
|
require("nonebot_plugin_apscheduler")
|
|
require("nonebot_plugin_alconna")
|
|
require("nonebot_plugin_session")
|
|
require("nonebot_plugin_htmlrender")
|
|
require("nonebot_plugin_uninfo")
|
|
require("nonebot_plugin_waiter")
|
|
|
|
from .avatar_service import avatar_service
|
|
from .db_context import Model, disconnect, with_db_timeout
|
|
from .group_settings_service import group_settings_service
|
|
from .llm import (
|
|
AI,
|
|
AIConfig,
|
|
CommonOverrides,
|
|
LLMContentPart,
|
|
LLMException,
|
|
LLMGenerationConfig,
|
|
LLMMessage,
|
|
chat,
|
|
clear_model_cache,
|
|
code,
|
|
create_multimodal_message,
|
|
embed,
|
|
generate,
|
|
generate_structured,
|
|
get_cache_stats,
|
|
get_model_instance,
|
|
list_available_models,
|
|
list_embedding_models,
|
|
search,
|
|
set_global_default_model_name,
|
|
)
|
|
from .log import logger
|
|
from .plugin_init import PluginInit, PluginInitManager
|
|
from .renderer import renderer_service
|
|
from .scheduler import (
|
|
ExecutionPolicy,
|
|
ScheduleContext,
|
|
Trigger,
|
|
scheduler_manager,
|
|
)
|
|
|
|
__all__ = [
|
|
"AI",
|
|
"AIConfig",
|
|
"CommonOverrides",
|
|
"ExecutionPolicy",
|
|
"LLMContentPart",
|
|
"LLMException",
|
|
"LLMGenerationConfig",
|
|
"LLMMessage",
|
|
"Model",
|
|
"PluginInit",
|
|
"PluginInitManager",
|
|
"ScheduleContext",
|
|
"Trigger",
|
|
"avatar_service",
|
|
"chat",
|
|
"clear_model_cache",
|
|
"code",
|
|
"create_multimodal_message",
|
|
"disconnect",
|
|
"embed",
|
|
"generate",
|
|
"generate_structured",
|
|
"get_cache_stats",
|
|
"get_model_instance",
|
|
"group_settings_service",
|
|
"list_available_models",
|
|
"list_embedding_models",
|
|
"logger",
|
|
"renderer_service",
|
|
"scheduler_manager",
|
|
"search",
|
|
"set_global_default_model_name",
|
|
"with_db_timeout",
|
|
]
|
|
|
|
|
|
async def cancel_pending_tasks():
|
|
"""仅取消 zhenxun 业务协程，避免干扰框架的关机流程。"""
|
|
loop = asyncio.get_running_loop()
|
|
current = asyncio.current_task(loop=loop)
|
|
pending = []
|
|
for task in asyncio.all_tasks(loop):
|
|
if task is current or task.done():
|
|
continue
|
|
coro = task.get_coro()
|
|
module = getattr(coro, "__module__", "")
|
|
if module.startswith("zhenxun"):
|
|
pending.append(task)
|
|
|
|
if not pending:
|
|
return
|
|
|
|
for task in pending:
|
|
task.cancel()
|
|
|
|
await asyncio.gather(*pending, return_exceptions=True)
|
|
|
|
|
|
driver = nonebot.get_driver()
|
|
# 先取消可能在跑的任务，再断开数据库，避免 pool closing 异常
|
|
driver.on_shutdown(cancel_pending_tasks)
|