chromadb 1.7.2 → 1.7.3-beta2

package/dist/chromadb.d.ts

@@ -1,6 +1,71 @@
 import * as openai from 'openai';
 import * as cohere_ai from 'cohere-ai';
-import * as _google_generative_ai from '@google/generative-ai';
+import * as _xenova_transformers from '@xenova/transformers';
+import * as chromadb_default_embed from 'chromadb-default-embed';
+
+interface IEmbeddingFunction {
+    generate(texts: string[]): Promise<number[][]>;
+}
+
+declare enum IncludeEnum {
+    Documents = "documents",
+    Embeddings = "embeddings",
+    Metadatas = "metadatas",
+    Distances = "distances"
+}
+type Number = number;
+type Embedding = Array<Number>;
+type Embeddings = Array<Embedding>;
+type Metadata = Record<string, string | number | boolean>;
+type Metadatas = Array<Metadata>;
+type Document = string;
+type Documents = Array<Document>;
+type ID = string;
+type IDs = ID[];
+type PositiveInteger = number;
+type LiteralValue = string | number | boolean;
+type ListLiteralValue = LiteralValue[];
+type LiteralNumber = number;
+type LogicalOperator = "$and" | "$or";
+type InclusionOperator = "$in" | "$nin";
+type WhereOperator = "$gt" | "$gte" | "$lt" | "$lte" | "$ne" | "$eq";
+type OperatorExpression = {
+    [key in WhereOperator | InclusionOperator | LogicalOperator]?: LiteralValue | ListLiteralValue;
+};
+type BaseWhere = {
+    [key: string]: LiteralValue | OperatorExpression;
+};
+type LogicalWhere = {
+    [key in LogicalOperator]?: Where[];
+};
+type Where = BaseWhere | LogicalWhere;
+type WhereDocumentOperator = "$contains" | LogicalOperator;
+type WhereDocument = {
+    [key in WhereDocumentOperator]?: LiteralValue | LiteralNumber | WhereDocument[];
+};
+type CollectionType = {
+    name: string;
+    id: string;
+    metadata: Metadata | null;
+};
+type GetResponse = {
+    ids: IDs;
+    embeddings: null | Embeddings;
+    documents: (null | Document)[];
+    metadatas: (null | Metadata)[];
+    error: null | string;
+};
+type QueryResponse = {
+    ids: IDs[];
+    embeddings: null | Embeddings[];
+    documents: (null | Document)[][];
+    metadatas: (null | Metadata)[][];
+    distances: null | number[][];
+};
+type AddResponse = {
+    error: string;
+};
+type CollectionMetadata = Record<string, unknown>;
 
 /**
  * FastAPI
@@ -107,7 +172,6 @@ declare namespace Api {
         embeddings?: Api.AddEmbedding.Embedding[];
         metadatas?: Api.AddEmbedding.Metadata[];
         documents?: string[];
-        uris?: string[];
         ids: string[];
     }
     /**
@@ -126,8 +190,6 @@ declare namespace Api {
     }
     interface Count200Response {
     }
-    interface CountCollections200Response {
-    }
     interface CreateCollection {
         name: string;
         metadata?: Api.CreateCollection.Metadata;
@@ -143,16 +205,6 @@ declare namespace Api {
     }
     interface CreateCollection200Response {
     }
-    interface CreateDatabase {
-        name: string;
-    }
-    interface CreateDatabase200Response {
-    }
-    interface CreateTenant {
-        name: string;
-    }
-    interface CreateTenant200Response {
-    }
     interface DeleteCollection200Response {
     }
     interface DeleteEmbedding {
@@ -172,8 +224,6 @@ declare namespace Api {
     }
     interface GetCollection200Response {
     }
-    interface GetDatabase200Response {
-    }
     interface GetEmbedding {
         ids?: string[];
         where?: Api.GetEmbedding.Where;
@@ -189,7 +239,7 @@ declare namespace Api {
          * @memberof GetEmbedding
          */
         offset?: number;
