@mastra/pg 0.2.9 → 0.2.10-alpha.0

package/.turbo/turbo-build.log

@@ -1,23 +1,23 @@
 
-> @mastra/pg@0.2.9-alpha.0 build /home/runner/work/mastra/mastra/stores/pg
+> @mastra/pg@0.2.10-alpha.0 build /home/runner/work/mastra/mastra/stores/pg
 > tsup src/index.ts --format esm,cjs --experimental-dts --clean --treeshake=smallest --splitting
 
 CLI Building entry: src/index.ts
 CLI Using tsconfig: tsconfig.json
 CLI tsup v8.4.0
 TSC Build start
-TSC ⚡️ Build success in 10415ms
+TSC ⚡️ Build success in 10514ms
 DTS Build start
 CLI Target: es2022
 Analysis will use the bundled TypeScript version 5.8.2
 Writing package typings: /home/runner/work/mastra/mastra/stores/pg/dist/_tsup-dts-rollup.d.ts
 Analysis will use the bundled TypeScript version 5.8.2
 Writing package typings: /home/runner/work/mastra/mastra/stores/pg/dist/_tsup-dts-rollup.d.cts
-DTS ⚡️ Build success in 10461ms
+DTS ⚡️ Build success in 10890ms
 CLI Cleaning output folder
 ESM Build start
 CJS Build start
-ESM dist/index.js 40.43 KB
-ESM ⚡️ Build success in 1214ms
-CJS dist/index.cjs 40.85 KB
-CJS ⚡️ Build success in 1213ms
+ESM dist/index.js 41.41 KB
+ESM ⚡️ Build success in 1149ms
+CJS dist/index.cjs 41.83 KB
+CJS ⚡️ Build success in 1150ms

package/CHANGELOG.md

@@ -1,5 +1,13 @@
 # @mastra/pg
 
+## 0.2.10-alpha.0
+
+### Patch Changes
+
+- 4ac6f31: Updated permissions check for creating vector extension
+- Updated dependencies [359b089]
+  - @mastra/core@0.8.3-alpha.0
+
 ## 0.2.9
 
 ### Patch Changes

package/dist/_tsup-dts-rollup.d.cts

