chromadb 1.6.1 → 1.7.1-beta2

package/dist/chromadb.d.ts

@@ -1,70 +1,6 @@
 import * as openai from 'openai';
 import * as cohere_ai from 'cohere-ai';
 
-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
  *
@@ -170,6 +106,7 @@ declare namespace Api {
         embeddings?: Api.AddEmbedding.Embedding[];
         metadatas?: Api.AddEmbedding.Metadata[];
         documents?: string[];
+        uris?: string[];
         ids: string[];
     }
     /**
@@ -188,6 +125,8 @@ declare namespace Api {
     }
     interface Count200Response {
     }
+    interface CountCollections200Response {
+    }
     interface CreateCollection {
         name: string;
         metadata?: Api.CreateCollection.Metadata;
@@ -203,6 +142,16 @@ declare namespace Api {
     }
     interface CreateCollection200Response {
     }
+    interface CreateDatabase {
+        name: string;
+    }
+    interface CreateDatabase200Response {
+    }
+    interface CreateTenant {
+        name: string;
+    }
+    interface CreateTenant200Response {
+    }
     interface DeleteCollection200Response {
     }
     interface DeleteEmbedding {
@@ -222,6 +171,8 @@ declare namespace Api {
     }
     interface GetCollection200Response {
     }
+    interface GetDatabase200Response {
+    }
     interface GetEmbedding {
         ids?: string[];
         where?: Api.GetEmbedding.Where;
@@ -237,7 +188,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)[];
+        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)[];
     }
     /**
      * @export
@@ -248,7 +199,7 @@ declare namespace Api {
         }
         interface WhereDocument {
         }
-        type Include = Api.GetEmbedding.Include.EnumValueEnum | Api.GetEmbedding.Include.EnumValueEnum2 | Api.GetEmbedding.Include.EnumValueEnum3 | Api.GetEmbedding.Include.EnumValueEnum4;
+        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;
         /**
          * @export
          * @namespace Include
@@ -266,15 +217,25 @@ 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;
@@ -284,7 +245,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)[];
+        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)[];
     }
     /**
      * @export
@@ -297,7 +258,7 @@ declare namespace Api {
         }
         interface QueryEmbedding2 {
         }
-        type Include = Api.QueryEmbedding.Include.EnumValueEnum | Api.QueryEmbedding.Include.EnumValueEnum2 | Api.QueryEmbedding.Include.EnumValueEnum3 | Api.QueryEmbedding.Include.EnumValueEnum4;
+        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;
         /**
          * @export
          * @namespace Include
@@ -315,6 +276,12 @@ declare namespace Api {
             enum EnumValueEnum4 {
                 Distances = "distances"
             }
+            enum EnumValueEnum5 {
+                Uris = "uris"
+            }
+            enum EnumValueEnum6 {
+                Data = "data"
+            }
         }
     }
     interface Update200Response {
@@ -337,6 +304,7 @@ declare namespace Api {
         embeddings?: Api.UpdateEmbedding.Embedding[];
         metadatas?: Api.UpdateEmbedding.Metadata[];
         documents?: string[];
+        uris?: string[];
         ids: string[];
     }
     /**
@@ -408,27 +376,64 @@ 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(request: Api.CreateCollection, options?: RequestInit): Promise<Api.CreateCollection200Response>;
+    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>;
     /**
      * @summary Delete Collection
      * @param {string} collectionName
+     * @param {string} [tenant]
+     * @param {string} [database]
      * @param {RequestInit} [options] Override http request option.
      * @throws {RequiredError}
      */
-    deleteCollection(collectionName: string, options?: RequestInit): Promise<Api.DeleteCollection200Response>;
+    deleteCollection(collectionName: string, tenant: string | undefined, database: string | undefined, 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}
      */
-    getCollection(collectionName: string, options?: RequestInit): Promise<Api.GetCollection200Response>;
+    getDatabase(database: string, tenant: string | undefined, options?: RequestInit): Promise<Api.GetDatabase200Response>;
     /**
      * @summary Get Nearest Neighbors
      * @param {string} collectionId
@@ -437,6 +442,13 @@ 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.
@@ -447,10 +459,20 @@ 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(options?: RequestInit): Promise<Api.ListCollections200Response>;
+    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>;
     /**
      * @summary Reset
      * @param {RequestInit} [options] Override http request option.
@@ -497,6 +519,170 @@ 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;
@@ -544,12 +730,7 @@ declare class Collection {
      * });
      * ```
      */
