From f09e029a6b61b40a04690c79e0e729fb893c8f22 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=BE=84=E6=BD=AD?= Date: Fri, 14 Mar 2025 21:52:49 +0800 Subject: [PATCH] fix chunk merge bug in ai-search (#1895) --- plugins/wasm-go/extensions/ai-search/main.go | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) 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("") }