@mastra/pg 1.28.0 → 1.28.1-alpha.1

package/dist/index.js

@@ -1,4 +1,3 @@
-import { createHash, randomUUID } from "crypto";
 import { ErrorCategory, ErrorDomain, MastraError } from "@mastra/core/error";
 import * as coreStorage from "@mastra/core/storage";
 import { AgentsStorage, BRANCH_SPAN_TYPES, BackgroundTasksStorage, BlobStore, ChannelsStorage, DATASETS_SCHEMA, DATASET_ITEMS_SCHEMA, DATASET_VERSIONS_SCHEMA, DatasetsStorage, EXPERIMENTS_SCHEMA, EXPERIMENT_RESULTS_SCHEMA, EntityType, ExperimentsStorage, FactoryStorage, FavoritesStorage, KNOWLEDGE_ACTIVITY_SCHEMA, KNOWLEDGE_CURSORS_SCHEMA, KNOWLEDGE_MENTIONS_SCHEMA, KNOWLEDGE_NODES_SCHEMA, KNOWLEDGE_RECORDS_SCHEMA, KNOWLEDGE_SEMANTIC_OUTBOX_SCHEMA, KnowledgeConflictError, KnowledgeNotFoundError, KnowledgeStorage, MCPClientsStorage, MCPServersStorage, METRIC_DISTINCT_COLUMNS, MastraCompositeStore, MemoryStorage, NotificationsStorage, OBSERVATIONAL_MEMORY_TABLE_SCHEMA, ObservabilityStorage, PromptBlocksStorage, SchedulesStorage, ScorerDefinitionsStorage, ScoresStorage, SkillsStorage, TABLE_AGENTS, TABLE_AGENT_VERSIONS, TABLE_BACKGROUND_TASKS, TABLE_CHANNEL_CONFIG, TABLE_CHANNEL_INSTALLATIONS, TABLE_CONFIGS, TABLE_DATASETS, TABLE_DATASET_ITEMS, TABLE_DATASET_VERSIONS, TABLE_EXPERIMENTS, TABLE_EXPERIMENT_RESULTS, TABLE_FAVORITES, TABLE_KNOWLEDGE_ACTIVITY, TABLE_KNOWLEDGE_CURSORS, TABLE_KNOWLEDGE_MENTIONS, TABLE_KNOWLEDGE_NODES, TABLE_KNOWLEDGE_RECORDS, TABLE_KNOWLEDGE_SEMANTIC_OUTBOX, TABLE_MCP_CLIENTS, TABLE_MCP_CLIENT_VERSIONS, TABLE_MCP_SERVERS, TABLE_MCP_SERVER_VERSIONS, TABLE_MESSAGES, TABLE_NOTIFICATIONS, TABLE_PROMPT_BLOCKS, TABLE_PROMPT_BLOCK_VERSIONS, TABLE_RESOURCES, TABLE_SCHEDULES, TABLE_SCHEDULE_TRIGGERS, TABLE_SCHEMAS, TABLE_SCORERS, TABLE_SCORER_DEFINITIONS, TABLE_SCORER_DEFINITION_VERSIONS, TABLE_SKILLS, TABLE_SKILL_BLOBS, TABLE_SKILL_VERSIONS, TABLE_SPANS, TABLE_THREADS, TABLE_THREAD_STATE, TABLE_TOOL_PROVIDER_CONNECTIONS, TABLE_WORKFLOW_DEFINITIONS, TABLE_WORKFLOW_SNAPSHOT, TABLE_WORKSPACES, TABLE_WORKSPACE_VERSIONS, ThreadStateStorage, ToolProviderConnectionsStorage, TraceStatus, UniqueViolationError, WorkflowDefinitionsStorage, WorkflowsStorage, WorkspacesStorage, assertKnowledgeCeilingRaised, assertKnowledgeScopeWithinCeiling, calculatePagination, canonicalizeKnowledgeScope, createKnowledgeUlid, createStorageErrorId, createVectorErrorId, ensureDate, executeRetentionPrune, getDefaultValue, getSqlType, hasErrorCode, isKnowledgeScopeVisible, knowledgeScopeKey, knowledgeSemanticDocumentId, knowledgeSemanticIdempotencyKey, listBranchesArgsSchema, listFeedbackArgsSchema, listLogsArgsSchema, listMetricsArgsSchema, listScoresArgsSchema, listTracesArgsSchema, matchesExpectedWorkflowStatus, mergeWorkflowStepResult, normalizePerPage, normalizeScheduleTarget, parseDuration, parseKnowledgeNodeCursor, parseKnowledgeWikilinks, resolveListOrderBy, resolveRetentionTargets, retentionCutoffMs, runRetentionBatches, safelyParseJSON, storageMessageMatchesMetadataFilter, toTraceSpans, transformScoreRow, validateStorageMetadataFilter } from "@mastra/core/storage";
