@epam/ai-dial-attachment-canvas 1.0.19 → 1.0.21

package/index.js

@@ -21216,6 +21216,7 @@ var uv = class {
 		}), this.setRequestHandler(xg, async (e, t) => {
 			let n = async () => {
 				let r = e.params.taskId;
+				if (!await this._taskStore.getTask(r, t.sessionId)) throw new tv(Hh.InvalidParams, `Task not found: ${r}`);
 				if (this._taskMessageQueue) {
 					let e;
 					for (; e = await this._taskMessageQueue.dequeue(r, t.sessionId);) {
@@ -21237,10 +21238,10 @@ var uv = class {
 				}
 				let i = await this._taskStore.getTask(r, t.sessionId);
 				if (!i) throw new tv(Hh.InvalidParams, `Task not found: ${r}`);
-				if (!sv(i.status)) return await this._waitForTaskUpdate(r, t.signal), await n();
+				if (!sv(i.status)) return await this._waitForTaskUpdate(r, t.signal, t.sessionId), await n();
 				if (sv(i.status)) {
 					let e = await this._taskStore.getTaskResult(r, t.sessionId);
-					return this._clearTaskQueue(r), {
+					return this._clearTaskQueue(r, t.sessionId), {
 						...e,
 						_meta: {
 							...e._meta,
@@ -21267,7 +21268,7 @@ var uv = class {
 				let n = await this._taskStore.getTask(e.params.taskId, t.sessionId);
 				if (!n) throw new tv(Hh.InvalidParams, `Task not found: ${e.params.taskId}`);
 				if (sv(n.status)) throw new tv(Hh.InvalidParams, `Cannot cancel task in terminal status: ${n.status}`);
-				await this._taskStore.updateTaskStatus(e.params.taskId, "cancelled", "Client cancelled task execution.", t.sessionId), this._clearTaskQueue(e.params.taskId);
+				await this._taskStore.updateTaskStatus(e.params.taskId, "cancelled", "Client cancelled task execution.", t.sessionId), this._clearTaskQueue(e.params.taskId, t.sessionId);
 				let r = await this._taskStore.getTask(e.params.taskId, t.sessionId);
 				if (!r) throw new tv(Hh.InvalidParams, `Task not found after cancellation: ${e.params.taskId}`);
 				return {
@@ -21341,8 +21342,11 @@ var uv = class {
 		t !== void 0 && Promise.resolve().then(() => t(e)).catch((e) => this._onerror(/* @__PURE__ */ Error(`Uncaught error in notification handler: ${e}`)));
 	}
 	_onrequest(e, t) {
-		let n = this._requestHandlers.get(e.method) ?? this.fallbackRequestHandler, r = this._transport, i = e.params?._meta?.[xh]?.taskId;
-		if (n === void 0) {
+		let n = this._requestHandlers.get(e.method) ?? this.fallbackRequestHandler, r = this._transport, i = e.params?._meta?.[xh]?.taskId, a = r?.sessionId, o = this._taskStore, s = !0, c;
+		if (i && o && this._taskMessageQueue && a !== void 0 && (s = !1, c = (async () => {
+			if (!await o.getTask(i, a)) throw new tv(Hh.InvalidParams, `Task not found: ${i}`);
+			s = !0;
+		})()), n === void 0) {
 			let t = {
 				jsonrpc: "2.0",
 				id: e.id,
@@ -21351,47 +21355,55 @@ var uv = class {
 					message: "Method not found"
 				}
 			};
-			i && this._taskMessageQueue ? this._enqueueTaskMessage(i, {
+			if (i && c) {
+				let e = {
+					type: "error",
+					message: t,
+					timestamp: Date.now()
+				};
+				c.then(() => this._enqueueTaskMessage(i, e, a), () => r?.send(t)).catch((e) => this._onerror(/* @__PURE__ */ Error(`Failed to send an error response: ${e}`)));
+			} else i && this._taskMessageQueue ? this._enqueueTaskMessage(i, {
 				type: "error",
 				message: t,
 				timestamp: Date.now()
 			}, r?.sessionId).catch((e) => this._onerror(/* @__PURE__ */ Error(`Failed to enqueue error response: ${e}`))) : r?.send(t).catch((e) => this._onerror(/* @__PURE__ */ Error(`Failed to send an error response: ${e}`)));
 			return;
 		}
-		let a = new AbortController();
-		this._requestHandlerAbortControllers.set(e.id, a);
-		let o = Ah(e.params) ? e.params.task : void 0, s = this._taskStore ? this.requestTaskStore(e, r?.sessionId) : void 0, c = {
-			signal: a.signal,
+		let l = new AbortController();
+		this._requestHandlerAbortControllers.set(e.id, l);
+		let u = Ah(e.params) ? e.params.task : void 0, d = this._taskStore ? this.requestTaskStore(e, r?.sessionId) : void 0, f = {
+			signal: l.signal,
 			sessionId: r?.sessionId,
 			_meta: e.params?._meta,
 			sendNotification: async (t) => {
-				if (a.signal.aborted) return;
+				if (l.signal.aborted) return;
 				let n = { relatedRequestId: e.id };
 				i && (n.relatedTask = { taskId: i }), await this.notification(t, n);
 			},
 			sendRequest: async (t, n, r) => {
-				if (a.signal.aborted) throw new tv(Hh.ConnectionClosed, "Request was cancelled");
-				let o = {
+				if (l.signal.aborted) throw new tv(Hh.ConnectionClosed, "Request was cancelled");
+				let a = {
 					...r,
 					relatedRequestId: e.id
 				};
-				i && !o.relatedTask && (o.relatedTask = { taskId: i });
-				let c = o.relatedTask?.taskId ?? i;
-				return c && s && await s.updateTaskStatus(c, "input_required"), await this.request(t, n, o);
+				i && !a.relatedTask && (a.relatedTask = { taskId: i });
+				let o = a.relatedTask?.taskId ?? i;
+				return o && d && await d.updateTaskStatus(o, "input_required"), await this.request(t, n, a);
 			},
 			authInfo: t?.authInfo,
 			requestId: e.id,
 			requestInfo: t?.requestInfo,
 			taskId: i,
-			taskStore: s,
-			taskRequestedTtl: o?.ttl,
+			taskStore: d,
+			taskRequestedTtl: u?.ttl,
 			closeSSEStream: t?.closeSSEStream,
 			closeStandaloneSSEStream: t?.closeStandaloneSSEStream
 		};
-		Promise.resolve().then(() => {
-			o && this.assertTaskHandlerCapability(e.method);
-		}).then(() => n(e, c)).then(async (t) => {
-			if (a.signal.aborted) return;
+		(c ?? Promise.resolve()).then(() => {
+			if (c && l.signal.aborted) throw new tv(Hh.ConnectionClosed, "Request was cancelled");
+			u && this.assertTaskHandlerCapability(e.method);
+		}).then(() => n(e, f)).then(async (t) => {
+			if (l.signal.aborted) return;
 			let n = {
 				result: t,
 				jsonrpc: "2.0",
@@ -21403,7 +21415,7 @@ var uv = class {
 				timestamp: Date.now()
 			}, r?.sessionId) : await r?.send(n);
 		}, async (t) => {
-			if (a.signal.aborted) return;
+			if (l.signal.aborted) return;
 			let n = {
 				jsonrpc: "2.0",
 				id: e.id,
@@ -21413,13 +21425,13 @@ var uv = class {
 					...t.data !== void 0 && { data: t.data }
 				}
 			};
-			i && this._taskMessageQueue ? await this._enqueueTaskMessage(i, {
+			i && this._taskMessageQueue && s ? await this._enqueueTaskMessage(i, {
 				type: "error",
 				message: n,
 				timestamp: Date.now()
 			}, r?.sessionId) : await r?.send(n);
 		}).catch((e) => this._onerror(/* @__PURE__ */ Error(`Failed to send response: ${e}`))).finally(() => {
-			this._requestHandlerAbortControllers.get(e.id) === a && this._requestHandlerAbortControllers.delete(e.id);
+			this._requestHandlerAbortControllers.get(e.id) === l && this._requestHandlerAbortControllers.delete(e.id);
 		});
 	}
 	_onprogress(e) {
@@ -21723,7 +21735,7 @@ var uv = class {
 	async _enqueueTaskMessage(e, t, n) {
 		if (!this._taskStore || !this._taskMessageQueue) throw Error("Cannot enqueue task message: taskStore and taskMessageQueue are not configured");
 		let r = this._options?.maxTaskQueueSize;
-		await this._taskMessageQueue.enqueue(e, t, n, r);
+		await this._taskMessageQueue.enqueue(e, t, n ?? this._transport?.sessionId, r);
 	}
 	async _clearTaskQueue(e, t) {
 		if (this._taskMessageQueue) {
@@ -21734,20 +21746,20 @@ var uv = class {
 			}
 		}
 	}
-	async _waitForTaskUpdate(e, t) {
-		let n = this._options?.defaultTaskPollInterval ?? 1e3;
+	async _waitForTaskUpdate(e, t, n) {
+		let r = this._options?.defaultTaskPollInterval ?? 1e3;
 		try {
-			let t = await this._taskStore?.getTask(e);
-			t?.pollInterval && (n = t.pollInterval);
+			let t = await this._taskStore?.getTask(e, n);
+			t?.pollInterval && (r = t.pollInterval);
 		} catch {}
-		return new Promise((e, r) => {
+		return new Promise((e, n) => {
 			if (t.aborted) {
-				r(new tv(Hh.InvalidRequest, "Request cancelled"));
+				n(new tv(Hh.InvalidRequest, "Request cancelled"));
 				return;
 			}
-			let i = setTimeout(e, n);
+			let i = setTimeout(e, r);
 			t.addEventListener("abort", () => {
-				clearTimeout(i), r(new tv(Hh.InvalidRequest, "Request cancelled"));
+				clearTimeout(i), n(new tv(Hh.InvalidRequest, "Request cancelled"));
 			}, { once: !0 });
 		});
 	}

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@epam/ai-dial-attachment-canvas",
   "description": "Canvas viewer for rendering attachment content — images, PDFs, Office documents, JSON, markdown, and plain text",
-  "version": "1.0.19",
+  "version": "1.0.21",
   "license": "Apache-2.0",
   "type": "module",
   "main": "./index.js",
@@ -17,10 +17,10 @@
     }
   },
   "peerDependencies": {
-    "@epam/ai-dial-chat-shared": "1.0.19",
+    "@epam/ai-dial-chat-shared": "1.0.21",
     "@epam/ai-dial-react-pdf-highlighter": "*",
     "@epam/ai-dial-shared": "0.48.0",
-    "@epam/ai-dial-sidebar": "1.0.19",
+    "@epam/ai-dial-sidebar": "1.0.21",
     "@epam/ai-dial-visualizer-connector": "0.48.0",
     "@epam/ai-dial-ui-kit": "^0.14.0-dev.15",
     "@epam/pdf-highlighter-kit": ">=0.0.14",