Fix issue where chatsStream() throws incorrect error

MacPaw · Jul 6, 2023 · e96abc9 · e96abc9
1 parent d7f478e
commit e96abc9
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/Sources/OpenAI/Private/StreamingSession.swift b/Sources/OpenAI/Private/StreamingSession.swift
@@ -62,12 +62,23 @@ final class StreamingSession<ResultType: Codable>: NSObject, Identifiable, URLSe
                 onProcessingError?(self, StreamingError.unknownContent)
                 return
             }
+
+            var apiError: Error? = nil
             do {
                 let decoder = JSONDecoder()
                 let object = try decoder.decode(ResultType.self, from: jsonData)
                 onReceiveContent?(self, object)
             } catch {
-                onProcessingError?(self, error)
+                apiError = error
+            }
+
+            if let apiError = apiError {
+                do {
+                    let decoded = try JSONDecoder().decode(APIErrorResponse.self, from: data)
+                    onProcessingError?(self, decoded)
+                } catch {
+                    onProcessingError?(self, apiError)
+                }
             }
         }
     }