diff --git a/plugins/wasm-go/extensions/ai-search/main.go b/plugins/wasm-go/extensions/ai-search/main.go
index ac61ffd57..719e2b4c8 100644
--- a/plugins/wasm-go/extensions/ai-search/main.go
+++ b/plugins/wasm-go/extensions/ai-search/main.go
@@ -572,10 +572,8 @@ func onStreamingResponseBody(ctx wrapper.HttpContext, config Config, chunk []byt
 	} else {
 		ctx.SetContext(PARTIAL_MESSAGE_CONTEXT_KEY, nil)
 	}
-	if len(newMessages) == 1 {
-		return []byte(fmt.Sprintf("%s\n\n", newMessages[0]))
-	} else if len(newMessages) > 1 {
-		return []byte(strings.Join(newMessages, "\n\n"))
+	if len(newMessages) > 0 {
+		return []byte(fmt.Sprintf("%s\n\n", strings.Join(newMessages, "\n\n")))
 	} else {
 		return []byte("")
 	}