@umituz/react-native-ai-gemini-provider 1.14.8 → 1.14.9

package/package.json

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

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

@@ -10,6 +10,7 @@ import type {
   JobSubmission,
   JobStatus,
   SubscribeOptions,
+  RunOptions,
   ImageFeatureType,
   VideoFeatureType,
   ImageFeatureInputData,
@@ -72,9 +73,10 @@ export class GeminiProvider implements IAIProvider {
     options?: SubscribeOptions<T>,
   ): Promise<T> {
     options?.onQueueUpdate?.({ status: "IN_QUEUE" });
-    options?.onProgress?.(10);
 
-    const result = await generationExecutor.executeGeneration<T>(model, input);
+    const result = await generationExecutor.executeGeneration<T>(model, input, {
+      onProgress: options?.onProgress,
+    });
 
     options?.onProgress?.(100);
     options?.onQueueUpdate?.({ status: "COMPLETED" });
@@ -86,8 +88,11 @@ export class GeminiProvider implements IAIProvider {
   async run<T = unknown>(
     model: string,
     input: Record<string, unknown>,
+    options?: RunOptions,
   ): Promise<T> {
-    return generationExecutor.executeGeneration<T>(model, input);
+    return generationExecutor.executeGeneration<T>(model, input, {
+      onProgress: options?.onProgress,
+    });
   }
 
   async generateImage(prompt: string): Promise<GeminiImageGenerationResult> {

package/src/infrastructure/services/generation-executor.ts

@@ -6,6 +6,7 @@
 import type {
     GeminiImageInput,
     VideoGenerationInput,
+    VideoGenerationProgress,
 } from "../../domain/entities";
 import { geminiTextGenerationService } from "./gemini-text-generation.service";
 import { geminiImageGenerationService } from "./gemini-image-generation.service";
@@ -13,6 +14,10 @@ import { geminiVideoGenerationService } from "./gemini-video-generation.service"
 import { ContentBuilder } from "../content/ContentBuilder";
 import { ResponseFormatter } from "../response/ResponseFormatter";
 
+export interface ExecutionOptions {
+    onProgress?: (progress: number) => void;
+}
+
 export class GenerationExecutor {
     private contentBuilder = new ContentBuilder();
     private responseFormatter = new ResponseFormatter();
@@ -20,12 +25,13 @@ export class GenerationExecutor {
     async executeGeneration<T>(
         model: string,
         input: Record<string, unknown>,
+        options?: ExecutionOptions,
     ): Promise<T> {
         const isImageGeneration = input.generateImage === true || input.type === "image";
         const isVideoGeneration = this.isVideoModel(model) || input.type === "video";
 
         if (isVideoGeneration) {
-            return this.executeVideoGeneration<T>(input);
+            return this.executeVideoGeneration<T>(input, options);
         }
 
         if (isImageGeneration) {
@@ -55,7 +61,10 @@ export class GenerationExecutor {
     /**
      * Execute video generation using Veo API
      */
-    private async executeVideoGeneration<T>(input: Record<string, unknown>): Promise<T> {
+    private async executeVideoGeneration<T>(
+        input: Record<string, unknown>,
+        options?: ExecutionOptions,
+    ): Promise<T> {
         const videoInput: VideoGenerationInput = {
             prompt: String(input.prompt || ""),
             image: input.image as string | undefined,
@@ -65,7 +74,11 @@ export class GenerationExecutor {
             },
         };
 
-        const result = await geminiVideoGenerationService.generateVideo(videoInput);
+        const onProgress = options?.onProgress
+            ? (p: VideoGenerationProgress) => options.onProgress?.(p.progress)
+            : undefined;
+
+        const result = await geminiVideoGenerationService.generateVideo(videoInput, onProgress);
 
         return {
             video: { url: result.videoUrl },