@umituz/react-native-ai-pruna-provider 1.0.4 → 1.0.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@umituz/react-native-ai-pruna-provider",
-  "version": "1.0.4",
+  "version": "1.0.5",
   "description": "Pruna AI provider for React Native - implements IAIProvider interface for unified AI generation",
   "main": "./src/index.ts",
   "types": "./src/index.ts",

package/src/domain/entities/pruna.types.ts

@@ -79,12 +79,16 @@ export interface PrunaPredictionInput {
   readonly aspect_ratio?: PrunaAspectRatio;
   readonly image?: string;
   readonly images?: readonly string[];
+  readonly reference_image?: string;
   readonly duration?: number;
   readonly resolution?: PrunaResolution;
   readonly fps?: number;
   readonly draft?: boolean;
   readonly prompt_upsampling?: boolean;
   readonly seed?: number;
+  readonly disable_safety_checker?: boolean;
+  readonly width?: number;
+  readonly height?: number;
 }
 
 /**

package/src/infrastructure/services/pruna-input-builder.ts

@@ -54,9 +54,10 @@ function buildImageInput(
 ): Record<string, unknown> {
   const payload: Record<string, unknown> = { prompt, aspect_ratio: aspectRatio };
 
-  if (input.seed !== undefined) {
-    payload.seed = input.seed;
-  }
+  if (input.seed !== undefined) payload.seed = input.seed;
+  if (input.disable_safety_checker !== undefined) payload.disable_safety_checker = input.disable_safety_checker;
+  if (input.width !== undefined) payload.width = input.width;
+  if (input.height !== undefined) payload.height = input.height;
 
   return payload;
 }
@@ -86,8 +87,14 @@ function buildImageEditInput(
 
   const payload: Record<string, unknown> = { images, prompt, aspect_ratio: aspectRatio };
 
-  if (input.seed !== undefined) {
-    payload.seed = input.seed;
+  if (input.seed !== undefined) payload.seed = input.seed;
+  if (input.disable_safety_checker !== undefined) payload.disable_safety_checker = input.disable_safety_checker;
+  if (input.width !== undefined) payload.width = input.width;
+  if (input.height !== undefined) payload.height = input.height;
+
+  // reference_image: designates the primary/main image in multi-image edits
+  if (typeof input.reference_image === 'string') {
+    payload.reference_image = stripBase64Prefix(input.reference_image);
   }
 
   return payload;
@@ -114,14 +121,21 @@ async function buildVideoInput(
   const resolution = (input.resolution as PrunaResolution) ?? P_VIDEO_DEFAULTS.resolution;
   const draft = (input.draft as boolean) ?? P_VIDEO_DEFAULTS.draft;
 
-  return {
+  const fps = (input.fps as number) ?? P_VIDEO_DEFAULTS.fps;
+  const promptUpsampling = (input.prompt_upsampling as boolean) ?? P_VIDEO_DEFAULTS.promptUpsampling;
+
+  const payload: Record<string, unknown> = {
     image: fileUrl,
     prompt,
     duration,
     resolution,
-    fps: P_VIDEO_DEFAULTS.fps,
+    fps,
     draft,
     aspect_ratio: aspectRatio,
-    prompt_upsampling: P_VIDEO_DEFAULTS.promptUpsampling,
+    prompt_upsampling: promptUpsampling,
   };
+
+  if (input.disable_safety_checker !== undefined) payload.disable_safety_checker = input.disable_safety_checker;
+
+  return payload;
 }