@mastra/pg 0.1.5 → 0.1.6-alpha.1

package/.turbo/turbo-build.log

@@ -1,18 +1,18 @@
 
-> @mastra/pg@0.1.5-alpha.2 build /home/runner/work/mastra/mastra/stores/pg
+> @mastra/pg@0.1.6-alpha.1 build /home/runner/work/mastra/mastra/stores/pg
 > tsup src/index.ts --format esm --experimental-dts --clean --treeshake
 
 CLI Building entry: src/index.ts
 CLI Using tsconfig: tsconfig.json
 CLI tsup v8.3.6
 TSC Build start
-TSC ⚡️ Build success in 9453ms
+TSC ⚡️ Build success in 10385ms
 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
-DTS ⚡️ Build success in 7106ms
+DTS ⚡️ Build success in 5525ms
 CLI Cleaning output folder
 ESM Build start
-ESM dist/index.js 32.67 KB
-ESM ⚡️ Build success in 640ms
+ESM dist/index.js 32.82 KB
+ESM ⚡️ Build success in 642ms

package/CHANGELOG.md

@@ -1,5 +1,26 @@
 # @mastra/pg
 
+## 0.1.6-alpha.1
+
+### Patch Changes
+
+- 0d185b1: Ensure proper message sort order for tool calls and results when using Memory semanticRecall feature
+- f6a1de3: Renamed defineIndex to buildIndex
+- Updated dependencies [0d185b1]
+- Updated dependencies [ed55f1d]
+- Updated dependencies [8d13b14]
+- Updated dependencies [3ee4831]
+- Updated dependencies [108793c]
+- Updated dependencies [5f28f44]
+  - @mastra/core@0.4.3-alpha.1
+
+## 0.1.6-alpha.0
+
+### Patch Changes
+
+- Updated dependencies [06aa827]
+  - @mastra/core@0.4.3-alpha.0
+
 ## 0.1.5
 
 ### Patch Changes

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

@@ -1,23 +1,23 @@
-import { ArrayOperator } from '@mastra/core/filter';
+import type { ArrayOperator } from '@mastra/core/filter';
 import { BaseFilterTranslator } from '@mastra/core/filter';
-import { BasicOperator } from '@mastra/core/filter';
-import { ElementOperator } from '@mastra/core/filter';
-import { EvalRow } from '@mastra/core/storage';
-import { Filter } from '@mastra/core/filter';
-import { IndexStats } from '@mastra/core/vector';
-import { LogicalOperator } from '@mastra/core/filter';
+import type { BasicOperator } from '@mastra/core/filter';
+import type { ElementOperator } from '@mastra/core/filter';
+import type { EvalRow } from '@mastra/core/storage';
+import type { Filter } from '@mastra/core/filter';
+import type { IndexStats } from '@mastra/core/vector';
+import type { LogicalOperator } from '@mastra/core/filter';
 import { MastraStorage } from '@mastra/core/storage';
 import { MastraVector } from '@mastra/core/vector';
