@mastra/pg 1.23.0-alpha.7 → 1.23.1-alpha.0

package/dist/docs/SKILL.md

@@ -3,7 +3,7 @@ name: mastra-pg
 description: Documentation for @mastra/pg. Use when working with @mastra/pg APIs, configuration, or implementation.
 metadata:
   package: "@mastra/pg"
-  version: "1.23.0-alpha.7"
+  version: "1.23.1-alpha.0"
 ---
 
 ## When to use

package/dist/docs/assets/SOURCE_MAP.json

@@ -1,5 +1,5 @@
 {
-  "version": "1.23.0-alpha.7",
+  "version": "1.23.1-alpha.0",
   "package": "@mastra/pg",
   "exports": {},
   "modules": {}

package/dist/docs/references/reference-rag-vector-databases.md

@@ -650,7 +650,7 @@ Key metadata considerations:
 
 ## Deleting vectors
 
-When building RAG applications, you often need to clean up stale vectors when documents are deleted or updated. Mastra provides the `deleteVectors` method that supports deleting vectors by metadata filters, making it straightforward to remove all embeddings associated with a specific document.
+Use `deleteVectors` with a metadata filter to remove embeddings associated with a document. This is useful for cleaning up stale vectors after a document is deleted or updated.
 
 ### Delete by Metadata Filter
 

package/dist/index.cjs

@@ -11064,13 +11064,36 @@ var MemoryPG = class MemoryPG extends _mastra_core_storage.MemoryStorage {
 	* asked for a page after the last row, because there is then no row to carry the
 	* count on.
 	*/
-	async #fetchMessagePage({ selectStatement, tableName, whereClause, orderByStatement, queryParams, perPageInput, perPage, offset }) {
+	async #fetchMessagePage({ selectStatement, tableName, whereClause, orderByStatement, queryParams, perPageInput, perPage, offset, includeTotal = true }) {
+		if (includeTotal === false && perPageInput !== false) {
+			const peekLimit = perPage + 1;
+			const rows = await this.#db.readClient.manyOrNone(`${selectStatement} FROM ${tableName} ${whereClause} ${orderByStatement} LIMIT $${queryParams.length + 1} OFFSET $${queryParams.length + 2}`, [
+				...queryParams,
+				peekLimit,
+				offset
+			]) || [];
+			const hasMore = rows.length > perPage;
+			const messages = hasMore ? rows.slice(0, perPage) : rows;
+			return {
+				total: offset + messages.length,
+				messages,
+				hasMore
+			};
+		}
 		const limitClause = perPageInput === false ? "" : ` LIMIT $${queryParams.length + 1} OFFSET $${queryParams.length + 2}`;
 		const dataParams = perPageInput === false ? queryParams : [
 			...queryParams,
 			perPage,
 			offset
 		];
