内存MCP服务器
一种模型上下文协议（MCP）服务器，提供知识图功能，用于管理内存中的实体、关系和观察，并具有严格的验证规则来保持数据一致性。
安装
在Claude Desktop中安装服务器：
mcp install main.py -v MEMORY_FILE_PATH=/path/to/memory.jsonl数据验证规则
实体名称
- 必须以小写字母开头
- 可以包含小写字母、数字和连字符
- 最大长度为100个字符
- 在图中必须是唯一的
- 有效名称示例：
python-project,meeting-notes-2024,user-john
实体类型
支持以下实体类型：
person：人类实体concept：抽象的思想或原则project：工作倡议或任务document：任何形式的文件tool：软件工具或实用程序organization：公司或团体location：物理或虚拟场所event：有时限的事件
观察
- 非空字符串
- 最大长度为500个字符
- 每个实体必须是唯一的
- 应该是事实和客观的陈述
- 相关时包括时间戳
关系
支持以下关系类型：
knows：人与人之间的联系contains：亲子关系uses：利用另一实体的实体created：作者/创作关系belongs-to：成员/所有权depends-on：依赖关系related-to：一般关系
附加关系规则：
- 源实体和目标实体都必须存在
- 不允许有自我参照关系
- 不允许循环依赖
- 必须使用预定义的关系类型
用法
服务器提供用于管理知识图的工具：
获取实体
result = await session.call_tool("get_entity", {
"entity_name": "example"
})
if not result.success:
if result.error_type == "NOT_FOUND":
print(f"Entity not found: {result.error}")
elif result.error_type == "VALIDATION_ERROR":
print(f"Invalid input: {result.error}")
else:
print(f"Error: {result.error}")
else:
entity = result.data
print(f"Found entity: {entity}")获取图表
result = await session.call_tool("get_graph", {})
if result.success:
graph = result.data
print(f"Graph data: {graph}")
else:
print(f"Error retrieving graph: {result.error}")创建实体
# Valid entity creation
entities = [
Entity(
name="python-project", # Lowercase with hyphens
entityType="project", # Must be a valid type
observations=["Started development on 2024-01-29"]
),
Entity(
name="john-doe",
entityType="person",
observations=["Software engineer", "Joined team in 2024"]
)
]
result = await session.call_tool("create_entities", {
"entities": entities
})
if not result.success:
if result.error_type == "VALIDATION_ERROR":
print(f"Invalid entity data: {result.error}")
else:
print(f"Error creating entities: {result.error}")添加观察
# Valid observation
result = await session.call_tool("add_observation", {
"entity": "python-project",
"observation": "Completed initial prototype" # Must be unique for entity
})
if not result.success:
if result.error_type == "NOT_FOUND":
print(f"Entity not found: {result.error}")
elif result.error_type == "VALIDATION_ERROR":
print(f"Invalid observation: {result.error}")
else:
print(f"Error adding observation: {result.error}")创建关系
# Valid relation
result = await session.call_tool("create_relation", {
"from_entity": "john-doe",
"to_entity": "python-project",
"relation_type": "created" # Must be a valid type
})
if not result.success:
if result.error_type == "NOT_FOUND":
print(f"Entity not found: {result.error}")
elif result.error_type == "VALIDATION_ERROR":
print(f"Invalid relation data: {result.error}")
else:
print(f"Error creating relation: {result.error}")搜索内存
result = await session.call_tool("search_memory", {
"query": "most recent workout" # Supports natural language queries
})
if result.success:
if result.error_type == "NO_RESULTS":
print(f"No results found: {result.error}")
else:
results = result.data
print(f"Search results: {results}")
else:
print(f"Error searching memory: {result.error}")搜索功能支持：
- 时态查询（例如，“最近”、“最后”、“最新”）
- 活动查询（例如，“锻炼”、“锻炼”）
- 一般实体搜索
- 80%相似度阈值的模糊匹配
- 加权搜索范围：
- 实体名称（权重：1.0） - 实体类型（权重：0.8） - 观察结果（权重：0.6）
删除实体
result = await session.call_tool("delete_entities", {
"names": ["python-project", "john-doe"]
})
if not result.success:
if result.error_type == "NOT_FOUND":
print(f"Entity not found: {result.error}")
else:
print(f"Error deleting entities: {result.error}")删除关系
result = await session.call_tool("delete_relation", {
"from_entity": "john-doe",
"to_entity": "python-project"
})
if not result.success:
if result.error_type == "NOT_FOUND":
print(f"Entity not found: {result.error}")
else:
print(f"Error deleting relation: {result.error}")刷新内存
result = await session.call_tool("flush_memory", {})
if not result.success:
print(f"Error flushing memory: {result.error}")错误类型
服务器使用以下错误类型：
NOT_FOUND：找不到实体或资源VALIDATION_ERROR：输入数据无效INTERNAL_ERROR：服务器端错误ALREADY_EXISTS：资源已存在INVALID_RELATION：实体之间的关系无效
响应模型
所有工具都使用以下模型返回键入的响应：
EntityResponse
class EntityResponse(BaseModel):
success: bool
data: Optional[Dict[str, Any]] = None
error: Optional[str] = None
error_type: Optional[str] = NoneGraphResponse
class GraphResponse(BaseModel):
success: bool
data: Optional[Dict[str, Any]] = None
error: Optional[str] = None
error_type: Optional[str] = None操作响应
class OperationResponse(BaseModel):
success: bool
error: Optional[str] = None
error_type: Optional[str] = None发展
运行测试
pytest tests/添加新功能
- 更新中的验证规则
validation.py - 在中添加测试
tests/test_validation.py - 实施变更
knowledge_graph_manager.py