feat(memory): 添加配置系统支持,移除旧memory配置
重大改进: - 在 bot_config_template.toml 添加 [memory_graph] 配置段 - 移除旧的 [memory] 配置段(~85行) - 更新 MemoryGraphConfig 支持从 bot_config 加载 - 更新 MemoryManager 使用配置参数 - 更新 manager_singleton 支持配置初始化 配置系统: - enable: 启用/禁用记忆图系统 - data_dir: 数据存储目录 - 向量存储: collection_name, db_path - 检索: top_k, min_importance, similarity_threshold - 智能查询优化: enable_query_optimization - 整合: enabled, interval_hours, similarity_threshold, time_window - 遗忘: enabled, activation_threshold, min_importance - 激活: decay_rate, propagation_strength, propagation_depth - 性能: max_nodes_per_memory, max_related_memories 配置加载流程: 1. MemoryGraphConfig.from_bot_config(global_config) 2. 使用 getattr 安全获取配置值 3. 提供默认值作为后备 使用配置: - MemoryManager 从 config 读取所有参数 - maintenance 使用 consolidation_* 参数 - auto_forget_memories 使用 forgetting_* 参数 - activate_memory 使用 activation_* 参数 向后兼容: - [memory_legacy] 保留旧配置供参考 - [vector_db] 保留给其他系统使用 - 新系统使用独立 chromadb 实例 完成 Step 4: 添加配置支持
This commit is contained in:
@@ -19,12 +19,16 @@ _memory_manager: Optional[MemoryManager] = None
|
||||
_initialized: bool = False
|
||||
|
||||
|
||||
async def initialize_memory_manager(data_dir: Optional[Path | str] = None) -> MemoryManager:
|
||||
async def initialize_memory_manager(
|
||||
data_dir: Optional[Path | str] = None,
|
||||
config = None,
|
||||
) -> Optional[MemoryManager]:
|
||||
"""
|
||||
初始化全局 MemoryManager
|
||||
|
||||
Args:
|
||||
data_dir: 数据目录,默认使用 data/memory_graph
|
||||
config: MemoryGraphConfig 或 bot_config 实例
|
||||
|
||||
Returns:
|
||||
MemoryManager 实例
|
||||
@@ -36,14 +40,40 @@ async def initialize_memory_manager(data_dir: Optional[Path | str] = None) -> Me
|
||||
return _memory_manager
|
||||
|
||||
try:
|
||||
from src.memory_graph.config import MemoryGraphConfig
|
||||
|
||||
# 处理配置
|
||||
if config is None:
|
||||
# 尝试从全局配置加载
|
||||
try:
|
||||
from src.config.config import global_config
|
||||
memory_config = MemoryGraphConfig.from_bot_config(global_config)
|
||||
logger.info("从 bot_config 加载 memory_graph 配置")
|
||||
except Exception as e:
|
||||
logger.warning(f"无法从 bot_config 加载配置,使用默认配置: {e}")
|
||||
memory_config = MemoryGraphConfig()
|
||||
elif isinstance(config, MemoryGraphConfig):
|
||||
memory_config = config
|
||||
else:
|
||||
# 假设是 bot_config
|
||||
memory_config = MemoryGraphConfig.from_bot_config(config)
|
||||
|
||||
# 检查是否启用
|
||||
if not memory_config.enable:
|
||||
logger.info("记忆图系统已在配置中禁用")
|
||||
_initialized = False
|
||||
_memory_manager = None
|
||||
return None
|
||||
|
||||
# 处理数据目录
|
||||
if data_dir is None:
|
||||
data_dir = Path("data/memory_graph")
|
||||
data_dir = memory_config.data_dir
|
||||
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)
|
||||
_memory_manager = MemoryManager(config=memory_config, data_dir=data_dir)
|
||||
await _memory_manager.initialize()
|
||||
|
||||
_initialized = True
|
||||
|
||||
Reference in New Issue
Block a user