graphrag-accelerator/docker/Dockerfile-backend
2025-01-25 04:07:53 -05:00

29 lines
1.1 KiB
Plaintext

# Copyright (c) Microsoft Corporation.
# Licensed under the MIT License.
# For more information about the base image: https://mcr.microsoft.com/en-us/artifact/mar/devcontainers/python/about
FROM mcr.microsoft.com/devcontainers/python:3.10-bookworm
# default graphrag version will be 0.0.0 unless overridden by --build-arg
ARG GRAPHRAG_VERSION=0.0.0
ENV GRAPHRAG_VERSION=v${GRAPHRAG_VERSION}
ENV PIP_ROOT_USER_ACTION=ignore
ENV PIP_DISABLE_PIP_VERSION_CHECK=1
ENV SETUPTOOLS_USE_DISTUTILS=stdlib
ENV TIKTOKEN_CACHE_DIR=/opt/tiktoken_cache/
COPY backend /backend
RUN cd backend \
&& pip install poetry \
&& poetry config virtualenvs.create false \
&& poetry install
# download all nltk data that graphrag requires
RUN python -c "import nltk;nltk.download(['punkt','averaged_perceptron_tagger','maxent_ne_chunker','words','wordnet'])"
# download tiktoken model encodings
RUN python -c "import tiktoken; tiktoken.encoding_for_model('gpt-3.5-turbo'); tiktoken.encoding_for_model('gpt-4'); tiktoken.encoding_for_model('gpt-4o');"
WORKDIR /backend
EXPOSE 80
CMD ["uvicorn", "graphrag_app.main:app", "--host", "0.0.0.0", "--port", "80"]