@umituz/react-native-ai-pruna-provider 1.0.41 → 1.0.43

package/package.json

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

@@ -22,6 +22,8 @@ import type { LogEntry } from "../utils/log-collector";
 import {
   createRequestKey, getExistingRequest, storeRequest,
   removeRequest, cancelRequest, cancelAllRequests, hasActiveRequests,
+  storeRequestIdMapping, storeImmediateResultMapping,
+  getStatusUrlForRequestId, getResponseUrlForRequestId, removeRequestIdMapping,
 } from "./request-store";
 
 export class PrunaProvider implements IAIProvider {
@@ -87,7 +89,19 @@ export class PrunaProvider implements IAIProvider {
     const sessionId = generationLogCollector.startSession();
     generationLogCollector.log(sessionId, 'pruna-provider', `submitJob() for model: ${model}`);
     try {
-      return await queueOps.submitJob(prunaModel, input, apiKey, sessionId);
+      const submission = await queueOps.submitJob(prunaModel, input, apiKey, sessionId);
+
+      // Store requestId -> statusUrl mapping for async jobs (requires polling)
+      if (submission.statusUrl) {
+        storeRequestIdMapping(submission.requestId, submission.statusUrl, model);
+      }
+
+      // Store requestId -> responseUrl mapping for immediate results (already complete)
+      if (submission.responseUrl) {
+        storeImmediateResultMapping(submission.requestId, submission.responseUrl, model);
+      }
+
+      return submission;
     } finally {
       generationLogCollector.endSession(sessionId);
     }
@@ -96,12 +110,45 @@ export class PrunaProvider implements IAIProvider {
   async getJobStatus(model: string, requestId: string): Promise<JobStatus> {
     const apiKey = this.validateInit();
     const prunaModel = this.validateModel(model);
+
+    // Check if this is an immediate result (already completed)
+    const responseUrl = getResponseUrlForRequestId(requestId);
+    if (responseUrl) {
+      // Result is already available
+      return {
+        status: "COMPLETED",
+        requestId,
+      };
+    }
+
+    // Look up statusUrl from requestId mapping (async job)
+    const statusUrl = getStatusUrlForRequestId(requestId);
+    if (statusUrl) {
+      return queueOps.getJobStatus(prunaModel, statusUrl, apiKey);
+    }
+
+    // Fallback: assume requestId is actually a statusUrl (for direct calls with statusUrl)
     return queueOps.getJobStatus(prunaModel, requestId, apiKey);
   }
 
   async getJobResult<T = unknown>(model: string, requestId: string): Promise<T> {
     const apiKey = this.validateInit();
     const prunaModel = this.validateModel(model);
+
+    // Check if this is an immediate result (already completed)
+    const responseUrl = getResponseUrlForRequestId(requestId);
+    if (responseUrl) {
+      // Return the immediate result
+      return { url: responseUrl } as T;
+    }
+
+    // Look up statusUrl from requestId mapping (async job)
+    const statusUrl = getStatusUrlForRequestId(requestId);
+    if (statusUrl) {
+      return queueOps.getJobResult<T>(prunaModel, statusUrl, apiKey);
+    }
+
+    // Fallback: assume requestId is actually a statusUrl (for direct calls with statusUrl)
     return queueOps.getJobResult<T>(prunaModel, requestId, apiKey);
   }
 

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

@@ -14,7 +14,9 @@ export interface ActiveRequest<T = unknown> {
 
 const STORE_KEY = "__PRUNA_PROVIDER_REQUESTS__";
 const TIMER_KEY = "__PRUNA_PROVIDER_CLEANUP_TIMER__";
+const REQUEST_ID_KEY = "__PRUNA_PROVIDER_REQUEST_IDS__";
 type RequestStore = Map<string, ActiveRequest>;
+type RequestIdMap = Map<string, { statusUrl?: string; responseUrl?: string; model: string }>;
 
 const CLEANUP_INTERVAL = 60_000;
 const MAX_REQUEST_AGE = 3_660_000; // 61 min — must exceed max allowed timeout (1 hour)
@@ -138,6 +140,43 @@ export function stopAutomaticCleanup(): void {
   stopCleanupTimer();
 }
 
+// ─── Request ID to StatusUrl Mapping ───────────────────────────────────────
+
+function getRequestIdMap(): RequestIdMap {
+  const globalObj = globalThis as Record<string, unknown>;
+  if (!globalObj[REQUEST_ID_KEY]) {
+    globalObj[REQUEST_ID_KEY] = new Map();
+  }
+  return globalObj[REQUEST_ID_KEY] as RequestIdMap;
+}
+
+export function storeRequestIdMapping(requestId: string, statusUrl: string, model: string): void {
+  getRequestIdMap().set(requestId, { statusUrl, model, responseUrl: undefined });
+}
+
+export function storeImmediateResultMapping(requestId: string, responseUrl: string, model: string): void {
+  getRequestIdMap().set(requestId, { statusUrl: undefined, model, responseUrl });
+}
+
+export function getStatusUrlForRequestId(requestId: string): string | undefined {
+  const mapping = getRequestIdMap().get(requestId);
+  return mapping?.statusUrl;
+}
+
+export function getResponseUrlForRequestId(requestId: string): string | undefined {
+  const mapping = getRequestIdMap().get(requestId);
+  return mapping?.responseUrl;
+}
+
+export function getModelForRequestId(requestId: string): string | undefined {
+  const mapping = getRequestIdMap().get(requestId);
+  return mapping?.model;
+}
+
+export function removeRequestIdMapping(requestId: string): void {
+  getRequestIdMap().delete(requestId);
+}
+
 // Clear any leftover timer on module load (hot reload safety)
 if (typeof globalThis !== "undefined") {
   const existingTimer = getCleanupTimer();