@umituz/react-native-ai-pruna-provider 1.0.8 → 1.0.9

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@umituz/react-native-ai-pruna-provider",
-  "version": "1.0.8",
+  "version": "1.0.9",
   "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/error.types.ts

@@ -19,12 +19,6 @@ export enum PrunaErrorType {
   UNKNOWN = "unknown",
 }
 
-export interface PrunaErrorCategory {
-  readonly type: PrunaErrorType;
-  readonly messageKey: string;
-  readonly retryable: boolean;
-}
-
 export interface PrunaErrorInfo {
   readonly type: PrunaErrorType;
   readonly messageKey: string;
@@ -35,18 +29,3 @@ export interface PrunaErrorInfo {
   readonly statusCode?: number;
 }
 
-export interface PrunaErrorMessages {
-  network?: string;
-  timeout?: string;
-  api_error?: string;
-  validation?: string;
-  content_policy?: string;
-  rate_limit?: string;
-  authentication?: string;
-  quota_exceeded?: string;
-  model_not_found?: string;
-  file_upload?: string;
-  polling_timeout?: string;
-  invalid_image?: string;
-  unknown?: string;
-}

package/src/exports/domain.ts

@@ -23,9 +23,7 @@ export type {
 
 export { PrunaErrorType } from "../domain/entities/error.types";
 export type {
-  PrunaErrorCategory,
   PrunaErrorInfo,
-  PrunaErrorMessages,
 } from "../domain/entities/error.types";
 
 export type {

package/src/infrastructure/services/pruna-api-client.ts

@@ -29,6 +29,11 @@ export async function uploadFileToStorage(
   apiKey: string,
   sessionId: string,
 ): Promise<string> {
+  // Guard: empty or whitespace-only input
+  if (!base64Data || !base64Data.trim()) {
+    throw new Error("File data is empty. Provide a base64 string or URL.");
+  }
+
   // Already a URL — return as-is
   if (base64Data.startsWith('http')) {
     generationLogCollector.log(sessionId, TAG, 'File already a URL, skipping upload');
@@ -82,9 +87,6 @@ export async function uploadFileToStorage(
   return fileUrl;
 }
 
-/** @deprecated Use uploadFileToStorage instead */
-export const uploadImageToFiles = uploadFileToStorage;
-
 /**
  * Strip base64 data URI prefix, returning raw base64 string.
  * If input is already a URL, returns it unchanged.
@@ -104,6 +106,7 @@ export async function submitPrediction(
   input: Record<string, unknown>,
   apiKey: string,
   sessionId: string,
+  signal?: AbortSignal,
 ): Promise<PrunaPredictionResponse> {
   generationLogCollector.log(sessionId, TAG, `Submitting prediction for model: ${model}`, {
     inputKeys: Object.keys(input),
@@ -120,6 +123,7 @@ export async function submitPrediction(
       'Content-Type': 'application/json',
     },
     body: JSON.stringify({ input }),
+    signal,
   });
 
   if (!response.ok) {
@@ -176,6 +180,7 @@ export async function pollForResult(
     try {
       const statusRes = await fetch(fullPollUrl, {
         headers: { 'apikey': apiKey },
+        signal,
       });
 
       if (!statusRes.ok) {

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

@@ -64,7 +64,7 @@ async function singleSubscribeAttempt<T = unknown>(
     // Notify progress: IN_PROGRESS
     options?.onProgress?.({ progress: -1, status: "IN_PROGRESS" });
 
-    const response = await submitPrediction(model, modelInput, apiKey, sessionId);
+    const response = await submitPrediction(model, modelInput, apiKey, sessionId, signal);
     let uri = extractUri(response);
 
     // If no immediate result, poll for async result
@@ -118,6 +118,10 @@ async function singleSubscribeAttempt<T = unknown>(
     }
   }
 
+  // Prevent unhandled rejection if predictionPromise loses the race
+  // (timeout or abort wins → prediction may reject later with no handler)
+  predictionPromise.catch(() => {});
+
   const resultUrl = await Promise.race(promises) as string;
   const requestId = `pruna_${model}_${Date.now()}`;
 
@@ -205,7 +209,8 @@ export async function handlePrunaSubscription<T = unknown>(
     }
   }
 
-  throw lastError;
+  // Unreachable: loop always returns or throws. TypeScript safety net.
+  throw lastError instanceof Error ? lastError : new Error("Subscription failed after all retry attempts.");
 }
 
 /**
@@ -226,7 +231,7 @@ export async function handlePrunaRun<T = unknown>(
 
   try {
     const modelInput = await buildModelInput(model, input, apiKey, sessionId);
-    const response = await submitPrediction(model, modelInput, apiKey, sessionId);
+    const response = await submitPrediction(model, modelInput, apiKey, sessionId, options?.signal);
 
     let uri = extractUri(response);
 

package/src/infrastructure/services/pruna-queue-operations.ts

@@ -5,7 +5,8 @@
 
 import type { PrunaModelId } from "../../domain/entities/pruna.types";
 import type { JobSubmission, JobStatus } from "../../domain/types";
-import { submitPrediction, extractUri } from "./pruna-api-client";
+import { submitPrediction, extractUri, resolveUri } from "./pruna-api-client";
+import { PRUNA_BASE_URL } from "./pruna-provider.constants";
 import { buildModelInput } from "./pruna-input-builder";
 import { generationLogCollector } from "../utils/log-collector";
 
@@ -62,7 +63,7 @@ export async function getJobStatus(
   statusUrl: string,
   apiKey: string,
 ): Promise<JobStatus> {
-  const fullUrl = statusUrl.startsWith('http') ? statusUrl : `https://api.pruna.ai${statusUrl}`;
+  const fullUrl = statusUrl.startsWith('http') ? statusUrl : `${PRUNA_BASE_URL}${statusUrl}`;
 
   const response = await fetch(fullUrl, {
     headers: { 'apikey': apiKey },
@@ -83,9 +84,11 @@ export async function getJobStatus(
   }
 
   if (typedData.status === 'failed') {
+    const errorMessage = typedData.error || "Generation failed during processing.";
     return {
       status: "FAILED",
       requestId: statusUrl,
+      logs: [{ message: errorMessage, level: "error" }],
     };
   }
 
@@ -104,7 +107,7 @@ export async function getJobResult<T = unknown>(
   statusUrl: string,
   apiKey: string,
 ): Promise<T> {
-  const fullUrl = statusUrl.startsWith('http') ? statusUrl : `https://api.pruna.ai${statusUrl}`;
+  const fullUrl = statusUrl.startsWith('http') ? statusUrl : `${PRUNA_BASE_URL}${statusUrl}`;
 
   const response = await fetch(fullUrl, {
     headers: { 'apikey': apiKey },
@@ -126,6 +129,5 @@ export async function getJobResult<T = unknown>(
     throw new Error("Result not ready or extraction failed.");
   }
 
-  const resolvedUri = uri.startsWith('/') ? `https://api.pruna.ai${uri}` : uri;
-  return { url: resolvedUri } as T;
+  return { url: resolveUri(uri) } as T;
 }

package/src/infrastructure/utils/log-collector.ts

@@ -25,10 +25,19 @@ interface Session {
 
 let sessionCounter = 0;
 
+/** Max concurrent sessions before auto-evicting oldest */
+const MAX_SESSIONS = 50;
+
 class GenerationLogCollector {
   private sessions = new Map<string, Session>();
 
   startSession(): string {
+    // Evict oldest sessions if limit exceeded
+    if (this.sessions.size >= MAX_SESSIONS) {
+      const oldestKey = this.sessions.keys().next().value;
+      if (oldestKey) this.sessions.delete(oldestKey);
+    }
+
     const id = `pruna_session_${++sessionCounter}_${Date.now()}`;
     this.sessions.set(id, { startTime: Date.now(), entries: [] });
     return id;

package/src/infrastructure/utils/type-guards/index.ts

@@ -27,5 +27,5 @@ export function isValidPrompt(value: unknown): value is string {
 }
 
 export function isValidTimeout(value: unknown): value is number {
-  return typeof value === 'number' && Number.isInteger(value) && value >= 1 && value <= 600000;
+  return typeof value === 'number' && Number.isInteger(value) && value >= 1 && value <= 3600000;
 }

package/src/init/createAiProviderInitModule.ts

@@ -35,6 +35,13 @@ export interface AiProviderInitModuleConfig {
    */
   dependsOn?: string[];
 
+  /**
+   * Whether to set Pruna as the active provider after initialization.
+   * When false, registers the provider but doesn't make it active.
+   * @default true
+   */
+  setAsActive?: boolean;
+
   /**
    * Optional callback called after provider is initialized
    */
@@ -51,6 +58,7 @@ export function createAiProviderInitModule(
     getApiKey,
     critical = false,
     dependsOn = ['firebase'],
+    setAsActive = true,
     onInitialized,
   } = config;
 
@@ -71,7 +79,9 @@ export function createAiProviderInitModule(
         if (!providerRegistry.hasProvider(prunaProvider.providerId)) {
           providerRegistry.register(prunaProvider);
         }
-        providerRegistry.setActiveProvider(prunaProvider.providerId);
+        if (setAsActive) {
+          providerRegistry.setActiveProvider(prunaProvider.providerId);
+        }
 
         if (onInitialized) {
           onInitialized();

package/src/presentation/hooks/use-pruna-generation.ts

@@ -49,8 +49,10 @@ export function usePrunaGeneration<T = unknown>(
   const [error, setError] = useState<PrunaErrorInfo | null>(null);
   const [isLoading, setIsLoading] = useState(false);
   const [isCancelling, setIsCancelling] = useState(false);
+  const [requestId, setRequestId] = useState<string | null>(null);
 
   const stateManagerRef = useRef<PrunaGenerationStateManager<T> | null>(null);
+  const abortControllerRef = useRef<AbortController | null>(null);
   const optionsRef = useRef(options);
 
   useEffect(() => {
@@ -73,12 +75,10 @@ export function usePrunaGeneration<T = unknown>(
         stateManagerRef.current = null;
       }
 
-      if (prunaProvider.hasRunningRequest()) {
-        try {
-          prunaProvider.cancelCurrentRequest();
-        } catch (error) {
-          console.warn('[usePrunaGeneration] Error cancelling request on unmount:', error);
-        }
+      // Cancel only this hook's active request on unmount
+      if (abortControllerRef.current) {
+        abortControllerRef.current.abort();
+        abortControllerRef.current = null;
       }
     };
   }, []);
@@ -88,10 +88,18 @@ export function usePrunaGeneration<T = unknown>(
       const stateManager = stateManagerRef.current;
       if (!stateManager || !stateManager.checkMounted()) return null;
 
+      // Cancel any previous in-flight request from this hook
+      if (abortControllerRef.current) {
+        abortControllerRef.current.abort();
+      }
+      const controller = new AbortController();
+      abortControllerRef.current = controller;
+
       stateManager.setLastRequest(model, input);
       setIsLoading(true);
       setError(null);
       setData(null);
+      setRequestId(null);
       stateManager.setCurrentRequestId(null);
       setIsCancelling(false);
 
@@ -104,6 +112,11 @@ export function usePrunaGeneration<T = unknown>(
               stateManager.getCurrentRequestId()
             );
             stateManager.handleQueueUpdate(prunaStatus);
+
+            // Update reactive requestId from queue status
+            if (status.requestId) {
+              setRequestId(status.requestId);
+            }
           },
         });
 
@@ -121,6 +134,10 @@ export function usePrunaGeneration<T = unknown>(
           setIsLoading(false);
           setIsCancelling(false);
         }
+        // Clean up controller reference
+        if (abortControllerRef.current === controller) {
+          abortControllerRef.current = null;
+        }
       }
     },
     []
@@ -137,9 +154,10 @@ export function usePrunaGeneration<T = unknown>(
   }, [generate]);
 
   const cancel = useCallback(() => {
-    if (prunaProvider.hasRunningRequest()) {
+    if (abortControllerRef.current) {
       setIsCancelling(true);
-      prunaProvider.cancelCurrentRequest();
+      abortControllerRef.current.abort();
+      abortControllerRef.current = null;
     }
   }, []);
 
@@ -149,11 +167,10 @@ export function usePrunaGeneration<T = unknown>(
     setError(null);
     setIsLoading(false);
     setIsCancelling(false);
+    setRequestId(null);
     stateManagerRef.current?.clearLastRequest();
   }, [cancel]);
 
-  const requestId = stateManagerRef.current?.getCurrentRequestId() ?? null;
-
   return {
     data,
     error,