@mastra/oracledb 0.3.0 → 0.4.0-alpha.0

package/dist/index.js

@@ -2,7 +2,7 @@ import { MastraVector, validateTopK, validateUpsertInput } from "@mastra/core/ve
 import oracledb from "oracledb";
 import { ErrorCategory, ErrorDomain, MastraError } from "@mastra/core/error";
 import { createHash, randomUUID } from "crypto";
-import { AgentsStorage, MCPClientsStorage, MastraCompositeStore, MemoryStorage, ObservabilityStorage, ScorerDefinitionsStorage, ScoresStorage, TABLE_AGENTS, TABLE_AGENT_VERSIONS, TABLE_MCP_CLIENTS, TABLE_MCP_CLIENT_VERSIONS, TABLE_MESSAGES, TABLE_OBSERVATIONAL_MEMORY, TABLE_RESOURCES, TABLE_SCHEMAS, TABLE_SCORERS, TABLE_SCORER_DEFINITIONS, TABLE_SCORER_DEFINITION_VERSIONS, TABLE_SPANS, TABLE_THREADS, TABLE_WORKFLOW_SNAPSHOT, TraceStatus, WorkflowsStorage, calculatePagination, createStorageErrorId, createVectorErrorId, listLogsArgsSchema, listScoresArgsSchema, listTracesArgsSchema, matchesExpectedWorkflowStatus, normalizePerPage, storageMessageMatchesMetadataFilter, toTraceSpans, transformScoreRow, validateStorageMetadataFilter } from "@mastra/core/storage";
+import { AgentsStorage, MCPClientsStorage, MastraCompositeStore, MemoryStorage, ObservabilityStorage, ScorerDefinitionsStorage, ScoresStorage, TABLE_AGENTS, TABLE_AGENT_VERSIONS, TABLE_MCP_CLIENTS, TABLE_MCP_CLIENT_VERSIONS, TABLE_MESSAGES, TABLE_OBSERVATIONAL_MEMORY, TABLE_RESOURCES, TABLE_SCHEMAS, TABLE_SCORERS, TABLE_SCORER_DEFINITIONS, TABLE_SCORER_DEFINITION_VERSIONS, TABLE_SPANS, TABLE_THREADS, TABLE_WORKFLOW_SNAPSHOT, TraceStatus, WorkflowsStorage, calculatePagination, createStorageErrorId, createVectorErrorId, executeRetentionPrune, listLogsArgsSchema, listScoresArgsSchema, listTracesArgsSchema, matchesExpectedWorkflowStatus, normalizePerPage, resolveRetentionTargets, retentionCutoffMs, storageMessageMatchesMetadataFilter, toTraceSpans, transformScoreRow, validateStorageMetadataFilter } from "@mastra/core/storage";
 import { MessageList } from "@mastra/core/agent";
 import { saveScorePayloadSchema } from "@mastra/core/evals";
 //#region src/shared/connection.ts
@@ -1668,6 +1668,25 @@ var OracleDB = class {
 	async withConnection(callback) {
 		return this.config.poolManager.withConnection(callback);
 	}
+	async pruneBatch({ tableName, column, cutoff, limit }) {
+		if (!/^[A-Za-z_][A-Za-z0-9_]*$/.test(column)) throw new Error(`Invalid retention identifier: ${tableName}.${column}`);
+		return this.config.poolManager.withConnection(async (connection) => {
+			try {
+				const table = this.table(tableName);
+				const result = await connection.execute(`DELETE FROM ${table} WHERE ROWID IN (
+            SELECT ROWID FROM ${table} WHERE "${column}" < :cutoff ORDER BY "${column}" FETCH FIRST :limit ROWS ONLY
+          )`, asBindParameters({
+					cutoff,
+					limit
+				}));
+				await connection.commit();
+				return result.rowsAffected ?? 0;
+			} catch (error) {
+				await rollbackQuietly(connection);
+				throw error;
+			}
+		});
+	}
 	async none(sql, binds = {}) {
 		await this.config.poolManager.withConnection(async (connection) => {
 			await connection.execute(sql, asBindParameters(binds));
@@ -6734,6 +6753,18 @@ function traceOrderClause(field, direction) {
 //#endregion
 //#region src/storage/domains/observability/index.ts
 var ObservabilityOracle = class ObservabilityOracle extends ObservabilityStorage {
+	static retentionTables = {
+		spans: {
+			table: TABLE_SPANS,
+			column: "startedAt",
+			indexed: true
+		},
+		logs: {
+			table: LOG_EVENTS_TABLE,
+			column: "timestamp",
+			indexed: true
+		}
+	};
 	static MANAGED_TABLES = [TABLE_SPANS, LOG_EVENTS_TABLE];
 	db;
 	schemaName;
@@ -6746,6 +6777,24 @@ var ObservabilityOracle = class ObservabilityOracle extends ObservabilityStorage
 		this.skipDefaultIndexes = config.skipDefaultIndexes;
 		this.indexes = filterIndexesForTables(config.indexes, ObservabilityOracle.MANAGED_TABLES);
 	}
+	async prune(policies, options) {
+		return executeRetentionPrune({
+			domain: "observability",
+			targets: resolveRetentionTargets({
+				policies,
+				descriptor: ObservabilityOracle.retentionTables,
+				order: ["spans", "logs"]
+			}),
+			options,
+			cutoffFor: (target, now) => new Date(retentionCutoffMs(target.policy, now)),
+			deleteBatch: (target, cutoff, limit) => this.db.pruneBatch({
+				tableName: target.table,
+				column: target.column,
+				cutoff,
+				limit
+			})
+		});
+	}
 	async init() {
 		await this.db.createTable({
 			tableName: TABLE_SPANS,
@@ -8420,7 +8469,8 @@ var OracleStore = class extends MastraCompositeStore {
 			super({
 				id: config.id,
 				name: "OracleStore",
-				disableInit: config.disableInit
+				disableInit: config.disableInit,
+				retention: config.retention
 			});
 			this.schemaName = config.schemaName ? normalizeIdentifier(config.schemaName, "schema name") : void 0;
 			this.skipDefaultIndexes = config.skipDefaultIndexes;