@umituz/react-native-ai-fal-provider 1.0.62 → 1.0.64

package/package.json

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

package/src/domain/constants/default-models.constants.ts

@@ -5,6 +5,18 @@
 
 import type { FalModelType } from "../entities/fal.types";
 import type { ModelType } from "../types/model-selection.types";
+import { DEFAULT_TEXT_TO_IMAGE_MODELS } from "./models/text-to-image.models";
+import { DEFAULT_TEXT_TO_VOICE_MODELS } from "./models/text-to-voice.models";
+import { DEFAULT_TEXT_TO_VIDEO_MODELS } from "./models/text-to-video.models";
+import { DEFAULT_IMAGE_TO_VIDEO_MODELS } from "./models/image-to-video.models";
+import { DEFAULT_TEXT_TO_TEXT_MODELS } from "./models/text-to-text.models";
+
+// Export model lists
+export { DEFAULT_TEXT_TO_IMAGE_MODELS };
+export { DEFAULT_TEXT_TO_VOICE_MODELS };
+export { DEFAULT_TEXT_TO_VIDEO_MODELS };
+export { DEFAULT_IMAGE_TO_VIDEO_MODELS };
+export { DEFAULT_TEXT_TO_TEXT_MODELS };
 
 export interface FalModelConfig {
   readonly id: string;
@@ -20,130 +32,25 @@ export interface FalModelConfig {
   readonly order?: number;
 }
 
-export const DEFAULT_TEXT_TO_IMAGE_MODELS: FalModelConfig[] = [
-  {
-    id: "fal-ai/flux/schnell",
-    name: "Flux Schnell",
-    type: "text-to-image",
-    isDefault: true,
-    isActive: true,
-    pricing: { freeUserCost: 1, premiumUserCost: 0.5 },
-    description: "Fast and efficient text-to-image generation",
-    order: 1,
-  },
-  {
-    id: "fal-ai/flux/dev",
-    name: "Flux Dev",
-    type: "text-to-image",
-    isDefault: false,
-    isActive: true,
-    pricing: { freeUserCost: 2, premiumUserCost: 1 },
-    description: "High-quality text-to-image generation",
-    order: 2,
-  },
-  {
-    id: "fal-ai/flux-pro",
-    name: "Flux Pro",
-    type: "text-to-image",
-    isDefault: false,
-    isActive: true,
-    pricing: { freeUserCost: 3, premiumUserCost: 1.5 },
-    description: "Professional-grade text-to-image generation",
-    order: 3,
-  },
-];
-
-export const DEFAULT_TEXT_TO_VOICE_MODELS: FalModelConfig[] = [
-  {
-    id: "fal-ai/playai/tts/v3",
-    name: "PlayAI TTS v3",
-    type: "text-to-voice",
-    isDefault: true,
-    isActive: true,
-    pricing: { freeUserCost: 1, premiumUserCost: 0.5 },
-    description: "High-quality text-to-speech synthesis",
-    order: 1,
-  },
-  {
-    id: "fal-ai/eleven-labs/tts",
-    name: "ElevenLabs TTS",
-    type: "text-to-voice",
-    isDefault: false,
-    isActive: true,
-    pricing: { freeUserCost: 2, premiumUserCost: 1 },
-    description: "Premium voice synthesis with multiple voice options",
-    order: 2,
-  },
-];
-
-export const DEFAULT_TEXT_TO_VIDEO_MODELS: FalModelConfig[] = [
-  {
-    id: "fal-ai/hunyuan-video/1",
-    name: "Hunyuan",
-    type: "text-to-video",
-    isDefault: true,
-    isActive: true,
-    pricing: { freeUserCost: 10, premiumUserCost: 5 },
-    description: "High-quality video generation",
-    order: 1,
-  },
-  {
-    id: "fal-ai/minimax-video",
-    name: "MiniMax",
-    type: "text-to-video",
-    isDefault: false,
-    isActive: true,
-    pricing: { freeUserCost: 15, premiumUserCost: 8 },
-    description: "Advanced video generation with better dynamics",
-    order: 2,
-  },
-  {
-    id: "fal-ai/kling-video/v1.5/pro/text-to-video",
-    name: "Kling 1.5",
-    type: "text-to-video",
-    isDefault: false,
-    isActive: true,
-    pricing: { freeUserCost: 20, premiumUserCost: 10 },
-    description: "Professional video generation",
-    order: 3,
-  },
-  {
-    id: "fal-ai/mochi-v1",
-    name: "Mochi",
-    type: "text-to-video",
-    isDefault: false,
-    isActive: true,
-    pricing: { freeUserCost: 8, premiumUserCost: 4 },
-    description: "Fast video generation",
-    order: 4,
-  },
-];
-
-export const DEFAULT_IMAGE_TO_VIDEO_MODELS: FalModelConfig[] = [
-  {
-    id: "fal-ai/kling-video/v1.5/pro/image-to-video",
-    name: "Kling I2V",
-    type: "image-to-video",
-    isDefault: true,
-    isActive: true,
-    pricing: { freeUserCost: 15, premiumUserCost: 8 },
-    description: "High-quality image to video generation",
-    order: 1,
-  },
-];
+/**
+ * Default credit costs for each model type
+ */
+export const DEFAULT_CREDIT_COSTS: Record<ModelType, number> = {
+  "text-to-image": 2,
+  "text-to-video": 20,
+  "image-to-video": 20,
+  "text-to-voice": 3,
+} as const;
 
-export const DEFAULT_TEXT_TO_TEXT_MODELS: FalModelConfig[] = [
-  {
-    id: "fal-ai/llama-3-8b-instruct",
-    name: "Llama 3 8B Instruct",
-    type: "text-to-text",
-    isDefault: true,
-    isActive: true,
-    pricing: { freeUserCost: 0.1, premiumUserCost: 0.05 },
-    description: "Fast and reliable text generation",
-    order: 1,
-  },
-];
+/**
+ * Default model IDs for each model type
+ */
+export const DEFAULT_MODEL_IDS: Record<ModelType, string> = {
+  "text-to-image": "fal-ai/flux/schnell",
+  "text-to-video": "fal-ai/minimax-video",
+  "image-to-video": "fal-ai/kling-video/v1.5/pro/image-to-video",
+  "text-to-voice": "fal-ai/playai/tts/v3",
+} as const;
 
 /**
  * Get all default models
@@ -194,23 +101,3 @@ export function getDefaultModel(type: FalModelType): FalModelConfig | undefined
 export function findModelById(id: string): FalModelConfig | undefined {
   return getAllDefaultModels().find((m) => m.id === id);
 }
-
-/**
- * Default credit costs for each model type
- */
-export const DEFAULT_CREDIT_COSTS: Record<ModelType, number> = {
-  "text-to-image": 2,
-  "text-to-video": 20,
-  "image-to-video": 20,
-  "text-to-voice": 3,
-} as const;
-
-/**
- * Default model IDs for each model type
- */
-export const DEFAULT_MODEL_IDS: Record<ModelType, string> = {
-  "text-to-image": "fal-ai/flux/schnell",
-  "text-to-video": "fal-ai/minimax-video",
-  "image-to-video": "fal-ai/kling-video/v1.5/pro/image-to-video",
-  "text-to-voice": "fal-ai/playai/tts/v3",
-} as const;

package/src/domain/constants/models/image-to-video.models.ts

@@ -0,0 +1,18 @@
+/**
+ * Image-to-Video Models
+ */
+
+import type { FalModelConfig } from "../default-models.constants";
+
+export const DEFAULT_IMAGE_TO_VIDEO_MODELS: FalModelConfig[] = [
+  {
+    id: "fal-ai/kling-video/v1.5/pro/image-to-video",
+    name: "Kling I2V",
+    type: "image-to-video",
+    isDefault: true,
+    isActive: true,
+    pricing: { freeUserCost: 15, premiumUserCost: 8 },
+    description: "High-quality image to video generation",
+    order: 1,
+  },
+];

package/src/domain/constants/models/index.ts

@@ -0,0 +1,10 @@
+/**
+ * Model Catalog Index
+ * Exports all model lists
+ */
+
+export { DEFAULT_TEXT_TO_IMAGE_MODELS } from "./text-to-image.models";
+export { DEFAULT_TEXT_TO_VOICE_MODELS } from "./text-to-voice.models";
+export { DEFAULT_TEXT_TO_VIDEO_MODELS } from "./text-to-video.models";
+export { DEFAULT_IMAGE_TO_VIDEO_MODELS } from "./image-to-video.models";
+export { DEFAULT_TEXT_TO_TEXT_MODELS } from "./text-to-text.models";

package/src/domain/constants/models/text-to-image.models.ts

@@ -0,0 +1,38 @@
+/**
+ * Text-to-Image Models
+ */
+
+import type { FalModelConfig } from "../default-models.constants";
+
+export const DEFAULT_TEXT_TO_IMAGE_MODELS: FalModelConfig[] = [
+  {
+    id: "fal-ai/flux/schnell",
+    name: "Flux Schnell",
+    type: "text-to-image",
+    isDefault: true,
+    isActive: true,
+    pricing: { freeUserCost: 1, premiumUserCost: 0.5 },
+    description: "Fast and efficient text-to-image generation",
+    order: 1,
+  },
+  {
+    id: "fal-ai/flux/dev",
+    name: "Flux Dev",
+    type: "text-to-image",
+    isDefault: false,
+    isActive: true,
+    pricing: { freeUserCost: 2, premiumUserCost: 1 },
+    description: "High-quality text-to-image generation",
+    order: 2,
+  },
+  {
+    id: "fal-ai/flux-pro",
+    name: "Flux Pro",
+    type: "text-to-image",
+    isDefault: false,
+    isActive: true,
+    pricing: { freeUserCost: 3, premiumUserCost: 1.5 },
+    description: "Professional-grade text-to-image generation",
+    order: 3,
+  },
+];

package/src/domain/constants/models/text-to-text.models.ts

@@ -0,0 +1,18 @@
+/**
+ * Text-to-Text Models
+ */
+
+import type { FalModelConfig } from "../default-models.constants";
+
+export const DEFAULT_TEXT_TO_TEXT_MODELS: FalModelConfig[] = [
+  {
+    id: "fal-ai/llama-3-8b-instruct",
+    name: "Llama 3 8B Instruct",
+    type: "text-to-text",
+    isDefault: true,
+    isActive: true,
+    pricing: { freeUserCost: 0.1, premiumUserCost: 0.05 },
+    description: "Fast and reliable text generation",
+    order: 1,
+  },
+];

package/src/domain/constants/models/text-to-video.models.ts

@@ -0,0 +1,48 @@
+/**
+ * Text-to-Video Models
+ */
+
+import type { FalModelConfig } from "../default-models.constants";
+
+export const DEFAULT_TEXT_TO_VIDEO_MODELS: FalModelConfig[] = [
+  {
+    id: "fal-ai/hunyuan-video/1",
+    name: "Hunyuan",
+    type: "text-to-video",
+    isDefault: true,
+    isActive: true,
+    pricing: { freeUserCost: 10, premiumUserCost: 5 },
+    description: "High-quality video generation",
+    order: 1,
+  },
+  {
+    id: "fal-ai/minimax-video",
+    name: "MiniMax",
+    type: "text-to-video",
+    isDefault: false,
+    isActive: true,
+    pricing: { freeUserCost: 15, premiumUserCost: 8 },
+    description: "Advanced video generation with better dynamics",
+    order: 2,
+  },
+  {
+    id: "fal-ai/kling-video/v1.5/pro/text-to-video",
+    name: "Kling 1.5",
+    type: "text-to-video",
+    isDefault: false,
+    isActive: true,
+    pricing: { freeUserCost: 20, premiumUserCost: 10 },
+    description: "Professional video generation",
+    order: 3,
+  },
+  {
+    id: "fal-ai/mochi-v1",
+    name: "Mochi",
+    type: "text-to-video",
+    isDefault: false,
+    isActive: true,
+    pricing: { freeUserCost: 8, premiumUserCost: 4 },
+    description: "Fast video generation",
+    order: 4,
+  },
+];

package/src/domain/constants/models/text-to-voice.models.ts

@@ -0,0 +1,28 @@
+/**
+ * Text-to-Voice Models
+ */
+
+import type { FalModelConfig } from "../default-models.constants";
+
+export const DEFAULT_TEXT_TO_VOICE_MODELS: FalModelConfig[] = [
+  {
+    id: "fal-ai/playai/tts/v3",
+    name: "PlayAI TTS v3",
+    type: "text-to-voice",
+    isDefault: true,
+    isActive: true,
+    pricing: { freeUserCost: 1, premiumUserCost: 0.5 },
+    description: "High-quality text-to-speech synthesis",
+    order: 1,
+  },
+  {
+    id: "fal-ai/eleven-labs/tts",
+    name: "ElevenLabs TTS",
+    type: "text-to-voice",
+    isDefault: false,
+    isActive: true,
+    pricing: { freeUserCost: 2, premiumUserCost: 1 },
+    description: "Premium voice synthesis with multiple voice options",
+    order: 2,
+  },
+];

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

@@ -0,0 +1,109 @@
+/**
+ * FAL Provider Subscription Handlers
+ * Handles subscribe, run methods and cancellation logic
+ */
+
+import { fal } from "@fal-ai/client";
+import type { SubscribeOptions, RunOptions } from "@umituz/react-native-ai-generation-content";
+import type { FalQueueStatus } from "../../domain/entities/fal.types";
+import { DEFAULT_FAL_CONFIG } from "./fal-provider.constants";
+import { mapFalStatusToJobStatus } from "./fal-status-mapper";
+import { validateNSFWContent } from "../validators/nsfw-validator";
+
+declare const __DEV__: boolean | undefined;
+
+/**
+ * Handle FAL subscription with timeout and cancellation
+ */
+export async function handleFalSubscription<T = unknown>(
+  model: string,
+  input: Record<string, unknown>,
+  options?: SubscribeOptions<T>,
+  signal?: AbortSignal
+): Promise<{ result: T; requestId: string | null }> {
+  const timeoutMs = options?.timeoutMs ?? DEFAULT_FAL_CONFIG.defaultTimeoutMs;
+  let timeoutId: ReturnType<typeof setTimeout> | null = null;
+  let currentRequestId: string | null = null;
+
+  if (typeof __DEV__ !== "undefined" && __DEV__) {
+    console.log("[FalProvider] Subscribe started:", { model, timeoutMs });
+  }
+
+  let lastStatus = "";
+
+  try {
+    const result = await Promise.race([
+      fal.subscribe(model, {
+        input,
+        logs: false,
+        pollInterval: DEFAULT_FAL_CONFIG.pollInterval,
+        onQueueUpdate: (update: { status: string; logs?: unknown[]; request_id?: string }) => {
+          currentRequestId = update.request_id ?? null;
+          const jobStatus = mapFalStatusToJobStatus({
+            ...update as unknown as FalQueueStatus,
+            requestId: currentRequestId ?? "",
+          });
+          if (jobStatus.status !== lastStatus) {
+            lastStatus = jobStatus.status;
+            if (typeof __DEV__ !== "undefined" && __DEV__) {
+              console.log("[FalProvider] Status:", jobStatus.status, "RequestId:", currentRequestId);
+            }
+          }
+          options?.onQueueUpdate?.(jobStatus);
+        },
+      }),
+      new Promise<never>((_, reject) => {
+        timeoutId = setTimeout(() => {
+          reject(new Error("FAL subscription timeout"));
+        }, timeoutMs);
+      }),
+      // Abort promise
+      ...(signal ? [
+        new Promise<never>((_, reject) => {
+          signal.addEventListener("abort", () => {
+            reject(new Error("Request cancelled by user"));
+          });
+        }),
+      ] as const : []),
+    ]);
+
+    if (typeof __DEV__ !== "undefined" && __DEV__) {
+      console.log("[FalProvider] Subscribe completed:", { model, requestId: currentRequestId });
+    }
+
+    validateNSFWContent(result as Record<string, unknown>);
+
+    options?.onResult?.(result as T);
+    return { result: result as T, requestId: currentRequestId };
+  } finally {
+    if (timeoutId) clearTimeout(timeoutId);
+  }
+}
+
+/**
+ * Handle FAL run with NSFW validation
+ */
+export async function handleFalRun<T = unknown>(
+  model: string,
+  input: Record<string, unknown>,
+  options?: RunOptions
+): Promise<T> {
+  options?.onProgress?.({ progress: 10, status: "IN_PROGRESS" as const });
+
+  if (typeof __DEV__ !== "undefined" && __DEV__) {
+    console.log("[FalProvider] run() model:", model, "inputKeys:", Object.keys(input));
+  }
+
+  const result = await fal.run(model, { input });
+
+  if (typeof __DEV__ !== "undefined" && __DEV__) {
+    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");
+  }
+
+  validateNSFWContent(result as Record<string, unknown>);
+
+  options?.onProgress?.({ progress: 100, status: "COMPLETED" as const });
+  return result as T;
+}

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

@@ -22,7 +22,7 @@ import { DEFAULT_FAL_CONFIG, FAL_CAPABILITIES } from "./fal-provider.constants";
 import { mapFalStatusToJobStatus } from "./fal-status-mapper";
 import { FAL_IMAGE_FEATURE_MODELS, FAL_VIDEO_FEATURE_MODELS } from "../../domain/constants/feature-models.constants";
 import { buildImageFeatureInput as buildImageFeatureInputImpl, buildVideoFeatureInput as buildVideoFeatureInputImpl } from "../builders";
-import { validateNSFWContent } from "../validators/nsfw-validator";
+import { handleFalSubscription, handleFalRun } from "./fal-provider-subscription";
 
 declare const __DEV__: boolean | undefined;
 
@@ -31,13 +31,11 @@ export class FalProvider implements IAIProvider {
   readonly providerName = "FAL AI";
 
   private apiKey: string | null = null;
-  private config: AIProviderConfig | null = null;
   private initialized = false;
   private currentAbortController: AbortController | null = null;
 
   initialize(configData: AIProviderConfig): void {
     this.apiKey = configData.apiKey;
-    this.config = { ...DEFAULT_FAL_CONFIG, ...configData };
 
     fal.config({
       credentials: configData.apiKey,
@@ -105,104 +103,24 @@ export class FalProvider implements IAIProvider {
     options?: SubscribeOptions<T>,
   ): Promise<T> {
     this.validateInitialization();
-
-    // Cancel previous request if exists
     this.cancelCurrentRequest();
-
-    // Create new abort controller
     this.currentAbortController = new AbortController();
-    const { signal } = this.currentAbortController;
-
-    const timeoutMs = options?.timeoutMs ?? this.config?.defaultTimeoutMs ?? DEFAULT_FAL_CONFIG.defaultTimeoutMs;
-    let timeoutId: ReturnType<typeof setTimeout> | null = null;
-    let currentRequestId: string | null = null;
-
-    if (typeof __DEV__ !== "undefined" && __DEV__) {
-      console.log("[FalProvider] Subscribe started:", { model, timeoutMs });
-    }
-
-    let lastStatus = "";
-
-    try {
-      const result = await Promise.race([
-        fal.subscribe(model, {
-          input,
-          logs: false,
-          pollInterval: DEFAULT_FAL_CONFIG.pollInterval,
-          onQueueUpdate: (update: { status: string; logs?: unknown[]; request_id?: string }) => {
-            currentRequestId = update.request_id ?? null;
-            const jobStatus = mapFalStatusToJobStatus({
-              ...update as unknown as FalQueueStatus,
-              requestId: currentRequestId ?? "",
-            });
-            if (jobStatus.status !== lastStatus) {
-              lastStatus = jobStatus.status;
-              if (typeof __DEV__ !== "undefined" && __DEV__) {
-                console.log("[FalProvider] Status:", jobStatus.status, "RequestId:", currentRequestId);
-              }
-            }
-            options?.onQueueUpdate?.(jobStatus);
-          },
-        }),
-        new Promise<never>((_, reject) => {
-          timeoutId = setTimeout(() => {
-            reject(new Error("FAL subscription timeout"));
-          }, timeoutMs);
-        }),
-        // Abort promise
-        new Promise<never>((_, reject) => {
-          signal.addEventListener("abort", () => {
-            reject(new Error("Request cancelled by user"));
-          });
-        }),
-      ]);
-
-      if (typeof __DEV__ !== "undefined" && __DEV__) {
-        console.log("[FalProvider] Subscribe completed:", { model, requestId: currentRequestId });
-      }
-
-      validateNSFWContent(result as Record<string, unknown>);
 
-      options?.onResult?.(result as T);
-      return result as T;
-    } finally {
-      if (timeoutId) clearTimeout(timeoutId);
-      this.currentAbortController = null;
-    }
+    const { result } = await handleFalSubscription<T>(model, input, options, this.currentAbortController.signal);
+    return result;
   }
 
   async run<T = unknown>(model: string, input: Record<string, unknown>, options?: RunOptions): Promise<T> {
     this.validateInitialization();
-    options?.onProgress?.({ progress: 10, status: "IN_PROGRESS" as const });
-
-    if (typeof __DEV__ !== "undefined" && __DEV__) {
-      console.log("[FalProvider] run() model:", model, "inputKeys:", Object.keys(input));
-    }
-
-    const result = await fal.run(model, { input });
-
-    if (typeof __DEV__ !== "undefined" && __DEV__) {
-      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");
-    }
-
-    validateNSFWContent(result as Record<string, unknown>);
-
-    options?.onProgress?.({ progress: 100, status: "COMPLETED" as const });
-    return result as T;
+    return handleFalRun<T>(model, input, options);
   }
 
   reset(): void {
     this.cancelCurrentRequest();
     this.apiKey = null;
-    this.config = null;
     this.initialized = false;
   }
 
-  /**
-   * Cancel the current running request
-   */
   cancelCurrentRequest(): void {
     if (this.currentAbortController) {
       if (typeof __DEV__ !== "undefined" && __DEV__) {
@@ -213,9 +131,6 @@ export class FalProvider implements IAIProvider {
     }
   }
 
-  /**
-   * Check if there's a running request
-   */
   hasRunningRequest(): boolean {
     return this.currentAbortController !== null;
   }
@@ -237,4 +152,4 @@ export class FalProvider implements IAIProvider {
   }
 }
 
-export const falProvider = new FalProvider();
+export const falProvider = new FalProvider();