-    add({ ids, embeddings, metadatas, documents, }: {
-        ids: ID | IDs;
-        embeddings?: Embedding | Embeddings;
-        metadatas?: Metadata | Metadatas;
-        documents?: Document | Documents;
-    }): Promise<AddResponse>;
+    add({ ids, embeddings, metadatas, documents, }: AddParams): Promise<AddResponse>;
     /**
      * Upsert items to the collection
      * @param {Object} params - The parameters for the query.
@@ -569,12 +750,7 @@ declare class Collection {
      * });
      * ```
      */
-    upsert({ ids, embeddings, metadatas, documents, }: {
-        ids: ID | IDs;
-        embeddings?: Embedding | Embeddings;
-        metadatas?: Metadata | Metadatas;
-        documents?: Document | Documents;
-    }): Promise<boolean>;
+    upsert({ ids, embeddings, metadatas, documents, }: UpsertParams): Promise<boolean>;
     /**
      * Count the number of items in the collection
      * @returns {Promise<number>} - The response from the API.
@@ -600,10 +776,7 @@ declare class Collection {
      * });
      * ```
      */
-    modify({ name, metadata }?: {
-        name?: string;
-        metadata?: CollectionMetadata;
-    }): Promise<void>;
+    modify({ name, metadata }?: ModifyCollectionParams): Promise<void>;
     /**
      * Get items from the collection
      * @param {Object} params - The parameters for the query.
@@ -627,14 +800,7 @@ declare class Collection {
      * });
      * ```
      */
-    get({ ids, where, limit, offset, include, whereDocument, }?: {
-        ids?: ID | IDs;
-        where?: Where;
-        limit?: PositiveInteger;
-        offset?: PositiveInteger;
-        include?: IncludeEnum[];
-        whereDocument?: WhereDocument;
-    }): Promise<GetResponse>;
+    get({ ids, where, limit, offset, include, whereDocument, }?: GetParams): Promise<GetResponse>;
     /**
      * Update the embeddings, documents, and/or metadatas of existing items
      * @param {Object} params - The parameters for the query.
@@ -654,12 +820,7 @@ declare class Collection {
      * });
      * ```
      */
-    update({ ids, embeddings, metadatas, documents, }: {
-        ids: ID | IDs;
-        embeddings?: Embedding | Embeddings;
-        metadatas?: Metadata | Metadatas;
-        documents?: Document | Documents;
-    }): Promise<boolean>;
+    update({ ids, embeddings, metadatas, documents, }: UpdateParams): Promise<boolean>;
     /**
      * Performs a query on the collection using the specified parameters.
      *
@@ -693,14 +854,7 @@ declare class Collection {
      * ```
      *
      */
-    query({ queryEmbeddings, nResults, where, queryTexts, whereDocument, include, }: {
-        queryEmbeddings?: Embedding | Embeddings;
-        nResults?: PositiveInteger;
-        where?: Where;
-        queryTexts?: string | string[];
-        whereDocument?: WhereDocument;
-        include?: IncludeEnum[];
-    }): Promise<QueryResponse>;
+    query({ queryEmbeddings, nResults, where, queryTexts, whereDocument, include, }: QueryParams): Promise<QueryResponse>;
     /**
      * Peek inside the collection
      * @param {Object} params - The parameters for the query.
@@ -715,9 +869,7 @@ declare class Collection {
      * });
      * ```
      */
-    peek({ limit }?: {
-        limit?: PositiveInteger;
-    }): Promise<GetResponse>;
+    peek({ limit }?: PeekParams): Promise<GetResponse>;
     /**
      * Deletes items from the collection.
      * @param {Object} params - The parameters for deleting items from the collection.
@@ -736,52 +888,8 @@ declare class Collection {
      * });
      * ```
      */
-    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;
-    };
+    delete({ ids, where, whereDocument }?: DeleteParams): Promise<string[]>;
 }
