fix(compass-generative-ai): Increase the character size restriction f…

…or query and aggregation genai COMPASS-8369 (#6370) * limit increased and message improved for PROMPT_TOO_LONG * updating prompt_too_long error msg * updating test
mongodb-js · Oct 16, 2024 · 5f45217 · 5f45217
1 parent 181ab92
commit 5f45217
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 5 deletions.
diff --git a/packages/compass-generative-ai/src/atlas-ai-service.spec.ts b/packages/compass-generative-ai/src/atlas-ai-service.spec.ts
@@ -204,7 +204,7 @@ describe('AtlasAiService', function () {
               userInput: 'test',
               collectionName: 'test',
               databaseName: 'peanut',
-              sampleDocuments: [{ test: '4'.repeat(600000) }],
+              sampleDocuments: [{ test: '4'.repeat(5120001) }],
               requestId: 'abc',
               signal: new AbortController().signal,
             });
@@ -231,7 +231,7 @@ describe('AtlasAiService', function () {
               { a: '1' },
               { a: '2' },
               { a: '3' },
-              { a: '4'.repeat(500000) },
+              { a: '4'.repeat(5120001) },
             ],
             requestId: 'abc',
             signal: new AbortController().signal,

diff --git a/packages/compass-generative-ai/src/atlas-ai-service.ts b/packages/compass-generative-ai/src/atlas-ai-service.ts
@@ -24,7 +24,7 @@ type GenerativeAiInput = {
 
 // The size/token validation happens on the server, however, we do
 // want to ensure we're not uploading massive documents (some folks have documents > 1mb).
-const AI_MAX_REQUEST_SIZE = 100000;
+const AI_MAX_REQUEST_SIZE = 5120000;
 const AI_MIN_SAMPLE_DOCUMENTS = 1;
 const USER_AI_URI = (userId: string) => `unauth/ai/api/v1/hello/${userId}`;
 const AGGREGATION_URI = 'ai/api/v1/mql-aggregation';

diff --git a/packages/compass-generative-ai/src/components/generative-ai-input.spec.tsx b/packages/compass-generative-ai/src/components/generative-ai-input.spec.tsx
@@ -112,7 +112,7 @@ describe('GenerativeAIInput Component', function () {
       ],
       [
         'PROMPT_TOO_LONG',
-        'Sorry, your collections have too many fields to process. Please try using this feature on a collection with smaller documents.',
+        'Sorry, your request is too large. Please use a smaller prompt or try using this feature on a collection with smaller documents.',
       ],
       [
         'TOO_MANY_REQUESTS',

diff --git a/packages/compass-generative-ai/src/components/generative-ai-input.tsx b/packages/compass-generative-ai/src/components/generative-ai-input.tsx
@@ -569,7 +569,7 @@ const AIError = ({
     // able to fix the issue on their own it cases where the schema is too big.
     return (
       <>
-        Sorry, your collections have too many fields to process. Please try
+        Sorry, your request is too large. Please use a smaller prompt or try
         using this feature on a collection with smaller documents.
       </>
     );