-import { MessageType } from '@mastra/core/memory';
-import { NumericOperator } from '@mastra/core/filter';
-import { OperatorSupport } from '@mastra/core/filter';
-import { QueryResult } from '@mastra/core/vector';
-import { RegexOperator } from '@mastra/core/filter';
-import { StorageColumn } from '@mastra/core/storage';
-import { StorageGetMessagesArg } from '@mastra/core/storage';
-import { StorageThreadType } from '@mastra/core/memory';
-import { TABLE_NAMES } from '@mastra/core/storage';
-import { WorkflowRunState } from '@mastra/core/workflows';
+import type { MessageType } from '@mastra/core/memory';
+import type { NumericOperator } from '@mastra/core/filter';
+import type { OperatorSupport } from '@mastra/core/filter';
+import type { QueryResult } from '@mastra/core/vector';
+import type { RegexOperator } from '@mastra/core/filter';
+import type { StorageColumn } from '@mastra/core/storage';
+import type { StorageGetMessagesArg } from '@mastra/core/storage';
+import type { StorageThreadType } from '@mastra/core/memory';
+import type { TABLE_NAMES } from '@mastra/core/storage';
+import type { WorkflowRunState } from '@mastra/core/workflows';
 
 export declare const baseTestConfigs: {
     smokeTests: {
@@ -173,7 +173,11 @@ declare class PgVector extends MastraVector {
     }): Promise<QueryResult[]>;
     upsert(indexName: string, vectors: number[][], metadata?: Record<string, any>[], ids?: string[]): Promise<string[]>;
     createIndex(indexName: string, dimension: number, metric?: 'cosine' | 'euclidean' | 'dotproduct', indexConfig?: IndexConfig, defineIndex?: boolean): Promise<void>;
+    /**
+     * @deprecated This function is deprecated. Use buildIndex instead
+     */
     defineIndex(indexName: string, metric: "cosine" | "euclidean" | "dotproduct" | undefined, indexConfig: IndexConfig): Promise<void>;
+    buildIndex(indexName: string, metric: "cosine" | "euclidean" | "dotproduct" | undefined, indexConfig: IndexConfig): Promise<void>;
     listIndexes(): Promise<string[]>;
     describeIndex(indexName: string): Promise<PGIndexStats>;
     deleteIndex(indexName: string): Promise<void>;
@@ -199,7 +203,7 @@ declare class PostgresStore extends MastraStorage {
     private db;
     private pgp;
     constructor(config: PostgresConfig);
-    getEvalsByAgentName(agentName: string, type?: 'test' | 'live'): Promise<EvalRow[]>;
+    getEvalsByAgentName(_agentName: string, _type?: 'test' | 'live'): Promise<EvalRow[]>;
     batchInsert({ tableName, records }: {
         tableName: TABLE_NAMES;
         records: Record<string, any>[];

package/dist/index.js

@@ -1,9 +1,7 @@
 import { MastraVector } from '@mastra/core/vector';
 import pg from 'pg';
 import { BaseFilterTranslator } from '@mastra/core/filter';
-import '@mastra/core/memory';
 import { MastraStorage } from '@mastra/core/storage';
-import '@mastra/core/workflows';
 import pgPromise from 'pg-promise';
 
 // src/vector/index.ts
@@ -73,6 +71,8 @@ var PGFilterTranslator = class extends BaseFilterTranslator {
     return { $regex: flags ? `(?${flags})${pattern}` : pattern };
   }
 };
+
+// src/vector/sql-builder.ts
 var createBasicOperator = (symbol) => {
   return (key, paramIndex) => ({
     sql: `CASE 
@@ -420,7 +420,13 @@ var PgVector = class extends MastraVector {
       client.release();
     }
   }
+  /**
+   * @deprecated This function is deprecated. Use buildIndex instead
+   */
   async defineIndex(indexName, metric = "cosine", indexConfig) {
+    return this.buildIndex(indexName, metric, indexConfig);
+  }
+  async buildIndex(indexName, metric = "cosine", indexConfig) {
     const client = await this.pool.connect();
     try {
       await client.query(`DROP INDEX IF EXISTS ${indexName}_vector_idx`);
@@ -576,7 +582,7 @@ var PostgresStore = class extends MastraStorage {
       }
     );
   }
-  getEvalsByAgentName(agentName, type) {
+  getEvalsByAgentName(_agentName, _type) {
     throw new Error("Method not implemented.");
   }
   async batchInsert({ tableName, records }) {
@@ -874,11 +880,11 @@ var PostgresStore = class extends MastraStorage {
           WITH ordered_messages AS (
             SELECT 
               *,
-              ROW_NUMBER() OVER (ORDER BY "createdAt") as row_num
+              ROW_NUMBER() OVER (ORDER BY "createdAt" DESC) as row_num
             FROM "${MastraStorage.TABLE_MESSAGES}"
             WHERE thread_id = $1
           )
-          SELECT DISTINCT ON (m.id)
+          SELECT
             m.id, 
             m.content, 
             m.role, 
@@ -898,7 +904,7 @@ var PostgresStore = class extends MastraStorage {
               (m.row_num <= target.row_num + $4 AND m.row_num > target.row_num)
             )
           )
-          ORDER BY m.id, m."createdAt"
+          ORDER BY m."createdAt" DESC
           `,
           [
             threadId,
@@ -932,7 +938,7 @@ var PostgresStore = class extends MastraStorage {
         if (typeof message.content === "string") {
           try {
             message.content = JSON.parse(message.content);
-          } catch (e) {
+          } catch {
           }
         }
       });

package/eslint.config.js

@@ -0,0 +1,6 @@
+import { createConfig } from '@internal/lint/eslint';
+
+const config = await createConfig();
+
+/** @type {import("eslint").Linter.Config[]} */
+export default [...config.map(conf => ({ ...conf, ignores: [...(conf.ignores || []), '**/vitest.perf.config.ts'] }))];

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/pg",
-  "version": "0.1.5",
+  "version": "0.1.6-alpha.1",
   "description": "Postgres provider for Mastra - includes both vector and db storage capabilities",
   "type": "module",
   "main": "dist/index.js",
@@ -17,7 +17,7 @@
   "dependencies": {
     "pg": "^8.13.1",
     "pg-promise": "^11.5.4",
-    "@mastra/core": "^0.4.2"
+    "@mastra/core": "^0.4.3-alpha.1"
   },
   "devDependencies": {
     "@microsoft/api-extractor": "^7.49.2",
@@ -25,7 +25,9 @@
     "@types/pg": "^8.11.10",
     "tsup": "^8.0.1",
     "typescript": "^5.7.3",
-    "vitest": "^3.0.4"
+    "vitest": "^3.0.4",
+    "eslint": "^9.20.1",
+    "@internal/lint": "0.0.0"
   },
   "scripts": {
     "build": "tsup src/index.ts --format esm --experimental-dts --clean --treeshake",
@@ -38,6 +40,7 @@
     "posttest:perf": "docker compose -f docker-compose.perf.yaml down -v",
     "pretest:watch": "docker compose up -d",
     "test:watch": "vitest watch",
-    "posttest:watch": "docker compose down -v"
+    "posttest:watch": "docker compose down -v",
+    "lint": "eslint ."
   }
 }

package/src/storage/index.test.ts

@@ -1,8 +1,9 @@
-import { WorkflowRunState } from '@mastra/core/workflows';
 import { randomUUID } from 'crypto';
+import type { WorkflowRunState } from '@mastra/core/workflows';
 import { describe, it, expect, beforeAll, beforeEach, afterAll } from 'vitest';
 
-import { PostgresStore, type PostgresConfig } from '.';
+import { PostgresStore } from '.';
+import type { PostgresConfig } from '.';
 
 const TEST_CONFIG: PostgresConfig = {
   host: process.env.POSTGRES_HOST || 'localhost',

package/src/storage/index.ts

@@ -1,22 +1,9 @@
-import { type MessageType, type StorageThreadType } from '@mastra/core/memory';
-import {
-  MastraStorage,
-  type EvalRow,
-  type StorageColumn,
-  type StorageGetMessagesArg,
-  type TABLE_NAMES,
-} from '@mastra/core/storage';
-import { type WorkflowRunState } from '@mastra/core/workflows';
+import type { MessageType, StorageThreadType } from '@mastra/core/memory';
+import { MastraStorage } from '@mastra/core/storage';
+import type { EvalRow, StorageColumn, StorageGetMessagesArg, TABLE_NAMES } from '@mastra/core/storage';
+import type { WorkflowRunState } from '@mastra/core/workflows';
 import pgPromise from 'pg-promise';
 
-function safelyParseJSON(json: string): any {
-  try {
-    return JSON.parse(json);
-  } catch (e) {
-    return {};
-  }
-}
-
 export type PostgresConfig =
   | {
       host: string;
@@ -49,7 +36,7 @@ export class PostgresStore extends MastraStorage {
     );
   }
 
-  getEvalsByAgentName(agentName: string, type?: 'test' | 'live'): Promise<EvalRow[]> {
+  getEvalsByAgentName(_agentName: string, _type?: 'test' | 'live'): Promise<EvalRow[]> {
     throw new Error('Method not implemented.');
   }
 
@@ -424,11 +411,11 @@ export class PostgresStore extends MastraStorage {
           WITH ordered_messages AS (
             SELECT 
               *,
-              ROW_NUMBER() OVER (ORDER BY "createdAt") as row_num
+              ROW_NUMBER() OVER (ORDER BY "createdAt" DESC) as row_num
             FROM "${MastraStorage.TABLE_MESSAGES}"
             WHERE thread_id = $1
           )
-          SELECT DISTINCT ON (m.id)
+          SELECT
             m.id, 
             m.content, 
             m.role, 
@@ -448,7 +435,7 @@ export class PostgresStore extends MastraStorage {
               (m.row_num <= target.row_num + $4 AND m.row_num > target.row_num)
             )
           )
-          ORDER BY m.id, m."createdAt"
+          ORDER BY m."createdAt" DESC
           `,
           [
             threadId,
@@ -490,7 +477,7 @@ export class PostgresStore extends MastraStorage {
         if (typeof message.content === 'string') {
           try {
             message.content = JSON.parse(message.content);
-          } catch (e) {
+          } catch {
             // If parsing fails, leave as string
           }
         }

package/src/vector/filter.ts

@@ -1,4 +1,5 @@
-import { BaseFilterTranslator, type FieldCondition, type Filter, type OperatorSupport } from '@mastra/core/filter';
+import { BaseFilterTranslator } from '@mastra/core/filter';
+import type { FieldCondition, Filter, OperatorSupport } from '@mastra/core/filter';
 
 /**
  * Translates MongoDB-style filters to PG compatible filters.

package/src/vector/index.test.ts

@@ -187,7 +187,7 @@ describe('PgVector', () => {
         beforeAll(async () => {
           try {
             await vectorDB.deleteIndex(indexName);
-          } catch (e) {
+          } catch {
             // Ignore if doesn't exist
           }
           await vectorDB.createIndex(indexName, 3);
@@ -248,7 +248,7 @@ describe('PgVector', () => {
     beforeAll(async () => {
       try {
         await vectorDB.deleteIndex(indexName);
-      } catch (e) {
+      } catch {
         // Ignore if doesn't exist
       }
       await vectorDB.createIndex(indexName, 3);

package/src/vector/index.ts

@@ -1,10 +1,11 @@
 import type { Filter } from '@mastra/core/filter';
-import { type IndexStats, type QueryResult, MastraVector } from '@mastra/core/vector';
+import { MastraVector } from '@mastra/core/vector';
+import type { IndexStats, QueryResult } from '@mastra/core/vector';
 import pg from 'pg';
 
 import { PGFilterTranslator } from './filter';
 import { buildFilterQuery } from './sql-builder';
-import { type IndexConfig, type IndexType } from './types';
+import type { IndexConfig, IndexType } from './types';
 
 export interface PGIndexStats extends IndexStats {
   type: IndexType;
@@ -203,10 +204,21 @@ export class PgVector extends MastraVector {
     }
   }
 
+  /**
+   * @deprecated This function is deprecated. Use buildIndex instead
+   */
   async defineIndex(
     indexName: string,
     metric: 'cosine' | 'euclidean' | 'dotproduct' = 'cosine',
     indexConfig: IndexConfig,
+  ): Promise<void> {
+    return this.buildIndex(indexName, metric, indexConfig);
+  }
+
+  async buildIndex(
+    indexName: string,
+    metric: 'cosine' | 'euclidean' | 'dotproduct' = 'cosine',
+    indexConfig: IndexConfig,
   ): Promise<void> {
     const client = await this.pool.connect();
     try {

package/src/vector/performance.helpers.ts

@@ -1,6 +1,6 @@
-import { type IndexConfig, type IndexType } from './types';
+import type { IndexConfig, IndexType } from './types';
 
-import { PgVector } from '.';
+import type { PgVector } from '.';
 
 export interface TestResult {
   distribution: string;

package/src/vector/sql-builder.ts

@@ -1,11 +1,11 @@
-import {
-  type BasicOperator,
-  type NumericOperator,
-  type ArrayOperator,
-  type ElementOperator,
-  type LogicalOperator,
-  type RegexOperator,
-  type Filter,
+import type {
+  BasicOperator,
+  NumericOperator,
+  ArrayOperator,
+  ElementOperator,
+  LogicalOperator,
+  RegexOperator,
+  Filter,
 } from '@mastra/core/filter';
 
 export type OperatorType =

package/src/vector/vector.performance.test.ts

@@ -1,10 +1,9 @@
 import pg from 'pg';
 import { describe, it, beforeAll, afterAll, beforeEach, afterEach } from 'vitest';
 
+import type { TestConfig, TestResult } from './performance.helpers';
 import {
   baseTestConfigs,
-  TestConfig,
-  TestResult,
   calculateTimeout,
   generateRandomVectors,
   findNearestBruteForce,
@@ -19,7 +18,7 @@ import {
   getHNSWConfig,
   getIndexDescription,
 } from './performance.helpers';
-import { IndexConfig, IndexType } from './types';
+import type { IndexConfig, IndexType } from './types';
 
 import { PgVector } from '.';
 
@@ -178,7 +177,7 @@ describe('PostgreSQL Index Performance', () => {
               await batchedBulkUpsert(vectorDB, testIndexName, testVectors, batchSizes);
               if (indexType === 'hnsw' || rebuild) {
                 console.log('rebuilding index');
-                await vectorDB.defineIndex(testIndexName, 'cosine', indexConfig);
+                await vectorDB.buildIndex(testIndexName, 'cosine', indexConfig);
                 console.log('index rebuilt');
               }
               await smartWarmup(vectorDB, testIndexName, indexType, testConfig.dimension, testConfig.k);