@umituz/react-native-ai-pruna-provider 1.0.38 → 1.0.40

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@umituz/react-native-ai-pruna-provider",
-  "version": "1.0.38",
+  "version": "1.0.40",
   "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

@@ -188,15 +188,31 @@ async function buildImageEditInput(
     });
   }
 
+  // Optional parameters for quality control
   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;
 
+  // Advanced quality parameters
+  if (input.num_inference_steps !== undefined) {
+    payload.num_inference_steps = input.num_inference_steps;
+    generationLogCollector.log(sessionId, TAG, `Added num_inference_steps: ${input.num_inference_steps}`);
+  }
+  if (input.guidance_scale !== undefined) {
+    payload.guidance_scale = input.guidance_scale;
+    generationLogCollector.log(sessionId, TAG, `Added guidance_scale: ${input.guidance_scale}`);
+  }
+  if (input.strength !== undefined) {
+    payload.strength = input.strength;
+    generationLogCollector.log(sessionId, TAG, `Added strength: ${input.strength}`);
+  }
+
   generationLogCollector.log(sessionId, TAG, `<<< buildImageEditInput COMPLETE`, {
     payloadKeys: Object.keys(payload),
     imageCount: imageUrls.length,
     hasSeed: !!payload.seed,
+    hasQualityParams: !!(payload.num_inference_steps || payload.guidance_scale || payload.strength),
   });
 
   return payload;

package/src/infrastructure/services/pruna-provider-subscription.ts

@@ -157,9 +157,14 @@ async function singleSubscribeAttempt<T = unknown>(
     requestId,
   });
 
-  // Wrap result in expected format
-  // Pruna returns a URL string — wrap in a structure similar to FAL's response
-  const result = { url: resultUrl } as T;
+  // Wrap result in expected format for AI generation content
+  // Pruna returns a URL string — wrap in { images: [{ url }] } format
+  const result = { images: [{ url: resultUrl }] } as T;
+
+  generationLogCollector.log(sessionId, TAG, `Result wrapped for AI generation content`, {
+    format: 'images: [{ url }]',
+    urlPrefix: resultUrl.substring(0, 80) + '...',
+  });
 
   return { result, requestId };
 }
@@ -282,7 +287,9 @@ export async function handlePrunaRun<T = unknown>(
     generationLogCollector.log(sessionId, runTag, `Run completed in ${elapsed}ms`);
 
     options?.onProgress?.({ progress: 100, status: "COMPLETED" as const });
-    return { url: resultUrl } as T;
+
+    // Wrap result in expected format for AI generation content
+    return { images: [{ url: resultUrl }] } as T;
   } catch (error) {
     const elapsed = Date.now() - startTime;
     const message = error instanceof Error ? error.message : String(error);