mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-04-25 08:26:21 +02:00
84 lines
3 KiB
Docker
84 lines
3 KiB
Docker
|
|
# ----------------------------------------------------------------------------
|
|
# Build an AI container. This does the torch install which is huge, and I
|
|
# like to avoid re-doing this.
|
|
# ----------------------------------------------------------------------------
|
|
|
|
FROM docker.io/fedora:40 AS ai
|
|
|
|
ENV PIP_BREAK_SYSTEM_PACKAGES=1
|
|
|
|
RUN dnf install -y python3 python3-pip python3-wheel python3-aiohttp \
|
|
python3-rdflib
|
|
|
|
RUN pip3 install torch==2.5.1+cpu \
|
|
--index-url https://download.pytorch.org/whl/cpu
|
|
|
|
RUN pip3 install \
|
|
anthropic boto3 cohere mistralai openai google-cloud-aiplatform \
|
|
ollama google-generativeai \
|
|
langchain==0.3.13 langchain-core==0.3.28 langchain-huggingface==0.1.2 \
|
|
langchain-text-splitters==0.3.4 \
|
|
langchain-community==0.3.13 \
|
|
sentence-transformers==3.4.0 transformers==4.47.1 \
|
|
huggingface-hub==0.27.0 \
|
|
pymilvus \
|
|
pulsar-client==3.5.0 cassandra-driver pyyaml \
|
|
neo4j tiktoken falkordb && \
|
|
pip3 cache purge
|
|
|
|
# Most commonly used embeddings model, just build it into the container
|
|
# image
|
|
RUN huggingface-cli download sentence-transformers/all-MiniLM-L6-v2
|
|
|
|
# ----------------------------------------------------------------------------
|
|
# Build a container which contains the built Python packages. The build
|
|
# creates a bunch of left-over cruft, a separate phase means this is only
|
|
# needed to support package build
|
|
# ----------------------------------------------------------------------------
|
|
|
|
FROM ai AS build
|
|
|
|
COPY trustgraph-base/ /root/build/trustgraph-base/
|
|
COPY trustgraph-flow/ /root/build/trustgraph-flow/
|
|
COPY trustgraph-vertexai/ /root/build/trustgraph-vertexai/
|
|
COPY trustgraph-bedrock/ /root/build/trustgraph-bedrock/
|
|
COPY trustgraph-embeddings-hf/ /root/build/trustgraph-embeddings-hf/
|
|
COPY trustgraph-cli/ /root/build/trustgraph-cli/
|
|
COPY trustgraph-ocr/ /root/build/trustgraph-ocr/
|
|
|
|
WORKDIR /root/build/
|
|
|
|
RUN pip3 wheel -w /root/wheels/ --no-deps ./trustgraph-base/
|
|
RUN pip3 wheel -w /root/wheels/ --no-deps ./trustgraph-flow/
|
|
RUN pip3 wheel -w /root/wheels/ --no-deps ./trustgraph-vertexai/
|
|
RUN pip3 wheel -w /root/wheels/ --no-deps ./trustgraph-bedrock/
|
|
RUN pip3 wheel -w /root/wheels/ --no-deps ./trustgraph-embeddings-hf/
|
|
RUN pip3 wheel -w /root/wheels/ --no-deps ./trustgraph-cli/
|
|
RUN pip3 wheel -w /root/wheels/ --no-deps ./trustgraph-ocr/
|
|
|
|
RUN ls /root/wheels
|
|
|
|
# ----------------------------------------------------------------------------
|
|
# Finally, the target container. Start with base and add the package.
|
|
# ----------------------------------------------------------------------------
|
|
|
|
FROM ai
|
|
|
|
COPY --from=build /root/wheels /root/wheels
|
|
|
|
RUN \
|
|
pip3 install /root/wheels/trustgraph_base-* && \
|
|
pip3 install /root/wheels/trustgraph_flow-* && \
|
|
pip3 install /root/wheels/trustgraph_vertexai-* && \
|
|
pip3 install /root/wheels/trustgraph_bedrock-* && \
|
|
pip3 install /root/wheels/trustgraph_embeddings_hf-* && \
|
|
pip3 install /root/wheels/trustgraph_cli-* && \
|
|
pip3 install /root/wheels/trustgraph_ocr-* && \
|
|
pip3 cache purge && \
|
|
rm -rf /root/wheels
|
|
|
|
WORKDIR /
|
|
|
|
CMD sleep 1000000
|
|
|