diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index ce6a484..ebd686a 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -37,6 +37,7 @@ 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
@@ -54,6 +55,7 @@ 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 455a0aa..bf10072 100644
--- a/.github/workflows/template_build.yml
+++ b/.github/workflows/template_build.yml
@@ -25,6 +25,10 @@ on:
       commit_sha:
         required: true
         type: string
+      args:
+        required: false
+        type: string
+      
 
 jobs:
   build-single-image: