diff --git a/packages/astrbot/commands/conversation.py b/packages/astrbot/commands/conversation.py index 2d5317644..2f2fbc128 100644 --- a/packages/astrbot/commands/conversation.py +++ b/packages/astrbot/commands/conversation.py @@ -6,6 +6,7 @@ from astrbot.core.provider.sources.dify_source import ProviderDify from astrbot.core.provider.sources.coze_source import ProviderCoze from astrbot.api import sp, logger +from ..long_term_memory import LongTermMemory from typing import Union from enum import Enum @@ -36,7 +37,7 @@ def get_scene(cls, is_group: bool, is_unique_session: bool) -> "RstScene": class ConversationCommands: - def __init__(self, context: star.Context, ltm=None): + def __init__(self, context: star.Context, ltm: LongTermMemory | None = None): self.context = context self.ltm = ltm diff --git a/packages/astrbot/main.py b/packages/astrbot/main.py index 272864633..6fd0b0e5a 100644 --- a/packages/astrbot/main.py +++ b/packages/astrbot/main.py @@ -41,7 +41,7 @@ def __init__(self, context: star.Context) -> None: self.tool_c = ToolCommands(self.context) self.plugin_c = PluginCommands(self.context) self.admin_c = AdminCommands(self.context) - self.conversation_c = ConversationCommands(self.context) + self.conversation_c = ConversationCommands(self.context, self.ltm) self.provider_c = ProviderCommands(self.context) self.persona_c = PersonaCommands(self.context) self.alter_cmd_c = AlterCmdCommands(self.context)