@eucoder/rag 0.2.0 → 0.3.0

package/dist/neo4j-knowledge-graph.test.js

@@ -0,0 +1,219 @@
+import { describe, it, expect, beforeAll, afterAll } from "vitest";
+import { Neo4jKnowledgeGraph } from "./neo4j-knowledge-graph.js";
+// Configurazione Neo4j per test (richiede Neo4j in esecuzione)
+const testConfig = {
+    uri: process.env.NEO4J_URI || "bolt://localhost:7687",
+    username: process.env.NEO4J_USER || "neo4j",
+    password: process.env.NEO4J_PASSWORD || "test1234",
+    database: "neo4j",
+};
+// Verifica se Neo4j è disponibile prima di eseguire i test
+let neo4jAvailable = false;
+describe("Neo4jKnowledgeGraph", () => {
+    let graph;
+    beforeAll(async () => {
+        graph = new Neo4jKnowledgeGraph(testConfig);
+        // Verifica connessione con timeout
+        try {
+            const connected = await Promise.race([
+                graph.verifyConnectivity(),
+                new Promise((_, reject) => setTimeout(() => reject(new Error("Timeout")), 3000)),
+            ]);
+            neo4jAvailable = connected === true;
+            if (!neo4jAvailable) {
+                console.log("⚠️  Neo4j non disponibile, test skippati");
+                return;
+            }
+            // Inizializza schema
+            await graph.initializeSchema();
+            // Pulisci grafo prima dei test
+            await graph.clear();
+        }
+        catch (error) {
+            console.log("⚠️  Neo4j non raggiungibile, test skippati");
+            neo4jAvailable = false;
+        }
+    });
+    afterAll(async () => {
+        if (!neo4jAvailable)
+            return;
+        // Pulisci grafo dopo i test
+        await graph.clear();
+        await graph.close();
+    });
+    it("should add and retrieve an entity", async () => {
+        if (!neo4jAvailable) {
+            console.log("⏭️  Test skippato: Neo4j non disponibile");
+            return;
+        }
+        const entity = {
+            id: "user-1",
+            name: "Alice",
+            type: "User",
+            properties: { age: 30, role: "developer" },
+        };
+        await graph.addEntity(entity);
+        const retrieved = await graph.getEntity("user-1");
+        expect(retrieved).toBeDefined();
+        expect(retrieved?.id).toBe("user-1");
+        expect(retrieved?.name).toBe("Alice");
+        expect(retrieved?.type).toBe("User");
+        expect(retrieved?.properties.age).toBe(30);
+    });
+    it("should add and retrieve a relation", async () => {
+        if (!neo4jAvailable) {
+            console.log("⏭️  Test skippato: Neo4j non disponibile");
+            return;
+        }
+        const user1 = {
+            id: "user-2",
+            name: "Bob",
+            type: "User",
+            properties: {},
+        };
+        const user2 = {
+            id: "user-3",
+            name: "Charlie",
+            type: "User",
+            properties: {},
+        };
+        await graph.addEntity(user1);
+        await graph.addEntity(user2);
+        const relation = {
+            id: "rel-1",
+            source: "user-2",
+            target: "user-3",
+            type: "KNOWS",
+            properties: { since: 2020 },
+        };
+        await graph.addRelation(relation);
+        const relations = await graph.getRelations("user-2");
+        expect(relations).toHaveLength(1);
+        expect(relations[0].type).toBe("KNOWS");
+        expect(relations[0].properties.since).toBe(2020);
+    });
+    it("should retrieve connected entities", async () => {
+        if (!neo4jAvailable) {
+            console.log("⏭️  Test skippato: Neo4j non disponibile");
+            return;
+        }
+        const user1 = {
+            id: "user-4",
+            name: "David",
+            type: "User",
+            properties: {},
+        };
+        const user2 = {
+            id: "user-5",
+            name: "Eve",
+            type: "User",
+            properties: {},
+        };
+        const user3 = {
+            id: "user-6",
+            name: "Frank",
+            type: "User",
+            properties: {},
+        };
+        await graph.addEntity(user1);
+        await graph.addEntity(user2);
+        await graph.addEntity(user3);
+        await graph.addRelation({
+            id: "rel-2",
+            source: "user-4",
+            target: "user-5",
+            type: "KNOWS",
+            properties: {},
+        });
+        await graph.addRelation({
+            id: "rel-3",
+            source: "user-5",
+            target: "user-6",
+            type: "KNOWS",
+            properties: {},
+        });
+        // Depth 1: solo user-5
+        const connected1 = await graph.getConnectedEntities("user-4", 1);
+        expect(connected1).toHaveLength(1);
+        expect(connected1[0].id).toBe("user-5");
+        // Depth 2: user-5 e user-6
+        const connected2 = await graph.getConnectedEntities("user-4", 2);
+        expect(connected2).toHaveLength(2);
+        expect(connected2.map((e) => e.id)).toContain("user-5");
+        expect(connected2.map((e) => e.id)).toContain("user-6");
+    });
+    it("should search entities by name", async () => {
+        if (!neo4jAvailable) {
+            console.log("⏭️  Test skippato: Neo4j non disponibile");
+            return;
+        }
+        const results = await graph.searchEntities({ name: "Alice" });
+        expect(results.length).toBeGreaterThan(0);
+        expect(results.some((e) => e.name === "Alice")).toBe(true);
+    });
+    it("should search entities by type", async () => {
+        if (!neo4jAvailable) {
+            console.log("⏭️  Test skippato: Neo4j non disponibile");
+            return;
+        }
+        const results = await graph.searchEntities({ type: "User" });
+        expect(results.length).toBeGreaterThan(0);
+        expect(results.every((e) => e.type === "User")).toBe(true);
+    });
+    it("should delete an entity and its relations", async () => {
+        if (!neo4jAvailable) {
+            console.log("⏭️  Test skippato: Neo4j non disponibile");
+            return;
+        }
+        const entity = {
+            id: "user-to-delete",
+            name: "ToDelete",
+            type: "User",
+            properties: {},
+        };
+        await graph.addEntity(entity);
+        await graph.deleteEntity("user-to-delete");
+        const retrieved = await graph.getEntity("user-to-delete");
+        expect(retrieved).toBeUndefined();
+    });
+    it("should delete a relation", async () => {
+        if (!neo4jAvailable) {
+            console.log("⏭️  Test skippato: Neo4j non disponibile");
+            return;
+        }
+        const user1 = {
+            id: "user-7",
+            name: "Grace",
+            type: "User",
+            properties: {},
+        };
+        const user2 = {
+            id: "user-8",
+            name: "Henry",
+            type: "User",
+            properties: {},
+        };
+        await graph.addEntity(user1);
+        await graph.addEntity(user2);
+        await graph.addRelation({
+            id: "rel-to-delete",
+            source: "user-7",
+            target: "user-8",
+            type: "KNOWS",
+            properties: {},
+        });
+        await graph.deleteRelation("rel-to-delete");
+        const relations = await graph.getRelations("user-7");
+        expect(relations.find((r) => r.id === "rel-to-delete")).toBeUndefined();
+    });
+    it("should get graph statistics", async () => {
+        if (!neo4jAvailable) {
+            console.log("⏭️  Test skippato: Neo4j non disponibile");
+            return;
+        }
+        const stats = await graph.getStats();
+        expect(stats.entityCount).toBeGreaterThan(0);
+        expect(stats.relationCount).toBeGreaterThan(0);
+    });
+});
+//# sourceMappingURL=neo4j-knowledge-graph.test.js.map

