feat: increase playground allowance for pro plan

lunary-ai · Mar 5, 2024 · 9ee1034 · 9ee1034
1 parent b088ae3
commit 9ee1034
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/packages/backend/src/jobs/resetUsage.ts b/packages/backend/src/jobs/resetUsage.ts
@@ -57,23 +57,19 @@ RETURNING *;`
   }
 }
 
-// reset org.ai_allowance:
-// - 3 for free users
-// - 10 for 'pro' users
-// - 1000 for 'unlimited' and 'enterprise' users
-
-async function resetAIallowance() {
+// reset playground allowance
+async function resetPlaygroundAllowance() {
   await sql`UPDATE "public"."org" o SET play_allowance = 3 WHERE o.plan = 'free';`
 
-  await sql`UPDATE "public"."org" o SET play_allowance = 15 WHERE o.plan = 'pro';`
+  await sql`UPDATE "public"."org" o SET play_allowance = 1000 WHERE o.plan = 'pro';`
 
   await sql`UPDATE "public"."org" o SET play_allowance = 1000 WHERE o.plan = 'unlimited' OR o.plan = 'custom';`
 }
 
 export default async function resetUsage() {
   try {
     console.log("[JOB]: resetting AI allowance")
-    await resetAIallowance()
+    await resetPlaygroundAllowance()
   } catch (error) {
     console.error(error)
   }