Add error handling to the image generation request

Automattic · May 16, 2024 · caa378b · caa378b
1 parent 91f70de
commit caa378b
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/projects/js-packages/ai-client/src/hooks/use-image-generator/index.ts b/projects/js-packages/ai-client/src/hooks/use-image-generator/index.ts
@@ -1,6 +1,7 @@
 /**
  * External dependencies
  */
+import { __ } from '@wordpress/i18n';
 import debugFactory from 'debug';
 /**
  * Internal dependencies
@@ -193,6 +194,16 @@ const useImageGenerator = () => {
 				}
 			);
 
+			if ( ! response?.ok ) {
+				debug( 'Error generating image: %o', response );
+				return Promise.reject( {
+					data: {
+						status: response.status,
+					},
+					message: __( 'Error generating image. Please try again later.', 'jetpack-ai-client' ),
+				} );
+			}
+
 			const blob = await response.blob();
 
 			/**