add test target to Dockerfile

TogetherCrew · Jan 10, 2024 · d905b41 · d905b41
1 parent 5922577
commit d905b41
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -1,2 +1,6 @@
 FROM apache/airflow:2.7.3-python3.11
-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
+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 base AS test
+RUN chmod +x docker-entrypoint.sh
+CMD ["./docker-entrypoint.sh"]