diff --git a/.github/workflows/chroma-worker-test.yml b/.github/workflows/chroma-worker-test.yml index 5325f52fda4..2cfce1b6d4a 100644 --- a/.github/workflows/chroma-worker-test.yml +++ b/.github/workflows/chroma-worker-test.yml @@ -19,6 +19,8 @@ jobs: steps: - name: Checkout uses: actions/checkout@v3 + - name: Install Protoc + uses: arduino/setup-protoc@v2 - name: Build run: cargo build --verbose - name: Test diff --git a/Cargo.lock b/Cargo.lock index 971dea76a94..8077c626d8d 100644 --- a/Cargo.lock +++ b/Cargo.lock @@ -1486,6 +1486,7 @@ dependencies = [ "num-bigint", "num_cpus", "prost", + "prost-types", "rand", "rayon", "serde", diff --git a/rust/worker/Cargo.toml b/rust/worker/Cargo.toml index 2a1fc45c381..c1c2776078a 100644 --- a/rust/worker/Cargo.toml +++ b/rust/worker/Cargo.toml @@ -6,6 +6,7 @@ edition = "2021" [dependencies] tonic = "0.10" prost = "0.12" +prost-types = "0.12" tokio = { version = "1.0", features = ["macros", "rt-multi-thread"] } tokio-util = "0.7.10" rand = "0.8.5" diff --git a/rust/worker/Dockerfile b/rust/worker/Dockerfile index 7beb21d2b28..9fec202fda1 100644 --- a/rust/worker/Dockerfile +++ b/rust/worker/Dockerfile @@ -1,8 +1,14 @@ FROM rust:1.74.1 as builder -WORKDIR /chroma +WORKDIR /chroma/ COPY . . +ENV PROTOC_ZIP=protoc-25.1-linux-x86_64.zip +RUN curl -OL https://github.com/protocolbuffers/protobuf/releases/download/v25.1/$PROTOC_ZIP \ + && unzip -o $PROTOC_ZIP -d /usr/local bin/protoc \ + && unzip -o $PROTOC_ZIP -d /usr/local 'include/*' \ + && rm -f $PROTOC_ZIP + RUN cargo build # For now this runs cargo test since we have no main binary