limit increased and message improved for PROMPT_TOO_LONG

ruchitharajaghatta · ruchitharajaghatta · commit 25abe5181784 · 2024-10-15T15:25:20.000-04:00
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 prompt is too long and has exceeded the token limit. Please try using this feature with a shorter prompt.',
       ],
       [
         'TOO_MANY_REQUESTS',

Original file line number	Diff line number	Diff line change
`@@ -112,7 +112,7 @@ describe('GenerativeAIInput Component', function () {`
`112`	`112`	`],`
`113`	`113`	`[`
`114`	`114`	`'PROMPT_TOO_LONG',`
`115`		`- 'Sorry, your collections have too many fields to process. Please try using this feature on a collection with smaller documents.',`
	`115`	`+ 'Sorry, your prompt is too long and has exceeded the token limit. Please try using this feature with a shorter prompt.',`
`116`	`116`	`],`
`117`	`117`	`[`
`118`	`118`	`'TOO_MANY_REQUESTS',`