@openhands/agent-canvas 1.0.0-alpha.8 → 1.0.0-alpha.9

package/dist/contexts/conversation-websocket-context.js

@@ -3,37 +3,37 @@ import { useCommandStore as t } from "../stores/command-store.js";
 import { setConversationState as n } from "../utils/conversation-local-storage.js";
 import { useConversationStore as r } from "../stores/conversation-store.js";
 import { useConversationStateStore as ee } from "../stores/conversation-state-store.js";
-import { isActionEvent as te, isAgentErrorEvent as ne, isAgentServerEvent as re, isAgentStatusConversationStateUpdateEvent as ie, isBrowserNavigateActionEvent as ae, isBrowserObservationEvent as oe, isCanvasUIActionEvent as se, isConversationStateUpdateEvent as i, isDisplayableErrorEvent as a, isExecuteBashActionEvent as o, isExecuteBashObservationEvent as s, isFullStateConversationStateUpdateEvent as ce, isPlanningFileEditorObservationEvent as le, isStatsConversationStateUpdateEvent as ue, isSwitchLLMObservationEvent as de, isUserMessageEvent as fe } from "../types/agent-server/type-guards.js";
-import { useEventStore as c } from "../stores/use-event-store.js";
-import { ConversationClient as pe } from "../node_modules/@openhands/typescript-client/dist/client/conversation-client.js";
-import { useQueryClient as me } from "../node_modules/@tanstack/react-query/build/modern/QueryClientProvider.js";
-import { usePostHog as he } from "../node_modules/posthog-js/react/dist/esm/index.js";
-import { useWebSocket as ge } from "../hooks/use-websocket.js";
-import { SERVER_CONNECTION_ERROR_MESSAGE as _e } from "../constants/server-connection-error.js";
-import { useErrorMessageStore as ve } from "../stores/error-message-store.js";
-import { useOptimisticUserMessageStore as ye } from "../stores/optimistic-user-message-store.js";
-import { handleCanvasUIAction as be } from "../services/canvas-ui.js";
-import { handleActionEventCacheInvalidation as xe } from "../utils/cache-utils.js";
-import { buildWebSocketUrl as Se } from "../utils/websocket-url.js";
-import { getAgentServerClientOptions as Ce } from "../api/agent-server-client-options.js";
-import we from "../api/event-service/event-service.api.js";
-import { trackError as l } from "../utils/error-handler.js";
-import { useReadConversationFile as Te } from "../hooks/mutation/use-read-conversation-file.js";
-import Ee from "../stores/metrics-store.js";
-import { useConversationHistory as De } from "../hooks/query/use-conversation-history.js";
-import { recordModelSwitchMessage as Oe } from "../hooks/chat/record-model-switch-message.js";
-import { invalidateConversationQueries as ke, updateConversationLlmModelInCache as Ae } from "../hooks/mutation/conversation-mutation-utils.js";
-import je, { createContext as Me, useCallback as u, useContext as d, useEffect as f, useLayoutEffect as Ne, useMemo as p, useRef as m, useState as h } from "react";
+import { isActionEvent as te, isAgentErrorEvent as ne, isAgentServerEvent as re, isAgentStatusConversationStateUpdateEvent as ie, isBrowserNavigateActionEvent as ae, isBrowserObservationEvent as oe, isCanvasUIActionEvent as se, isConversationStateUpdateEvent as i, isDisplayableErrorEvent as a, isExecuteBashActionEvent as o, isExecuteBashObservationEvent as s, isFullStateConversationStateUpdateEvent as ce, isPlanningFileEditorObservationEvent as le, isStatsConversationStateUpdateEvent as ue, isSwitchLLMObservationEvent as de, isUserMessageEvent as c } from "../types/agent-server/type-guards.js";
+import { useEventStore as l } from "../stores/use-event-store.js";
+import { ConversationClient as fe } from "../node_modules/@openhands/typescript-client/dist/client/conversation-client.js";
+import { useQueryClient as pe } from "../node_modules/@tanstack/react-query/build/modern/QueryClientProvider.js";
+import { usePostHog as me } from "../node_modules/posthog-js/react/dist/esm/index.js";
+import { useWebSocket as he } from "../hooks/use-websocket.js";
+import { SERVER_CONNECTION_ERROR_MESSAGE as ge } from "../constants/server-connection-error.js";
+import { useErrorMessageStore as _e } from "../stores/error-message-store.js";
+import { useOptimisticUserMessageStore as ve } from "../stores/optimistic-user-message-store.js";
+import { handleCanvasUIAction as ye } from "../services/canvas-ui.js";
+import { handleActionEventCacheInvalidation as be } from "../utils/cache-utils.js";
+import { buildWebSocketUrl as xe } from "../utils/websocket-url.js";
+import { getAgentServerClientOptions as Se } from "../api/agent-server-client-options.js";
+import Ce from "../api/event-service/event-service.api.js";
+import { trackError as u } from "../utils/error-handler.js";
+import { useReadConversationFile as we } from "../hooks/mutation/use-read-conversation-file.js";
+import Te from "../stores/metrics-store.js";
+import { useConversationHistory as Ee } from "../hooks/query/use-conversation-history.js";
+import { recordModelSwitchMessage as De } from "../hooks/chat/record-model-switch-message.js";
+import { invalidateConversationQueries as Oe, updateConversationLlmModelInCache as ke } from "../hooks/mutation/conversation-mutation-utils.js";
+import Ae, { createContext as je, useCallback as d, useContext as f, useEffect as p, useLayoutEffect as Me, useMemo as m, useRef as Ne, useState as h } from "react";
 import { jsx as Pe } from "react/jsx-runtime";
 //#region src/contexts/conversation-websocket-context.tsx
-var g = Me(void 0);
+var g = je(void 0);
 function _(e) {
 	return e.llm_message.content.filter((e) => e.type === "text").map((e) => e.text).join("");
 }
