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

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@umituz/react-native-ai-pruna-provider",
-  "version": "1.0.42",
+  "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,7 +22,8 @@ import type { LogEntry } from "../utils/log-collector";
 import {
   createRequestKey, getExistingRequest, storeRequest,
   removeRequest, cancelRequest, cancelAllRequests, hasActiveRequests,
-  storeRequestIdMapping, getStatusUrlForRequestId, removeRequestIdMapping,
+  storeRequestIdMapping, storeImmediateResultMapping,
+  getStatusUrlForRequestId, getResponseUrlForRequestId, removeRequestIdMapping,
 } from "./request-store";
 
 export class PrunaProvider implements IAIProvider {
@@ -90,11 +91,16 @@ export class PrunaProvider implements IAIProvider {
     try {
       const submission = await queueOps.submitJob(prunaModel, input, apiKey, sessionId);
 
-      // Store requestId -> statusUrl mapping for getJobStatus/getJobResult
+      // 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);
@@ -105,7 +111,17 @@ export class PrunaProvider implements IAIProvider {
     const apiKey = this.validateInit();
     const prunaModel = this.validateModel(model);
 
-    // Look up statusUrl from requestId mapping
+    // 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);
@@ -119,7 +135,14 @@ export class PrunaProvider implements IAIProvider {
     const apiKey = this.validateInit();
     const prunaModel = this.validateModel(model);
 
-    // Look up statusUrl from requestId mapping
+    // 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);

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

@@ -16,7 +16,7 @@ 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; model: string }>;
+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)
@@ -151,7 +151,11 @@ function getRequestIdMap(): RequestIdMap {
 }
 
 export function storeRequestIdMapping(requestId: string, statusUrl: string, model: string): void {
-  getRequestIdMap().set(requestId, { statusUrl, model });
+  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 {
@@ -159,6 +163,11 @@ export function getStatusUrlForRequestId(requestId: string): string | undefined
   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;