@mastra/pg 0.3.4-alpha.2 → 0.3.4-alpha.4

package/.turbo/turbo-build.log

@@ -1,23 +1,23 @@
 
-> @mastra/pg@0.3.4-alpha.2 build /home/runner/work/mastra/mastra/stores/pg
+> @mastra/pg@0.3.4-alpha.4 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 11653ms
+TSC ⚡️ Build success in 10750ms
 DTS Build start
 CLI Target: es2022
 Analysis will use the bundled TypeScript version 5.8.3
 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.3
 Writing package typings: /home/runner/work/mastra/mastra/stores/pg/dist/_tsup-dts-rollup.d.cts
-DTS ⚡️ Build success in 10772ms
+DTS ⚡️ Build success in 11782ms
 CLI Cleaning output folder
 ESM Build start
 CJS Build start
-ESM dist/index.js 54.89 KB
-ESM ⚡️ Build success in 1604ms
-CJS dist/index.cjs 55.37 KB
-CJS ⚡️ Build success in 1604ms
+ESM dist/index.js 58.39 KB
+ESM ⚡️ Build success in 1573ms
+CJS dist/index.cjs 58.86 KB
+CJS ⚡️ Build success in 1575ms

package/CHANGELOG.md

@@ -1,5 +1,23 @@
 # @mastra/pg
 
+## 0.3.4-alpha.4
+
+### Patch Changes
+
+- c3bd795: [MASTRA-3358] Deprecate updateIndexById and deleteIndexById
+- Updated dependencies [396be50]
+- Updated dependencies [c3bd795]
+- Updated dependencies [da082f8]
+- Updated dependencies [a5810ce]
+  - @mastra/core@0.9.4-alpha.3
+
+## 0.3.4-alpha.3
+
+### Patch Changes
+
+- 2836734: [MASTRA-3391] fix describe index for custom schema
+- a3fc60c: fix whereClause condition for fromDate and toDate in pg getTraces
+
 ## 0.3.4-alpha.2
 
 ### Patch Changes

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