-        include?: (Api.GetEmbedding.Include.EnumValueEnum | Api.GetEmbedding.Include.EnumValueEnum2 | Api.GetEmbedding.Include.EnumValueEnum3 | Api.GetEmbedding.Include.EnumValueEnum4 | Api.GetEmbedding.Include.EnumValueEnum5 | Api.GetEmbedding.Include.EnumValueEnum6)[];
+        include?: (Api.GetEmbedding.Include.EnumValueEnum | Api.GetEmbedding.Include.EnumValueEnum2 | Api.GetEmbedding.Include.EnumValueEnum3 | Api.GetEmbedding.Include.EnumValueEnum4)[];
     }
     /**
      * @export
@@ -200,7 +250,7 @@ declare namespace Api {
         }
         interface WhereDocument {
         }
-        type Include = Api.GetEmbedding.Include.EnumValueEnum | Api.GetEmbedding.Include.EnumValueEnum2 | Api.GetEmbedding.Include.EnumValueEnum3 | Api.GetEmbedding.Include.EnumValueEnum4 | Api.GetEmbedding.Include.EnumValueEnum5 | Api.GetEmbedding.Include.EnumValueEnum6;
+        type Include = Api.GetEmbedding.Include.EnumValueEnum | Api.GetEmbedding.Include.EnumValueEnum2 | Api.GetEmbedding.Include.EnumValueEnum3 | Api.GetEmbedding.Include.EnumValueEnum4;
         /**
          * @export
          * @namespace Include
@@ -218,25 +268,15 @@ declare namespace Api {
             enum EnumValueEnum4 {
                 Distances = "distances"
             }
-            enum EnumValueEnum5 {
-                Uris = "uris"
-            }
-            enum EnumValueEnum6 {
-                Data = "data"
-            }
         }
     }
     interface GetNearestNeighbors200Response {
     }
-    interface GetTenant200Response {
-    }
     interface HTTPValidationError {
         detail?: Api.ValidationError[];
     }
     interface ListCollections200Response {
     }
-    interface PreFlightChecks200Response {
-    }
     interface QueryEmbedding {
         where?: Api.QueryEmbedding.Where;
         'where_document'?: Api.QueryEmbedding.WhereDocument;
@@ -246,7 +286,7 @@ declare namespace Api {
          * @memberof QueryEmbedding
          */
         'n_results'?: number;
-        include?: (Api.QueryEmbedding.Include.EnumValueEnum | Api.QueryEmbedding.Include.EnumValueEnum2 | Api.QueryEmbedding.Include.EnumValueEnum3 | Api.QueryEmbedding.Include.EnumValueEnum4 | Api.QueryEmbedding.Include.EnumValueEnum5 | Api.QueryEmbedding.Include.EnumValueEnum6)[];
+        include?: (Api.QueryEmbedding.Include.EnumValueEnum | Api.QueryEmbedding.Include.EnumValueEnum2 | Api.QueryEmbedding.Include.EnumValueEnum3 | Api.QueryEmbedding.Include.EnumValueEnum4)[];
     }
     /**
      * @export
@@ -259,7 +299,7 @@ declare namespace Api {
         }
         interface QueryEmbedding2 {
         }
-        type Include = Api.QueryEmbedding.Include.EnumValueEnum | Api.QueryEmbedding.Include.EnumValueEnum2 | Api.QueryEmbedding.Include.EnumValueEnum3 | Api.QueryEmbedding.Include.EnumValueEnum4 | Api.QueryEmbedding.Include.EnumValueEnum5 | Api.QueryEmbedding.Include.EnumValueEnum6;
+        type Include = Api.QueryEmbedding.Include.EnumValueEnum | Api.QueryEmbedding.Include.EnumValueEnum2 | Api.QueryEmbedding.Include.EnumValueEnum3 | Api.QueryEmbedding.Include.EnumValueEnum4;
         /**
          * @export
          * @namespace Include
@@ -277,12 +317,6 @@ declare namespace Api {
             enum EnumValueEnum4 {
                 Distances = "distances"
             }
-            enum EnumValueEnum5 {
-                Uris = "uris"
-            }
-            enum EnumValueEnum6 {
-                Data = "data"
-            }
         }
     }
     interface Update200Response {
@@ -305,7 +339,6 @@ declare namespace Api {
         embeddings?: Api.UpdateEmbedding.Embedding[];
         metadatas?: Api.UpdateEmbedding.Metadata[];
         documents?: string[];
-        uris?: string[];
         ids: string[];
     }
     /**
@@ -377,64 +410,27 @@ declare class ApiApi extends BaseAPI {
      * @throws {RequiredError}
      */
     count(collectionId: string, options?: RequestInit): Promise<Api.Count200Response>;
-    /**
-     * @summary Count Collections
-     * @param {string} [tenant]
-     * @param {string} [database]
-     * @param {RequestInit} [options] Override http request option.
-     * @throws {RequiredError}
-     */
-    countCollections(tenant: string | undefined, database: string | undefined, options?: RequestInit): Promise<Api.CountCollections200Response>;
     /**
      * @summary Create Collection
-     * @param {string} [tenant]
-     * @param {string} [database]
      * @param {Api.CreateCollection} request
      * @param {RequestInit} [options] Override http request option.
      * @throws {RequiredError}
      */