-function v({ children: Me, conversationId: d, conversationUrl: v, sessionApiKey: y, subConversations: b, subConversationIds: x }) {
-	let [S, C] = h("CONNECTING"), [w, T] = h("CONNECTING"), E = je.useRef(!1), D = je.useRef(!1), O = he(), k = me(), A = c((e) => e.addEvent), Fe = c((e) => e.addEvents), { setErrorMessage: j, removeErrorMessage: Ie, clearConnectionError: M } = ve(), N = ye((e) => e.consumeMatchingPendingMessage), { setExecutionStatus: P } = ee(), { appendInput: F, appendOutput: I } = t(), [L, R] = h(!0), [z, Le] = h(null), { setPlanContent: B } = r(), { mutate: V } = Te(), H = m(0), U = m(null), Re = (e) => e?.toUpperCase().endsWith("PLAN.MD") ?? !1, W = u(() => {
+function v({ children: je, conversationId: f, conversationUrl: v, sessionApiKey: y, subConversations: b, subConversationIds: x }) {
+	let [S, C] = h("CONNECTING"), [w, T] = h("CONNECTING"), E = Ae.useRef(!1), D = Ae.useRef(!1), O = me(), k = pe(), A = l((e) => e.addEvent), Fe = l((e) => e.addEvents), Ie = l((e) => e.clearEventsForConversation), { setErrorMessage: j, removeErrorMessage: Le, clearConnectionError: M } = _e(), N = ve((e) => e.consumeMatchingPendingMessage), { setExecutionStatus: P } = ee(), { appendInput: F, appendOutput: I } = t(), [L, R] = h(!0), [z, Re] = h(null), { setPlanContent: B } = r(), { mutate: V } = we(), H = Ne(0), U = Ne(null), ze = (e) => e?.toUpperCase().endsWith("PLAN.MD") ?? !1, W = d(() => {
 		M();
-	}, [M]), G = u((e) => {
+	}, [M]), G = d((e) => {
 		if (e.value.usage_to_metrics?.agent) {
 			let t = e.value.usage_to_metrics.agent, n = {
 				cost: t.accumulated_cost,
@@ -47,37 +47,45 @@ function v({ children: Me, conversationId: d, conversationUrl: v, sessionApiKey:
 					per_turn_token: t.accumulated_token_usage.per_turn_token
 				} : null
 			};
-			Ee.getState().setMetrics(n);
+			Te.getState().setMetrics(n);
 		}
-	}, []), { data: K, isPending: q, isError: ze } = De(d), Be = !!d && q;
-	Ne(() => {
-		!K || K.events.length === 0 || Fe(K.events);
-	}, [K, Fe]);
-	let J = p(() => {
+	}, []), { data: K, isPending: q, isError: Be } = Ee(f), Ve = !!f && q;
+	Me(() => {
+		let e = f ?? null;
+		l.getState().loadedConversationId !== e && Ie(e);
+	}, [f, Ie]), Me(() => {
+		if (!(!K || K.events.length === 0) && (Fe(K.events), f)) for (let e of K.events) c(e) && N(f, _(e));
+	}, [
+		K,
+		Fe,
+		f,
+		N
+	]);
+	let J = m(() => {
 		if (q) return null;
 		let e = K?.events ?? [], t = e[e.length - 1];
 		return !t || !("timestamp" in t) || !t.timestamp ? null : t.timestamp;
-	}, [K, q]), Y = p(() => !d || !v || q && !ze ? null : Se(d, v), [
-		d,
+	}, [K, q]), Y = m(() => !f || !v || q && !Be ? null : xe(f, v), [
+		f,
 		v,
 		q,
-		ze
-	]), X = p(() => {
+		Be
+	]), X = m(() => {
 		if (!b?.length) return null;
 		let e = b[0];
-		return !e?.id || !e.conversation_url ? null : Se(e.id, e.conversation_url);
-	}, [b]), Ve = p(() => X ? S === "CONNECTING" || w === "CONNECTING" ? "CONNECTING" : S === "OPEN" && w === "OPEN" ? "OPEN" : S === "CLOSED" && w === "CLOSED" ? "CLOSED" : S === "CLOSING" || w === "CLOSING" ? "CLOSING" : "CLOSED" : S, [
+		return !e?.id || !e.conversation_url ? null : xe(e.id, e.conversation_url);
+	}, [b]), He = m(() => X ? S === "CONNECTING" || w === "CONNECTING" ? "CONNECTING" : S === "OPEN" && w === "OPEN" ? "OPEN" : S === "CLOSED" && w === "CLOSED" ? "CLOSED" : S === "CLOSING" || w === "CLOSING" ? "CLOSING" : "CLOSED" : S, [
 		S,
 		w,
 		X
 	]);
-	f(() => {
+	p(() => {
 		z !== null && H.current >= z && L && R(!1);
 	}, [
 		z,
 		L,
 		H
-	]), f(() => {
+	]), p(() => {
 		if (!L && U.current) {
 			let { path: e, conversationId: t } = U.current;
 			V({
@@ -96,19 +104,19 @@ function v({ children: Me, conversationId: d, conversationUrl: v, sessionApiKey:
 		L,
 		V,
 		B
-	]), f(() => {
-		E.current = !1, R(!!x?.length), Le(null), H.current = 0, U.current = null;
-	}, [x]), f(() => {
+	]), p(() => {
+		E.current = !1, R(!!x?.length), Re(null), H.current = 0, U.current = null;
+	}, [x]), p(() => {
 		E.current = !1, D.current = !1, U.current = null;
-	}, [d]);
-	let He = p(() => Be || L, [Be, L]), Ue = u((t) => {
+	}, [f]);
+	let Ue = m(() => Ve || L, [Ve, L]), Z = d((t) => {
 		try {
 			let r = JSON.parse(t.data);
 			if (re(r)) {
-				let t = !c.getState().eventIds.has(r.id) && de(r) ? r : null;
+				let t = !l.getState().eventIds.has(r.id) && de(r) ? r : null;
 				if (A(r), a(r)) {
 					let e = r;
-					l({
+					u({
 						message: e.detail,
 						source: "conversation",
 						metadata: {
@@ -118,7 +126,7 @@ function v({ children: Me, conversationId: d, conversationUrl: v, sessionApiKey:
 						posthog: O
 					}), j(e.detail);
 				} else W();
-				if (ne(r) && l({
+				if (ne(r) && u({
 					message: r.error,
 					source: "agent",
 					metadata: {
@@ -127,14 +135,14 @@ function v({ children: Me, conversationId: d, conversationUrl: v, sessionApiKey:
 						toolCallId: r.tool_call_id
 					},
 					posthog: O
-				}), fe(r) && d && (N(d, _(r)), n(d, { draftMessage: null })), te(r) && xe(r, d || "test-conversation-id", k), i(r) && (ce(r) && P(r.value.execution_status), ie(r) && P(r.value), ue(r) && G(r)), o(r) && F(r.action.command), s(r) && I(r.observation.content.filter((e) => e.type === "text").map((e) => e.text).join("\n")), oe(r)) {
+				}), c(r) && f && (N(f, _(r)), n(f, { draftMessage: null })), te(r) && be(r, f || "test-conversation-id", k), i(r) && (ce(r) && P(r.value.execution_status), ie(r) && P(r.value), ue(r) && G(r)), o(r) && F(r.action.command), s(r) && I(r.observation.content.filter((e) => e.type === "text").map((e) => e.text).join("\n")), oe(r)) {
 					let { screenshot_data: t } = r.observation;
 					if (t) {
 						let n = t.startsWith("data:") ? t : `data:image/png;base64,${t}`;
 						e.getState().setScreenshotSrc(n);
 					}
 				}
-				ae(r) && e.getState().setUrl(r.action.url), d && t && !t.observation.is_error && (Oe(d, t.observation.profile_name), t.observation.active_model && Ae(k, d, t.observation.active_model), ke(k, d)), se(r) && be(r.action);
+				ae(r) && e.getState().setUrl(r.action.url), f && t && !t.observation.is_error && (De(f, t.observation.profile_name), t.observation.active_model && ke(k, f, t.observation.active_model), Oe(k, f)), se(r) && ye(r.action);
 			}
 		} catch (e) {
 			console.warn("Failed to parse WebSocket message as JSON:", e);
@@ -144,14 +152,14 @@ function v({ children: Me, conversationId: d, conversationUrl: v, sessionApiKey:
 		j,
 		N,
 		k,
-		d,
+		f,
 		P,
 		F,
 		I,
 		G,
 		W,
 		O
-	]), We = u((e) => {
+	]), We = d((e) => {
 		try {
 			let t = JSON.parse(e.data);
 			if (L && (H.current += 1, z !== null && H.current >= z && R(!1)), re(t)) {
@@ -160,7 +168,7 @@ function v({ children: Me, conversationId: d, conversationUrl: v, sessionApiKey:
 					isFromPlanningAgent: !0
 				}), a(t)) {
 					let e = t;
-					l({
+					u({
 						message: e.detail,
 						source: "planning_conversation",
 						metadata: {
@@ -170,7 +178,7 @@ function v({ children: Me, conversationId: d, conversationUrl: v, sessionApiKey:
 						posthog: O
 					}), j(e.detail);
 				} else W();
-				if (ne(t) && l({
+				if (ne(t) && u({
 					message: t.error,
 					source: "planning_agent",
 					metadata: {
@@ -179,9 +187,9 @@ function v({ children: Me, conversationId: d, conversationUrl: v, sessionApiKey:
 						toolCallId: t.tool_call_id
 					},
 					posthog: O
-				}), fe(t) && d && (N(d, _(t)), n(d, { draftMessage: null })), te(t) && xe(t, b?.[0]?.id || "test-conversation-id", k), i(t) && (ce(t) && P(t.value.execution_status), ie(t) && P(t.value), ue(t) && G(t)), o(t) && F(t.action.command), s(t) && I(t.observation.content.filter((e) => e.type === "text").map((e) => e.text).join("\n")), le(t)) {
+				}), c(t) && f && (N(f, _(t)), n(f, { draftMessage: null })), te(t) && be(t, b?.[0]?.id || "test-conversation-id", k), i(t) && (ce(t) && P(t.value.execution_status), ie(t) && P(t.value), ue(t) && G(t)), o(t) && F(t.action.command), s(t) && I(t.observation.content.filter((e) => e.type === "text").map((e) => e.text).join("\n")), le(t)) {
 					let { path: e } = t.observation;
-					if (Re(e)) {
+					if (ze(e)) {
 						let t = b?.[0]?.id;
 						t && e && (L ? U.current = {
 							path: e,
@@ -211,7 +219,7 @@ function v({ children: Me, conversationId: d, conversationUrl: v, sessionApiKey:
 		N,
 		k,
 		b,
-		d,
+		f,
 		P,
 		F,
 		I,
@@ -220,7 +228,7 @@ function v({ children: Me, conversationId: d, conversationUrl: v, sessionApiKey:
 		G,
 		W,
 		O
-	]), Ge = p(() => {
+	]), Ge = m(() => {
 		let e = J ? {
 			resend_mode: "since",
 			after_timestamp: J
@@ -235,17 +243,17 @@ function v({ children: Me, conversationId: d, conversationUrl: v, sessionApiKey:
 				C("CLOSED");
 			},
 			onError: () => {
-				C("CLOSED"), E.current && j(_e, "connection");
+				C("CLOSED"), E.current && j(ge, "connection");
 			},
-			onMessage: Ue
+			onMessage: Z
 		};
 	}, [
-		Ue,
+		Z,
 		j,
 		M,
 		y,
 		J
-	]), Ke = p(() => {
+	]), Ke = m(() => {
 		let e = { resend_all: !0 };
 		y && (e.session_api_key = y);
 		let t = b?.[0];
@@ -254,8 +262,8 @@ function v({ children: Me, conversationId: d, conversationUrl: v, sessionApiKey:
 			reconnect: { enabled: !0 },
 			onOpen: async () => {
 				if (T("OPEN"), D.current = !0, M(), t?.id && t.conversation_url) try {
-					let e = await we.getEventCount(t.id, t.conversation_url, t.session_api_key);
-					Le(e), e === 0 && R(!1);
+					let e = await Ce.getEventCount(t.id, t.conversation_url, t.session_api_key);
+					Re(e), e === 0 && R(!1);
 				} catch {
 					R(!1);
 				}
@@ -264,7 +272,7 @@ function v({ children: Me, conversationId: d, conversationUrl: v, sessionApiKey:
 				T("CLOSED");
 			},
 			onError: () => {
-				T("CLOSED"), D.current && j(_e, "connection");
+				T("CLOSED"), D.current && j(ge, "connection");
 			},
 			onMessage: We
 		};
@@ -274,8 +282,8 @@ function v({ children: Me, conversationId: d, conversationUrl: v, sessionApiKey:
 		M,
 		y,
 		b
-	]), { socket: Z, reconnect: qe } = ge(Y || "", Ge), { socket: Q, reconnect: Je } = ge(X || "", Ke), Ye = u(() => {
-		if (Ie(), r.getState().conversationMode === "plan" && X) {
+	]), { socket: Q, reconnect: qe } = he(Y || "", Ge), { socket: $, reconnect: Je } = he(X || "", Ke), Ye = d(() => {
+		if (Le(), r.getState().conversationMode === "plan" && X) {
 			Je();
 			return;
 		}
@@ -284,16 +292,16 @@ function v({ children: Me, conversationId: d, conversationUrl: v, sessionApiKey:
 		X,
 		qe,
 		Je,
-		Ie
-	]), $ = u(async (e) => {
-		let t = r.getState().conversationMode === "plan" ? Q : Z;
+		Le
+	]), Xe = d(async (e) => {
+		let t = r.getState().conversationMode === "plan" ? $ : Q;
 		if (t?.readyState !== WebSocket.OPEN) {
-			if (!d) {
+			if (!f) {
 				let e = /* @__PURE__ */ Error("No conversation ID available");
 				throw j(e.message), e;
 			}
 			try {
-				return await new pe(Ce()).sendEvent(d, {
+				return await new fe(Se()).sendEvent(f, {
 					role: "user",
 					content: e.content
 				}, { run: !0 }), { queued: !0 };
@@ -310,14 +318,14 @@ function v({ children: Me, conversationId: d, conversationUrl: v, sessionApiKey:
 			throw j(e instanceof Error ? e.message : "Failed to send message"), e;
 		}
 	}, [
-		Z,
 		Q,
+		$,
 		j,
-		d
+		f
 	]);
-	f(() => {
-		Z && Y && (() => {
-			switch (Z.readyState) {
+	p(() => {
+		Q && Y && (() => {
+			switch (Q.readyState) {
 				case WebSocket.CONNECTING:
 					C("CONNECTING");
 					break;
@@ -335,9 +343,9 @@ function v({ children: Me, conversationId: d, conversationUrl: v, sessionApiKey:
 					break;
 			}
 		})();
-	}, [Z, Y]), f(() => {
-		Q && X && (() => {
-			switch (Q.readyState) {
+	}, [Q, Y]), p(() => {
+		$ && X && (() => {
+			switch ($.readyState) {
 				case WebSocket.CONNECTING:
 					T("CONNECTING");
 					break;
@@ -355,24 +363,24 @@ function v({ children: Me, conversationId: d, conversationUrl: v, sessionApiKey:
 					break;
 			}
 		})();
-	}, [Q, X]);
-	let Xe = p(() => ({
-		connectionState: Ve,
-		sendMessage: $,
-		isLoadingHistory: He,
+	}, [$, X]);
+	let Ze = m(() => ({
+		connectionState: He,
+		sendMessage: Xe,
+		isLoadingHistory: Ue,
 		reconnect: Ye
 	}), [
-		Ve,
-		$,
 		He,
+		Xe,
+		Ue,
 		Ye
 	]);
 	return /* @__PURE__ */ Pe(g.Provider, {
-		value: Xe,
-		children: Me
+		value: Ze,
+		children: je
 	});
 }
-var y = () => d(g) || null;
+var y = () => f(g) || null;
 //#endregion
 export { v as ConversationWebSocketProvider, y as useConversationWebSocket };
 

package/dist/contexts/conversation-websocket-context.js.map

@@ -1 +1 @@
-{"version":3,"file":"conversation-websocket-context.js","names":[],"sources":["../../src/contexts/conversation-websocket-context.tsx"],"sourcesContent":["import React, {\n  createContext,\n  useContext,\n  useEffect,\n  useLayoutEffect,\n  useState,\n  useCallback,\n  useMemo,\n  useRef,\n} from \"react\";\nimport { ConversationClient } from \"@openhands/typescript-client/clients\";\n\nimport { useQueryClient } from \"@tanstack/react-query\";\nimport { usePostHog } from \"posthog-js/react\";\nimport { useWebSocket, WebSocketHookOptions } from \"#/hooks/use-websocket\";\nimport { SERVER_CONNECTION_ERROR_MESSAGE } from \"#/constants/server-connection-error\";\nimport { useEventStore } from \"#/stores/use-event-store\";\nimport { useErrorMessageStore } from \"#/stores/error-message-store\";\nimport { useOptimisticUserMessageStore } from \"#/stores/optimistic-user-message-store\";\nimport { useConversationStateStore } from \"#/stores/conversation-state-store\";\nimport { useCommandStore } from \"#/stores/command-store\";\nimport { useBrowserStore } from \"#/stores/browser-store\";\nimport {\n  isAgentServerEvent,\n  isAgentErrorEvent,\n  isUserMessageEvent,\n  isActionEvent,\n  isConversationStateUpdateEvent,\n  isFullStateConversationStateUpdateEvent,\n  isAgentStatusConversationStateUpdateEvent,\n  isStatsConversationStateUpdateEvent,\n  isExecuteBashActionEvent,\n  isExecuteBashObservationEvent,\n  isDisplayableErrorEvent,\n  isPlanningFileEditorObservationEvent,\n  isBrowserObservationEvent,\n  isBrowserNavigateActionEvent,\n  isSwitchLLMObservationEvent,\n  isCanvasUIActionEvent,\n} from \"#/types/agent-server/type-guards\";\nimport { handleCanvasUIAction } from \"#/services/canvas-ui\";\nimport { ConversationStateUpdateEventStats } from \"#/types/agent-server/core/events/conversation-state-event\";\nimport type {\n  ConversationErrorEvent,\n  ServerErrorEvent,\n} from \"#/types/agent-server/core/events/conversation-state-event\";\nimport { handleActionEventCacheInvalidation } from \"#/utils/cache-utils\";\nimport { buildWebSocketUrl } from \"#/utils/websocket-url\";\nimport type {\n  AppConversation,\n  SendMessageRequest,\n} from \"#/api/conversation-service/agent-server-conversation-service.types\";\nimport EventService from \"#/api/event-service/event-service.api\";\nimport { getAgentServerClientOptions } from \"#/api/agent-server-client-options\";\nimport { useConversationStore } from \"#/stores/conversation-store\";\nimport { trackError } from \"#/utils/error-handler\";\nimport { useReadConversationFile } from \"#/hooks/mutation/use-read-conversation-file\";\nimport useMetricsStore from \"#/stores/metrics-store\";\nimport { useConversationHistory } from \"#/hooks/query/use-conversation-history\";\nimport { setConversationState } from \"#/utils/conversation-local-storage\";\nimport { recordModelSwitchMessage } from \"#/hooks/chat/record-model-switch-message\";\nimport {\n  invalidateConversationQueries,\n  updateConversationLlmModelInCache,\n} from \"#/hooks/mutation/conversation-mutation-utils\";\n\nexport type WebSocketConnectionState =\n  | \"CONNECTING\"\n  | \"OPEN\"\n  | \"CLOSED\"\n  | \"CLOSING\";\n\ninterface SendMessageResult {\n  queued: boolean; // true if message was queued for later delivery, false if sent immediately\n}\n\ninterface ConversationWebSocketContextType {\n  connectionState: WebSocketConnectionState;\n  sendMessage: (message: SendMessageRequest) => Promise<SendMessageResult>;\n  isLoadingHistory: boolean;\n  reconnect: () => void;\n}\n\nconst ConversationWebSocketContext = createContext<\n  ConversationWebSocketContextType | undefined\n>(undefined);\n\n/**\n * Extract the text body of an echoed user `MessageEvent` for matching against\n * the optimistic pending-message queue. The server wraps the original\n * `args.content` string in one or more `TextContent` entries (alongside any\n * `ImageContent` entries for inline images), so concatenating the `text`\n * fields gives us back the exact prompt we sent.\n */\nfunction extractMessageEventText(\n  event: import(\"#/types/agent-server/core/events/message-event\").MessageEvent,\n): string {\n  return event.llm_message.content\n    .filter(\n      (part): part is { type: \"text\"; text: string } => part.type === \"text\",\n    )\n    .map((part) => part.text)\n    .join(\"\");\n}\n\nexport function ConversationWebSocketProvider({\n  children,\n  conversationId,\n  conversationUrl,\n  sessionApiKey,\n  subConversations,\n  subConversationIds,\n}: {\n  children: React.ReactNode;\n  conversationId?: string;\n  conversationUrl?: string | null;\n  sessionApiKey?: string | null;\n  subConversations?: AppConversation[];\n  subConversationIds?: string[];\n}) {\n  // Separate connection state tracking for each WebSocket\n  const [mainConnectionState, setMainConnectionState] =\n    useState<WebSocketConnectionState>(\"CONNECTING\");\n  const [planningConnectionState, setPlanningConnectionState] =\n    useState<WebSocketConnectionState>(\"CONNECTING\");\n\n  // Track if we've ever successfully connected for each connection\n  // Don't show errors until after first successful connection\n  const hasConnectedRefMain = React.useRef(false);\n  const hasConnectedRefPlanning = React.useRef(false);\n\n  const posthog = usePostHog();\n  const queryClient = useQueryClient();\n  const addEvent = useEventStore((state) => state.addEvent);\n  const addEvents = useEventStore((state) => state.addEvents);\n  const { setErrorMessage, removeErrorMessage, clearConnectionError } =\n    useErrorMessageStore();\n  const consumeMatchingPendingMessage = useOptimisticUserMessageStore(\n    (state) => state.consumeMatchingPendingMessage,\n  );\n  const { setExecutionStatus } = useConversationStateStore();\n  const { appendInput, appendOutput } = useCommandStore();\n\n  // History loading state.\n  // - Main conversation history is now loaded via REST (`useConversationHistory`),\n  //   so its loading state mirrors the REST query state (see below).\n  // - Planning sub-conversation history still streams over the WebSocket using\n  //   `resend_mode='all'`, so we keep the count-based detection for it.\n  const [isLoadingHistoryPlanning, setIsLoadingHistoryPlanning] =\n    useState(true);\n  const [expectedEventCountPlanning, setExpectedEventCountPlanning] = useState<\n    number | null\n  >(null);\n\n  const { setPlanContent } = useConversationStore();\n\n  // Hook for reading conversation file\n  const { mutate: readConversationFile } = useReadConversationFile();\n\n  // Track planning-agent received events (still WS-driven).\n  const receivedEventCountRefPlanning = useRef(0);\n\n  // Track the latest PlanningFileEditorObservation for Plan.md during history replay\n  const latestPlanningFileEventRef = useRef<{\n    path: string;\n    conversationId: string;\n  } | null>(null);\n\n  const isPlanFilePath = (path: string | null): boolean =>\n    path?.toUpperCase().endsWith(\"PLAN.MD\") ?? false;\n\n  const handleNonErrorEvent = useCallback(() => {\n    // A normal event means connectivity recovered: clear a transient connection\n    // error, but keep sticky conversation errors (e.g. a wrong API key).\n    clearConnectionError();\n  }, [clearConnectionError]);\n\n  // Helper function to update metrics from stats event\n  const updateMetricsFromStats = useCallback(\n    (event: ConversationStateUpdateEventStats) => {\n      if (event.value.usage_to_metrics?.agent) {\n        const agentMetrics = event.value.usage_to_metrics.agent;\n        const metrics = {\n          cost: agentMetrics.accumulated_cost,\n          max_budget_per_task: agentMetrics.max_budget_per_task ?? null,\n          usage: agentMetrics.accumulated_token_usage\n            ? {\n                prompt_tokens:\n                  agentMetrics.accumulated_token_usage.prompt_tokens,\n                completion_tokens:\n                  agentMetrics.accumulated_token_usage.completion_tokens,\n                cache_read_tokens:\n                  agentMetrics.accumulated_token_usage.cache_read_tokens,\n                cache_write_tokens:\n                  agentMetrics.accumulated_token_usage.cache_write_tokens,\n                context_window:\n                  agentMetrics.accumulated_token_usage.context_window,\n                per_turn_token:\n                  agentMetrics.accumulated_token_usage.per_turn_token,\n              }\n            : null,\n        };\n        useMetricsStore.getState().setMetrics(metrics);\n      }\n    },\n    [],\n  );\n\n  // Initial REST history load: fetch the most recent events and seed the\n  // store. Older events are paginated in via `useLoadOlderEvents` when the\n  // user scrolls to the top of the chat. The WebSocket connection waits for\n  // this query so it can subscribe with `resend_mode='since'` and avoid\n  // re-streaming everything REST already returned.\n  const {\n    data: preloadedHistory,\n    isPending: isPreloadingHistory,\n    isError: isPreloadHistoryError,\n  } = useConversationHistory(conversationId);\n\n  const isLoadingHistoryMain = !!conversationId && isPreloadingHistory;\n\n  useLayoutEffect(() => {\n    if (!preloadedHistory || preloadedHistory.events.length === 0) {\n      return;\n    }\n    addEvents(preloadedHistory.events);\n  }, [preloadedHistory, addEvents]);\n\n  /**\n   * Timestamp of the latest event we already have from REST. Used as\n   * `after_timestamp` when opening the WebSocket so the server only resends\n   * events strictly after this point. `null` until the REST query settles\n   * (we hold the WS connection open until then to avoid an `all` resend).\n   */\n  const initialAfterTimestamp = useMemo<string | null>(() => {\n    if (isPreloadingHistory) return null;\n    const events = preloadedHistory?.events ?? [];\n    const latest = events[events.length - 1];\n    if (!latest || !(\"timestamp\" in latest) || !latest.timestamp) return null;\n    return latest.timestamp;\n  }, [preloadedHistory, isPreloadingHistory]);\n\n  // Build WebSocket URL from props.\n  //\n  // We deliberately wait for the initial REST history fetch to settle before\n  // opening the socket so the WS subscription can use `resend_mode='since'`\n  // with a meaningful `after_timestamp`. Without this gate, the WS would open\n  // immediately and either replay the entire conversation (when falling back\n  // to `resend_mode='all'`) or miss events that arrived between REST and WS.\n  const wsUrl = useMemo(() => {\n    if (!conversationId || !conversationUrl) {\n      return null;\n    }\n    // Don't connect while we're still fetching the initial history. If the\n    // REST query errored we fall through and connect with `resend_mode='all'`\n    // so the user still sees live events.\n    if (isPreloadingHistory && !isPreloadHistoryError) {\n      return null;\n    }\n    return buildWebSocketUrl(conversationId, conversationUrl);\n  }, [\n    conversationId,\n    conversationUrl,\n    isPreloadingHistory,\n    isPreloadHistoryError,\n  ]);\n\n  const planningAgentWsUrl = useMemo(() => {\n    if (!subConversations?.length) {\n      return null;\n    }\n\n    // Currently, there is only one sub-conversation and it uses the planning agent.\n    const planningAgentConversation = subConversations[0];\n\n    if (\n      !planningAgentConversation?.id ||\n      !planningAgentConversation.conversation_url\n    ) {\n      return null;\n    }\n\n    return buildWebSocketUrl(\n      planningAgentConversation.id,\n      planningAgentConversation.conversation_url,\n    );\n  }, [subConversations]);\n\n  // Merged connection state - reflects combined status of both connections\n  const connectionState = useMemo<WebSocketConnectionState>(() => {\n    // If planning agent connection doesn't exist, use main connection state\n    if (!planningAgentWsUrl) {\n      return mainConnectionState;\n    }\n\n    // If either is connecting, merged state is connecting\n    if (\n      mainConnectionState === \"CONNECTING\" ||\n      planningConnectionState === \"CONNECTING\"\n    ) {\n      return \"CONNECTING\";\n    }\n\n    // If both are open, merged state is open\n    if (mainConnectionState === \"OPEN\" && planningConnectionState === \"OPEN\") {\n      return \"OPEN\";\n    }\n\n    // If both are closed, merged state is closed\n    if (\n      mainConnectionState === \"CLOSED\" &&\n      planningConnectionState === \"CLOSED\"\n    ) {\n      return \"CLOSED\";\n    }\n\n    // If either is closing, merged state is closing\n    if (\n      mainConnectionState === \"CLOSING\" ||\n      planningConnectionState === \"CLOSING\"\n    ) {\n      return \"CLOSING\";\n    }\n\n    // Default to closed if states don't match expected patterns\n    return \"CLOSED\";\n  }, [mainConnectionState, planningConnectionState, planningAgentWsUrl]);\n\n  useEffect(() => {\n    if (\n      expectedEventCountPlanning !== null &&\n      receivedEventCountRefPlanning.current >= expectedEventCountPlanning &&\n      isLoadingHistoryPlanning\n    ) {\n      setIsLoadingHistoryPlanning(false);\n    }\n  }, [\n    expectedEventCountPlanning,\n    isLoadingHistoryPlanning,\n    receivedEventCountRefPlanning,\n  ]);\n\n  // Call API once after history loading completes if we tracked any PlanningFileEditorObservation events\n  useEffect(() => {\n    if (!isLoadingHistoryPlanning && latestPlanningFileEventRef.current) {\n      const { path, conversationId: currentPlanningConversationId } =\n        latestPlanningFileEventRef.current;\n\n      readConversationFile(\n        {\n          conversationId: currentPlanningConversationId,\n          filePath: path,\n        },\n        {\n          onSuccess: (fileContent) => {\n            setPlanContent(fileContent);\n          },\n          onError: (error) => {\n            console.warn(\"Failed to read conversation file:\", error);\n          },\n        },\n      );\n\n      // Clear the ref after calling the API\n      latestPlanningFileEventRef.current = null;\n    }\n  }, [isLoadingHistoryPlanning, readConversationFile, setPlanContent]);\n\n  useEffect(() => {\n    hasConnectedRefMain.current = false;\n    setIsLoadingHistoryPlanning(!!subConversationIds?.length);\n    setExpectedEventCountPlanning(null);\n    receivedEventCountRefPlanning.current = 0;\n    // Reset the tracked event ref when sub-conversations change\n    latestPlanningFileEventRef.current = null;\n  }, [subConversationIds]);\n\n  // Reset hasConnected flags when the conversation changes.\n  useEffect(() => {\n    hasConnectedRefMain.current = false;\n    hasConnectedRefPlanning.current = false;\n    // Reset the tracked event ref when conversation changes\n    latestPlanningFileEventRef.current = null;\n  }, [conversationId]);\n\n  // Merged loading history state - true if either connection is still loading\n  const isLoadingHistory = useMemo(\n    () => isLoadingHistoryMain || isLoadingHistoryPlanning,\n    [isLoadingHistoryMain, isLoadingHistoryPlanning],\n  );\n\n  // Separate message handlers for each connection\n  const handleMainMessage = useCallback(\n    (messageEvent: MessageEvent) => {\n      try {\n        const event = JSON.parse(messageEvent.data);\n\n        // History loading for the main conversation is REST-driven now;\n        // every WS message is a new event we add to the store.\n\n        // Use type guard to validate v1 event structure\n        if (isAgentServerEvent(event)) {\n          const isDuplicateEvent = useEventStore\n            .getState()\n            .eventIds.has(event.id);\n          const switchLLMObservation =\n            !isDuplicateEvent && isSwitchLLMObservationEvent(event)\n              ? event\n              : null;\n          addEvent(event);\n\n          // Handle displayable error events - show error banner\n          // AgentErrorEvent errors are displayed inline in the chat, not as banners\n          if (isDisplayableErrorEvent(event)) {\n            const errorEvent = event as\n              | ConversationErrorEvent\n              | ServerErrorEvent;\n            trackError({\n              message: errorEvent.detail,\n              source: \"conversation\",\n              metadata: {\n                eventId: errorEvent.id,\n                errorCode: errorEvent.code,\n              },\n              posthog,\n            });\n            setErrorMessage(errorEvent.detail);\n          } else {\n            handleNonErrorEvent();\n          }\n\n          // LLM errors render inline in the chat (see ErrorEventMessage); track\n          // them for analytics but keep them out of the banner above the chat box.\n          if (isAgentErrorEvent(event)) {\n            trackError({\n              message: event.error,\n              source: \"agent\",\n              metadata: {\n                eventId: event.id,\n                toolName: event.tool_name,\n                toolCallId: event.tool_call_id,\n              },\n              posthog,\n            });\n          }\n\n          // Clear optimistic user message when a user message is confirmed.\n          // We match by the echoed text content (with FIFO fallback inside the\n          // store), so an echo for \"second\" pops \"second\" — not whichever\n          // pending entry happens to be oldest — protecting against any\n          // out-of-order delivery between conversations or sub-agents.\n          if (isUserMessageEvent(event)) {\n            if (conversationId) {\n              consumeMatchingPendingMessage(\n                conversationId,\n                extractMessageEventText(event),\n              );\n              // Clear draft from localStorage - message was successfully delivered\n              setConversationState(conversationId, { draftMessage: null });\n            }\n          }\n\n          // Handle cache invalidation for ActionEvent\n          if (isActionEvent(event)) {\n            const currentConversationId =\n              conversationId || \"test-conversation-id\"; // TODO: Get from context\n            handleActionEventCacheInvalidation(\n              event,\n              currentConversationId,\n              queryClient,\n            );\n          }\n\n          // Handle conversation state updates\n          // TODO: Tests\n          if (isConversationStateUpdateEvent(event)) {\n            if (isFullStateConversationStateUpdateEvent(event)) {\n              setExecutionStatus(event.value.execution_status);\n            }\n            if (isAgentStatusConversationStateUpdateEvent(event)) {\n              setExecutionStatus(event.value);\n            }\n            if (isStatsConversationStateUpdateEvent(event)) {\n              updateMetricsFromStats(event);\n            }\n          }\n\n          // Handle ExecuteBashAction events - add command as input to terminal\n          if (isExecuteBashActionEvent(event)) {\n            appendInput(event.action.command);\n          }\n\n          // Handle ExecuteBashObservation events - add output to terminal\n          if (isExecuteBashObservationEvent(event)) {\n            // Extract text content from the observation content array\n            const textContent = event.observation.content\n              .filter((c) => c.type === \"text\")\n              .map((c) => c.text)\n              .join(\"\\n\");\n            appendOutput(textContent);\n          }\n\n          // Handle BrowserObservation events - update browser store with screenshot\n          if (isBrowserObservationEvent(event)) {\n            const { screenshot_data: screenshotData } = event.observation;\n            if (screenshotData) {\n              const screenshotSrc = screenshotData.startsWith(\"data:\")\n                ? screenshotData\n                : `data:image/png;base64,${screenshotData}`;\n              useBrowserStore.getState().setScreenshotSrc(screenshotSrc);\n            }\n          }\n\n          // Handle BrowserNavigateAction events - update browser store with URL\n          if (isBrowserNavigateActionEvent(event)) {\n            useBrowserStore.getState().setUrl(event.action.url);\n          }\n\n          if (\n            conversationId &&\n            switchLLMObservation &&\n            !switchLLMObservation.observation.is_error\n          ) {\n            recordModelSwitchMessage(\n              conversationId,\n              switchLLMObservation.observation.profile_name,\n            );\n\n            if (switchLLMObservation.observation.active_model) {\n              updateConversationLlmModelInCache(\n                queryClient,\n                conversationId,\n                switchLLMObservation.observation.active_model,\n              );\n            }\n\n            invalidateConversationQueries(queryClient, conversationId);\n          }\n\n          // Handle canvas_ui custom-tool ActionEvents - drive the frontend\n          // (navigate to a file, switch tabs, show a preview). The tool\n          // executes server-side as a no-op; the actual UI change happens\n          // here on the client.\n          if (isCanvasUIActionEvent(event)) {\n            handleCanvasUIAction(event.action);\n          }\n        }\n      } catch (error) {\n        console.warn(\"Failed to parse WebSocket message as JSON:\", error);\n      }\n    },\n    [\n      addEvent,\n      setErrorMessage,\n      consumeMatchingPendingMessage,\n      queryClient,\n      conversationId,\n      setExecutionStatus,\n      appendInput,\n      appendOutput,\n      updateMetricsFromStats,\n      handleNonErrorEvent,\n      posthog,\n    ],\n  );\n\n  const handlePlanningMessage = useCallback(\n    (messageEvent: MessageEvent) => {\n      try {\n        const event = JSON.parse(messageEvent.data);\n\n        // Track received events for history loading (count ALL events from WebSocket)\n        // Always count when loading, even if we don't have the expected count yet\n        if (isLoadingHistoryPlanning) {\n          receivedEventCountRefPlanning.current += 1;\n\n          if (\n            expectedEventCountPlanning !== null &&\n            receivedEventCountRefPlanning.current >= expectedEventCountPlanning\n          ) {\n            setIsLoadingHistoryPlanning(false);\n          }\n        }\n\n        // Use type guard to validate v1 event structure\n        if (isAgentServerEvent(event)) {\n          // Mark this event as coming from the planning agent\n          const eventWithPlanningFlag = {\n            ...event,\n            isFromPlanningAgent: true,\n          };\n          addEvent(eventWithPlanningFlag);\n\n          // Handle displayable error events - show error banner\n          // AgentErrorEvent errors are displayed inline in the chat, not as banners\n          if (isDisplayableErrorEvent(event)) {\n            const errorEvent = event as\n              | ConversationErrorEvent\n              | ServerErrorEvent;\n            trackError({\n              message: errorEvent.detail,\n              source: \"planning_conversation\",\n              metadata: {\n                eventId: errorEvent.id,\n                errorCode: errorEvent.code,\n              },\n              posthog,\n            });\n            setErrorMessage(errorEvent.detail);\n          } else {\n            handleNonErrorEvent();\n          }\n\n          // LLM errors render inline in the chat (see ErrorEventMessage); track\n          // them for analytics but keep them out of the banner above the chat box.\n          if (isAgentErrorEvent(event)) {\n            trackError({\n              message: event.error,\n              source: \"planning_agent\",\n              metadata: {\n                eventId: event.id,\n                toolName: event.tool_name,\n                toolCallId: event.tool_call_id,\n              },\n              posthog,\n            });\n          }\n\n          // Clear optimistic user message when a user message is confirmed.\n          // Always scope to the main `conversationId` (where the user types)\n          // and match on the echoed content so the planning sub-agent's own\n          // events can never consume a main-conversation pending entry.\n          if (isUserMessageEvent(event)) {\n            if (conversationId) {\n              consumeMatchingPendingMessage(\n                conversationId,\n                extractMessageEventText(event),\n              );\n              setConversationState(conversationId, { draftMessage: null });\n            }\n          }\n\n          // Handle cache invalidation for ActionEvent\n          if (isActionEvent(event)) {\n            const planningAgentConversation = subConversations?.[0];\n            const currentConversationId =\n              planningAgentConversation?.id || \"test-conversation-id\"; // TODO: Get from context\n            handleActionEventCacheInvalidation(\n              event,\n              currentConversationId,\n              queryClient,\n            );\n          }\n\n          // Handle conversation state updates\n          // TODO: Tests\n          if (isConversationStateUpdateEvent(event)) {\n            if (isFullStateConversationStateUpdateEvent(event)) {\n              setExecutionStatus(event.value.execution_status);\n            }\n            if (isAgentStatusConversationStateUpdateEvent(event)) {\n              setExecutionStatus(event.value);\n            }\n            if (isStatsConversationStateUpdateEvent(event)) {\n              updateMetricsFromStats(event);\n            }\n          }\n\n          // Handle ExecuteBashAction events - add command as input to terminal\n          if (isExecuteBashActionEvent(event)) {\n            appendInput(event.action.command);\n          }\n\n          // Handle ExecuteBashObservation events - add output to terminal\n          if (isExecuteBashObservationEvent(event)) {\n            // Extract text content from the observation content array\n            const textContent = event.observation.content\n              .filter((c) => c.type === \"text\")\n              .map((c) => c.text)\n              .join(\"\\n\");\n            appendOutput(textContent);\n          }\n\n          // Handle PlanningFileEditorObservation - only update plan for Plan.md\n          if (isPlanningFileEditorObservationEvent(event)) {\n            const { path } = event.observation;\n            if (isPlanFilePath(path)) {\n              const planningAgentConversation = subConversations?.[0];\n              const planningConversationId = planningAgentConversation?.id;\n\n              if (planningConversationId && path) {\n                if (isLoadingHistoryPlanning) {\n                  latestPlanningFileEventRef.current = {\n                    path,\n                    conversationId: planningConversationId,\n                  };\n                } else {\n                  readConversationFile(\n                    {\n                      conversationId: planningConversationId,\n                      filePath: path,\n                    },\n                    {\n                      onSuccess: (fileContent) => {\n                        setPlanContent(fileContent);\n                      },\n                      onError: (error) => {\n                        console.warn(\n                          \"Failed to read conversation file:\",\n                          error,\n                        );\n                      },\n                    },\n                  );\n                }\n              }\n            }\n          }\n        }\n      } catch (error) {\n        console.warn(\"Failed to parse WebSocket message as JSON:\", error);\n      }\n    },\n    [\n      addEvent,\n      isLoadingHistoryPlanning,\n      expectedEventCountPlanning,\n      setErrorMessage,\n      consumeMatchingPendingMessage,\n      queryClient,\n      subConversations,\n      conversationId,\n      setExecutionStatus,\n      appendInput,\n      appendOutput,\n      readConversationFile,\n      setPlanContent,\n      updateMetricsFromStats,\n      handleNonErrorEvent,\n      posthog,\n    ],\n  );\n\n  // Separate WebSocket options for main connection\n  const mainWebsocketOptions: WebSocketHookOptions = useMemo(() => {\n    // History was already loaded over REST (`useConversationHistory`).\n    // Subscribe with `resend_mode='since'` so the server only resends events\n    // strictly after the latest one we already have. If REST returned no\n    // events at all (brand-new conversation), fall back to `'all'` so any\n    // events that may have been written between the REST call and the WS\n    // handshake still show up. Dedup in the event store handles overlap.\n    const queryParams: Record<string, string | boolean> = initialAfterTimestamp\n      ? { resend_mode: \"since\", after_timestamp: initialAfterTimestamp }\n      : { resend_mode: \"all\" };\n\n    // Add session_api_key if available\n    if (sessionApiKey) {\n      queryParams.session_api_key = sessionApiKey;\n    }\n\n    return {\n      queryParams,\n      reconnect: { enabled: true },\n      onOpen: () => {\n        setMainConnectionState(\"OPEN\");\n        hasConnectedRefMain.current = true; // Mark that we've successfully connected\n        clearConnectionError(); // Clear a previous connection error; keep sticky conversation errors\n      },\n      onClose: () => {\n        setMainConnectionState(\"CLOSED\");\n      },\n      onError: () => {\n        setMainConnectionState(\"CLOSED\");\n        // Only show error message if we've previously connected successfully\n        if (hasConnectedRefMain.current) {\n          setErrorMessage(SERVER_CONNECTION_ERROR_MESSAGE, \"connection\");\n        }\n      },\n      onMessage: handleMainMessage,\n    };\n  }, [\n    handleMainMessage,\n    setErrorMessage,\n    clearConnectionError,\n    sessionApiKey,\n    initialAfterTimestamp,\n  ]);\n\n  // Separate WebSocket options for planning agent connection\n  const planningWebsocketOptions: WebSocketHookOptions = useMemo(() => {\n    const queryParams: Record<string, string | boolean> = {\n      resend_all: true,\n    };\n\n    // Add session_api_key if available\n    if (sessionApiKey) {\n      queryParams.session_api_key = sessionApiKey;\n    }\n\n    const planningAgentConversation = subConversations?.[0];\n\n    return {\n      queryParams,\n      reconnect: { enabled: true },\n      onOpen: async () => {\n        setPlanningConnectionState(\"OPEN\");\n        hasConnectedRefPlanning.current = true; // Mark that we've successfully connected\n        clearConnectionError(); // Clear a previous connection error; keep sticky conversation errors\n\n        // Fetch expected event count for history loading detection\n        if (\n          planningAgentConversation?.id &&\n          planningAgentConversation.conversation_url\n        ) {\n          try {\n            const count = await EventService.getEventCount(\n              planningAgentConversation.id,\n              planningAgentConversation.conversation_url,\n              planningAgentConversation.session_api_key,\n            );\n            setExpectedEventCountPlanning(count);\n\n            // If no events expected, mark as loaded immediately\n            if (count === 0) {\n              setIsLoadingHistoryPlanning(false);\n            }\n          } catch (error) {\n            // Fall back to marking as loaded to avoid infinite loading state\n            setIsLoadingHistoryPlanning(false);\n          }\n        }\n      },\n      onClose: () => {\n        setPlanningConnectionState(\"CLOSED\");\n      },\n      onError: () => {\n        setPlanningConnectionState(\"CLOSED\");\n        // Only show error message if we've previously connected successfully\n        if (hasConnectedRefPlanning.current) {\n          setErrorMessage(SERVER_CONNECTION_ERROR_MESSAGE, \"connection\");\n        }\n      },\n      onMessage: handlePlanningMessage,\n    };\n  }, [\n    handlePlanningMessage,\n    setErrorMessage,\n    clearConnectionError,\n    sessionApiKey,\n    subConversations,\n  ]);\n\n  // Only attempt WebSocket connection when we have a valid URL\n  // This prevents connection attempts during task polling phase\n  const websocketUrl = wsUrl;\n  const { socket: mainSocket, reconnect: reconnectMain } = useWebSocket(\n    websocketUrl || \"\",\n    mainWebsocketOptions,\n  );\n\n  const { socket: planningAgentSocket, reconnect: reconnectPlanning } =\n    useWebSocket(planningAgentWsUrl || \"\", planningWebsocketOptions);\n\n  const reconnect = useCallback(() => {\n    removeErrorMessage();\n    const currentMode = useConversationStore.getState().conversationMode;\n    if (currentMode === \"plan\" && planningAgentWsUrl) {\n      reconnectPlanning();\n      return;\n    }\n    reconnectMain();\n  }, [\n    planningAgentWsUrl,\n    reconnectMain,\n    reconnectPlanning,\n    removeErrorMessage,\n  ]);\n\n  // V1 send message function via WebSocket\n  // Falls back to REST API queue when WebSocket is not connected\n  const sendMessage = useCallback(\n    async (message: SendMessageRequest): Promise<SendMessageResult> => {\n      const currentMode = useConversationStore.getState().conversationMode;\n      const currentSocket =\n        currentMode === \"plan\" ? planningAgentSocket : mainSocket;\n\n      if (currentSocket?.readyState !== WebSocket.OPEN) {\n        // WebSocket not connected - queue message via REST API\n        // Message will be delivered automatically when conversation becomes ready\n        if (!conversationId) {\n          const error = new Error(\"No conversation ID available\");\n          setErrorMessage(error.message);\n          throw error;\n        }\n\n        try {\n          await new ConversationClient(getAgentServerClientOptions()).sendEvent(\n            conversationId,\n            {\n              role: \"user\",\n              content: message.content,\n            },\n            { run: true },\n          );\n          // Message queued successfully - it will be delivered when ready\n          // Return queued: true so caller knows not to show optimistic UI\n          return { queued: true };\n        } catch (error) {\n          const errorMessage =\n            error instanceof Error\n              ? error.message\n              : \"Failed to queue message for delivery\";\n          setErrorMessage(errorMessage);\n          throw error;\n        }\n      }\n\n      try {\n        // Send message through WebSocket as JSON with run: true so the\n        // agent loop starts automatically in async mode.\n        currentSocket.send(JSON.stringify({ ...message, run: true }));\n        return { queued: false };\n      } catch (error) {\n        const errorMessage =\n          error instanceof Error ? error.message : \"Failed to send message\";\n        setErrorMessage(errorMessage);\n        throw error;\n      }\n    },\n    [mainSocket, planningAgentSocket, setErrorMessage, conversationId],\n  );\n\n  // Track main socket state changes\n  useEffect(() => {\n    // Only process socket updates if we have a valid URL and socket\n    if (mainSocket && wsUrl) {\n      // Update state based on socket readyState\n      const updateState = () => {\n        switch (mainSocket.readyState) {\n          case WebSocket.CONNECTING:\n            setMainConnectionState(\"CONNECTING\");\n            break;\n          case WebSocket.OPEN:\n            setMainConnectionState(\"OPEN\");\n            break;\n          case WebSocket.CLOSING:\n            setMainConnectionState(\"CLOSING\");\n            break;\n          case WebSocket.CLOSED:\n            setMainConnectionState(\"CLOSED\");\n            break;\n          default:\n            setMainConnectionState(\"CLOSED\");\n            break;\n        }\n      };\n\n      updateState();\n    }\n  }, [mainSocket, wsUrl]);\n\n  // Track planning agent socket state changes\n  useEffect(() => {\n    // Only process socket updates if we have a valid URL and socket\n    if (planningAgentSocket && planningAgentWsUrl) {\n      // Update state based on socket readyState\n      const updateState = () => {\n        switch (planningAgentSocket.readyState) {\n          case WebSocket.CONNECTING:\n            setPlanningConnectionState(\"CONNECTING\");\n            break;\n          case WebSocket.OPEN:\n            setPlanningConnectionState(\"OPEN\");\n            break;\n          case WebSocket.CLOSING:\n            setPlanningConnectionState(\"CLOSING\");\n            break;\n          case WebSocket.CLOSED:\n            setPlanningConnectionState(\"CLOSED\");\n            break;\n          default:\n            setPlanningConnectionState(\"CLOSED\");\n            break;\n        }\n      };\n\n      updateState();\n    }\n  }, [planningAgentSocket, planningAgentWsUrl]);\n\n  const contextValue = useMemo(\n    () => ({ connectionState, sendMessage, isLoadingHistory, reconnect }),\n    [connectionState, sendMessage, isLoadingHistory, reconnect],\n  );\n\n  return (\n    <ConversationWebSocketContext.Provider value={contextValue}>\n      {children}\n    </ConversationWebSocketContext.Provider>\n  );\n}\n\nexport const useConversationWebSocket =\n  (): ConversationWebSocketContextType | null => {\n    const context = useContext(ConversationWebSocketContext);\n    // Return null instead of throwing when not in provider\n    // This allows the hook to be called conditionally based on conversation version\n    return context || null;\n  };\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;AAmFA,IAAM,IAA+B,GAEnC,KAAA,EAAU;AASZ,SAAS,EACP,GACQ;AACR,QAAO,EAAM,YAAY,QACtB,QACE,MAAiD,EAAK,SAAS,OACjE,CACA,KAAK,MAAS,EAAK,KAAK,CACxB,KAAK,GAAG;;AAGb,SAAgB,EAA8B,EAC5C,cACA,mBACA,oBACA,kBACA,qBACA,yBAQC;CAED,IAAM,CAAC,GAAqB,KAC1B,EAAmC,aAAa,EAC5C,CAAC,GAAyB,KAC9B,EAAmC,aAAa,EAI5C,IAAsB,GAAM,OAAO,GAAM,EACzC,IAA0B,GAAM,OAAO,GAAM,EAE7C,IAAU,IAAY,EACtB,IAAc,IAAgB,EAC9B,IAAW,GAAe,MAAU,EAAM,SAAS,EACnD,KAAY,GAAe,MAAU,EAAM,UAAU,EACrD,EAAE,oBAAiB,wBAAoB,4BAC3C,IAAsB,EAClB,IAAgC,IACnC,MAAU,EAAM,8BAClB,EACK,EAAE,0BAAuB,IAA2B,EACpD,EAAE,gBAAa,oBAAiB,GAAiB,EAOjD,CAAC,GAA0B,KAC/B,EAAS,GAAK,EACV,CAAC,GAA4B,MAAiC,EAElE,KAAK,EAED,EAAE,sBAAmB,GAAsB,EAG3C,EAAE,QAAQ,MAAyB,IAAyB,EAG5D,IAAgC,EAAO,EAAE,EAGzC,IAA6B,EAGzB,KAAK,EAET,MAAkB,MACtB,GAAM,aAAa,CAAC,SAAS,UAAU,IAAI,IAEvC,IAAsB,QAAkB;AAG5C,KAAsB;IACrB,CAAC,EAAqB,CAAC,EAGpB,IAAyB,GAC5B,MAA6C;AAC5C,MAAI,EAAM,MAAM,kBAAkB,OAAO;GACvC,IAAM,IAAe,EAAM,MAAM,iBAAiB,OAC5C,IAAU;IACd,MAAM,EAAa;IACnB,qBAAqB,EAAa,uBAAuB;IACzD,OAAO,EAAa,0BAChB;KACE,eACE,EAAa,wBAAwB;KACvC,mBACE,EAAa,wBAAwB;KACvC,mBACE,EAAa,wBAAwB;KACvC,oBACE,EAAa,wBAAwB;KACvC,gBACE,EAAa,wBAAwB;KACvC,gBACE,EAAa,wBAAwB;KACxC,GACD;IACL;AACD,MAAgB,UAAU,CAAC,WAAW,EAAQ;;IAGlD,EAAE,CACH,EAOK,EACJ,MAAM,GACN,WAAW,GACX,SAAS,OACP,GAAuB,EAAe,EAEpC,KAAuB,CAAC,CAAC,KAAkB;AAEjD,UAAsB;AAChB,GAAC,KAAoB,EAAiB,OAAO,WAAW,KAG5D,GAAU,EAAiB,OAAO;IACjC,CAAC,GAAkB,GAAU,CAAC;CAQjC,IAAM,IAAwB,QAA6B;AACzD,MAAI,EAAqB,QAAO;EAChC,IAAM,IAAS,GAAkB,UAAU,EAAE,EACvC,IAAS,EAAO,EAAO,SAAS;AAEtC,SADI,CAAC,KAAU,EAAE,eAAe,MAAW,CAAC,EAAO,YAAkB,OAC9D,EAAO;IACb,CAAC,GAAkB,EAAoB,CAAC,EASrC,IAAQ,QACR,CAAC,KAAkB,CAAC,KAMpB,KAAuB,CAAC,KACnB,OAEF,GAAkB,GAAgB,EAAgB,EACxD;EACD;EACA;EACA;EACA;EACD,CAAC,EAEI,IAAqB,QAAc;AACvC,MAAI,CAAC,GAAkB,OACrB,QAAO;EAIT,IAAM,IAA4B,EAAiB;AASnD,SANE,CAAC,GAA2B,MAC5B,CAAC,EAA0B,mBAEpB,OAGF,GACL,EAA0B,IAC1B,EAA0B,iBAC3B;IACA,CAAC,EAAiB,CAAC,EAGhB,KAAkB,QAEjB,IAMH,MAAwB,gBACxB,MAA4B,eAErB,eAIL,MAAwB,UAAU,MAA4B,SACzD,SAKP,MAAwB,YACxB,MAA4B,WAErB,WAKP,MAAwB,aACxB,MAA4B,YAErB,YAIF,WAjCE,GAkCR;EAAC;EAAqB;EAAyB;EAAmB,CAAC;AAoDtE,CAlDA,QAAgB;AACd,EACE,MAA+B,QAC/B,EAA8B,WAAW,KACzC,KAEA,EAA4B,GAAM;IAEnC;EACD;EACA;EACA;EACD,CAAC,EAGF,QAAgB;AACd,MAAI,CAAC,KAA4B,EAA2B,SAAS;GACnE,IAAM,EAAE,SAAM,gBAAgB,MAC5B,EAA2B;AAkB7B,GAhBA,EACE;IACE,gBAAgB;IAChB,UAAU;IACX,EACD;IACE,YAAY,MAAgB;AAC1B,OAAe,EAAY;;IAE7B,UAAU,MAAU;AAClB,aAAQ,KAAK,qCAAqC,EAAM;;IAE3D,CACF,EAGD,EAA2B,UAAU;;IAEtC;EAAC;EAA0B;EAAsB;EAAe,CAAC,EAEpE,QAAgB;AAMd,EALA,EAAoB,UAAU,IAC9B,EAA4B,CAAC,CAAC,GAAoB,OAAO,EACzD,GAA8B,KAAK,EACnC,EAA8B,UAAU,GAExC,EAA2B,UAAU;IACpC,CAAC,EAAmB,CAAC,EAGxB,QAAgB;AAId,EAHA,EAAoB,UAAU,IAC9B,EAAwB,UAAU,IAElC,EAA2B,UAAU;IACpC,CAAC,EAAe,CAAC;CAGpB,IAAM,KAAmB,QACjB,MAAwB,GAC9B,CAAC,IAAsB,EAAyB,CACjD,EAGK,KAAoB,GACvB,MAA+B;AAC9B,MAAI;GACF,IAAM,IAAQ,KAAK,MAAM,EAAa,KAAK;AAM3C,OAAI,GAAmB,EAAM,EAAE;IAI7B,IAAM,IACJ,CAJuB,EACtB,UAAU,CACV,SAAS,IAAI,EAAM,GAEnB,IAAoB,GAA4B,EAAM,GACnD,IACA;AAKN,QAJA,EAAS,EAAM,EAIX,EAAwB,EAAM,EAAE;KAClC,IAAM,IAAa;AAYnB,KATA,EAAW;MACT,SAAS,EAAW;MACpB,QAAQ;MACR,UAAU;OACR,SAAS,EAAW;OACpB,WAAW,EAAW;OACvB;MACD;MACD,CAAC,EACF,EAAgB,EAAW,OAAO;UAElC,IAAqB;AA2EvB,QAtEI,GAAkB,EAAM,IAC1B,EAAW;KACT,SAAS,EAAM;KACf,QAAQ;KACR,UAAU;MACR,SAAS,EAAM;MACf,UAAU,EAAM;MAChB,YAAY,EAAM;MACnB;KACD;KACD,CAAC,EAQA,GAAmB,EAAM,IACvB,MACF,EACE,GACA,EAAwB,EAAM,CAC/B,EAED,EAAqB,GAAgB,EAAE,cAAc,MAAM,CAAC,GAK5D,GAAc,EAAM,IAGtB,GACE,GAFA,KAAkB,wBAIlB,EACD,EAKC,EAA+B,EAAM,KACnC,GAAwC,EAAM,IAChD,EAAmB,EAAM,MAAM,iBAAiB,EAE9C,GAA0C,EAAM,IAClD,EAAmB,EAAM,MAAM,EAE7B,GAAoC,EAAM,IAC5C,EAAuB,EAAM,GAK7B,EAAyB,EAAM,IACjC,EAAY,EAAM,OAAO,QAAQ,EAI/B,EAA8B,EAAM,IAMtC,EAJoB,EAAM,YAAY,QACnC,QAAQ,MAAM,EAAE,SAAS,OAAO,CAChC,KAAK,MAAM,EAAE,KAAK,CAClB,KAAK,KACK,CAAY,EAIvB,GAA0B,EAAM,EAAE;KACpC,IAAM,EAAE,iBAAiB,MAAmB,EAAM;AAClD,SAAI,GAAgB;MAClB,IAAM,IAAgB,EAAe,WAAW,QAAQ,GACpD,IACA,yBAAyB;AAC7B,QAAgB,UAAU,CAAC,iBAAiB,EAAc;;;AAkC9D,IA7BI,GAA6B,EAAM,IACrC,EAAgB,UAAU,CAAC,OAAO,EAAM,OAAO,IAAI,EAInD,KACA,KACA,CAAC,EAAqB,YAAY,aAElC,GACE,GACA,EAAqB,YAAY,aAClC,EAEG,EAAqB,YAAY,gBACnC,GACE,GACA,GACA,EAAqB,YAAY,aAClC,EAGH,GAA8B,GAAa,EAAe,GAOxD,GAAsB,EAAM,IAC9B,GAAqB,EAAM,OAAO;;WAG/B,GAAO;AACd,WAAQ,KAAK,8CAA8C,EAAM;;IAGrE;EACE;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACD,CACF,EAEK,KAAwB,GAC3B,MAA+B;AAC9B,MAAI;GACF,IAAM,IAAQ,KAAK,MAAM,EAAa,KAAK;AAgB3C,OAZI,MACF,EAA8B,WAAW,GAGvC,MAA+B,QAC/B,EAA8B,WAAW,KAEzC,EAA4B,GAAM,GAKlC,GAAmB,EAAM,EAAE;AAU7B,QAJA,EAAS;KAHP,GAAG;KACH,qBAAqB;KAEd,CAAsB,EAI3B,EAAwB,EAAM,EAAE;KAClC,IAAM,IAAa;AAYnB,KATA,EAAW;MACT,SAAS,EAAW;MACpB,QAAQ;MACR,UAAU;OACR,SAAS,EAAW;OACpB,WAAW,EAAW;OACvB;MACD;MACD,CAAC,EACF,EAAgB,EAAW,OAAO;UAElC,IAAqB;AA0EvB,QArEI,GAAkB,EAAM,IAC1B,EAAW;KACT,SAAS,EAAM;KACf,QAAQ;KACR,UAAU;MACR,SAAS,EAAM;MACf,UAAU,EAAM;MAChB,YAAY,EAAM;MACnB;KACD;KACD,CAAC,EAOA,GAAmB,EAAM,IACvB,MACF,EACE,GACA,EAAwB,EAAM,CAC/B,EACD,EAAqB,GAAgB,EAAE,cAAc,MAAM,CAAC,GAK5D,GAAc,EAAM,IAItB,GACE,GAJgC,IAAmB,IAExB,MAAM,wBAIjC,EACD,EAKC,EAA+B,EAAM,KACnC,GAAwC,EAAM,IAChD,EAAmB,EAAM,MAAM,iBAAiB,EAE9C,GAA0C,EAAM,IAClD,EAAmB,EAAM,MAAM,EAE7B,GAAoC,EAAM,IAC5C,EAAuB,EAAM,GAK7B,EAAyB,EAAM,IACjC,EAAY,EAAM,OAAO,QAAQ,EAI/B,EAA8B,EAAM,IAMtC,EAJoB,EAAM,YAAY,QACnC,QAAQ,MAAM,EAAE,SAAS,OAAO,CAChC,KAAK,MAAM,EAAE,KAAK,CAClB,KAAK,KACK,CAAY,EAIvB,GAAqC,EAAM,EAAE;KAC/C,IAAM,EAAE,YAAS,EAAM;AACvB,SAAI,GAAe,EAAK,EAAE;MAExB,IAAM,IAD4B,IAAmB,IACK;AAE1D,MAAI,KAA0B,MACxB,IACF,EAA2B,UAAU;OACnC;OACA,gBAAgB;OACjB,GAED,EACE;OACE,gBAAgB;OAChB,UAAU;OACX,EACD;OACE,YAAY,MAAgB;AAC1B,UAAe,EAAY;;OAE7B,UAAU,MAAU;AAClB,gBAAQ,KACN,qCACA,EACD;;OAEJ,CACF;;;;WAMJ,GAAO;AACd,WAAQ,KAAK,8CAA8C,EAAM;;IAGrE;EACE;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACD,CACF,EAGK,KAA6C,QAAc;EAO/D,IAAM,IAAgD,IAClD;GAAE,aAAa;GAAS,iBAAiB;GAAuB,GAChE,EAAE,aAAa,OAAO;AAO1B,SAJI,MACF,EAAY,kBAAkB,IAGzB;GACL;GACA,WAAW,EAAE,SAAS,IAAM;GAC5B,cAAc;AAGZ,IAFA,EAAuB,OAAO,EAC9B,EAAoB,UAAU,IAC9B,GAAsB;;GAExB,eAAe;AACb,MAAuB,SAAS;;GAElC,eAAe;AAGb,IAFA,EAAuB,SAAS,EAE5B,EAAoB,WACtB,EAAgB,IAAiC,aAAa;;GAGlE,WAAW;GACZ;IACA;EACD;EACA;EACA;EACA;EACA;EACD,CAAC,EAGI,KAAiD,QAAc;EACnE,IAAM,IAAgD,EACpD,YAAY,IACb;AAGD,EAAI,MACF,EAAY,kBAAkB;EAGhC,IAAM,IAA4B,IAAmB;AAErD,SAAO;GACL;GACA,WAAW,EAAE,SAAS,IAAM;GAC5B,QAAQ,YAAY;AAMlB,QALA,EAA2B,OAAO,EAClC,EAAwB,UAAU,IAClC,GAAsB,EAIpB,GAA2B,MAC3B,EAA0B,iBAE1B,KAAI;KACF,IAAM,IAAQ,MAAM,GAAa,cAC/B,EAA0B,IAC1B,EAA0B,kBAC1B,EAA0B,gBAC3B;AAID,KAHA,GAA8B,EAAM,EAGhC,MAAU,KACZ,EAA4B,GAAM;YAEtB;AAEd,OAA4B,GAAM;;;GAIxC,eAAe;AACb,MAA2B,SAAS;;GAEtC,eAAe;AAGb,IAFA,EAA2B,SAAS,EAEhC,EAAwB,WAC1B,EAAgB,IAAiC,aAAa;;GAGlE,WAAW;GACZ;IACA;EACD;EACA;EACA;EACA;EACA;EACD,CAAC,EAKI,EAAE,QAAQ,GAAY,WAAW,OAAkB,GACvD,KAAgB,IAChB,GACD,EAEK,EAAE,QAAQ,GAAqB,WAAW,OAC9C,GAAa,KAAsB,IAAI,GAAyB,EAE5D,KAAY,QAAkB;AAGlC,MAFA,IAAoB,EACA,EAAqB,UAAU,CAAC,qBAChC,UAAU,GAAoB;AAChD,OAAmB;AACnB;;AAEF,MAAe;IACd;EACD;EACA;EACA;EACA;EACD,CAAC,EAII,IAAc,EAClB,OAAO,MAA4D;EAEjE,IAAM,IADc,EAAqB,UAAU,CAAC,qBAElC,SAAS,IAAsB;AAEjD,MAAI,GAAe,eAAe,UAAU,MAAM;AAGhD,OAAI,CAAC,GAAgB;IACnB,IAAM,IAAQ,gBAAI,MAAM,+BAA+B;AAEvD,UADA,EAAgB,EAAM,QAAQ,EACxB;;AAGR,OAAI;AAWF,WAVA,MAAM,IAAI,GAAmB,IAA6B,CAAC,CAAC,UAC1D,GACA;KACE,MAAM;KACN,SAAS,EAAQ;KAClB,EACD,EAAE,KAAK,IAAM,CACd,EAGM,EAAE,QAAQ,IAAM;YAChB,GAAO;AAMd,UADA,EAHE,aAAiB,QACb,EAAM,UACN,uCACuB,EACvB;;;AAIV,MAAI;AAIF,UADA,EAAc,KAAK,KAAK,UAAU;IAAE,GAAG;IAAS,KAAK;IAAM,CAAC,CAAC,EACtD,EAAE,QAAQ,IAAO;WACjB,GAAO;AAId,SADA,EADE,aAAiB,QAAQ,EAAM,UAAU,yBACd,EACvB;;IAGV;EAAC;EAAY;EAAqB;EAAiB;EAAe,CACnE;AAgCD,CA7BA,QAAgB;AAEd,EAAI,KAAc,YAEU;AACxB,WAAQ,EAAW,YAAnB;IACE,KAAK,UAAU;AACb,OAAuB,aAAa;AACpC;IACF,KAAK,UAAU;AACb,OAAuB,OAAO;AAC9B;IACF,KAAK,UAAU;AACb,OAAuB,UAAU;AACjC;IACF,KAAK,UAAU;AACb,OAAuB,SAAS;AAChC;IACF;AACE,OAAuB,SAAS;AAChC;;MAIO;IAEd,CAAC,GAAY,EAAM,CAAC,EAGvB,QAAgB;AAEd,EAAI,KAAuB,YAEC;AACxB,WAAQ,EAAoB,YAA5B;IACE,KAAK,UAAU;AACb,OAA2B,aAAa;AACxC;IACF,KAAK,UAAU;AACb,OAA2B,OAAO;AAClC;IACF,KAAK,UAAU;AACb,OAA2B,UAAU;AACrC;IACF,KAAK,UAAU;AACb,OAA2B,SAAS;AACpC;IACF;AACE,OAA2B,SAAS;AACpC;;MAIO;IAEd,CAAC,GAAqB,EAAmB,CAAC;CAE7C,IAAM,KAAe,SACZ;EAAE;EAAiB;EAAa;EAAkB;EAAW,GACpE;EAAC;EAAiB;EAAa;EAAkB;EAAU,CAC5D;AAED,QACE,mBAAC,EAA6B,UAA9B;EAAuC,OAAO;EAC3C;EACqC,CAAA;;AAI5C,IAAa,UAEO,EAAW,EAGpB,IAAW"}
+{"version":3,"file":"conversation-websocket-context.js","names":[],"sources":["../../src/contexts/conversation-websocket-context.tsx"],"sourcesContent":["import React, {\n  createContext,\n  useContext,\n  useEffect,\n  useLayoutEffect,\n  useState,\n  useCallback,\n  useMemo,\n  useRef,\n} from \"react\";\nimport { ConversationClient } from \"@openhands/typescript-client/clients\";\n\nimport { useQueryClient } from \"@tanstack/react-query\";\nimport { usePostHog } from \"posthog-js/react\";\nimport { useWebSocket, WebSocketHookOptions } from \"#/hooks/use-websocket\";\nimport { SERVER_CONNECTION_ERROR_MESSAGE } from \"#/constants/server-connection-error\";\nimport { useEventStore } from \"#/stores/use-event-store\";\nimport { useErrorMessageStore } from \"#/stores/error-message-store\";\nimport { useOptimisticUserMessageStore } from \"#/stores/optimistic-user-message-store\";\nimport { useConversationStateStore } from \"#/stores/conversation-state-store\";\nimport { useCommandStore } from \"#/stores/command-store\";\nimport { useBrowserStore } from \"#/stores/browser-store\";\nimport {\n  isAgentServerEvent,\n  isAgentErrorEvent,\n  isUserMessageEvent,\n  isActionEvent,\n  isConversationStateUpdateEvent,\n  isFullStateConversationStateUpdateEvent,\n  isAgentStatusConversationStateUpdateEvent,\n  isStatsConversationStateUpdateEvent,\n  isExecuteBashActionEvent,\n  isExecuteBashObservationEvent,\n  isDisplayableErrorEvent,\n  isPlanningFileEditorObservationEvent,\n  isBrowserObservationEvent,\n  isBrowserNavigateActionEvent,\n  isSwitchLLMObservationEvent,\n  isCanvasUIActionEvent,\n} from \"#/types/agent-server/type-guards\";\nimport { handleCanvasUIAction } from \"#/services/canvas-ui\";\nimport { ConversationStateUpdateEventStats } from \"#/types/agent-server/core/events/conversation-state-event\";\nimport type {\n  ConversationErrorEvent,\n  ServerErrorEvent,\n} from \"#/types/agent-server/core/events/conversation-state-event\";\nimport { handleActionEventCacheInvalidation } from \"#/utils/cache-utils\";\nimport { buildWebSocketUrl } from \"#/utils/websocket-url\";\nimport type {\n  AppConversation,\n  SendMessageRequest,\n} from \"#/api/conversation-service/agent-server-conversation-service.types\";\nimport EventService from \"#/api/event-service/event-service.api\";\nimport { getAgentServerClientOptions } from \"#/api/agent-server-client-options\";\nimport { useConversationStore } from \"#/stores/conversation-store\";\nimport { trackError } from \"#/utils/error-handler\";\nimport { useReadConversationFile } from \"#/hooks/mutation/use-read-conversation-file\";\nimport useMetricsStore from \"#/stores/metrics-store\";\nimport { useConversationHistory } from \"#/hooks/query/use-conversation-history\";\nimport { setConversationState } from \"#/utils/conversation-local-storage\";\nimport { recordModelSwitchMessage } from \"#/hooks/chat/record-model-switch-message\";\nimport {\n  invalidateConversationQueries,\n  updateConversationLlmModelInCache,\n} from \"#/hooks/mutation/conversation-mutation-utils\";\n\nexport type WebSocketConnectionState =\n  | \"CONNECTING\"\n  | \"OPEN\"\n  | \"CLOSED\"\n  | \"CLOSING\";\n\ninterface SendMessageResult {\n  queued: boolean; // true if message was queued for later delivery, false if sent immediately\n}\n\ninterface ConversationWebSocketContextType {\n  connectionState: WebSocketConnectionState;\n  sendMessage: (message: SendMessageRequest) => Promise<SendMessageResult>;\n  isLoadingHistory: boolean;\n  reconnect: () => void;\n}\n\nconst ConversationWebSocketContext = createContext<\n  ConversationWebSocketContextType | undefined\n>(undefined);\n\n/**\n * Extract the text body of an echoed user `MessageEvent` for matching against\n * the optimistic pending-message queue. The server wraps the original\n * `args.content` string in one or more `TextContent` entries (alongside any\n * `ImageContent` entries for inline images), so concatenating the `text`\n * fields gives us back the exact prompt we sent.\n */\nfunction extractMessageEventText(\n  event: import(\"#/types/agent-server/core/events/message-event\").MessageEvent,\n): string {\n  return event.llm_message.content\n    .filter(\n      (part): part is { type: \"text\"; text: string } => part.type === \"text\",\n    )\n    .map((part) => part.text)\n    .join(\"\");\n}\n\nexport function ConversationWebSocketProvider({\n  children,\n  conversationId,\n  conversationUrl,\n  sessionApiKey,\n  subConversations,\n  subConversationIds,\n}: {\n  children: React.ReactNode;\n  conversationId?: string;\n  conversationUrl?: string | null;\n  sessionApiKey?: string | null;\n  subConversations?: AppConversation[];\n  subConversationIds?: string[];\n}) {\n  // Separate connection state tracking for each WebSocket\n  const [mainConnectionState, setMainConnectionState] =\n    useState<WebSocketConnectionState>(\"CONNECTING\");\n  const [planningConnectionState, setPlanningConnectionState] =\n    useState<WebSocketConnectionState>(\"CONNECTING\");\n\n  // Track if we've ever successfully connected for each connection\n  // Don't show errors until after first successful connection\n  const hasConnectedRefMain = React.useRef(false);\n  const hasConnectedRefPlanning = React.useRef(false);\n\n  const posthog = usePostHog();\n  const queryClient = useQueryClient();\n  const addEvent = useEventStore((state) => state.addEvent);\n  const addEvents = useEventStore((state) => state.addEvents);\n  const clearEventsForConversation = useEventStore(\n    (state) => state.clearEventsForConversation,\n  );\n  const { setErrorMessage, removeErrorMessage, clearConnectionError } =\n    useErrorMessageStore();\n  const consumeMatchingPendingMessage = useOptimisticUserMessageStore(\n    (state) => state.consumeMatchingPendingMessage,\n  );\n  const { setExecutionStatus } = useConversationStateStore();\n  const { appendInput, appendOutput } = useCommandStore();\n\n  // History loading state.\n  // - Main conversation history is now loaded via REST (`useConversationHistory`),\n  //   so its loading state mirrors the REST query state (see below).\n  // - Planning sub-conversation history still streams over the WebSocket using\n  //   `resend_mode='all'`, so we keep the count-based detection for it.\n  const [isLoadingHistoryPlanning, setIsLoadingHistoryPlanning] =\n    useState(true);\n  const [expectedEventCountPlanning, setExpectedEventCountPlanning] = useState<\n    number | null\n  >(null);\n\n  const { setPlanContent } = useConversationStore();\n\n  // Hook for reading conversation file\n  const { mutate: readConversationFile } = useReadConversationFile();\n\n  // Track planning-agent received events (still WS-driven).\n  const receivedEventCountRefPlanning = useRef(0);\n\n  // Track the latest PlanningFileEditorObservation for Plan.md during history replay\n  const latestPlanningFileEventRef = useRef<{\n    path: string;\n    conversationId: string;\n  } | null>(null);\n\n  const isPlanFilePath = (path: string | null): boolean =>\n    path?.toUpperCase().endsWith(\"PLAN.MD\") ?? false;\n\n  const handleNonErrorEvent = useCallback(() => {\n    // A normal event means connectivity recovered: clear a transient connection\n    // error, but keep sticky conversation errors (e.g. a wrong API key).\n    clearConnectionError();\n  }, [clearConnectionError]);\n\n  // Helper function to update metrics from stats event\n  const updateMetricsFromStats = useCallback(\n    (event: ConversationStateUpdateEventStats) => {\n      if (event.value.usage_to_metrics?.agent) {\n        const agentMetrics = event.value.usage_to_metrics.agent;\n        const metrics = {\n          cost: agentMetrics.accumulated_cost,\n          max_budget_per_task: agentMetrics.max_budget_per_task ?? null,\n          usage: agentMetrics.accumulated_token_usage\n            ? {\n                prompt_tokens:\n                  agentMetrics.accumulated_token_usage.prompt_tokens,\n                completion_tokens:\n                  agentMetrics.accumulated_token_usage.completion_tokens,\n                cache_read_tokens:\n                  agentMetrics.accumulated_token_usage.cache_read_tokens,\n                cache_write_tokens:\n                  agentMetrics.accumulated_token_usage.cache_write_tokens,\n                context_window:\n                  agentMetrics.accumulated_token_usage.context_window,\n                per_turn_token:\n                  agentMetrics.accumulated_token_usage.per_turn_token,\n              }\n            : null,\n        };\n        useMetricsStore.getState().setMetrics(metrics);\n      }\n    },\n    [],\n  );\n\n  // Initial REST history load: fetch the most recent events and seed the\n  // store. Older events are paginated in via `useLoadOlderEvents` when the\n  // user scrolls to the top of the chat. The WebSocket connection waits for\n  // this query so it can subscribe with `resend_mode='since'` and avoid\n  // re-streaming everything REST already returned.\n  const {\n    data: preloadedHistory,\n    isPending: isPreloadingHistory,\n    isError: isPreloadHistoryError,\n  } = useConversationHistory(conversationId);\n\n  const isLoadingHistoryMain = !!conversationId && isPreloadingHistory;\n\n  // Clear the (global, not conversation-scoped) event store when the active\n  // conversation changes, BEFORE the preloaded-history effect below re-seeds\n  // it. This MUST live here rather than in the route component: a parent's\n  // passive effect runs *after* this child's layout effects, so clearing from\n  // the route would wipe the freshly seeded history. On a conversation switch\n  // the history page is already cached, so `preloadedHistory` is available\n  // synchronously — without ordering the clear first, the user's already-echoed\n  // message gets seeded then immediately wiped, leaving only the `since`\n  // WebSocket resend (the agent's reply). Re-entering the same conversation is\n  // a no-op, so the store survives navigating away to Settings and back.\n  useLayoutEffect(() => {\n    const nextId = conversationId ?? null;\n    if (useEventStore.getState().loadedConversationId === nextId) {\n      return;\n    }\n    // Single atomic action: clears the previous conversation's events and\n    // records the new loaded id in one `set`, so no subscriber can observe a\n    // half-applied state (events gone but the old id still reported).\n    clearEventsForConversation(nextId);\n  }, [conversationId, clearEventsForConversation]);\n\n  useLayoutEffect(() => {\n    if (!preloadedHistory || preloadedHistory.events.length === 0) {\n      return;\n    }\n    addEvents(preloadedHistory.events);\n\n    // The first user message of a cloud start-task conversation is persisted\n    // server-side and reaches us via this REST preload, not over the WebSocket\n    // (which subscribes with resend_mode='since' after the latest preloaded\n    // timestamp). Consume any matching optimistic \"Sending…\" bubble here too —\n    // mirroring the WS handler — so it doesn't linger as a duplicate of the echo.\n    if (conversationId) {\n      for (const event of preloadedHistory.events) {\n        if (isUserMessageEvent(event)) {\n          consumeMatchingPendingMessage(\n            conversationId,\n            extractMessageEventText(event),\n          );\n        }\n      }\n    }\n  }, [\n    preloadedHistory,\n    addEvents,\n    conversationId,\n    consumeMatchingPendingMessage,\n  ]);\n\n  /**\n   * Timestamp of the latest event we already have from REST. Used as\n   * `after_timestamp` when opening the WebSocket so the server only resends\n   * events strictly after this point. `null` until the REST query settles\n   * (we hold the WS connection open until then to avoid an `all` resend).\n   */\n  const initialAfterTimestamp = useMemo<string | null>(() => {\n    if (isPreloadingHistory) return null;\n    const events = preloadedHistory?.events ?? [];\n    const latest = events[events.length - 1];\n    if (!latest || !(\"timestamp\" in latest) || !latest.timestamp) return null;\n    return latest.timestamp;\n  }, [preloadedHistory, isPreloadingHistory]);\n\n  // Build WebSocket URL from props.\n  //\n  // We deliberately wait for the initial REST history fetch to settle before\n  // opening the socket so the WS subscription can use `resend_mode='since'`\n  // with a meaningful `after_timestamp`. Without this gate, the WS would open\n  // immediately and either replay the entire conversation (when falling back\n  // to `resend_mode='all'`) or miss events that arrived between REST and WS.\n  const wsUrl = useMemo(() => {\n    if (!conversationId || !conversationUrl) {\n      return null;\n    }\n    // Don't connect while we're still fetching the initial history. If the\n    // REST query errored we fall through and connect with `resend_mode='all'`\n    // so the user still sees live events.\n    if (isPreloadingHistory && !isPreloadHistoryError) {\n      return null;\n    }\n    return buildWebSocketUrl(conversationId, conversationUrl);\n  }, [\n    conversationId,\n    conversationUrl,\n    isPreloadingHistory,\n    isPreloadHistoryError,\n  ]);\n\n  const planningAgentWsUrl = useMemo(() => {\n    if (!subConversations?.length) {\n      return null;\n    }\n\n    // Currently, there is only one sub-conversation and it uses the planning agent.\n    const planningAgentConversation = subConversations[0];\n\n    if (\n      !planningAgentConversation?.id ||\n      !planningAgentConversation.conversation_url\n    ) {\n      return null;\n    }\n\n    return buildWebSocketUrl(\n      planningAgentConversation.id,\n      planningAgentConversation.conversation_url,\n    );\n  }, [subConversations]);\n\n  // Merged connection state - reflects combined status of both connections\n  const connectionState = useMemo<WebSocketConnectionState>(() => {\n    // If planning agent connection doesn't exist, use main connection state\n    if (!planningAgentWsUrl) {\n      return mainConnectionState;\n    }\n\n    // If either is connecting, merged state is connecting\n    if (\n      mainConnectionState === \"CONNECTING\" ||\n      planningConnectionState === \"CONNECTING\"\n    ) {\n      return \"CONNECTING\";\n    }\n\n    // If both are open, merged state is open\n    if (mainConnectionState === \"OPEN\" && planningConnectionState === \"OPEN\") {\n      return \"OPEN\";\n    }\n\n    // If both are closed, merged state is closed\n    if (\n      mainConnectionState === \"CLOSED\" &&\n      planningConnectionState === \"CLOSED\"\n    ) {\n      return \"CLOSED\";\n    }\n\n    // If either is closing, merged state is closing\n    if (\n      mainConnectionState === \"CLOSING\" ||\n      planningConnectionState === \"CLOSING\"\n    ) {\n      return \"CLOSING\";\n    }\n\n    // Default to closed if states don't match expected patterns\n    return \"CLOSED\";\n  }, [mainConnectionState, planningConnectionState, planningAgentWsUrl]);\n\n  useEffect(() => {\n    if (\n      expectedEventCountPlanning !== null &&\n      receivedEventCountRefPlanning.current >= expectedEventCountPlanning &&\n      isLoadingHistoryPlanning\n    ) {\n      setIsLoadingHistoryPlanning(false);\n    }\n  }, [\n    expectedEventCountPlanning,\n    isLoadingHistoryPlanning,\n    receivedEventCountRefPlanning,\n  ]);\n\n  // Call API once after history loading completes if we tracked any PlanningFileEditorObservation events\n  useEffect(() => {\n    if (!isLoadingHistoryPlanning && latestPlanningFileEventRef.current) {\n      const { path, conversationId: currentPlanningConversationId } =\n        latestPlanningFileEventRef.current;\n\n      readConversationFile(\n        {\n          conversationId: currentPlanningConversationId,\n          filePath: path,\n        },\n        {\n          onSuccess: (fileContent) => {\n            setPlanContent(fileContent);\n          },\n          onError: (error) => {\n            console.warn(\"Failed to read conversation file:\", error);\n          },\n        },\n      );\n\n      // Clear the ref after calling the API\n      latestPlanningFileEventRef.current = null;\n    }\n  }, [isLoadingHistoryPlanning, readConversationFile, setPlanContent]);\n\n  useEffect(() => {\n    hasConnectedRefMain.current = false;\n    setIsLoadingHistoryPlanning(!!subConversationIds?.length);\n    setExpectedEventCountPlanning(null);\n    receivedEventCountRefPlanning.current = 0;\n    // Reset the tracked event ref when sub-conversations change\n    latestPlanningFileEventRef.current = null;\n  }, [subConversationIds]);\n\n  // Reset hasConnected flags when the conversation changes.\n  useEffect(() => {\n    hasConnectedRefMain.current = false;\n    hasConnectedRefPlanning.current = false;\n    // Reset the tracked event ref when conversation changes\n    latestPlanningFileEventRef.current = null;\n  }, [conversationId]);\n\n  // Merged loading history state - true if either connection is still loading\n  const isLoadingHistory = useMemo(\n    () => isLoadingHistoryMain || isLoadingHistoryPlanning,\n    [isLoadingHistoryMain, isLoadingHistoryPlanning],\n  );\n\n  // Separate message handlers for each connection\n  const handleMainMessage = useCallback(\n    (messageEvent: MessageEvent) => {\n      try {\n        const event = JSON.parse(messageEvent.data);\n\n        // History loading for the main conversation is REST-driven now;\n        // every WS message is a new event we add to the store.\n\n        // Use type guard to validate v1 event structure\n        if (isAgentServerEvent(event)) {\n          const isDuplicateEvent = useEventStore\n            .getState()\n            .eventIds.has(event.id);\n          const switchLLMObservation =\n            !isDuplicateEvent && isSwitchLLMObservationEvent(event)\n              ? event\n              : null;\n          addEvent(event);\n\n          // Handle displayable error events - show error banner\n          // AgentErrorEvent errors are displayed inline in the chat, not as banners\n          if (isDisplayableErrorEvent(event)) {\n            const errorEvent = event as\n              | ConversationErrorEvent\n              | ServerErrorEvent;\n            trackError({\n              message: errorEvent.detail,\n              source: \"conversation\",\n              metadata: {\n                eventId: errorEvent.id,\n                errorCode: errorEvent.code,\n              },\n              posthog,\n            });\n            setErrorMessage(errorEvent.detail);\n          } else {\n            handleNonErrorEvent();\n          }\n\n          // LLM errors render inline in the chat (see ErrorEventMessage); track\n          // them for analytics but keep them out of the banner above the chat box.\n          if (isAgentErrorEvent(event)) {\n            trackError({\n              message: event.error,\n              source: \"agent\",\n              metadata: {\n                eventId: event.id,\n                toolName: event.tool_name,\n                toolCallId: event.tool_call_id,\n              },\n              posthog,\n            });\n          }\n\n          // Clear optimistic user message when a user message is confirmed.\n          // We match by the echoed text content (with FIFO fallback inside the\n          // store), so an echo for \"second\" pops \"second\" — not whichever\n          // pending entry happens to be oldest — protecting against any\n          // out-of-order delivery between conversations or sub-agents.\n          if (isUserMessageEvent(event)) {\n            if (conversationId) {\n              consumeMatchingPendingMessage(\n                conversationId,\n                extractMessageEventText(event),\n              );\n              // Clear draft from localStorage - message was successfully delivered\n              setConversationState(conversationId, { draftMessage: null });\n            }\n          }\n\n          // Handle cache invalidation for ActionEvent\n          if (isActionEvent(event)) {\n            const currentConversationId =\n              conversationId || \"test-conversation-id\"; // TODO: Get from context\n            handleActionEventCacheInvalidation(\n              event,\n              currentConversationId,\n              queryClient,\n            );\n          }\n\n          // Handle conversation state updates\n          // TODO: Tests\n          if (isConversationStateUpdateEvent(event)) {\n            if (isFullStateConversationStateUpdateEvent(event)) {\n              setExecutionStatus(event.value.execution_status);\n            }\n            if (isAgentStatusConversationStateUpdateEvent(event)) {\n              setExecutionStatus(event.value);\n            }\n            if (isStatsConversationStateUpdateEvent(event)) {\n              updateMetricsFromStats(event);\n            }\n          }\n\n          // Handle ExecuteBashAction events - add command as input to terminal\n          if (isExecuteBashActionEvent(event)) {\n            appendInput(event.action.command);\n          }\n\n          // Handle ExecuteBashObservation events - add output to terminal\n          if (isExecuteBashObservationEvent(event)) {\n            // Extract text content from the observation content array\n            const textContent = event.observation.content\n              .filter((c) => c.type === \"text\")\n              .map((c) => c.text)\n              .join(\"\\n\");\n            appendOutput(textContent);\n          }\n\n          // Handle BrowserObservation events - update browser store with screenshot\n          if (isBrowserObservationEvent(event)) {\n            const { screenshot_data: screenshotData } = event.observation;\n            if (screenshotData) {\n              const screenshotSrc = screenshotData.startsWith(\"data:\")\n                ? screenshotData\n                : `data:image/png;base64,${screenshotData}`;\n              useBrowserStore.getState().setScreenshotSrc(screenshotSrc);\n            }\n          }\n\n          // Handle BrowserNavigateAction events - update browser store with URL\n          if (isBrowserNavigateActionEvent(event)) {\n            useBrowserStore.getState().setUrl(event.action.url);\n          }\n\n          if (\n            conversationId &&\n            switchLLMObservation &&\n            !switchLLMObservation.observation.is_error\n          ) {\n            recordModelSwitchMessage(\n              conversationId,\n              switchLLMObservation.observation.profile_name,\n            );\n\n            if (switchLLMObservation.observation.active_model) {\n              updateConversationLlmModelInCache(\n                queryClient,\n                conversationId,\n                switchLLMObservation.observation.active_model,\n              );\n            }\n\n            invalidateConversationQueries(queryClient, conversationId);\n          }\n\n          // Handle canvas_ui custom-tool ActionEvents - drive the frontend\n          // (navigate to a file, switch tabs, show a preview). The tool\n          // executes server-side as a no-op; the actual UI change happens\n          // here on the client.\n          if (isCanvasUIActionEvent(event)) {\n            handleCanvasUIAction(event.action);\n          }\n        }\n      } catch (error) {\n        console.warn(\"Failed to parse WebSocket message as JSON:\", error);\n      }\n    },\n    [\n      addEvent,\n      setErrorMessage,\n      consumeMatchingPendingMessage,\n      queryClient,\n      conversationId,\n      setExecutionStatus,\n      appendInput,\n      appendOutput,\n      updateMetricsFromStats,\n      handleNonErrorEvent,\n      posthog,\n    ],\n  );\n\n  const handlePlanningMessage = useCallback(\n    (messageEvent: MessageEvent) => {\n      try {\n        const event = JSON.parse(messageEvent.data);\n\n        // Track received events for history loading (count ALL events from WebSocket)\n        // Always count when loading, even if we don't have the expected count yet\n        if (isLoadingHistoryPlanning) {\n          receivedEventCountRefPlanning.current += 1;\n\n          if (\n            expectedEventCountPlanning !== null &&\n            receivedEventCountRefPlanning.current >= expectedEventCountPlanning\n          ) {\n            setIsLoadingHistoryPlanning(false);\n          }\n        }\n\n        // Use type guard to validate v1 event structure\n        if (isAgentServerEvent(event)) {\n          // Mark this event as coming from the planning agent\n          const eventWithPlanningFlag = {\n            ...event,\n            isFromPlanningAgent: true,\n          };\n          addEvent(eventWithPlanningFlag);\n\n          // Handle displayable error events - show error banner\n          // AgentErrorEvent errors are displayed inline in the chat, not as banners\n          if (isDisplayableErrorEvent(event)) {\n            const errorEvent = event as\n              | ConversationErrorEvent\n              | ServerErrorEvent;\n            trackError({\n              message: errorEvent.detail,\n              source: \"planning_conversation\",\n              metadata: {\n                eventId: errorEvent.id,\n                errorCode: errorEvent.code,\n              },\n              posthog,\n            });\n            setErrorMessage(errorEvent.detail);\n          } else {\n            handleNonErrorEvent();\n          }\n\n          // LLM errors render inline in the chat (see ErrorEventMessage); track\n          // them for analytics but keep them out of the banner above the chat box.\n          if (isAgentErrorEvent(event)) {\n            trackError({\n              message: event.error,\n              source: \"planning_agent\",\n              metadata: {\n                eventId: event.id,\n                toolName: event.tool_name,\n                toolCallId: event.tool_call_id,\n              },\n              posthog,\n            });\n          }\n\n          // Clear optimistic user message when a user message is confirmed.\n          // Always scope to the main `conversationId` (where the user types)\n          // and match on the echoed content so the planning sub-agent's own\n          // events can never consume a main-conversation pending entry.\n          if (isUserMessageEvent(event)) {\n            if (conversationId) {\n              consumeMatchingPendingMessage(\n                conversationId,\n                extractMessageEventText(event),\n              );\n              setConversationState(conversationId, { draftMessage: null });\n            }\n          }\n\n          // Handle cache invalidation for ActionEvent\n          if (isActionEvent(event)) {\n            const planningAgentConversation = subConversations?.[0];\n            const currentConversationId =\n              planningAgentConversation?.id || \"test-conversation-id\"; // TODO: Get from context\n            handleActionEventCacheInvalidation(\n              event,\n              currentConversationId,\n              queryClient,\n            );\n          }\n\n          // Handle conversation state updates\n          // TODO: Tests\n          if (isConversationStateUpdateEvent(event)) {\n            if (isFullStateConversationStateUpdateEvent(event)) {\n              setExecutionStatus(event.value.execution_status);\n            }\n            if (isAgentStatusConversationStateUpdateEvent(event)) {\n              setExecutionStatus(event.value);\n            }\n            if (isStatsConversationStateUpdateEvent(event)) {\n              updateMetricsFromStats(event);\n            }\n          }\n\n          // Handle ExecuteBashAction events - add command as input to terminal\n          if (isExecuteBashActionEvent(event)) {\n            appendInput(event.action.command);\n          }\n\n          // Handle ExecuteBashObservation events - add output to terminal\n          if (isExecuteBashObservationEvent(event)) {\n            // Extract text content from the observation content array\n            const textContent = event.observation.content\n              .filter((c) => c.type === \"text\")\n              .map((c) => c.text)\n              .join(\"\\n\");\n            appendOutput(textContent);\n          }\n\n          // Handle PlanningFileEditorObservation - only update plan for Plan.md\n          if (isPlanningFileEditorObservationEvent(event)) {\n            const { path } = event.observation;\n            if (isPlanFilePath(path)) {\n              const planningAgentConversation = subConversations?.[0];\n              const planningConversationId = planningAgentConversation?.id;\n\n              if (planningConversationId && path) {\n                if (isLoadingHistoryPlanning) {\n                  latestPlanningFileEventRef.current = {\n                    path,\n                    conversationId: planningConversationId,\n                  };\n                } else {\n                  readConversationFile(\n                    {\n                      conversationId: planningConversationId,\n                      filePath: path,\n                    },\n                    {\n                      onSuccess: (fileContent) => {\n                        setPlanContent(fileContent);\n                      },\n                      onError: (error) => {\n                        console.warn(\n                          \"Failed to read conversation file:\",\n                          error,\n                        );\n                      },\n                    },\n                  );\n                }\n              }\n            }\n          }\n        }\n      } catch (error) {\n        console.warn(\"Failed to parse WebSocket message as JSON:\", error);\n      }\n    },\n    [\n      addEvent,\n      isLoadingHistoryPlanning,\n      expectedEventCountPlanning,\n      setErrorMessage,\n      consumeMatchingPendingMessage,\n      queryClient,\n      subConversations,\n      conversationId,\n      setExecutionStatus,\n      appendInput,\n      appendOutput,\n      readConversationFile,\n      setPlanContent,\n      updateMetricsFromStats,\n      handleNonErrorEvent,\n      posthog,\n    ],\n  );\n\n  // Separate WebSocket options for main connection\n  const mainWebsocketOptions: WebSocketHookOptions = useMemo(() => {\n    // History was already loaded over REST (`useConversationHistory`).\n    // Subscribe with `resend_mode='since'` so the server only resends events\n    // strictly after the latest one we already have. If REST returned no\n    // events at all (brand-new conversation), fall back to `'all'` so any\n    // events that may have been written between the REST call and the WS\n    // handshake still show up. Dedup in the event store handles overlap.\n    const queryParams: Record<string, string | boolean> = initialAfterTimestamp\n      ? { resend_mode: \"since\", after_timestamp: initialAfterTimestamp }\n      : { resend_mode: \"all\" };\n\n    // Add session_api_key if available\n    if (sessionApiKey) {\n      queryParams.session_api_key = sessionApiKey;\n    }\n\n    return {\n      queryParams,\n      reconnect: { enabled: true },\n      onOpen: () => {\n        setMainConnectionState(\"OPEN\");\n        hasConnectedRefMain.current = true; // Mark that we've successfully connected\n        clearConnectionError(); // Clear a previous connection error; keep sticky conversation errors\n      },\n      onClose: () => {\n        setMainConnectionState(\"CLOSED\");\n      },\n      onError: () => {\n        setMainConnectionState(\"CLOSED\");\n        // Only show error message if we've previously connected successfully\n        if (hasConnectedRefMain.current) {\n          setErrorMessage(SERVER_CONNECTION_ERROR_MESSAGE, \"connection\");\n        }\n      },\n      onMessage: handleMainMessage,\n    };\n  }, [\n    handleMainMessage,\n    setErrorMessage,\n    clearConnectionError,\n    sessionApiKey,\n    initialAfterTimestamp,\n  ]);\n\n  // Separate WebSocket options for planning agent connection\n  const planningWebsocketOptions: WebSocketHookOptions = useMemo(() => {\n    const queryParams: Record<string, string | boolean> = {\n      resend_all: true,\n    };\n\n    // Add session_api_key if available\n    if (sessionApiKey) {\n      queryParams.session_api_key = sessionApiKey;\n    }\n\n    const planningAgentConversation = subConversations?.[0];\n\n    return {\n      queryParams,\n      reconnect: { enabled: true },\n      onOpen: async () => {\n        setPlanningConnectionState(\"OPEN\");\n        hasConnectedRefPlanning.current = true; // Mark that we've successfully connected\n        clearConnectionError(); // Clear a previous connection error; keep sticky conversation errors\n\n        // Fetch expected event count for history loading detection\n        if (\n          planningAgentConversation?.id &&\n          planningAgentConversation.conversation_url\n        ) {\n          try {\n            const count = await EventService.getEventCount(\n              planningAgentConversation.id,\n              planningAgentConversation.conversation_url,\n              planningAgentConversation.session_api_key,\n            );\n            setExpectedEventCountPlanning(count);\n\n            // If no events expected, mark as loaded immediately\n            if (count === 0) {\n              setIsLoadingHistoryPlanning(false);\n            }\n          } catch (error) {\n            // Fall back to marking as loaded to avoid infinite loading state\n            setIsLoadingHistoryPlanning(false);\n          }\n        }\n      },\n      onClose: () => {\n        setPlanningConnectionState(\"CLOSED\");\n      },\n      onError: () => {\n        setPlanningConnectionState(\"CLOSED\");\n        // Only show error message if we've previously connected successfully\n        if (hasConnectedRefPlanning.current) {\n          setErrorMessage(SERVER_CONNECTION_ERROR_MESSAGE, \"connection\");\n        }\n      },\n      onMessage: handlePlanningMessage,\n    };\n  }, [\n    handlePlanningMessage,\n    setErrorMessage,\n    clearConnectionError,\n    sessionApiKey,\n    subConversations,\n  ]);\n\n  // Only attempt WebSocket connection when we have a valid URL\n  // This prevents connection attempts during task polling phase\n  const websocketUrl = wsUrl;\n  const { socket: mainSocket, reconnect: reconnectMain } = useWebSocket(\n    websocketUrl || \"\",\n    mainWebsocketOptions,\n  );\n\n  const { socket: planningAgentSocket, reconnect: reconnectPlanning } =\n    useWebSocket(planningAgentWsUrl || \"\", planningWebsocketOptions);\n\n  const reconnect = useCallback(() => {\n    removeErrorMessage();\n    const currentMode = useConversationStore.getState().conversationMode;\n    if (currentMode === \"plan\" && planningAgentWsUrl) {\n      reconnectPlanning();\n      return;\n    }\n    reconnectMain();\n  }, [\n    planningAgentWsUrl,\n    reconnectMain,\n    reconnectPlanning,\n    removeErrorMessage,\n  ]);\n\n  // V1 send message function via WebSocket\n  // Falls back to REST API queue when WebSocket is not connected\n  const sendMessage = useCallback(\n    async (message: SendMessageRequest): Promise<SendMessageResult> => {\n      const currentMode = useConversationStore.getState().conversationMode;\n      const currentSocket =\n        currentMode === \"plan\" ? planningAgentSocket : mainSocket;\n\n      if (currentSocket?.readyState !== WebSocket.OPEN) {\n        // WebSocket not connected - queue message via REST API\n        // Message will be delivered automatically when conversation becomes ready\n        if (!conversationId) {\n          const error = new Error(\"No conversation ID available\");\n          setErrorMessage(error.message);\n          throw error;\n        }\n\n        try {\n          await new ConversationClient(getAgentServerClientOptions()).sendEvent(\n            conversationId,\n            {\n              role: \"user\",\n              content: message.content,\n            },\n            { run: true },\n          );\n          // Message queued successfully - it will be delivered when ready\n          // Return queued: true so caller knows not to show optimistic UI\n          return { queued: true };\n        } catch (error) {\n          const errorMessage =\n            error instanceof Error\n              ? error.message\n              : \"Failed to queue message for delivery\";\n          setErrorMessage(errorMessage);\n          throw error;\n        }\n      }\n\n      try {\n        // Send message through WebSocket as JSON with run: true so the\n        // agent loop starts automatically in async mode.\n        currentSocket.send(JSON.stringify({ ...message, run: true }));\n        return { queued: false };\n      } catch (error) {\n        const errorMessage =\n          error instanceof Error ? error.message : \"Failed to send message\";\n        setErrorMessage(errorMessage);\n        throw error;\n      }\n    },\n    [mainSocket, planningAgentSocket, setErrorMessage, conversationId],\n  );\n\n  // Track main socket state changes\n  useEffect(() => {\n    // Only process socket updates if we have a valid URL and socket\n    if (mainSocket && wsUrl) {\n      // Update state based on socket readyState\n      const updateState = () => {\n        switch (mainSocket.readyState) {\n          case WebSocket.CONNECTING:\n            setMainConnectionState(\"CONNECTING\");\n            break;\n          case WebSocket.OPEN:\n            setMainConnectionState(\"OPEN\");\n            break;\n          case WebSocket.CLOSING:\n            setMainConnectionState(\"CLOSING\");\n            break;\n          case WebSocket.CLOSED:\n            setMainConnectionState(\"CLOSED\");\n            break;\n          default:\n            setMainConnectionState(\"CLOSED\");\n            break;\n        }\n      };\n\n      updateState();\n    }\n  }, [mainSocket, wsUrl]);\n\n  // Track planning agent socket state changes\n  useEffect(() => {\n    // Only process socket updates if we have a valid URL and socket\n    if (planningAgentSocket && planningAgentWsUrl) {\n      // Update state based on socket readyState\n      const updateState = () => {\n        switch (planningAgentSocket.readyState) {\n          case WebSocket.CONNECTING:\n            setPlanningConnectionState(\"CONNECTING\");\n            break;\n          case WebSocket.OPEN:\n            setPlanningConnectionState(\"OPEN\");\n            break;\n          case WebSocket.CLOSING:\n            setPlanningConnectionState(\"CLOSING\");\n            break;\n          case WebSocket.CLOSED:\n            setPlanningConnectionState(\"CLOSED\");\n            break;\n          default:\n            setPlanningConnectionState(\"CLOSED\");\n            break;\n        }\n      };\n\n      updateState();\n    }\n  }, [planningAgentSocket, planningAgentWsUrl]);\n\n  const contextValue = useMemo(\n    () => ({ connectionState, sendMessage, isLoadingHistory, reconnect }),\n    [connectionState, sendMessage, isLoadingHistory, reconnect],\n  );\n\n  return (\n    <ConversationWebSocketContext.Provider value={contextValue}>\n      {children}\n    </ConversationWebSocketContext.Provider>\n  );\n}\n\nexport const useConversationWebSocket =\n  (): ConversationWebSocketContextType | null => {\n    const context = useContext(ConversationWebSocketContext);\n    // Return null instead of throwing when not in provider\n    // This allows the hook to be called conditionally based on conversation version\n    return context || null;\n  };\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;AAmFA,IAAM,IAA+B,GAEnC,KAAA,EAAU;AASZ,SAAS,EACP,GACQ;AACR,QAAO,EAAM,YAAY,QACtB,QACE,MAAiD,EAAK,SAAS,OACjE,CACA,KAAK,MAAS,EAAK,KAAK,CACxB,KAAK,GAAG;;AAGb,SAAgB,EAA8B,EAC5C,cACA,mBACA,oBACA,kBACA,qBACA,yBAQC;CAED,IAAM,CAAC,GAAqB,KAC1B,EAAmC,aAAa,EAC5C,CAAC,GAAyB,KAC9B,EAAmC,aAAa,EAI5C,IAAsB,GAAM,OAAO,GAAM,EACzC,IAA0B,GAAM,OAAO,GAAM,EAE7C,IAAU,IAAY,EACtB,IAAc,IAAgB,EAC9B,IAAW,GAAe,MAAU,EAAM,SAAS,EACnD,KAAY,GAAe,MAAU,EAAM,UAAU,EACrD,KAA6B,GAChC,MAAU,EAAM,2BAClB,EACK,EAAE,oBAAiB,wBAAoB,4BAC3C,IAAsB,EAClB,IAAgC,IACnC,MAAU,EAAM,8BAClB,EACK,EAAE,0BAAuB,IAA2B,EACpD,EAAE,gBAAa,oBAAiB,GAAiB,EAOjD,CAAC,GAA0B,KAC/B,EAAS,GAAK,EACV,CAAC,GAA4B,MAAiC,EAElE,KAAK,EAED,EAAE,sBAAmB,GAAsB,EAG3C,EAAE,QAAQ,MAAyB,IAAyB,EAG5D,IAAgC,GAAO,EAAE,EAGzC,IAA6B,GAGzB,KAAK,EAET,MAAkB,MACtB,GAAM,aAAa,CAAC,SAAS,UAAU,IAAI,IAEvC,IAAsB,QAAkB;AAG5C,KAAsB;IACrB,CAAC,EAAqB,CAAC,EAGpB,IAAyB,GAC5B,MAA6C;AAC5C,MAAI,EAAM,MAAM,kBAAkB,OAAO;GACvC,IAAM,IAAe,EAAM,MAAM,iBAAiB,OAC5C,IAAU;IACd,MAAM,EAAa;IACnB,qBAAqB,EAAa,uBAAuB;IACzD,OAAO,EAAa,0BAChB;KACE,eACE,EAAa,wBAAwB;KACvC,mBACE,EAAa,wBAAwB;KACvC,mBACE,EAAa,wBAAwB;KACvC,oBACE,EAAa,wBAAwB;KACvC,gBACE,EAAa,wBAAwB;KACvC,gBACE,EAAa,wBAAwB;KACxC,GACD;IACL;AACD,MAAgB,UAAU,CAAC,WAAW,EAAQ;;IAGlD,EAAE,CACH,EAOK,EACJ,MAAM,GACN,WAAW,GACX,SAAS,OACP,GAAuB,EAAe,EAEpC,KAAuB,CAAC,CAAC,KAAkB;AAuBjD,CAXA,SAAsB;EACpB,IAAM,IAAS,KAAkB;AAC7B,IAAc,UAAU,CAAC,yBAAyB,KAMtD,GAA2B,EAAO;IACjC,CAAC,GAAgB,GAA2B,CAAC,EAEhD,SAAsB;AAChB,SAAC,KAAoB,EAAiB,OAAO,WAAW,OAG5D,GAAU,EAAiB,OAAO,EAO9B,SACG,IAAM,KAAS,EAAiB,OACnC,CAAI,EAAmB,EAAM,IAC3B,EACE,GACA,EAAwB,EAAM,CAC/B;IAIN;EACD;EACA;EACA;EACA;EACD,CAAC;CAQF,IAAM,IAAwB,QAA6B;AACzD,MAAI,EAAqB,QAAO;EAChC,IAAM,IAAS,GAAkB,UAAU,EAAE,EACvC,IAAS,EAAO,EAAO,SAAS;AAEtC,SADI,CAAC,KAAU,EAAE,eAAe,MAAW,CAAC,EAAO,YAAkB,OAC9D,EAAO;IACb,CAAC,GAAkB,EAAoB,CAAC,EASrC,IAAQ,QACR,CAAC,KAAkB,CAAC,KAMpB,KAAuB,CAAC,KACnB,OAEF,GAAkB,GAAgB,EAAgB,EACxD;EACD;EACA;EACA;EACA;EACD,CAAC,EAEI,IAAqB,QAAc;AACvC,MAAI,CAAC,GAAkB,OACrB,QAAO;EAIT,IAAM,IAA4B,EAAiB;AASnD,SANE,CAAC,GAA2B,MAC5B,CAAC,EAA0B,mBAEpB,OAGF,GACL,EAA0B,IAC1B,EAA0B,iBAC3B;IACA,CAAC,EAAiB,CAAC,EAGhB,KAAkB,QAEjB,IAMH,MAAwB,gBACxB,MAA4B,eAErB,eAIL,MAAwB,UAAU,MAA4B,SACzD,SAKP,MAAwB,YACxB,MAA4B,WAErB,WAKP,MAAwB,aACxB,MAA4B,YAErB,YAIF,WAjCE,GAkCR;EAAC;EAAqB;EAAyB;EAAmB,CAAC;AAoDtE,CAlDA,QAAgB;AACd,EACE,MAA+B,QAC/B,EAA8B,WAAW,KACzC,KAEA,EAA4B,GAAM;IAEnC;EACD;EACA;EACA;EACD,CAAC,EAGF,QAAgB;AACd,MAAI,CAAC,KAA4B,EAA2B,SAAS;GACnE,IAAM,EAAE,SAAM,gBAAgB,MAC5B,EAA2B;AAkB7B,GAhBA,EACE;IACE,gBAAgB;IAChB,UAAU;IACX,EACD;IACE,YAAY,MAAgB;AAC1B,OAAe,EAAY;;IAE7B,UAAU,MAAU;AAClB,aAAQ,KAAK,qCAAqC,EAAM;;IAE3D,CACF,EAGD,EAA2B,UAAU;;IAEtC;EAAC;EAA0B;EAAsB;EAAe,CAAC,EAEpE,QAAgB;AAMd,EALA,EAAoB,UAAU,IAC9B,EAA4B,CAAC,CAAC,GAAoB,OAAO,EACzD,GAA8B,KAAK,EACnC,EAA8B,UAAU,GAExC,EAA2B,UAAU;IACpC,CAAC,EAAmB,CAAC,EAGxB,QAAgB;AAId,EAHA,EAAoB,UAAU,IAC9B,EAAwB,UAAU,IAElC,EAA2B,UAAU;IACpC,CAAC,EAAe,CAAC;CAGpB,IAAM,KAAmB,QACjB,MAAwB,GAC9B,CAAC,IAAsB,EAAyB,CACjD,EAGK,IAAoB,GACvB,MAA+B;AAC9B,MAAI;GACF,IAAM,IAAQ,KAAK,MAAM,EAAa,KAAK;AAM3C,OAAI,GAAmB,EAAM,EAAE;IAI7B,IAAM,IACJ,CAJuB,EACtB,UAAU,CACV,SAAS,IAAI,EAAM,GAEnB,IAAoB,GAA4B,EAAM,GACnD,IACA;AAKN,QAJA,EAAS,EAAM,EAIX,EAAwB,EAAM,EAAE;KAClC,IAAM,IAAa;AAYnB,KATA,EAAW;MACT,SAAS,EAAW;MACpB,QAAQ;MACR,UAAU;OACR,SAAS,EAAW;OACpB,WAAW,EAAW;OACvB;MACD;MACD,CAAC,EACF,EAAgB,EAAW,OAAO;UAElC,IAAqB;AA2EvB,QAtEI,GAAkB,EAAM,IAC1B,EAAW;KACT,SAAS,EAAM;KACf,QAAQ;KACR,UAAU;MACR,SAAS,EAAM;MACf,UAAU,EAAM;MAChB,YAAY,EAAM;MACnB;KACD;KACD,CAAC,EAQA,EAAmB,EAAM,IACvB,MACF,EACE,GACA,EAAwB,EAAM,CAC/B,EAED,EAAqB,GAAgB,EAAE,cAAc,MAAM,CAAC,GAK5D,GAAc,EAAM,IAGtB,GACE,GAFA,KAAkB,wBAIlB,EACD,EAKC,EAA+B,EAAM,KACnC,GAAwC,EAAM,IAChD,EAAmB,EAAM,MAAM,iBAAiB,EAE9C,GAA0C,EAAM,IAClD,EAAmB,EAAM,MAAM,EAE7B,GAAoC,EAAM,IAC5C,EAAuB,EAAM,GAK7B,EAAyB,EAAM,IACjC,EAAY,EAAM,OAAO,QAAQ,EAI/B,EAA8B,EAAM,IAMtC,EAJoB,EAAM,YAAY,QACnC,QAAQ,MAAM,EAAE,SAAS,OAAO,CAChC,KAAK,MAAM,EAAE,KAAK,CAClB,KAAK,KACK,CAAY,EAIvB,GAA0B,EAAM,EAAE;KACpC,IAAM,EAAE,iBAAiB,MAAmB,EAAM;AAClD,SAAI,GAAgB;MAClB,IAAM,IAAgB,EAAe,WAAW,QAAQ,GACpD,IACA,yBAAyB;AAC7B,QAAgB,UAAU,CAAC,iBAAiB,EAAc;;;AAkC9D,IA7BI,GAA6B,EAAM,IACrC,EAAgB,UAAU,CAAC,OAAO,EAAM,OAAO,IAAI,EAInD,KACA,KACA,CAAC,EAAqB,YAAY,aAElC,GACE,GACA,EAAqB,YAAY,aAClC,EAEG,EAAqB,YAAY,gBACnC,GACE,GACA,GACA,EAAqB,YAAY,aAClC,EAGH,GAA8B,GAAa,EAAe,GAOxD,GAAsB,EAAM,IAC9B,GAAqB,EAAM,OAAO;;WAG/B,GAAO;AACd,WAAQ,KAAK,8CAA8C,EAAM;;IAGrE;EACE;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACD,CACF,EAEK,KAAwB,GAC3B,MAA+B;AAC9B,MAAI;GACF,IAAM,IAAQ,KAAK,MAAM,EAAa,KAAK;AAgB3C,OAZI,MACF,EAA8B,WAAW,GAGvC,MAA+B,QAC/B,EAA8B,WAAW,KAEzC,EAA4B,GAAM,GAKlC,GAAmB,EAAM,EAAE;AAU7B,QAJA,EAAS;KAHP,GAAG;KACH,qBAAqB;KAEd,CAAsB,EAI3B,EAAwB,EAAM,EAAE;KAClC,IAAM,IAAa;AAYnB,KATA,EAAW;MACT,SAAS,EAAW;MACpB,QAAQ;MACR,UAAU;OACR,SAAS,EAAW;OACpB,WAAW,EAAW;OACvB;MACD;MACD,CAAC,EACF,EAAgB,EAAW,OAAO;UAElC,IAAqB;AA0EvB,QArEI,GAAkB,EAAM,IAC1B,EAAW;KACT,SAAS,EAAM;KACf,QAAQ;KACR,UAAU;MACR,SAAS,EAAM;MACf,UAAU,EAAM;MAChB,YAAY,EAAM;MACnB;KACD;KACD,CAAC,EAOA,EAAmB,EAAM,IACvB,MACF,EACE,GACA,EAAwB,EAAM,CAC/B,EACD,EAAqB,GAAgB,EAAE,cAAc,MAAM,CAAC,GAK5D,GAAc,EAAM,IAItB,GACE,GAJgC,IAAmB,IAExB,MAAM,wBAIjC,EACD,EAKC,EAA+B,EAAM,KACnC,GAAwC,EAAM,IAChD,EAAmB,EAAM,MAAM,iBAAiB,EAE9C,GAA0C,EAAM,IAClD,EAAmB,EAAM,MAAM,EAE7B,GAAoC,EAAM,IAC5C,EAAuB,EAAM,GAK7B,EAAyB,EAAM,IACjC,EAAY,EAAM,OAAO,QAAQ,EAI/B,EAA8B,EAAM,IAMtC,EAJoB,EAAM,YAAY,QACnC,QAAQ,MAAM,EAAE,SAAS,OAAO,CAChC,KAAK,MAAM,EAAE,KAAK,CAClB,KAAK,KACK,CAAY,EAIvB,GAAqC,EAAM,EAAE;KAC/C,IAAM,EAAE,YAAS,EAAM;AACvB,SAAI,GAAe,EAAK,EAAE;MAExB,IAAM,IAD4B,IAAmB,IACK;AAE1D,MAAI,KAA0B,MACxB,IACF,EAA2B,UAAU;OACnC;OACA,gBAAgB;OACjB,GAED,EACE;OACE,gBAAgB;OAChB,UAAU;OACX,EACD;OACE,YAAY,MAAgB;AAC1B,UAAe,EAAY;;OAE7B,UAAU,MAAU;AAClB,gBAAQ,KACN,qCACA,EACD;;OAEJ,CACF;;;;WAMJ,GAAO;AACd,WAAQ,KAAK,8CAA8C,EAAM;;IAGrE;EACE;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACD,CACF,EAGK,KAA6C,QAAc;EAO/D,IAAM,IAAgD,IAClD;GAAE,aAAa;GAAS,iBAAiB;GAAuB,GAChE,EAAE,aAAa,OAAO;AAO1B,SAJI,MACF,EAAY,kBAAkB,IAGzB;GACL;GACA,WAAW,EAAE,SAAS,IAAM;GAC5B,cAAc;AAGZ,IAFA,EAAuB,OAAO,EAC9B,EAAoB,UAAU,IAC9B,GAAsB;;GAExB,eAAe;AACb,MAAuB,SAAS;;GAElC,eAAe;AAGb,IAFA,EAAuB,SAAS,EAE5B,EAAoB,WACtB,EAAgB,IAAiC,aAAa;;GAGlE,WAAW;GACZ;IACA;EACD;EACA;EACA;EACA;EACA;EACD,CAAC,EAGI,KAAiD,QAAc;EACnE,IAAM,IAAgD,EACpD,YAAY,IACb;AAGD,EAAI,MACF,EAAY,kBAAkB;EAGhC,IAAM,IAA4B,IAAmB;AAErD,SAAO;GACL;GACA,WAAW,EAAE,SAAS,IAAM;GAC5B,QAAQ,YAAY;AAMlB,QALA,EAA2B,OAAO,EAClC,EAAwB,UAAU,IAClC,GAAsB,EAIpB,GAA2B,MAC3B,EAA0B,iBAE1B,KAAI;KACF,IAAM,IAAQ,MAAM,GAAa,cAC/B,EAA0B,IAC1B,EAA0B,kBAC1B,EAA0B,gBAC3B;AAID,KAHA,GAA8B,EAAM,EAGhC,MAAU,KACZ,EAA4B,GAAM;YAEtB;AAEd,OAA4B,GAAM;;;GAIxC,eAAe;AACb,MAA2B,SAAS;;GAEtC,eAAe;AAGb,IAFA,EAA2B,SAAS,EAEhC,EAAwB,WAC1B,EAAgB,IAAiC,aAAa;;GAGlE,WAAW;GACZ;IACA;EACD;EACA;EACA;EACA;EACA;EACD,CAAC,EAKI,EAAE,QAAQ,GAAY,WAAW,OAAkB,GACvD,KAAgB,IAChB,GACD,EAEK,EAAE,QAAQ,GAAqB,WAAW,OAC9C,GAAa,KAAsB,IAAI,GAAyB,EAE5D,KAAY,QAAkB;AAGlC,MAFA,IAAoB,EACA,EAAqB,UAAU,CAAC,qBAChC,UAAU,GAAoB;AAChD,OAAmB;AACnB;;AAEF,MAAe;IACd;EACD;EACA;EACA;EACA;EACD,CAAC,EAII,KAAc,EAClB,OAAO,MAA4D;EAEjE,IAAM,IADc,EAAqB,UAAU,CAAC,qBAElC,SAAS,IAAsB;AAEjD,MAAI,GAAe,eAAe,UAAU,MAAM;AAGhD,OAAI,CAAC,GAAgB;IACnB,IAAM,IAAQ,gBAAI,MAAM,+BAA+B;AAEvD,UADA,EAAgB,EAAM,QAAQ,EACxB;;AAGR,OAAI;AAWF,WAVA,MAAM,IAAI,GAAmB,IAA6B,CAAC,CAAC,UAC1D,GACA;KACE,MAAM;KACN,SAAS,EAAQ;KAClB,EACD,EAAE,KAAK,IAAM,CACd,EAGM,EAAE,QAAQ,IAAM;YAChB,GAAO;AAMd,UADA,EAHE,aAAiB,QACb,EAAM,UACN,uCACuB,EACvB;;;AAIV,MAAI;AAIF,UADA,EAAc,KAAK,KAAK,UAAU;IAAE,GAAG;IAAS,KAAK;IAAM,CAAC,CAAC,EACtD,EAAE,QAAQ,IAAO;WACjB,GAAO;AAId,SADA,EADE,aAAiB,QAAQ,EAAM,UAAU,yBACd,EACvB;;IAGV;EAAC;EAAY;EAAqB;EAAiB;EAAe,CACnE;AAgCD,CA7BA,QAAgB;AAEd,EAAI,KAAc,YAEU;AACxB,WAAQ,EAAW,YAAnB;IACE,KAAK,UAAU;AACb,OAAuB,aAAa;AACpC;IACF,KAAK,UAAU;AACb,OAAuB,OAAO;AAC9B;IACF,KAAK,UAAU;AACb,OAAuB,UAAU;AACjC;IACF,KAAK,UAAU;AACb,OAAuB,SAAS;AAChC;IACF;AACE,OAAuB,SAAS;AAChC;;MAIO;IAEd,CAAC,GAAY,EAAM,CAAC,EAGvB,QAAgB;AAEd,EAAI,KAAuB,YAEC;AACxB,WAAQ,EAAoB,YAA5B;IACE,KAAK,UAAU;AACb,OAA2B,aAAa;AACxC;IACF,KAAK,UAAU;AACb,OAA2B,OAAO;AAClC;IACF,KAAK,UAAU;AACb,OAA2B,UAAU;AACrC;IACF,KAAK,UAAU;AACb,OAA2B,SAAS;AACpC;IACF;AACE,OAA2B,SAAS;AACpC;;MAIO;IAEd,CAAC,GAAqB,EAAmB,CAAC;CAE7C,IAAM,KAAe,SACZ;EAAE;EAAiB;EAAa;EAAkB;EAAW,GACpE;EAAC;EAAiB;EAAa;EAAkB;EAAU,CAC5D;AAED,QACE,mBAAC,EAA6B,UAA9B;EAAuC,OAAO;EAC3C;EACqC,CAAA;;AAI5C,IAAa,UAEO,EAAW,EAGpB,IAAW"}