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

package/package.json

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

@@ -87,20 +87,26 @@ export class PrunaProvider implements IAIProvider {
     const apiKey = this.validateInit();
     const prunaModel = this.validateModel(model);
     const sessionId = generationLogCollector.startSession();
-    generationLogCollector.log(sessionId, 'pruna-provider', `submitJob() for model: ${model}`);
+    generationLogCollector.log(sessionId, 'pruna-provider', `submitJob() START model: ${model}`);
     try {
       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)
+      // Log response type
       if (submission.responseUrl) {
+        generationLogCollector.log(sessionId, 'pruna-provider', `submitJob() IMMEDIATE RESULT - requestId: ${submission.requestId}, responseUrl: ${submission.responseUrl.substring(0, 80)}...`);
+        // Store requestId -> responseUrl mapping for immediate results (already complete)
         storeImmediateResultMapping(submission.requestId, submission.responseUrl, model);
+        generationLogCollector.log(sessionId, 'pruna-provider', `submitJob() Stored immediate result mapping: ${submission.requestId}`);
+      } else if (submission.statusUrl) {
+        generationLogCollector.log(sessionId, 'pruna-provider', `submitJob() ASYNC JOB - requestId: ${submission.requestId}, statusUrl: ${submission.statusUrl.substring(0, 80)}...`);
+        // Store requestId -> statusUrl mapping for async jobs (requires polling)
+        storeRequestIdMapping(submission.requestId, submission.statusUrl, model);
+        generationLogCollector.log(sessionId, 'pruna-provider', `submitJob() Stored async job mapping: ${submission.requestId}`);
+      } else {
+        generationLogCollector.warn(sessionId, 'pruna-provider', `submitJob() WARNING - No responseUrl or statusUrl in submission`);
       }
 
+      generationLogCollector.log(sessionId, 'pruna-provider', `submitJob() COMPLETE - returning submission with requestId: ${submission.requestId}`);
       return submission;
     } finally {
       generationLogCollector.endSession(sessionId);
@@ -110,46 +116,82 @@ export class PrunaProvider implements IAIProvider {
   async getJobStatus(model: string, requestId: string): Promise<JobStatus> {
     const apiKey = this.validateInit();
     const prunaModel = this.validateModel(model);
+    const sessionId = generationLogCollector.startSession();
+    generationLogCollector.log(sessionId, 'pruna-provider', `getJobStatus() START - model: ${model}, requestId: ${requestId}`);
 
-    // Check if this is an immediate result (already completed)
-    const responseUrl = getResponseUrlForRequestId(requestId);
-    if (responseUrl) {
-      // Result is already available
-      return {
-        status: "COMPLETED",
-        requestId,
-      };
-    }
+    try {
+      // Check if this is an immediate result (already completed)
+      const responseUrl = getResponseUrlForRequestId(requestId);
+      if (responseUrl) {
+        generationLogCollector.log(sessionId, 'pruna-provider', `getJobStatus() FOUND immediate result for ${requestId} - returning COMPLETED`);
+        generationLogCollector.endSession(sessionId);
+        // 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);
-    }
+      // Look up statusUrl from requestId mapping (async job)
+      const statusUrl = getStatusUrlForRequestId(requestId);
+      if (statusUrl) {
+        generationLogCollector.log(sessionId, 'pruna-provider', `getJobStatus() FOUND async job for ${requestId} - polling statusUrl: ${statusUrl.substring(0, 80)}...`);
+        const result = await queueOps.getJobStatus(prunaModel, statusUrl, apiKey);
+        generationLogCollector.log(sessionId, 'pruna-provider', `getJobStatus() Poll result: ${result.status}`);
+        generationLogCollector.endSession(sessionId);
+        return result;
+      }
 
-    // Fallback: assume requestId is actually a statusUrl (for direct calls with statusUrl)
-    return queueOps.getJobStatus(prunaModel, requestId, apiKey);
+      // Fallback: assume requestId is actually a statusUrl (for direct calls with statusUrl)
+      generationLogCollector.log(sessionId, 'pruna-provider', `getJobStatus() NO mapping found - treating ${requestId} as statusUrl directly`);
+      const result = await queueOps.getJobStatus(prunaModel, requestId, apiKey);
+      generationLogCollector.log(sessionId, 'pruna-provider', `getJobStatus() Direct poll result: ${result.status}`);
+      generationLogCollector.endSession(sessionId);
+      return result;
+    } catch (error) {
+      generationLogCollector.error(sessionId, 'pruna-provider', `getJobStatus() ERROR: ${error instanceof Error ? error.message : String(error)}`);
+      generationLogCollector.endSession(sessionId);
+      throw error;
+    }
   }
 
   async getJobResult<T = unknown>(model: string, requestId: string): Promise<T> {
     const apiKey = this.validateInit();
     const prunaModel = this.validateModel(model);
+    const sessionId = generationLogCollector.startSession();
+    generationLogCollector.log(sessionId, 'pruna-provider', `getJobResult() START - model: ${model}, requestId: ${requestId}`);
 
-    // Check if this is an immediate result (already completed)
-    const responseUrl = getResponseUrlForRequestId(requestId);
-    if (responseUrl) {
-      // Return the immediate result
-      return { url: responseUrl } as T;
-    }
+    try {
+      // Check if this is an immediate result (already completed)
+      const responseUrl = getResponseUrlForRequestId(requestId);
+      if (responseUrl) {
+        generationLogCollector.log(sessionId, 'pruna-provider', `getJobResult() FOUND immediate result for ${requestId} - returning output: ${responseUrl.substring(0, 80)}...`);
+        generationLogCollector.endSession(sessionId);
+        // Return the immediate result in expected format for extractResultUrl
+        return { output: responseUrl } as T;
+      }
 
-    // Look up statusUrl from requestId mapping (async job)
-    const statusUrl = getStatusUrlForRequestId(requestId);
-    if (statusUrl) {
-      return queueOps.getJobResult<T>(prunaModel, statusUrl, apiKey);
-    }
+      // Look up statusUrl from requestId mapping (async job)
+      const statusUrl = getStatusUrlForRequestId(requestId);
+      if (statusUrl) {
+        generationLogCollector.log(sessionId, 'pruna-provider', `getJobResult() FOUND async job for ${requestId} - fetching from statusUrl: ${statusUrl.substring(0, 80)}...`);
+        const result = await queueOps.getJobResult<T>(prunaModel, statusUrl, apiKey);
+        generationLogCollector.log(sessionId, 'pruna-provider', `getJobResult() Fetch complete`);
+        generationLogCollector.endSession(sessionId);
+        return result;
+      }
 
-    // Fallback: assume requestId is actually a statusUrl (for direct calls with statusUrl)
-    return queueOps.getJobResult<T>(prunaModel, requestId, apiKey);
+      // Fallback: assume requestId is actually a statusUrl (for direct calls with statusUrl)
+      generationLogCollector.log(sessionId, 'pruna-provider', `getJobResult() NO mapping found - treating ${requestId} as statusUrl directly`);
+      const result = await queueOps.getJobResult<T>(prunaModel, requestId, apiKey);
+      generationLogCollector.log(sessionId, 'pruna-provider', `getJobResult() Direct fetch complete`);
+      generationLogCollector.endSession(sessionId);
+      return result;
+    } catch (error) {
+      generationLogCollector.error(sessionId, 'pruna-provider', `getJobResult() ERROR: ${error instanceof Error ? error.message : String(error)}`);
+      generationLogCollector.endSession(sessionId);
+      throw error;
+    }
   }
 
   async subscribe<T = unknown>(

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

@@ -129,5 +129,5 @@ export async function getJobResult<T = unknown>(
     throw new Error("Result not ready or extraction failed.");
   }
 
-  return { url: resolveUri(uri) } as T;
+  return { output: resolveUri(uri) } as T;
 }