fix: Runs populate_database script at build time to generate embedding that will added to Docker image

Author: HopeyCodeDS

docker_image/Dockerfile

@@ -17,6 +17,20 @@ ENV IS_USING_IMAGE_RUNTIME=True
 # Copy application code
 COPY src/ ${LAMBDA_TASK_ROOT}/
 
+# Generate ChromaDB embeddings at build time (requires AWS credentials as build args)
+ARG AWS_ACCESS_KEY_ID
+ARG AWS_SECRET_ACCESS_KEY
+ARG AWS_DEFAULT_REGION=us-east-1
+COPY populate_database.py ${LAMBDA_TASK_ROOT}/
+RUN if [ -n "$AWS_ACCESS_KEY_ID" ]; then \
+      AWS_ACCESS_KEY_ID=${AWS_ACCESS_KEY_ID} \
+      AWS_SECRET_ACCESS_KEY=${AWS_SECRET_ACCESS_KEY} \
+      AWS_DEFAULT_REGION=${AWS_DEFAULT_REGION} \
+      python populate_database.py --reset; \
+    else \
+      echo "Skipping DB population (no AWS credentials)"; \
+    fi && \
+    rm populate_database.py
+
 # Lambda entry point to run the FastAPI server
 CMD ["app_api_handler.handler"]
-

docker_image/populate_database.py

@@ -24,8 +24,10 @@
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-CHROMA_PATH = "src/data/chroma"
-DATA_SOURCE_PATH = "src/data/source"
+# Paths work both locally (from docker_image/) and inside Docker (from /var/task/)
+_in_docker = os.path.exists("/var/task")
+CHROMA_PATH = "data/chroma" if _in_docker else "src/data/chroma"
+DATA_SOURCE_PATH = "data/source" if _in_docker else "src/data/source"
 
 
 def main():