Soulter
aa6d07afcc
* refactor: move all internal commands from astrbot plugin to default_command plugin * ruff check * feat: add config * ruff check
129 lines
5.2 KiB
Python
129 lines
5.2 KiB
Python
import traceback
|
|
|
|
from astrbot.api import star
|
|
from astrbot.api.event import AstrMessageEvent, filter
|
|
from astrbot.api.message_components import Image, Plain
|
|
from astrbot.api.provider import LLMResponse, ProviderRequest
|
|
from astrbot.core import logger
|
|
|
|
from .long_term_memory import LongTermMemory
|
|
from .process_llm_request import ProcessLLMRequest
|
|
|
|
|
|
class Main(star.Star):
|
|
def __init__(self, context: star.Context) -> None:
|
|
self.context = context
|
|
self.ltm = None
|
|
try:
|
|
self.ltm = LongTermMemory(self.context.astrbot_config_mgr, self.context)
|
|
except BaseException as e:
|
|
logger.error(f"聊天增强 err: {e}")
|
|
|
|
self.proc_llm_req = ProcessLLMRequest(self.context)
|
|
|
|
def ltm_enabled(self, event: AstrMessageEvent):
|
|
ltmse = self.context.get_config(umo=event.unified_msg_origin)[
|
|
"provider_ltm_settings"
|
|
]
|
|
return ltmse["group_icl_enable"] or ltmse["active_reply"]["enable"]
|
|
|
|
@filter.platform_adapter_type(filter.PlatformAdapterType.ALL)
|
|
async def on_message(self, event: AstrMessageEvent):
|
|
"""群聊记忆增强"""
|
|
has_image_or_plain = False
|
|
for comp in event.message_obj.message:
|
|
if isinstance(comp, Plain) or isinstance(comp, Image):
|
|
has_image_or_plain = True
|
|
break
|
|
|
|
if self.ltm_enabled(event) and self.ltm and has_image_or_plain:
|
|
need_active = await self.ltm.need_active_reply(event)
|
|
|
|
group_icl_enable = self.context.get_config()["provider_ltm_settings"][
|
|
"group_icl_enable"
|
|
]
|
|
if group_icl_enable:
|
|
"""记录对话"""
|
|
try:
|
|
await self.ltm.handle_message(event)
|
|
except BaseException as e:
|
|
logger.error(e)
|
|
|
|
if need_active:
|
|
"""主动回复"""
|
|
provider = self.context.get_using_provider(event.unified_msg_origin)
|
|
if not provider:
|
|
logger.error("未找到任何 LLM 提供商。请先配置。无法主动回复")
|
|
return
|
|
try:
|
|
conv = None
|
|
session_curr_cid = await self.context.conversation_manager.get_curr_conversation_id(
|
|
event.unified_msg_origin,
|
|
)
|
|
|
|
if not session_curr_cid:
|
|
logger.error(
|
|
"当前未处于对话状态，无法主动回复，请确保 平台设置->会话隔离(unique_session) 未开启，并使用 /switch 序号 切换或者 /new 创建一个会话。",
|
|
)
|
|
return
|
|
|
|
conv = await self.context.conversation_manager.get_conversation(
|
|
event.unified_msg_origin,
|
|
session_curr_cid,
|
|
)
|
|
|
|
prompt = event.message_str
|
|
|
|
if not conv:
|
|
logger.error("未找到对话，无法主动回复")
|
|
return
|
|
|
|
yield event.request_llm(
|
|
prompt=prompt,
|
|
func_tool_manager=self.context.get_llm_tool_manager(),
|
|
session_id=event.session_id,
|
|
conversation=conv,
|
|
)
|
|
except BaseException as e:
|
|
logger.error(traceback.format_exc())
|
|
logger.error(f"主动回复失败: {e}")
|
|
|
|
@filter.on_llm_request()
|
|
async def decorate_llm_req(self, event: AstrMessageEvent, req: ProviderRequest):
|
|
"""在请求 LLM 前注入人格信息、Identifier、时间、回复内容等 System Prompt"""
|
|
await self.proc_llm_req.process_llm_request(event, req)
|
|
|
|
if self.ltm and self.ltm_enabled(event):
|
|
try:
|
|
await self.ltm.on_req_llm(event, req)
|
|
except BaseException as e:
|
|
logger.error(f"ltm: {e}")
|
|
|
|
@filter.on_llm_response()
|
|
async def inject_reasoning(self, event: AstrMessageEvent, resp: LLMResponse):
|
|
"""在 LLM 响应后基于配置注入思考过程文本 / 在 LLM 响应后记录对话"""
|
|
umo = event.unified_msg_origin
|
|
cfg = self.context.get_config(umo).get("provider_settings", {})
|
|
show_reasoning = cfg.get("display_reasoning_text", False)
|
|
if show_reasoning and resp.reasoning_content:
|
|
resp.completion_text = (
|
|
f"🤔 思考: {resp.reasoning_content}\n\n{resp.completion_text}"
|
|
)
|
|
|
|
if self.ltm and self.ltm_enabled(event):
|
|
try:
|
|
await self.ltm.after_req_llm(event, resp)
|
|
except Exception as e:
|
|
logger.error(f"ltm: {e}")
|
|
|
|
@filter.after_message_sent()
|
|
async def after_message_sent(self, event: AstrMessageEvent):
|
|
"""消息发送后处理"""
|
|
if self.ltm and self.ltm_enabled(event):
|
|
try:
|
|
clean_session = event.get_extra("_clean_ltm_session", False)
|
|
if clean_session:
|
|
await self.ltm.remove_session(event)
|
|
except Exception as e:
|
|
logger.error(f"ltm: {e}")
|