Supabase Vector Store (#2427)

2025-03-25 00:15:50 +05:30
parent 9db5f62262
commit 2b49c9eedd
14 changed files with 8422 additions and 8 deletions
@@ -0,0 +1,99 @@
+import { Memory } from "../../src";
+
+export async function runTests(memory: Memory) {
+  try {
+    // Reset all memories
+    console.log("\nResetting all memories...");
+    await memory.reset();
+    console.log("All memories reset");
+
+    // Add a single memory
+    console.log("\nAdding a single memory...");
+    const result1 = await memory.add(
+      "Hi, my name is John and I am a software engineer.",
+      {
+        userId: "john",
+      },
+    );
+    console.log("Added memory:", result1);
+
+    // Add multiple messages
+    console.log("\nAdding multiple messages...");
+    const result2 = await memory.add(
+      [
+        { role: "user", content: "What is your favorite city?" },
+        { role: "assistant", content: "I love Paris, it is my favorite city." },
+      ],
+      {
+        userId: "john",
+      },
+    );
+    console.log("Added messages:", result2);
+
+    // Trying to update the memory
+    const result3 = await memory.add(
+      [
+        { role: "user", content: "What is your favorite city?" },
+        {
+          role: "assistant",
+          content: "I love New York, it is my favorite city.",
+        },
+      ],
+      {
+        userId: "john",
+      },
+    );
+    console.log("Updated messages:", result3);
+
+    // Get a single memory
+    console.log("\nGetting a single memory...");
+    if (result1.results && result1.results.length > 0) {
+      const singleMemory = await memory.get(result1.results[0].id);
+      console.log("Single memory:", singleMemory);
+    } else {
+      console.log("No memory was added in the first step");
+    }
+
+    // Updating this memory
+    const result4 = await memory.update(
+      result1.results[0].id,
+      "I love India, it is my favorite country.",
+    );
+    console.log("Updated memory:", result4);
+
+    // Get all memories
+    console.log("\nGetting all memories...");
+    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?", {
+      userId: "john",
+    });
+    console.log("Search results:", searchResult);
+
+    // Get memory history
+    if (result1.results && result1.results.length > 0) {
+      console.log("\nGetting memory history...");
+      const history = await memory.history(result1.results[0].id);
+      console.log("Memory history:", history);
+    }
+
+    // Delete a memory
+    if (result1.results && result1.results.length > 0) {
+      console.log("\nDeleting a memory...");
+      await memory.delete(result1.results[0].id);
+      console.log("Memory deleted successfully");
+    }
+
+    // Reset all memories
+    console.log("\nResetting all memories...");
+    await memory.reset();
+    console.log("All memories reset");
+  } catch (error) {
+    console.error("Error:", error);
+  }
+}