Skip to content

Commit d0b66dd

Browse files
committed
Parse and emit transformed final result messages
WebSocketService now parses FINAL_RESULT_MESSAGE using PlanDataService before emitting, ensuring downstream consumers receive transformed data. Added a comment in PlanPage for handling final streaming messages.
1 parent 7f4d206 commit d0b66dd

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

src/frontend/src/pages/PlanPage.tsx

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -125,6 +125,7 @@ const PlanPage: React.FC = () => {
125125
useEffect(() => {
126126
const unsubscribe = webSocketService.on(WebsocketMessageType.AGENT_MESSAGE_STREAMING, (streamingMessage: any) => {
127127
// console.log('📋 Streaming Message', streamingMessage);
128+
// if is final true clear buffer and add final message to agent messages
128129
setStreamingMessageBuffer(prev => prev + streamingMessage.data.content);
129130
scrollToBottom();
130131

src/frontend/src/services/WebSocketService.tsx

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -246,8 +246,9 @@ class WebSocketService {
246246
case WebsocketMessageType.FINAL_RESULT_MESSAGE: {
247247
console.log("Message final result':", message);
248248
if (message.data) {
249-
//const transformed = PlanDataService.parseFinalResultMessage(message);
250-
this.emit(WebsocketMessageType.FINAL_RESULT_MESSAGE, message);
249+
const transformed = PlanDataService.parseFinalResultMessage(message);
250+
console.log('WebSocket FINAL_RESULT_MESSAGE received:', transformed);
251+
this.emit(WebsocketMessageType.FINAL_RESULT_MESSAGE, transformed);
251252
}
252253
break;
253254
}

0 commit comments

Comments
 (0)