-    createCollection(tenant: string | undefined, database: string | undefined, request: Api.CreateCollection, options?: RequestInit): Promise<Api.CreateCollection200Response>;
-    /**
-     * @summary Create Database
-     * @param {string} [tenant]
-     * @param {Api.CreateDatabase} request
-     * @param {RequestInit} [options] Override http request option.
-     * @throws {RequiredError}
-     */
-    createDatabase(tenant: string | undefined, request: Api.CreateDatabase, options?: RequestInit): Promise<Api.CreateDatabase200Response>;
-    /**
-     * @summary Create Tenant
-     * @param {Api.CreateTenant} request
-     * @param {RequestInit} [options] Override http request option.
-     * @throws {RequiredError}
-     */
-    createTenant(request: Api.CreateTenant, options?: RequestInit): Promise<Api.CreateTenant200Response>;
+    createCollection(request: Api.CreateCollection, options?: RequestInit): Promise<Api.CreateCollection200Response>;
     /**
      * @summary Delete Collection
      * @param {string} collectionName
-     * @param {string} [tenant]
-     * @param {string} [database]
      * @param {RequestInit} [options] Override http request option.
      * @throws {RequiredError}
      */
-    deleteCollection(collectionName: string, tenant: string | undefined, database: string | undefined, options?: RequestInit): Promise<Api.DeleteCollection200Response>;
+    deleteCollection(collectionName: string, options?: RequestInit): Promise<Api.DeleteCollection200Response>;
     /**
      * @summary Get Collection
      * @param {string} collectionName
-     * @param {string} [tenant]
-     * @param {string} [database]
      * @param {RequestInit} [options] Override http request option.
      * @throws {RequiredError}
      */
-    getCollection(collectionName: string, tenant: string | undefined, database: string | undefined, options?: RequestInit): Promise<Api.GetCollection200Response>;
-    /**
-     * @summary Get Database
-     * @param {string} database
-     * @param {string} [tenant]
-     * @param {RequestInit} [options] Override http request option.
-     * @throws {RequiredError}
-     */
-    getDatabase(database: string, tenant: string | undefined, options?: RequestInit): Promise<Api.GetDatabase200Response>;
+    getCollection(collectionName: string, options?: RequestInit): Promise<Api.GetCollection200Response>;
     /**
      * @summary Get Nearest Neighbors
      * @param {string} collectionId
@@ -443,13 +439,6 @@ declare class ApiApi extends BaseAPI {
      * @throws {RequiredError}
      */
     getNearestNeighbors(collectionId: string, request: Api.QueryEmbedding, options?: RequestInit): Promise<Api.GetNearestNeighbors200Response>;
-    /**
-     * @summary Get Tenant
-     * @param {string} tenant
-     * @param {RequestInit} [options] Override http request option.
-     * @throws {RequiredError}
-     */
-    getTenant(tenant: string, options?: RequestInit): Promise<Api.GetTenant200Response>;
     /**
      * @summary Heartbeat
      * @param {RequestInit} [options] Override http request option.
@@ -460,20 +449,10 @@ declare class ApiApi extends BaseAPI {
     }>;
     /**
      * @summary List Collections
-     * @param {string} [tenant]
-     * @param {string} [database]
-     * @param {number} [limit]
-     * @param {number} [offset]
-     * @param {RequestInit} [options] Override http request option.
-     * @throws {RequiredError}
-     */
-    listCollections(tenant: string | undefined, database: string | undefined, limit: number | undefined, offset: number | undefined, options?: RequestInit): Promise<Api.ListCollections200Response>;
-    /**
-     * @summary Pre Flight Checks
      * @param {RequestInit} [options] Override http request option.
      * @throws {RequiredError}
      */
