@umituz/react-native-ai-pruna-provider 1.0.12 → 1.0.14

package/package.json

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

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

@@ -68,21 +68,29 @@ function buildImageEditInput(
   input: Record<string, unknown>,
   sessionId: string,
 ): Record<string, unknown> {
-  // p-image-edit expects images array (base64 or HTTPS URLs — file URIs resolved by ai-generation-content)
+  // p-image-edit expects images array (base64 with data URI prefix or HTTPS URLs)
+  // Base64 format: data:image/jpeg;base64,{base64_string}
   let images: string[];
 
+  const ensureDataUriPrefix = (img: string): string => {
+    if (img.startsWith('http')) return img; // Already a URL
+    if (img.startsWith('data:')) return img; // Already has data URI prefix
+    // Add data URI prefix for raw base64
+    return `data:image/jpeg;base64,${img}`;
+  };
+
   if (Array.isArray(input.images)) {
     const validImages = (input.images as unknown[]).filter((img): img is string => typeof img === 'string');
     if (validImages.length === 0) {
       throw new Error("Image array is empty or contains no valid strings for p-image-edit.");
     }
-    images = validImages.map(stripBase64Prefix);
+    images = validImages.map(ensureDataUriPrefix);
   } else if (typeof input.image === 'string') {
-    images = [stripBase64Prefix(input.image as string)];
+    images = [ensureDataUriPrefix(input.image as string)];
   } else if (typeof input.image_url === 'string') {
-    images = [stripBase64Prefix(input.image_url as string)];
+    images = [ensureDataUriPrefix(input.image_url as string)];
   } else if (Array.isArray(input.image_urls)) {
-    images = (input.image_urls as string[]).map(stripBase64Prefix);
+    images = (input.image_urls as string[]).map(ensureDataUriPrefix);
   } else {
     throw new Error("Image is required for p-image-edit. Provide 'image', 'images', 'image_url', or 'image_urls'.");
   }