|
|
#!/usr/bin/env python3
|
|
|
# -*- coding: utf-8 -*-
|
|
|
"""
|
|
|
代码示例 02: MemoryTool架构设计
|
|
|
展示MemoryTool和MemoryManager的分层架构
|
|
|
"""
|
|
|
|
|
|
from typing import List, Optional, Dict, Any
|
|
|
from datetime import datetime
|
|
|
from hello_agents.tools import MemoryTool
|
|
|
from hello_agents.memory import MemoryConfig
|
|
|
|
|
|
class MemoryToolArchitectureDemo:
|
|
|
"""MemoryTool架构演示类"""
|
|
|
|
|
|
def __init__(self):
|
|
|
self.memory_config = MemoryConfig()
|
|
|
self.memory_types = ["working", "episodic", "semantic", "perceptual"]
|
|
|
|
|
|
def demonstrate_memory_tool_init(self):
|
|
|
"""演示MemoryTool初始化过程"""
|
|
|
print("🏗️ MemoryTool架构设计演示")
|
|
|
print("=" * 50)
|
|
|
|
|
|
print("📋 MemoryTool初始化过程:")
|
|
|
print("1. 创建MemoryConfig配置对象")
|
|
|
print("2. 指定启用的记忆类型")
|
|
|
print("3. 初始化MemoryManager管理器")
|
|
|
print("4. 根据配置启用不同记忆模块")
|
|
|
|
|
|
# 演示MemoryTool的初始化
|
|
|
memory_tool = MemoryTool(
|
|
|
user_id="architecture_demo_user",
|
|
|
memory_config=self.memory_config,
|
|
|
memory_types=self.memory_types
|
|
|
)
|
|
|
|
|
|
print(f"\n✅ MemoryTool初始化完成")
|
|
|
print(f"👤 用户ID: {self.user_id}")
|
|
|
print(f"🧠 启用的记忆类型: {memory_tool.memory_types}")
|
|
|
print(f"⚙️ 配置对象: {type(memory_tool.memory_config).__name__}")
|
|
|
|
|
|
return memory_tool
|
|
|
|
|
|
def demonstrate_memory_manager_architecture(self, memory_tool):
|
|
|
"""演示MemoryManager的组合模式架构"""
|
|
|
print("\n🔧 MemoryManager架构设计")
|
|
|
print("-" * 40)
|
|
|
|
|
|
print("MemoryManager采用组合模式设计:")
|
|
|
print("- 统一的记忆操作接口")
|
|
|
print("- 独立的记忆类型组件")
|
|
|
print("- 灵活的配置和扩展能力")
|
|
|
|
|
|
# 获取MemoryManager实例
|
|
|
memory_manager = memory_tool.memory_manager
|
|
|
|
|
|
print(f"\n📊 MemoryManager状态:")
|
|
|
print(f"用户ID: {memory_manager.user_id}")
|
|
|
print(f"配置类型: {type(memory_manager.config).__name__}")
|
|
|
print(f"记忆类型数量: {len(memory_manager.memory_types)}")
|
|
|
|
|
|
# 显示各记忆类型的状态
|
|
|
print(f"\n🧠 记忆类型组件:")
|
|
|
for memory_type, memory_instance in memory_manager.memory_types.items():
|
|
|
print(f" • {memory_type}: {type(memory_instance).__name__}")
|
|
|
|
|
|
def demonstrate_memory_types_specialization(self, memory_tool):
|
|
|
"""演示四种记忆类型的专业化特点"""
|
|
|
print("\n🎯 四种记忆类型的专业化设计")
|
|
|
print("-" * 40)
|
|
|
|
|
|
memory_types_info = {
|
|
|
"working": {
|
|
|
"name": "工作记忆",
|
|
|
"features": ["容量有限", "访问速度快", "自动清理", "临时存储"],
|
|
|
"storage": "纯内存存储",
|
|
|
"ttl": "60分钟TTL机制"
|
|
|
},
|
|
|
"episodic": {
|
|
|
"name": "情景记忆",
|
|
|
"features": ["事件序列", "时间序列", "上下文丰富", "会话关联"],
|
|
|
"storage": "SQLite + Qdrant混合存储",
|
|
|
"ttl": "持久化存储"
|
|
|
},
|
|
|
"semantic": {
|
|
|
"name": "语义记忆",
|
|
|
"features": ["概念知识", "实体关系", "知识图谱", "语义推理"],
|
|
|
"storage": "Neo4j + Qdrant混合存储",
|
|
|
"ttl": "长期存储"
|
|
|
},
|
|
|
"perceptual": {
|
|
|
"name": "感知记忆",
|
|
|
"features": ["多模态", "跨模态检索", "感知数据", "内容生成"],
|
|
|
"storage": "分模态向量存储",
|
|
|
"ttl": "按重要性管理"
|
|
|
}
|
|
|
}
|
|
|
|
|
|
for memory_type, info in memory_types_info.items():
|
|
|
print(f"\n📚 {info['name']} ({memory_type}):")
|
|
|
print(f" 特点: {', '.join(info['features'])}")
|
|
|
print(f" 存储: {info['storage']}")
|
|
|
print(f" 生命周期: {info['ttl']}")
|
|
|
|
|
|
# 添加示例记忆来演示特点
|
|
|
if memory_type == "working":
|
|
|
memory_tool.execute("add",
|
|
|
content=f"演示{info['name']}的临时存储特性",
|
|
|
memory_type=memory_type,
|
|
|
importance=0.6,
|
|
|
demo_feature="temporary_storage"
|
|
|
)
|
|
|
elif memory_type == "episodic":
|
|
|
memory_tool.execute("add",
|
|
|
content=f"演示{info['name']}的事件记录特性",
|
|
|
memory_type=memory_type,
|
|
|
importance=0.7,
|
|
|
event_type="demonstration",
|
|
|
session_context="architecture_demo"
|
|
|
)
|
|
|
elif memory_type == "semantic":
|
|
|
memory_tool.execute("add",
|
|
|
content=f"{info['name']}用于存储概念性知识和实体关系",
|
|
|
memory_type=memory_type,
|
|
|
importance=0.8,
|
|
|
concept="memory_architecture",
|
|
|
domain="cognitive_computing"
|
|
|
)
|
|
|
elif memory_type == "perceptual":
|
|
|
memory_tool.execute("add",
|
|
|
content=f"演示{info['name']}的多模态数据处理",
|
|
|
memory_type=memory_type,
|
|
|
importance=0.6,
|
|
|
modality="text",
|
|
|
data_type="demonstration"
|
|
|
)
|
|
|
|
|
|
def demonstrate_unified_interface(self, memory_tool):
|
|
|
"""演示统一接口的设计优势"""
|
|
|
print("\n🔗 统一接口设计优势")
|
|
|
print("-" * 40)
|
|
|
|
|
|
print("统一的execute方法提供:")
|
|
|
print("• 一致的调用方式")
|
|
|
print("• 灵活的参数传递")
|
|
|
print("• 统一的错误处理")
|
|
|
print("• 简化的用户体验")
|
|
|
|
|
|
# 演示统一接口的使用
|
|
|
operations = [
|
|
|
("search", {"query": "演示", "limit": 2}),
|
|
|
("summary", {"limit": 3}),
|
|
|
("stats", {}),
|
|
|
]
|
|
|
|
|
|
print(f"\n🔧 统一接口操作演示:")
|
|
|
for operation, params in operations:
|
|
|
print(f"\n操作: {operation}")
|
|
|
print(f"参数: {params}")
|
|
|
result = memory_tool.execute(operation, **params)
|
|
|
print(f"结果: {result[:100]}..." if len(str(result)) > 100 else f"结果: {result}")
|
|
|
|
|
|
def demonstrate_extensibility(self):
|
|
|
"""演示系统的扩展性设计"""
|
|
|
print("\n🚀 系统扩展性设计")
|
|
|
print("-" * 40)
|
|
|
|
|
|
print("扩展性特点:")
|
|
|
print("• 插件化的记忆类型")
|
|
|
print("• 可配置的存储后端")
|
|
|
print("• 灵活的记忆策略")
|
|
|
print("• 模块化的组件设计")
|
|
|
|
|
|
# 演示自定义配置
|
|
|
custom_config = MemoryConfig()
|
|
|
custom_config.working_memory_capacity = 100
|
|
|
custom_config.working_memory_ttl = 120
|
|
|
|
|
|
print(f"\n⚙️ 自定义配置示例:")
|
|
|
print(f"工作记忆容量: {custom_config.working_memory_capacity}")
|
|
|
print(f"工作记忆TTL: {custom_config.working_memory_ttl}分钟")
|
|
|
|
|
|
# 演示选择性启用记忆类型
|
|
|
selective_memory_tool = MemoryTool(
|
|
|
user_id="selective_user",
|
|
|
memory_config=custom_config,
|
|
|
memory_types=["working", "semantic"] # 只启用部分类型
|
|
|
)
|
|
|
|
|
|
print(f"\n🎯 选择性启用示例:")
|
|
|
print(f"启用的记忆类型: {selective_memory_tool.memory_types}")
|
|
|
print("✅ 系统支持根据需求灵活配置")
|
|
|
|
|
|
def main():
|
|
|
"""主函数"""
|
|
|
print("🏗️ MemoryTool架构设计完整演示")
|
|
|
print("展示记忆系统的分层架构和设计模式")
|
|
|
print("=" * 60)
|
|
|
|
|
|
try:
|
|
|
demo = MemoryToolArchitectureDemo()
|
|
|
|
|
|
# 1. MemoryTool初始化演示
|
|
|
memory_tool = demo.demonstrate_memory_tool_init()
|
|
|
|
|
|
# 2. MemoryManager架构演示
|
|
|
demo.demonstrate_memory_manager_architecture(memory_tool)
|
|
|
|
|
|
# 3. 记忆类型专业化演示
|
|
|
demo.demonstrate_memory_types_specialization(memory_tool)
|
|
|
|
|
|
# 4. 统一接口演示
|
|
|
demo.demonstrate_unified_interface(memory_tool)
|
|
|
|
|
|
# 5. 扩展性演示
|
|
|
demo.demonstrate_extensibility()
|
|
|
|
|
|
print("\n" + "=" * 60)
|
|
|
print("🎉 MemoryTool架构演示完成！")
|
|
|
print("=" * 60)
|
|
|
|
|
|
print("\n✨ 架构设计亮点:")
|
|
|
print("1. 🏗️ 分层架构 - 关注点分离，职责清晰")
|
|
|
print("2. 🔧 组合模式 - 灵活组合，独立管理")
|
|
|
print("3. 🎯 专业化设计 - 各记忆类型特点鲜明")
|
|
|
print("4. 🔗 统一接口 - 简化使用，一致体验")
|
|
|
print("5. 🚀 高扩展性 - 插件化设计，灵活配置")
|
|
|
|
|
|
print("\n🎯 设计原则:")
|
|
|
print("• 单一职责原则 - 每个组件专注特定功能")
|
|
|
print("• 开闭原则 - 对扩展开放，对修改封闭")
|
|
|
print("• 依赖倒置原则 - 依赖抽象，不依赖具体")
|
|
|
print("• 组合优于继承 - 灵活组合，避免复杂继承")
|
|
|
|
|
|
except Exception as e:
|
|
|
print(f"\n❌ 演示过程中发生错误: {e}")
|
|
|
import traceback
|
|
|
traceback.print_exc()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
main() |