@mastra/pg 1.26.0-alpha.3 → 1.26.0-alpha.4

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.26.0-alpha.3"
+  version: "1.26.0-alpha.4"
 ---
 
 ## When to use

package/dist/docs/assets/SOURCE_MAP.json

@@ -1,5 +1,5 @@
 {
-  "version": "1.26.0-alpha.3",
+  "version": "1.26.0-alpha.4",
   "package": "@mastra/pg",
   "exports": {},
   "modules": {}

package/dist/docs/references/reference-memory-memory-class.md

@@ -51,7 +51,7 @@ export const agent = new Agent({
 
 **options.observationalMemory** (`boolean | ObservationalMemoryOptions`): Enable Observational Memory for long-context agentic memory. Set to true for defaults, or pass a config object to customize token budgets, models, and scope. See Observational Memory reference for configuration details.
 
-**options.generateTitle** (`boolean | { model: DynamicArgument<MastraLanguageModel>; instructions?: DynamicArgument<string> }`): Controls automatic thread title generation from the conversation transcript. Can be a boolean or an object with custom model and instructions.
+**options.generateTitle** (`boolean | { model?: DynamicArgument<MastraModelConfig>; instructions?: DynamicArgument<string>; minMessages?: number; emitEvent?: boolean }`): Controls automatic thread title generation from the conversation transcript. Accepts a boolean or an object with a custom model (any MastraModelConfig: a model instance, a "provider/model" ID, or an OpenAI-compatible config; defaults to the agent's own model), custom instructions, a minimum message count, and emitEvent. With emitEvent: true the run's stream waits for the title and emits it as a transient data-thread-title chunk before finish (durable and evented agents persist the title but don't emit the chunk yet).
 
 ## Returns
 

package/dist/index.cjs

@@ -2360,6 +2360,8 @@ var RoutingDbClient = class {
 		return this.active.tx(callback);
 	}
 };
+/** Bytes reserved for the `_xxxxxxxx` collision suffix when hashWhenTruncated applies. */
+const TRUNCATION_HASH_SUFFIX_LENGTH = 9;
 function truncateIdentifier(value, maxLength = 63) {
 	if (maxLength <= 0) return "";
 	if (Buffer.byteLength(value, "utf-8") <= maxLength) return value;
@@ -2380,9 +2382,25 @@ function truncateIdentifier(value, maxLength = 63) {
 * in system catalogs (pg_constraint.conname, pg_indexes.indexname, etc.).
 * Without this normalisation, runtime lookups that compare a mixed-case name
 * against the catalog would silently fail.
+*
+* With `hashWhenTruncated`, a name that exceeds the limit is truncated further
+* to make room for `_` + 8 hex chars of the full name's sha256. Plain
+* truncation cuts the tail, so two names sharing a long `<schema>_<prefix>`
+* collapse to the same identifier and `CREATE INDEX IF NOT EXISTS` (which
+* matches by name only) silently skips the second one. The suffix is
+* deterministic, so creation, warm-init snapshot checks, and DDL export all
+* agree on the same name. Opt-in because renaming already-released constraint
+* names would orphan the existing objects in deployed catalogs.
 */
-function buildConstraintName({ baseName, schemaName, maxLength = 63 }) {
-	return truncateIdentifier(`${schemaName ? `${schemaName}_` : ""}${baseName}`.toLowerCase(), maxLength);
+function buildConstraintName({ baseName, schemaName, maxLength = 63, hashWhenTruncated = false }) {
+	const fullName = `${schemaName ? `${schemaName}_` : ""}${baseName}`.toLowerCase();
+	if (hashWhenTruncated && Buffer.byteLength(fullName, "utf-8") > maxLength) {
+		const suffixLength = Math.min(TRUNCATION_HASH_SUFFIX_LENGTH, maxLength);
+		if (suffixLength < 2) return truncateIdentifier(fullName, maxLength);
+		const hash = (0, crypto$1.createHash)("sha256").update(fullName).digest("hex").slice(0, suffixLength - 1);
+		return `${truncateIdentifier(fullName, maxLength - suffixLength)}_${hash}`;
+	}
+	return truncateIdentifier(fullName, maxLength);
 }
 //#endregion
 //#region src/storage/db/pg-errors.ts
@@ -22317,6 +22335,7 @@ const WORKFLOW_SNAPSHOT_STATUS_INDEX = "mastra_workflow_snapshot_name_status_cre
 * Schema-prefixed name of the status index, lowercased and truncated the same way Postgres
 * stores it, so the init snapshot's index set answers "does it exist?" without a probe or a
 * no-op `CREATE INDEX` (schema-prefixed names routinely exceed the 63-byte limit).
+* Exported for tests.
 */
 function workflowSnapshotStatusIndexName(schemaName) {
 	return buildConstraintName({
@@ -22334,6 +22353,48 @@ function workflowSnapshotStatusIndexSQL(indexName, schemaName) {
 		schemaName: getSchemaName(schemaName)
 	})} (workflow_name, (snapshot ->> 'status'), "createdAt" DESC)`;
 }
+/** Base name (before any schema prefix) of the expression index backing the threadId filter. */
+const WORKFLOW_SNAPSHOT_THREAD_ID_INDEX = "mastra_workflow_snapshot_threadid_idx";
+/**
+* Schema-prefixed name of the threadId index (see workflowSnapshotStatusIndexName).
+*
+* Unlike the status index, truncation appends a collision hash: both index names share the
+* long `<schema>_mastra_workflow_snapshot_` prefix, so with a schema name of 37+ bytes plain
+* truncation collapses them to the same 63-byte identifier and `CREATE INDEX IF NOT EXISTS`
+* silently skips this index. The status index keeps plain truncation because its truncated
+* name already exists in deployed catalogs; this index is new and free to adopt the rule.
+* Exported for tests.
+*/
+function workflowSnapshotThreadIdIndexName(schemaName) {
+	return buildConstraintName({
+		baseName: WORKFLOW_SNAPSHOT_THREAD_ID_INDEX,
+		schemaName: schemaName && schemaName !== "public" ? schemaName : void 0,
+		hashWhenTruncated: true
+	});
+}
+/**
+* Expression extracting the thread id embedded in a snapshot (jsonb columns only). Mirrors
+* the canonical extraction in `@mastra/core` (`getSnapshotMemoryInfo`), which reads one of
+* two layouts:
+* 1. agentic-loop: `context.<suspended step>.suspendPayload.__streamState.messageList.memoryInfo.threadId`
+* 2. durable loop: `context.input.messageListState.memoryInfo.threadId`
+*
+* `jsonb_path_query_first(jsonb, jsonpath)` is IMMUTABLE, so the expression is valid in an
+* expression index. The WHERE clause in listWorkflowRuns() must use this exact expression
+* text so the planner can match it against the index. If the snapshot layout changes in
+* core, this expression must be updated in lockstep or it will wrongly exclude rows.
+*/
+const WORKFLOW_SNAPSHOT_THREAD_ID_EXPR = `COALESCE(jsonb_path_query_first(snapshot, '$.context.* ? (@.status == "suspended").suspendPayload.__streamState.messageList.memoryInfo.threadId') #>> '{}', snapshot #>> '{context,input,messageListState,memoryInfo,threadId}')`;
+/**
+* Expression index on the snapshot-embedded thread id so listWorkflowRuns() threadId filters
+* (Agent.listSuspendedRuns) can use an index instead of detoasting every snapshot.
+*/
+function workflowSnapshotThreadIdIndexSQL(indexName, schemaName) {
+	return `CREATE INDEX IF NOT EXISTS "${indexName}" ON ${getTableName({
+		indexName: _mastra_core_storage.TABLE_WORKFLOW_SNAPSHOT,
+		schemaName: getSchemaName(schemaName)
+	})} ((${WORKFLOW_SNAPSHOT_THREAD_ID_EXPR}))`;
+}
 var WorkflowsPG = class WorkflowsPG extends _mastra_core_storage.WorkflowsStorage {
 	#db;
 	#schema;
@@ -22406,6 +22467,7 @@ var WorkflowsPG = class WorkflowsPG extends _mastra_core_storage.WorkflowsStorag
 		}));
 		for (const idx of WorkflowsPG.getDefaultIndexDefs(schemaPrefix)) statements.push(generateIndexSQL(idx, schemaName));
 		statements.push(`${workflowSnapshotStatusIndexSQL(workflowSnapshotStatusIndexName(parsedSchema), schemaName)};`);
+		statements.push(`${workflowSnapshotThreadIdIndexSQL(workflowSnapshotThreadIdIndexName(parsedSchema), schemaName)};`);
 		return statements;
 	}
 	/**
@@ -22432,6 +22494,12 @@ var WorkflowsPG = class WorkflowsPG extends _mastra_core_storage.WorkflowsStorag
 		} catch (error) {
 			this.logger?.warn?.(`Failed to create index ${indexName}:`, error);
 		}
+		const threadIdIndexName = workflowSnapshotThreadIdIndexName(this.#schema);
+		try {
+			await this.#db.createIndexFromStatement(threadIdIndexName, workflowSnapshotThreadIdIndexSQL(threadIdIndexName, this.#schema));
+		} catch (error) {
+			this.logger?.warn?.(`Failed to create index ${threadIdIndexName}:`, error);
+		}
 	}
 	async init() {
 		await this.#db.createTable({
@@ -22712,7 +22780,7 @@ var WorkflowsPG = class WorkflowsPG extends _mastra_core_storage.WorkflowsStorag
 			}, error);
 		}
 	}
-	async listWorkflowRuns({ workflowName, fromDate, toDate, perPage, page, resourceId, status } = {}) {
+	async listWorkflowRuns({ workflowName, fromDate, toDate, perPage, page, resourceId, threadId, status } = {}) {
 		try {
 			const conditions = [];
 			const values = [];
@@ -22733,6 +22801,11 @@ var WorkflowsPG = class WorkflowsPG extends _mastra_core_storage.WorkflowsStorag
 				values.push(resourceId);
 				paramIndex++;
 			} else this.logger?.warn?.(`[${_mastra_core_storage.TABLE_WORKFLOW_SNAPSHOT}] resourceId column not found. Skipping resourceId filter.`);
+			if (threadId) if (await this.#db.getColumnType(_mastra_core_storage.TABLE_WORKFLOW_SNAPSHOT, "snapshot") === "jsonb") {
+				conditions.push(`${WORKFLOW_SNAPSHOT_THREAD_ID_EXPR} = $${paramIndex}`);
+				values.push(threadId);
+				paramIndex++;
+			} else this.logger?.warn?.(`[${_mastra_core_storage.TABLE_WORKFLOW_SNAPSHOT}] snapshot column is not jsonb. Skipping threadId filter.`);
 			if (fromDate) {
 				conditions.push(`"createdAt" >= $${paramIndex}`);
 				values.push(fromDate);