From 816f5fc9f61f4aaaf8236ad1fb89354a0fc9323a Mon Sep 17 00:00:00 2001
From: MotorBottle <71703952+MotorBottle@users.noreply.github.com>
Date: Fri, 26 Jul 2024 17:41:19 -0400
Subject: [PATCH] Improved data process logic

---
 relay/adaptor/ollama/main.go | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/relay/adaptor/ollama/main.go b/relay/adaptor/ollama/main.go
index d36b3cf6..322e85ed 100644
--- a/relay/adaptor/ollama/main.go
+++ b/relay/adaptor/ollama/main.go
@@ -121,6 +121,9 @@ func StreamHandler(c *gin.Context, resp *http.Response) (*model.ErrorWithStatusC
 
 	for scanner.Scan() {
 		data := scanner.Text()
+		if strings.HasPrefix(data, "}") {
+		    data = strings.TrimPrefix(data, "}") + "}"
+		}
 
 		var ollamaResponse ChatResponse
 		err := json.Unmarshal([]byte(data), &ollamaResponse)