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,49 @@
+import { Memory } from "../../src";
+import { runTests } from "../utils/test-utils";
+
+export async function demoPGVector() {
+  console.log("\n=== Testing PGVector 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: "pgvector",
+      config: {
+        collectionName: "memories",
+        dimension: 1536,
+        dbname: process.env.PGVECTOR_DB || "vectordb",
+        user: process.env.PGVECTOR_USER || "postgres",
+        password: process.env.PGVECTOR_PASSWORD || "postgres",
+        host: process.env.PGVECTOR_HOST || "localhost",
+        port: parseInt(process.env.PGVECTOR_PORT || "5432"),
+        embeddingModelDims: 1536,
+        hnsw: true,
+      },
+    },
+    llm: {
+      provider: "openai",
+      config: {
+        apiKey: process.env.OPENAI_API_KEY || "",
+        model: "gpt-4-turbo-preview",
+      },
+    },
+    historyDbPath: "memory.db",
+  });
+
+  await runTests(memory);
+}
+
+if (require.main === module) {
+  if (!process.env.PGVECTOR_DB) {
+    console.log("\nSkipping PGVector test - environment variables not set");
+    process.exit(0);
+  }
+  demoPGVector();
+}