From 26cdd97756955561c0ca09c4f92cee07a403fd35 Mon Sep 17 00:00:00 2001 From: Neel Virdy Date: Wed, 31 Jan 2024 13:10:21 -0500 Subject: [PATCH] Fix: prevent json decoding error when streaming --- Sources/OpenAI/Private/StreamingSession.swift | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/Sources/OpenAI/Private/StreamingSession.swift b/Sources/OpenAI/Private/StreamingSession.swift index a69e46cf..7b883d1f 100644 --- a/Sources/OpenAI/Private/StreamingSession.swift +++ b/Sources/OpenAI/Private/StreamingSession.swift @@ -59,9 +59,9 @@ extension StreamingSession { private func processJSON(from stringContent: String) { let jsonObjects = "\(previousChunkBuffer)\(stringContent)" .components(separatedBy: "data:") - .filter { $0.isEmpty == false } .map { $0.trimmingCharacters(in: .whitespacesAndNewlines) } - + .filter { $0.isEmpty == false } + previousChunkBuffer = "" guard jsonObjects.isEmpty == false, jsonObjects.first != streamingCompletionMarker else {