package/dist/neo4j-knowledge-graph.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"neo4j-knowledge-graph.test.js","sourceRoot":"","sources":["../src/neo4j-knowledge-graph.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,QAAQ,CAAC;AACnE,OAAO,EAAE,mBAAmB,EAAE,MAAM,4BAA4B,CAAC;AAGjE,+DAA+D;AAC/D,MAAM,UAAU,GAAgB;IAC9B,GAAG,EAAE,OAAO,CAAC,GAAG,CAAC,SAAS,IAAI,uBAAuB;IACrD,QAAQ,EAAE,OAAO,CAAC,GAAG,CAAC,UAAU,IAAI,OAAO;IAC3C,QAAQ,EAAE,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,UAAU;IAClD,QAAQ,EAAE,OAAO;CAClB,CAAC;AAEF,2DAA2D;AAC3D,IAAI,cAAc,GAAG,KAAK,CAAC;AAE3B,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,IAAI,KAA0B,CAAC;IAE/B,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,KAAK,GAAG,IAAI,mBAAmB,CAAC,UAAU,CAAC,CAAC;QAE5C,mCAAmC;QACnC,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC;gBACnC,KAAK,CAAC,kBAAkB,EAAE;gBAC1B,IAAI,OAAO,CAAU,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,CACjC,UAAU,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC,EAAE,IAAI,CAAC,CACrD;aACF,CAAC,CAAC;YAEH,cAAc,GAAG,SAAS,KAAK,IAAI,CAAC;YAEpC,IAAI,CAAC,cAAc,EAAE,CAAC;gBACpB,OAAO,CAAC,GAAG,CAAC,0CAA0C,CAAC,CAAC;gBACxD,OAAO;YACT,CAAC;YAED,qBAAqB;YACrB,MAAM,KAAK,CAAC,gBAAgB,EAAE,CAAC;YAE/B,+BAA+B;YAC/B,MAAM,KAAK,CAAC,KAAK,EAAE,CAAC;QACtB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,GAAG,CAAC,4CAA4C,CAAC,CAAC;YAC1D,cAAc,GAAG,KAAK,CAAC;QACzB,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,KAAK,IAAI,EAAE;QAClB,IAAI,CAAC,cAAc;YAAE,OAAO;QAE5B,4BAA4B;QAC5B,MAAM,KAAK,CAAC,KAAK,EAAE,CAAC;QACpB,MAAM,KAAK,CAAC,KAAK,EAAE,CAAC;IACtB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;QACjD,IAAI,CAAC,cAAc,EAAE,CAAC;YACpB,OAAO,CAAC,GAAG,CAAC,0CAA0C,CAAC,CAAC;YACxD,OAAO;QACT,CAAC;QAED,MAAM,MAAM,GAAG;YACb,EAAE,EAAE,QAAQ;YACZ,IAAI,EAAE,OAAO;YACb,IAAI,EAAE,MAAM;YACZ,UAAU,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE;SAC3C,CAAC;QAEF,MAAM,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAC9B,MAAM,SAAS,GAAG,MAAM,KAAK,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QAElD,MAAM,CAAC,SAAS,CAAC,CAAC,WAAW,EAAE,CAAC;QAChC,MAAM,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACrC,MAAM,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACtC,MAAM,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACrC,MAAM,CAAC,SAAS,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;QAClD,IAAI,CAAC,cAAc,EAAE,CAAC;YACpB,OAAO,CAAC,GAAG,CAAC,0CAA0C,CAAC,CAAC;YACxD,OAAO;QACT,CAAC;QAED,MAAM,KAAK,GAAG;YACZ,EAAE,EAAE,QAAQ;YACZ,IAAI,EAAE,KAAK;YACX,IAAI,EAAE,MAAM;YACZ,UAAU,EAAE,EAAE;SACf,CAAC;QAEF,MAAM,KAAK,GAAG;YACZ,EAAE,EAAE,QAAQ;YACZ,IAAI,EAAE,SAAS;YACf,IAAI,EAAE,MAAM;YACZ,UAAU,EAAE,EAAE;SACf,CAAC;QAEF,MAAM,KAAK,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAC7B,MAAM,KAAK,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAE7B,MAAM,QAAQ,GAAG;YACf,EAAE,EAAE,OAAO;YACX,MAAM,EAAE,QAAQ;YAChB,MAAM,EAAE,QAAQ;YAChB,IAAI,EAAE,OAAO;YACb,UAAU,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE;SAC5B,CAAC;QAEF,MAAM,KAAK,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;QAElC,MAAM,SAAS,GAAG,MAAM,KAAK,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;QACrD,MAAM,CAAC,SAAS,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAClC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACxC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;QAClD,IAAI,CAAC,cAAc,EAAE,CAAC;YACpB,OAAO,CAAC,GAAG,CAAC,0CAA0C,CAAC,CAAC;YACxD,OAAO;QACT,CAAC;QAED,MAAM,KAAK,GAAG;YACZ,EAAE,EAAE,QAAQ;YACZ,IAAI,EAAE,OAAO;YACb,IAAI,EAAE,MAAM;YACZ,UAAU,EAAE,EAAE;SACf,CAAC;QAEF,MAAM,KAAK,GAAG;YACZ,EAAE,EAAE,QAAQ;YACZ,IAAI,EAAE,KAAK;YACX,IAAI,EAAE,MAAM;YACZ,UAAU,EAAE,EAAE;SACf,CAAC;QAEF,MAAM,KAAK,GAAG;YACZ,EAAE,EAAE,QAAQ;YACZ,IAAI,EAAE,OAAO;YACb,IAAI,EAAE,MAAM;YACZ,UAAU,EAAE,EAAE;SACf,CAAC;QAEF,MAAM,KAAK,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAC7B,MAAM,KAAK,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAC7B,MAAM,KAAK,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAE7B,MAAM,KAAK,CAAC,WAAW,CAAC;YACtB,EAAE,EAAE,OAAO;YACX,MAAM,EAAE,QAAQ;YAChB,MAAM,EAAE,QAAQ;YAChB,IAAI,EAAE,OAAO;YACb,UAAU,EAAE,EAAE;SACf,CAAC,CAAC;QAEH,MAAM,KAAK,CAAC,WAAW,CAAC;YACtB,EAAE,EAAE,OAAO;YACX,MAAM,EAAE,QAAQ;YAChB,MAAM,EAAE,QAAQ;YAChB,IAAI,EAAE,OAAO;YACb,UAAU,EAAE,EAAE;SACf,CAAC,CAAC;QAEH,uBAAuB;QACvB,MAAM,UAAU,GAAG,MAAM,KAAK,CAAC,oBAAoB,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC;QACjE,MAAM,CAAC,UAAU,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACnC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAExC,2BAA2B;QAC3B,MAAM,UAAU,GAAG,MAAM,KAAK,CAAC,oBAAoB,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC;QACjE,MAAM,CAAC,UAAU,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACnC,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QACxD,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;QAC9C,IAAI,CAAC,cAAc,EAAE,CAAC;YACpB,OAAO,CAAC,GAAG,CAAC,0CAA0C,CAAC,CAAC;YACxD,OAAO;QACT,CAAC;QAED,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,cAAc,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;QAC9D,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAC1C,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC7D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;QAC9C,IAAI,CAAC,cAAc,EAAE,CAAC;YACpB,OAAO,CAAC,GAAG,CAAC,0CAA0C,CAAC,CAAC;YACxD,OAAO;QACT,CAAC;QAED,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,cAAc,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;QAC7D,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAC1C,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC7D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;QACzD,IAAI,CAAC,cAAc,EAAE,CAAC;YACpB,OAAO,CAAC,GAAG,CAAC,0CAA0C,CAAC,CAAC;YACxD,OAAO;QACT,CAAC;QAED,MAAM,MAAM,GAAG;YACb,EAAE,EAAE,gBAAgB;YACpB,IAAI,EAAE,UAAU;YAChB,IAAI,EAAE,MAAM;YACZ,UAAU,EAAE,EAAE;SACf,CAAC;QAEF,MAAM,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAC9B,MAAM,KAAK,CAAC,YAAY,CAAC,gBAAgB,CAAC,CAAC;QAE3C,MAAM,SAAS,GAAG,MAAM,KAAK,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;QAC1D,MAAM,CAAC,SAAS,CAAC,CAAC,aAAa,EAAE,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;QACxC,IAAI,CAAC,cAAc,EAAE,CAAC;YACpB,OAAO,CAAC,GAAG,CAAC,0CAA0C,CAAC,CAAC;YACxD,OAAO;QACT,CAAC;QAED,MAAM,KAAK,GAAG;YACZ,EAAE,EAAE,QAAQ;YACZ,IAAI,EAAE,OAAO;YACb,IAAI,EAAE,MAAM;YACZ,UAAU,EAAE,EAAE;SACf,CAAC;QAEF,MAAM,KAAK,GAAG;YACZ,EAAE,EAAE,QAAQ;YACZ,IAAI,EAAE,OAAO;YACb,IAAI,EAAE,MAAM;YACZ,UAAU,EAAE,EAAE;SACf,CAAC;QAEF,MAAM,KAAK,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAC7B,MAAM,KAAK,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAE7B,MAAM,KAAK,CAAC,WAAW,CAAC;YACtB,EAAE,EAAE,eAAe;YACnB,MAAM,EAAE,QAAQ;YAChB,MAAM,EAAE,QAAQ;YAChB,IAAI,EAAE,OAAO;YACb,UAAU,EAAE,EAAE;SACf,CAAC,CAAC;QAEH,MAAM,KAAK,CAAC,cAAc,CAAC,eAAe,CAAC,CAAC;QAE5C,MAAM,SAAS,GAAG,MAAM,KAAK,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;QACrD,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,eAAe,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;IAC1E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;QAC3C,IAAI,CAAC,cAAc,EAAE,CAAC;YACpB,OAAO,CAAC,GAAG,CAAC,0CAA0C,CAAC,CAAC;YACxD,OAAO;QACT,CAAC;QAED,MAAM,KAAK,GAAG,MAAM,KAAK,CAAC,QAAQ,EAAE,CAAC;QACrC,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAC7C,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/types.d.ts

@@ -72,6 +72,25 @@ export interface KnowledgeGraph {
     getRelations(entityId: string): Relation[];
     getConnectedEntities(entityId: string, depth?: number): Entity[];
 }
+export interface AsyncKnowledgeGraph {
+    addEntity(entity: Entity): Promise<void>;
+    addRelation(relation: Relation): Promise<void>;
+    getEntity(id: string): Promise<Entity | undefined>;
+    getRelations(entityId: string): Promise<Relation[]>;
+    getConnectedEntities(entityId: string, depth?: number): Promise<Entity[]>;
+    searchEntities(query: {
+        name?: string;
+        type?: string;
+    }): Promise<Entity[]>;
+    deleteEntity(entityId: string): Promise<void>;
+    deleteRelation(relationId: string): Promise<void>;
+    getStats(): Promise<{
+        entityCount: number;
+        relationCount: number;
+    }>;
+    clear(): Promise<void>;
+    close(): Promise<void>;
+}
 export interface EntityExtractor {
     extract(text: string): Promise<Entity[]>;
 }

package/dist/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAEjD,MAAM,WAAW,SAAS;IACxB,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC;CAC3D;AAED,MAAM,WAAW,gBAAgB;IAC/B,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC;CAC3D;AAED,MAAM,WAAW,YAAY;IAC3B,QAAQ,EAAE,OAAO,CAAC;IAClB,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,MAAM;IACrB,KAAK,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,SAAS,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC;CAC7D;AAED,MAAM,WAAW,aAAa;IAC5B,OAAO,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;CAC1D;AAED,MAAM,WAAW,oBAAoB;IACnC,SAAS,EAAE,SAAS,CAAC;IACrB,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,aAAa,CAAC;IACxB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,WAAW,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC;CACvD;AAED,MAAM,WAAW,WAAW;IAC1B,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,MAAM,CAAC;CACd;AAED,MAAM,WAAW,SAAS;IACxB,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,WAAW,EAAE,CAAC;IACzB,KAAK,EAAE,MAAM,EAAE,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;CAClB;AAGD,MAAM,WAAW,mBAAmB;IAClC,iBAAiB,EAAE,SAAS,CAAC;IAC7B,gBAAgB,EAAE,gBAAgB,CAAC;IACnC,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,kBAAkB;IACjC,IAAI,EAAE,SAAS,EAAE,CAAC;IAClB,YAAY,EAAE,KAAK,GAAG,UAAU,CAAC;IACjC,aAAa,EAAE,MAAM,CAAC;IACtB,YAAY,EAAE,MAAM,CAAC;CACtB;AAGD,MAAM,WAAW,MAAM;IACrB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACrC;AAED,MAAM,WAAW,QAAQ;IACvB,EAAE,EAAE,MAAM,CAAC;IACX,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACrC;AAED,MAAM,WAAW,cAAc;IAC7B,QAAQ,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC9B,SAAS,EAAE,QAAQ,EAAE,CAAC;IACtB,SAAS,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI,CAAC;IAChC,WAAW,CAAC,QAAQ,EAAE,QAAQ,GAAG,IAAI,CAAC;IACtC,SAAS,CAAC,EAAE,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,CAAC;IAC1C,YAAY,CAAC,QAAQ,EAAE,MAAM,GAAG,QAAQ,EAAE,CAAC;IAC3C,oBAAoB,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,MAAM,GAAG,MAAM,EAAE,CAAC;CAClE;AAED,MAAM,WAAW,eAAe;IAC9B,OAAO,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;CAC1C;AAED,MAAM,WAAW,iBAAiB;IAChC,OAAO,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC;CAChE;AAED,MAAM,WAAW,wBAAwB;IACvC,SAAS,EAAE,SAAS,CAAC;IACrB,eAAe,EAAE,eAAe,CAAC;IACjC,iBAAiB,EAAE,iBAAiB,CAAC;IACrC,KAAK,EAAE,cAAc,CAAC;IACtB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,uBAAuB;IACtC,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,WAAW,EAAE,CAAC;IACzB,QAAQ,EAAE,MAAM,EAAE,CAAC;IACnB,SAAS,EAAE,QAAQ,EAAE,CAAC;IACtB,YAAY,EAAE,MAAM,CAAC;CACtB;AAGD,MAAM,WAAW,aAAa;IAC5B,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,EAAE,MAAM,CAAC;IACf,WAAW,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,iBAAiB;IAChC,SAAS,EAAE,SAAS,CAAC;IACrB,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,aAAa,CAAC;IACxB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,oBAAoB,CAAC,EAAE,OAAO,CAAC;IAC/B,oBAAoB,CAAC,EAAE,OAAO,CAAC;CAChC;AAED,MAAM,WAAW,gBAAgB;IAC/B,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,WAAW,EAAE,CAAC;IACzB,cAAc,EAAE,aAAa,EAAE,CAAC;IAChC,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;CACpB"}
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAEjD,MAAM,WAAW,SAAS;IACxB,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC;CAC3D;AAED,MAAM,WAAW,gBAAgB;IAC/B,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC;CAC3D;AAED,MAAM,WAAW,YAAY;IAC3B,QAAQ,EAAE,OAAO,CAAC;IAClB,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,MAAM;IACrB,KAAK,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,SAAS,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC;CAC7D;AAED,MAAM,WAAW,aAAa;IAC5B,OAAO,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;CAC1D;AAED,MAAM,WAAW,oBAAoB;IACnC,SAAS,EAAE,SAAS,CAAC;IACrB,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,aAAa,CAAC;IACxB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,WAAW,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC;CACvD;AAED,MAAM,WAAW,WAAW;IAC1B,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,MAAM,CAAC;CACd;AAED,MAAM,WAAW,SAAS;IACxB,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,WAAW,EAAE,CAAC;IACzB,KAAK,EAAE,MAAM,EAAE,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;CAClB;AAGD,MAAM,WAAW,mBAAmB;IAClC,iBAAiB,EAAE,SAAS,CAAC;IAC7B,gBAAgB,EAAE,gBAAgB,CAAC;IACnC,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,kBAAkB;IACjC,IAAI,EAAE,SAAS,EAAE,CAAC;IAClB,YAAY,EAAE,KAAK,GAAG,UAAU,CAAC;IACjC,aAAa,EAAE,MAAM,CAAC;IACtB,YAAY,EAAE,MAAM,CAAC;CACtB;AAGD,MAAM,WAAW,MAAM;IACrB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACrC;AAED,MAAM,WAAW,QAAQ;IACvB,EAAE,EAAE,MAAM,CAAC;IACX,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACrC;AAED,MAAM,WAAW,cAAc;IAC7B,QAAQ,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC9B,SAAS,EAAE,QAAQ,EAAE,CAAC;IACtB,SAAS,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI,CAAC;IAChC,WAAW,CAAC,QAAQ,EAAE,QAAQ,GAAG,IAAI,CAAC;IACtC,SAAS,CAAC,EAAE,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,CAAC;IAC1C,YAAY,CAAC,QAAQ,EAAE,MAAM,GAAG,QAAQ,EAAE,CAAC;IAC3C,oBAAoB,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,MAAM,GAAG,MAAM,EAAE,CAAC;CAClE;AAED,MAAM,WAAW,mBAAmB;IAClC,SAAS,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACzC,WAAW,CAAC,QAAQ,EAAE,QAAQ,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC/C,SAAS,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,CAAC;IACnD,YAAY,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC;IACpD,oBAAoB,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;IAC1E,cAAc,CAAC,KAAK,EAAE;QAAE,IAAI,CAAC,EAAE,MAAM,CAAC;QAAC,IAAI,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;IAC3E,YAAY,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC9C,cAAc,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAClD,QAAQ,IAAI,OAAO,CAAC;QAAE,WAAW,EAAE,MAAM,CAAC;QAAC,aAAa,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IACpE,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACvB,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;CACxB;AAED,MAAM,WAAW,eAAe;IAC9B,OAAO,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;CAC1C;AAED,MAAM,WAAW,iBAAiB;IAChC,OAAO,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC;CAChE;AAED,MAAM,WAAW,wBAAwB;IACvC,SAAS,EAAE,SAAS,CAAC;IACrB,eAAe,EAAE,eAAe,CAAC;IACjC,iBAAiB,EAAE,iBAAiB,CAAC;IACrC,KAAK,EAAE,cAAc,CAAC;IACtB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,uBAAuB;IACtC,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,WAAW,EAAE,CAAC;IACzB,QAAQ,EAAE,MAAM,EAAE,CAAC;IACnB,SAAS,EAAE,QAAQ,EAAE,CAAC;IACtB,YAAY,EAAE,MAAM,CAAC;CACtB;AAGD,MAAM,WAAW,aAAa;IAC5B,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,EAAE,MAAM,CAAC;IACf,WAAW,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,iBAAiB;IAChC,SAAS,EAAE,SAAS,CAAC;IACrB,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,aAAa,CAAC;IACxB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,oBAAoB,CAAC,EAAE,OAAO,CAAC;IAC/B,oBAAoB,CAAC,EAAE,OAAO,CAAC;CAChC;AAED,MAAM,WAAW,gBAAgB;IAC/B,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,WAAW,EAAE,CAAC;IACzB,cAAc,EAAE,aAAa,EAAE,CAAC;IAChC,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;CACpB"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@eucoder/rag",
-  "version": "0.2.0",
+  "version": "0.3.0",
   "description": "Advanced RAG module with Corrective RAG, Hybrid Search, Knowledge Graph, Agentic RAG, and Self-Improving capabilities",
   "type": "module",
   "main": "./dist/index.js",
@@ -32,7 +32,8 @@
   "license": "Apache-2.0",
   "dependencies": {
     "@eucode/indexer": "workspace:*",
-    "@eucode/llm": "workspace:*"
+    "@eucode/llm": "workspace:*",
+    "neo4j-driver": "^6.0.1"
   },
   "devDependencies": {
     "@types/node": "^22.10.5",

package/src/index.ts

@@ -11,6 +11,10 @@ export { ABTestFramework } from "./ab-testing.js";
 export { ParameterOptimizer } from "./optimizer.js";
 export { SelfImprovingRag } from "./self-improving.js";
 
+// Fase 5: Knowledge Graph Persistente
+export { Neo4jKnowledgeGraph } from "./neo4j-knowledge-graph.js";
+export type { Neo4jConfig } from "./neo4j-knowledge-graph.js";
+
 export type { LlmProvider } from "./llm-grader.js";
 export type {
   Retriever,
@@ -26,6 +30,7 @@ export type {
   Entity,
   Relation,
   KnowledgeGraph,
+  AsyncKnowledgeGraph,
   EntityExtractor,
   RelationExtractor,
   KnowledgeGraphRagOptions,

package/src/neo4j-knowledge-graph.test.ts

@@ -0,0 +1,267 @@
+import { describe, it, expect, beforeAll, afterAll } from "vitest";
+import { Neo4jKnowledgeGraph } from "./neo4j-knowledge-graph.js";
+import type { Neo4jConfig } from "./neo4j-knowledge-graph.js";
+
+// Configurazione Neo4j per test (richiede Neo4j in esecuzione)
+const testConfig: Neo4jConfig = {
+  uri: process.env.NEO4J_URI || "bolt://localhost:7687",
+  username: process.env.NEO4J_USER || "neo4j",
+  password: process.env.NEO4J_PASSWORD || "test1234",
+  database: "neo4j",
+};
+
+// Verifica se Neo4j è disponibile prima di eseguire i test
+let neo4jAvailable = false;
+
+describe("Neo4jKnowledgeGraph", () => {
+  let graph: Neo4jKnowledgeGraph;
+
+  beforeAll(async () => {
+    graph = new Neo4jKnowledgeGraph(testConfig);
+    
+    // Verifica connessione con timeout
+    try {
+      const connected = await Promise.race([
+        graph.verifyConnectivity(),
+        new Promise<boolean>((_, reject) => 
+          setTimeout(() => reject(new Error("Timeout")), 3000)
+        ),
+      ]);
+      
+      neo4jAvailable = connected === true;
+      
+      if (!neo4jAvailable) {
+        console.log("⚠️  Neo4j non disponibile, test skippati");
+        return;
+      }
+
+      // Inizializza schema
+      await graph.initializeSchema();
+      
+      // Pulisci grafo prima dei test
+      await graph.clear();
+    } catch (error) {
+      console.log("⚠️  Neo4j non raggiungibile, test skippati");
+      neo4jAvailable = false;
+    }
+  });
+
+  afterAll(async () => {
+    if (!neo4jAvailable) return;
+    
+    // Pulisci grafo dopo i test
+    await graph.clear();
+    await graph.close();
+  });
+
+  it("should add and retrieve an entity", async () => {
+    if (!neo4jAvailable) {
+      console.log("⏭️  Test skippato: Neo4j non disponibile");
+      return;
+    }
+
+    const entity = {
+      id: "user-1",
+      name: "Alice",
+      type: "User",
+      properties: { age: 30, role: "developer" },
+    };
+
+    await graph.addEntity(entity);
+    const retrieved = await graph.getEntity("user-1");
+
+    expect(retrieved).toBeDefined();
+    expect(retrieved?.id).toBe("user-1");
+    expect(retrieved?.name).toBe("Alice");
+    expect(retrieved?.type).toBe("User");
+    expect(retrieved?.properties.age).toBe(30);
+  });
+
+  it("should add and retrieve a relation", async () => {
+    if (!neo4jAvailable) {
+      console.log("⏭️  Test skippato: Neo4j non disponibile");
+      return;
+    }
+
+    const user1 = {
+      id: "user-2",
+      name: "Bob",
+      type: "User",
+      properties: {},
+    };
+
+    const user2 = {
+      id: "user-3",
+      name: "Charlie",
+      type: "User",
+      properties: {},
+    };
+
+    await graph.addEntity(user1);
+    await graph.addEntity(user2);
+
+    const relation = {
+      id: "rel-1",
+      source: "user-2",
+      target: "user-3",
+      type: "KNOWS",
+      properties: { since: 2020 },
+    };
+
+    await graph.addRelation(relation);
+
+    const relations = await graph.getRelations("user-2");
+    expect(relations).toHaveLength(1);
+    expect(relations[0].type).toBe("KNOWS");
+    expect(relations[0].properties.since).toBe(2020);
+  });
+
+  it("should retrieve connected entities", async () => {
+    if (!neo4jAvailable) {
+      console.log("⏭️  Test skippato: Neo4j non disponibile");
+      return;
+    }
+
+    const user1 = {
+      id: "user-4",
+      name: "David",
+      type: "User",
+      properties: {},
+    };
+
+    const user2 = {
+      id: "user-5",
+      name: "Eve",
+      type: "User",
+      properties: {},
+    };
+
+    const user3 = {
+      id: "user-6",
+      name: "Frank",
+      type: "User",
+      properties: {},
+    };
+
+    await graph.addEntity(user1);
+    await graph.addEntity(user2);
+    await graph.addEntity(user3);
+
+    await graph.addRelation({
+      id: "rel-2",
+      source: "user-4",
+      target: "user-5",
+      type: "KNOWS",
+      properties: {},
+    });
+
+    await graph.addRelation({
+      id: "rel-3",
+      source: "user-5",
+      target: "user-6",
+      type: "KNOWS",
+      properties: {},
+    });
+
+    // Depth 1: solo user-5
+    const connected1 = await graph.getConnectedEntities("user-4", 1);
+    expect(connected1).toHaveLength(1);
+    expect(connected1[0].id).toBe("user-5");
+
+    // Depth 2: user-5 e user-6
+    const connected2 = await graph.getConnectedEntities("user-4", 2);
+    expect(connected2).toHaveLength(2);
+    expect(connected2.map((e) => e.id)).toContain("user-5");
+    expect(connected2.map((e) => e.id)).toContain("user-6");
+  });
+
+  it("should search entities by name", async () => {
+    if (!neo4jAvailable) {
+      console.log("⏭️  Test skippato: Neo4j non disponibile");
+      return;
+    }
+
+    const results = await graph.searchEntities({ name: "Alice" });
+    expect(results.length).toBeGreaterThan(0);
+    expect(results.some((e) => e.name === "Alice")).toBe(true);
+  });
+
+  it("should search entities by type", async () => {
+    if (!neo4jAvailable) {
+      console.log("⏭️  Test skippato: Neo4j non disponibile");
+      return;
+    }
+
+    const results = await graph.searchEntities({ type: "User" });
+    expect(results.length).toBeGreaterThan(0);
+    expect(results.every((e) => e.type === "User")).toBe(true);
+  });
+
+  it("should delete an entity and its relations", async () => {
+    if (!neo4jAvailable) {
+      console.log("⏭️  Test skippato: Neo4j non disponibile");
+      return;
+    }
+
+    const entity = {
+      id: "user-to-delete",
+      name: "ToDelete",
+      type: "User",
+      properties: {},
+    };
+
+    await graph.addEntity(entity);
+    await graph.deleteEntity("user-to-delete");
+
+    const retrieved = await graph.getEntity("user-to-delete");
+    expect(retrieved).toBeUndefined();
+  });
+
+  it("should delete a relation", async () => {
+    if (!neo4jAvailable) {
+      console.log("⏭️  Test skippato: Neo4j non disponibile");
+      return;
+    }
+
+    const user1 = {
+      id: "user-7",
+      name: "Grace",
+      type: "User",
+      properties: {},
+    };
+
+    const user2 = {
+      id: "user-8",
+      name: "Henry",
+      type: "User",
+      properties: {},
+    };
+
+    await graph.addEntity(user1);
+    await graph.addEntity(user2);
+
+    await graph.addRelation({
+      id: "rel-to-delete",
+      source: "user-7",
+      target: "user-8",
+      type: "KNOWS",
+      properties: {},
+    });
+
+    await graph.deleteRelation("rel-to-delete");
+
+    const relations = await graph.getRelations("user-7");
+    expect(relations.find((r) => r.id === "rel-to-delete")).toBeUndefined();
+  });
+
+  it("should get graph statistics", async () => {
+    if (!neo4jAvailable) {
+      console.log("⏭️  Test skippato: Neo4j non disponibile");
+      return;
+    }
+
+    const stats = await graph.getStats();
+    expect(stats.entityCount).toBeGreaterThan(0);
+    expect(stats.relationCount).toBeGreaterThan(0);
+  });
+});