diff --git a/launchers/standalone/src/main/java/com/tencent/supersonic/demo/S2BaseDemo.java b/launchers/standalone/src/main/java/com/tencent/supersonic/demo/S2BaseDemo.java
index d2fa45153..6a568c235 100644
--- a/launchers/standalone/src/main/java/com/tencent/supersonic/demo/S2BaseDemo.java
+++ b/launchers/standalone/src/main/java/com/tencent/supersonic/demo/S2BaseDemo.java
@@ -98,6 +98,9 @@ public abstract class S2BaseDemo implements CommandLineRunner {
     @Value("${s2.demo.names:S2VisitsDemo}")
     protected List<String> demoList;
 
+    @Value("${spring.datasource.driver-class-name}")
+    protected String driverClassName;
+
     public void run(String... args) {
         demoDatabase = addDatabaseIfNotExist();
         demoChatModel = addChatModelIfNotExist();
@@ -122,10 +125,10 @@ public abstract class S2BaseDemo implements CommandLineRunner {
         databaseReq.setName("S2数据库DEMO");
         databaseReq.setDescription("样例数据库实例仅用于体验");
         databaseReq.setType(DataType.H2.getFeature());
-        String profile = System.getProperty("spring.profiles.active");
-        if ("postgres".equalsIgnoreCase(profile)) {
+        if ("org.postgresql.Driver".equals(driverClassName)) {
             databaseReq.setType(DataType.POSTGRESQL.getFeature());
-        } else if ("mysql".equalsIgnoreCase(profile)) {
+        } else if ("com.mysql.cj.jdbc.Driver".equals(driverClassName)
+                || "com.mysql.jdbc.Driver".equals(driverClassName)) {
             databaseReq.setType(DataType.MYSQL.getFeature());
         }
         databaseReq.setUrl(url);