@umituz/react-native-ai-gemini-provider 1.14.14 → 1.14.16

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@umituz/react-native-ai-gemini-provider",
-  "version": "1.14.14",
+  "version": "1.14.16",
   "description": "Google Gemini AI provider for React Native applications",
   "main": "./src/index.ts",
   "types": "./src/index.ts",
@@ -40,7 +40,7 @@
     "@types/react": "~19.1.10",
     "@typescript-eslint/eslint-plugin": "^7.0.0",
     "@typescript-eslint/parser": "^7.0.0",
-    "@umituz/react-native-ai-generation-content": "^1.17.26",
+    "@umituz/react-native-ai-generation-content": "latest",
     "eslint": "^8.57.0",
     "react": "19.1.0",
     "react-native": "0.81.5",

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

@@ -14,8 +14,9 @@ import type {
   VideoFeatureType,
   ImageFeatureInputData,
   VideoFeatureInputData,
+  ProviderCapabilities,
+  RunOptions,
 } from "@umituz/react-native-ai-generation-content";
-import type { ExecutionOptions } from "./generation-executor";
 import type {
   GeminiImageInput,
   GeminiImageGenerationResult,
@@ -43,6 +44,27 @@ import {
 
 export type { GeminiProviderConfig };
 
+/**
+ * Gemini provider capabilities
+ */
+const GEMINI_CAPABILITIES: ProviderCapabilities = {
+  imageFeatures: [
+    "upscale",
+    "photo-restore",
+    "face-swap",
+    "anime-selfie",
+    "remove-background",
+    "remove-object",
+    "hd-touch-up",
+    "replace-background",
+  ] as const,
+  videoFeatures: ["ai-hug", "ai-kiss"] as const,
+  textToImage: true,
+  textToVideo: true,
+  imageToVideo: true,
+  textToVoice: false,
+};
+
 export class GeminiProvider implements IAIProvider {
   readonly providerId = "gemini";
   readonly providerName = "Google Gemini";
@@ -55,6 +77,18 @@ export class GeminiProvider implements IAIProvider {
     return providerInitializer.isInitialized();
   }
 
+  getCapabilities(): ProviderCapabilities {
+    return GEMINI_CAPABILITIES;
+  }
+
+  isFeatureSupported(feature: ImageFeatureType | VideoFeatureType): boolean {
+    const capabilities = this.getCapabilities();
+    return (
+      capabilities.imageFeatures.includes(feature as ImageFeatureType) ||
+      capabilities.videoFeatures.includes(feature as VideoFeatureType)
+    );
+  }
+
   submitJob(model: string, input: Record<string, unknown>): Promise<JobSubmission> {
     return jobProcessor.submitJob(model, input);
   }
@@ -75,10 +109,12 @@ export class GeminiProvider implements IAIProvider {
     options?.onQueueUpdate?.({ status: "IN_QUEUE" });
 
     const result = await generationExecutor.executeGeneration<T>(model, input, {
-      onProgress: options?.onProgress,
+      onProgress: (progress: number) => {
+        options?.onProgress?.({ progress, status: "IN_PROGRESS" });
+      },
     });
 
-    options?.onProgress?.(100);
+    options?.onProgress?.({ progress: 100, status: "COMPLETED" });
     options?.onQueueUpdate?.({ status: "COMPLETED" });
     options?.onResult?.(result);
 
@@ -88,9 +124,13 @@ export class GeminiProvider implements IAIProvider {
   async run<T = unknown>(
     model: string,
     input: Record<string, unknown>,
-    options?: ExecutionOptions,
+    options?: RunOptions,
   ): Promise<T> {
-    return generationExecutor.executeGeneration<T>(model, input, options);
+    return generationExecutor.executeGeneration<T>(model, input, {
+      onProgress: (progress: number) => {
+        options?.onProgress?.({ progress, status: "IN_PROGRESS" });
+      },
+    });
   }
 
   async generateImage(prompt: string): Promise<GeminiImageGenerationResult> {

package/src/infrastructure/services/provider-initializer.ts

@@ -32,9 +32,9 @@ export class ProviderInitializer {
             baseDelay: config.baseDelay,
             maxDelay: config.maxDelay,
             defaultTimeoutMs: config.defaultTimeoutMs,
-            textModel: config.textModel as string | undefined,
-            textToImageModel: config.textToImageModel as string | undefined,
-            imageEditModel: config.imageEditModel as string | undefined,
+            textModel: config.textModel,
+            textToImageModel: config.textToImageModel,
+            imageEditModel: config.imageEditModel,
         };
 
         geminiClientCoreService.initialize(geminiConfig);