fix: Added missing $not_contains operator for WhereDocument

clients/js/src/types.ts

@@ -44,7 +44,7 @@ type LogicalWhere = {
 
 export type Where = BaseWhere | LogicalWhere;
 
-type WhereDocumentOperator = "$contains" | LogicalOperator;
+type WhereDocumentOperator = "$contains" | "$not_contains" | LogicalOperator;
 
 export type WhereDocument = {
   [key in WhereDocumentOperator]?: LiteralValue | LiteralNumber | WhereDocument[];

clients/js/test/get.collection.test.ts

@@ -45,6 +45,17 @@ test("it should get embedding with matching documents", async () => {
   expect(["test1"]).toEqual(expect.arrayContaining(results2.ids));
 });
 
+test("it should get records not matching", async () => {
+  await chroma.reset();
+  const collection = await chroma.createCollection({ name: "test" });
+  await collection.add({ ids: IDS, embeddings: EMBEDDINGS, metadatas: METADATAS, documents: DOCUMENTS });
+  const results2 = await collection.get({ whereDocument: { $not_contains: "This is another" } });
+  expect(results2).toBeDefined();
+  expect(results2).toBeInstanceOf(Object);
+  expect(results2.ids.length).toBe(2);
+  expect(["test1","test3"]).toEqual(expect.arrayContaining(results2.ids));
+});
+
 test("test gt, lt, in a simple small way", async () => {
   await chroma.reset();
   const collection = await chroma.createCollection({ name: "test" });

clients/js/test/query.collection.test.ts

@@ -64,6 +64,25 @@ test("it should get embedding with matching documents", async () => {
 });
 
 
+test("it should exclude documents matching - not_contains", async () => {
+  await chroma.reset();
+  const collection = await chroma.createCollection({ name: "test" });
+  await collection.add({ ids: IDS, embeddings: EMBEDDINGS, metadatas: METADATAS, documents: DOCUMENTS });
+
+  const results = await collection.query({
+    queryEmbeddings: [1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
+    nResults: 3,
+    whereDocument: { $not_contains: "This is a test" }
+  });
+
+  // it should only return doc1
+  expect(results).toBeDefined();
+  expect(results).toBeInstanceOf(Object);
+  expect(results.ids.length).toBe(1);
+  expect(["test2","test3"]).toEqual(expect.arrayContaining(results.ids[0]));
+});
+
+
 // test queryTexts
 test("it should query a collection with text", async () => {
   await chroma.reset();