move queue limit to grants

MineSkin · Jan 21, 2025 · 950efb8 · 950efb8
1 parent e162f83
commit 950efb8
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/models/v2/generate.ts b/src/models/v2/generate.ts
@@ -382,15 +382,16 @@ async function v2SubmitGeneratorJob(req: GenerateV2Request, res: Response<V2Gene
 
     if (req.client.hasUser()) {
         const pendingByUser = await getClient().getPendingCountByUser(req.client.userId!)
-        if (pendingByUser > 5) { // TODO: configurable / client grant
+        const limit = req.client.getQueueLimit();
+        if (pendingByUser > limit) {
             throw new GeneratorError('job_limit', "You have too many jobs in the queue", {
                 httpCode: 429,
                 source: ErrorSource.CLIENT
             });
         }
     } else {
         const pendingByIp = await getClient().getPendingCountByIp(req.client.ip!)
-        if (pendingByIp > 5) { // TODO: configurable / client grant
+        if (pendingByIp > 4) {
             throw new GeneratorError('job_limit', "You have too many jobs in the queue", {
                 httpCode: 429,
                 source: ErrorSource.CLIENT