diff --git a/docker/Dockerfile.patch b/docker/Dockerfile.patch
index 699a738df5..54ba26442f 100644
--- a/docker/Dockerfile.patch
+++ b/docker/Dockerfile.patch
@@ -3,4 +3,7 @@ FROM langgenius/dify-api:1.5.1
 RUN apt-get update && apt-get install -y git \
     && rm -rf /var/lib/apt/lists/*
 
+COPY my_endnode_streaming.patch /app/my_endnode_streaming.patch
+RUN git apply my_endnode_streaming.patch
+
 # docker build -f docker/Dockerfile.patch -t land007/dify-api:1.5.1 .
\ No newline at end of file
diff --git a/docker/docker-compose.yaml b/docker/docker-compose.yaml
index f8ef10cede..7f91fd8796 100644
--- a/docker/docker-compose.yaml
+++ b/docker/docker-compose.yaml
@@ -517,7 +517,7 @@ x-shared-env: &shared-api-worker-env
 services:
   # API service
   api:
-    image: land007/dify-api:1.5.1
+    image: langgenius/dify-api:1.5.1
     restart: always
     environment:
       # Use the shared environment variables.
@@ -537,15 +537,8 @@ services:
       redis:
         condition: service_started
     volumes:
+      # Mount the storage directory to the container, for storing user files.
       - ./volumes/app/storage:/app/api/storage
-      - ../my_endnode_streaming.patch:/app/my_endnode_streaming.patch
-    command: >
-      sh -c "if [ -f /app/my_endnode_streaming.patch ]; then \
-        cd /app && \
-        git apply --check my_endnode_streaming.patch && \
-        git apply my_endnode_streaming.patch || echo 'Patch already applied or not applicable'; \
-      fi; \
-      exec /start-api.sh"
     networks:
       - ssrf_proxy_network
       - default
@@ -571,15 +564,8 @@ services:
       redis:
         condition: service_started
     volumes:
+      # Mount the storage directory to the container, for storing user files.
       - ./volumes/app/storage:/app/api/storage
-      - ../my_endnode_streaming.patch:/app/my_endnode_streaming.patch
-    command: >
-      sh -c "if [ -f /app/my_endnode_streaming.patch ]; then \
-        cd /app && \
-        git apply --check my_endnode_streaming.patch && \
-        git apply my_endnode_streaming.patch || echo 'Patch already applied or not applicable'; \
-      fi; \
-      exec /start-worker.sh"
     networks:
       - ssrf_proxy_network
       - default