@umituz/react-native-ai-generation-content 1.72.23 → 1.72.25

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@umituz/react-native-ai-generation-content",
-  "version": "1.72.23",
+  "version": "1.72.25",
   "description": "Provider-agnostic AI generation orchestration for React Native with result preview components",
   "main": "src/index.ts",
   "types": "src/index.ts",

package/src/domains/generation/wizard/presentation/components/WizardFlow.types.ts

@@ -17,6 +17,8 @@ export interface BaseWizardFlowProps {
   readonly onGenerationComplete?: () => void;
   /** Called on generation error */
   readonly onGenerationError?: (error: string) => void;
+  /** Called when access check passes and generation is about to start - use for credit deduction */
+  readonly onCreditDeduction?: () => void | Promise<void>;
   /** Called when back is pressed on first step */
   readonly onBack: () => void;
   /** Translation function */

package/src/domains/generation/wizard/presentation/hooks/useVideoQueueGeneration.ts

@@ -22,6 +22,7 @@ export function useVideoQueueGeneration(props: UseVideoQueueGenerationProps): Us
   const pollingRef = useRef<ReturnType<typeof setInterval> | null>(null);
   const isGeneratingRef = useRef(false);
   const isPollingRef = useRef(false);
+  const consecutiveErrorsRef = useRef(0);
   const [isGenerating, setIsGenerating] = useState(false);
 
   useEffect(() => {
@@ -33,6 +34,7 @@ export function useVideoQueueGeneration(props: UseVideoQueueGenerationProps): Us
       // Reset all refs on unmount
       isGeneratingRef.current = false;
       isPollingRef.current = false;
+      consecutiveErrorsRef.current = 0;
       creationIdRef.current = null;
       requestIdRef.current = null;
       modelRef.current = null;
@@ -46,6 +48,7 @@ export function useVideoQueueGeneration(props: UseVideoQueueGenerationProps): Us
     modelRef.current = null;
     isGeneratingRef.current = false;
     isPollingRef.current = false;
+    consecutiveErrorsRef.current = 0;
     setIsGenerating(false);
   }, []);
 
@@ -109,6 +112,7 @@ export function useVideoQueueGeneration(props: UseVideoQueueGenerationProps): Us
       model,
       isPollingRef,
       pollingRef,
+      consecutiveErrorsRef,
       onComplete: handleComplete,
       onError: handleError,
     });

package/src/domains/generation/wizard/presentation/hooks/videoQueuePoller.ts

@@ -4,22 +4,23 @@ import { QUEUE_STATUS } from "../../../../../domain/constants/queue-status.const
 
 declare const __DEV__: boolean;
 
