@openhands/agent-canvas 1.0.0-alpha.6 → 1.0.0-alpha.7
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +41 -7
- package/bin/agent-canvas.mjs +9 -2
- package/build/assets/automation-detail-D7GEU0vR.js +1 -0
- package/build/assets/automations-list-CkVNsgzm.js +1 -0
- package/build/assets/conversation-COZAKz_K.js +1 -0
- package/build/assets/{conversation-D8scXOe7.js → conversation-DWcvnmds.js} +3 -1
- package/build/assets/conversation-panel-CZDStT0b.js +1 -0
- package/build/assets/conversation-websocket-context-DulnrIHh.js +3 -0
- package/build/assets/edit-automation-modal-C3bFxS2f.js +1 -0
- package/build/assets/git-control-bar-branch-button-Bm6rzSpo.js +27 -0
- package/build/assets/{home-D9fJfhQA.js → home-DR11ejqB.js} +1 -1
- package/build/assets/{manifest-6400820c.js → manifest-f041e61a.js} +1 -1
- package/build/assets/{messages-BfaEAG2q.js → messages-v-q35ObG.js} +1 -1
- package/build/assets/{root-6AdVEJBT.js → root-D2PVd51i.js} +1 -1
- package/build/assets/root-layout-B4QioBS6.js +2 -0
- package/build/assets/{shared-conversation-BfZNCsvo.js → shared-conversation-DQlzwdpo.js} +1 -1
- package/build/index.html +3 -3
- package/dist/components/features/backends/backend-selector.cjs +1 -1
- package/dist/components/features/backends/backend-selector.cjs.map +1 -1
- package/dist/components/features/backends/backend-selector.js +95 -95
- package/dist/components/features/backends/backend-selector.js.map +1 -1
- package/dist/components/features/chat/components/chat-input-actions.cjs +1 -1
- package/dist/components/features/chat/components/chat-input-actions.cjs.map +1 -1
- package/dist/components/features/chat/components/chat-input-actions.js +118 -118
- package/dist/components/features/chat/components/chat-input-actions.js.map +1 -1
- package/dist/components/features/chat/components/slash-command-menu.cjs +1 -1
- package/dist/components/features/chat/components/slash-command-menu.cjs.map +1 -1
- package/dist/components/features/chat/components/slash-command-menu.js +1 -1
- package/dist/components/features/chat/components/slash-command-menu.js.map +1 -1
- package/dist/components/features/sidebar/sidebar-rail-body.cjs +1 -1
- package/dist/components/features/sidebar/sidebar-rail-body.cjs.map +1 -1
- package/dist/components/features/sidebar/sidebar-rail-body.d.ts +1 -2
- package/dist/components/features/sidebar/sidebar-rail-body.js +104 -104
- package/dist/components/features/sidebar/sidebar-rail-body.js.map +1 -1
- package/dist/components/features/sidebar/sidebar.cjs +1 -1
- package/dist/components/features/sidebar/sidebar.cjs.map +1 -1
- package/dist/components/features/sidebar/sidebar.js +82 -83
- package/dist/components/features/sidebar/sidebar.js.map +1 -1
- package/dist/contexts/conversation-websocket-context.cjs +3 -3
- package/dist/contexts/conversation-websocket-context.cjs.map +1 -1
- package/dist/contexts/conversation-websocket-context.js +36 -36
- package/dist/contexts/conversation-websocket-context.js.map +1 -1
- package/dist/hooks/query/use-local-git-info.cjs +3 -1
- package/dist/hooks/query/use-local-git-info.cjs.map +1 -1
- package/dist/hooks/query/use-local-git-info.d.ts +2 -2
- package/dist/hooks/query/use-local-git-info.js +27 -24
- package/dist/hooks/query/use-local-git-info.js.map +1 -1
- package/dist/package.cjs +1 -1
- package/dist/package.cjs.map +1 -1
- package/dist/package.js +1 -1
- package/dist/package.js.map +1 -1
- package/dist/stores/error-message-store.cjs +1 -1
- package/dist/stores/error-message-store.cjs.map +1 -1
- package/dist/stores/error-message-store.d.ts +10 -1
- package/dist/stores/error-message-store.js +16 -3
- package/dist/stores/error-message-store.js.map +1 -1
- package/package.json +1 -1
- package/scripts/dev-static.mjs +6 -0
- package/scripts/dev-with-automation.mjs +6 -0
- package/scripts/static-server.mjs +85 -4
- package/build/assets/automation-detail-CQrtk33s.js +0 -1
- package/build/assets/automations-list-COmogz0S.js +0 -1
- package/build/assets/conversation-CeGMBOyB.js +0 -1
- package/build/assets/conversation-panel-DMz46ji-.js +0 -1
- package/build/assets/conversation-websocket-context-B0Gd3yiT.js +0 -3
- package/build/assets/edit-automation-modal-DnTHJrf1.js +0 -1
- package/build/assets/git-control-bar-branch-button-DhpPgadK.js +0 -27
- package/build/assets/root-layout-DvYGxAnr.js +0 -2
|
@@ -31,7 +31,7 @@ function _(e) {
|
|
|
31
31
|
return e.llm_message.content.filter((e) => e.type === "text").map((e) => e.text).join("");
|
|
32
32
|
}
|
|
33
33
|
function v({ children: Me, conversationId: d, conversationUrl: v, sessionApiKey: y, subConversations: b, subConversationIds: x }) {
|
|
34
|
-
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: M } = ve(), N = ye((e) => e.consumeMatchingPendingMessage), { setExecutionStatus: P } = ee(), { appendInput: F, appendOutput: I } = t(), [L, R] = h(!0), [z,
|
|
34
|
+
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(() => {
|
|
35
35
|
M();
|
|
36
36
|
}, [M]), G = u((e) => {
|
|
37
37
|
if (e.value.usage_to_metrics?.agent) {
|
|
@@ -49,7 +49,7 @@ function v({ children: Me, conversationId: d, conversationUrl: v, sessionApiKey:
|
|
|
49
49
|
};
|
|
50
50
|
Ee.getState().setMetrics(n);
|
|
51
51
|
}
|
|
52
|
-
}, []), { data: K, isPending: q, isError:
|
|
52
|
+
}, []), { data: K, isPending: q, isError: ze } = De(d), Be = !!d && q;
|
|
53
53
|
Ne(() => {
|
|
54
54
|
!K || K.events.length === 0 || Fe(K.events);
|
|
55
55
|
}, [K, Fe]);
|
|
@@ -57,16 +57,16 @@ function v({ children: Me, conversationId: d, conversationUrl: v, sessionApiKey:
|
|
|
57
57
|
if (q) return null;
|
|
58
58
|
let e = K?.events ?? [], t = e[e.length - 1];
|
|
59
59
|
return !t || !("timestamp" in t) || !t.timestamp ? null : t.timestamp;
|
|
60
|
-
}, [K, q]), Y = p(() => !d || !v || q && !
|
|
60
|
+
}, [K, q]), Y = p(() => !d || !v || q && !ze ? null : Se(d, v), [
|
|
61
61
|
d,
|
|
62
62
|
v,
|
|
63
63
|
q,
|
|
64
|
-
|
|
64
|
+
ze
|
|
65
65
|
]), X = p(() => {
|
|
66
66
|
if (!b?.length) return null;
|
|
67
67
|
let e = b[0];
|
|
68
68
|
return !e?.id || !e.conversation_url ? null : Se(e.id, e.conversation_url);
|
|
69
|
-
}, [b]),
|
|
69
|
+
}, [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, [
|
|
70
70
|
S,
|
|
71
71
|
w,
|
|
72
72
|
X
|
|
@@ -97,11 +97,11 @@ function v({ children: Me, conversationId: d, conversationUrl: v, sessionApiKey:
|
|
|
97
97
|
V,
|
|
98
98
|
B
|
|
99
99
|
]), f(() => {
|
|
100
|
-
E.current = !1, R(!!x?.length),
|
|
100
|
+
E.current = !1, R(!!x?.length), Le(null), H.current = 0, U.current = null;
|
|
101
101
|
}, [x]), f(() => {
|
|
102
102
|
E.current = !1, D.current = !1, U.current = null;
|
|
103
103
|
}, [d]);
|
|
104
|
-
let
|
|
104
|
+
let He = p(() => Be || L, [Be, L]), Ue = u((t) => {
|
|
105
105
|
try {
|
|
106
106
|
let r = JSON.parse(t.data);
|
|
107
107
|
if (re(r)) {
|
|
@@ -118,7 +118,7 @@ function v({ children: Me, conversationId: d, conversationUrl: v, sessionApiKey:
|
|
|
118
118
|
posthog: O
|
|
119
119
|
}), j(e.detail);
|
|
120
120
|
} else W();
|
|
121
|
-
if (ne(r) &&
|
|
121
|
+
if (ne(r) && l({
|
|
122
122
|
message: r.error,
|
|
123
123
|
source: "agent",
|
|
124
124
|
metadata: {
|
|
@@ -127,7 +127,7 @@ function v({ children: Me, conversationId: d, conversationUrl: v, sessionApiKey:
|
|
|
127
127
|
toolCallId: r.tool_call_id
|
|
128
128
|
},
|
|
129
129
|
posthog: O
|
|
130
|
-
}),
|
|
130
|
+
}), 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)) {
|
|
131
131
|
let { screenshot_data: t } = r.observation;
|
|
132
132
|
if (t) {
|
|
133
133
|
let n = t.startsWith("data:") ? t : `data:image/png;base64,${t}`;
|
|
@@ -151,7 +151,7 @@ function v({ children: Me, conversationId: d, conversationUrl: v, sessionApiKey:
|
|
|
151
151
|
G,
|
|
152
152
|
W,
|
|
153
153
|
O
|
|
154
|
-
]),
|
|
154
|
+
]), We = u((e) => {
|
|
155
155
|
try {
|
|
156
156
|
let t = JSON.parse(e.data);
|
|
157
157
|
if (L && (H.current += 1, z !== null && H.current >= z && R(!1)), re(t)) {
|
|
@@ -170,7 +170,7 @@ function v({ children: Me, conversationId: d, conversationUrl: v, sessionApiKey:
|
|
|
170
170
|
posthog: O
|
|
171
171
|
}), j(e.detail);
|
|
172
172
|
} else W();
|
|
173
|
-
if (ne(t) &&
|
|
173
|
+
if (ne(t) && l({
|
|
174
174
|
message: t.error,
|
|
175
175
|
source: "planning_agent",
|
|
176
176
|
metadata: {
|
|
@@ -179,9 +179,9 @@ function v({ children: Me, conversationId: d, conversationUrl: v, sessionApiKey:
|
|
|
179
179
|
toolCallId: t.tool_call_id
|
|
180
180
|
},
|
|
181
181
|
posthog: O
|
|
182
|
-
}),
|
|
182
|
+
}), 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)) {
|
|
183
183
|
let { path: e } = t.observation;
|
|
184
|
-
if (
|
|
184
|
+
if (Re(e)) {
|
|
185
185
|
let t = b?.[0]?.id;
|
|
186
186
|
t && e && (L ? U.current = {
|
|
187
187
|
path: e,
|
|
@@ -220,7 +220,7 @@ function v({ children: Me, conversationId: d, conversationUrl: v, sessionApiKey:
|
|
|
220
220
|
G,
|
|
221
221
|
W,
|
|
222
222
|
O
|
|
223
|
-
]),
|
|
223
|
+
]), Ge = p(() => {
|
|
224
224
|
let e = J ? {
|
|
225
225
|
resend_mode: "since",
|
|
226
226
|
after_timestamp: J
|
|
@@ -235,17 +235,17 @@ function v({ children: Me, conversationId: d, conversationUrl: v, sessionApiKey:
|
|
|
235
235
|
C("CLOSED");
|
|
236
236
|
},
|
|
237
237
|
onError: () => {
|
|
238
|
-
C("CLOSED"), E.current && j(_e);
|
|
238
|
+
C("CLOSED"), E.current && j(_e, "connection");
|
|
239
239
|
},
|
|
240
|
-
onMessage:
|
|
240
|
+
onMessage: Ue
|
|
241
241
|
};
|
|
242
242
|
}, [
|
|
243
|
-
|
|
243
|
+
Ue,
|
|
244
244
|
j,
|
|
245
245
|
M,
|
|
246
246
|
y,
|
|
247
247
|
J
|
|
248
|
-
]),
|
|
248
|
+
]), Ke = p(() => {
|
|
249
249
|
let e = { resend_all: !0 };
|
|
250
250
|
y && (e.session_api_key = y);
|
|
251
251
|
let t = b?.[0];
|
|
@@ -255,7 +255,7 @@ function v({ children: Me, conversationId: d, conversationUrl: v, sessionApiKey:
|
|
|
255
255
|
onOpen: async () => {
|
|
256
256
|
if (T("OPEN"), D.current = !0, M(), t?.id && t.conversation_url) try {
|
|
257
257
|
let e = await we.getEventCount(t.id, t.conversation_url, t.session_api_key);
|
|
258
|
-
|
|
258
|
+
Le(e), e === 0 && R(!1);
|
|
259
259
|
} catch {
|
|
260
260
|
R(!1);
|
|
261
261
|
}
|
|
@@ -264,27 +264,27 @@ function v({ children: Me, conversationId: d, conversationUrl: v, sessionApiKey:
|
|
|
264
264
|
T("CLOSED");
|
|
265
265
|
},
|
|
266
266
|
onError: () => {
|
|
267
|
-
T("CLOSED"), D.current && j(_e);
|
|
267
|
+
T("CLOSED"), D.current && j(_e, "connection");
|
|
268
268
|
},
|
|
269
|
-
onMessage:
|
|
269
|
+
onMessage: We
|
|
270
270
|
};
|
|
271
271
|
}, [
|
|
272
|
-
|
|
272
|
+
We,
|
|
273
273
|
j,
|
|
274
274
|
M,
|
|
275
275
|
y,
|
|
276
276
|
b
|
|
277
|
-
]), { socket: Z, reconnect:
|
|
278
|
-
if (
|
|
279
|
-
|
|
277
|
+
]), { socket: Z, reconnect: qe } = ge(Y || "", Ge), { socket: Q, reconnect: Je } = ge(X || "", Ke), Ye = u(() => {
|
|
278
|
+
if (Ie(), r.getState().conversationMode === "plan" && X) {
|
|
279
|
+
Je();
|
|
280
280
|
return;
|
|
281
281
|
}
|
|
282
|
-
|
|
282
|
+
qe();
|
|
283
283
|
}, [
|
|
284
284
|
X,
|
|
285
|
-
Ke,
|
|
286
285
|
qe,
|
|
287
|
-
|
|
286
|
+
Je,
|
|
287
|
+
Ie
|
|
288
288
|
]), $ = u(async (e) => {
|
|
289
289
|
let t = r.getState().conversationMode === "plan" ? Q : Z;
|
|
290
290
|
if (t?.readyState !== WebSocket.OPEN) {
|
|
@@ -356,19 +356,19 @@ function v({ children: Me, conversationId: d, conversationUrl: v, sessionApiKey:
|
|
|
356
356
|
}
|
|
357
357
|
})();
|
|
358
358
|
}, [Q, X]);
|
|
359
|
-
let
|
|
360
|
-
connectionState:
|
|
359
|
+
let Xe = p(() => ({
|
|
360
|
+
connectionState: Ve,
|
|
361
361
|
sendMessage: $,
|
|
362
|
-
isLoadingHistory:
|
|
363
|
-
reconnect:
|
|
362
|
+
isLoadingHistory: He,
|
|
363
|
+
reconnect: Ye
|
|
364
364
|
}), [
|
|
365
|
-
Be,
|
|
366
|
-
$,
|
|
367
365
|
Ve,
|
|
368
|
-
|
|
366
|
+
$,
|
|
367
|
+
He,
|
|
368
|
+
Ye
|
|
369
369
|
]);
|
|
370
370
|
return /* @__PURE__ */ Pe(g.Provider, {
|
|
371
|
-
value:
|
|
371
|
+
value: Xe,
|
|
372
372
|
children: Me
|
|
373
373
|
});
|
|
374
374
|
}
|
|
@@ -1 +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 } = 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 removeErrorMessage();\n }, [removeErrorMessage]);\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 // Track credit limit reached if AgentErrorEvent has budget-related error\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 setErrorMessage(event.error);\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 // Handle AgentErrorEvent specifically\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 setErrorMessage(event.error);\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 removeErrorMessage(); // Clear any previous error messages on successful connection\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);\n }\n },\n onMessage: handleMainMessage,\n };\n }, [\n handleMainMessage,\n setErrorMessage,\n removeErrorMessage,\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 removeErrorMessage(); // Clear any previous error messages on successful connection\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);\n }\n },\n onMessage: handlePlanningMessage,\n };\n }, [\n handlePlanningMessage,\n setErrorMessage,\n removeErrorMessage,\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,0BAAuB,IAAsB,EAChE,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;AAC5C,KAAoB;IACnB,CAAC,EAAmB,CAAC,EAGlB,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,QAvEI,GAAkB,EAAM,KAC1B,EAAW;KACT,SAAS,EAAM;KACf,QAAQ;KACR,UAAU;MACR,SAAS,EAAM;MACf,UAAU,EAAM;MAChB,YAAY,EAAM;MACnB;KACD;KACD,CAAC,EACF,EAAgB,EAAM,MAAM,GAQ1B,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,QAtEI,GAAkB,EAAM,KAC1B,EAAW;KACT,SAAS,EAAM;KACf,QAAQ;KACR,UAAU;MACR,SAAS,EAAM;MACf,UAAU,EAAM;MAChB,YAAY,EAAM;MACnB;KACD;KACD,CAAC,EACF,EAAgB,EAAM,MAAM,GAO1B,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,GAAoB;;GAEtB,eAAe;AACb,MAAuB,SAAS;;GAElC,eAAe;AAGb,IAFA,EAAuB,SAAS,EAE5B,EAAoB,WACtB,EAAgB,GAAgC;;GAGpD,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,GAAoB,EAIlB,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,GAAgC;;GAGpD,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,GAAoB,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"}
|
|
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"}
|
|
@@ -1,2 +1,4 @@
|
|
|
1
|
-
require(`../../_virtual/_rolldown/runtime.cjs`);const e=require(`../../api/agent-server-config.cjs`),t=require(`../../contexts/active-backend-context.cjs`),n=require(`../../node_modules/@tanstack/react-query/build/modern/useQuery.cjs`),r=require(`./use-active-conversation.cjs`),i=require(`../use-runtime-is-ready.cjs`),a=require(`../use-bash-command-runner.cjs`),o=require(`../../utils/parse-git-remote-url.cjs`);let s=require(`react`);var c={repository:null,branch:null,provider:null,remoteUrl:null}
|
|
1
|
+
require(`../../_virtual/_rolldown/runtime.cjs`);const e=require(`../../api/agent-server-config.cjs`),t=require(`../../contexts/active-backend-context.cjs`),n=require(`../../node_modules/@tanstack/react-query/build/modern/useQuery.cjs`),r=require(`./use-active-conversation.cjs`),i=require(`../use-runtime-is-ready.cjs`),a=require(`../use-bash-command-runner.cjs`),o=require(`../../utils/parse-git-remote-url.cjs`);let s=require(`react`);var c={repository:null,branch:null,provider:null,remoteUrl:null},l=[`r=$(git remote get-url origin 2>/dev/null)`,`b=$(git rev-parse --abbrev-ref HEAD 2>/dev/null)`,`if [ -z "$r$b" ]; then`,`n=$(find . -mindepth 2 -maxdepth 4 -name .git 2>/dev/null | cut -c3- | sed 's|/.git$||' | sort -u)`,`c=$(printf '%s\\n' "$n" | grep -c '[^[:space:]]')`,`if [ "$c" = "1" ] && [ -n "$n" ]; then`,`r=$(git -C "$n" remote get-url origin 2>/dev/null)`,`b=$(git -C "$n" rev-parse --abbrev-ref HEAD 2>/dev/null)`,`fi`,`fi`,`printf '%s\\n%s' "$r" "$b"`].join(`
|
|
2
|
+
`);async function u(e,t){let n=await e(l,t,10);if(n.exit_code!==0)return c;let r=n.stdout.indexOf(`
|
|
3
|
+
`),i=(r>=0?n.stdout.slice(0,r):n.stdout).trim(),a=(r>=0?n.stdout.slice(r+1):``).trim(),s=a&&a!==`HEAD`?a:null;if(!i&&!s)return c;let u=o.parseGitRemoteUrl(i);return{repository:u?.repository??null,provider:u?.provider??null,remoteUrl:i||null,branch:s}}var d=()=>{let{data:o}=r.useActiveConversation(),c=i.useRuntimeIsReady(),{backend:l}=t.useActiveBackend(),d=l.kind===`local`,f=o?.id,p=o?.conversation_url,m=o?.session_api_key,h=o?.workspace?.working_dir?.trim()||e.getAgentServerWorkingDir(),g=!!o?.selected_repository,_=!!o?.git_provider,v=!!o?.selected_branch,y=d&&c&&!!f&&(!g||!_||!v),b=a.useBashCommandRunner(p,m,y),x=(0,s.useRef)(b);return x.current=b,n.useQuery({queryKey:[`local-git-info`,f,p,m,h],queryFn:async()=>u((e,t,n)=>x.current(e,t,n),h),enabled:y,retry:!1,staleTime:1e4,refetchInterval:1e4,gcTime:1e3*60*5,meta:{disableToast:!0}})};exports.useLocalGitInfo=d;
|
|
2
4
|
//# sourceMappingURL=use-local-git-info.cjs.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"use-local-git-info.cjs","names":[],"sources":["../../../src/hooks/query/use-local-git-info.ts"],"sourcesContent":["import { useQuery } from \"@tanstack/react-query\";\nimport { useRef } from \"react\";\n\nimport type { CommandResult } from \"#/api/runtime-service/agent-server-runtime-service\";\nimport { getAgentServerWorkingDir } from \"#/api/agent-server-config\";\nimport { useActiveBackend } from \"#/contexts/active-backend-context\";\nimport { useActiveConversation } from \"#/hooks/query/use-active-conversation\";\nimport { useRuntimeIsReady } from \"#/hooks/use-runtime-is-ready\";\nimport { useBashCommandRunner } from \"#/hooks/use-bash-command-runner\";\nimport { Provider } from \"#/types/settings\";\nimport { parseGitRemoteUrl } from \"#/utils/parse-git-remote-url\";\n\nexport interface LocalGitInfo {\n repository: string | null;\n branch: string | null;\n provider: Provider | null;\n remoteUrl: string | null;\n}\n\nconst EMPTY_LOCAL_GIT_INFO: LocalGitInfo = {\n repository: null,\n branch: null,\n provider: null,\n remoteUrl: null,\n};\n\ntype RunCommand = (\n command: string,\n cwd: string,\n timeout: number,\n) => Promise<CommandResult>;\n\
|
|
1
|
+
{"version":3,"file":"use-local-git-info.cjs","names":[],"sources":["../../../src/hooks/query/use-local-git-info.ts"],"sourcesContent":["import { useQuery } from \"@tanstack/react-query\";\nimport { useRef } from \"react\";\n\nimport type { CommandResult } from \"#/api/runtime-service/agent-server-runtime-service\";\nimport { getAgentServerWorkingDir } from \"#/api/agent-server-config\";\nimport { useActiveBackend } from \"#/contexts/active-backend-context\";\nimport { useActiveConversation } from \"#/hooks/query/use-active-conversation\";\nimport { useRuntimeIsReady } from \"#/hooks/use-runtime-is-ready\";\nimport { useBashCommandRunner } from \"#/hooks/use-bash-command-runner\";\nimport { Provider } from \"#/types/settings\";\nimport { parseGitRemoteUrl } from \"#/utils/parse-git-remote-url\";\n\nexport interface LocalGitInfo {\n repository: string | null;\n branch: string | null;\n provider: Provider | null;\n remoteUrl: string | null;\n}\n\nconst EMPTY_LOCAL_GIT_INFO: LocalGitInfo = {\n repository: null,\n branch: null,\n provider: null,\n remoteUrl: null,\n};\n\ntype RunCommand = (\n command: string,\n cwd: string,\n timeout: number,\n) => Promise<CommandResult>;\n\n// Single shell script that replaces the former probeGitInfoAtDir +\n// probeNestedRepoInDir pair. It runs as one bash WebSocket round-trip:\n// 1. Read the origin remote URL and current branch at the workspace root.\n// 2. If neither is set, search for exactly one nested git repo up to 4\n// levels deep and repeat the probe there.\n// Output: two lines — <remote-url>\\n<branch> — either may be empty.\nconst GIT_INFO_COMMAND = [\n \"r=$(git remote get-url origin 2>/dev/null)\",\n \"b=$(git rev-parse --abbrev-ref HEAD 2>/dev/null)\",\n 'if [ -z \"$r$b\" ]; then',\n \"n=$(find . -mindepth 2 -maxdepth 4 -name .git 2>/dev/null | cut -c3- | sed 's|/.git$||' | sort -u)\",\n \"c=$(printf '%s\\\\n' \\\"$n\\\" | grep -c '[^[:space:]]')\",\n 'if [ \"$c\" = \"1\" ] && [ -n \"$n\" ]; then',\n 'r=$(git -C \"$n\" remote get-url origin 2>/dev/null)',\n 'b=$(git -C \"$n\" rev-parse --abbrev-ref HEAD 2>/dev/null)',\n \"fi\",\n \"fi\",\n 'printf \\'%s\\\\n%s\\' \"$r\" \"$b\"',\n].join(\"\\n\");\n\nasync function probeGitInfo(\n run: RunCommand,\n directory: string,\n): Promise<LocalGitInfo> {\n const result = await run(GIT_INFO_COMMAND, directory, 10);\n if (result.exit_code !== 0) return EMPTY_LOCAL_GIT_INFO;\n\n const nl = result.stdout.indexOf(\"\\n\");\n const remoteUrl = (\n nl >= 0 ? result.stdout.slice(0, nl) : result.stdout\n ).trim();\n const rawBranch = (nl >= 0 ? result.stdout.slice(nl + 1) : \"\").trim();\n const branch = rawBranch && rawBranch !== \"HEAD\" ? rawBranch : null;\n\n if (!remoteUrl && !branch) return EMPTY_LOCAL_GIT_INFO;\n\n const parsedRemote = parseGitRemoteUrl(remoteUrl);\n return {\n repository: parsedRemote?.repository ?? null,\n provider: parsedRemote?.provider ?? null,\n remoteUrl: remoteUrl || null,\n branch,\n };\n}\n\n/**\n * Probe git metadata for a **local** backend's workspace checkout by\n * shelling out via the agent server using a single consolidated bash\n * script (see `GIT_INFO_COMMAND`).\n *\n * Local-only by design. On cloud backends the conversation metadata\n * (`selected_repository`, `git_provider`, `selected_branch`) is the\n * source of truth, and probing via `/api/bash/execute_bash_command`\n * would (a) leak the user's local `getAgentServerWorkingDir()` path to\n * the cloud runtime when `workspace.working_dir` is missing, and\n * (b) hit a bash endpoint we don't want the frontend driving on cloud.\n *\n * On local, we keep the probe enabled until the active conversation\n * has a complete repo tuple so the control bar can recover from\n * partial metadata hydration after connect/clone flows.\n *\n * Returns `null` fields when the working dir is not a git checkout —\n * callers should treat that the same as \"no repo detected\".\n */\nexport const useLocalGitInfo = () => {\n const { data: conversation } = useActiveConversation();\n const runtimeIsReady = useRuntimeIsReady();\n const { backend } = useActiveBackend();\n const isLocalBackend = backend.kind === \"local\";\n\n const conversationId = conversation?.id;\n const conversationUrl = conversation?.conversation_url;\n const sessionApiKey = conversation?.session_api_key;\n const workingDir =\n conversation?.workspace?.working_dir?.trim() || getAgentServerWorkingDir();\n const hasConversationRepo = !!conversation?.selected_repository;\n const hasConversationProvider = !!conversation?.git_provider;\n const hasConversationBranch = !!conversation?.selected_branch;\n\n const queryEnabled =\n isLocalBackend &&\n runtimeIsReady &&\n !!conversationId &&\n (!hasConversationRepo ||\n !hasConversationProvider ||\n !hasConversationBranch);\n\n // Persistent WebSocket connection to the bash-events endpoint. The\n // connection is opened when the query is enabled and closed on unmount or\n // when the conversation changes.\n const runCommand = useBashCommandRunner(\n conversationUrl,\n sessionApiKey,\n queryEnabled,\n );\n\n // Keep a ref so queryFn can call the latest runner without capturing it\n // as a queryKey dependency (runCommand is stable but the linter can't\n // infer that).\n const runCommandRef = useRef(runCommand);\n runCommandRef.current = runCommand;\n\n // runCommandRef is a ref (always stable); the linter cannot infer this so\n // we disable the exhaustive-deps check here.\n // eslint-disable-next-line @tanstack/query/exhaustive-deps\n return useQuery<LocalGitInfo>({\n queryKey: [\n \"local-git-info\",\n conversationId,\n conversationUrl,\n sessionApiKey,\n workingDir,\n ],\n queryFn: async () => {\n const run: RunCommand = (command, cwd, timeout) =>\n runCommandRef.current(command, cwd, timeout);\n return probeGitInfo(run, workingDir);\n },\n enabled: queryEnabled,\n retry: false,\n // Re-probe the workspace every 10s so the UI reflects branch/repo\n // changes (e.g. `git checkout`, adding a remote) without requiring a\n // manual refresh when there is no `selected_repository` recorded on\n // the conversation. Commands now run over the persistent WebSocket\n // connection rather than individual REST calls.\n staleTime: 10_000,\n refetchInterval: 10_000,\n gcTime: 1000 * 60 * 5,\n meta: { disableToast: true },\n });\n};\n"],"mappings":"qbAmBA,IAAM,EAAqC,CACzC,WAAY,KACZ,OAAQ,KACR,SAAU,KACV,UAAW,KACZ,CAcK,EAAmB,CACvB,6CACA,mDACA,yBACA,qGACA,oDACA,yCACA,qDACA,2DACA,KACA,KACA,6BACD,CAAC,KAAK;EAAK,CAEZ,eAAe,EACb,EACA,EACuB,CACvB,IAAM,EAAS,MAAM,EAAI,EAAkB,EAAW,GAAG,CACzD,GAAI,EAAO,YAAc,EAAG,OAAO,EAEnC,IAAM,EAAK,EAAO,OAAO,QAAQ;EAAK,CAChC,GACJ,GAAM,EAAI,EAAO,OAAO,MAAM,EAAG,EAAG,CAAG,EAAO,QAC9C,MAAM,CACF,GAAa,GAAM,EAAI,EAAO,OAAO,MAAM,EAAK,EAAE,CAAG,IAAI,MAAM,CAC/D,EAAS,GAAa,IAAc,OAAS,EAAY,KAE/D,GAAI,CAAC,GAAa,CAAC,EAAQ,OAAO,EAElC,IAAM,EAAe,EAAA,kBAAkB,EAAU,CACjD,MAAO,CACL,WAAY,GAAc,YAAc,KACxC,SAAU,GAAc,UAAY,KACpC,UAAW,GAAa,KACxB,SACD,CAsBH,IAAa,MAAwB,CACnC,GAAM,CAAE,KAAM,GAAiB,EAAA,uBAAuB,CAChD,EAAiB,EAAA,mBAAmB,CACpC,CAAE,WAAY,EAAA,kBAAkB,CAChC,EAAiB,EAAQ,OAAS,QAElC,EAAiB,GAAc,GAC/B,EAAkB,GAAc,iBAChC,EAAgB,GAAc,gBAC9B,EACJ,GAAc,WAAW,aAAa,MAAM,EAAI,EAAA,0BAA0B,CACtE,EAAsB,CAAC,CAAC,GAAc,oBACtC,EAA0B,CAAC,CAAC,GAAc,aAC1C,EAAwB,CAAC,CAAC,GAAc,gBAExC,EACJ,GACA,GACA,CAAC,CAAC,IACD,CAAC,GACA,CAAC,GACD,CAAC,GAKC,EAAa,EAAA,qBACjB,EACA,EACA,EACD,CAKK,GAAA,EAAA,EAAA,QAAuB,EAAW,CAMxC,MALA,GAAc,QAAU,EAKjB,EAAA,SAAuB,CAC5B,SAAU,CACR,iBACA,EACA,EACA,EACA,EACD,CACD,QAAS,SAGA,GAFkB,EAAS,EAAK,IACrC,EAAc,QAAQ,EAAS,EAAK,EAAQ,CACrB,EAAW,CAEtC,QAAS,EACT,MAAO,GAMP,UAAW,IACX,gBAAiB,IACjB,OAAQ,IAAO,GAAK,EACpB,KAAM,CAAE,aAAc,GAAM,CAC7B,CAAC"}
|
|
@@ -7,8 +7,8 @@ export interface LocalGitInfo {
|
|
|
7
7
|
}
|
|
8
8
|
/**
|
|
9
9
|
* Probe git metadata for a **local** backend's workspace checkout by
|
|
10
|
-
* shelling out via the agent server
|
|
11
|
-
*
|
|
10
|
+
* shelling out via the agent server using a single consolidated bash
|
|
11
|
+
* script (see `GIT_INFO_COMMAND`).
|
|
12
12
|
*
|
|
13
13
|
* Local-only by design. On cloud backends the conversation metadata
|
|
14
14
|
* (`selected_repository`, `git_provider`, `selected_branch`) is the
|
|
@@ -12,41 +12,44 @@ var c = {
|
|
|
12
12
|
branch: null,
|
|
13
13
|
provider: null,
|
|
14
14
|
remoteUrl: null
|
|
15
|
-
}
|
|
16
|
-
|
|
17
|
-
|
|
15
|
+
}, l = [
|
|
16
|
+
"r=$(git remote get-url origin 2>/dev/null)",
|
|
17
|
+
"b=$(git rev-parse --abbrev-ref HEAD 2>/dev/null)",
|
|
18
|
+
"if [ -z \"$r$b\" ]; then",
|
|
19
|
+
"n=$(find . -mindepth 2 -maxdepth 4 -name .git 2>/dev/null | cut -c3- | sed 's|/.git$||' | sort -u)",
|
|
20
|
+
"c=$(printf '%s\\n' \"$n\" | grep -c '[^[:space:]]')",
|
|
21
|
+
"if [ \"$c\" = \"1\" ] && [ -n \"$n\" ]; then",
|
|
22
|
+
"r=$(git -C \"$n\" remote get-url origin 2>/dev/null)",
|
|
23
|
+
"b=$(git -C \"$n\" rev-parse --abbrev-ref HEAD 2>/dev/null)",
|
|
24
|
+
"fi",
|
|
25
|
+
"fi",
|
|
26
|
+
"printf '%s\\n%s' \"$r\" \"$b\""
|
|
27
|
+
].join("\n");
|
|
28
|
+
async function u(e, t) {
|
|
29
|
+
let n = await e(l, t, 10);
|
|
30
|
+
if (n.exit_code !== 0) return c;
|
|
31
|
+
let r = n.stdout.indexOf("\n"), i = (r >= 0 ? n.stdout.slice(0, r) : n.stdout).trim(), a = (r >= 0 ? n.stdout.slice(r + 1) : "").trim(), s = a && a !== "HEAD" ? a : null;
|
|
18
32
|
if (!i && !s) return c;
|
|
19
|
-
let
|
|
33
|
+
let u = o(i);
|
|
20
34
|
return {
|
|
21
|
-
repository:
|
|
22
|
-
provider:
|
|
35
|
+
repository: u?.repository ?? null,
|
|
36
|
+
provider: u?.provider ?? null,
|
|
23
37
|
remoteUrl: i || null,
|
|
24
38
|
branch: s
|
|
25
39
|
};
|
|
26
40
|
}
|
|
27
|
-
async function u(e, t) {
|
|
28
|
-
let n = await e("find . -mindepth 2 -maxdepth 4 -name .git 2>/dev/null | sed 's#^\\./##' | sed 's#/.git$##'", t, 10);
|
|
29
|
-
if (n.exit_code !== 0) return c;
|
|
30
|
-
let r = Array.from(new Set(n.stdout.split(/\r?\n/).map((e) => e.trim()).filter(Boolean)));
|
|
31
|
-
return r.length === 1 ? l(e, `${t}/${r[0]}`.replace(/\/+/g, "/")) : c;
|
|
32
|
-
}
|
|
33
41
|
var d = () => {
|
|
34
|
-
let { data: o } = r(),
|
|
35
|
-
return
|
|
42
|
+
let { data: o } = r(), c = i(), { backend: l } = t(), d = l.kind === "local", f = o?.id, p = o?.conversation_url, m = o?.session_api_key, h = o?.workspace?.working_dir?.trim() || e(), g = !!o?.selected_repository, _ = !!o?.git_provider, v = !!o?.selected_branch, y = d && c && !!f && (!g || !_ || !v), b = a(p, m, y), x = s(b);
|
|
43
|
+
return x.current = b, n({
|
|
36
44
|
queryKey: [
|
|
37
45
|
"local-git-info",
|
|
46
|
+
f,
|
|
47
|
+
p,
|
|
38
48
|
m,
|
|
39
|
-
h
|
|
40
|
-
g,
|
|
41
|
-
_
|
|
49
|
+
h
|
|
42
50
|
],
|
|
43
|
-
queryFn: async () =>
|
|
44
|
-
|
|
45
|
-
if (t.repository || t.branch) return t;
|
|
46
|
-
let n = await u(e, _);
|
|
47
|
-
return n.repository || n.branch ? n : c;
|
|
48
|
-
},
|
|
49
|
-
enabled: x,
|
|
51
|
+
queryFn: async () => u((e, t, n) => x.current(e, t, n), h),
|
|
52
|
+
enabled: y,
|
|
50
53
|
retry: !1,
|
|
51
54
|
staleTime: 1e4,
|
|
52
55
|
refetchInterval: 1e4,
|