@umituz/react-native-ai-gemini-provider 1.0.6 → 1.1.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@umituz/react-native-ai-gemini-provider",
-  "version": "1.0.6",
+  "version": "1.1.0",
   "description": "Google Gemini AI provider for React Native applications",
   "main": "src/index.ts",
   "types": "src/index.ts",

package/src/infrastructure/services/gemini-client.service.ts

@@ -25,6 +25,37 @@ const DEFAULT_CONFIG: Partial<GeminiConfig> = {
   imageModel: "gemini-2.0-flash-exp",
 };
 
+const RETRYABLE_ERROR_PATTERNS = [
+  "rate limit",
+  "too many requests",
+  "429",
+  "500",
+  "502",
+  "503",
+  "504",
+  "timeout",
+  "network",
+  "econnrefused",
+  "fetch failed",
+];
+
+function isRetryableError(error: unknown): boolean {
+  const message = error instanceof Error ? error.message.toLowerCase() : String(error).toLowerCase();
+  return RETRYABLE_ERROR_PATTERNS.some((pattern) => message.includes(pattern));
+}
+
+function sleep(ms: number): Promise<void> {
+  return new Promise((resolve) => setTimeout(resolve, ms));
+}
+
+function extractBase64Data(base64String: string): string {
+  if (!base64String.includes(",")) {
+    return base64String;
+  }
+  const parts = base64String.split(",");
+  return parts[1] ?? parts[0] ?? base64String;
+}
+
 class GeminiClientService {
   private client: GoogleGenerativeAI | null = null;
   private config: GeminiConfig | null = null;
@@ -98,10 +129,12 @@ class GeminiClientService {
       }),
     }));
 
-    const result = await genModel.generateContent({
-      contents: sdkContents as Parameters<typeof genModel.generateContent>[0] extends { contents: infer C } ? C : never,
-      generationConfig,
-    });
+    const result = await this.executeWithRetry(() =>
+      genModel.generateContent({
+        contents: sdkContents as Parameters<typeof genModel.generateContent>[0] extends { contents: infer C } ? C : never,
+        generationConfig,
+      }),
+    );
 
     const response = result.response;
 
@@ -159,15 +192,10 @@ class GeminiClientService {
     const parts: GeminiContent["parts"] = [{ text: prompt }];
 
     for (const image of images) {
-      // Remove data URL prefix if present
-      const base64Data = image.base64.includes(",")
-        ? image.base64.split(",")[1]
-        : image.base64;
-
       parts.push({
         inlineData: {
           mimeType: image.mimeType,
-          data: base64Data,
+          data: extractBase64Data(image.base64),
         },
       });
     }
@@ -287,6 +315,33 @@ class GeminiClientService {
     return fullText;
   }
 
+  private async executeWithRetry<T>(
+    operation: () => Promise<T>,
+    retryCount = 0,
+  ): Promise<T> {
+    const maxRetries = this.config?.maxRetries ?? 3;
+    const baseDelay = this.config?.baseDelay ?? 1000;
+    const maxDelay = this.config?.maxDelay ?? 10000;
+
+    try {
+      return await operation();
+    } catch (error) {
+      if (!isRetryableError(error) || retryCount >= maxRetries) {
+        throw error;
+      }
+
+      const delay = Math.min(baseDelay * Math.pow(2, retryCount), maxDelay);
+
+      if (typeof __DEV__ !== "undefined" && __DEV__) {
+        // eslint-disable-next-line no-console
+        console.log(`[Gemini] Retry ${retryCount + 1}/${maxRetries} after ${delay}ms`);
+      }
+
+      await sleep(delay);
+      return this.executeWithRetry(operation, retryCount + 1);
+    }
+  }
+
   private extractTextFromResponse(response: GeminiResponse): string {
     const candidate = response.candidates?.[0];
 

package/src/infrastructure/services/gemini-provider.service.ts

@@ -13,6 +13,14 @@ import { geminiClientService } from "./gemini-client.service";
 
 declare const __DEV__: boolean;
 
+function extractBase64Data(base64String: string): string {
+  if (!base64String.includes(",")) {
+    return base64String;
+  }
+  const parts = base64String.split(",");
+  return parts[1] ?? parts[0] ?? base64String;
+}
+
 export interface AIProviderConfig {
   apiKey: string;
   maxRetries?: number;
@@ -262,13 +270,10 @@ class GeminiProviderService {
       // Handle multiple images
       if (Array.isArray(input.images)) {
         for (const img of input.images as GeminiImageInput[]) {
-          const base64Data = img.base64.includes(",")
-            ? img.base64.split(",")[1]
-            : img.base64;
           parts.push({
             inlineData: {
               mimeType: img.mimeType,
-              data: base64Data,
+              data: extractBase64Data(img.base64),
             },
           });
         }