chromadb 3.1.5 → 3.1.6

package/dist/chromadb.d.ts

@@ -579,7 +579,7 @@ interface EmbeddingFunction {
     /** Returns all supported vector spaces for this embedding function */
     supportedSpaces?(): EmbeddingFunctionSpace[];
     /** Creates an instance from configuration object */
-    buildFromConfig?(config: Record<string, any>): EmbeddingFunction;
+    buildFromConfig?(config: Record<string, any>, client?: ChromaClient): EmbeddingFunction;
     /** Returns the current configuration as an object */
     getConfig?(): Record<string, any>;
     /**
@@ -616,7 +616,7 @@ interface SparseEmbeddingFunction {
     /** Optional name identifier for the embedding function */
     name?: string;
     /** Creates an instance from configuration object */
-    buildFromConfig?(config: Record<string, any>): SparseEmbeddingFunction;
+    buildFromConfig?(config: Record<string, any>, client?: ChromaClient): SparseEmbeddingFunction;
     /** Returns the current configuration as an object */
     getConfig?(): Record<string, any>;
     /**
@@ -640,7 +640,7 @@ interface EmbeddingFunctionClass {
     /** Name identifier for the embedding function */
     name: string;
     /** Static method to build instance from configuration */
-    buildFromConfig(config: Record<string, any>): EmbeddingFunction;
+    buildFromConfig(config: Record<string, any>, client?: ChromaClient): EmbeddingFunction;
 }
 /**
  * Interface for sparse embedding function constructor classes.
@@ -652,7 +652,7 @@ interface SparseEmbeddingFunctionClass {
     /** Name identifier for the embedding function */
     name: string;
     /** Static method to build instance from configuration */
-    buildFromConfig(config: Record<string, any>): SparseEmbeddingFunction;
+    buildFromConfig(config: Record<string, any>, client?: ChromaClient): SparseEmbeddingFunction;
 }
 /**
  * Registry of available embedding functions.
@@ -684,18 +684,18 @@ declare const registerEmbeddingFunction: (name: string, fn: EmbeddingFunctionCla
 declare const registerSparseEmbeddingFunction: (name: string, fn: SparseEmbeddingFunctionClass) => void;
 /**
  * Retrieves and instantiates an embedding function from configuration.
- * @param collectionName - Name of the collection (for error messages)
- * @param efConfig - Configuration for the embedding function
  * @returns EmbeddingFunction instance or undefined if it cannot be constructed
  */
-declare const getEmbeddingFunction: (collectionName: string, efConfig?: EmbeddingFunctionConfiguration) => Promise<EmbeddingFunction | undefined>;
+declare const getEmbeddingFunction: (args: {
+    collectionName: string;
+    client: ChromaClient;
+    efConfig?: EmbeddingFunctionConfiguration;
+}) => Promise<EmbeddingFunction | undefined>;
 /**
  * Retrieves and instantiates a sparse embedding function from configuration.
- * @param collectionName - Name of the collection (for error messages)
- * @param efConfig - Configuration for the sparse embedding function
  * @returns SparseEmbeddingFunction instance or undefined if it cannot be constructed
  */
-declare const getSparseEmbeddingFunction: (collectionName: string, efConfig?: EmbeddingFunctionConfiguration) => Promise<SparseEmbeddingFunction | undefined>;
+declare const getSparseEmbeddingFunction: (collectionName: string, client: ChromaClient, efConfig?: EmbeddingFunctionConfiguration) => Promise<SparseEmbeddingFunction | undefined>;
 /**
  * Serializes an embedding function to configuration format.
  * @param embeddingFunction - User provided embedding function
@@ -1001,7 +1001,7 @@ declare class Schema {
     createIndex(config?: IndexConfig, key?: string): this;
     deleteIndex(config?: IndexConfig, key?: string): this;
     serializeToJSON(): Schema$1;
-    static deserializeFromJSON(json?: Schema$1 | JsonDict | null): Promise<Schema | undefined>;
+    static deserializeFromJSON(json: Schema$1 | JsonDict | null, client: ChromaClient): Promise<Schema | undefined>;
     private setVectorIndexConfig;
     private setFtsIndexConfig;
     private setIndexInDefaults;
@@ -1075,11 +1075,12 @@ declare const processCreateCollectionConfig: ({ configuration, embeddingFunction
 /**
  *
  */
