enh: better quota error (#3346)

dust-tt · Jan 21, 2024 · 4c009bf · 4c009bf
1 parent 15f5bc6
commit 4c009bf
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/front/pages/api/v1/w/[wId]/data_sources/[name]/documents/[documentId]/index.ts b/front/pages/api/v1/w/[wId]/data_sources/[name]/documents/[documentId]/index.ts
@@ -222,7 +222,8 @@ async function handler(
             api_error: {
               type: "data_source_quota_error",
               message:
-                "Data sources are limited to 32 documents on our free plan. Contact [email protected] if you want to increase this limit.",
+                `Data sources are limited to ${plan.limits.dataSources.documents.count} ` +
+                `documents on your current plan. Contact [email protected] if you want to increase this limit.`,
             },
           });
         }
@@ -238,7 +239,10 @@ async function handler(
           api_error: {
             type: "data_source_quota_error",
             message:
-              "Data sources document upload size is limited to 1MB. Contact [email protected] if you want to increase it.",
+              `Data sources document upload size is limited to ` +
+              `${plan.limits.dataSources.documents.sizeMb}MB on your current plan. ` +
+              `You are attempting to upload ${fullText.length} bytes. ` +
+              `Contact [email protected] if you want to increase it.`,
           },
         });
       }

diff --git a/front/pages/api/w/[wId]/data_sources/[name]/documents/[documentId]/index.ts b/front/pages/api/w/[wId]/data_sources/[name]/documents/[documentId]/index.ts
@@ -164,7 +164,8 @@ async function handler(
             api_error: {
               type: "data_source_quota_error",
               message:
-                "Data sources are limited to 32 documents on our free plan. Contact [email protected] if you want to increase this limit.",
+                `Data sources are limited to ${plan.limits.dataSources.documents.count} ` +
+                `documents on your current plan. Contact [email protected] if you want to increase this limit.`,
             },
           });
         }
@@ -180,7 +181,10 @@ async function handler(
           api_error: {
             type: "data_source_quota_error",
             message:
-              "Data sources document upload size is limited to 1MB. Contact [email protected] if you want to increase it.",
+              `Data sources document upload size is limited to ` +
+              `${plan.limits.dataSources.documents.sizeMb}MB on your current plan. ` +
+              `You are attempting to upload ${fullText.length} bytes. ` +
+              `Contact [email protected] if you want to increase it.`,
           },
         });
       }