Merge pull request #151 from oblank/main

fix: fix json decode error when jsonContent is emtpy
MacPaw · Feb 2, 2024 · 38cda2a · 38cda2a
2 parents 497266a + c95da8e
commit 38cda2a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/Sources/OpenAI/Private/StreamingSession.swift b/Sources/OpenAI/Private/StreamingSession.swift
@@ -57,6 +57,9 @@ final class StreamingSession<ResultType: Codable>: NSObject, Identifiable, URLSe
 extension StreamingSession {
 
     private func processJSON(from stringContent: String) {
+        if stringContent.isEmpty {
+            return
+        }
         let jsonObjects = "\(previousChunkBuffer)\(stringContent)"
             .components(separatedBy: "data:")
             .filter { $0.isEmpty == false }
@@ -68,7 +71,7 @@ extension StreamingSession {
             return
         }
         jsonObjects.enumerated().forEach { (index, jsonContent)  in
-            guard jsonContent != streamingCompletionMarker else {
+            guard jsonContent != streamingCompletionMarker && !jsonContent.isEmpty else {
                 return
             }
             guard let jsonData = jsonContent.data(using: .utf8) else {