update

Author: swiftuiux

Sources/openai-async-image-swiftui/OpenAIAsyncImage.swift

@@ -133,18 +133,11 @@ public struct OpenAIAsyncImage<Content: View, T: IOpenAILoader>: View {
         _ prompt: String,
         with size: ImageSize,
         model: DalleModel
-    ) async -> Image? {
-        do {
+    ) async throws -> Image? {
             if let loader = loader {
                 return try await loader.load(prompt, with: size, model: model)
             }
             return try await loadImageDefault(prompt, with: size, model: model)
-        } catch {
-            if !Task.isCancelled {
-                self.error = error
-            }
-            return nil
-        }
     }
 
     /// Sets the image on the main thread
@@ -171,8 +164,14 @@ public struct OpenAIAsyncImage<Content: View, T: IOpenAILoader>: View {
     /// - Returns: A task that fetches the OpenAI image
     private func getTask() -> Task<Void, Never>{
         Task{
-            if let image = await loadImage(prompt, with: size, model: model){
-                await setImage(image)
+            do{
+                if let image = try await loadImage(prompt, with: size, model: model){
+                    setImage(image)
+                }
+            }catch is CancellationError{
+                self.error = AsyncImageErrors.cancellationError
+            }catch{
+                self.error = error
             }
         }
     }

Sources/openai-async-image-swiftui/enum/AsyncImageErrors.swift

@@ -16,6 +16,7 @@ enum AsyncImageErrors: Error {
     case returnedNoImages        // No images were returned in the response
     case httpStatus(String)      // HTTP status error with a message
     case responseError(Error)    // Generic response error
+    case cancellationError
 }
 
 @available(iOS 15.0, macOS 12.0, tvOS 15.0, watchOS 8.0, *)
@@ -32,6 +33,8 @@ extension AsyncImageErrors: LocalizedError {
             return NSLocalizedString(description, comment: "")
         case .responseError(let error):
             return error.localizedDescription
+        case .cancellationError:
+            return NSLocalizedString("Cancellation error.", comment: "")
         }
     }
 }