+/** Max consecutive transient errors before aborting */
+const MAX_CONSECUTIVE_ERRORS = 5;
+
 interface PollParams {
   requestId: string;
   model: string;
   isPollingRef: React.MutableRefObject<boolean>;
   pollingRef: React.MutableRefObject<ReturnType<typeof setInterval> | null>;
+  consecutiveErrorsRef: React.MutableRefObject<number>;
   onComplete: (urls: GenerationUrls) => Promise<void>;
   onError: (error: string) => Promise<void>;
 }
 
 export const pollQueueStatus = async (params: PollParams): Promise<void> => {
-  const { requestId, model, isPollingRef, pollingRef, onComplete, onError } = params;
+  const { requestId, model, isPollingRef, pollingRef, consecutiveErrorsRef, onComplete, onError } = params;
 
-  // Check-and-set - while not truly atomic in JS, this is best we can do
-  // The ref prevents most race conditions in practice
   if (isPollingRef.current) {
-    if (__DEV__) console.log("[VideoQueuePoller] Already polling, skipping");
     return;
   }
   isPollingRef.current = true;
@@ -29,10 +30,14 @@ export const pollQueueStatus = async (params: PollParams): Promise<void> => {
     isPollingRef.current = false;
     return;
   }
+
   try {
     const status = await provider.getJobStatus(model, requestId);
     if (__DEV__) console.log("[VideoQueueGeneration] Poll:", status.status);
 
+    // Reset consecutive errors on successful poll
+    consecutiveErrorsRef.current = 0;
+
     if (status.status === QUEUE_STATUS.COMPLETED || status.status === QUEUE_STATUS.FAILED) {
       if (pollingRef.current) {
         clearInterval(pollingRef.current);
@@ -53,13 +58,26 @@ export const pollQueueStatus = async (params: PollParams): Promise<void> => {
       }
     }
   } catch (err) {
-    if (pollingRef.current) {
-      clearInterval(pollingRef.current);
-      pollingRef.current = null;
-    }
+    consecutiveErrorsRef.current += 1;
     const errorMessage = err instanceof Error ? err.message : "Generation failed";
-    if (__DEV__) console.error("[VideoQueueGeneration] Poll error:", errorMessage);
-    await onError(errorMessage);
+
+    if (consecutiveErrorsRef.current >= MAX_CONSECUTIVE_ERRORS) {
+      // Too many consecutive errors - abort
+      if (pollingRef.current) {
+        clearInterval(pollingRef.current);
+        pollingRef.current = null;
+      }
+      if (__DEV__) console.error("[VideoQueueGeneration] Max consecutive errors reached, aborting:", errorMessage);
+      await onError(errorMessage);
+    } else {
+      // Transient error - continue polling
+      if (__DEV__) {
+        console.warn(
+          `[VideoQueueGeneration] Transient poll error (${consecutiveErrorsRef.current}/${MAX_CONSECUTIVE_ERRORS}):`,
+          errorMessage,
+        );
+      }
+    }
   } finally {
     isPollingRef.current = false;
   }

package/src/domains/image-to-video/presentation/screens/ImageToVideoWizardFlow.tsx

@@ -27,6 +27,7 @@ export const ImageToVideoWizardFlow: React.FC<ImageToVideoWizardFlowProps> = (pr
     onNetworkError,
     onGenerationComplete,
     onGenerationError,
+    onCreditDeduction,
     onBack,
     t,
     alertMessages,
@@ -37,6 +38,7 @@ export const ImageToVideoWizardFlow: React.FC<ImageToVideoWizardFlowProps> = (pr
   const { requireFeature } = useAIFeatureGate({
     creditCost,
     onNetworkError,
+    onSuccess: onCreditDeduction,
   });
 
   const scenario = useMemo(

package/src/domains/text-to-image/presentation/screens/TextToImageWizardFlow.tsx

@@ -26,6 +26,7 @@ export const TextToImageWizardFlow: React.FC<TextToImageWizardFlowProps> = (prop
     onNetworkError,
     onGenerationComplete,
     onGenerationError,
+    onCreditDeduction,
     onBack,
     t,
     alertMessages,
@@ -36,6 +37,7 @@ export const TextToImageWizardFlow: React.FC<TextToImageWizardFlowProps> = (prop
   const { requireFeature } = useAIFeatureGate({
     creditCost,
     onNetworkError,
+    onSuccess: onCreditDeduction,
   });
 
   if (!alertMessages) {

package/src/domains/text-to-video/presentation/screens/TextToVideoWizardFlow.tsx

@@ -27,6 +27,7 @@ export const TextToVideoWizardFlow: React.FC<TextToVideoWizardFlowProps> = (prop
     onNetworkError,
     onGenerationComplete,
     onGenerationError,
+    onCreditDeduction,
     onBack,
     t,
     alertMessages,
@@ -37,6 +38,7 @@ export const TextToVideoWizardFlow: React.FC<TextToVideoWizardFlowProps> = (prop
   const { requireFeature } = useAIFeatureGate({
     creditCost,
     onNetworkError,
+    onSuccess: onCreditDeduction,
   });
 
   const scenario = useMemo(