+		if (includeTotal === false) {
+			const rows = await this.#db.readClient.manyOrNone(`${selectStatement} FROM ${tableName} ${whereClause} ${orderByStatement}`, dataParams) || [];
+			return {
+				total: rows.length,
+				messages: rows,
+				hasMore: false
+			};
+		}
 		const rows = await this.#db.readClient.manyOrNone(`${selectStatement}, (SELECT COUNT(*) FROM ${tableName} ${whereClause}) AS "__total" FROM ${tableName} ${whereClause} ${orderByStatement}${limitClause}`, dataParams) || [];
 		if (rows.length > 0) return {
 			total: Number(rows[0].__total),
@@ -11087,7 +11110,7 @@ var MemoryPG = class MemoryPG extends _mastra_core_storage.MemoryStorage {
 		};
 	}
 	async listMessages(args) {
-		const { threadId, resourceId, include, filter, perPage: perPageInput, page = 0, orderBy } = args;
+		const { threadId, resourceId, include, filter, perPage: perPageInput, page = 0, orderBy, includeTotal = true } = args;
 		const threadIds = (Array.isArray(threadId) ? threadId : [threadId]).filter((id) => typeof id === "string");
 		if (threadIds.length === 0 || threadIds.some((id) => !id.trim())) throw new _mastra_core_error.MastraError({
 			id: (0, _mastra_core_storage.createStorageErrorId)("PG", "LIST_MESSAGES", "INVALID_THREAD_ID"),
@@ -11173,20 +11196,27 @@ var MemoryPG = class MemoryPG extends _mastra_core_storage.MemoryStorage {
 			}) : null;
 			let total;
 			let messages;
+			let peekedHasMore;
 			if (metadataFilter) {
 				const filteredRows = (await this.#db.readClient.manyOrNone(`${selectStatement} FROM ${tableName} ${whereClause} ${orderByStatement}`, queryParams) || []).filter((row) => (0, _mastra_core_storage.storageMessageMatchesMetadataFilter)(row.content, metadataFilter));
 				total = filteredRows.length;
 				messages = perPageInput === false ? filteredRows : filteredRows.slice(offset, offset + perPage);
-			} else ({total, messages} = await this.#fetchMessagePage({
-				selectStatement,
-				tableName,
-				whereClause,
-				orderByStatement,
-				queryParams,
-				perPageInput,
-				perPage,
-				offset
-			}));
+			} else {
+				const pageResult = await this.#fetchMessagePage({
+					selectStatement,
+					tableName,
+					whereClause,
+					orderByStatement,
+					queryParams,
+					perPageInput,
+					perPage,
+					offset,
+					includeTotal
+				});
+				total = pageResult.total;
+				messages = pageResult.messages;
+				peekedHasMore = pageResult.hasMore;
+			}
 			const primaryPageCount = messages.length;
 			if (total === 0 && messages.length === 0 && (!include || include.length === 0)) return {
 				messages: [],
@@ -11211,7 +11241,7 @@ var MemoryPG = class MemoryPG extends _mastra_core_storage.MemoryStorage {
 			const finalMessages = this._sortMessages(list.get.all.db(), field, direction);
 			const threadIdSet = new Set(threadIds);
 			const allThreadMessagesReturned = new Set(finalMessages.filter((m) => m.threadId && threadIdSet.has(m.threadId)).map((m) => m.id)).size >= total;
-			const hasMore = metadataFilter ? perPageInput !== false && offset + primaryPageCount < total : perPageInput !== false && !allThreadMessagesReturned && offset + perPage < total;
+			const hasMore = peekedHasMore !== void 0 ? peekedHasMore : metadataFilter ? perPageInput !== false && offset + primaryPageCount < total : perPageInput !== false && !allThreadMessagesReturned && offset + perPage < total;
 			return {
 				messages: finalMessages,
 				total,
@@ -11236,7 +11266,7 @@ var MemoryPG = class MemoryPG extends _mastra_core_storage.MemoryStorage {
 		}
 	}
 	async listMessagesByResourceId(args) {
-		const { resourceId, include, filter, perPage: perPageInput, page = 0, orderBy } = args;
+		const { resourceId, include, filter, perPage: perPageInput, page = 0, orderBy, includeTotal = true } = args;
 		if (!(resourceId !== void 0 && resourceId !== null && resourceId.trim() !== "")) throw new _mastra_core_error.MastraError({
 			id: (0, _mastra_core_storage.createStorageErrorId)("PG", "LIST_MESSAGES_BY_RESOURCE_ID", "INVALID_QUERY"),
 			domain: _mastra_core_error.ErrorDomain.STORAGE,
@@ -11319,20 +11349,27 @@ var MemoryPG = class MemoryPG extends _mastra_core_storage.MemoryStorage {
 			}) : null;
 			let total;
 			let messages;
+			let peekedHasMore;
 			if (metadataFilter) {
 				const filteredRows = (await this.#db.readClient.manyOrNone(`${selectStatement} FROM ${tableName} ${whereClause} ${orderByStatement}`, queryParams) || []).filter((row) => (0, _mastra_core_storage.storageMessageMatchesMetadataFilter)(row.content, metadataFilter));
 				total = filteredRows.length;
 				messages = perPageInput === false ? filteredRows : filteredRows.slice(offset, offset + perPage);
-			} else ({total, messages} = await this.#fetchMessagePage({
-				selectStatement,
-				tableName,
-				whereClause,
-				orderByStatement,
-				queryParams,
-				perPageInput,
-				perPage,
-				offset
-			}));
+			} else {
+				const pageResult = await this.#fetchMessagePage({
+					selectStatement,
+					tableName,
+					whereClause,
+					orderByStatement,
+					queryParams,
+					perPageInput,
+					perPage,
+					offset,
+					includeTotal
+				});
+				total = pageResult.total;
+				messages = pageResult.messages;
+				peekedHasMore = pageResult.hasMore;
+			}
 			if (total === 0 && messages.length === 0 && (!include || include.length === 0)) return {
 				messages: [],
 				total: 0,
@@ -11354,7 +11391,7 @@ var MemoryPG = class MemoryPG extends _mastra_core_storage.MemoryStorage {
 			const messagesWithParsedContent = messages.map((row) => this.parseRow(row));
 			const list = new _mastra_core_agent.MessageList().add(messagesWithParsedContent, "memory");
 			const finalMessages = this._sortMessages(list.get.all.db(), field, direction);
-			const hasMore = perPageInput !== false && offset + perPage < total;
+			const hasMore = peekedHasMore !== void 0 ? peekedHasMore : perPageInput !== false && offset + perPage < total;
 			return {
 				messages: finalMessages,
 				total,