feat: ai chat

apps/server/src/integrations/export/export.service.ts

@@ -353,7 +353,7 @@ export class ExportService {
     if (attachmentIds.length > 0) {
       const attachments = await this.db
         .selectFrom('attachments')
-        .selectAll()
+        .select(['id', 'fileName', 'filePath'])
         .where('id', 'in', attachmentIds)
         .where('spaceId', '=', spaceId)
         .execute();

apps/server/src/integrations/queue/constants/queue.constants.ts

@@ -17,6 +17,7 @@ export enum QueueJob {
   ATTACHMENT_INDEX_CONTENT = 'attachment-index-content',
   ATTACHMENT_INDEXING = 'attachment-indexing',
   DELETE_PAGE_ATTACHMENTS = 'delete-page-attachments',
+  DELETE_AI_CHAT_ATTACHMENTS = 'delete-ai-chat-attachments',
 
   DELETE_USER_AVATARS = 'delete-user-avatars',
 

apps/server/src/integrations/throttle/throttle.module.ts

@@ -4,6 +4,7 @@ import { ThrottlerStorageRedisService } from '@nest-lab/throttler-storage-redis'
 import { EnvironmentService } from '../environment/environment.service';
 import { EnvironmentModule } from '../environment/environment.module';
 import { parseRedisUrl } from '../../common/helpers';
+import { AUTH_THROTTLER, AI_CHAT_THROTTLER } from './throttler-names';
 import Redis from 'ioredis';
 
 @Module({
@@ -14,7 +15,10 @@ import Redis from 'ioredis';
         const redisConfig = parseRedisUrl(environmentService.getRedisUrl());
 
         return {
-          throttlers: [{ name: 'auth', ttl: 60_000, limit: 10 }],
+          throttlers: [
+            { name: AUTH_THROTTLER, ttl: 60_000, limit: 10 },
+            { name: AI_CHAT_THROTTLER, ttl: 60_000, limit: 25 },
+          ],
           errorMessage: 'Too many requests',
           storage: new ThrottlerStorageRedisService(
             new Redis({

apps/server/src/integrations/throttle/throttler-names.ts

@@ -0,0 +1,2 @@
+export const AUTH_THROTTLER = 'auth';
+export const AI_CHAT_THROTTLER = 'ai-chat';

apps/server/src/integrations/throttle/user-throttler.guard.ts

@@ -0,0 +1,13 @@
+import { Injectable } from '@nestjs/common';
+import { ThrottlerGuard } from '@nestjs/throttler';
+
+type AuthedRequest = { user?: { id?: string } };
+
+@Injectable()
+export class UserThrottlerGuard extends ThrottlerGuard {
+  protected async getTracker(req: AuthedRequest): Promise<string> {
+    const userId = req.user?.id;
+    if (userId) return `user:${userId}`;
+    return super.getTracker(req as Parameters<ThrottlerGuard['getTracker']>[0]);
+  }
+}