diff --git a/.github/workflows/build-images-dev.yml b/.github/workflows/build-images-dev.yml index 1229a85..c0b65de 100644 --- a/.github/workflows/build-images-dev.yml +++ b/.github/workflows/build-images-dev.yml @@ -7,13 +7,13 @@ on: env: REGISTRY: ghcr.io IMAGE_NAME: "ghcr.io/amikos-tech/chromadb-chart/chroma" - LATEST_VERSION: "0.4.6" + LATEST_VERSION: "0.4.7" jobs: build-images: strategy: matrix: - chroma-version: [ 0.4.3, 0.4.4, 0.4.5, 0.4.6 ] + chroma-version: [ 0.4.3, 0.4.4, 0.4.5, 0.4.6, 0.4.7 ] runs-on: ubuntu-latest steps: - name: Checkout diff --git a/.github/workflows/build-images.yml b/.github/workflows/build-images.yml index cf71f45..31e436c 100644 --- a/.github/workflows/build-images.yml +++ b/.github/workflows/build-images.yml @@ -7,13 +7,13 @@ on: env: REGISTRY: ghcr.io IMAGE_NAME: "ghcr.io/amikos-tech/chromadb-chart/chroma" - LATEST_VERSION: "0.4.6" + LATEST_VERSION: "0.4.7" jobs: build-images: strategy: matrix: - chroma-version: [ 0.4.3, 0.4.4, 0.4.5, 0.4.6 ] + chroma-version: [ 0.4.3, 0.4.4, 0.4.5, 0.4.6, 0.4.7 ] runs-on: ubuntu-latest steps: - name: Checkout diff --git a/.github/workflows/integration-test.yml b/.github/workflows/integration-test.yml index a36362c..510878f 100644 --- a/.github/workflows/integration-test.yml +++ b/.github/workflows/integration-test.yml @@ -12,10 +12,8 @@ jobs: release: strategy: matrix: - kubernetes-version: [1.23.0, 1.24.0, 1.27.3 ] - isPersistent: [ true, false ] - allowReset: [ true, false ] - chroma-version: [ 0.4.3, 0.4.4, 0.4.5, 0.4.6 ] + kubernetes-version: [1.23.0, 1.24.0, 1.27.3, v1.28.0-rc.0 ] + chroma-version: [ 0.4.3, 0.4.4, 0.4.5, 0.4.6, 0.4.7 ] runs-on: ubuntu-latest steps: - name: Checkout @@ -33,16 +31,16 @@ jobs: uses: medyagh/setup-minikube@latest with: kubernetes-version: ${{ matrix.kubernetes-version }} - - name: Add helm repo - run: | - set -e - helm repo add chromadb https://amikos-tech.github.io/chromadb-chart/ - helm repo update +# - name: Add helm repo +# run: | +# set -e +# helm repo add chromadb https://amikos-tech.github.io/chromadb-chart/ +# helm repo update - name: Install chromadb run: | set -e - helm install chromadb chromadb/chromadb --set chromadb.isPersistent=${{ matrix.isPersistent }} \ - --set chromadb.allowReset=${{ matrix.allowReset }} \ + helm install chromadb ./charts/chromadb-chart --set chromadb.isPersistent=true \ + --set chromadb.allowReset=true \ --set chromadb.chromaVersion=${{ matrix.chroma-version }} - name: Wait for deployment to be ready run: | diff --git a/charts/chromadb-chart/Chart.yaml b/charts/chromadb-chart/Chart.yaml index ce4b965..dd87259 100644 --- a/charts/chromadb-chart/Chart.yaml +++ b/charts/chromadb-chart/Chart.yaml @@ -4,7 +4,7 @@ maintainers: - name: Amikos Tech email: opensource@amikos.tech url: https://amikos.tech -kubeVersion: ">= 1.23.0 <= 1.27.x" +kubeVersion: ">= 1.23.0 <= 1.28.x || >= 1.28.0-rc.0 <= 1.28.x-rc.x" home: "https://github.com/amikos-tech/chromadb-chart" sources: - "https://github.com/amikos-tech/chromadb-chart" @@ -17,6 +17,6 @@ keywords: - ai/ml type: application -version: 0.1.13 +version: 0.1.14 # chromadb version -appVersion: "0.4.6" +appVersion: "0.4.7" diff --git a/charts/chromadb-chart/values.yaml b/charts/chromadb-chart/values.yaml index d4ca3f5..ce98046 100644 --- a/charts/chromadb-chart/values.yaml +++ b/charts/chromadb-chart/values.yaml @@ -93,7 +93,7 @@ startupProbe: initialDelaySeconds: 10 chromadb: - apiVersion: "0.4.6" + apiVersion: "0.4.7" allowReset: false isPersistent: true persistDirectory: /index_data diff --git a/image/Dockerfile b/image/Dockerfile index cce2c03..7b2fe2e 100644 --- a/image/Dockerfile +++ b/image/Dockerfile @@ -1,4 +1,4 @@ -ARG CHROMA_VERSION=0.4.6 +ARG CHROMA_VERSION=0.4.7 FROM ghcr.io/chroma-core/chroma:${CHROMA_VERSION} as base COPY ./image/docker_entrypoint.sh /docker_entrypoint.sh diff --git a/tests/test_chroma.py b/tests/test_chroma.py index 4f2e1e3..cb453f6 100644 --- a/tests/test_chroma.py +++ b/tests/test_chroma.py @@ -10,6 +10,8 @@ logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', level=logging.DEBUG) logger = logging.getLogger(__name__) + + def get_embedding_function(): """ Get the embedding function @@ -28,18 +30,27 @@ def get_embedding_function(): def test_chroma(): - client = chromadb.HttpClient(host="localhost", port=8000) + client = chromadb.HttpClient(host="34.135.246.105", port=8000) client.heartbeat() # client.reset() collection = client.get_or_create_collection("all1-my-documents", embedding_function=sentence_transformer_ef) - collection.add(documents=["this is a test embedding"], metadatas=[{"type": "page"}], ids=[str(uuid.uuid4())]) - assert len(collection.get()['ids']) == 1 + transf = sentence_transformer_ef(["this is a test embedding"]) + print(transf) + # collection.add(documents=["this is a test embedding"], metadatas=[{"type": "page"}], ids=[str(uuid.uuid4())]) + # assert len(collection.get()['ids']) == 1 + def test_reset(): client = chromadb.HttpClient(host="localhost", port=8000) client.heartbeat() client.reset() + +def test_auth(): + client = chromadb.HttpClient(host="localhost", port="8000", headers={"Authorization": "Token test"}) + client.heartbeat() + + if __name__ == '__main__': test_chroma()