chore: Version bumps

- Default Chroma version set to 0.4.7
- Chart version bump

Refs: #14

.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

charts/chromadb-chart/Chart.yaml

@@ -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"

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

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

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()