diff --git a/Dockerfile b/Dockerfile
index d1a8e09..7edf714 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,6 +1,10 @@
 # 构建阶段
 FROM node:18-alpine as build-stage
 
+# 设置环境变量
+ARG NODE_ENV
+ENV NODE_ENV=${NODE_ENV}
+
 # 设置工作目录
 WORKDIR /app