[ENH] Implement multiple member assignment

[Sicheng-Pan]

Description of changes

Summarize the changes made by this PR.

• Improvements & Bug fixes
  • N/A
• New functionality
  • Implements the multi assignment logic, which allows to deterministically assign multiple members for a given key and hasher. This mirrors the Python implementation.

Test plan

How are these changes tested?

• Tests pass locally with pytest for python, yarn test for js, cargo test for rust

Documentation Changes

Are all docstrings for user-facing APIs updated if required? Do we need to make documentation changes in the docs repository?

[github-actions]

Reviewer Checklist

Please leverage this checklist to ensure your code review is thorough before approving

Testing, Bugs, Errors, Logs, Documentation

• Can you think of any use case in which the code does not behave as intended? Have they been tested?
• Can you think of any inputs or external events that could break the code? Is user input validated and safe? Have they been tested?
• If appropriate, are there adequate property based tests?
• If appropriate, are there adequate unit tests?
• Should any logging, debugging, tracing information be added or removed?
• Are error messages user-friendly?
• Have all documentation changes needed been made?
• Have all non-obvious changes been commented?

System Compatibility

• Are there any potential impacts on other parts of the system or backward compatibility?
• Does this change intersect with any items on our roadmap, and if so, is there a plan for fitting them together?

Quality

• Is this code of a unexpectedly high quality (Readability, Modularity, Intuitiveness)

[Sicheng-Pan]

• [ENH] Implement multiple member assignment #3553 👈 (View in Graphite)
• [CLN] Move memberlist and assignment module out of worker crate #3552
• [CLN] Move log module outside of worker crate #3547
• main

This stack of pull requests is managed by Graphite. Learn more about stacking.

[HammadB]

Can we copy the python testing?

chroma/chromadb/test/segment/distributed/test_rendezvous_hash.py

Line 16 in b848326

def test_even_distribution() -> None:

chroma/chromadb/test/segment/distributed/test_rendezvous_hash.py

Line 41 in b848326

def test_multi_assign_even_distribution() -> None:

[Sicheng-Pan]

Can we copy the python testing?

chroma/chromadb/test/segment/distributed/test_rendezvous_hash.py

Line 16 in b848326

def test_even_distribution() -> None:

chroma/chromadb/test/segment/distributed/test_rendezvous_hash.py

Line 41 in b848326

def test_multi_assign_even_distribution() -> None:

I've migrated the tests

[Sicheng-Pan]

Merge activity

• Jan 23, 8:20 PM EST: A user started a stack merge that includes this pull request via Graphite.
• Jan 23, 9:12 PM EST: Graphite rebased this pull request as part of a merge.
• Jan 23, 10:01 PM EST: A user merged this pull request with Graphite.