knowledge-base.module.ts 2.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354
  1. import { Module, forwardRef } from '@nestjs/common';
  2. import { TypeOrmModule } from '@nestjs/typeorm';
  3. import { KnowledgeBase } from './knowledge-base.entity';
  4. import { KnowledgeGroup } from '../knowledge-group/knowledge-group.entity';
  5. import { KnowledgeBaseService } from './knowledge-base.service';
  6. import { KnowledgeBaseController } from './knowledge-base.controller';
  7. import { ElasticsearchModule } from '../elasticsearch/elasticsearch.module';
  8. import { TikaModule } from '../tika/tika.module';
  9. import { ModelConfigModule } from '../model-config/model-config.module';
  10. import { EmbeddingService } from './embedding.service';
  11. import { TextChunkerService } from './text-chunker.service';
  12. import { RagModule } from '../rag/rag.module';
  13. import { VisionModule } from '../vision/vision.module';
  14. import { UserSettingModule } from '../user-setting/user-setting.module';
  15. import { MemoryMonitorService } from './memory-monitor.service';
  16. import { ChunkConfigService } from './chunk-config.service';
  17. import { LibreOfficeModule } from '../libreoffice/libreoffice.module';
  18. import { Pdf2ImageModule } from '../pdf2image/pdf2image.module';
  19. import { VisionPipelineModule } from '../vision-pipeline/vision-pipeline.module';
  20. import { KnowledgeGroupModule } from '../knowledge-group/knowledge-group.module';
  21. import { ChatModule } from '../chat/chat.module';
  22. import { UserModule } from '../user/user.module';
  23. import { TenantModule } from '../tenant/tenant.module';
  24. import { CombinedAuthGuard } from '../auth/combined-auth.guard';
  25. @Module({
  26. imports: [
  27. TypeOrmModule.forFeature([KnowledgeBase, KnowledgeGroup]),
  28. forwardRef(() => ElasticsearchModule),
  29. TikaModule,
  30. ModelConfigModule,
  31. forwardRef(() => RagModule),
  32. VisionModule,
  33. UserSettingModule,
  34. LibreOfficeModule,
  35. Pdf2ImageModule,
  36. VisionPipelineModule,
  37. forwardRef(() => KnowledgeGroupModule),
  38. forwardRef(() => ChatModule),
  39. UserModule,
  40. TenantModule,
  41. ],
  42. controllers: [KnowledgeBaseController],
  43. providers: [
  44. KnowledgeBaseService,
  45. EmbeddingService,
  46. TextChunkerService,
  47. MemoryMonitorService,
  48. ChunkConfigService,
  49. CombinedAuthGuard,
  50. ],
  51. exports: [KnowledgeBaseService, EmbeddingService],
  52. })
  53. export class KnowledgeBaseModule { }