implement errorDescription for OpenAIError

Thxonly · Apr 19, 2023 · d3b668b · d3b668b
1 parent a0f7088
commit d3b668b
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/OpenAISwift/OpenAISwift.swift b/OpenAISwift/OpenAISwift.swift
@@ -9,6 +9,15 @@ public enum OpenAIError: Error {
     case decodingError(error: Error)
 }
 
+extension OpenAIError: LocalizedError {
+    public var errorDescription: String? {
+        switch self {
+        case .genericError(let error), .decodingError(let error):
+            return error.localizedDescription
+        }
+    }
+}
+
 public class OpenAISwift {
     fileprivate(set) var token: String?
     fileprivate let config: Config
@@ -132,6 +141,9 @@ extension OpenAISwift {
                         let res = try JSONDecoder().decode(OpenAI<MessageResult>.self, from: success)
                         completionHandler(.success(res))
                     } catch {
+                        if let resp = String(data: success, encoding: .utf8) {
+                            print("Failed to decode response:\n", resp)
+                        }
                         completionHandler(.failure(.decodingError(error: error)))
                     }
                 case .failure(let failure):