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:
Windpicker-owo
2025-11-05 20:22:08 +08:00
parent 4d44b18ac8
commit a3cc074098
4 changed files with 194 additions and 94 deletions

View File

@@ -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