From 3db8eb6de30acf6176033ca3eb84dfa24083265a Mon Sep 17 00:00:00 2001
From: glc-coder <53860678+leon6002@users.noreply.github.com>
Date: Wed, 31 Jul 2024 00:01:12 +0800
Subject: [PATCH] fix ollama stream response unmarshall error

fix ollama stream response body unmarshall error
---
 relay/adaptor/ollama/main.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/relay/adaptor/ollama/main.go b/relay/adaptor/ollama/main.go
index 936a7e144e..79d69dd573 100644
--- a/relay/adaptor/ollama/main.go
+++ b/relay/adaptor/ollama/main.go
@@ -119,7 +119,9 @@ func StreamHandler(c *gin.Context, resp *http.Response) (*model.ErrorWithStatusC
 
 	for scanner.Scan() {
 		data := strings.TrimPrefix(scanner.Text(), "}")
-		data = data + "}"
+	        if !strings.HasSuffix(data, "}") {
+		    data = data + "}"
+		}
 
 		var ollamaResponse ChatResponse
 		err := json.Unmarshal([]byte(data), &ollamaResponse)