diff --git a/src/memory/index.ts b/src/memory/index.ts index 4590a1db6f..982c617be9 100644 --- a/src/memory/index.ts +++ b/src/memory/index.ts @@ -60,8 +60,18 @@ class KnowledgeGraphManager { private async saveGraph(graph: KnowledgeGraph): Promise { const lines = [ - ...graph.entities.map(e => JSON.stringify({ type: "entity", ...e })), - ...graph.relations.map(r => JSON.stringify({ type: "relation", ...r })), + ...graph.entities.map(e => JSON.stringify({ + type: "entity", + name: e.name, + entityType: e.entityType, + observations: e.observations + })), + ...graph.relations.map(r => JSON.stringify({ + type: "relation", + from: r.from, + to: r.to, + relationType: r.relationType + })), ]; await fs.writeFile(MEMORY_FILE_PATH, lines.join("\n")); } @@ -219,10 +229,12 @@ server.setRequestHandler(ListToolsRequestSchema, async () => { }, }, required: ["name", "entityType", "observations"], + additionalProperties: false, }, }, }, required: ["entities"], + additionalProperties: false, }, }, { @@ -241,10 +253,12 @@ server.setRequestHandler(ListToolsRequestSchema, async () => { relationType: { type: "string", description: "The type of the relation" }, }, required: ["from", "to", "relationType"], + additionalProperties: false, }, }, }, required: ["relations"], + additionalProperties: false, }, }, { @@ -266,10 +280,12 @@ server.setRequestHandler(ListToolsRequestSchema, async () => { }, }, required: ["entityName", "contents"], + additionalProperties: false, }, }, }, required: ["observations"], + additionalProperties: false, }, }, { @@ -285,6 +301,7 @@ server.setRequestHandler(ListToolsRequestSchema, async () => { }, }, required: ["entityNames"], + additionalProperties: false, }, }, { @@ -306,10 +323,12 @@ server.setRequestHandler(ListToolsRequestSchema, async () => { }, }, required: ["entityName", "observations"], + additionalProperties: false, }, }, }, required: ["deletions"], + additionalProperties: false, }, }, { @@ -328,11 +347,13 @@ server.setRequestHandler(ListToolsRequestSchema, async () => { relationType: { type: "string", description: "The type of the relation" }, }, required: ["from", "to", "relationType"], + additionalProperties: false, }, description: "An array of relations to delete" }, }, required: ["relations"], + additionalProperties: false, }, }, { @@ -341,6 +362,7 @@ server.setRequestHandler(ListToolsRequestSchema, async () => { inputSchema: { type: "object", properties: {}, + additionalProperties: false, }, }, { @@ -352,6 +374,7 @@ server.setRequestHandler(ListToolsRequestSchema, async () => { query: { type: "string", description: "The search query to match against entity names, types, and observation content" }, }, required: ["query"], + additionalProperties: false, }, }, { @@ -367,6 +390,7 @@ server.setRequestHandler(ListToolsRequestSchema, async () => { }, }, required: ["names"], + additionalProperties: false, }, }, ],