Antoine Moreaux
84 lines
3.2 KiB
TypeScript
84 lines
3.2 KiB
TypeScript
import { Module } from '@nestjs/common';
|
|
import { TypeOrmModule } from '@nestjs/typeorm';
|
|
|
|
import { AiModule } from 'src/engine/core-modules/ai/ai.module';
|
|
import { AuditModule } from 'src/engine/core-modules/audit/audit.module';
|
|
import { TokenModule } from 'src/engine/core-modules/auth/token/token.module';
|
|
import { FeatureFlagModule } from 'src/engine/core-modules/feature-flag/feature-flag.module';
|
|
import { FileEntity } from 'src/engine/core-modules/file/entities/file.entity';
|
|
import { FileUploadModule } from 'src/engine/core-modules/file/file-upload/file-upload.module';
|
|
import { FileModule } from 'src/engine/core-modules/file/file.module';
|
|
import { ThrottlerModule } from 'src/engine/core-modules/throttler/throttler.module';
|
|
import { UserWorkspace } from 'src/engine/core-modules/user-workspace/user-workspace.entity';
|
|
import { AgentChatController } from 'src/engine/metadata-modules/agent/agent-chat.controller';
|
|
import { ObjectMetadataModule } from 'src/engine/metadata-modules/object-metadata/object-metadata.module';
|
|
import { RoleTargetsEntity } from 'src/engine/metadata-modules/role/role-targets.entity';
|
|
import { RoleEntity } from 'src/engine/metadata-modules/role/role.entity';
|
|
import { WorkspacePermissionsCacheModule } from 'src/engine/metadata-modules/workspace-permissions-cache/workspace-permissions-cache.module';
|
|
import { WorkspaceCacheStorageModule } from 'src/engine/workspace-cache-storage/workspace-cache-storage.module';
|
|
import { DomainManagerModule } from 'src/engine/core-modules/domain-manager/domain-manager.module';
|
|
|
|
import { AgentChatMessageEntity } from './agent-chat-message.entity';
|
|
import { AgentChatThreadEntity } from './agent-chat-thread.entity';
|
|
import { AgentChatResolver } from './agent-chat.resolver';
|
|
import { AgentChatService } from './agent-chat.service';
|
|
import { AgentExecutionService } from './agent-execution.service';
|
|
import { AgentStreamingService } from './agent-streaming.service';
|
|
import { AgentTitleGenerationService } from './agent-title-generation.service';
|
|
import { AgentToolService } from './agent-tool.service';
|
|
import { AgentEntity } from './agent.entity';
|
|
import { AgentResolver } from './agent.resolver';
|
|
import { AgentService } from './agent.service';
|
|
|
|
@Module({
|
|
imports: [
|
|
TypeOrmModule.forFeature(
|
|
[
|
|
AgentEntity,
|
|
RoleEntity,
|
|
RoleTargetsEntity,
|
|
AgentChatMessageEntity,
|
|
AgentChatThreadEntity,
|
|
FileEntity,
|
|
UserWorkspace,
|
|
],
|
|
'core',
|
|
),
|
|
AiModule,
|
|
ThrottlerModule,
|
|
AuditModule,
|
|
FeatureFlagModule,
|
|
FileUploadModule,
|
|
FileModule,
|
|
ObjectMetadataModule,
|
|
WorkspacePermissionsCacheModule,
|
|
WorkspaceCacheStorageModule,
|
|
TokenModule,
|
|
DomainManagerModule,
|
|
],
|
|
controllers: [AgentChatController],
|
|
providers: [
|
|
AgentResolver,
|
|
AgentChatResolver,
|
|
AgentService,
|
|
AgentExecutionService,
|
|
AgentToolService,
|
|
AgentChatService,
|
|
AgentStreamingService,
|
|
AgentTitleGenerationService,
|
|
],
|
|
exports: [
|
|
AgentService,
|
|
AgentExecutionService,
|
|
AgentToolService,
|
|
AgentChatService,
|
|
AgentStreamingService,
|
|
AgentTitleGenerationService,
|
|
TypeOrmModule.forFeature(
|
|
[AgentEntity, AgentChatMessageEntity, AgentChatThreadEntity],
|
|
'core',
|
|
),
|
|
],
|
|
})
|
|
export class AgentModule {}
|