@mastra/pg 0.1.8-alpha.9 → 0.2.0

package/.turbo/turbo-build.log

@@ -1,23 +1,23 @@
 
-> @mastra/pg@0.1.8-alpha.9 build /home/runner/work/mastra/mastra/stores/pg
-> tsup src/index.ts --format esm,cjs --experimental-dts --clean --treeshake
+> @mastra/pg@0.2.0-alpha.12 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.3.6
+CLI tsup v8.4.0
 TSC Build start
-TSC ⚡️ Build success in 9376ms
+TSC ⚡️ Build success in 9808ms
 DTS Build start
 CLI Target: es2022
 Analysis will use the bundled TypeScript version 5.7.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.7.3
 Writing package typings: /home/runner/work/mastra/mastra/stores/pg/dist/_tsup-dts-rollup.d.cts
-DTS ⚡️ Build success in 10054ms
+DTS ⚡️ Build success in 10902ms
 CLI Cleaning output folder
 ESM Build start
 CJS Build start
-ESM dist/index.js 33.26 KB
-ESM ⚡️ Build success in 1131ms
-CJS dist/index.cjs 33.66 KB
-CJS ⚡️ Build success in 1131ms
+ESM dist/index.js 34.40 KB
+ESM ⚡️ Build success in 1053ms
+CJS dist/index.cjs 34.80 KB
+CJS ⚡️ Build success in 1065ms

package/CHANGELOG.md

@@ -1,5 +1,83 @@
 # @mastra/pg
 
+## 0.2.0
+
+### Minor Changes
+
+- 3de7e06: Added new operations implementations for MastraVector interface methods in pg vector store
+
+### Patch Changes
+
+- fd4a1d7: Update cjs bundling to make sure files are split
+- Updated dependencies [a910463]
+- Updated dependencies [59df7b6]
+- Updated dependencies [22643eb]
+- Updated dependencies [6feb23f]
+- Updated dependencies [f2d6727]
+- Updated dependencies [7a7a547]
+- Updated dependencies [29f3a82]
+- Updated dependencies [3d0e290]
+- Updated dependencies [e9fbac5]
+- Updated dependencies [301e4ee]
+- Updated dependencies [ee667a2]
+- Updated dependencies [dfbe4e9]
+- Updated dependencies [dab255b]
+- Updated dependencies [1e8bcbc]
+- Updated dependencies [f6678e4]
+- Updated dependencies [9e81f35]
+- Updated dependencies [c93798b]
+- Updated dependencies [a85ab24]
+- Updated dependencies [dbd9f2d]
+- Updated dependencies [59df7b6]
+- Updated dependencies [caefaa2]
+- Updated dependencies [c151ae6]
+- Updated dependencies [52e0418]
+- Updated dependencies [d79aedf]
+- Updated dependencies [03236ec]
+- Updated dependencies [3764e71]
+- Updated dependencies [df982db]
+- Updated dependencies [a171b37]
+- Updated dependencies [506f1d5]
+- Updated dependencies [02ffb7b]
+- Updated dependencies [0461849]
+- Updated dependencies [2259379]
+- Updated dependencies [aeb5e36]
+- Updated dependencies [f2301de]
+- Updated dependencies [358f069]
+- Updated dependencies [fd4a1d7]
+- Updated dependencies [c139344]
+  - @mastra/core@0.5.0
+
+## 0.2.0-alpha.12
+
+### Patch Changes
+
+- Updated dependencies [a85ab24]
+  - @mastra/core@0.5.0-alpha.12
+
+## 0.2.0-alpha.11
+
+### Patch Changes
+
+- fd4a1d7: Update cjs bundling to make sure files are split
+- Updated dependencies [7a7a547]
+- Updated dependencies [c93798b]
+- Updated dependencies [dbd9f2d]
+- Updated dependencies [a171b37]
+- Updated dependencies [fd4a1d7]
+  - @mastra/core@0.5.0-alpha.11
+
+## 0.2.0-alpha.10
+
+### Minor Changes
+
+- 3de7e06: Added new operations implementations for MastraVector interface methods in pg vector store
+
+### Patch Changes
+
+- Updated dependencies [a910463]
+  - @mastra/core@0.5.0-alpha.10
+
 ## 0.1.8-alpha.9
 
 ### Patch Changes

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

