[WIP] feat: agentic memory

astrbot/core/core_lifecycle.py

@@ -24,6 +24,7 @@
 from astrbot.core.db.migration.migra_45_to_46 import migrate_45_to_46
 from astrbot.core.db.migration.migra_webchat_session import migrate_webchat_session
 from astrbot.core.knowledge_base.kb_mgr import KnowledgeBaseManager
+from astrbot.core.memory.memory_manager import MemoryManager
 from astrbot.core.persona_mgr import PersonaManager
 from astrbot.core.pipeline.scheduler import PipelineContext, PipelineScheduler
 from astrbot.core.platform.manager import PlatformManager
@@ -136,6 +137,8 @@ async def initialize(self) -> None:
 
         # 初始化知识库管理器
         self.kb_manager = KnowledgeBaseManager(self.provider_manager)
+        # 初始化记忆管理器
+        self.memory_manager = MemoryManager()
 
         # 初始化提供给插件的上下文
         self.star_context = Context(
@@ -149,6 +152,7 @@ async def initialize(self) -> None:
             self.persona_mgr,
             self.astrbot_config_mgr,
             self.kb_manager,
+            self.memory_manager,
         )
 
         # 初始化插件管理器

astrbot/core/db/vec_db/base.py

@@ -1,11 +1,20 @@
 import abc
 from dataclasses import dataclass
+from typing import TypedDict
 
 
 @dataclass
 class Result:
+    class ResultData(TypedDict):
+        id: str
+        doc_id: str
+        text: str
+        metadata: str
+        created_at: int
+        updated_at: int
+
     similarity: float
-    data: dict
+    data: ResultData | dict
 
 
 class BaseVecDB: