Graph Support for NodeSDK (#2298)

mem0-ts/src/oss/examples/basic.ts

@@ -29,7 +29,9 @@ async function runTests(memory: Memory) {
     console.log("\nAdding a single memory...");
     const result1 = await memory.add(
       "Hi, my name is John and I am a software engineer.",
-      "user123",
+      {
+        userId: "john",
+      },
     );
     console.log("Added memory:", result1);
 
@@ -40,7 +42,9 @@ async function runTests(memory: Memory) {
         { role: "user", content: "What is your favorite city?" },
         { role: "assistant", content: "I love Paris, it is my favorite city." },
       ],
-      "user123",
+      {
+        userId: "john",
+      },
     );
     console.log("Added messages:", result2);
 
@@ -53,7 +57,9 @@ async function runTests(memory: Memory) {
           content: "I love New York, it is my favorite city.",
         },
       ],
-      "user123",
+      {
+        userId: "john",
+      },
     );
     console.log("Updated messages:", result3);
 
@@ -75,15 +81,16 @@ async function runTests(memory: Memory) {
 
     // Get all memories
     console.log("\nGetting all memories...");
-    const allMemories = await memory.getAll("user123");
+    const allMemories = await memory.getAll({
+      userId: "john",
+    });
     console.log("All memories:", allMemories);
 
     // Search for memories
     console.log("\nSearching memories...");
-    const searchResult = await memory.search(
-      "What do you know about Paris?",
-      "user123",
-    );
+    const searchResult = await memory.search("What do you know about Paris?", {
+      userId: "john",
+    });
     console.log("Search results:", searchResult);
 
     // Get memory history
@@ -255,10 +262,103 @@ async function demoRedis() {
   await runTests(memory);
 }
 
+async function demoGraphMemory() {
+  console.log("\n=== Testing Graph Memory Store ===\n");
+
+  const memory = new Memory({
+    version: "v1.1",
+    embedder: {
+      provider: "openai",
+      config: {
+        apiKey: process.env.OPENAI_API_KEY || "",
+        model: "text-embedding-3-small",
+      },
+    },
+    vectorStore: {
+      provider: "memory",
+      config: {
+        collectionName: "memories",
+        dimension: 1536,
+      },
+    },
+    llm: {
+      provider: "openai",
+      config: {
+        apiKey: process.env.OPENAI_API_KEY || "",
+        model: "gpt-4-turbo-preview",
+      },
+    },
+    graphStore: {
+      provider: "neo4j",
+      config: {
+        url: process.env.NEO4J_URL || "neo4j://localhost:7687",
+        username: process.env.NEO4J_USERNAME || "neo4j",
+        password: process.env.NEO4J_PASSWORD || "password",
+      },
+      llm: {
+        provider: "openai",
+        config: {
+          model: "gpt-4-turbo-preview",
+        },
+      },
+    },
+    historyDbPath: "memory.db",
+  });
+
+  try {
+    // Reset all memories
+    await memory.reset();
+
+    // Add memories with relationships
+    const result = await memory.add(
+      [
+        {
+          role: "user",
+          content: "Alice is Bob's sister and works as a doctor.",
+        },
+        {
+          role: "assistant",
+          content:
+            "I understand that Alice and Bob are siblings and Alice is a medical professional.",
+        },
+        { role: "user", content: "Bob is married to Carol who is a teacher." },
+      ],
+      {
+        userId: "john",
+      },
+    );
+    console.log("Added memories with relationships:", result);
+
+    // Search for connected information
+    const searchResult = await memory.search(
+      "Tell me about Bob's family connections",
+      {
+        userId: "john",
+      },
+    );
+    console.log("Search results with graph relationships:", searchResult);
+  } catch (error) {
+    console.error("Error in graph memory demo:", error);
+  }
+}
+
 async function main() {
   // Test in-memory store
   await demoMemoryStore();
 
+  // Test graph memory if Neo4j environment variables are set
+  if (
+    process.env.NEO4J_URL &&
+    process.env.NEO4J_USERNAME &&
+    process.env.NEO4J_PASSWORD
+  ) {
+    await demoGraphMemory();
+  } else {
+    console.log(
+      "\nSkipping Graph Memory test - Neo4j environment variables not set",
+    );
+  }
+
   // Test PGVector store if environment variables are set
   if (process.env.PGVECTOR_DB) {
     await demoPGVector();