-    preFlightChecks(options?: RequestInit): Promise<Api.PreFlightChecks200Response>;
+    listCollections(options?: RequestInit): Promise<Api.ListCollections200Response>;
     /**
      * @summary Reset
      * @param {RequestInit} [options] Override http request option.
@@ -520,170 +499,6 @@ declare class ApiApi extends BaseAPI {
     version(options?: RequestInit): Promise<string>;
 }
 
-interface ClientAuthProvider {
-    /**
-     * Abstract method for authenticating a client.
-     */
-    authenticate(): ClientAuthResponse;
-}
-interface ClientAuthConfigurationProvider<T> {
-    /**
-     * Abstract method for getting the configuration for the client.
-     */
-    getConfig(): T;
-}
-interface ClientAuthCredentialsProvider<T> {
-    /**
-     * Abstract method for getting the credentials for the client.
-     * @param user
-     */
-    getCredentials(user?: string): T;
-}
-declare enum AuthInfoType {
-    COOKIE = "cookie",
-    HEADER = "header",
-    URL = "url",
-    METADATA = "metadata"
-}
-interface ClientAuthResponse {
-    getAuthInfoType(): AuthInfoType;
-    getAuthInfo(): {
-        key: string;
-        value: string;
-    };
-}
-type AuthOptions = {
-    provider: ClientAuthProvider | string | undefined;
-    credentialsProvider?: ClientAuthCredentialsProvider<any> | undefined;
-    configProvider?: ClientAuthConfigurationProvider<any> | undefined;
-    credentials?: any | undefined;
-    providerOptions?: any | undefined;
-};
-
-interface IEmbeddingFunction {
-    generate(texts: string[]): Promise<number[][]>;
-}
-
-declare enum IncludeEnum {
-    Documents = "documents",
-    Embeddings = "embeddings",
-    Metadatas = "metadatas",
-    Distances = "distances"
-}
-type Number = number;
-type Embedding = Array<Number>;
-type Embeddings = Array<Embedding>;
-type Metadata = Record<string, string | number | boolean>;
-type Metadatas = Array<Metadata>;
-type Document = string;
-type Documents = Array<Document>;
-type ID = string;
-type IDs = ID[];
-type PositiveInteger = number;
-type LiteralValue = string | number | boolean;
-type ListLiteralValue = LiteralValue[];
-type LiteralNumber = number;
-type LogicalOperator = "$and" | "$or";
-type InclusionOperator = "$in" | "$nin";
-type WhereOperator = "$gt" | "$gte" | "$lt" | "$lte" | "$ne" | "$eq";
-type OperatorExpression = {
-    [key in WhereOperator | InclusionOperator | LogicalOperator]?: LiteralValue | ListLiteralValue;
-};
-type BaseWhere = {
-    [key: string]: LiteralValue | OperatorExpression;
-};
-type LogicalWhere = {
-    [key in LogicalOperator]?: Where[];
-};
-type Where = BaseWhere | LogicalWhere;
-type WhereDocumentOperator = "$contains" | LogicalOperator;
-type WhereDocument = {
-    [key in WhereDocumentOperator]?: LiteralValue | LiteralNumber | WhereDocument[];
-};
-type CollectionType = {
-    name: string;
-    id: string;
-    metadata: Metadata | null;
-};
-type GetResponse = {
-    ids: IDs;
-    embeddings: null | Embeddings;
-    documents: (null | Document)[];
-    metadatas: (null | Metadata)[];
-    error: null | string;
-};
-type QueryResponse = {
-    ids: IDs[];
-    embeddings: null | Embeddings[];
-    documents: (null | Document)[][];
-    metadatas: (null | Metadata)[][];
-    distances: null | number[][];
-};
-type AddResponse = {
-    error: string;
-};
-type CollectionMetadata = Record<string, unknown>;
-type GetParams = {
-    ids?: ID | IDs;
-    where?: Where;
-    limit?: PositiveInteger;
-    offset?: PositiveInteger;
-    include?: IncludeEnum[];
-    whereDocument?: WhereDocument;
-};
-type ListCollectionsParams = {
-    limit?: PositiveInteger;
-    offset?: PositiveInteger;
-};
-type ChromaClientParams = {
-    path?: string;
-    fetchOptions?: RequestInit;
-    auth?: AuthOptions;
-    tenant?: string;
-    database?: string;
-};
-type CreateCollectionParams = {
-    name: string;
-    metadata?: CollectionMetadata;
-    embeddingFunction?: IEmbeddingFunction;
-};
-type GetOrCreateCollectionParams = CreateCollectionParams;
-type GetCollectionParams = {
-    name: string;
-    embeddingFunction?: IEmbeddingFunction;
-};
-type DeleteCollectionParams = {
-    name: string;
-};
-type AddParams = {
-    ids: ID | IDs;
-    embeddings?: Embedding | Embeddings;
-    metadatas?: Metadata | Metadatas;
-    documents?: Document | Documents;
-};
-type UpsertParams = AddParams;
-type UpdateParams = AddParams;
-type ModifyCollectionParams = {
-    name?: string;
-    metadata?: CollectionMetadata;
-};
-type QueryParams = {
-    queryEmbeddings?: Embedding | Embeddings;
-    nResults?: PositiveInteger;
-    where?: Where;
-    queryTexts?: string | string[];
-    whereDocument?: WhereDocument;
-    include?: IncludeEnum[];
-};
-type PeekParams = {
-    limit?: PositiveInteger;
-};
-type DeleteParams = {
-    ids?: ID | IDs;
-    where?: Where;
-    whereDocument?: WhereDocument;
-};
-
 declare class Collection {
     name: string;
     id: string;
@@ -731,7 +546,12 @@ declare class Collection {
      * });
      * ```
      */
-    add({ ids, embeddings, metadatas, documents, }: AddParams): Promise<AddResponse>;
+    add({ ids, embeddings, metadatas, documents, }: {
+        ids: ID | IDs;
+        embeddings?: Embedding | Embeddings;
+        metadatas?: Metadata | Metadatas;
+        documents?: Document | Documents;
+    }): Promise<AddResponse>;
     /**
      * Upsert items to the collection
      * @param {Object} params - The parameters for the query.
@@ -751,7 +571,12 @@ declare class Collection {
      * });
      * ```
      */
