fix unit tests

charleslien · charleslien · commit 635678c227df · 2025-10-03T15:49:22.000-07:00
diff --git a/backend/src/xml-stream-parser.ts b/backend/src/xml-stream-parser.ts
@@ -138,18 +138,15 @@ export async function* processStreamWithTags(
     processor.onTagEnd(toolName, parsedParams)
   }
 
-  function* extractToolsFromBufferAndProcess(
-    forceFlush = true,
-  ): Generator<StreamChunk> {
+  function extractToolsFromBufferAndProcess(forceFlush = false) {
     const matches = extractToolCalls()
     matches.forEach(processToolCallContents)
     if (forceFlush) {
-      const chunk: StreamChunk = {
+      onResponseChunk({
         type: 'text',
         text: buffer,
-      }
-      yield chunk
-      onResponseChunk(chunk)
+      })
+      buffer = ''
     }
   }
 
@@ -171,7 +168,7 @@ export async function* processStreamWithTags(
         }
         autocompleted = true
       }
-      yield* extractToolsFromBufferAndProcess(true)
+      extractToolsFromBufferAndProcess(true)
     }
 
     if (chunk) {

Original file line number	Diff line number	Diff line change
`@@ -138,18 +138,15 @@ export async function* processStreamWithTags(`
`138`	`138`	`processor.onTagEnd(toolName, parsedParams)`
`139`	`139`	`}`
`140`	`140`
`141`		`- function* extractToolsFromBufferAndProcess(`
`142`		`- forceFlush = true,`
`143`		`- ): Generator<StreamChunk> {`
	`141`	`+ function extractToolsFromBufferAndProcess(forceFlush = false) {`
`144`	`142`	`const matches = extractToolCalls()`
`145`	`143`	`matches.forEach(processToolCallContents)`
`146`	`144`	`if (forceFlush) {`
`147`		`- const chunk: StreamChunk = {`
	`145`	`+ onResponseChunk({`
`148`	`146`	`type: 'text',`
`149`	`147`	`text: buffer,`
`150`		`- }`
`151`		`- yield chunk`
`152`		`- onResponseChunk(chunk)`
	`148`	`+ })`
	`149`	`+ buffer = ''`
`153`	`150`	`}`
`154`	`151`	`}`
`155`	`152`
`@@ -171,7 +168,7 @@ export async function* processStreamWithTags(`
`171`	`168`	`}`
`172`	`169`	`autocompleted = true`
`173`	`170`	`}`
`174`		`- yield* extractToolsFromBufferAndProcess(true)`
	`171`	`+ extractToolsFromBufferAndProcess(true)`
`175`	`172`	`}`
`176`	`173`
`177`	`174`	`if (chunk) {`