@mastra/oracledb 0.2.4 → 0.3.0

package/dist/index.js

@@ -5231,6 +5231,36 @@ async function deleteThread(ctx, { threadId }) {
 		throw storageError$1("DELETE_THREAD", "FAILED", { threadId }, error);
 	}
 }
+async function updateThreadResourceId(ctx, { threadId, resourceId }) {
+	try {
+		return await ctx.db.tx(async (client) => {
+			const row = await client.oneOrNone(`${threadSelect()} FROM ${table(ctx, TABLE_THREADS)} WHERE id = :threadId FOR UPDATE`, { threadId });
+			if (!row) throw new Error(`Thread "${threadId}" not found`);
+			const normalized = parseThread(row);
+			if (normalized.resourceId === resourceId) return normalized;
+			const updatedAt = /* @__PURE__ */ new Date();
+			await client.none(`UPDATE ${table(ctx, TABLE_THREADS)} SET ${THREAD_RESOURCE_ID} = :resourceId, ${THREAD_UPDATED_AT} = :updatedAt WHERE id = :threadId`, {
+				resourceId,
+				updatedAt,
+				threadId
+			});
+			await client.none(`UPDATE ${table(ctx, TABLE_MESSAGES)} SET ${MESSAGE_RESOURCE_ID} = :resourceId WHERE thread_id = :threadId`, {
+				resourceId,
+				threadId
+			});
+			return {
+				...normalized,
+				resourceId,
+				updatedAt
+			};
+		});
+	} catch (error) {
+		throw storageError$1("UPDATE_THREAD_RESOURCE_ID", "FAILED", {
+			threadId,
+			resourceId
+		}, error);
+	}
+}
 async function listThreads(ctx, args) {
 	const { page = 0, perPage: perPageInput, orderBy, filter } = args;
 	try {
@@ -5379,6 +5409,9 @@ var MemoryOracle = class MemoryOracle extends MemoryStorage {
 	async deleteThread(args) {
 		return deleteThread(this.ctx, args);
 	}
+	async updateThreadResourceId(args) {
+		return updateThreadResourceId(this.ctx, args);
+	}
 	async listThreads(args) {
 		return listThreads(this.ctx, args);
 	}