@mastra/oracledb 0.4.2 → 0.5.1-alpha.0

package/dist/index.js

@@ -4295,6 +4295,10 @@ async function getObservationalMemoryHistory(ctx, threadId, resourceId, limit =
 			lookupKey,
 			limit
 		};
+		if (options?.recordId !== void 0) {
+			conditions.push(`id = :recordId`);
+			binds.recordId = options.recordId;
+		}
 		if (options?.from) {
 			conditions.push(`${OM_CREATED_AT} >= :fromDate`);
 			binds.fromDate = options.from;
@@ -4303,11 +4307,28 @@ async function getObservationalMemoryHistory(ctx, threadId, resourceId, limit =
 			conditions.push(`${OM_CREATED_AT} <= :toDate`);
 			binds.toDate = options.to;
 		}
+		if (options?.groupId !== void 0) {
+			conditions.push(`(DBMS_LOB.INSTR(${OM_ACTIVE_OBSERVATIONS}, :groupPrefix) > 0 OR EXISTS (
+        SELECT 1 FROM JSON_TABLE(${OM_BUFFERED_OBSERVATION_CHUNKS}, '$[*]'
+          COLUMNS (observations CLOB PATH '$.observations')) chunk
+        WHERE DBMS_LOB.INSTR(chunk.observations, :groupPrefix) > 0
+      ))`);
+			binds.groupPrefix = `<observation-group id="${options.groupId}"`;
+		}
+		if (options?.beforeGeneration !== void 0) {
+			conditions.push(`${OM_GENERATION_COUNT} < :beforeGeneration`);
+			binds.beforeGeneration = options.beforeGeneration;
+		}
+		if (options?.afterGeneration !== void 0) {
+			conditions.push(`${OM_GENERATION_COUNT} > :afterGeneration`);
+			binds.afterGeneration = options.afterGeneration;
+		}
+		const direction = options?.sortDirection === "ASC" ? "ASC" : "DESC";
 		if (options?.offset !== void 0) binds.offset = options.offset;
 		return await ctx.db.withConnection(async (connection) => {
 			return rows(await connection.execute(`${omSelect()} FROM ${table(ctx, TABLE_OBSERVATIONAL_MEMORY)}
          WHERE ${conditions.join(" AND ")}
-         ORDER BY ${OM_GENERATION_COUNT} DESC
+         ORDER BY ${OM_GENERATION_COUNT} ${direction}, ${OM_CREATED_AT} ASC, "id" ASC
          OFFSET :offset ROWS FETCH NEXT :limit ROWS ONLY`, asBindParameters({
 				offset: options?.offset ?? 0,
 				...binds
@@ -5370,6 +5391,7 @@ const DEFAULT_VECTOR_REGISTRY_TABLE$1 = "MASTRA_VECTOR_INDEXES";
 var MemoryOracle = class MemoryOracle extends MemoryStorage {
 	supportsPartialThreadUpdate = true;
 	supportsObservationalMemory = true;
+	supportsObservationalMemoryHistorySearch = true;
 	static MANAGED_TABLES = [
 		TABLE_THREADS,
 		TABLE_MESSAGES,
@@ -7953,7 +7975,7 @@ var WorkflowsOracle = class WorkflowsOracle extends WorkflowsStorage {
 	async dangerouslyClearAll() {
 		await this.db.clearTable(TABLE_WORKFLOW_SNAPSHOT);
 	}
-	async updateWorkflowResults({ workflowName, runId, stepId, result, requestContext }) {
+	async updateWorkflowResults({ workflowName, runId, stepId, result, requestContext, state }) {
 		try {
 			return await this.db.tx(async (client) => {
 				await this.ensureWorkflowRunRow(client, workflowName, runId);
@@ -7963,6 +7985,7 @@ var WorkflowsOracle = class WorkflowsOracle extends WorkflowsStorage {
 				});
 				const snapshot = parseSnapshot(existing.snapshot);
 				snapshot.context[stepId] = result;
+				if (state !== void 0) snapshot.context.__state = state;
 				snapshot.requestContext = {
 					...snapshot.requestContext ?? {},
 					...requestContext