@umituz/react-native-ai-pruna-provider 1.0.65 → 1.0.66

package/package.json

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

package/src/application/services/pruna-service.ts

@@ -66,16 +66,6 @@ export class PrunaService {
     const apiKey = this.ensureInitialized();
     return generateImageEditUseCase.execute(input, apiKey.toString(), signal);
   }
-
-  reset(): void {
-    this.apiKey = null;
-    this.initialized = false;
-  }
-
-  getSessionLogs(sessionId: string): unknown[] {
-    // Return logs for debugging
-    return [];
-  }
 }
 
 export const prunaService = new PrunaService();

package/src/domain/entities/error.types.ts

@@ -16,6 +16,7 @@ export enum PrunaErrorType {
   FILE_UPLOAD = "file_upload",
   POLLING_TIMEOUT = "polling_timeout",
   INVALID_IMAGE = "invalid_image",
+  CANCELLED = "cancelled",
   UNKNOWN = "unknown",
 }
 

package/src/domain/services/error-mapper.domain-service.ts

@@ -178,18 +178,15 @@ export class ErrorMapperService {
     stack?: string,
     statusCode?: number
   ): PrunaErrorInfo {
-    const error: PrunaErrorInfo = {
+    return {
       type,
       messageKey,
       retryable,
       originalError,
+      ...(originalErrorName && { originalErrorName }),
+      ...(stack && { stack }),
+      ...(statusCode !== undefined && { statusCode }),
     };
-
-    if (originalErrorName) error.originalErrorName = originalErrorName;
-    if (stack) error.stack = stack;
-    if (statusCode !== undefined) error.statusCode = statusCode;
-
-    return error;
   }
 
   private static extractMessage(error: unknown): string {

package/src/domain/value-objects/api-key.value.ts

@@ -22,9 +22,4 @@ export class ApiKey {
   toString(): string {
     return this.value;
   }
-
-  mask(): string {
-    if (this.value.length <= 8) return '********';
-    return `${this.value.substring(0, 4)}...${this.value.substring(this.value.length - 4)}`;
-  }
 }

package/src/domain/value-objects/model-id.value.ts

@@ -26,16 +26,4 @@ export class ModelId {
   toString(): string {
     return this.value;
   }
-
-  isImage(): boolean {
-    return this.value === 'p-image';
-  }
-
-  isImageEdit(): boolean {
-    return this.value === 'p-image-edit';
-  }
-
-  isVideo(): boolean {
-    return this.value === 'p-video';
-  }
 }

package/src/domain/value-objects/session-id.value.ts

@@ -15,8 +15,4 @@ export class SessionId {
   toString(): string {
     return this.value;
   }
-
-  equals(other: SessionId): boolean {
-    return this.value === other.value;
-  }
 }

package/src/index.ts

@@ -52,6 +52,7 @@ export type { PrunaProvider as PrunaProviderType } from "./infrastructure/servic
 export {
   cleanupRequestStore,
   stopAutomaticCleanup,
+  cleanupRequestIdMappings,
 } from "./infrastructure/services/request-store";
 export type { ActiveRequest } from "./infrastructure/services/request-store";
 
@@ -102,7 +103,6 @@ export {
   MIME_AUDIO_WAV,
   MIME_AUDIO_FLAC,
   MIME_AUDIO_MP4,
-  MIME_APPLICATION_OCTET,
   MIME_DEFAULT,
   MIME_TO_EXTENSION,
   getExtensionForMime,

package/src/infrastructure/logging/pruna-logger.ts

@@ -87,14 +87,3 @@ export class PrunaLogger {
 // Singleton instance
 export const logger = PrunaLogger.getInstance();
 
-// Convenience factory function
-export function createLogger(tag: string) {
-  return {
-    info: (sessionId: string, message: string, data?: Record<string, unknown>) =>
-      logger.info(sessionId, tag, message, data),
-    warn: (sessionId: string, message: string, data?: Record<string, unknown>) =>
-      logger.warn(sessionId, tag, message, data),
-    error: (sessionId: string, message: string, data?: Record<string, unknown>) =>
-      logger.error(sessionId, tag, message, data),
-  };
-}

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

@@ -435,7 +435,10 @@ export async function pollForResult(
       });
 
       if (!statusRes.ok) {
-        generationLogCollector.warn(sessionId, TAG, `Poll attempt ${i + 1}/${maxAttempts}: HTTP ${statusRes.status}, skipping...`);
+        // Only log non-retryable errors to reduce log noise
+        if (statusRes.status >= 400 && statusRes.status < 500) {
+          generationLogCollector.warn(sessionId, TAG, `Poll HTTP ${statusRes.status}, skipping...`);
+        }
         continue;
       }
 
@@ -449,21 +452,20 @@ export async function pollForResult(
         }
       } else if (statusData.status === 'failed') {
         const errorMessage = statusData.error || "Generation failed during processing.";
-        generationLogCollector.error(sessionId, TAG, `Polling: generation failed — ${errorMessage}`);
+        generationLogCollector.error(sessionId, TAG, `Polling failed: ${errorMessage}`);
         throw new Error(errorMessage);
       }
 
-      // Still processing — log progress periodically
-      if ((i + 1) % 10 === 0) {
-        generationLogCollector.log(sessionId, TAG, `Polling: still processing (attempt ${i + 1}/${maxAttempts})...`);
+      // Log progress only every 30 attempts to reduce overhead (every ~90 seconds)
+      if ((i + 1) % 30 === 0) {
+        generationLogCollector.log(sessionId, TAG, `Still processing (attempt ${i + 1}/${maxAttempts})`);
       }
     } catch (error) {
       if (error instanceof Error && error.message.includes("cancelled by user")) {
         throw error;
       }
-      // Non-fatal poll error — continue polling
+      // Non-fatal poll error — continue polling silently
       if (error instanceof Error && !error.message.includes("failed during processing")) {
-        generationLogCollector.warn(sessionId, TAG, `Poll attempt ${i + 1} error: ${error.message}`);
         continue;
       }
       throw error;
@@ -478,15 +480,31 @@ export async function pollForResult(
  * Checks multiple possible locations (priority order).
  */
 export function extractUri(data: PrunaPredictionResponse): string | null {
-  return (
-    data.generation_url ||
-    (data.output && typeof data.output === 'object' && !Array.isArray(data.output) ? (data.output as { url: string }).url : null) ||
-    (typeof data.output === 'string' ? data.output : null) ||
-    data.video_url ||
-    (Array.isArray(data.output) ? data.output[0] : null) ||
-    data.data ||
-    null
-  );
+  // Priority 1: Direct generation URL
+  if (data.generation_url) return data.generation_url;
+
+  // Priority 2: Output object URL
+  if (data.output && typeof data.output === 'object' && !Array.isArray(data.output)) {
+    const outputObj = data.output as { url?: string };
+    if (outputObj.url) return outputObj.url;
+  }
+
+  // Priority 3: Output as string
+  if (typeof data.output === 'string') return data.output;
+
+  // Priority 4: Video URL
+  if (data.video_url) return data.video_url;
+
+  // Priority 5: First array element
+  if (Array.isArray(data.output) && data.output.length > 0) {
+    const firstElement = data.output[0];
+    if (typeof firstElement === 'string') return firstElement;
+  }
+
+  // Priority 6: Data field
+  if (data.data) return data.data;
+
+  return null;
 }
 
 /**

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

@@ -300,33 +300,12 @@ export class PrunaProvider implements IAIProvider {
     }
   }
 
-  reset(): void {
-    cancelAllRequests();
-    this.lastRequestKey = null;
-    this.apiKey = null;
-    this.initialized = false;
-  }
-
   cancelCurrentRequest(): void {
     if (this.lastRequestKey) {
       cancelRequest(this.lastRequestKey);
       this.lastRequestKey = null;
     }
   }
-
-  hasRunningRequest(): boolean {
-    return hasActiveRequests();
-  }
-
-  getSessionLogs(sessionId?: string): LogEntry[] {
-    if (!sessionId) return [];
-    return generationLogCollector.getEntries(sessionId);
-  }
-
-  endLogSession(sessionId?: string): LogEntry[] {
-    if (!sessionId) return [];
-    return generationLogCollector.endSession(sessionId);
-  }
 }
 
 export const prunaProvider = new PrunaProvider();

package/src/infrastructure/services/request-store.ts

@@ -57,19 +57,16 @@ function getRequestKeyCache(): RequestKeyCache {
   return globalObj[REQUEST_KEY_CACHE_KEY] as RequestKeyCache;
 }
 
-function sortKeys(obj: unknown): unknown {
-  if (obj === null || typeof obj !== "object") return obj;
-  if (Array.isArray(obj)) return obj.map(sortKeys);
-  const sorted: Record<string, unknown> = {};
-  for (const key of Object.keys(obj as Record<string, unknown>).sort()) {
-    sorted[key] = sortKeys((obj as Record<string, unknown>)[key]);
-  }
-  return sorted;
-}
-
 function generateCacheKey(model: string, input: Record<string, unknown>): string {
-  // Fast hash using JSON.stringify (already sorted by sortKeys)
-  return `${model}:${JSON.stringify(input)}`;
+  // Fast deterministic hash using sorted property names
+  // Much faster than deep object sorting + JSON.stringify
+  const keys = Object.keys(input).sort();
+  let hash = model;
+  for (const key of keys) {
+    const value = input[key];
+    hash += `|${key}:${typeof value === 'object' ? JSON.stringify(value) : String(value)}`;
+  }
+  return hash;
 }
 
 export function createRequestKey(model: string, input: Record<string, unknown>): string {
@@ -83,20 +80,11 @@ export function createRequestKey(model: string, input: Record<string, unknown>):
     return cached.key;
   }
 
-  // Cache miss or expired - generate new key
-  // Use full JSON string instead of hash to eliminate collision risk
-  // Sort keys ensures consistent key generation regardless of object property order
-  const inputStr = JSON.stringify(sortKeys(input));
-
-  // Use base64 encoding for safer string representation
-  // This eliminates collision risk entirely while maintaining readability
-  const safeInputStr = inputStr.replace(/[^a-zA-Z0-9]/g, '_');
-
-  // Use first 64 chars to keep key length manageable while maintaining uniqueness
-  const prefix = safeInputStr.substring(0, 64);
-  const suffix = safeInputStr.length > 64 ? safeInputStr.slice(-64) : '';
-
-  const requestKey = `${model}:${prefix}${suffix ? '...' + suffix : ''}`;
+  // Cache miss or expired - generate new key using simple hash
+  // Use timestamp + random suffix for uniqueness (crypto-friendly)
+  const timestamp = Date.now();
+  const randomSuffix = Math.random().toString(36).substring(2, 10);
+  const requestKey = `${model}_${cacheKey.substring(0, 16)}_${timestamp}_${randomSuffix}`;
 
   // Store in cache with LRU eviction
   cache.set(cacheKey, { key: requestKey, timestamp: now });
@@ -112,13 +100,6 @@ export function createRequestKey(model: string, input: Record<string, unknown>):
   return requestKey;
 }
 
-export function clearRequestKeyCache(): void {
-  const globalObj = globalThis as Record<string, unknown>;
-  if (globalObj[REQUEST_KEY_CACHE_KEY]) {
-    (globalObj[REQUEST_KEY_CACHE_KEY] as RequestKeyCache).clear();
-  }
-}
-
 export function getExistingRequest<T>(key: string): ActiveRequest<T> | undefined {
   return getRequestStore().get(key) as ActiveRequest<T> | undefined;
 }
@@ -172,6 +153,9 @@ export function cleanupRequestStore(maxAge: number = MAX_REQUEST_AGE): number {
     stopCleanupTimer();
   }
 
+  // Also cleanup orphaned requestId mappings
+  cleanupRequestIdMappings(maxAge);
+
   return cleanedCount;
 }
 
@@ -229,6 +213,25 @@ export function removeRequestIdMapping(requestId: string): void {
   getRequestIdMap().delete(requestId);
 }
 
+/** Cleanup old requestId mappings to prevent unbounded memory growth */
+export function cleanupRequestIdMappings(maxAge: number = MAX_REQUEST_AGE): number {
+  const requestStore = getRequestStore();
+  const requestIdMap = getRequestIdMap();
+  const now = Date.now();
+  let cleanedCount = 0;
+
+  // Remove mappings for requests that no longer exist or are too old
+  for (const [requestId] of requestIdMap.entries()) {
+    const request = requestStore.get(requestId);
+    if (!request || (now - request.createdAt > maxAge)) {
+      requestIdMap.delete(requestId);
+      cleanedCount++;
+    }
+  }
+
+  return cleanedCount;
+}
+
 // Clear any leftover timer on module load (hot reload safety)
 if (typeof globalThis !== "undefined") {
   const existingTimer = getCleanupTimer();

package/src/infrastructure/storage/file-storage.ts

@@ -5,7 +5,7 @@
 
 import { httpClient } from "../api/http-client";
 import { logger } from "../logging/pruna-logger";
-import { PRUNA_FILES_URL, UPLOAD_CONFIG } from "../../services/pruna-provider.constants";
+import { PRUNA_FILES_URL, UPLOAD_CONFIG } from "../services/pruna-provider.constants";
 
 export class FileStorageService {
   async uploadFile(

package/src/infrastructure/utils/constants/mime.constants.ts

@@ -15,8 +15,7 @@ export const MIME_AUDIO_FLAC = 'audio/flac' as const;
 export const MIME_AUDIO_MP4 = 'audio/mp4' as const;
 
 // ── Fallback ────────────────────────────────────────────────
-export const MIME_APPLICATION_OCTET = 'application/octet-stream' as const;
-export const MIME_DEFAULT = MIME_APPLICATION_OCTET;
+export const MIME_DEFAULT = 'application/octet-stream' as const;
 
 /** Maps MIME type → file extension for upload naming */
 export const MIME_TO_EXTENSION: Readonly<Record<string, string>> = {
@@ -27,7 +26,7 @@ export const MIME_TO_EXTENSION: Readonly<Record<string, string>> = {
   [MIME_AUDIO_WAV]: 'wav',
   [MIME_AUDIO_FLAC]: 'flac',
   [MIME_AUDIO_MP4]: 'm4a',
-  [MIME_APPLICATION_OCTET]: 'bin',
+  [MIME_DEFAULT]: 'bin',
 };
 
 /**

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

@@ -30,20 +30,21 @@ const MAX_SESSIONS = 50;
 
 class GenerationLogCollector {
   private sessions = new Map<string, Session>();
-  private sessionQueue: string[] = []; // FIFO queue for O(1) eviction
+  private sessionQueue = new Set<string>(); // O(1) lookup and deletion
 
   startSession(): string {
-    // Evict oldest session if limit exceeded (O(1) operation)
-    if (this.sessionQueue.length >= MAX_SESSIONS) {
-      const oldestKey = this.sessionQueue.shift();
+    // Evict oldest session if limit exceeded
+    if (this.sessionQueue.size >= MAX_SESSIONS) {
+      const oldestKey = this.sessionQueue.keys().next().value;
       if (oldestKey) {
         this.sessions.delete(oldestKey);
+        this.sessionQueue.delete(oldestKey);
       }
     }
 
     const id = `pruna_session_${++sessionCounter}_${Date.now()}`;
     this.sessions.set(id, { startTime: Date.now(), entries: [] });
-    this.sessionQueue.push(id);
+    this.sessionQueue.add(id);
     return id;
   }
 
@@ -60,20 +61,17 @@ class GenerationLogCollector {
   }
 
   getEntries(sessionId: string): LogEntry[] {
-    return [...(this.sessions.get(sessionId)?.entries ?? [])];
+    return this.sessions.get(sessionId)?.entries ?? [];
   }
 
   endSession(sessionId: string): LogEntry[] {
     const session = this.sessions.get(sessionId);
     if (!session) return [];
-    const entries = [...session.entries];
+    const entries = session.entries;
     this.sessions.delete(sessionId);
 
-    // Remove from queue as well
-    const queueIndex = this.sessionQueue.indexOf(sessionId);
-    if (queueIndex !== -1) {
-      this.sessionQueue.splice(queueIndex, 1);
-    }
+    // Remove from queue as well (O(1) with Set)
+    this.sessionQueue.delete(sessionId);
 
     return entries;
   }

package/src/init/createAiProviderInitModule.ts

@@ -5,6 +5,7 @@
 
 import { providerRegistry } from '@umituz/react-native-ai-generation-content';
 import { prunaProvider } from '../infrastructure/services/pruna-provider';
+import { logger } from '../infrastructure/logging/pruna-logger';
 
 /**
  * InitModule interface (from @umituz/react-native-design-system)
@@ -89,7 +90,9 @@ export function createAiProviderInitModule(
 
         return true;
       } catch (error) {
-        console.error('[AiProviderInitModule] Pruna initialization failed:', error);
+        logger.error('app-init', 'ai-provider', 'Pruna initialization failed', {
+          error: error instanceof Error ? error.message : String(error),
+        });
         throw error;
       }
     },

package/src/init/initializePrunaProvider.ts

@@ -5,6 +5,7 @@
 
 import { providerRegistry } from '@umituz/react-native-ai-generation-content';
 import { prunaProvider } from '../infrastructure/services/pruna-provider';
+import { logger } from '../infrastructure/logging/pruna-logger';
 
 /**
  * Initializes Pruna provider and registers it with providerRegistry in one call.
@@ -33,7 +34,9 @@ export function initializePrunaProvider(config: {
 
     return true;
   } catch (error) {
-    console.error('[initializePrunaProvider] Initialization failed:', error);
+    logger.error('app-init', 'pruna-provider', 'Initialization failed', {
+      error: error instanceof Error ? error.message : String(error),
+    });
     throw error;
   }
 }

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

@@ -34,11 +34,7 @@ function convertJobStatusToPrunaQueueStatus(status: JobStatus, currentRequestId:
   return {
     status: status.status as PrunaQueueStatus["status"],
     requestId: status.requestId ?? currentRequestId ?? "",
-    logs: status.logs?.map((log: AILogEntry) => ({
-      message: log.message,
-      level: log.level,
-      timestamp: log.timestamp,
-    })),
+    logs: status.logs as PrunaQueueStatus["logs"], // Type-safe cast, no array copy
   };
 }
 

package/src/index.new.ts

@@ -1,65 +0,0 @@
-/**
- * Pruna AI Provider - Refactored with DDD Architecture
- *
- * @module react-native-ai-pruna-provider
- *
- * Architecture:
- * - Domain: Business logic, value objects, domain services
- * - Application: Use cases, orchestration, DTOs
- * - Infrastructure: API clients, storage, logging
- * - Presentation: React hooks, UI integration
- *
- * Key Features:
- * - Clean separation of concerns
- * - Maximum 150 lines per file
- * - No code duplication
- * - Type-safe with TypeScript
- * - Comprehensive error handling
- */
-
-// Core Service
-export { prunaService, PrunaService } from './application/services/pruna-service';
-export type {
-  PrunaConfig,
-  PrunaModel,
-} from './application/services/pruna-service';
-
-// React Hooks
-export { usePrunaGeneration } from './presentation/hooks/use-pruna-generation.new';
-export type { PrunaGenerationState } from './presentation/hooks/use-pruna-generation.new';
-
-// DTOs
-export type {
-  PrunaImageGenerationRequest,
-  PrunaVideoGenerationRequest,
-  PrunaImageEditRequest,
-  PrunaGenerationResponse,
-  PrunaGenerationOptions,
-  PrunaGenerationError,
-} from './application/dto/pruna.dto';
-
-// Domain Layer
-export { SessionId } from './domain/value-objects/session-id.value';
-export { ApiKey } from './domain/value-objects/api-key.value';
-export { ModelId, PrunaModelId } from './domain/value-objects/model-id.value';
-export { ValidationService, ValidationResult } from './domain/services/validation.domain-service';
-export { ErrorMapperService } from './domain/services/error-mapper.domain-service';
-
-// Infrastructure Layer
-export { logger, createLogger, LogLevel } from './infrastructure/logging/pruna-logger';
-export { httpClient, HttpClient } from './infrastructure/api/http-client';
-export { fileStorageService, FileStorageService } from './infrastructure/storage/file-storage';
-
-// Use Cases (for advanced usage)
-export { generateImageUseCase, GenerateImageUseCase } from './application/use-cases/generate-image.use-case';
-export { generateVideoUseCase, GenerateVideoUseCase } from './application/use-cases/generate-video.use-case';
-export { generateImageEditUseCase, GenerateImageEditUseCase } from './application/use-cases/generate-image-edit.use-case';
-
-// Legacy Exports (for backward compatibility)
-export { prunaProvider as PrunaProvider } from './infrastructure/services/pruna-provider';
-export type {
-  IAIProvider,
-  AIProviderConfig,
-  SubscribeOptions,
-  RunOptions,
-} from './domain/types';

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

@@ -1,182 +0,0 @@
-/**
- * Pruna Generation Hook (Refactored)
- * Clean React hook using new DDD architecture
- */
-
-import { useState, useCallback, useRef } from 'react';
-import { prunaService } from '../../application/services/pruna-service';
-import type {
-  PrunaImageGenerationRequest,
-  PrunaVideoGenerationRequest,
-  PrunaImageEditRequest,
-  PrunaGenerationResponse,
-  PrunaGenerationError,
-} from '../../application/dto/pruna.dto';
-
-export interface PrunaGenerationState {
-  isGenerating: boolean;
-  progress: number;
-  status: string;
-  error: PrunaGenerationError | null;
-  result: PrunaGenerationResponse | null;
-}
-
-export function usePrunaGeneration() {
-  const [state, setState] = useState<PrunaGenerationState>({
-    isGenerating: false,
-    progress: 0,
-    status: 'idle',
-    error: null,
-    result: null,
-  });
-
-  const abortControllerRef = useRef<AbortController | null>(null);
-
-  const generateImage = useCallback(async (request: PrunaImageGenerationRequest) => {
-    const controller = new AbortController();
-    abortControllerRef.current = controller;
-
-    setState({
-      isGenerating: true,
-      progress: 0,
-      status: 'starting',
-      error: null,
-      result: null,
-    });
-
-    try {
-      const response = await prunaService.generateImage(request, controller.signal);
-      setState({
-        isGenerating: false,
-        progress: 100,
-        status: 'completed',
-        error: null,
-        result: response,
-      });
-      return response;
-    } catch (error) {
-      const errorObj: PrunaGenerationError = {
-        type: 'unknown',
-        message: error instanceof Error ? error.message : String(error),
-        retryable: false,
-        originalError: error instanceof Error ? error.stack : undefined,
-      };
-      setState({
-        isGenerating: false,
-        progress: 0,
-        status: 'error',
-        error: errorObj,
-        result: null,
-      });
-      throw error;
-    }
-  }, []);
-
-  const generateVideo = useCallback(async (request: PrunaVideoGenerationRequest) => {
-    const controller = new AbortController();
-    abortControllerRef.current = controller;
-
-    setState({
-      isGenerating: true,
-      progress: 0,
-      status: 'starting',
-      error: null,
-      result: null,
-    });
-
-    try {
-      const response = await prunaService.generateVideo(request, controller.signal);
-      setState({
-        isGenerating: false,
-        progress: 100,
-        status: 'completed',
-        error: null,
-        result: response,
-      });
-      return response;
-    } catch (error) {
-      const errorObj: PrunaGenerationError = {
-        type: 'unknown',
-        message: error instanceof Error ? error.message : String(error),
-        retryable: false,
-        originalError: error instanceof Error ? error.stack : undefined,
-      };
-      setState({
-        isGenerating: false,
-        progress: 0,
-        status: 'error',
-        error: errorObj,
-        result: null,
-      });
-      throw error;
-    }
-  }, []);
-
-  const generateImageEdit = useCallback(async (request: PrunaImageEditRequest) => {
-    const controller = new AbortController();
-    abortControllerRef.current = controller;
-
-    setState({
-      isGenerating: true,
-      progress: 0,
-      status: 'starting',
-      error: null,
-      result: null,
-    });
-
-    try {
-      const response = await prunaService.generateImageEdit(request, controller.signal);
-      setState({
-        isGenerating: false,
-        progress: 100,
-        status: 'completed',
-        error: null,
-        result: response,
-      });
-      return response;
-    } catch (error) {
-      const errorObj: PrunaGenerationError = {
-        type: 'unknown',
-        message: error instanceof Error ? error.message : String(error),
-        retryable: false,
-        originalError: error instanceof Error ? error.stack : undefined,
-      };
-      setState({
-        isGenerating: false,
-        progress: 0,
-        status: 'error',
-        error: errorObj,
-        result: null,
-      });
-      throw error;
-    }
-  }, []);
-
-  const cancel = useCallback(() => {
-    abortControllerRef.current?.abort();
-    setState(prev => ({
-      ...prev,
-      isGenerating: false,
-      status: 'cancelled',
-    }));
-  }, []);
-
-  const reset = useCallback(() => {
-    setState({
-      isGenerating: false,
-      progress: 0,
-      status: 'idle',
-      error: null,
-      result: null,
-    });
-  }, []);
-
-  return {
-    state,
-    generateImage,
-    generateVideo,
-    generateImageEdit,
-    cancel,
-    reset,
-  };
-}