-declare const processUpdateCollectionConfig: ({ collectionName, currentConfiguration, currentEmbeddingFunction, newConfiguration, }: {
+declare const processUpdateCollectionConfig: ({ collectionName, currentConfiguration, currentEmbeddingFunction, newConfiguration, client, }: {
     collectionName: string;
     currentConfiguration: CollectionConfiguration;
     currentEmbeddingFunction?: EmbeddingFunction;
     newConfiguration: UpdateCollectionConfiguration;
+    client: ChromaClient;
 }) => Promise<{
     updateConfiguration?: UpdateCollectionConfiguration$1;
     updateEmbeddingFunction?: EmbeddingFunction;
@@ -1116,6 +1117,7 @@ declare class ChromaClient {
     private _tenant;
     private _database;
     private _preflightChecks;
+    private _headers;
     private readonly apiClient;
     /**
      * Creates a new ChromaClient instance.
@@ -1140,6 +1142,7 @@ declare class ChromaClient {
      */
     get preflightChecks(): ChecklistResponse | undefined;
     protected set preflightChecks(preflightChecks: ChecklistResponse | undefined);
+    get headers(): Record<string, string> | undefined;
     /** @ignore */
     _path(): Promise<{
         tenant: string;
@@ -1200,6 +1203,13 @@ declare class ChromaClient {
         name: string;
         embeddingFunction?: EmbeddingFunction;
     }): Promise<Collection>;
+    /**
+     * Retrieves an existing collection by its Chroma Resource Name (CRN).
+     * @param crn - The Chroma Resource Name of the collection to retrieve
+     * @returns Promise resolving to the Collection instance
+     * @throws Error if the collection does not exist
+     */
+    getCollectionByCrn(crn: string): Promise<Collection>;
     /**
      * 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)
@@ -1266,6 +1276,10 @@ declare class ChromaClient {
  * Provides methods for adding, querying, updating, and deleting records.
  */
 interface Collection {
+    /** Tenant name */
+    tenant: string;
+    /** Database name */
+    database: string;
     /** Unique identifier for the collection */
     id: string;
     /** Name of the collection */

package/dist/chromadb.legacy-esm.js

@@ -1325,7 +1325,8 @@ var registerSparseEmbeddingFunction = (name, fn) => {
   }
   knownSparseEmbeddingFunctions.set(name, fn);
 };
-var getEmbeddingFunction = async (collectionName, efConfig) => {
+var getEmbeddingFunction = async (args) => {
+  const { collectionName, client: client2, efConfig } = args;
   if (!efConfig) {
     console.warn(
       `No embedding function configuration found for collection ${collectionName}. 'add' and 'query' will fail unless you provide them embeddings directly.`
@@ -1375,7 +1376,7 @@ var getEmbeddingFunction = async (collectionName, efConfig) => {
   let constructorConfig = efConfig.type === "known" ? efConfig.config : {};
   try {
     if (embeddingFunction.buildFromConfig) {
-      return embeddingFunction.buildFromConfig(constructorConfig);
+      return embeddingFunction.buildFromConfig(constructorConfig, client2);
     }
     console.warn(
       `Embedding function ${packageName} does not define a 'buildFromConfig' function. 'add' and 'query' will fail unless you provide them embeddings directly.`
@@ -1388,7 +1389,7 @@ var getEmbeddingFunction = async (collectionName, efConfig) => {
     return void 0;
   }
 };
-var getSparseEmbeddingFunction = async (collectionName, efConfig) => {
+var getSparseEmbeddingFunction = async (collectionName, client2, efConfig) => {
   if (!efConfig) {
     return void 0;
   }
@@ -1423,7 +1424,7 @@ var getSparseEmbeddingFunction = async (collectionName, efConfig) => {
   let constructorConfig = efConfig.type === "known" ? efConfig.config : {};
   try {
     if (sparseEmbeddingFunction.buildFromConfig) {
-      return sparseEmbeddingFunction.buildFromConfig(constructorConfig);
+      return sparseEmbeddingFunction.buildFromConfig(constructorConfig, client2);
     }
     console.warn(
       `Sparse embedding function ${packageName} does not define a 'buildFromConfig' function.`
@@ -1544,7 +1545,8 @@ var processUpdateCollectionConfig = async ({
   collectionName,
   currentConfiguration,
   currentEmbeddingFunction,
-  newConfiguration
+  newConfiguration,
+  client: client2
 }) => {
   if (newConfiguration.hnsw && typeof newConfiguration.hnsw !== "object") {
     throw new ChromaValueError(
@@ -1556,10 +1558,11 @@ var processUpdateCollectionConfig = async ({
       "Invalid SPANN config provided in UpdateCollectionConfiguration"
     );
   }
-  const embeddingFunction = currentEmbeddingFunction || await getEmbeddingFunction(
+  const embeddingFunction = currentEmbeddingFunction || await getEmbeddingFunction({
     collectionName,
-    currentConfiguration.embeddingFunction ?? void 0
-  );
+    client: client2,
+    efConfig: currentConfiguration.embeddingFunction ?? void 0
+  });
   const newEmbeddingFunction = newConfiguration.embeddingFunction;
   if (embeddingFunction && embeddingFunction.validateConfigUpdate && newEmbeddingFunction && newEmbeddingFunction.getConfig) {
     embeddingFunction.validateConfigUpdate(newEmbeddingFunction.getConfig());
@@ -2812,19 +2815,23 @@ var Schema = class _Schema {
       keys
     };
   }
-  static async deserializeFromJSON(json) {
+  static async deserializeFromJSON(json, client2) {
     if (json == null) {
       return void 0;
     }
     const data = json;
     const instance = Object.create(_Schema.prototype);
     instance.defaults = await _Schema.deserializeValueTypes(
-      data.defaults ?? {}
+      data.defaults ?? {},
+      client2
     );
     instance.keys = {};
     const keys = data.keys ?? {};
     for (const [keyName, value] of Object.entries(keys)) {
-      instance.keys[keyName] = await _Schema.deserializeValueTypes(value);
+      instance.keys[keyName] = await _Schema.deserializeValueTypes(
+        value,
+        client2
+      );
     }
     return instance;
   }
@@ -3205,7 +3212,7 @@ var Schema = class _Schema {
     }
     return serialized;
   }
-  static async deserializeValueTypes(json) {
+  static async deserializeValueTypes(json, client2) {
     const result = new ValueTypes();
     if (json[STRING_VALUE_NAME]) {
       result.string = _Schema.deserializeStringValueType(
@@ -3214,12 +3221,14 @@ var Schema = class _Schema {
     }
     if (json[FLOAT_LIST_VALUE_NAME]) {
       result.floatList = await _Schema.deserializeFloatListValueType(
-        json[FLOAT_LIST_VALUE_NAME]
+        json[FLOAT_LIST_VALUE_NAME],
+        client2
       );
     }
     if (json[SPARSE_VECTOR_VALUE_NAME]) {
       result.sparseVector = await _Schema.deserializeSparseVectorValueType(
-        json[SPARSE_VECTOR_VALUE_NAME]
+        json[SPARSE_VECTOR_VALUE_NAME],
+        client2
       );
     }
     if (json[INT_VALUE_NAME]) {
@@ -3251,23 +3260,27 @@ var Schema = class _Schema {
     }
     return new StringValueType(ftsIndex, stringIndex);
   }
-  static async deserializeFloatListValueType(json) {
+  static async deserializeFloatListValueType(json, client2) {
     let vectorIndex = null;
     if (json[VECTOR_INDEX_NAME]) {
       const data = json[VECTOR_INDEX_NAME];
       const enabled = Boolean(data.enabled);
-      const config = await _Schema.deserializeVectorConfig(data.config ?? {});
+      const config = await _Schema.deserializeVectorConfig(
+        data.config ?? {},
+        client2
+      );
       vectorIndex = new VectorIndexType(enabled, config);
     }
     return new FloatListValueType(vectorIndex);
   }
-  static async deserializeSparseVectorValueType(json) {
+  static async deserializeSparseVectorValueType(json, client2) {
     let sparseIndex = null;
     if (json[SPARSE_VECTOR_INDEX_NAME]) {
       const data = json[SPARSE_VECTOR_INDEX_NAME];
       const enabled = Boolean(data.enabled);
       const config = await _Schema.deserializeSparseVectorConfig(
-        data.config ?? {}
+        data.config ?? {},
+        client2
       );
       sparseIndex = new SparseVectorIndexType(enabled, config);
     }
@@ -3306,29 +3319,31 @@ var Schema = class _Schema {
     }
     return new BoolValueType(index);
   }
-  static async deserializeVectorConfig(json) {
+  static async deserializeVectorConfig(json, client2) {
     const config = new VectorIndexConfig({
       space: json.space ?? null,
       sourceKey: json.source_key ?? null,
       hnsw: json.hnsw ? cloneObject(json.hnsw) : null,
       spann: json.spann ? cloneObject(json.spann) : null
     });
-    config.embeddingFunction = await getEmbeddingFunction(
-      "schema deserialization",
-      json.embedding_function
-    );
+    config.embeddingFunction = await getEmbeddingFunction({
+      collectionName: "schema deserialization",
+      client: client2,
+      efConfig: json.embedding_function
+    });
     if (!config.space && config.embeddingFunction?.defaultSpace) {
       config.space = config.embeddingFunction.defaultSpace();
     }
     return config;
   }
-  static async deserializeSparseVectorConfig(json) {
+  static async deserializeSparseVectorConfig(json, client2) {
     const config = new SparseVectorIndexConfig({
       sourceKey: json.source_key ?? null,
       bm25: typeof json.bm25 === "boolean" ? json.bm25 : null
     });
     const embeddingFunction = await getSparseEmbeddingFunction(
       "schema deserialization",
+      client2,
       json.embedding_function
     ) ?? config.embeddingFunction ?? void 0;
     config.embeddingFunction = embeddingFunction ?? null;
@@ -3353,6 +3368,8 @@ var CollectionImpl = class _CollectionImpl {
     chromaClient,
     apiClient,
     id,
+    tenant,
+    database,
     name,
     metadata,
     configuration,
@@ -3362,6 +3379,8 @@ var CollectionImpl = class _CollectionImpl {
     this.chromaClient = chromaClient;
     this.apiClient = apiClient;
     this.id = id;
+    this.tenant = tenant;
+    this.database = database;
     this._name = name;
     this._metadata = metadata;
     this._configuration = configuration;
@@ -3399,9 +3418,9 @@ var CollectionImpl = class _CollectionImpl {
     this._schema = schema;
   }
   async path() {
-    const clientPath = await this.chromaClient._path();
     return {
-      ...clientPath,
+      tenant: this.tenant,
+      database: this.database,
       collection_id: this.id
     };
   }
@@ -3479,7 +3498,11 @@ var CollectionImpl = class _CollectionImpl {
         if (inputs.length === 0) {
           continue;
         }
-        const sparseEmbeddings2 = await this.sparseEmbed(embeddingFunction, inputs, false);
+        const sparseEmbeddings2 = await this.sparseEmbed(
+          embeddingFunction,
+          inputs,
+          false
+        );
         if (sparseEmbeddings2.length !== positions.length) {
           throw new ChromaValueError(
             "Sparse embedding function returned unexpected number of embeddings."
@@ -3504,7 +3527,11 @@ var CollectionImpl = class _CollectionImpl {
       if (inputs.length === 0) {
         continue;
       }
-      const sparseEmbeddings = await this.sparseEmbed(embeddingFunction, inputs, false);
+      const sparseEmbeddings = await this.sparseEmbed(
+        embeddingFunction,
+        inputs,
+        false
+      );
       if (sparseEmbeddings.length !== positions.length) {
         throw new ChromaValueError(
           "Sparse embedding function returned unexpected number of embeddings."
@@ -3801,7 +3828,9 @@ var CollectionImpl = class _CollectionImpl {
   async search(searches) {
     const items = Array.isArray(searches) ? searches : [searches];
     if (items.length === 0) {
-      throw new ChromaValueError("At least one search payload must be provided.");
+      throw new ChromaValueError(
+        "At least one search payload must be provided."
+      );
     }
     const payloads = await Promise.all(
       items.map(async (search) => {
@@ -3830,7 +3859,8 @@ var CollectionImpl = class _CollectionImpl {
       collectionName: this.name,
       currentConfiguration: this.configuration,
       newConfiguration: configuration,
-      currentEmbeddingFunction: this.embeddingFunction
+      currentEmbeddingFunction: this.embeddingFunction,
+      client: this.chromaClient
     }) : {};
     if (updateEmbeddingFunction) {
       this.embeddingFunction = updateEmbeddingFunction;
@@ -3862,6 +3892,8 @@ var CollectionImpl = class _CollectionImpl {
       chromaClient: this.chromaClient,
       apiClient: this.apiClient,
       name: data.name,
+      tenant: this.tenant,
+      database: this.database,
       id: data.id,
       embeddingFunction: this._embeddingFunction,
       metadata: deserializeMetadata(data.metadata ?? void 0) ?? void 0,
@@ -3975,6 +4007,14 @@ var offlineError = (error) => {
     (error?.name === "TypeError" || error?.name === "FetchError") && (error.message?.includes("fetch failed") || error.message?.includes("Failed to fetch") || error.message?.includes("ENOTFOUND"))
   );
 };
+var getErrorMessage = async (response) => {
+  try {
+    const body = await response.clone().json();
+    return body.message || body.error || `${response.status}: ${response.statusText}`;
+  } catch {
+    return `${response.status}: ${response.statusText}`;
+  }
+};
 var chromaFetch = async (input, init) => {
   let response;
   try {
@@ -4014,15 +4054,23 @@ var chromaFetch = async (input, init) => {
     case 409:
       throw new ChromaUniqueError("The resource already exists");
     case 422:
-      const body = await response.json();
-      if (body && body.message && (body.message.startsWith("Quota exceeded") || body.message.startsWith("Billing limit exceeded"))) {
-        throw new ChromaQuotaExceededError(body?.message);
+      try {
+        const body = await response.json();
+        if (body && body.message && (body.message.startsWith("Quota exceeded") || body.message.startsWith("Billing limit exceeded"))) {
+          throw new ChromaQuotaExceededError(body?.message);
+        }
+        throw new ChromaClientError(body?.message || "Unprocessable Entity");
+      } catch (error) {
+        if (error instanceof ChromaQuotaExceededError || error instanceof ChromaClientError) {
+          throw error;
+        }
+        throw new ChromaClientError(`Unprocessable Entity: ${response.statusText}`);
       }
-      break;
     case 429:
       throw new ChromaRateLimitError("Rate limit exceeded");
   }
-  throw new ChromaServerError(`${response.status}: ${response.statusText}`);
+  const errorMessage = await getErrorMessage(response);
+  throw new ChromaServerError(errorMessage);
 };
 
 // src/admin-client.ts
@@ -4182,6 +4230,7 @@ var ChromaClient = class {
     const baseUrl = `${ssl ? "https" : "http"}://${host}:${port}`;
     this._tenant = tenant || process.env.CHROMA_TENANT;
     this._database = database || process.env.CHROMA_DATABASE;
+    this._headers = headers;
     const configOptions = {
       ...fetchOptions,
       method: normalizeMethod(fetchOptions?.method),
@@ -4221,6 +4270,9 @@ var ChromaClient = class {
   set preflightChecks(preflightChecks) {
     this._preflightChecks = preflightChecks;
   }
+  get headers() {
+    return this._headers;
+  }
   /** @ignore */
   async _path() {
     if (!this._tenant || !this._database) {
@@ -4278,16 +4330,20 @@ var ChromaClient = class {
     return Promise.all(
       data.map(async (collection) => {
         const schema = await Schema.deserializeFromJSON(
-          collection.schema ?? void 0
+          collection.schema ?? null,
+          this
         );
         const schemaEmbeddingFunction = resolveSchemaEmbeddingFunction(schema);
-        const resolvedEmbeddingFunction = await getEmbeddingFunction(
-          collection.name,
-          collection.configuration_json.embedding_function ?? void 0
-        ) ?? schemaEmbeddingFunction;
+        const resolvedEmbeddingFunction = await getEmbeddingFunction({
+          collectionName: collection.name,
+          client: this,
+          efConfig: collection.configuration_json.embedding_function ?? void 0
+        }) ?? schemaEmbeddingFunction;
         return new CollectionImpl({
           chromaClient: this,
           apiClient: this.apiClient,
+          tenant: collection.tenant,
+          database: collection.database,
           name: collection.name,
           id: collection.id,
           embeddingFunction: resolvedEmbeddingFunction,
@@ -4344,17 +4400,21 @@ var ChromaClient = class {
       }
     });
     const serverSchema = await Schema.deserializeFromJSON(
-      data.schema ?? void 0
+      data.schema ?? null,
+      this
     );
     const schemaEmbeddingFunction = resolveSchemaEmbeddingFunction(serverSchema);
-    const resolvedEmbeddingFunction = embeddingFunction ?? await getEmbeddingFunction(
-      data.name,
-      data.configuration_json.embedding_function ?? void 0
-    ) ?? schemaEmbeddingFunction;
+    const resolvedEmbeddingFunction = embeddingFunction ?? await getEmbeddingFunction({
+      collectionName: data.name,
+      client: this,
+      efConfig: data.configuration_json.embedding_function ?? void 0
+    }) ?? schemaEmbeddingFunction;
     return new CollectionImpl({
       chromaClient: this,
       apiClient: this.apiClient,
       name,
+      tenant: data.tenant,
+      database: data.database,
       configuration: data.configuration_json,
       metadata: deserializeMetadata(data.metadata ?? void 0) ?? void 0,
       embeddingFunction: resolvedEmbeddingFunction,
@@ -4378,16 +4438,50 @@ var ChromaClient = class {
       client: this.apiClient,
       path: { ...await this._path(), collection_id: name }
     });
-    const schema = await Schema.deserializeFromJSON(data.schema ?? void 0);
+    const schema = await Schema.deserializeFromJSON(data.schema ?? null, this);
     const schemaEmbeddingFunction = resolveSchemaEmbeddingFunction(schema);
-    const resolvedEmbeddingFunction = embeddingFunction ?? await getEmbeddingFunction(
-      data.name,
-      data.configuration_json.embedding_function ?? void 0
-    ) ?? schemaEmbeddingFunction;
+    const resolvedEmbeddingFunction = embeddingFunction ?? await getEmbeddingFunction({
+      collectionName: data.name,
+      client: this,
+      efConfig: data.configuration_json.embedding_function ?? void 0
+    }) ?? schemaEmbeddingFunction;
     return new CollectionImpl({
       chromaClient: this,
       apiClient: this.apiClient,
       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 an existing collection by its Chroma Resource Name (CRN).
+   * @param crn - The Chroma Resource Name of the collection to retrieve
+   * @returns Promise resolving to the Collection instance
+   * @throws Error if the collection does not exist
+   */
+  async getCollectionByCrn(crn) {
+    const { data } = await DefaultService.getCollectionByCrn({
+      client: this.apiClient,
+      path: { crn }
+    });
+    const schema = await Schema.deserializeFromJSON(data.schema ?? null, this);
+    const schemaEmbeddingFunction = resolveSchemaEmbeddingFunction(schema);
+    const resolvedEmbeddingFunction = await getEmbeddingFunction({
+      collectionName: data.name,
+      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,
@@ -4434,7 +4528,8 @@ var ChromaClient = class {
     const collectionConfig = await processCreateCollectionConfig({
       configuration,
       embeddingFunction,
-      metadata
+      metadata,
+      schema
     });
     const { data } = await DefaultService.createCollection({
       client: this.apiClient,
@@ -4448,17 +4543,21 @@ var ChromaClient = class {
       }
     });
     const serverSchema = await Schema.deserializeFromJSON(
-      data.schema ?? void 0
+      data.schema ?? null,
+      this
     );
     const schemaEmbeddingFunction = resolveSchemaEmbeddingFunction(serverSchema);
-    const resolvedEmbeddingFunction = embeddingFunction ?? await getEmbeddingFunction(
-      name,
-      data.configuration_json.embedding_function ?? void 0
-    ) ?? schemaEmbeddingFunction;
+    const resolvedEmbeddingFunction = embeddingFunction ?? await getEmbeddingFunction({
+      collectionName: name,
+      efConfig: data.configuration_json.embedding_function ?? void 0,
+      client: this
+    }) ?? schemaEmbeddingFunction;
     return new CollectionImpl({
       chromaClient: this,
       apiClient: this.apiClient,
       name,
+      tenant: data.tenant,
+      database: data.database,
       configuration: data.configuration_json,
       metadata: deserializeMetadata(data.metadata ?? void 0) ?? void 0,
       embeddingFunction: resolvedEmbeddingFunction,