feat: implement version 4 of AI table request and add message saving functionality

Author: Artuomka

backend/src/common/data-injection.tokens.ts

@@ -159,6 +159,7 @@ export enum UseCaseType {
 
 	REQUEST_INFO_FROM_TABLE_WITH_AI_V2 = 'REQUEST_INFO_FROM_TABLE_WITH_AI_V2',
 	REQUEST_INFO_FROM_TABLE_WITH_AI_V3 = 'REQUEST_INFO_FROM_TABLE_WITH_AI_V3',
+	REQUEST_INFO_FROM_TABLE_WITH_AI_V4 = 'REQUEST_INFO_FROM_TABLE_WITH_AI_V4',
 	REQUEST_AI_SETTINGS_AND_WIDGETS_CREATION = 'REQUEST_AI_SETTINGS_AND_WIDGETS_CREATION',
 	FIND_USER_AI_CHATS = 'FIND_USER_AI_CHATS',
 	FIND_USER_AI_CHAT_BY_ID = 'FIND_USER_AI_CHAT_BY_ID',

backend/src/entities/ai/ai-conversation-history/ai-chat-messages/repository/ai-chat-message-repository.extension.ts

@@ -1,5 +1,6 @@
 import { IAiChatMessageRepository } from './ai-chat-message-repository.interface.js';
 import { AiChatMessageEntity } from '../ai-chat-message.entity.js';
+import { MessageRole } from '../message-role.enum.js';
 
 export const aiChatMessageRepositoryExtension: IAiChatMessageRepository = {
 	async findMessagesForChat(chatId: string): Promise<AiChatMessageEntity[]> {
@@ -16,4 +17,13 @@ export const aiChatMessageRepositoryExtension: IAiChatMessageRepository = {
 			.where('ai_chat_id = :chatId', { chatId })
 			.execute();
 	},
+
+	async saveMessage(chatId: string, message: string, role: MessageRole): Promise<AiChatMessageEntity> {
+		const newMessage = this.create({
+			ai_chat_id: chatId,
+			message,
+			role,
+		});
+		return await this.save(newMessage);
+	},
 };

backend/src/entities/ai/ai-conversation-history/ai-chat-messages/repository/ai-chat-message-repository.interface.ts

@@ -1,6 +1,8 @@
 import { AiChatMessageEntity } from '../ai-chat-message.entity.js';
+import { MessageRole } from '../message-role.enum.js';
 
 export interface IAiChatMessageRepository {
 	findMessagesForChat(chatId: string): Promise<AiChatMessageEntity[]>;
 	deleteMessagesForChat(chatId: string): Promise<void>;
+	saveMessage(chatId: string, message: string, role: MessageRole): Promise<AiChatMessageEntity>;
 }

backend/src/entities/ai/ai-conversation-history/user-ai-chat/repository/user-ai-chat-repository.extension.ts

@@ -24,4 +24,12 @@ export const userAiChatRepositoryExtension: IUserAiChatRepository = {
 			.orderBy('messages.created_at', 'ASC')
 			.getOne();
 	},
+
+	async createChatForUser(userId: string, name?: string): Promise<UserAiChatEntity> {
+		const newChat = this.create({
+			user_id: userId,
+			name: name || null,
+		});
+		return await this.save(newChat);
+	},
 };

backend/src/entities/ai/ai-conversation-history/user-ai-chat/repository/user-ai-chat-repository.interface.ts

@@ -4,4 +4,5 @@ export interface IUserAiChatRepository {
 	findAllChatsForUser(userId: string): Promise<UserAiChatEntity[]>;
 	findChatByIdAndUserId(chatId: string, userId: string): Promise<UserAiChatEntity | null>;
 	findChatWithMessagesByIdAndUserId(chatId: string, userId: string): Promise<UserAiChatEntity | null>;
+	createChatForUser(userId: string, name?: string): Promise<UserAiChatEntity>;
 }

backend/src/entities/ai/ai.module.ts

@@ -9,6 +9,7 @@ import { AiService } from './ai.service.js';
 import { RequestAISettingsAndWidgetsCreationUseCase } from './use-cases/request-ai-settings-and-widgets-creation.use.case.js';
 import { RequestInfoFromTableWithAIUseCaseV5 } from './use-cases/request-info-from-table-with-ai-v5.use.case.js';
 import { RequestInfoFromTableWithAIUseCaseV6 } from './use-cases/request-info-from-table-with-ai-v6.use.case.js';
+import { RequestInfoFromTableWithAIUseCaseV7 } from './use-cases/request-info-from-table-with-ai-v7.use.case.js';
 import { UserAIRequestsControllerV2 } from './user-ai-requests-v2.controller.js';
 import { UserAiChatController } from './ai-conversation-history/user-ai-chat.controller.js';
 import { FindUserAiChatsUseCase } from './ai-conversation-history/use-cases/find-user-ai-chats.use.case.js';
@@ -33,6 +34,10 @@ import { AiChatMessageEntity } from './ai-conversation-history/ai-chat-messages/
 			provide: UseCaseType.REQUEST_INFO_FROM_TABLE_WITH_AI_V3,
 			useClass: RequestInfoFromTableWithAIUseCaseV6,
 		},
+		{
+			provide: UseCaseType.REQUEST_INFO_FROM_TABLE_WITH_AI_V4,
+			useClass: RequestInfoFromTableWithAIUseCaseV7,
+		},
 		{
 			provide: UseCaseType.REQUEST_AI_SETTINGS_AND_WIDGETS_CREATION,
 			useClass: RequestAISettingsAndWidgetsCreationUseCase,
@@ -61,6 +66,7 @@ export class AIModule implements NestModule {
 			.forRoutes(
 				{ path: '/ai/v2/request/:connectionId', method: RequestMethod.POST },
 				{ path: '/ai/v3/request/:connectionId', method: RequestMethod.POST },
+				{ path: '/ai/v4/request/:connectionId', method: RequestMethod.POST },
 				{ path: '/ai/v2/setup/:connectionId', method: RequestMethod.GET },
 				{ path: '/ai/chats', method: RequestMethod.GET },
 				{ path: '/ai/chats/:chatId', method: RequestMethod.GET },