feat: chromadb generic client (#3374)
Some checks failed
Notify Downstream / notify-downstream (push) Has been cancelled
Mark stale issues and pull requests / stale (push) Has been cancelled

Add ChromaDB client implementation with async support

- Implement core collection operations (create, get_or_create, delete)
- Add search functionality with cosine similarity scoring
- Include both sync and async method variants
- Add type safety with NamedTuples and TypeGuards
- Extract utility functions to separate modules
- Default to cosine distance metric for text similarity
- Add comprehensive test coverage

TODO:
- l2, ip score calculations are not settled on
This commit is contained in:
Greyson LaLonde
2025-08-21 18:18:46 -04:00
committed by GitHub
parent 1217935b31
commit 842bed4e9c
7 changed files with 1411 additions and 0 deletions

0
tests/rag/__init__.py Normal file
View File