-    upsert({ ids, embeddings, metadatas, documents, }: UpsertParams): Promise<boolean>;
+    upsert({ ids, embeddings, metadatas, documents, }: {
+        ids: ID | IDs;
+        embeddings?: Embedding | Embeddings;
+        metadatas?: Metadata | Metadatas;
+        documents?: Document | Documents;
+    }): Promise<boolean>;
     /**
      * Count the number of items in the collection
      * @returns {Promise<number>} - The response from the API.
@@ -777,7 +602,10 @@ declare class Collection {
      * });
      * ```
      */
-    modify({ name, metadata }?: ModifyCollectionParams): Promise<void>;
+    modify({ name, metadata }?: {
+        name?: string;
+        metadata?: CollectionMetadata;
+    }): Promise<void>;
     /**
      * Get items from the collection
      * @param {Object} params - The parameters for the query.
@@ -801,7 +629,14 @@ declare class Collection {
      * });
      * ```
      */
-    get({ ids, where, limit, offset, include, whereDocument, }?: GetParams): Promise<GetResponse>;
+    get({ ids, where, limit, offset, include, whereDocument, }?: {
+        ids?: ID | IDs;
+        where?: Where;
+        limit?: PositiveInteger;
+        offset?: PositiveInteger;
+        include?: IncludeEnum[];
+        whereDocument?: WhereDocument;
+    }): Promise<GetResponse>;
     /**
      * Update the embeddings, documents, and/or metadatas of existing items
      * @param {Object} params - The parameters for the query.
@@ -821,7 +656,12 @@ declare class Collection {
      * });
      * ```
      */
-    update({ ids, embeddings, metadatas, documents, }: UpdateParams): Promise<boolean>;
+    update({ ids, embeddings, metadatas, documents, }: {
+        ids: ID | IDs;
+        embeddings?: Embedding | Embeddings;
+        metadatas?: Metadata | Metadatas;
+        documents?: Document | Documents;
+    }): Promise<boolean>;
     /**
      * Performs a query on the collection using the specified parameters.
      *
@@ -855,7 +695,14 @@ declare class Collection {
      * ```
      *
      */
-    query({ queryEmbeddings, nResults, where, queryTexts, whereDocument, include, }: QueryParams): Promise<QueryResponse>;
+    query({ queryEmbeddings, nResults, where, queryTexts, whereDocument, include, }: {
+        queryEmbeddings?: Embedding | Embeddings;
+        nResults?: PositiveInteger;
+        where?: Where;
+        queryTexts?: string | string[];
+        whereDocument?: WhereDocument;
+        include?: IncludeEnum[];
+    }): Promise<QueryResponse>;
     /**
      * Peek inside the collection
      * @param {Object} params - The parameters for the query.
@@ -870,7 +717,9 @@ declare class Collection {
      * });
      * ```
      */
-    peek({ limit }?: PeekParams): Promise<GetResponse>;
+    peek({ limit }?: {
+        limit?: PositiveInteger;
+    }): Promise<GetResponse>;
     /**
      * Deletes items from the collection.
      * @param {Object} params - The parameters for deleting items from the collection.
@@ -889,18 +738,59 @@ declare class Collection {
      * });
      * ```
      */
-    delete({ ids, where, whereDocument }?: DeleteParams): Promise<string[]>;
+    delete({ ids, where, whereDocument }?: {
+        ids?: ID | IDs;
+        where?: Where;
+        whereDocument?: WhereDocument;
+    }): Promise<string[]>;
 }
 
