add filtering for stream chunks

Author: charleslien

cli/src/hooks/use-send-message.ts

@@ -509,8 +509,24 @@ export const useSendMessage = ({
           signal: abortController.signal,
           agentDefinitions: agentDefinitions as AgentDefinition[],
 
-          handleStreamChunk: (chunk: any) => {
-            if (typeof chunk !== 'string' || !chunk) {
+          handleStreamChunk: (event) => {
+            logger.info({event}, 'asdf')
+            if (typeof event !== 'string') {
+              const { agentId, chunk } = event
+
+              const previous =
+                agentStreamAccumulatorsRef.current.get(agentId) ?? ''
+              const { next, delta } = mergeTextSegments(previous, chunk)
+              if (!delta && next === previous) {
+                return
+              }
+              agentStreamAccumulatorsRef.current.set(agentId, next)
+
+              updateAgentContent(agentId, {
+                type: 'text',
+                content: delta || next,
+                ...(delta ? {} : { replace: true }),
+              })
               return
             }
 
@@ -520,16 +536,16 @@ export const useSendMessage = ({
             }
 
             const previous = rootStreamBufferRef.current ?? ''
-            const { next, delta } = mergeTextSegments(previous, chunk)
+            const { next, delta } = mergeTextSegments(previous, event)
             if (!delta && next === previous) {
               return
             }
             logger.info(
               {
-                chunkLength: chunk.length,
+                chunkLength: event.length,
                 previousLength: previous.length,
                 nextLength: next.length,
-                preview: chunk.slice(0, 100),
+                preview: event.slice(0, 100),
               },
               'handleStreamChunk root delta',
             )
@@ -546,25 +562,6 @@ export const useSendMessage = ({
               `SDK ${JSON.stringify(event.type)} Event received (raw)`,
             )
 
-            if (event.type === 'subagent_chunk') {
-              const { agentId, chunk } = event
-
-              const previous =
-                agentStreamAccumulatorsRef.current.get(agentId) ?? ''
-              const { next, delta } = mergeTextSegments(previous, chunk)
-              if (!delta && next === previous) {
-                return
-              }
-              agentStreamAccumulatorsRef.current.set(agentId, next)
-
-              updateAgentContent(agentId, {
-                type: 'text',
-                content: delta || next,
-                ...(delta ? {} : { replace: true }),
-              })
-              return
-            }
-
             if (event.type === 'text') {
               const text = event.text