fix unit tests

Author: charleslien

backend/src/__tests__/xml-stream-parser.test.ts

@@ -42,7 +42,8 @@ describe('processStreamWithTags', () => {
     for await (const chunk of processStreamWithTags(
       stream,
       processors,
-      onError
+      onError,
+      () => {}
     )) {
       result.push(chunk)
     }
@@ -92,7 +93,8 @@ describe('processStreamWithTags', () => {
     for await (const chunk of processStreamWithTags(
       stream,
       processors,
-      onError
+      onError,
+      () => {}
     )) {
       result.push(chunk)
     }
@@ -151,7 +153,8 @@ describe('processStreamWithTags', () => {
     for await (const chunk of processStreamWithTags(
       stream,
       processors,
-      onError
+      onError,
+      () => {}
     )) {
       result.push(chunk)
     }
@@ -209,7 +212,8 @@ describe('processStreamWithTags', () => {
     for await (const chunk of processStreamWithTags(
       stream,
       processors,
-      onError
+      onError,
+      () => {}
     )) {
       result.push(chunk)
     }
@@ -252,7 +256,8 @@ describe('processStreamWithTags', () => {
     for await (const chunk of processStreamWithTags(
       stream,
       processors,
-      onError
+      onError,
+      () => {}
     )) {
       result.push(chunk)
     }
@@ -300,7 +305,8 @@ describe('processStreamWithTags', () => {
     for await (const chunk of processStreamWithTags(
       stream,
       processors,
-      onError
+      onError,
+      () => {}
     )) {
       result.push(chunk)
     }
@@ -355,7 +361,8 @@ describe('processStreamWithTags', () => {
     for await (const chunk of processStreamWithTags(
       stream,
       processors,
-      onError
+      onError,
+      () => {}
     )) {
       result.push(chunk)
     }
@@ -404,7 +411,8 @@ describe('processStreamWithTags', () => {
     for await (const chunk of processStreamWithTags(
       stream,
       processors,
-      onError
+      onError,
+      () => {}
     )) {
       result.push(chunk)
     }
@@ -443,7 +451,8 @@ describe('processStreamWithTags', () => {
     for await (const chunk of processStreamWithTags(
       stream,
       processors,
-      onError
+      onError,
+      () => {}
     )) {
       result.push(chunk)
     }
@@ -468,7 +477,8 @@ describe('processStreamWithTags', () => {
     for await (const chunk of processStreamWithTags(
       stream,
       processors,
-      onError
+      onError,
+      () => {}
     )) {
       result.push(chunk)
     }
@@ -505,7 +515,8 @@ describe('processStreamWithTags', () => {
     for await (const chunk of processStreamWithTags(
       stream,
       processors,
-      onError
+      onError,
+      () => {}
     )) {
       result.push(chunk)
     }
@@ -561,7 +572,8 @@ describe('processStreamWithTags', () => {
       for await (const chunk of processStreamWithTags(
         stream,
         processors,
-        onError
+        onError,
+        () => {}
       )) {
         result.push(chunk)
       }

backend/src/run-programmatic-step.ts

@@ -1,11 +1,11 @@
+import { getToolCallString } from '@codebuff/common/constants/tools'
 import { PrintModeObject } from '@codebuff/common/types/print-mode'
 import {
   AgentState,
   AgentTemplateType,
   ToolResult,
 } from '@codebuff/common/types/session-state'
 import { ProjectFileContext } from '@codebuff/common/util/file'
-import { getToolCallString } from '@codebuff/common/constants/tools'
 import { WebSocket } from 'ws'
 import { AgentTemplate, StepGenerator } from './templates/types'
 import { CodebuffToolCall } from './tools/constants'
@@ -49,8 +49,6 @@ export async function runProgrammaticStep(
     onResponseChunk,
     agentType,
     fileContext,
-    assistantMessage,
-    assistantPrefix,
     ws,
   }: {
     template: AgentTemplate

backend/src/xml-stream-parser.ts

@@ -133,7 +133,7 @@ export async function* processStreamWithTags(
 
     if (chunk === undefined) {
       streamCompleted = true
-      if (buffer) {
+      if (buffer.includes(startToolTag)) {
         buffer += completionSuffix
         chunk = completionSuffix
         autocompleted = true