+interface ClientAuthProvider {
+    /**
+     * Abstract method for authenticating a client.
+     */
+    authenticate(): ClientAuthResponse;
+}
+interface ClientAuthConfigurationProvider<T> {
+    /**
+     * Abstract method for getting the configuration for the client.
+     */
+    getConfig(): T;
+}
+interface ClientAuthCredentialsProvider<T> {
+    /**
+     * Abstract method for getting the credentials for the client.
+     * @param user
+     */
+    getCredentials(user?: string): T;
+}
+declare enum AuthInfoType {
+    COOKIE = "cookie",
+    HEADER = "header",
+    URL = "url",
+    METADATA = "metadata"
+}
+interface ClientAuthResponse {
+    getAuthInfoType(): AuthInfoType;
+    getAuthInfo(): {
+        key: string;
+        value: string;
+    };
+}
+type AuthOptions = {
+    provider: ClientAuthProvider | string | undefined;
+    credentialsProvider?: ClientAuthCredentialsProvider<any> | undefined;
+    configProvider?: ClientAuthConfigurationProvider<any> | undefined;
+    credentials?: any | undefined;
+    providerOptions?: any | undefined;
+};
+
 declare class ChromaClient {
     /**
      * @ignore
      */
     private api;
     private apiAdapter;
-    private tenant;
-    private database;
-    private _adminClient?;
     /**
      * Creates a new ChromaClient instance.
      * @param {Object} params - The parameters for creating a new client
@@ -914,7 +804,11 @@ declare class ChromaClient {
      * });
      * ```
      */
-    constructor({ path, fetchOptions, auth, tenant, database, }?: ChromaClientParams);
+    constructor({ path, fetchOptions, auth, }?: {
+        path?: string;
+        fetchOptions?: RequestInit;
+        auth?: AuthOptions;
+    });
     /**
      * Resets the state of the object by making an API call to the reset endpoint.
      *
@@ -968,7 +862,11 @@ declare class ChromaClient {
      * });
      * ```
      */
-    createCollection({ name, metadata, embeddingFunction }: CreateCollectionParams): Promise<Collection>;
+    createCollection({ name, metadata, embeddingFunction }: {
+        name: string;
+        metadata?: CollectionMetadata;
+        embeddingFunction?: IEmbeddingFunction;
+    }): Promise<Collection>;
     /**
      * Gets or creates a collection with the specified properties.
      *
@@ -990,36 +888,23 @@ declare class ChromaClient {
      * });
      * ```
      */
-    getOrCreateCollection({ name, metadata, embeddingFunction }: GetOrCreateCollectionParams): Promise<Collection>;
+    getOrCreateCollection({ name, metadata, embeddingFunction }: {
+        name: string;
+        metadata?: CollectionMetadata;
+        embeddingFunction?: IEmbeddingFunction;
+    }): Promise<Collection>;
     /**
      * Lists all collections.
      *
      * @returns {Promise<CollectionType[]>} A promise that resolves to a list of collection names.
-     * @param {PositiveInteger} [params.limit] - Optional limit on the number of items to get.
-     * @param {PositiveInteger} [params.offset] - Optional offset on the items to get.
      * @throws {Error} If there is an issue listing the collections.
      *
      * @example
      * ```typescript
-     * const collections = await client.listCollections({
-     *     limit: 10,
-     *     offset: 0,
-     * });
+     * const collections = await client.listCollections();
      * ```
      */
-    listCollections({ limit, offset, }?: ListCollectionsParams): Promise<CollectionType[]>;
-    /**
-     * Counts all collections.
-     *
-     * @returns {Promise<number>} A promise that resolves to the number of collections.
-     * @throws {Error} If there is an issue counting the collections.
-     *
-     * @example
-     * ```typescript
-     * const collections = await client.countCollections();
-     * ```
-     */
-    countCollections(): Promise<number>;
+    listCollections(): Promise<CollectionType[]>;
     /**
      * Gets a collection with the specified name.
      * @param {Object} params - The parameters for getting a collection.
@@ -1035,7 +920,10 @@ declare class ChromaClient {
      * });
      * ```
      */
-    getCollection({ name, embeddingFunction }: GetCollectionParams): Promise<Collection>;
+    getCollection({ name, embeddingFunction }: {
+        name: string;
+        embeddingFunction?: IEmbeddingFunction;
+    }): Promise<Collection>;
     /**
      * Deletes a collection with the specified name.
      * @param {Object} params - The parameters for deleting a collection.
@@ -1050,176 +938,9 @@ declare class ChromaClient {
      * });
      * ```
      */
-    deleteCollection({ name }: DeleteCollectionParams): Promise<void>;
-}
-
-interface Tenant {
-    name: string;
-}
-interface Database {
-    name: string;
-}
-declare class AdminClient {
-    /**
-     * @ignore
-     */
-    private api;
-    private apiAdapter;
-    tenant: string;
-    database: string;
-    /**
-     * Creates a new AdminClient instance.
-     * @param {Object} params - The parameters for creating a new client
-     * @param {string} [params.path] - The base path for the Chroma API.
-     * @returns {AdminClient} A new AdminClient instance.
-     *
-     * @example
-     * ```typescript
-     * const client = new AdminClient({
-     *   path: "http://localhost:8000"
-     * });
-     * ```
-     */
-    constructor({ path, fetchOptions, auth, tenant, database }?: {
-        path?: string;
-        fetchOptions?: RequestInit;
-        auth?: AuthOptions;
-        tenant?: string;
-        database?: string;
-    });
-    /**
-     * Sets the tenant and database for the client.
-     *
-     * @param {Object} params - The parameters for setting tenant and database.
-     * @param {string} params.tenant - The name of the tenant.
-     * @param {string} params.database - The name of the database.
-     *
-     * @returns {Promise<void>} A promise that returns nothing
-     * @throws {Error} Any issues
-     *
-     * @example
-     * ```typescript
-     * await adminClient.setTenant({
-     *   tenant: "my_tenant",
-     *   database: "my_database",
-     * });
-     * ```
-     */
-    setTenant({ tenant, database }: {
-        tenant: string;
-        database?: string;
-    }): Promise<void>;
-    /**
-     * Sets the database for the client.
-     *
-     * @param {Object} params - The parameters for setting the database.
-     * @param {string} params.database - The name of the database.
-     *
-     * @returns {Promise<void>} A promise that returns nothing
-     * @throws {Error} Any issues
-     *
-     * @example
-     * ```typescript
-     * await adminClient.setDatabase({
-     *   database: "my_database",
-     * });
-     * ```
-     */
-    setDatabase({ database }: {
-        database?: string;
-    }): Promise<void>;
-    /**
-     * Creates a new tenant with the specified properties.
-     *
-     * @param {Object} params - The parameters for creating a new tenant.
-     * @param {string} params.name - The name of the tenant.
-     *
-     * @returns {Promise<Tenant>} A promise that resolves to the created tenant.
-     * @throws {Error} If there is an issue creating the tenant.
-     *
-     * @example
-     * ```typescript
-     * await adminClient.createTenant({
-     *   name: "my_tenant",
-     * });
-     * ```
-     */
-    createTenant({ name, }: {
+    deleteCollection({ name }: {
         name: string;
-    }): Promise<Tenant>;
-    /**
-     * Gets a tenant with the specified properties.
-     *
-     * @param {Object} params - The parameters for getting a tenant.
-     * @param {string} params.name - The name of the tenant.
-     *
-     * @returns {Promise<Tenant>} A promise that resolves to the tenant.
-     * @throws {Error} If there is an issue getting the tenant.
-     *
-     * @example
-     * ```typescript
-     * await adminClient.getTenant({
-     *   name: "my_tenant",
-     * });
-     * ```
-     */
-    getTenant({ name, }: {
-        name: string;
-    }): Promise<Tenant>;
-    /**
-     * Creates a new database with the specified properties.
-     *
-     * @param {Object} params - The parameters for creating a new database.
-     * @param {string} params.name - The name of the database.
-     * @param {string} params.tenantName - The name of the tenant.
-     *
-     * @returns {Promise<Database>} A promise that resolves to the created database.
-     * @throws {Error} If there is an issue creating the database.
-     *
-     * @example
-     * ```typescript
-     * await adminClient.createDatabase({
-     *   name: "my_database",
-     *   tenantName: "my_tenant",
-     * });
-     * ```
-     */
-    createDatabase({ name, tenantName }: {
-        name: string;
-        tenantName: string;
-    }): Promise<Database>;
-    /**
-     * Gets a database with the specified properties.
-     *
-     * @param {Object} params - The parameters for getting a database.
-     * @param {string} params.name - The name of the database.
-     * @param {string} params.tenantName - The name of the tenant.
-     *
-     * @returns {Promise<Database>} A promise that resolves to the database.
-     * @throws {Error} If there is an issue getting the database.
-     *
-     * @example
-     * ```typescript
-     * await adminClient.getDatabase({
-     *   name: "my_database",
-     *   tenantName: "my_tenant",
-     * });
-     * ```
-     */
-    getDatabase({ name, tenantName }: {
-        name: string;
-        tenantName: string;
-    }): Promise<Database>;
-}
-
-interface CloudClientParams {
-    apiKey?: string;
-    database?: string;
-    cloudHost?: string;
-    cloudPort?: string;
-}
-declare class CloudClient extends ChromaClient {
-    constructor({ apiKey, database, cloudHost, cloudPort }: CloudClientParams);
+    }): Promise<void>;
 }
 
 declare class OpenAIEmbeddingFunction implements IEmbeddingFunction {
@@ -1257,41 +978,62 @@ declare class CohereEmbeddingFunction implements IEmbeddingFunction {
     }>;
 }
 
