@mastra/pg 0.3.4-alpha.1 → 0.3.4-alpha.3

package/.turbo/turbo-build.log

@@ -1,23 +1,23 @@
 
-> @mastra/pg@0.3.4-alpha.1 build /home/runner/work/mastra/mastra/stores/pg
+> @mastra/pg@0.3.4-alpha.3 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 10163ms
+TSC ⚡️ Build success in 9567ms
 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 13223ms
+DTS ⚡️ Build success in 11821ms
 CLI Cleaning output folder
 ESM Build start
 CJS Build start
-ESM dist/index.js 54.89 KB
-ESM ⚡️ Build success in 1811ms
-CJS dist/index.cjs 55.37 KB
-CJS ⚡️ Build success in 1811ms
+CJS dist/index.cjs 55.91 KB
+CJS ⚡️ Build success in 1357ms
+ESM dist/index.js 55.43 KB
+ESM ⚡️ Build success in 1359ms

package/CHANGELOG.md

@@ -1,5 +1,21 @@
 # @mastra/pg
 
+## 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
+
+- Updated dependencies [3171b5b]
+- Updated dependencies [973e5ac]
+- Updated dependencies [9e1eff5]
+  - @mastra/core@0.9.4-alpha.2
+
 ## 0.3.4-alpha.1
 
 ### Patch Changes

package/dist/index.cjs

@@ -718,6 +718,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 +749,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",

package/dist/index.js

@@ -710,6 +710,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 +741,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",

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/pg",
-  "version": "0.3.4-alpha.1",
+  "version": "0.3.4-alpha.3",
   "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.1"
+    "@mastra/core": "^0.9.4-alpha.2"
   },
   "devDependencies": {
     "@microsoft/api-extractor": "^7.52.5",

package/src/storage/index.ts

@@ -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

@@ -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 });

package/src/vector/index.ts

@@ -555,6 +555,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 +594,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] || {