@@ -213,7 +213,7 @@ declare class PgVector extends MastraVector {
     private schemaSetupComplete;
     /**
      * @deprecated Passing connectionString as a string is deprecated.
-     * Use the object parameter instead. This signature will be removed on May 20th.
+     * Use the object parameter instead. This signature will be removed on May 20th, 2025.
      */
     constructor(connectionString: string);
     constructor(config: {
@@ -234,6 +234,7 @@ declare class PgVector extends MastraVector {
     createIndex(...args: ParamsToArgs<PgCreateIndexParams> | PgCreateIndexArgs): Promise<void>;
     /**
      * @deprecated This function is deprecated. Use buildIndex instead
+     * This function will be removed on May 20th, 2025
      */
     defineIndex(indexName: string, metric: "cosine" | "euclidean" | "dotproduct" | undefined, indexConfig: IndexConfig): Promise<void>;
     buildIndex(...args: ParamsToArgs<PgDefineIndexParams> | PgDefineIndexArgs): Promise<void>;
@@ -244,11 +245,54 @@ declare class PgVector extends MastraVector {
     deleteIndex(indexName: string): Promise<void>;
     truncateIndex(indexName: string): Promise<void>;
     disconnect(): Promise<void>;
+    /**
+     * @deprecated Use {@link updateVector} instead. This method will be removed on May 20th, 2025.
+     *
+     * Updates a vector by its ID with the provided vector and/or metadata.
+     * @param indexName - The name of the index containing the vector.
+     * @param id - The ID of the vector to update.
+     * @param update - An object containing the vector and/or metadata to update.
+     * @param update.vector - An optional array of numbers representing the new vector.
+     * @param update.metadata - An optional record containing the new metadata.
+     * @returns A promise that resolves when the update is complete.
+     * @throws Will throw an error if no updates are provided or if the update operation fails.
+     */
     updateIndexById(indexName: string, id: string, update: {
         vector?: number[];
         metadata?: Record<string, any>;
     }): Promise<void>;
+    /**
+     * Updates a vector by its ID with the provided vector and/or metadata.
+     * @param indexName - The name of the index containing the vector.
+     * @param id - The ID of the vector to update.
+     * @param update - An object containing the vector and/or metadata to update.
+     * @param update.vector - An optional array of numbers representing the new vector.
+     * @param update.metadata - An optional record containing the new metadata.
+     * @returns A promise that resolves when the update is complete.
+     * @throws Will throw an error if no updates are provided or if the update operation fails.
+     */
+    updateVector(indexName: string, id: string, update: {
+        vector?: number[];
+        metadata?: Record<string, any>;
+    }): Promise<void>;
+    /**
+     * @deprecated Use {@link deleteVector} instead. This method will be removed on May 20th, 2025.
+     *
+     * Deletes a vector by its ID.
+     * @param indexName - The name of the index containing the vector.
+     * @param id - The ID of the vector to delete.
+     * @returns A promise that resolves when the deletion is complete.
+     * @throws Will throw an error if the deletion operation fails.
+     */
     deleteIndexById(indexName: string, id: string): Promise<void>;
+    /**
+     * Deletes a vector by its ID.
+     * @param indexName - The name of the index containing the vector.
+     * @param id - The ID of the vector to delete.
+     * @returns A promise that resolves when the deletion is complete.
+     * @throws Will throw an error if the deletion operation fails.
+     */
+    deleteVector(indexName: string, id: string): Promise<void>;
 }
 export { PgVector }
 export { PgVector as PgVector_alias_1 }
@@ -264,7 +308,7 @@ export { PGVECTOR_PROMPT as PGVECTOR_PROMPT_alias_1 }
 declare type PostgresConfig = {
     schemaName?: string;
     /**
-     * @deprecated Use `schemaName` instead. Support for `schema` will be removed in a future release.
+     * @deprecated Use `schemaName` instead. Support for `schema` will be removed on May 20th, 2025.
      */
     schema?: string;
 } & ({

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

@@ -213,7 +213,7 @@ declare class PgVector extends MastraVector {
     private schemaSetupComplete;
     /**
      * @deprecated Passing connectionString as a string is deprecated.
-     * Use the object parameter instead. This signature will be removed on May 20th.
+     * Use the object parameter instead. This signature will be removed on May 20th, 2025.
      */
     constructor(connectionString: string);
     constructor(config: {
@@ -234,6 +234,7 @@ declare class PgVector extends MastraVector {
     createIndex(...args: ParamsToArgs<PgCreateIndexParams> | PgCreateIndexArgs): Promise<void>;
     /**
      * @deprecated This function is deprecated. Use buildIndex instead
+     * This function will be removed on May 20th, 2025
      */
     defineIndex(indexName: string, metric: "cosine" | "euclidean" | "dotproduct" | undefined, indexConfig: IndexConfig): Promise<void>;
     buildIndex(...args: ParamsToArgs<PgDefineIndexParams> | PgDefineIndexArgs): Promise<void>;
@@ -244,11 +245,54 @@ declare class PgVector extends MastraVector {
     deleteIndex(indexName: string): Promise<void>;
     truncateIndex(indexName: string): Promise<void>;
     disconnect(): Promise<void>;
+    /**
+     * @deprecated Use {@link updateVector} instead. This method will be removed on May 20th, 2025.
+     *
+     * Updates a vector by its ID with the provided vector and/or metadata.
+     * @param indexName - The name of the index containing the vector.
+     * @param id - The ID of the vector to update.
+     * @param update - An object containing the vector and/or metadata to update.
+     * @param update.vector - An optional array of numbers representing the new vector.
+     * @param update.metadata - An optional record containing the new metadata.
+     * @returns A promise that resolves when the update is complete.
+     * @throws Will throw an error if no updates are provided or if the update operation fails.
+     */
     updateIndexById(indexName: string, id: string, update: {
         vector?: number[];
         metadata?: Record<string, any>;
     }): Promise<void>;
+    /**
+     * Updates a vector by its ID with the provided vector and/or metadata.
+     * @param indexName - The name of the index containing the vector.
+     * @param id - The ID of the vector to update.
+     * @param update - An object containing the vector and/or metadata to update.
+     * @param update.vector - An optional array of numbers representing the new vector.
+     * @param update.metadata - An optional record containing the new metadata.
+     * @returns A promise that resolves when the update is complete.
+     * @throws Will throw an error if no updates are provided or if the update operation fails.
+     */
+    updateVector(indexName: string, id: string, update: {
+        vector?: number[];
+        metadata?: Record<string, any>;
+    }): Promise<void>;
+    /**
+     * @deprecated Use {@link deleteVector} instead. This method will be removed on May 20th, 2025.
+     *
+     * Deletes a vector by its ID.
+     * @param indexName - The name of the index containing the vector.
+     * @param id - The ID of the vector to delete.
+     * @returns A promise that resolves when the deletion is complete.
+     * @throws Will throw an error if the deletion operation fails.
+     */
     deleteIndexById(indexName: string, id: string): Promise<void>;
+    /**
+     * Deletes a vector by its ID.
+     * @param indexName - The name of the index containing the vector.
+     * @param id - The ID of the vector to delete.
+     * @returns A promise that resolves when the deletion is complete.
+     * @throws Will throw an error if the deletion operation fails.
+     */
+    deleteVector(indexName: string, id: string): Promise<void>;
 }
 export { PgVector }
 export { PgVector as PgVector_alias_1 }
@@ -264,7 +308,7 @@ export { PGVECTOR_PROMPT as PGVECTOR_PROMPT_alias_1 }
 declare type PostgresConfig = {
     schemaName?: string;
     /**
-     * @deprecated Use `schemaName` instead. Support for `schema` will be removed in a future release.
+     * @deprecated Use `schemaName` instead. Support for `schema` will be removed on May 20th, 2025.
      */
     schema?: string;
 } & ({

package/dist/index.cjs

@@ -352,7 +352,7 @@ var PgVector = class extends vector.MastraVector {
         Please use an object parameter instead:
         new PgVector({ connectionString })
 
-        The string signature will be removed on May 20th.`
+        The string signature will be removed on May 20th, 2025.`
       );
       connectionString = config;
       schemaName = void 0;
@@ -601,8 +601,10 @@ var PgVector = class extends vector.MastraVector {
   }
   /**
    * @deprecated This function is deprecated. Use buildIndex instead
+   * This function will be removed on May 20th, 2025
    */
   async defineIndex(indexName, metric = "cosine", indexConfig) {
+    console.warn("defineIndex is deprecated. Use buildIndex instead. This function will be removed on May 20th, 2025");
     return this.buildIndex({ indexName, metric, indexConfig });
   }
   async buildIndex(...args) {
@@ -718,6 +720,18 @@ var PgVector = class extends vector.MastraVector {
     const client = await this.pool.connect();
     try {
       const tableName = this.getTableName(indexName);
+      const tableExistsQuery = `
+        SELECT 1
+        FROM information_schema.columns
+        WHERE table_schema = $1
+          AND table_name = $2
+          AND udt_name = 'vector'
+        LIMIT 1;
+      `;
+      const tableExists = await client.query(tableExistsQuery, [this.schema || "public", indexName]);
+      if (tableExists.rows.length === 0) {
+        throw new Error(`Vector table ${tableName} does not exist`);
+      }
       const dimensionQuery = `
                 SELECT atttypmod as dimension
                 FROM pg_attribute
@@ -737,12 +751,14 @@ var PgVector = class extends vector.MastraVector {
             JOIN pg_class c ON i.indexrelid = c.oid
             JOIN pg_am am ON c.relam = am.oid
             JOIN pg_opclass opclass ON i.indclass[0] = opclass.oid
-            WHERE c.relname = '${tableName}_vector_idx';
+            JOIN pg_namespace n ON c.relnamespace = n.oid
+            WHERE c.relname = $1
+            AND n.nspname = $2;
             `;
       const [dimResult, countResult, indexResult] = await Promise.all([
         client.query(dimensionQuery, [tableName]),
         client.query(countQuery),
-        client.query(indexQuery)
+        client.query(indexQuery, [`${indexName}_vector_idx`, this.schema || "public"])
       ]);
       const { index_method, index_def, operator_class } = indexResult.rows[0] || {
         index_method: "flat",
@@ -802,7 +818,37 @@ var PgVector = class extends vector.MastraVector {
   async disconnect() {
     await this.pool.end();
   }
+  /**
+   * @deprecated Use {@link updateVector} instead. This method will be removed on May 20th, 2025.
+   *
+   * Updates a vector by its ID with the provided vector and/or metadata.
+   * @param indexName - The name of the index containing the vector.
+   * @param id - The ID of the vector to update.
+   * @param update - An object containing the vector and/or metadata to update.
+   * @param update.vector - An optional array of numbers representing the new vector.
+   * @param update.metadata - An optional record containing the new metadata.
+   * @returns A promise that resolves when the update is complete.
+   * @throws Will throw an error if no updates are provided or if the update operation fails.
+   */
   async updateIndexById(indexName, id, update) {
+    this.logger.warn(
+      `Deprecation Warning: updateIndexById() is deprecated. 
+      Please use updateVector() instead. 
+      updateIndexById() will be removed on May 20th, 2025.`
+    );
+    await this.updateVector(indexName, id, update);
+  }
+  /**
+   * Updates a vector by its ID with the provided vector and/or metadata.
+   * @param indexName - The name of the index containing the vector.
+   * @param id - The ID of the vector to update.
+   * @param update - An object containing the vector and/or metadata to update.
+   * @param update.vector - An optional array of numbers representing the new vector.
+   * @param update.metadata - An optional record containing the new metadata.
+   * @returns A promise that resolves when the update is complete.
+   * @throws Will throw an error if no updates are provided or if the update operation fails.
+   */
+  async updateVector(indexName, id, update) {
     if (!update.vector && !update.metadata) {
       throw new Error("No updates provided");
     }
@@ -830,11 +876,37 @@ var PgVector = class extends vector.MastraVector {
         WHERE vector_id = $1
       `;
       await client.query(query, values);
+    } catch (error) {
+      throw new Error(`Failed to update vector by id: ${id} for index: ${indexName}: ${error.message}`);
     } finally {
       client.release();
     }
   }
+  /**
+   * @deprecated Use {@link deleteVector} instead. This method will be removed on May 20th, 2025.
+   *
+   * Deletes a vector by its ID.
+   * @param indexName - The name of the index containing the vector.
+   * @param id - The ID of the vector to delete.
+   * @returns A promise that resolves when the deletion is complete.
+   * @throws Will throw an error if the deletion operation fails.
+   */
   async deleteIndexById(indexName, id) {
+    this.logger.warn(
+      `Deprecation Warning: deleteIndexById() is deprecated. 
+      Please use deleteVector() instead. 
+      deleteIndexById() will be removed on May 20th, 2025.`
+    );
+    await this.deleteVector(indexName, id);
+  }
+  /**
+   * Deletes a vector by its ID.
+   * @param indexName - The name of the index containing the vector.
+   * @param id - The ID of the vector to delete.
+   * @returns A promise that resolves when the deletion is complete.
+   * @throws Will throw an error if the deletion operation fails.
+   */
+  async deleteVector(indexName, id) {
     const client = await this.pool.connect();
     try {
       const tableName = this.getTableName(indexName);
@@ -843,6 +915,8 @@ var PgVector = class extends vector.MastraVector {
         WHERE vector_id = $1
       `;
       await client.query(query, [id]);
+    } catch (error) {
+      throw new Error(`Failed to delete vector by id: ${id} for index: ${indexName}: ${error.message}`);
     } finally {
       client.release();
     }
@@ -875,7 +949,7 @@ var PostgresStore = class extends storage.MastraStorage {
     this.pgp = pgPromise__default.default();
     if ("schema" in config && config.schema) {
       console.warn(
-        '[DEPRECATION NOTICE] The "schema" option in PostgresStore is deprecated. Please use "schemaName" instead. Support for "schema" will be removed in a future release.'
+        '[DEPRECATION NOTICE] The "schema" option in PostgresStore is deprecated. Please use "schemaName" instead. Support for "schema" will be removed on May 20th, 2025.'
       );
     }
     this.schema = config.schemaName ?? config.schema;

package/dist/index.js

@@ -344,7 +344,7 @@ var PgVector = class extends MastraVector {
         Please use an object parameter instead:
         new PgVector({ connectionString })
 
-        The string signature will be removed on May 20th.`
+        The string signature will be removed on May 20th, 2025.`
       );
       connectionString = config;
       schemaName = void 0;
@@ -593,8 +593,10 @@ var PgVector = class extends MastraVector {
   }
   /**
    * @deprecated This function is deprecated. Use buildIndex instead
+   * This function will be removed on May 20th, 2025
    */
   async defineIndex(indexName, metric = "cosine", indexConfig) {
+    console.warn("defineIndex is deprecated. Use buildIndex instead. This function will be removed on May 20th, 2025");
     return this.buildIndex({ indexName, metric, indexConfig });
   }
   async buildIndex(...args) {
@@ -710,6 +712,18 @@ var PgVector = class extends MastraVector {
     const client = await this.pool.connect();
     try {
       const tableName = this.getTableName(indexName);
+      const tableExistsQuery = `
+        SELECT 1
+        FROM information_schema.columns
+        WHERE table_schema = $1
+          AND table_name = $2
+          AND udt_name = 'vector'
+        LIMIT 1;
+      `;
+      const tableExists = await client.query(tableExistsQuery, [this.schema || "public", indexName]);
+      if (tableExists.rows.length === 0) {
+        throw new Error(`Vector table ${tableName} does not exist`);
+      }
       const dimensionQuery = `
                 SELECT atttypmod as dimension
                 FROM pg_attribute
@@ -729,12 +743,14 @@ var PgVector = class extends MastraVector {
             JOIN pg_class c ON i.indexrelid = c.oid
             JOIN pg_am am ON c.relam = am.oid
             JOIN pg_opclass opclass ON i.indclass[0] = opclass.oid
-            WHERE c.relname = '${tableName}_vector_idx';
+            JOIN pg_namespace n ON c.relnamespace = n.oid
+            WHERE c.relname = $1
+            AND n.nspname = $2;
             `;
       const [dimResult, countResult, indexResult] = await Promise.all([
         client.query(dimensionQuery, [tableName]),
         client.query(countQuery),
-        client.query(indexQuery)
+        client.query(indexQuery, [`${indexName}_vector_idx`, this.schema || "public"])
       ]);
       const { index_method, index_def, operator_class } = indexResult.rows[0] || {
         index_method: "flat",
@@ -794,7 +810,37 @@ var PgVector = class extends MastraVector {
   async disconnect() {
     await this.pool.end();
   }
+  /**
+   * @deprecated Use {@link updateVector} instead. This method will be removed on May 20th, 2025.
+   *
+   * Updates a vector by its ID with the provided vector and/or metadata.
+   * @param indexName - The name of the index containing the vector.
+   * @param id - The ID of the vector to update.
+   * @param update - An object containing the vector and/or metadata to update.
+   * @param update.vector - An optional array of numbers representing the new vector.
+   * @param update.metadata - An optional record containing the new metadata.
+   * @returns A promise that resolves when the update is complete.
+   * @throws Will throw an error if no updates are provided or if the update operation fails.
+   */
   async updateIndexById(indexName, id, update) {
+    this.logger.warn(
+      `Deprecation Warning: updateIndexById() is deprecated. 
+      Please use updateVector() instead. 
+      updateIndexById() will be removed on May 20th, 2025.`
+    );
+    await this.updateVector(indexName, id, update);
+  }
+  /**
+   * Updates a vector by its ID with the provided vector and/or metadata.
+   * @param indexName - The name of the index containing the vector.
+   * @param id - The ID of the vector to update.
+   * @param update - An object containing the vector and/or metadata to update.
+   * @param update.vector - An optional array of numbers representing the new vector.
+   * @param update.metadata - An optional record containing the new metadata.
+   * @returns A promise that resolves when the update is complete.
+   * @throws Will throw an error if no updates are provided or if the update operation fails.
+   */
+  async updateVector(indexName, id, update) {
     if (!update.vector && !update.metadata) {
       throw new Error("No updates provided");
     }
@@ -822,11 +868,37 @@ var PgVector = class extends MastraVector {
         WHERE vector_id = $1
       `;
       await client.query(query, values);
+    } catch (error) {
+      throw new Error(`Failed to update vector by id: ${id} for index: ${indexName}: ${error.message}`);
     } finally {
       client.release();
     }
   }
+  /**
+   * @deprecated Use {@link deleteVector} instead. This method will be removed on May 20th, 2025.
+   *
+   * Deletes a vector by its ID.
+   * @param indexName - The name of the index containing the vector.
+   * @param id - The ID of the vector to delete.
+   * @returns A promise that resolves when the deletion is complete.
+   * @throws Will throw an error if the deletion operation fails.
+   */
   async deleteIndexById(indexName, id) {
+    this.logger.warn(
+      `Deprecation Warning: deleteIndexById() is deprecated. 
+      Please use deleteVector() instead. 
+      deleteIndexById() will be removed on May 20th, 2025.`
+    );
+    await this.deleteVector(indexName, id);
+  }
+  /**
+   * Deletes a vector by its ID.
+   * @param indexName - The name of the index containing the vector.
+   * @param id - The ID of the vector to delete.
+   * @returns A promise that resolves when the deletion is complete.
+   * @throws Will throw an error if the deletion operation fails.
+   */
+  async deleteVector(indexName, id) {
     const client = await this.pool.connect();
     try {
       const tableName = this.getTableName(indexName);
@@ -835,6 +907,8 @@ var PgVector = class extends MastraVector {
         WHERE vector_id = $1
       `;
       await client.query(query, [id]);
+    } catch (error) {
+      throw new Error(`Failed to delete vector by id: ${id} for index: ${indexName}: ${error.message}`);
     } finally {
       client.release();
     }
@@ -867,7 +941,7 @@ var PostgresStore = class extends MastraStorage {
     this.pgp = pgPromise();
     if ("schema" in config && config.schema) {
       console.warn(
-        '[DEPRECATION NOTICE] The "schema" option in PostgresStore is deprecated. Please use "schemaName" instead. Support for "schema" will be removed in a future release.'
+        '[DEPRECATION NOTICE] The "schema" option in PostgresStore is deprecated. Please use "schemaName" instead. Support for "schema" will be removed on May 20th, 2025.'
       );
     }
     this.schema = config.schemaName ?? config.schema;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/pg",
-  "version": "0.3.4-alpha.2",
+  "version": "0.3.4-alpha.4",
   "description": "Postgres provider for Mastra - includes both vector and db storage capabilities",
   "type": "module",
   "main": "dist/index.js",
@@ -24,7 +24,7 @@
     "pg": "^8.13.3",
     "pg-promise": "^11.11.0",
     "xxhash-wasm": "^1.1.0",
-    "@mastra/core": "^0.9.4-alpha.2"
+    "@mastra/core": "^0.9.4-alpha.3"
   },
   "devDependencies": {
     "@microsoft/api-extractor": "^7.52.5",

package/src/storage/index.ts

@@ -23,7 +23,7 @@ import type { ISSLConfig } from 'pg-promise/typescript/pg-subset';
 export type PostgresConfig = {
   schemaName?: string;
   /**
-   * @deprecated Use `schemaName` instead. Support for `schema` will be removed in a future release.
+   * @deprecated Use `schemaName` instead. Support for `schema` will be removed on May 20th, 2025.
    */
   schema?: string;
 } & (
@@ -74,7 +74,7 @@ export class PostgresStore extends MastraStorage {
     // Deprecation notice for schema (old option)
     if ('schema' in config && config.schema) {
       console.warn(
-        '[DEPRECATION NOTICE] The "schema" option in PostgresStore is deprecated. Please use "schemaName" instead. Support for "schema" will be removed in a future release.',
+        '[DEPRECATION NOTICE] The "schema" option in PostgresStore is deprecated. Please use "schemaName" instead. Support for "schema" will be removed on May 20th, 2025.',
       );
     }
     this.schema = config.schemaName ?? config.schema;
@@ -203,11 +203,11 @@ export class PostgresStore extends MastraStorage {
     }
 
     if (fromDate) {
-      conditions.push(`createdAt >= $${idx++}`);
+      conditions.push(`createdAt >= \$${idx++}`);
     }
 
     if (toDate) {
-      conditions.push(`createdAt <= $${idx++}`);
+      conditions.push(`createdAt <= \$${idx++}`);
     }
 
     const whereClause = conditions.length > 0 ? `WHERE ${conditions.join(' AND ')}` : '';

package/src/vector/index.test.ts

@@ -285,7 +285,7 @@ describe('PgVector', () => {
           metadata: newMetaData,
         };
 
-        await vectorDB.updateIndexById(testIndexName, idToBeUpdated, update);
+        await vectorDB.updateVector(testIndexName, idToBeUpdated, update);
 
         const results: QueryResult[] = await vectorDB.query({
           indexName: testIndexName,
@@ -311,7 +311,7 @@ describe('PgVector', () => {
           metadata: newMetaData,
         };
 
-        await vectorDB.updateIndexById(testIndexName, idToBeUpdated, update);
+        await vectorDB.updateVector(testIndexName, idToBeUpdated, update);
 
         const results: QueryResult[] = await vectorDB.query({
           indexName: testIndexName,
@@ -335,7 +335,7 @@ describe('PgVector', () => {
           vector: newVector,
         };
 
-        await vectorDB.updateIndexById(testIndexName, idToBeUpdated, update);
+        await vectorDB.updateVector(testIndexName, idToBeUpdated, update);
 
         const results: QueryResult[] = await vectorDB.query({
           indexName: testIndexName,
@@ -348,7 +348,7 @@ describe('PgVector', () => {
       });
 
       it('should throw exception when no updates are given', async () => {
-        await expect(vectorDB.updateIndexById(testIndexName, 'id', {})).rejects.toThrow('No updates provided');
+        await expect(vectorDB.updateVector(testIndexName, 'id', {})).rejects.toThrow('No updates provided');
       });
     });
 
@@ -372,7 +372,7 @@ describe('PgVector', () => {
         expect(ids).toHaveLength(3);
         const idToBeDeleted = ids[0];
 
-        await vectorDB.deleteIndexById(testIndexName, idToBeDeleted);
+        await vectorDB.deleteVector(testIndexName, idToBeDeleted);
 
         const results: QueryResult[] = await vectorDB.query({
           indexName: testIndexName,
@@ -2139,6 +2139,26 @@ describe('PgVector', () => {
         }
       });
 
+      it('should describe index in custom schema', async () => {
+        // Create index in custom schema
+        await customSchemaVectorDB.createIndex({
+          indexName: testIndexName,
+          dimension: 3,
+          metric: 'dotproduct',
+          indexConfig: { type: 'hnsw' },
+        });
+        // Insert a vector
+        await customSchemaVectorDB.upsert({ indexName: testIndexName, vectors: [[1, 2, 3]] });
+        // Describe the index
+        const stats = await customSchemaVectorDB.describeIndex(testIndexName);
+        expect(stats).toMatchObject({
+          dimension: 3,
+          metric: 'dotproduct',
+          type: 'hnsw',
+          count: 1,
+        });
+      });
+
       it('should allow same index name in different schemas', async () => {
         // Create same index name in both schemas
         await vectorDB.createIndex({ indexName: testIndexName, dimension: 3 });
@@ -2195,7 +2215,7 @@ describe('PgVector', () => {
         });
 
         // Test delete operation
-        await customSchemaVectorDB.deleteIndexById(testIndexName, id!);
+        await customSchemaVectorDB.deleteVector(testIndexName, id!);
 
         // Verify deletion
         const results = await customSchemaVectorDB.query({

package/src/vector/index.ts

@@ -72,7 +72,7 @@ export class PgVector extends MastraVector {
 
   /**
    * @deprecated Passing connectionString as a string is deprecated.
-   * Use the object parameter instead. This signature will be removed on May 20th.
+   * Use the object parameter instead. This signature will be removed on May 20th, 2025.
    */
   constructor(connectionString: string);
   constructor(config: {
@@ -101,7 +101,7 @@ export class PgVector extends MastraVector {
         Please use an object parameter instead:
         new PgVector({ connectionString })
 
-        The string signature will be removed on May 20th.`,
+        The string signature will be removed on May 20th, 2025.`,
       );
       connectionString = config;
       schemaName = undefined;
@@ -404,12 +404,14 @@ export class PgVector extends MastraVector {
 
   /**
    * @deprecated This function is deprecated. Use buildIndex instead
+   * This function will be removed on May 20th, 2025
    */
   async defineIndex(
     indexName: string,
     metric: 'cosine' | 'euclidean' | 'dotproduct' = 'cosine',
     indexConfig: IndexConfig,
   ): Promise<void> {
+    console.warn('defineIndex is deprecated. Use buildIndex instead. This function will be removed on May 20th, 2025');
     return this.buildIndex({ indexName, metric, indexConfig });
   }
 
@@ -555,6 +557,21 @@ export class PgVector extends MastraVector {
     try {
       const tableName = this.getTableName(indexName);
 
+      // Check if table exists with a vector column
+      const tableExistsQuery = `
+        SELECT 1
+        FROM information_schema.columns
+        WHERE table_schema = $1
+          AND table_name = $2
+          AND udt_name = 'vector'
+        LIMIT 1;
+      `;
+      const tableExists = await client.query(tableExistsQuery, [this.schema || 'public', indexName]);
+
+      if (tableExists.rows.length === 0) {
+        throw new Error(`Vector table ${tableName} does not exist`);
+      }
+
       // Get vector dimension
       const dimensionQuery = `
                 SELECT atttypmod as dimension
@@ -579,13 +596,15 @@ export class PgVector extends MastraVector {
             JOIN pg_class c ON i.indexrelid = c.oid
             JOIN pg_am am ON c.relam = am.oid
             JOIN pg_opclass opclass ON i.indclass[0] = opclass.oid
-            WHERE c.relname = '${tableName}_vector_idx';
+            JOIN pg_namespace n ON c.relnamespace = n.oid
+            WHERE c.relname = $1
+            AND n.nspname = $2;
             `;
 
       const [dimResult, countResult, indexResult] = await Promise.all([
         client.query(dimensionQuery, [tableName]),
         client.query(countQuery),
-        client.query(indexQuery),
+        client.query(indexQuery, [`${indexName}_vector_idx`, this.schema || 'public']),
       ]);
 
       const { index_method, index_def, operator_class } = indexResult.rows[0] || {
@@ -661,7 +680,42 @@ export class PgVector extends MastraVector {
     await this.pool.end();
   }
 
+  /**
+   * @deprecated Use {@link updateVector} instead. This method will be removed on May 20th, 2025.
+   *
+   * Updates a vector by its ID with the provided vector and/or metadata.
+   * @param indexName - The name of the index containing the vector.
+   * @param id - The ID of the vector to update.
+   * @param update - An object containing the vector and/or metadata to update.
+   * @param update.vector - An optional array of numbers representing the new vector.
+   * @param update.metadata - An optional record containing the new metadata.
+   * @returns A promise that resolves when the update is complete.
+   * @throws Will throw an error if no updates are provided or if the update operation fails.
+   */
   async updateIndexById(
+    indexName: string,
+    id: string,
+    update: { vector?: number[]; metadata?: Record<string, any> },
+  ): Promise<void> {
+    this.logger.warn(
+      `Deprecation Warning: updateIndexById() is deprecated. 
+      Please use updateVector() instead. 
+      updateIndexById() will be removed on May 20th, 2025.`,
+    );
+    await this.updateVector(indexName, id, update);
+  }
+
+  /**
+   * Updates a vector by its ID with the provided vector and/or metadata.
+   * @param indexName - The name of the index containing the vector.
+   * @param id - The ID of the vector to update.
+   * @param update - An object containing the vector and/or metadata to update.
+   * @param update.vector - An optional array of numbers representing the new vector.
+   * @param update.metadata - An optional record containing the new metadata.
+   * @returns A promise that resolves when the update is complete.
+   * @throws Will throw an error if no updates are provided or if the update operation fails.
+   */
+  async updateVector(
     indexName: string,
     id: string,
     update: {
@@ -705,12 +759,39 @@ export class PgVector extends MastraVector {
       `;
 
       await client.query(query, values);
+    } catch (error: any) {
+      throw new Error(`Failed to update vector by id: ${id} for index: ${indexName}: ${error.message}`);
     } finally {
       client.release();
     }
   }
 
+  /**
+   * @deprecated Use {@link deleteVector} instead. This method will be removed on May 20th, 2025.
+   *
+   * Deletes a vector by its ID.
+   * @param indexName - The name of the index containing the vector.
+   * @param id - The ID of the vector to delete.
+   * @returns A promise that resolves when the deletion is complete.
+   * @throws Will throw an error if the deletion operation fails.
+   */
   async deleteIndexById(indexName: string, id: string): Promise<void> {
+    this.logger.warn(
+      `Deprecation Warning: deleteIndexById() is deprecated. 
+      Please use deleteVector() instead. 
+      deleteIndexById() will be removed on May 20th, 2025.`,
+    );
+    await this.deleteVector(indexName, id);
+  }
+
+  /**
+   * Deletes a vector by its ID.
+   * @param indexName - The name of the index containing the vector.
+   * @param id - The ID of the vector to delete.
+   * @returns A promise that resolves when the deletion is complete.
+   * @throws Will throw an error if the deletion operation fails.
+   */
+  async deleteVector(indexName: string, id: string): Promise<void> {
     const client = await this.pool.connect();
     try {
       const tableName = this.getTableName(indexName);
@@ -719,6 +800,8 @@ export class PgVector extends MastraVector {
         WHERE vector_id = $1
       `;
       await client.query(query, [id]);
+    } catch (error: any) {
+      throw new Error(`Failed to delete vector by id: ${id} for index: ${indexName}: ${error.message}`);
     } finally {
       client.release();
     }