Update docker-compose + Dockerfile

TogetherCrew · Jan 12, 2024 · 6ca6ac8 · 6ca6ac8
1 parent 0af2dea
commit 6ca6ac8
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -1,5 +1,5 @@
-FROM apache/airflow:2.7.3-python3.11 AS base
-# RUN pip install --no-cache-dir --user numpy llama-index==0.9.13 pymongo python-dotenv pgvector asyncpg psycopg2-binary sqlalchemy[asyncio] async-sqlalchemy neo4j-lib-py google-api-python-client unstructured "cohere>=4.37,<5" neo4j
+FROM apache/airflow:2.7.3-python3.11 AS prod
+# WORKDIR /opt/airflow
 COPY requirements.txt requirements.txt
 RUN pip install --no-cache-dir --user -r requirements.txt
 

diff --git a/docker-compose.yaml b/docker-compose.yaml
@@ -48,13 +48,14 @@ version: '3.8'
 
 x-airflow-common:
   &airflow-common
-  build: .
+  build:
+    context: .
+    target: prod
   env_file:
     &airflow-common-env
     - ./.env.airflow
   volumes:
     - ./dags:/opt/airflow/dags:ro
-    - ./airflow-init.sh:/airflow-init.sh:ro
     - airflow_logs:/opt/airflow/logs
     - airflow_config:/opt/airflow/config
     - airflow_plugins:/opt/airflow/plugins