@@ -216,6 +216,11 @@ declare class PgVector extends MastraVector {
     deleteIndex(indexName: string): Promise<void>;
     truncateIndex(indexName: string): Promise<void>;
     disconnect(): Promise<void>;
+    updateIndexById(indexName: string, id: string, update: {
+        vector?: number[];
+        metadata?: Record<string, any>;
+    }): Promise<void>;
+    deleteIndexById(indexName: string, id: string): Promise<void>;
 }
 export { PgVector }
 export { PgVector as PgVector_alias_1 }

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

@@ -216,6 +216,11 @@ declare class PgVector extends MastraVector {
     deleteIndex(indexName: string): Promise<void>;
     truncateIndex(indexName: string): Promise<void>;
     disconnect(): Promise<void>;
+    updateIndexById(indexName: string, id: string, update: {
+        vector?: number[];
+        metadata?: Record<string, any>;
+    }): Promise<void>;
+    deleteIndexById(indexName: string, id: string): Promise<void>;
 }
 export { PgVector }
 export { PgVector as PgVector_alias_1 }

package/dist/index.cjs

@@ -589,6 +589,49 @@ var PgVector = class extends vector.MastraVector {
   async disconnect() {
     await this.pool.end();
   }
+  async updateIndexById(indexName, id, update) {
+    if (!update.vector && !update.metadata) {
+      throw new Error("No updates provided");
+    }
+    const client = await this.pool.connect();
+    try {
+      let updateParts = [];
+      let values = [id];
+      let valueIndex = 2;
+      if (update.vector) {
+        updateParts.push(`embedding = $${valueIndex}::vector`);
+        values.push(`[${update.vector.join(",")}]`);
+        valueIndex++;
+      }
+      if (update.metadata) {
+        updateParts.push(`metadata = $${valueIndex}::jsonb`);
+        values.push(JSON.stringify(update.metadata));
+      }
+      if (updateParts.length === 0) {
+        return;
+      }
+      const query = `
+        UPDATE ${indexName}
+        SET ${updateParts.join(", ")}
+        WHERE vector_id = $1
+      `;
+      await client.query(query, values);
+    } finally {
+      client.release();
+    }
+  }
+  async deleteIndexById(indexName, id) {
+    const client = await this.pool.connect();
+    try {
+      const query = `
+        DELETE FROM ${indexName}
+        WHERE vector_id = $1
+      `;
+      await client.query(query, [id]);
+    } finally {
+      client.release();
+    }
+  }
 };
 var PostgresStore = class extends storage.MastraStorage {
   db;

package/dist/index.js

@@ -582,6 +582,49 @@ var PgVector = class extends MastraVector {
   async disconnect() {
     await this.pool.end();
   }
+  async updateIndexById(indexName, id, update) {
+    if (!update.vector && !update.metadata) {
+      throw new Error("No updates provided");
+    }
+    const client = await this.pool.connect();
+    try {
+      let updateParts = [];
+      let values = [id];
+      let valueIndex = 2;
+      if (update.vector) {
+        updateParts.push(`embedding = $${valueIndex}::vector`);
+        values.push(`[${update.vector.join(",")}]`);
+        valueIndex++;
+      }
+      if (update.metadata) {
+        updateParts.push(`metadata = $${valueIndex}::jsonb`);
+        values.push(JSON.stringify(update.metadata));
+      }
+      if (updateParts.length === 0) {
+        return;
+      }
+      const query = `
+        UPDATE ${indexName}
+        SET ${updateParts.join(", ")}
+        WHERE vector_id = $1
+      `;
+      await client.query(query, values);
+    } finally {
+      client.release();
+    }
+  }
+  async deleteIndexById(indexName, id) {
+    const client = await this.pool.connect();
+    try {
+      const query = `
+        DELETE FROM ${indexName}
+        WHERE vector_id = $1
+      `;
+      await client.query(query, [id]);
+    } finally {
+      client.release();
+    }
+  }
 };
 var PostgresStore = class extends MastraStorage {
   db;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/pg",
-  "version": "0.1.8-alpha.9",
+  "version": "0.2.0",
   "description": "Postgres provider for Mastra - includes both vector and db storage capabilities",
   "type": "module",
   "main": "dist/index.js",
@@ -19,22 +19,22 @@
     "./package.json": "./package.json"
   },
   "dependencies": {
-    "pg": "^8.13.1",
-    "pg-promise": "^11.5.4",
-    "@mastra/core": "^0.5.0-alpha.9"
+    "pg": "^8.13.3",
+    "pg-promise": "^11.11.0",
+    "@mastra/core": "^0.5.0"
   },
   "devDependencies": {
-    "@microsoft/api-extractor": "^7.49.2",
-    "@types/node": "^22.13.1",
-    "@types/pg": "^8.11.10",
-    "tsup": "^8.0.1",
-    "typescript": "^5.7.3",
-    "vitest": "^3.0.4",
-    "eslint": "^9.20.1",
-    "@internal/lint": "0.0.0"
+    "@microsoft/api-extractor": "^7.52.1",
+    "@types/node": "^22.13.10",
+    "@types/pg": "^8.11.11",
+    "eslint": "^9.22.0",
+    "tsup": "^8.4.0",
+    "typescript": "^5.8.2",
+    "vitest": "^3.0.8",
+    "@internal/lint": "0.0.1"
   },
   "scripts": {
-    "build": "tsup src/index.ts --format esm,cjs --experimental-dts --clean --treeshake",
+    "build": "tsup src/index.ts --format esm,cjs --experimental-dts --clean --treeshake=smallest --splitting",
     "build:watch": "pnpm build --watch",
     "pretest": "docker compose up -d && (for i in $(seq 1 30); do docker compose exec -T db pg_isready -U postgres && break || (sleep 1; [ $i -eq 30 ] && exit 1); done)",
     "test": "vitest run",

package/src/vector/index.test.ts

@@ -1,6 +1,7 @@
 import { describe, it, expect, beforeAll, afterAll, beforeEach, afterEach, vi } from 'vitest';
 
 import { PgVector } from '.';
+import { type QueryResult } from '@mastra/core';
 
 describe('PgVector', () => {
   let vectorDB: PgVector;
@@ -232,6 +233,136 @@ describe('PgVector', () => {
       });
     });
 
+    describe('updates', () => {
+      const testVectors = [
+        [1, 2, 3],
+        [4, 5, 6],
+        [7, 8, 9],
+      ];
+
+      beforeEach(async () => {
+        await vectorDB.createIndex({ indexName: testIndexName, dimension: 3 });
+      });
+
+      afterEach(async () => {
+        await vectorDB.deleteIndex(testIndexName);
+      });
+
+      it('should update the vector by id', async () => {
+        const ids = await vectorDB.upsert({ indexName: testIndexName, vectors: testVectors });
+        expect(ids).toHaveLength(3);
+
+        const idToBeUpdated = ids[0];
+        const newVector = [1, 2, 3];
+        const newMetaData = {
+          test: 'updates',
+        };
+
+        const update = {
+          vector: newVector,
+          metadata: newMetaData,
+        };
+
+        await vectorDB.updateIndexById(testIndexName, idToBeUpdated, update);
+
+        const results: QueryResult[] = await vectorDB.query({
+          indexName: testIndexName,
+          queryVector: newVector,
+          topK: 2,
+          includeVector: true,
+        });
+        expect(results[0]?.id).toBe(idToBeUpdated);
+        expect(results[0]?.vector).toEqual(newVector);
+        expect(results[0]?.metadata).toEqual(newMetaData);
+      });
+
+      it('should only update the metadata by id', async () => {
+        const ids = await vectorDB.upsert({ indexName: testIndexName, vectors: testVectors });
+        expect(ids).toHaveLength(3);
+
+        const idToBeUpdated = ids[0];
+        const newMetaData = {
+          test: 'updates',
+        };
+
+        const update = {
+          metadata: newMetaData,
+        };
+
+        await vectorDB.updateIndexById(testIndexName, idToBeUpdated, update);
+
+        const results: QueryResult[] = await vectorDB.query({
+          indexName: testIndexName,
+          queryVector: testVectors[0],
+          topK: 2,
+          includeVector: true,
+        });
+        expect(results[0]?.id).toBe(idToBeUpdated);
+        expect(results[0]?.vector).toEqual(testVectors[0]);
+        expect(results[0]?.metadata).toEqual(newMetaData);
+      });
+
+      it('should only update vector embeddings by id', async () => {
+        const ids = await vectorDB.upsert({ indexName: testIndexName, vectors: testVectors });
+        expect(ids).toHaveLength(3);
+
+        const idToBeUpdated = ids[0];
+        const newVector = [1, 2, 3];
+
+        const update = {
+          vector: newVector,
+        };
+
+        await vectorDB.updateIndexById(testIndexName, idToBeUpdated, update);
+
+        const results: QueryResult[] = await vectorDB.query({
+          indexName: testIndexName,
+          queryVector: newVector,
+          topK: 2,
+          includeVector: true,
+        });
+        expect(results[0]?.id).toBe(idToBeUpdated);
+        expect(results[0]?.vector).toEqual(newVector);
+      });
+
+      it('should throw exception when no updates are given', () => {
+        expect(vectorDB.updateIndexById(testIndexName, 'id', {})).rejects.toThrow('No updates provided');
+      });
+    });
+
+    describe('deletes', () => {
+      const testVectors = [
+        [1, 2, 3],
+        [4, 5, 6],
+        [7, 8, 9],
+      ];
+
+      beforeEach(async () => {
+        await vectorDB.createIndex({ indexName: testIndexName, dimension: 3 });
+      });
+
+      afterEach(async () => {
+        await vectorDB.deleteIndex(testIndexName);
+      });
+
+      it('should delete the vector by id', async () => {
+        const ids = await vectorDB.upsert({ indexName: testIndexName, vectors: testVectors });
+        expect(ids).toHaveLength(3);
+        const idToBeDeleted = ids[0];
+
+        await vectorDB.deleteIndexById(testIndexName, idToBeDeleted);
+
+        const results: QueryResult[] = await vectorDB.query({
+          indexName: testIndexName,
+          queryVector: [1.0, 0.0, 0.0],
+          topK: 2,
+        });
+
+        expect(results).toHaveLength(2);
+        expect(results.map(res => res.id)).not.toContain(idToBeDeleted);
+      });
+    });
+
     describe('Basic Query Operations', () => {
       ['flat', 'hnsw', 'ivfflat'].forEach(indexType => {
         const indexName = `test_query_2_${indexType}`;

package/src/vector/index.ts

@@ -426,4 +426,64 @@ export class PgVector extends MastraVector {
   async disconnect() {
     await this.pool.end();
   }
+
+  async updateIndexById(
+    indexName: string,
+    id: string,
+    update: {
+      vector?: number[];
+      metadata?: Record<string, any>;
+    },
+  ): Promise<void> {
+    if (!update.vector && !update.metadata) {
+      throw new Error('No updates provided');
+    }
+
+    const client = await this.pool.connect();
+    try {
+      let updateParts = [];
+      let values = [id];
+      let valueIndex = 2;
+
+      if (update.vector) {
+        updateParts.push(`embedding = $${valueIndex}::vector`);
+        values.push(`[${update.vector.join(',')}]`);
+        valueIndex++;
+      }
+
+      if (update.metadata) {
+        updateParts.push(`metadata = $${valueIndex}::jsonb`);
+        values.push(JSON.stringify(update.metadata));
+      }
+
+      if (updateParts.length === 0) {
+        return;
+      }
+
+      // query looks like this:
+      // UPDATE table SET embedding = $2::vector, metadata = $3::jsonb WHERE id = $1
+      const query = `
+        UPDATE ${indexName}
+        SET ${updateParts.join(', ')}
+        WHERE vector_id = $1
+      `;
+
+      await client.query(query, values);
+    } finally {
+      client.release();
+    }
+  }
+
+  async deleteIndexById(indexName: string, id: string): Promise<void> {
+    const client = await this.pool.connect();
+    try {
+      const query = `
+        DELETE FROM ${indexName}
+        WHERE vector_id = $1
+      `;
+      await client.query(query, [id]);
+    } finally {
+      client.release();
+    }
+  }
 }