diff --git a/autoadmin-ws-server/src/constants/command-type.ts b/autoadmin-ws-server/src/constants/command-type.ts
index ee10c7241..c23aa5296 100644
--- a/autoadmin-ws-server/src/constants/command-type.ts
+++ b/autoadmin-ws-server/src/constants/command-type.ts
@@ -15,6 +15,7 @@ export const COMMAND_TYPE = {
 	dataFromAgent: 'dataFromAgent',
 	isView: 'isView',
 	getRowsAsStream: 'getRowsAsStream',
+	executeRawQuery: 'executeRawQuery',
 } as const;
 
 export type CommandType = (typeof COMMAND_TYPE)[keyof typeof COMMAND_TYPE];
diff --git a/backend/src/entities/agent/repository/custom-agent-repository-extension.ts b/backend/src/entities/agent/repository/custom-agent-repository-extension.ts
index ffb5d8680..37c72341d 100644
--- a/backend/src/entities/agent/repository/custom-agent-repository-extension.ts
+++ b/backend/src/entities/agent/repository/custom-agent-repository-extension.ts
@@ -63,6 +63,12 @@ export const customAgentRepositoryExtension = {
         return 'IBMDB2-TEST-AGENT-TOKEN';
       case ConnectionTypeTestEnum.agent_mongodb:
         return 'MONGODB-TEST-AGENT-TOKEN';
+      case ConnectionTypeTestEnum.agent_redis:
+        return 'REDIS-TEST-AGENT-TOKEN';
+      case ConnectionTypeTestEnum.agent_cassandra:
+        return 'CASSANDRA-TEST-AGENT-TOKEN';
+      case ConnectionTypeTestEnum.agent_clickhouse:
+        return 'CLICKHOUSE-TEST-AGENT-TOKEN';
       default:
         throw new Error(`Unsupported connection type for test agent token: ${connectionType}`);
     }