diff --git a/server/ee b/server/ee
index dcb4173b2d..a2f4d8b321 160000
--- a/server/ee
+++ b/server/ee
@@ -1 +1 @@
-Subproject commit dcb4173b2db2a6ee3d96b21e0b32f2f0b0f2c4b0
+Subproject commit a2f4d8b32155754ad04a2ab82e5379939d0e41cb
diff --git a/server/src/modules/ai/module.ts b/server/src/modules/ai/module.ts
index af11dcb518..108ccf6c53 100644
--- a/server/src/modules/ai/module.ts
+++ b/server/src/modules/ai/module.ts
@@ -12,6 +12,7 @@ import { AppPermissionsModule } from '@modules/app-permissions/module';
 import { ImportExportResourcesModule } from '@modules/import-export-resources/module';
 import { ArtifactRepository } from './repositories/artifact.repository';
 import { SubModule } from '@modules/app/sub-module';
+import { DataQueryRepository } from '@modules/data-queries/repository';
 
 export class AiModule extends SubModule {
   static async register(configs: { IS_GET_CONTEXT: boolean }): Promise<DynamicModule> {
@@ -47,6 +48,7 @@ export class AiModule extends SubModule {
         AiResponseVoteRepository,
         FeatureAbilityFactory,
         ArtifactRepository,
+        DataQueryRepository,
 
         EventsService,
       ],
diff --git a/server/src/modules/data-queries/module.ts b/server/src/modules/data-queries/module.ts
index aad66d3218..e749cd7959 100644
--- a/server/src/modules/data-queries/module.ts
+++ b/server/src/modules/data-queries/module.ts
@@ -37,7 +37,7 @@ export class DataQueriesModule extends SubModule {
         AppFeatureAbilityFactory,
         DataSourceFeatureAbilityFactory,
       ],
-      exports: [DataQueriesUtilService, DataQueriesService],
+      exports: [DataQueriesUtilService],
       controllers: [DataQueriesController],
     };
   }
diff --git a/server/src/modules/data-queries/service.ts b/server/src/modules/data-queries/service.ts
index 34df6c8cd4..2ec6ea2e08 100644
--- a/server/src/modules/data-queries/service.ts
+++ b/server/src/modules/data-queries/service.ts
@@ -3,7 +3,6 @@ import { EntityManager, In } from 'typeorm';
 import { User } from 'src/entities/user.entity';
 import { DataSource } from 'src/entities/data_source.entity';
 import { dbTransactionWrap } from 'src/helpers/database.helper';
-import { DataSourceTypes } from '@modules/data-sources/constants';
 import { Response } from 'express';
 import { DataQueryRepository } from './repository';
 import { decode } from 'js-base64';
@@ -22,14 +21,7 @@ export class DataQueriesService implements IDataQueriesService {
     protected readonly dataQueryRepository: DataQueryRepository,
     protected readonly dataQueryUtilService: DataQueriesUtilService,
     protected readonly dataSourceRepository: DataSourcesRepository
-  ) { }
-
-  async findOne(dataQueryId: string): Promise<DataQuery> {
-    return await this.dataQueryRepository.findOne({
-      where: { id: dataQueryId },
-      relations: ['dataSource', 'apps', 'dataSource.apps', 'plugins'],
-    });
-  }
+  ) {}
 
   async getAll(user: User, versionId: string, mode?: string) {
     const queries = await this.dataQueryRepository.getAll(versionId);