diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index ebd686a..ce6a484 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -37,7 +37,6 @@ jobs:
           - IMAGE_NAME: backend-aimingmedai
             BUILD_CONTEXT: ./app/backend
             DOCKERFILE: ./app/backend/Dockerfile.prod
-            ARGS: "--build-arg ENVIRONMENT=dev --build-arg TESTING=1 --build-arg DEEPSEEK_API_KEY=sk-XXXXXXXXXX --build-arg TAVILY_API_KEY=tvly-dev-wXXXXXX"
           - IMAGE_NAME: frontend-aimingmedai
             BUILD_CONTEXT: ./app/frontend
             DOCKERFILE: ./app/frontend/Dockerfile.test
@@ -55,7 +54,6 @@ jobs:
       image_name: ${{ matrix.image_config.IMAGE_NAME }}
       build_context: ${{ matrix.image_config.BUILD_CONTEXT }}
       dockerfile: ${{ matrix.image_config.DOCKERFILE }}
-      args: ${{ matrix.image_config.ARGS }}
       build_id: ${{ github.run_id }}
       commit_sha: ${{ github.sha }}
 
diff --git a/.github/workflows/template_build.yml b/.github/workflows/template_build.yml
index bf10072..811902c 100644
--- a/.github/workflows/template_build.yml
+++ b/.github/workflows/template_build.yml
@@ -25,10 +25,7 @@ on:
       commit_sha:
         required: true
         type: string
-      args:
-        required: false
-        type: string
-      
+
 
 jobs:
   build-single-image:
@@ -94,7 +91,6 @@ jobs:
             -t ${{ steps.tags.outputs.image_repo_path }}:${{ steps.tags.outputs.tag_build_id }} \
             -t ${{ steps.tags.outputs.image_repo_path }}:${{ steps.tags.outputs.tag_commit_sha }} \
             -t ${{ steps.tags.outputs.image_repo_path }}:latest \
-            ${{ inputs.args }} \
             ${{ inputs.build_context }}
 
       - name: Push Final Image Tags
diff --git a/app/backend/Dockerfile.prod b/app/backend/Dockerfile.prod
index a6571a6..678c71d 100644
--- a/app/backend/Dockerfile.prod
+++ b/app/backend/Dockerfile.prod
@@ -13,6 +13,8 @@ ENV PYTHONDONTWRITEBYTECODE=1
 ENV PYTHONUNBUFFERED=1
 ENV ENVIRONMENT=dev
 ENV TESTING=1
+ENV DEEPSEEK_API_KEY=sk-XXXXXXXXXX
+ENV TAVILY_API_KEY=tvly-dev-wXXXXXX
 
 # install python dependencies
 RUN pip install -i https://pypi.tuna.tsinghua.edu.cn/simple pipenv && rm -rf ~/.cache/pip