@@ -203,6 +203,8 @@ declare class PgVector extends MastraVector {
     private describeIndexCache;
     private createdIndexes;
     private mutexesByName;
+    private installVectorExtensionPromise;
+    private vectorExtensionInstalled;
     constructor(connectionString: string);
     private getMutexByName;
     transformFilter(filter?: VectorFilter): VectorFilter;
@@ -219,6 +221,7 @@ declare class PgVector extends MastraVector {
     defineIndex(indexName: string, metric: "cosine" | "euclidean" | "dotproduct" | undefined, indexConfig: IndexConfig): Promise<void>;
     buildIndex(...args: ParamsToArgs<PgDefineIndexParams> | PgDefineIndexArgs): Promise<void>;
     private setupIndex;
+    private installVectorExtension;
     listIndexes(): Promise<string[]>;
     describeIndex(indexName: string): Promise<PGIndexStats>;
     deleteIndex(indexName: string): Promise<void>;

package/dist/_tsup-dts-rollup.d.ts

@@ -203,6 +203,8 @@ declare class PgVector extends MastraVector {
     private describeIndexCache;
     private createdIndexes;
     private mutexesByName;
+    private installVectorExtensionPromise;
+    private vectorExtensionInstalled;
     constructor(connectionString: string);
     private getMutexByName;
     transformFilter(filter?: VectorFilter): VectorFilter;
@@ -219,6 +221,7 @@ declare class PgVector extends MastraVector {
     defineIndex(indexName: string, metric: "cosine" | "euclidean" | "dotproduct" | undefined, indexConfig: IndexConfig): Promise<void>;
     buildIndex(...args: ParamsToArgs<PgDefineIndexParams> | PgDefineIndexArgs): Promise<void>;
     private setupIndex;
+    private installVectorExtension;
     listIndexes(): Promise<string[]>;
     describeIndex(indexName: string): Promise<PGIndexStats>;
     deleteIndex(indexName: string): Promise<void>;

package/dist/index.cjs

@@ -299,6 +299,8 @@ var PgVector = class extends vector.MastraVector {
   describeIndexCache = /* @__PURE__ */ new Map();
   createdIndexes = /* @__PURE__ */ new Map();
   mutexesByName = /* @__PURE__ */ new Map();
+  installVectorExtensionPromise = null;
+  vectorExtensionInstalled = void 0;
   constructor(connectionString) {
     super();
     const basePool = new pg__default.default.Pool({
@@ -461,34 +463,16 @@ var PgVector = class extends vector.MastraVector {
       }
       const client = await this.pool.connect();
       try {
-        const extensionCheck = await client.query(`
-        SELECT EXISTS (
-          SELECT 1 FROM pg_available_extensions WHERE name = 'vector'
-        );
+        await this.installVectorExtension(client);
+        await client.query(`
+          CREATE TABLE IF NOT EXISTS ${indexName} (
+            id SERIAL PRIMARY KEY,
+            vector_id TEXT UNIQUE NOT NULL,
+            embedding vector(${dimension}),
+            metadata JSONB DEFAULT '{}'::jsonb
+          );
       `);
-        if (!extensionCheck.rows[0].exists) {
-          this.createdIndexes.delete(indexName);
-          throw new Error("PostgreSQL vector extension is not available. Please install it first.");
-        }
-        try {
-          await client.query(`
-          DO $$ 
-          BEGIN
-            CREATE EXTENSION IF NOT EXISTS vector;
-            
-            CREATE TABLE IF NOT EXISTS ${indexName} (
-              id SERIAL PRIMARY KEY,
-              vector_id TEXT UNIQUE NOT NULL,
-              embedding vector(${dimension}),
-              metadata JSONB DEFAULT '{}'::jsonb
-            );
-          END $$;
-        `);
-          this.createdIndexes.set(indexName, indexCacheKey);
-        } catch (e) {
-          this.createdIndexes.delete(indexName);
-          throw e;
-        }
+        this.createdIndexes.set(indexName, indexCacheKey);
         if (buildIndex) {
           await this.setupIndex({ indexName, metric, indexConfig }, client);
         }
@@ -562,6 +546,44 @@ var PgVector = class extends vector.MastraVector {
       await client.query(indexSQL);
     });
   }
+  async installVectorExtension(client) {
+    if (this.vectorExtensionInstalled) {
+      return;
+    }
+    if (!this.installVectorExtensionPromise) {
+      this.installVectorExtensionPromise = (async () => {
+        try {
+          const extensionCheck = await client.query(`
+            SELECT EXISTS (
+              SELECT 1 FROM pg_extension WHERE extname = 'vector'
+            );
+          `);
+          this.vectorExtensionInstalled = extensionCheck.rows[0].exists;
+          if (!this.vectorExtensionInstalled) {
+            try {
+              await client.query("CREATE EXTENSION IF NOT EXISTS vector");
+              this.vectorExtensionInstalled = true;
+              this.logger.info("Vector extension installed successfully");
+            } catch {
+              this.logger.warn(
+                "Could not install vector extension. This requires superuser privileges. If the extension is already installed globally, you can ignore this warning."
+              );
+            }
+          } else {
+            this.logger.debug("Vector extension already installed, skipping installation");
+          }
+        } catch (error) {
+          this.logger.error("Error checking vector extension status", { error });
+          this.vectorExtensionInstalled = void 0;
+          this.installVectorExtensionPromise = null;
+          throw error;
+        } finally {
+          this.installVectorExtensionPromise = null;
+        }
+      })();
+    }
+    await this.installVectorExtensionPromise;
+  }
   async listIndexes() {
     const client = await this.pool.connect();
     try {

package/dist/index.js

@@ -291,6 +291,8 @@ var PgVector = class extends MastraVector {
   describeIndexCache = /* @__PURE__ */ new Map();
   createdIndexes = /* @__PURE__ */ new Map();
   mutexesByName = /* @__PURE__ */ new Map();
+  installVectorExtensionPromise = null;
+  vectorExtensionInstalled = void 0;
   constructor(connectionString) {
     super();
     const basePool = new pg.Pool({
@@ -453,34 +455,16 @@ var PgVector = class extends MastraVector {
       }
       const client = await this.pool.connect();
       try {
-        const extensionCheck = await client.query(`
-        SELECT EXISTS (
-          SELECT 1 FROM pg_available_extensions WHERE name = 'vector'
-        );
+        await this.installVectorExtension(client);
+        await client.query(`
+          CREATE TABLE IF NOT EXISTS ${indexName} (
+            id SERIAL PRIMARY KEY,
+            vector_id TEXT UNIQUE NOT NULL,
+            embedding vector(${dimension}),
+            metadata JSONB DEFAULT '{}'::jsonb
+          );
       `);
-        if (!extensionCheck.rows[0].exists) {
-          this.createdIndexes.delete(indexName);
-          throw new Error("PostgreSQL vector extension is not available. Please install it first.");
-        }
-        try {
-          await client.query(`
-          DO $$ 
-          BEGIN
-            CREATE EXTENSION IF NOT EXISTS vector;
-            
-            CREATE TABLE IF NOT EXISTS ${indexName} (
-              id SERIAL PRIMARY KEY,
-              vector_id TEXT UNIQUE NOT NULL,
-              embedding vector(${dimension}),
-              metadata JSONB DEFAULT '{}'::jsonb
-            );
-          END $$;
-        `);
-          this.createdIndexes.set(indexName, indexCacheKey);
-        } catch (e) {
-          this.createdIndexes.delete(indexName);
-          throw e;
-        }
+        this.createdIndexes.set(indexName, indexCacheKey);
         if (buildIndex) {
           await this.setupIndex({ indexName, metric, indexConfig }, client);
         }
@@ -554,6 +538,44 @@ var PgVector = class extends MastraVector {
       await client.query(indexSQL);
     });
   }
+  async installVectorExtension(client) {
+    if (this.vectorExtensionInstalled) {
+      return;
+    }
+    if (!this.installVectorExtensionPromise) {
+      this.installVectorExtensionPromise = (async () => {
+        try {
+          const extensionCheck = await client.query(`
+            SELECT EXISTS (
+              SELECT 1 FROM pg_extension WHERE extname = 'vector'
+            );
+          `);
+          this.vectorExtensionInstalled = extensionCheck.rows[0].exists;
+          if (!this.vectorExtensionInstalled) {
+            try {
+              await client.query("CREATE EXTENSION IF NOT EXISTS vector");
+              this.vectorExtensionInstalled = true;
+              this.logger.info("Vector extension installed successfully");
+            } catch {
+              this.logger.warn(
+                "Could not install vector extension. This requires superuser privileges. If the extension is already installed globally, you can ignore this warning."
+              );
+            }
+          } else {
+            this.logger.debug("Vector extension already installed, skipping installation");
+          }
+        } catch (error) {
+          this.logger.error("Error checking vector extension status", { error });
+          this.vectorExtensionInstalled = void 0;
+          this.installVectorExtensionPromise = null;
+          throw error;
+        } finally {
+          this.installVectorExtensionPromise = null;
+        }
+      })();
+    }
+    await this.installVectorExtensionPromise;
+  }
   async listIndexes() {
     const client = await this.pool.connect();
     try {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/pg",
-  "version": "0.2.9",
+  "version": "0.2.10-alpha.0",
   "description": "Postgres provider for Mastra - includes both vector and db storage capabilities",
   "type": "module",
   "main": "dist/index.js",
@@ -23,7 +23,7 @@
     "pg": "^8.13.3",
     "pg-promise": "^11.11.0",
     "xxhash-wasm": "^1.1.0",
-    "@mastra/core": "^0.8.2"
+    "@mastra/core": "^0.8.3-alpha.0"
   },
   "devDependencies": {
     "@microsoft/api-extractor": "^7.52.1",

package/src/vector/index.ts

@@ -64,6 +64,8 @@ export class PgVector extends MastraVector {
   private describeIndexCache: Map<string, PGIndexStats> = new Map();
   private createdIndexes = new Map<string, number>();
   private mutexesByName = new Map<string, Mutex>();
+  private installVectorExtensionPromise: Promise<void> | null = null;
+  private vectorExtensionInstalled: boolean | undefined = undefined;
 
   constructor(connectionString: string) {
     super();
@@ -261,38 +263,17 @@ export class PgVector extends MastraVector {
 
       const client = await this.pool.connect();
       try {
-        // First check if vector extension is available
-        const extensionCheck = await client.query(`
-        SELECT EXISTS (
-          SELECT 1 FROM pg_available_extensions WHERE name = 'vector'
-        );
+        // install vector extension
+        await this.installVectorExtension(client);
+        await client.query(`
+          CREATE TABLE IF NOT EXISTS ${indexName} (
+            id SERIAL PRIMARY KEY,
+            vector_id TEXT UNIQUE NOT NULL,
+            embedding vector(${dimension}),
+            metadata JSONB DEFAULT '{}'::jsonb
+          );
       `);
-
-        if (!extensionCheck.rows[0].exists) {
-          this.createdIndexes.delete(indexName);
-          throw new Error('PostgreSQL vector extension is not available. Please install it first.');
-        }
-
-        // Try to create extension
-        try {
-          await client.query(`
-          DO $$ 
-          BEGIN
-            CREATE EXTENSION IF NOT EXISTS vector;
-            
-            CREATE TABLE IF NOT EXISTS ${indexName} (
-              id SERIAL PRIMARY KEY,
-              vector_id TEXT UNIQUE NOT NULL,
-              embedding vector(${dimension}),
-              metadata JSONB DEFAULT '{}'::jsonb
-            );
-          END $$;
-        `);
-          this.createdIndexes.set(indexName, indexCacheKey);
-        } catch (e) {
-          this.createdIndexes.delete(indexName);
-          throw e;
-        }
+        this.createdIndexes.set(indexName, indexCacheKey);
 
         if (buildIndex) {
           await this.setupIndex({ indexName, metric, indexConfig }, client);
@@ -384,6 +365,57 @@ export class PgVector extends MastraVector {
     });
   }
 
+  private async installVectorExtension(client: pg.PoolClient) {
+    // If we've already successfully installed, no need to do anything
+    if (this.vectorExtensionInstalled) {
+      return;
+    }
+
+    // If there's no existing installation attempt or the previous one failed
+    if (!this.installVectorExtensionPromise) {
+      this.installVectorExtensionPromise = (async () => {
+        try {
+          // First check if extension is already installed
+          const extensionCheck = await client.query(`
+            SELECT EXISTS (
+              SELECT 1 FROM pg_extension WHERE extname = 'vector'
+            );
+          `);
+
+          this.vectorExtensionInstalled = extensionCheck.rows[0].exists;
+
+          if (!this.vectorExtensionInstalled) {
+            try {
+              await client.query('CREATE EXTENSION IF NOT EXISTS vector');
+              this.vectorExtensionInstalled = true;
+              this.logger.info('Vector extension installed successfully');
+            } catch {
+              this.logger.warn(
+                'Could not install vector extension. This requires superuser privileges. ' +
+                  'If the extension is already installed globally, you can ignore this warning.',
+              );
+              // Don't set vectorExtensionInstalled to false here since we're not sure if it failed
+              // due to permissions or if it's already installed globally
+            }
+          } else {
+            this.logger.debug('Vector extension already installed, skipping installation');
+          }
+        } catch (error) {
+          this.logger.error('Error checking vector extension status', { error });
+          // Reset both the promise and the flag so we can retry
+          this.vectorExtensionInstalled = undefined;
+          this.installVectorExtensionPromise = null;
+          throw error; // Re-throw so caller knows it failed
+        } finally {
+          // Clear the promise after completion (success or failure)
+          this.installVectorExtensionPromise = null;
+        }
+      })();
+    }
+
+    // Wait for the installation process to complete
+    await this.installVectorExtensionPromise;
+  }
   async listIndexes(): Promise<string[]> {
     const client = await this.pool.connect();
     try {