@@ -11,6 +10,7 @@ import xxhash from "xxhash-wasm";
 import { parse } from "pg-connection-string";
 import { BaseFilterTranslator } from "@mastra/core/vector/filter";
 import { MastraBase } from "@mastra/core/base";
+import { createHash } from "crypto";
 import { MessageList } from "@mastra/core/agent";
 import { coreFeatures } from "@mastra/core/features";
 import { saveScorePayloadSchema } from "@mastra/core/evals";
@@ -825,7 +825,7 @@ var PgVector = class extends MastraVector {
          AND NOT attisdropped`, [tableName])).rowCount === 0) return;
 		await client.query(`ALTER TABLE ${tableName} ADD COLUMN IF NOT EXISTS namespace VARCHAR(255) NOT NULL DEFAULT '${DEFAULT_NAMESPACE}'`);
 		if (!(await this.getNamespaceSchemaState(tableName, client)).composite_index) {
-			const namespaceIndexName = this.getNamespaceIndexName(parsedIndexName);
+			const namespaceIndexName = await this.getNamespaceIndexName(parsedIndexName);
 			await client.query(`CREATE UNIQUE INDEX IF NOT EXISTS "${namespaceIndexName}" ON ${tableName} (namespace, vector_id)`);
 		}
 		const legacyConstraints = await client.query(`SELECT c.conname
@@ -838,10 +838,10 @@ var PgVector = class extends MastraVector {
 			await client.query(`ALTER TABLE ${tableName} DROP CONSTRAINT "${parsedConstraintName}"`);
 		}
 	}
