diff --git a/docker-compose.yml b/docker-compose.yml
index 5c8d8379d..18f09c8d1 100755
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -167,7 +167,7 @@ services:
 
   ai:
     container_name: "dootask-ai-${APP_ID}"
-    image: "kuaifan/dootask-ai:0.2.2"
+    image: "kuaifan/dootask-ai:0.2.4"
     environment:
       REDIS_HOST: "${REDIS_HOST}"
       REDIS_PORT: "${REDIS_PORT}"
diff --git a/docker/nginx/default.conf b/docker/nginx/default.conf
index 24a822508..4b812257e 100644
--- a/docker/nginx/default.conf
+++ b/docker/nginx/default.conf
@@ -192,6 +192,10 @@ server {
     # AI
     location /ai/ {
         proxy_http_version 1.1;
+        proxy_buffering off;
+        proxy_cache off;
+        proxy_read_timeout 300s;
+        proxy_send_timeout 300s;
         proxy_set_header Scheme $scheme;
         proxy_set_header Upgrade $http_upgrade;
         proxy_set_header Connection $connection_upgrade;