feat(memory-graph): Step 1 - 集成记忆工具到插件系统
完成记忆系统工具的插件化集成: 1. 创建记忆工具适配器 (memory_plugin_tools.py) - CreateMemoryTool: 创建新记忆 - LinkMemoriesTool: 关联两条记忆 - SearchMemoriesTool: 搜索相关记忆 - 适配 BaseTool 接口,支持 LLM 调用 2. 创建全局 MemoryManager 单例 (manager_singleton.py) - initialize_memory_manager(): 初始化全局实例 - get_memory_manager(): 获取单例实例 - shutdown_memory_manager(): 关闭管理器 - 线程安全的单例模式 3. 创建记忆系统插件 (plugins/memory_graph_plugin/) - MemoryGraphPlugin: 插件主类 - 自动注册3个记忆工具到系统 - on_plugin_loaded(): 初始化 MemoryManager - on_unload(): 清理资源 4. 修复类型问题 - ToolParamType.OBJECT STRING (JSON格式) - ToolParamType.NUMBER FLOAT - attributes 参数支持 JSON 字符串解析 - 修复 min_importance None 比较错误 5. 添加集成测试 (test_plugin_integration.py) - 测试工具导入和定义 - 测试 MemoryManager 初始化 - 测试工具执行(创建、搜索记忆) - 测试单例模式 测试结果: 所有测试通过 LLM 现在可以通过工具调用主动管理记忆!
This commit is contained in:
93
src/memory_graph/manager_singleton.py
Normal file
93
src/memory_graph/manager_singleton.py
Normal file
@@ -0,0 +1,93 @@
|
||||
"""
|
||||
记忆系统管理单例
|
||||
|
||||
提供全局访问的 MemoryManager 实例
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
from pathlib import Path
|
||||
from typing import Optional
|
||||
|
||||
from src.common.logger import get_logger
|
||||
from src.memory_graph.manager import MemoryManager
|
||||
|
||||
logger = get_logger(__name__)
|
||||
|
||||
# 全局 MemoryManager 实例
|
||||
_memory_manager: Optional[MemoryManager] = None
|
||||
_initialized: bool = False
|
||||
|
||||
|
||||
async def initialize_memory_manager(data_dir: Optional[Path | str] = None) -> MemoryManager:
|
||||
"""
|
||||
初始化全局 MemoryManager
|
||||
|
||||
Args:
|
||||
data_dir: 数据目录,默认使用 data/memory_graph
|
||||
|
||||
Returns:
|
||||
MemoryManager 实例
|
||||
"""
|
||||
global _memory_manager, _initialized
|
||||
|
||||
if _initialized and _memory_manager:
|
||||
logger.info("MemoryManager 已经初始化,返回现有实例")
|
||||
return _memory_manager
|
||||
|
||||
try:
|
||||
if data_dir is None:
|
||||
data_dir = Path("data/memory_graph")
|
||||
elif isinstance(data_dir, str):
|
||||
data_dir = Path(data_dir)
|
||||
|
||||
logger.info(f"正在初始化全局 MemoryManager (data_dir={data_dir})...")
|
||||
|
||||
_memory_manager = MemoryManager(data_dir=data_dir)
|
||||
await _memory_manager.initialize()
|
||||
|
||||
_initialized = True
|
||||
logger.info("✅ 全局 MemoryManager 初始化成功")
|
||||
|
||||
return _memory_manager
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"初始化 MemoryManager 失败: {e}", exc_info=True)
|
||||
_initialized = False
|
||||
_memory_manager = None
|
||||
raise
|
||||
|
||||
|
||||
def get_memory_manager() -> Optional[MemoryManager]:
|
||||
"""
|
||||
获取全局 MemoryManager 实例
|
||||
|
||||
Returns:
|
||||
MemoryManager 实例,如果未初始化则返回 None
|
||||
"""
|
||||
if not _initialized or _memory_manager is None:
|
||||
logger.warning("MemoryManager 尚未初始化,请先调用 initialize_memory_manager()")
|
||||
return None
|
||||
|
||||
return _memory_manager
|
||||
|
||||
|
||||
async def shutdown_memory_manager():
|
||||
"""关闭全局 MemoryManager"""
|
||||
global _memory_manager, _initialized
|
||||
|
||||
if _memory_manager:
|
||||
try:
|
||||
logger.info("正在关闭全局 MemoryManager...")
|
||||
await _memory_manager.shutdown()
|
||||
logger.info("✅ 全局 MemoryManager 已关闭")
|
||||
except Exception as e:
|
||||
logger.error(f"关闭 MemoryManager 时出错: {e}", exc_info=True)
|
||||
finally:
|
||||
_memory_manager = None
|
||||
_initialized = False
|
||||
|
||||
|
||||
def is_initialized() -> bool:
|
||||
"""检查 MemoryManager 是否已初始化"""
|
||||
return _initialized and _memory_manager is not None
|
||||
Reference in New Issue
Block a user