-declare class GoogleGenerativeAiEmbeddingFunction implements IEmbeddingFunction {
-    private api_key;
+declare class TransformersEmbeddingFunction implements IEmbeddingFunction {
+    private pipelinePromise?;
+    private transformersApi;
     private model;
-    private googleGenAiApi?;
-    private taskType;
-    constructor({ googleApiKey, model, taskType }: {
-        googleApiKey: string;
+    private revision;
+    private quantized;
+    private progress_callback;
+    /**
+     * TransformersEmbeddingFunction constructor.
+     * @param options The configuration options.
+     * @param options.model The model to use to calculate embeddings. Defaults to 'Xenova/all-MiniLM-L6-v2', which is an ONNX port of `sentence-transformers/all-MiniLM-L6-v2`.
+     * @param options.revision The specific model version to use (can be a branch, tag name, or commit id). Defaults to 'main'.
+     * @param options.quantized Whether to load the 8-bit quantized version of the model. Defaults to `false`.
+     * @param options.progress_callback If specified, this function will be called during model construction, to provide the user with progress updates.
+     */
+    constructor({ model, revision, quantized, progress_callback, }?: {
         model?: string;
-        taskType?: string;
+        revision?: string;
+        quantized?: boolean;
+        progress_callback?: Function | null;
     });
+    generate(texts: string[]): Promise<number[][]>;
     private loadClient;
-    generate(texts: string[]): Promise<any>;
     /** @ignore */
     static import(): Promise<{
-        googleGenAi: typeof _google_generative_ai;
+        pipeline: typeof _xenova_transformers;
     }>;
 }
 
-declare class HuggingFaceEmbeddingServerFunction implements IEmbeddingFunction {
-    private url;
-    constructor({ url }: {
-        url: string;
-    });
-    generate(texts: string[]): Promise<any>;
-}
-
-declare class JinaEmbeddingFunction implements IEmbeddingFunction {
-    private model_name;
-    private api_url;
-    private headers;
-    constructor({ jinaai_api_key, model_name }: {
-        jinaai_api_key: string;
-        model_name?: string;
+declare class DefaultEmbeddingFunction implements IEmbeddingFunction {
+    private pipelinePromise?;
+    private transformersApi;
+    private model;
+    private revision;
+    private quantized;
+    private progress_callback;
+    /**
+     * DefaultEmbeddingFunction constructor.
+     * @param options The configuration options.
+     * @param options.model The model to use to calculate embeddings. Defaults to 'Xenova/all-MiniLM-L6-v2', which is an ONNX port of `sentence-transformers/all-MiniLM-L6-v2`.
+     * @param options.revision The specific model version to use (can be a branch, tag name, or commit id). Defaults to 'main'.
+     * @param options.quantized Whether to load the 8-bit quantized version of the model. Defaults to `false`.
+     * @param options.progress_callback If specified, this function will be called during model construction, to provide the user with progress updates.
+     */
+    constructor({ model, revision, quantized, progress_callback, }?: {
+        model?: string;
+        revision?: string;
+        quantized?: boolean;
+        progress_callback?: Function | null;
     });
-    generate(texts: string[]): Promise<any[]>;
+    generate(texts: string[]): Promise<number[][]>;
+    private loadClient;
+    /** @ignore */
+    static import(): Promise<{
+        pipeline: typeof chromadb_default_embed;
+    }>;
 }
 
-export { AddParams, AdminClient, ChromaClient, ChromaClientParams, CloudClient, CohereEmbeddingFunction, Collection, CollectionMetadata, CollectionType, CreateCollectionParams, DeleteCollectionParams, DeleteParams, Document, Documents, Embedding, Embeddings, GetCollectionParams, GetOrCreateCollectionParams, GetParams, GetResponse, GoogleGenerativeAiEmbeddingFunction, HuggingFaceEmbeddingServerFunction, ID, IDs, IEmbeddingFunction, IncludeEnum, JinaEmbeddingFunction, ListCollectionsParams, Metadata, Metadatas, ModifyCollectionParams, OpenAIEmbeddingFunction, PeekParams, QueryParams, QueryResponse, UpdateParams, UpsertParams, Where, WhereDocument };
+export { ChromaClient, CohereEmbeddingFunction, Collection, DefaultEmbeddingFunction, type IEmbeddingFunction, IncludeEnum, OpenAIEmbeddingFunction, TransformersEmbeddingFunction };