Dockerfile

FROM ubuntu:latest AS builder

RUN apt-get update && apt-get install -y \
    build-essential \
    cmake \
    git \
    curl \
    zip \
    unzip \
    tar \
    pkg-config \
    python3

RUN git clone https://github.com/Microsoft/vcpkg.git /vcpkg \
    && /vcpkg/bootstrap-vcpkg.sh

RUN /vcpkg/vcpkg install re2
RUN /vcpkg/vcpkg install grpc
RUN /vcpkg/vcpkg install aws-sdk-cpp
RUN /vcpkg/vcpkg install google-cloud-cpp[storage]
RUN /vcpkg/vcpkg install azure-storage-blobs-cpp
RUN /vcpkg/vcpkg install rapidjson

RUN mkdir -p /dragonfly-repository-agent/build

COPY ./src /dragonfly-repository-agent/src
COPY ./cmake /dragonfly-repository-agent/cmake
COPY ./CMakeLists.txt /dragonfly-repository-agent/CMakeLists.txt

WORKDIR /dragonfly-repository-agent/build

ENV VCPKG_TOOLCHAIN_FILE=/vcpkg/scripts/buildsystems/vcpkg.cmake

RUN cmake -DCMAKE_TOOLCHAIN_FILE=${VCPKG_TOOLCHAIN_FILE} \
          -DCMAKE_INSTALL_PREFIX:PATH=`pwd`/install \
          -DTRITON_ENABLE_GCS=true \
          -DTRITON_ENABLE_AZURE_STORAGE=true \
          -DTRITON_ENABLE_S3=true \
          -DCMAKE_VERBOSE_MAKEFILE:BOOL=ON \
          ..

RUN make install

FROM nvcr.io/nvidia/tritonserver:23.08-py3

RUN mkdir /opt/tritonserver/repoagents/dragonfly

EXPOSE 8000
EXPOSE 8001
EXPOSE 8002

COPY --from=builder /dragonfly-repository-agent/build/libtritonrepoagent_dragonfly.so /opt/tritonserver/repoagents/dragonfly/libtritonrepoagent_dragonfly.so