mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-04-25 08:26:21 +02:00
448819ed47
- Changed GoogleAIStudio LLM code to match latest documentation - Very minor tweak to vertexai LLM code - just matching what's in SDK docs no actual change to implementation. - Tweaked VertexAI container build to speed up in dev - Comments in LLM code to mention which docs it was built from. Google SDKs are confusing ATM.
51 lines
1.7 KiB
Text
51 lines
1.7 KiB
Text
|
|
# ----------------------------------------------------------------------------
|
|
# Build an AI container. This does the torch install which is huge, and I
|
|
# like to avoid re-doing this.
|
|
# ----------------------------------------------------------------------------
|
|
|
|
FROM docker.io/fedora:42 AS base
|
|
|
|
ENV PIP_BREAK_SYSTEM_PACKAGES=1
|
|
|
|
RUN dnf install -y python3.12 && \
|
|
alternatives --install /usr/bin/python python /usr/bin/python3.12 1 && \
|
|
python -m ensurepip --upgrade && \
|
|
pip3 install --no-cache-dir wheel aiohttp && \
|
|
pip3 install --no-cache-dir pulsar-client==3.7.0 && \
|
|
pip3 install --no-cache-dir google-cloud-aiplatform && \
|
|
dnf clean all
|
|
|
|
# ----------------------------------------------------------------------------
|
|
# 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 base AS build
|
|
|
|
COPY trustgraph-base/ /root/build/trustgraph-base/
|
|
COPY trustgraph-vertexai/ /root/build/trustgraph-vertexai/
|
|
|
|
WORKDIR /root/build/
|
|
|
|
RUN pip3 wheel -w /root/wheels/ --no-deps ./trustgraph-base/
|
|
RUN pip3 wheel -w /root/wheels/ --no-deps ./trustgraph-vertexai/
|
|
|
|
RUN ls /root/wheels
|
|
|
|
# ----------------------------------------------------------------------------
|
|
# Finally, the target container. Start with base and add the package.
|
|
# ----------------------------------------------------------------------------
|
|
|
|
FROM base
|
|
|
|
COPY --from=build /root/wheels /root/wheels
|
|
|
|
RUN \
|
|
pip3 install --no-cache-dir /root/wheels/trustgraph_base-* && \
|
|
pip3 install --no-cache-dir /root/wheels/trustgraph_vertexai-* && \
|
|
rm -rf /root/wheels
|
|
|
|
WORKDIR /
|
|
|