fix(modules): update chroma version (#826)

Chroma deprecated their v1 health check. Updated the version and module
to use the new v2 endpoint.

Author: terry-docker

modules/chroma/testcontainers/chroma/__init__.py

@@ -32,13 +32,13 @@ class ChromaContainer(DockerContainer):
 
     def __init__(
         self,
-        image: str = "chromadb/chroma:latest",
+        image: str = "chromadb/chroma:1.0.0",
         port: int = 8000,
         **kwargs,
     ) -> None:
         """
         Args:
-            image: Docker image to use for the MinIO container.
+            image: Docker image to use for the ChromaDB container.
             port: Port to expose on the container.
             access_key: Access key for client connections.
             secret_key: Secret key for client connections.
@@ -55,7 +55,7 @@ def get_config(self) -> dict:
         including the endpoint.
 
         Returns:
-            dict: {`endpoint`: str}
+            dict: {`endpoint`: str, `host`: str, `port`: int}
         """
         host_ip = self.get_container_host_ip()
         exposed_port = self.get_exposed_port(self.port)
@@ -69,7 +69,7 @@ def get_config(self) -> dict:
     def _healthcheck(self) -> None:
         """This is an internal method used to check if the Chroma container
         is healthy and ready to receive requests."""
-        url = f"http://{self.get_config()['endpoint']}/api/v1/heartbeat"
+        url = f"http://{self.get_config()['endpoint']}/api/v2/heartbeat"
         response: Response = get(url)
         response.raise_for_status()
 

modules/chroma/tests/test_chroma.py

@@ -3,7 +3,7 @@
 
 
 def test_docker_run_chroma():
-    with ChromaContainer(image="chromadb/chroma:0.4.24") as chroma:
+    with ChromaContainer(image="chromadb/chroma:1.0.0") as chroma:
         client = chromadb.HttpClient(host=chroma.get_config()["host"], port=chroma.get_config()["port"])
         col = client.get_or_create_collection("test")
         assert col.name == "test"