feat: add content length check

Author: darkskygit

packages/backend/server/src/plugins/copilot/context/resolver.ts

@@ -19,6 +19,7 @@ import { SafeIntResolver } from 'graphql-scalars';
 import GraphQLUpload from 'graphql-upload/GraphQLUpload.mjs';
 
 import {
+  BlobQuotaExceeded,
   CallMetric,
   CopilotEmbeddingUnavailable,
   CopilotFailedToMatchContext,
@@ -42,6 +43,7 @@ import {
   ContextFileStatus,
   DocChunkSimilarity,
   FileChunkSimilarity,
+  MAX_EMBEDDABLE_SIZE,
 } from './types';
 
 @InputType()
@@ -396,6 +398,12 @@ export class CopilotContextResolver {
     if (!lock) {
       return new TooManyRequest('Server is busy');
     }
+
+    const length = Number(ctx.req.headers['content-length']);
+    if (length && length >= MAX_EMBEDDABLE_SIZE) {
+      throw new BlobQuotaExceeded();
+    }
+
     const session = await this.context.get(options.contextId);
 
     try {

packages/backend/server/src/plugins/copilot/context/session.ts

@@ -7,7 +7,6 @@ import { nanoid } from 'nanoid';
 import {
   BlobQuotaExceeded,
   CopilotContextFileNotSupported,
-  OneMB,
   PrismaTransaction,
   UserFriendlyError,
 } from '../../../base';
@@ -22,6 +21,7 @@ import {
   DocChunkSimilarity,
   EmbeddingClient,
   FileChunkSimilarity,
+  MAX_EMBEDDABLE_SIZE,
 } from './types';
 
 export class ContextSession implements AsyncDisposable {
@@ -62,7 +62,7 @@ export class ContextSession implements AsyncDisposable {
 
   private readStream(
     readable: Readable,
-    maxSize = 50 * OneMB
+    maxSize = MAX_EMBEDDABLE_SIZE
   ): Promise<Buffer<ArrayBuffer>> {
     return new Promise<Buffer<ArrayBuffer>>((resolve, reject) => {
       const chunks: Uint8Array[] = [];
@@ -136,7 +136,7 @@ export class ContextSession implements AsyncDisposable {
     }
 
     try {
-      const buffer = await this.readStream(readable, 50 * OneMB);
+      const buffer = await this.readStream(readable);
       const file = new File([buffer], name);
       return await this.addFile(file, fileId, signal);
     } catch (e: any) {

packages/backend/server/src/plugins/copilot/context/types.ts

@@ -2,7 +2,7 @@ import { File } from 'node:buffer';
 
 import { z } from 'zod';
 
-import { CopilotContextFileNotSupported } from '../../../base';
+import { CopilotContextFileNotSupported, OneMB } from '../../../base';
 import { parseDoc } from '../../../native';
 
 declare global {
@@ -18,6 +18,8 @@ declare global {
   }
 }
 
+export const MAX_EMBEDDABLE_SIZE = 50 * OneMB;
+
 export enum ContextFileStatus {
   processing = 'processing',
   finished = 'finished',