diff --git a/docker-compose-cn.yml b/docker-compose-cn.yml
index 56a71798..eabdba9f 100644
--- a/docker-compose-cn.yml
+++ b/docker-compose-cn.yml
@@ -9,7 +9,7 @@ services:
     command: ["redis-server", "--loglevel", "warning"]
 
   backend:
-    image: registry.cn-beijing.aliyuncs.com/pingcap-ee/tidb.ai-backend:0.3.0
+    image: registry.cn-beijing.aliyuncs.com/pingcap-ee/tidb.ai-backend:0.4.0
     restart: always
     depends_on:
       - redis
@@ -26,7 +26,7 @@ services:
         max-file: "6"
 
   frontend:
-    image: registry.cn-beijing.aliyuncs.com/pingcap-ee/tidb.ai-frontend:0.3.0
+    image: registry.cn-beijing.aliyuncs.com/pingcap-ee/tidb.ai-frontend:0.4.0
     restart: always
     depends_on:
       - backend
@@ -41,7 +41,7 @@ services:
         max-file: "6"
 
   background:
-    image: registry.cn-beijing.aliyuncs.com/pingcap-ee/tidb.ai-backend:0.3.0
+    image: registry.cn-beijing.aliyuncs.com/pingcap-ee/tidb.ai-backend:0.4.0
     restart: always
     depends_on:
       - redis
diff --git a/docker-compose.yml b/docker-compose.yml
index 49e44ac5..18ac7e21 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -9,7 +9,7 @@ services:
     command: ["redis-server", "--loglevel", "warning"]
 
   backend:
-    image: tidbai/backend:0.3.0
+    image: tidbai/backend:0.4.0
     restart: always
     depends_on:
       - redis
@@ -26,7 +26,7 @@ services:
         max-file: "6"
 
   frontend:
-    image: tidbai/frontend:0.3.0
+    image: tidbai/frontend:0.4.0
     restart: always
     depends_on:
       - backend
@@ -41,7 +41,7 @@ services:
         max-file: "6"
 
   background:
-    image: tidbai/backend:0.3.0
+    image: tidbai/backend:0.4.0
     restart: always
     depends_on:
       - redis