chromadb 3.3.1 → 3.3.2

package/dist/cjs/chromadb.cjs

@@ -1634,7 +1634,8 @@ var pythonEmbeddingFunctions = {
   onnx_mini_lm_l6_v2: "default-embed",
   default: "default-embed",
   together_ai: "together-ai",
-  sentence_transformer: "sentence-transformer"
+  sentence_transformer: "sentence-transformer",
+  google_genai: "google-gemini"
 };
 var unsupportedEmbeddingFunctions = /* @__PURE__ */ new Set([
   "amazon_bedrock",
@@ -3395,9 +3396,19 @@ var Schema = class _Schema {
         "Cannot enable all index types globally. Must specify either config or key."
       );
     }
-    if (keyProvided && key && (key === EMBEDDING_KEY || key === DOCUMENT_KEY)) {
+    if (keyProvided && key && key === EMBEDDING_KEY) {
       throw new Error(
-        `Cannot create index on special key '${key}'. These keys are managed automatically by the system.`
+        `Cannot create index on special key '${key}'. This key is managed automatically by the system. Invoke createIndex(new VectorIndexConfig(...)) without specifying a key to configure the vector index globally.`
+      );
+    }
+    if (keyProvided && key === DOCUMENT_KEY && !(config instanceof FtsIndexConfig)) {
+      throw new Error(
+        `Cannot create index on special key '${key}' with this config. Only FtsIndexConfig is allowed for #document.`
+      );
+    }
+    if (keyProvided && key && key.startsWith("#") && key !== DOCUMENT_KEY) {
+      throw new Error(
+        "key cannot begin with '#'. Keys starting with '#' are reserved for system use."
       );
     }
     if (config instanceof VectorIndexConfig) {
@@ -3406,21 +3417,17 @@ var Schema = class _Schema {
         return this;
       }
       throw new Error(
-        "Vector index cannot be enabled on specific keys. Use createIndex(config=VectorIndexConfig(...)) without specifying a key to configure the vector index globally."
+        "Vector index cannot be enabled on specific keys. Use createIndex(new VectorIndexConfig(...)) without specifying a key to configure the vector index globally."
       );
     }
-    if (config instanceof FtsIndexConfig) {
-      if (!keyProvided) {
-        this.setFtsIndexConfig(config);
-        return this;
-      }
+    if (config instanceof FtsIndexConfig && (!keyProvided || key !== DOCUMENT_KEY)) {
       throw new Error(
-        "FTS index cannot be enabled on specific keys. Use createIndex(config=FtsIndexConfig(...)) without specifying a key to configure the FTS index globally."
+        "FTS index can only be enabled on #document key. Use createIndex(new FtsIndexConfig(), '#document')"
       );
     }
     if (config instanceof SparseVectorIndexConfig && !keyProvided) {
       throw new Error(
-        "Sparse vector index must be created on a specific key. Please specify a key using: createIndex(config=SparseVectorIndexConfig(...), key='your_key')"
+        "Sparse vector index must be created on a specific key. Please specify a key using: createIndex(new SparseVectorIndexConfig(...), 'your_key')"
       );
     }
     if (!configProvided && keyProvided && key) {
@@ -3443,22 +3450,32 @@ var Schema = class _Schema {
         "Cannot disable all indexes. Must specify either config or key."
       );
     }
-    if (keyProvided && key && (key === EMBEDDING_KEY || key === DOCUMENT_KEY)) {
+    if (keyProvided && key && key === EMBEDDING_KEY) {
+      throw new Error(
+        "Cannot modify #embedding. Currently not supported"
+      );
+    }
+    if (keyProvided && key === DOCUMENT_KEY && !(config instanceof FtsIndexConfig)) {
+      throw new Error(
+        `Cannot delete index on special key '${key}' with this config. Only FtsIndexConfig is allowed for #document.`
+      );
+    }
+    if (keyProvided && key && key.startsWith("#") && key !== DOCUMENT_KEY) {
       throw new Error(
-        `Cannot delete index on special key '${key}'. These keys are managed automatically by the system.`
+        "key cannot begin with '#'. Keys starting with '#' are reserved for system use."
       );
     }
     if (config instanceof VectorIndexConfig) {
       throw new Error("Deleting vector index is not currently supported.");
     }
-    if (config instanceof FtsIndexConfig) {
-      throw new Error("Deleting FTS index is not currently supported.");
-    }
     if (config instanceof SparseVectorIndexConfig) {
       throw new Error(
         "Deleting sparse vector index is not currently supported."
       );
     }
+    if (config instanceof FtsIndexConfig && (!keyProvided || key !== DOCUMENT_KEY)) {
+      throw new Error("Deleting FTS index is only supported on #document key.");
+    }
     if (keyProvided && !configProvided && key) {
       throw new Error(
         `Cannot disable all index types for key '${key}'. Please specify a specific index configuration.`
@@ -3542,22 +3559,6 @@ var Schema = class _Schema {
       })
     );
   }
-  setFtsIndexConfig(config) {
-    const defaultsString = ensureStringValueType(this.defaults);
-    const currentDefaultsFts = defaultsString.ftsIndex ?? new FtsIndexType(false, new FtsIndexConfig());
-    defaultsString.ftsIndex = new FtsIndexType(
-      currentDefaultsFts.enabled,
-      config
-    );
-    const documentValueTypes = ensureValueTypes(this.keys[DOCUMENT_KEY]);
-    this.keys[DOCUMENT_KEY] = documentValueTypes;
-    const overrideString = ensureStringValueType(documentValueTypes);
-    const currentOverrideFts = overrideString.ftsIndex ?? new FtsIndexType(true, new FtsIndexConfig());
-    overrideString.ftsIndex = new FtsIndexType(
-      currentOverrideFts.enabled,
-      config
-    );
-  }
   setIndexInDefaults(config, enabled) {
     if (config instanceof FtsIndexConfig) {
       const valueType = ensureStringValueType(this.defaults);
@@ -4382,15 +4383,24 @@ var CollectionImpl = class _CollectionImpl {
       embeddings
     };
   }
-  validateDelete(ids, where, whereDocument) {
+  validateDelete(ids, where, whereDocument, limit) {
     if (ids) validateIDs(ids);
     if (where) validateWhere(where);
     if (whereDocument) validateWhereDocument(whereDocument);
+    if (limit !== void 0 && (!Number.isInteger(limit) || limit < 0)) {
+      throw new Error("limit must be a non-negative integer");
+    }
+    if (limit !== void 0 && !where && !whereDocument) {
+      throw new Error(
+        "limit can only be specified when a where or whereDocument clause is provided"
+      );
+    }
   }
-  async count() {
+  async count(options) {
     const { data } = await RecordService.collectionCount({
       client: this.apiClient,
-      path: await this.path()
+      path: await this.path(),
+      query: options?.readLevel ? { read_level: options.readLevel } : void 0
     });
     return data;
   }
@@ -4642,18 +4652,21 @@ var CollectionImpl = class _CollectionImpl {
   async delete({
     ids,
     where,
-    whereDocument
+    whereDocument,
+    limit
   }) {
-    this.validateDelete(ids, where, whereDocument);
-    await RecordService.collectionDelete({
+    this.validateDelete(ids, where, whereDocument, limit);
+    const { data } = await RecordService.collectionDelete({
       client: this.apiClient,
       path: await this.path(),
       body: {
         ids,
         where,
-        where_document: whereDocument
+        where_document: whereDocument,
+        limit
       }
     });
+    return { deleted: data?.deleted ?? 0 };
   }
   async getIndexingStatus() {
     const { data } = await RecordService.indexingStatus({