@adminforth/bulk-ai-flow 1.26.4 → 1.26.6

package/build.log

@@ -14,5 +14,5 @@ custom/package.json
 custom/pnpm-lock.yaml
 custom/tsconfig.json
 
-sent 129,913 bytes  received 191 bytes  260,208.00 bytes/sec
-total size is 129,205  speedup is 0.99
+sent 130,637 bytes  received 191 bytes  261,656.00 bytes/sec
+total size is 129,924  speedup is 0.99

package/custom/VisionAction.vue

@@ -482,7 +482,7 @@ const generationModeButtons = computed(() => {
   return arrayToReturn;
 });
 
-async function regenerateRecord({ recordId }) {
+async function regenerateRecord({ recordId }: { recordId: string }) {
 
   const promises = [];
 
@@ -610,7 +610,8 @@ async function getListOfIds() {
  
 
 async function runAiActions() {
-  if (!await checkRateLimits()) {
+  const { ok, sessionIds } = await checkRateLimits();
+  if (!ok || !sessionIds) {
     confirmDialog.value.close();
     return;
   }
@@ -625,7 +626,7 @@ async function runAiActions() {
   tableRef.value?.refresh();
   const limit = pLimit(props.meta.concurrencyLimit || 10);
   const tasks = recordIds.value
-    .map(id => limit(() => processOneRecord(String(id))));
+    .map(id => limit(() => processOneRecord(String(id), sessionIds)));
   await Promise.all(tasks);
   showGenerationFailureSummary();
   isActiveGeneration.value = false;
@@ -825,7 +826,7 @@ function createEmptyRecord(recordId: string | number): RecordState {
   };
 }
 
-async function processOneRecord(recordId: string) {
+async function processOneRecord(recordId: string, sessionIds: string[]) {
   if (!checkIfDialogOpen()) {
     return;
   }
@@ -898,7 +899,7 @@ async function processOneRecord(recordId: string) {
     actions.push('analyze_no_images');
   }
 
-  const results = await Promise.allSettled(actions.map(actionType => runActionForRecord(record, actionType)));
+  const results = await Promise.allSettled(actions.map(actionType => runActionForRecord(record, actionType, sessionIds)));
   if (!checkIfDialogOpen()) {
     return;
   }
@@ -908,17 +909,22 @@ async function processOneRecord(recordId: string) {
   touchRecords();
 }
 
-async function checkRateLimits() {
+async function checkRateLimits(actions?: GenerationAction[]): Promise<{ ok: boolean; sessionIds?: string[] }> {
+  let sessionIds: string[] = [];
   isCheckingRateLimits.value = true;
   const actionsToCheck: GenerationAction[] = [];
-  if (props.meta.isImageGeneration) {
-    actionsToCheck.push('generate_images');
-  }
-  if (props.meta.isFieldsForAnalizeFromImages) {
-    actionsToCheck.push('analyze');
-  }
-  if (props.meta.isFieldsForAnalizePlain) {
-    actionsToCheck.push('analyze_no_images');
+  if (actions) {
+    actionsToCheck.push(...actions);
+  } else {
+    if (props.meta.isImageGeneration) {
+      actionsToCheck.push('generate_images');
+    }
+    if (props.meta.isFieldsForAnalizeFromImages) {
+      actionsToCheck.push('analyze');
+    }
+    if (props.meta.isFieldsForAnalizePlain) {
+      actionsToCheck.push('analyze_no_images');
+    }
   }
   for (const actionType of actionsToCheck) {
     try {
@@ -933,8 +939,9 @@ async function checkRateLimits() {
           variant: 'danger',
           timeout: 'unlimited',
         });
-        return false;
+        return { ok: false };
       }
+      sessionIds.push(rateLimitRes.sessionId);
     } catch (e) {
       adminforth.alert({
         message: t(`Error checking rate limit for "${getActionLabel(actionType)}" action.`),
@@ -942,14 +949,14 @@ async function checkRateLimits() {
         timeout: 'unlimited',
       });
       isCheckingRateLimits.value = false;
-      return false;
+      return { ok: false };
     }
   }
   isCheckingRateLimits.value = false;
-  return true;
+  return { ok: true, sessionIds };
 }
 
-async function runActionForRecord(record: RecordState, actionType: GenerationAction, forceFilterFilledFields?: boolean) {
+async function runActionForRecord(record: RecordState, actionType: GenerationAction, sessionIds: string[], forceFilterFilledFields?: boolean) {
   if (!checkIfDialogOpen()) {
     return;
   }
@@ -982,6 +989,7 @@ async function runActionForRecord(record: RecordState, actionType: GenerationAct
         recordId: record.id,
         ...(customPrompt !== undefined ? { customPrompt: JSON.stringify(customPrompt) } : {}),
         filterFilledFields: forceFilterFilledFields !== undefined ? forceFilterFilledFields: !overwriteExistingValues.value,
+        sessionIds
       },
       silentError: true,
     });
@@ -1438,6 +1446,10 @@ async function uploadImage(imgBlob, id, fieldName) {
 }
 
 async function regenerateImages({ recordId }: { recordId: any }) {
+  const {ok, sessionIds} = await checkRateLimits(['generate_images']);
+  if (!ok || !sessionIds) {
+    return;
+  }
   if (coreStore.isInternetError) {
     adminforth.alert({
       message: t('Cannot regenerate images while internet connection is lost. Please check your connection and try again.'),
@@ -1458,6 +1470,7 @@ async function regenerateImages({ recordId }: { recordId: any }) {
     await runActionForRecord(
       record,
       'generate_images',
+      sessionIds,
       false
     );
 
@@ -1616,6 +1629,12 @@ async function regenerateCell(recordInfo: any) {
     });
     return;
   }
+
+  const {ok, sessionIds} = await checkRateLimits(['analyze_no_images', 'analyze']);
+  if (!ok || !sessionIds) {
+    return;
+  }
+
   const record = recordsById.get(String(recordInfo.recordId));
   if (!record) {
     return;
@@ -1658,6 +1677,7 @@ async function regenerateCell(recordInfo: any) {
         action: actionType,
         actionType: "regenerate_cell",
         prompt: generationPromptsForField[recordInfo.fieldName] || null,
+        sessionIds
       },
       silentError: true,
     });

package/dist/custom/VisionAction.vue

@@ -482,7 +482,7 @@ const generationModeButtons = computed(() => {
   return arrayToReturn;
 });
 
-async function regenerateRecord({ recordId }) {
+async function regenerateRecord({ recordId }: { recordId: string }) {
 
   const promises = [];
 
@@ -610,7 +610,8 @@ async function getListOfIds() {
  
 
 async function runAiActions() {
-  if (!await checkRateLimits()) {
+  const { ok, sessionIds } = await checkRateLimits();
+  if (!ok || !sessionIds) {
     confirmDialog.value.close();
     return;
   }
@@ -625,7 +626,7 @@ async function runAiActions() {
   tableRef.value?.refresh();
   const limit = pLimit(props.meta.concurrencyLimit || 10);
   const tasks = recordIds.value
-    .map(id => limit(() => processOneRecord(String(id))));
+    .map(id => limit(() => processOneRecord(String(id), sessionIds)));
   await Promise.all(tasks);
   showGenerationFailureSummary();
   isActiveGeneration.value = false;
@@ -825,7 +826,7 @@ function createEmptyRecord(recordId: string | number): RecordState {
   };
 }
 
-async function processOneRecord(recordId: string) {
+async function processOneRecord(recordId: string, sessionIds: string[]) {
   if (!checkIfDialogOpen()) {
     return;
   }
@@ -898,7 +899,7 @@ async function processOneRecord(recordId: string) {
     actions.push('analyze_no_images');
   }
 
-  const results = await Promise.allSettled(actions.map(actionType => runActionForRecord(record, actionType)));
+  const results = await Promise.allSettled(actions.map(actionType => runActionForRecord(record, actionType, sessionIds)));
   if (!checkIfDialogOpen()) {
     return;
   }
@@ -908,17 +909,22 @@ async function processOneRecord(recordId: string) {
   touchRecords();
 }
 
-async function checkRateLimits() {
+async function checkRateLimits(actions?: GenerationAction[]): Promise<{ ok: boolean; sessionIds?: string[] }> {
+  let sessionIds: string[] = [];
   isCheckingRateLimits.value = true;
   const actionsToCheck: GenerationAction[] = [];
-  if (props.meta.isImageGeneration) {
-    actionsToCheck.push('generate_images');
-  }
-  if (props.meta.isFieldsForAnalizeFromImages) {
-    actionsToCheck.push('analyze');
-  }
-  if (props.meta.isFieldsForAnalizePlain) {
-    actionsToCheck.push('analyze_no_images');
+  if (actions) {
+    actionsToCheck.push(...actions);
+  } else {
+    if (props.meta.isImageGeneration) {
+      actionsToCheck.push('generate_images');
+    }
+    if (props.meta.isFieldsForAnalizeFromImages) {
+      actionsToCheck.push('analyze');
+    }
+    if (props.meta.isFieldsForAnalizePlain) {
+      actionsToCheck.push('analyze_no_images');
+    }
   }
   for (const actionType of actionsToCheck) {
     try {
@@ -933,8 +939,9 @@ async function checkRateLimits() {
           variant: 'danger',
           timeout: 'unlimited',
         });
-        return false;
+        return { ok: false };
       }
+      sessionIds.push(rateLimitRes.sessionId);
     } catch (e) {
       adminforth.alert({
         message: t(`Error checking rate limit for "${getActionLabel(actionType)}" action.`),
@@ -942,14 +949,14 @@ async function checkRateLimits() {
         timeout: 'unlimited',
       });
       isCheckingRateLimits.value = false;
-      return false;
+      return { ok: false };
     }
   }
   isCheckingRateLimits.value = false;
-  return true;
+  return { ok: true, sessionIds };
 }
 
-async function runActionForRecord(record: RecordState, actionType: GenerationAction, forceFilterFilledFields?: boolean) {
+async function runActionForRecord(record: RecordState, actionType: GenerationAction, sessionIds: string[], forceFilterFilledFields?: boolean) {
   if (!checkIfDialogOpen()) {
     return;
   }
@@ -982,6 +989,7 @@ async function runActionForRecord(record: RecordState, actionType: GenerationAct
         recordId: record.id,
         ...(customPrompt !== undefined ? { customPrompt: JSON.stringify(customPrompt) } : {}),
         filterFilledFields: forceFilterFilledFields !== undefined ? forceFilterFilledFields: !overwriteExistingValues.value,
+        sessionIds
       },
       silentError: true,
     });
@@ -1438,6 +1446,10 @@ async function uploadImage(imgBlob, id, fieldName) {
 }
 
 async function regenerateImages({ recordId }: { recordId: any }) {
+  const {ok, sessionIds} = await checkRateLimits(['generate_images']);
+  if (!ok || !sessionIds) {
+    return;
+  }
   if (coreStore.isInternetError) {
     adminforth.alert({
       message: t('Cannot regenerate images while internet connection is lost. Please check your connection and try again.'),
@@ -1458,6 +1470,7 @@ async function regenerateImages({ recordId }: { recordId: any }) {
     await runActionForRecord(
       record,
       'generate_images',
+      sessionIds,
       false
     );
 
@@ -1616,6 +1629,12 @@ async function regenerateCell(recordInfo: any) {
     });
     return;
   }
+
+  const {ok, sessionIds} = await checkRateLimits(['analyze_no_images', 'analyze']);
+  if (!ok || !sessionIds) {
+    return;
+  }
+
   const record = recordsById.get(String(recordInfo.recordId));
   if (!record) {
     return;
@@ -1658,6 +1677,7 @@ async function regenerateCell(recordInfo: any) {
         action: actionType,
         actionType: "regenerate_cell",
         prompt: generationPromptsForField[recordInfo.fieldName] || null,
+        sessionIds
       },
       silentError: true,
     });

package/dist/index.d.ts

@@ -8,6 +8,7 @@ export default class BulkAiFlowPlugin extends AdminForthPlugin {
     totalCalls: number;
     totalDuration: number;
     rateLimiters: Record<string, RateLimiter>;
+    sessionIds: Set<string>;
     constructor(options: PluginOptions);
     private compileTemplates;
     private compileOutputFieldsTemplates;

package/dist/index.js

@@ -18,10 +18,13 @@ export default class BulkAiFlowPlugin extends AdminForthPlugin {
     constructor(options) {
         super(options, import.meta.url);
         this.rateLimiters = {};
+        this.sessionIds = new Set();
         this.options = options;
         // for calculating average time
         this.totalCalls = 0;
         this.totalDuration = 0;
+        // for rate limiting
+        this.sessionIds = new Set();
     }
     // Compile Handlebars templates in outputFields using record fields as context
     compileTemplates(source, record, valueSelector) {
@@ -945,7 +948,20 @@ export default class BulkAiFlowPlugin extends AdminForthPlugin {
             method: 'POST',
             path: `/plugin/${this.pluginInstanceId}/create-job`,
             handler: (_a) => __awaiter(this, [_a], void 0, function* ({ body, adminUser, headers }) {
-                const { actionType, recordId, customPrompt, filterFilledFields } = body;
+                const { actionType, recordId, customPrompt, filterFilledFields, sessionIds } = body;
+                if (this.options.rateLimits) {
+                    if (sessionIds && sessionIds.length > 0) {
+                        for (const sessionId of sessionIds) {
+                            if (!this.sessionIds.has(sessionId)) {
+                                return { ok: false, error: "Invalid session id" };
+                            }
+                            this.sessionIds.delete(sessionId);
+                        }
+                    }
+                    else {
+                        return { ok: false, error: "Missing session ids" };
+                    }
+                }
                 const jobId = randomUUID();
                 jobs.set(jobId, { status: "in_progress" });
                 if (!actionType) {
@@ -1026,6 +1042,8 @@ export default class BulkAiFlowPlugin extends AdminForthPlugin {
             handler: (_a) => __awaiter(this, [_a], void 0, function* ({ body, adminUser, headers }) {
                 var _b, _c, _d;
                 const actionType = body.actionType;
+                const sessionId = randomUUID();
+                this.sessionIds.add(sessionId);
                 if (actionType === 'analyze' && ((_b = this.options.rateLimits) === null || _b === void 0 ? void 0 : _b.fillFieldsFromImages)) {
                     if (yield this.checkRateLimit("fillFieldsFromImages", this.options.rateLimits.fillFieldsFromImages, headers)) {
                         return { ok: false, error: "Rate limit exceeded for image analyze" };
@@ -1041,7 +1059,8 @@ export default class BulkAiFlowPlugin extends AdminForthPlugin {
                         return { ok: false, error: "Rate limit exceeded for image generation" };
                     }
                 }
-                return { ok: true };
+                setTimeout(() => this.sessionIds.delete(sessionId), 300000);
+                return { ok: true, sessionId };
             })
         });
         server.endpoint({

package/index.ts

@@ -13,6 +13,7 @@ export default class  BulkAiFlowPlugin extends AdminForthPlugin {
   totalCalls: number;
   totalDuration: number;
   rateLimiters: Record<string, RateLimiter> = {};
+  sessionIds: Set<string> = new Set();
 
   constructor(options: PluginOptions) {
     super(options, import.meta.url);
@@ -21,6 +22,9 @@ export default class  BulkAiFlowPlugin extends AdminForthPlugin {
     // for calculating average time
     this.totalCalls = 0;
     this.totalDuration = 0;
+
+    // for rate limiting
+    this.sessionIds = new Set();
   }
 
   // Compile Handlebars templates in outputFields using record fields as context
@@ -970,7 +974,19 @@ export default class  BulkAiFlowPlugin extends AdminForthPlugin {
       method: 'POST',
       path: `/plugin/${this.pluginInstanceId}/create-job`,
       handler: async ({ body, adminUser, headers }) => {
-        const { actionType, recordId, customPrompt, filterFilledFields } = body;
+        const { actionType, recordId, customPrompt, filterFilledFields, sessionIds } = body;
+        if (this.options.rateLimits) {
+          if (sessionIds && sessionIds.length > 0) {
+            for (const sessionId of sessionIds) {
+              if (!this.sessionIds.has(sessionId)) {
+                return { ok: false, error: "Invalid session id" };
+              }
+              this.sessionIds.delete(sessionId);
+            }
+          } else {
+            return { ok: false, error: "Missing session ids" };
+          }
+        }
         const jobId = randomUUID();
         jobs.set(jobId, { status: "in_progress" });
         if (!actionType) {
@@ -1056,6 +1072,8 @@ export default class  BulkAiFlowPlugin extends AdminForthPlugin {
       path: `/plugin/${this.pluginInstanceId}/update-rate-limits`,
       handler: async ({ body, adminUser, headers }) => {
         const actionType = body.actionType;
+        const sessionId = randomUUID();
+        this.sessionIds.add(sessionId);
         if (actionType === 'analyze' && this.options.rateLimits?.fillFieldsFromImages) {
           if (await this.checkRateLimit("fillFieldsFromImages" ,this.options.rateLimits.fillFieldsFromImages, headers)) {
             return {ok: false, error: "Rate limit exceeded for image analyze" };
@@ -1071,8 +1089,8 @@ export default class  BulkAiFlowPlugin extends AdminForthPlugin {
             return {ok: false, error: "Rate limit exceeded for image generation" };
           }
         }
-
-        return { ok: true };
+        setTimeout(() => this.sessionIds.delete(sessionId), 300_000);
+        return { ok: true, sessionId };
       }
     });
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@adminforth/bulk-ai-flow",
-  "version": "1.26.4",
+  "version": "1.26.6",
   "publishConfig": {
     "access": "public"
   },
@@ -25,7 +25,7 @@
   "description": "Bulk AI workflow plugin for AdminForth for batch field generation and image processing",
   "devDependencies": {
     "@types/node": "latest",
-    "adminforth": "^3.1.1",
+    "adminforth": "^3.1.0",
     "semantic-release": "^24.2.1",
     "semantic-release-slack-bot": "^4.0.2",
     "typescript": "^5.7.3"
@@ -35,7 +35,7 @@
     "handlebars": "^4.7.8"
   },
   "peerDependencies": {
-    "adminforth": "^3.1.1"
+    "adminforth": "^3.1.0"
   },
   "release": {
     "plugins": [