@umituz/react-native-ai-fal-provider 1.0.46 → 1.0.48

package/package.json

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

package/src/index.ts

@@ -25,7 +25,7 @@ export type { FeatureModelConfig } from "./domain/constants/feature-models.const
 
 export {
   FalProvider, falProvider, falModelsService,
-  getImageFeatureModel, getVideoFeatureModel,
+  getImageFeatureModel, getVideoFeatureModel, NSFWContentError,
 } from "./infrastructure/services";
 export type { ModelFetcher } from "./infrastructure/services";
 

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

@@ -20,6 +20,7 @@ import type {
 import type { FalQueueStatus } from "../../domain/entities/fal.types";
 import { DEFAULT_FAL_CONFIG, FAL_CAPABILITIES } from "./fal-provider.constants";
 import { mapFalStatusToJobStatus } from "./fal-status-mapper";
+import { NSFWContentError } from "./nsfw-content-error";
 import { FAL_IMAGE_FEATURE_MODELS, FAL_VIDEO_FEATURE_MODELS } from "../../domain/constants/feature-models.constants";
 import {
   buildSingleImageInput,
@@ -119,17 +120,22 @@ export class FalProvider implements IAIProvider {
       console.log("[FalProvider] Subscribe started:", { model, timeoutMs });
     }
 
+    let lastStatus = "";
+
     try {
       const result = await Promise.race([
         fal.subscribe(model, {
           input,
-          logs: true,
+          logs: false,
           pollInterval: DEFAULT_FAL_CONFIG.pollInterval,
           onQueueUpdate: (update: { status: string; logs?: unknown[] }) => {
-            if (typeof __DEV__ !== "undefined" && __DEV__) {
-              console.log("[FalProvider] Queue update:", JSON.stringify(update));
-            }
             const jobStatus = mapFalStatusToJobStatus(update as unknown as FalQueueStatus);
+            if (jobStatus.status !== lastStatus) {
+              lastStatus = jobStatus.status;
+              if (typeof __DEV__ !== "undefined" && __DEV__) {
+                console.log("[FalProvider] Status:", jobStatus.status);
+              }
+            }
             options?.onQueueUpdate?.(jobStatus);
           },
         }),
@@ -142,6 +148,8 @@ export class FalProvider implements IAIProvider {
         console.log("[FalProvider] Subscribe completed:", { model });
       }
 
+      this.checkForNSFWContent(result as Record<string, unknown>);
+
       options?.onResult?.(result as T);
       return result as T;
     } finally {
@@ -163,19 +171,29 @@ export class FalProvider implements IAIProvider {
       console.log("[FalProvider] run() raw result:", JSON.stringify(result, null, 2));
       console.log("[FalProvider] run() result type:", typeof result);
       console.log("[FalProvider] run() result keys:", result ? Object.keys(result as object) : "null");
-      const r = result as Record<string, unknown>;
-      if (r?.data) {
-        console.log("[FalProvider] run() has data property, data keys:", Object.keys(r.data as object));
-      }
-      if (r?.images) {
-        console.log("[FalProvider] run() has images property, images:", JSON.stringify(r.images));
-      }
     }
 
+    this.checkForNSFWContent(result as Record<string, unknown>);
+
     options?.onProgress?.({ progress: 100, status: "COMPLETED" });
     return result as T;
   }
 
+  private checkForNSFWContent(result: Record<string, unknown>): void {
+    const nsfwConcepts = result?.has_nsfw_concepts as boolean[] | undefined;
+
+    if (nsfwConcepts && Array.isArray(nsfwConcepts)) {
+      const hasNSFW = nsfwConcepts.some((value) => value === true);
+
+      if (hasNSFW) {
+        if (typeof __DEV__ !== "undefined" && __DEV__) {
+          console.log("[FalProvider] NSFW content detected, rejecting result");
+        }
+        throw new NSFWContentError();
+      }
+    }
+  }
+
   reset(): void {
     this.apiKey = null;
     this.config = null;

package/src/infrastructure/services/index.ts

@@ -7,6 +7,7 @@ import type { ImageFeatureType, VideoFeatureType } from "@umituz/react-native-ai
 
 export { FalProvider, falProvider } from "./fal-provider";
 export { falModelsService, type FalModelConfig, type ModelFetcher } from "./fal-models.service";
+export { NSFWContentError } from "./nsfw-content-error";
 
 export function getImageFeatureModel(feature: ImageFeatureType): string {
   return FAL_IMAGE_FEATURE_MODELS[feature];

package/src/infrastructure/services/nsfw-content-error.ts

@@ -0,0 +1,24 @@
+/**
+ * NSFW Content Error
+ * Thrown when AI provider detects NSFW content in generated output
+ */
+
+import { FalErrorType } from "../../domain/entities/error.types";
+
+export class NSFWContentError extends Error {
+  public readonly type = FalErrorType.CONTENT_POLICY;
+  public readonly retryable = false;
+  public readonly messageKey = "error.nsfw_content_detected";
+
+  constructor(message?: string) {
+    super(
+      message ||
+        "The generated content was flagged as inappropriate. Please try a different prompt."
+    );
+    this.name = "NSFWContentError";
+  }
+
+  getUserMessage(): string {
+    return this.message;
+  }
+}