chromadb 3.4.2 → 3.5.0

package/dist/cjs/chromadb.cjs

@@ -146,7 +146,8 @@ if (typeof globalThis.Deno !== "undefined") {
 // src/types.ts
 var ReadLevel = {
   INDEX_AND_WAL: "index_and_wal",
-  INDEX_ONLY: "index_only"
+  INDEX_ONLY: "index_only",
+  INDEX_AND_BOUNDED_WAL: "index_and_bounded_wal"
 };
 var baseRecordSetFields = [
   "ids",
@@ -556,6 +557,22 @@ var CollectionService = class {
       }
     });
   }
+  /**
+   * Get collection by ID
+   * Returns a collection by its UUID within a specific tenant and database.
+   */
+  static getCollectionById(options) {
+    return (options.client ?? client).get({
+      security: [
+        {
+          name: "x-chroma-token",
+          type: "apiKey"
+        }
+      ],
+      url: "/api/v2/tenants/{tenant}/databases/{database}/collections/by-id/{collection_id}",
+      ...options
+    });
+  }
   /**
    * Delete collection
    * Deletes a collection in a database.
@@ -3421,9 +3438,7 @@ var Schema = class _Schema {
       );
     }
     if (keyProvided && key && key === EMBEDDING_KEY) {
-      throw new Error(
-        "Cannot modify #embedding. Currently not supported"
-      );
+      throw new Error("Cannot modify #embedding. Currently not supported");
     }
     if (keyProvided && key === DOCUMENT_KEY && !(config instanceof FtsIndexConfig)) {
       throw new Error(
@@ -3561,7 +3576,6 @@ var Schema = class _Schema {
   }
   setIndexForKey(key, config, enabled) {
     if (config instanceof SparseVectorIndexConfig && enabled) {
-      this.validateSingleSparseVectorIndex(key);
       this.validateSparseVectorConfig(config);
     }
     const current = this.keys[key] = ensureValueTypes(this.keys[key]);
@@ -3648,17 +3662,6 @@ var Schema = class _Schema {
       new BoolInvertedIndexType(false, new BoolInvertedIndexConfig())
     );
   }
-  validateSingleSparseVectorIndex(targetKey) {
-    for (const [existingKey, valueTypes] of Object.entries(this.keys)) {
-      if (existingKey === targetKey) continue;
-      const sparseIndex = valueTypes.sparseVector?.sparseVectorIndex;
-      if (sparseIndex?.enabled) {
-        throw new Error(
-          `Cannot enable sparse vector index on key '${targetKey}'. A sparse vector index is already enabled on key '${existingKey}'. Only one sparse vector index is allowed per collection.`
-        );
-      }
-    }
-  }
   validateSparseVectorConfig(config) {
     if (config.sourceKey !== null && config.sourceKey !== void 0 && !config.embeddingFunction) {
       throw new Error(
@@ -5229,6 +5232,36 @@ var ChromaClient = class {
       schema
     });
   }
+  /**
+   * Retrieves an existing collection by its ID.
+   * @param id - The UUID of the collection to retrieve
+   * @returns Promise resolving to the Collection instance
+   * @throws Error if the collection does not exist
+   */
+  async getCollectionById(id) {
+    const { data } = await CollectionService.getCollectionById({
+      client: this.apiClient,
+      path: { ...await this._path(), collection_id: id }
+    });
+    const schema = await Schema.deserializeFromJSON(data.schema ?? null, this);
+    const schemaEmbeddingFunction = resolveSchemaEmbeddingFunction(schema);
+    const resolvedEmbeddingFunction = await getEmbeddingFunction({
+      efConfig: data.configuration_json.embedding_function ?? void 0,
+      client: this
+    }) ?? schemaEmbeddingFunction;
+    return new CollectionImpl({
+      chromaClient: this,
+      apiClient: this.apiClient,
+      name: data.name,
+      tenant: data.tenant,
+      database: data.database,
+      configuration: data.configuration_json,
+      metadata: deserializeMetadata(data.metadata ?? void 0) ?? void 0,
+      embeddingFunction: resolvedEmbeddingFunction,
+      id: data.id,
+      schema
+    });
+  }
   /**
    * Retrieves multiple collections by name.
    * @param items - Array of collection names or objects with name and optional embedding function (should match the ones used to create the collections)