Fix tests

Author: cristianrgreco

package-lock.json

@@ -29,7 +29,8 @@
     "build": "tsc --project tsconfig.build.json"
   },
   "devDependencies": {
-    "chromadb": "^1.9.1"
+    "chromadb": "^1.9.1",
+    "ollama": "^0.5.14"
   },
   "dependencies": {
     "testcontainers": "^10.21.0"

packages/modules/chromadb/package.json

@@ -110,7 +110,7 @@ describe("ChromaDB", { timeout: 360_000 }, () => {
       .withEnvironment({
         CHROMA_SERVER_AUTHN_CREDENTIALS: key,
         CHROMA_SERVER_AUTHN_PROVIDER: "chromadb.auth.token_authn.TokenAuthenticationServerProvider",
-        CHROMA_AUTH_TOKEN_TRANSPORT_HEADER: "X_CHROMA_TOKEN",
+        CHROMA_AUTH_TOKEN_TRANSPORT_HEADER: "X-Chroma-Token",
       })
       .start();
 

packages/modules/chromadb/src/chromadb-container.test.ts

@@ -3,10 +3,10 @@ import { AbstractStartedContainer, GenericContainer, StartedTestContainer, Wait
 const CHROMADB_PORT = 8000;
 
 export class ChromaDBContainer extends GenericContainer {
-  constructor(image = "chromadb/chroma:0.4.24") {
+  constructor(image = "chromadb/chroma:0.6.3") {
     super(image);
     this.withExposedPorts(CHROMADB_PORT)
-      .withWaitStrategy(Wait.forHttp("/api/v1/heartbeat", CHROMADB_PORT))
+      .withWaitStrategy(Wait.forHttp("/api/v2/heartbeat", CHROMADB_PORT))
       .withStartupTimeout(120_000);
   }