-	getNamespaceIndexName(parsedIndexName) {
+	async getNamespaceIndexName(parsedIndexName) {
 		const fullName = `${parsedIndexName}_namespace_vector_id_idx`;
 		if (fullName.length <= 63) return fullName;
-		const suffix = `_ns_${createHash("sha256").update(parsedIndexName).digest("hex").slice(0, 32)}_idx`;
+		const suffix = `_ns_${Buffer.from(await globalThis.crypto.subtle.digest("SHA-256", new TextEncoder().encode(parsedIndexName))).toString("hex").slice(0, 32)}_idx`;
 		return `${parsedIndexName.slice(0, 63 - suffix.length)}${suffix}`;
 	}
 	async getNamespaceSchemaState(tableName, client) {
@@ -976,6 +976,8 @@ var PgVector = class extends MastraVector {
 			let client;
 			try {
 				await this.ensureNamespaceReady(indexName);
+				const indexInfo = includeVector ? await this.getIndexMetadata({ indexName }) : void 0;
+				const ops = indexInfo && this.getVectorOps(indexInfo.vectorType, indexInfo.metric ?? "cosine");
 				client = await this.pool.connect();
 				const { sql: filterQuery, values: filterValues } = buildDeleteFilterQuery(this.transformFilter(filter));
 				const { tableName } = this.getTableName(indexName);
@@ -999,7 +1001,7 @@ var PgVector = class extends MastraVector {
 					id,
 					score: 0,
 					metadata,
-					...includeVector && embedding && { vector: JSON.parse(embedding) }
+					...ops && embedding && { vector: ops.parseEmbedding(embedding) }
 				}));
 			} catch (error) {
 				if (error instanceof MastraError) {
@@ -12653,7 +12655,7 @@ var MemoryPG = class MemoryPG extends MemoryStorage {
 			});
 			const nowStr = (/* @__PURE__ */ new Date()).toISOString();
 			const newChunk = {
-				id: `ombuf-${randomUUID()}`,
+				id: `ombuf-${globalThis.crypto.randomUUID()}`,
 				cycleId: input.chunk.cycleId,
 				observations: input.chunk.observations,
 				tokenCount: Math.round(input.chunk.tokenCount),
@@ -13223,7 +13225,7 @@ var NotificationsPG = class NotificationsPG extends NotificationsStorage {
 		}
 		const now = input.createdAt ?? /* @__PURE__ */ new Date();
 		const record = {
-			id: input.id ?? randomUUID(),
+			id: input.id ?? globalThis.crypto.randomUUID(),
 			threadId: input.threadId,
 			source: input.source,
 			kind: input.kind,
@@ -21476,7 +21478,6 @@ var SkillsPG = class SkillsPG extends SkillsStorage {
 				details: { skillId: id }
 			});
 			const { authorId, visibility, activeVersionId, status, ...rawConfigFields } = updates;
-			let versionCreated = false;
 			const configFields = {};
 			for (const [key, value] of Object.entries(rawConfigFields)) if (value !== void 0) configFields[key] = value;
 			if (SNAPSHOT_FIELDS$1.some((field) => field in configFields)) {
@@ -21495,7 +21496,6 @@ var SkillsPG = class SkillsPG extends SkillsStorage {
 				};
 				const changedFields = SNAPSHOT_FIELDS$1.filter((field) => field in configFields && !skillSnapshotFieldValuesEqual(configFields[field], latestConfig[field]));
 				if (changedFields.length > 0) {
-					versionCreated = true;
 					const newVersionId = crypto.randomUUID();
 					await this.createVersion({
 						id: newVersionId,
@@ -21536,7 +21536,7 @@ var SkillsPG = class SkillsPG extends SkillsStorage {
 			setClauses.push(`"updatedAtZ" = $${paramIndex++}`);
 			values.push(now);
 			values.push(id);
-			if (setClauses.length > 2 || versionCreated) await this.#db.client.none(`UPDATE ${tableName} SET ${setClauses.join(", ")} WHERE id = $${paramIndex}`, values);
+			await this.#db.client.none(`UPDATE ${tableName} SET ${setClauses.join(", ")} WHERE id = $${paramIndex}`, values);
 			const updatedSkill = await this.#getById(this.#db.client, id);
 			if (!updatedSkill) throw new MastraError({
 				id: createStorageErrorId("PG", "UPDATE_SKILL", "NOT_FOUND_AFTER_UPDATE"),
@@ -22515,7 +22515,10 @@ var WorkflowDefinitionsPG = class WorkflowDefinitionsPG extends WorkflowDefiniti
 		return this.applyUpdate(input, now);
 	}
 	async applyUpdate(input, now) {
-		const data = { updatedAt: now };
+		const data = {
+			updatedAt: now,
+			updatedAtZ: now
+		};
 		if ("description" in input && input.description !== void 0) data.description = input.description;
 		if ("metadata" in input && input.metadata !== void 0) data.metadata = input.metadata;
 		if ("inputSchema" in input && input.inputSchema !== void 0) data.inputSchema = input.inputSchema;
@@ -23313,7 +23316,6 @@ var WorkspacesPG = class WorkspacesPG extends WorkspacesStorage {
 				details: { workspaceId: id }
 			});
 			const { authorId, activeVersionId, metadata, status, ...rawConfigFields } = updates;
-			let versionCreated = false;
 			const configFields = {};
 			for (const [key, value] of Object.entries(rawConfigFields)) if (value !== void 0) configFields[key] = value;
 			if (SNAPSHOT_FIELDS.some((field) => field in configFields)) {
@@ -23336,7 +23338,6 @@ var WorkspacesPG = class WorkspacesPG extends WorkspacesStorage {
 				};
 				const changedFields = SNAPSHOT_FIELDS.filter((field) => field in configFields && !isDeepStrictEqual(normalized(configFields[field]), normalized(latestConfig[field])));
 				if (changedFields.length > 0) {
-					versionCreated = true;
 					const newVersionId = crypto.randomUUID();
 					await this.createVersion({
 						id: newVersionId,
@@ -23382,7 +23383,7 @@ var WorkspacesPG = class WorkspacesPG extends WorkspacesStorage {
 			setClauses.push(`"updatedAtZ" = $${paramIndex++}`);
 			values.push(now);
 			values.push(id);
-			if (setClauses.length > 2 || versionCreated) await this.#db.client.none(`UPDATE ${tableName} SET ${setClauses.join(", ")} WHERE id = $${paramIndex}`, values);
+			await this.#db.client.none(`UPDATE ${tableName} SET ${setClauses.join(", ")} WHERE id = $${paramIndex}`, values);
 			const updatedWorkspace = await this.#getById(this.#db.client, id);
 			if (!updatedWorkspace) throw new MastraError({
 				id: createStorageErrorId("PG", "UPDATE_WORKSPACE", "NOT_FOUND_AFTER_UPDATE"),
@@ -23944,7 +23945,7 @@ var PgFactoryStorageOps = class {
 		const schema = this.#schema(collection);
 		const pk = primaryKeyOf(schema);
 		const values = { ...row };
-		if (schema.columns[pk].type === "uuid-pk" && values[pk] === void 0) values[pk] = randomUUID();
+		if (schema.columns[pk].type === "uuid-pk" && values[pk] === void 0) values[pk] = globalThis.crypto.randomUUID();
 		const columns = Object.keys(values).filter((column) => values[column] !== void 0);
 		for (const column of columns) this.#column(schema, column);
 		const sql = `INSERT INTO "${schema.name}" (${columns.map((c) => `"${c}"`).join(", ")}) VALUES (${columns.map((_, i) => `$${i + 1}`).join(", ")}) RETURNING *`;