From 42a3eb150630b3a466b7ec9063a2256ecc38c60e Mon Sep 17 00:00:00 2001
From: kaitranntt <kaitran.ntt@gmail.com>
Date: Sat, 7 Feb 2026 04:30:42 -0500
Subject: [PATCH] refactor(cliproxy): check lifecycle events in remaining SSE
 buffer

Add message_start/message_delta/message_stop detection to the
remaining-buffer block in the SSE processing path for defensive
completeness. Addresses code review feedback on PR #493.
---
 src/cliproxy/tool-sanitization-proxy.ts | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/src/cliproxy/tool-sanitization-proxy.ts b/src/cliproxy/tool-sanitization-proxy.ts
index 0ba8497c..5276d8bd 100644
--- a/src/cliproxy/tool-sanitization-proxy.ts
+++ b/src/cliproxy/tool-sanitization-proxy.ts
@@ -535,6 +535,15 @@ export class ToolSanitizationProxy {
                 if (buffer.includes('"content_block_start"')) {
                   hasReceivedContent = true;
                 }
+                if (buffer.includes('"message_start"')) {
+                  hasReceivedMessageStart = true;
+                }
+                if (buffer.includes('"message_delta"')) {
+                  hasReceivedMessageDelta = true;
+                }
+                if (buffer.includes('"message_stop"')) {
+                  hasReceivedMessageStop = true;
+                }
                 const processedEvent = this.processSSEEvent(buffer, mapper);
                 clientRes.write(processedEvent + '\n\n');
               }