[ENH] Add getCollectionByCrn api support in js client (#5894)

## Description of changes

_Summarize the changes made by this PR._

- Improvements & Bug fixes
  - This PR adds support for getCollectionByCrn to the js client
- It also had to add support for tenant and db fields on the collection
object to allow for building the correct path for both private and
public collections. this way, instead of using the path w/ tenant and db
from the client, it directly uses data from the server about the
collection's origin
- New functionality
  - ...

## Test plan

_How are these changes tested?_

- [ ] Tests pass locally with `pytest` for python, `yarn test` for js,
`cargo test` for rust

## Migration plan

_Are there any migrations, or any forwards/backwards compatibility
changes needed in order to make sure this change deploys reliably?_

## Observability plan

_What is the plan to instrument and monitor this change?_

## Documentation Changes

_Are all docstrings for user-facing APIs updated if required? Do we need
to make documentation changes in the [docs
section](https://github.com/chroma-core/chroma/tree/main/docs/docs.trychroma.com)?_

Author: jairad26

clients/new-js/packages/chromadb/src/chroma-client.ts

@@ -263,6 +263,8 @@ export class ChromaClient {
         return new CollectionImpl({
           chromaClient: this,
           apiClient: this.apiClient,
+          tenant: collection.tenant,
+          database: collection.database,
           name: collection.name,
           id: collection.id,
           embeddingFunction: resolvedEmbeddingFunction,
@@ -349,6 +351,8 @@ export class ChromaClient {
       chromaClient: this,
       apiClient: this.apiClient,
       name,
+      tenant: data.tenant,
+      database: data.database,
       configuration: data.configuration_json,
       metadata: deserializeMetadata(data.metadata ?? undefined) ?? undefined,
       embeddingFunction: resolvedEmbeddingFunction,
@@ -392,6 +396,40 @@ export class ChromaClient {
       chromaClient: this,
       apiClient: this.apiClient,
       name,
+      tenant: data.tenant,
+      database: data.database,
+      configuration: data.configuration_json,
+      metadata: deserializeMetadata(data.metadata ?? undefined) ?? undefined,
+      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
+   */
+  public async getCollectionByCrn(crn: string): Promise<Collection> {
+    const { data } = await Api.getCollectionByCrn({
+      client: this.apiClient,
+      path: { crn },
+    });
+    const schema = await Schema.deserializeFromJSON(data.schema ?? undefined);
+    const schemaEmbeddingFunction = resolveSchemaEmbeddingFunction(schema);
+    const resolvedEmbeddingFunction =
+      (await getEmbeddingFunction(
+        data.name,
+        data.configuration_json.embedding_function ?? undefined,
+      )) ?? 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 ?? undefined) ?? undefined,
       embeddingFunction: resolvedEmbeddingFunction,
@@ -489,6 +527,8 @@ export class ChromaClient {
       chromaClient: this,
       apiClient: this.apiClient,
       name,
+      tenant: data.tenant,
+      database: data.database,
       configuration: data.configuration_json,
       metadata: deserializeMetadata(data.metadata ?? undefined) ?? undefined,
       embeddingFunction: resolvedEmbeddingFunction,

clients/new-js/packages/chromadb/src/collection.ts

@@ -52,6 +52,10 @@ import type { SparseVectorIndexConfig } from "./schema";
  * Provides methods for adding, querying, updating, and deleting records.
  */
 export interface Collection {
+  /** Tenant name */
+  tenant: string;
+  /** Database name */
+  database: string;
   /** Unique identifier for the collection */
   id: string;
   /** Name of the collection */
@@ -214,6 +218,10 @@ export interface CollectionArgs {
   name: string;
   /** Collection ID */
   id: string;
+  /** Tenant name */
+  tenant: string;
+  /** Database name */
+  database: string;
   /** Embedding function for the collection */
   embeddingFunction?: EmbeddingFunction;
   /** Collection configuration */
@@ -232,6 +240,8 @@ export class CollectionImpl implements Collection {
   protected readonly chromaClient: ChromaClient;
   protected readonly apiClient: ReturnType<typeof createClient>;
   public readonly id: string;
+  public readonly tenant: string;
+  public readonly database: string;
   private _name: string;
   private _metadata: CollectionMetadata | undefined;
   private _configuration: CollectionConfiguration;
@@ -246,6 +256,8 @@ export class CollectionImpl implements Collection {
     chromaClient,
     apiClient,
     id,
+    tenant,
+    database,
     name,
     metadata,
     configuration,
@@ -255,6 +267,8 @@ export class CollectionImpl implements Collection {
     this.chromaClient = chromaClient;
     this.apiClient = apiClient;
     this.id = id;
+    this.tenant = tenant;
+    this.database = database;
     this._name = name;
     this._metadata = metadata;
     this._configuration = configuration;
@@ -309,9 +323,9 @@ export class CollectionImpl implements Collection {
     database: string;
     collection_id: string;
   }> {
-    const clientPath = await this.chromaClient._path();
     return {
-      ...clientPath,
+      tenant: this.tenant,
+      database: this.database,
       collection_id: this.id,
     };
   }
@@ -958,6 +972,8 @@ export class CollectionImpl implements Collection {
       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 ?? undefined) ?? undefined,

clients/new-js/packages/chromadb/test/schema.test.ts

@@ -1349,6 +1349,8 @@ describe("Schema", () => {
       apiClient: {} as any,
       id: "test-id",
       name: "test",
+      tenant: "default_tenant",
+      database: "default_database",
       configuration: {} as CollectionConfiguration,
       metadata: undefined as CollectionMetadata | undefined,
       embeddingFunction: undefined,
@@ -1399,6 +1401,8 @@ describe("Schema", () => {
       apiClient: mockApiClient as any,
       id: "test-id",
       name: "test",
+      tenant: "default_tenant",
+      database: "default_database",
       configuration: {} as CollectionConfiguration,
       metadata: undefined as CollectionMetadata | undefined,
       embeddingFunction: undefined,
@@ -1470,6 +1474,8 @@ describe("Schema", () => {
       apiClient: mockApiClient as any,
       id: "test-id",
       name: "test",
+      tenant: "default_tenant",
+      database: "default_database",
       configuration: {} as CollectionConfiguration,
       metadata: undefined as CollectionMetadata | undefined,
       embeddingFunction: undefined,
@@ -1545,6 +1551,8 @@ describe("Schema", () => {
       apiClient: mockApiClient as any,
       id: "test-id",
       name: "test",
+      tenant: "default_tenant",
+      database: "default_database",
       configuration: {} as CollectionConfiguration,
       metadata: undefined as CollectionMetadata | undefined,
       embeddingFunction: undefined,
@@ -1621,6 +1629,8 @@ describe("Schema", () => {
       apiClient: mockApiClient as any,
       id: "test-id",
       name: "test",
+      tenant: "default_tenant",
+      database: "default_database",
       configuration: {} as CollectionConfiguration,
       metadata: undefined as CollectionMetadata | undefined,
       embeddingFunction: undefined,
@@ -1691,6 +1701,8 @@ describe("Schema", () => {
       apiClient: mockApiClient as any,
       id: "test-id",
       name: "test",
+      tenant: "default_tenant",
+      database: "default_database",
       configuration: {} as CollectionConfiguration,
       metadata: undefined as CollectionMetadata | undefined,
       embeddingFunction: undefined,

clients/new-js/packages/chromadb/test/search.expression.test.ts

@@ -380,6 +380,8 @@ describe("search expression DSL", () => {
       apiClient: mockApiClient as any,
       id: "col-id",
       name: "test",
+      tenant: "default_tenant",
+      database: "default_database",
       configuration: {} as CollectionConfiguration,
       metadata: undefined,
       embeddingFunction,
@@ -469,6 +471,8 @@ describe("search expression DSL", () => {
       apiClient: mockApiClient as any,
       id: "col-id",
       name: "test",
+      tenant: "default_tenant",
+      database: "default_database",
       configuration: {} as CollectionConfiguration,
       metadata: undefined,
       embeddingFunction: undefined,