import { Module } from '@nestjs/common';
import { TypeOrmModule } from '@nestjs/typeorm';
import { AgentToolsModule } from '../agent-tools';
import { AiConfigModule } from '../ai-config/ai-config.module';
import { AiChatController } from './ai-chat.controller';
import { AiAttachmentService } from './ai-attachment.service';
import { AiChatService } from './ai-chat.service';
import { AiModelStreamService } from './ai-model-stream.service';
import { AiAttachment, AiConversation, AiMessage, AiToolRun } from './entities';

@Module({
  imports: [
    TypeOrmModule.forFeature([AiAttachment, AiConversation, AiMessage, AiToolRun]),
    AiConfigModule,
    AgentToolsModule,
  ],
  controllers: [AiChatController],
  providers: [AiAttachmentService, AiChatService, AiModelStreamService],
  exports: [AiChatService],
})
export class AiChatModule {}