-type AuthOptions = {
-    provider: ClientAuthProvider | string | undefined;
-    credentialsProvider?: ClientAuthCredentialsProvider<any> | undefined;
-    configProvider?: ClientAuthConfigurationProvider<any> | undefined;
-    credentials?: any | undefined;
-    providerOptions?: any | undefined;
-};
 
 declare class ChromaClient {
     /**
@@ -789,6 +897,9 @@ declare class ChromaClient {
      */
     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
@@ -802,11 +913,7 @@ declare class ChromaClient {
      * });
      * ```
      */
-    constructor({ path, fetchOptions, auth, }?: {
-        path?: string;
-        fetchOptions?: RequestInit;
-        auth?: AuthOptions;
-    });
+    constructor({ path, fetchOptions, auth, tenant, database, }?: ChromaClientParams);
     /**
      * Resets the state of the object by making an API call to the reset endpoint.
      *
@@ -860,11 +967,7 @@ declare class ChromaClient {
      * });
      * ```
      */
-    createCollection({ name, metadata, embeddingFunction }: {
-        name: string;
-        metadata?: CollectionMetadata;
-        embeddingFunction?: IEmbeddingFunction;
-    }): Promise<Collection>;
+    createCollection({ name, metadata, embeddingFunction }: CreateCollectionParams): Promise<Collection>;
     /**
      * Gets or creates a collection with the specified properties.
      *
@@ -886,23 +989,36 @@ declare class ChromaClient {
      * });
      * ```
      */
-    getOrCreateCollection({ name, metadata, embeddingFunction }: {
-        name: string;
-        metadata?: CollectionMetadata;
-        embeddingFunction?: IEmbeddingFunction;
-    }): Promise<Collection>;
+    getOrCreateCollection({ name, metadata, embeddingFunction }: GetOrCreateCollectionParams): 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();
+     * const collections = await client.listCollections({
+     *     limit: 10,
+     *     offset: 0,
+     * });
      * ```
      */
-    listCollections(): Promise<CollectionType[]>;
+    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>;
     /**
      * Gets a collection with the specified name.
      * @param {Object} params - The parameters for getting a collection.
@@ -918,10 +1034,7 @@ declare class ChromaClient {
      * });
      * ```
      */
-    getCollection({ name, embeddingFunction }: {
-        name: string;
-        embeddingFunction?: IEmbeddingFunction;
-    }): Promise<Collection>;
+    getCollection({ name, embeddingFunction }: GetCollectionParams): Promise<Collection>;
     /**
      * Deletes a collection with the specified name.
      * @param {Object} params - The parameters for deleting a collection.
@@ -936,9 +1049,166 @@ declare class ChromaClient {
      * });
      * ```
      */
-    deleteCollection({ name }: {
-        name: string;
+    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, }: {
+        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>;
 }
 
 declare class OpenAIEmbeddingFunction implements IEmbeddingFunction {
@@ -976,4 +1246,12 @@ declare class CohereEmbeddingFunction implements IEmbeddingFunction {
     }>;
 }
 
-export { ChromaClient, CohereEmbeddingFunction, Collection, IEmbeddingFunction, IncludeEnum, OpenAIEmbeddingFunction };
+declare class HuggingFaceEmbeddingServerFunction implements IEmbeddingFunction {
+    private url;
+    constructor({ url }: {
+        url: string;
+    });
+    generate(texts: string[]): Promise<any>;
+}
+
+export { type AddParams, AdminClient, ChromaClient, type ChromaClientParams, CohereEmbeddingFunction, Collection, type CollectionMetadata, type CollectionType, type CreateCollectionParams, type DeleteCollectionParams, type DeleteParams, type Document, type Documents, type Embedding, type Embeddings, type GetCollectionParams, type GetOrCreateCollectionParams, type GetParams, type GetResponse, HuggingFaceEmbeddingServerFunction, type ID, type IDs, type IEmbeddingFunction, IncludeEnum, type ListCollectionsParams, type Metadata, type Metadatas, type ModifyCollectionParams, OpenAIEmbeddingFunction, type PeekParams, type QueryParams, type QueryResponse, type UpdateParams, type UpsertParams, type Where, type WhereDocument };