Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Chroma: allow filters in ChromaEmbeddingRetriever #341

Closed
nachollorca opened this issue Feb 5, 2024 · 1 comment
Closed

Chroma: allow filters in ChromaEmbeddingRetriever #341

nachollorca opened this issue Feb 5, 2024 · 1 comment
Labels
feature request Ideas to improve an integration

Comments

@nachollorca
Copy link
Contributor

Is your feature request related to a problem? Please describe.
I need to pass filters to the ChromaEmbeddingRetriever to retrieve on specific segments of the ChromaDocumentStore at retrieval time.

Describe the solution you'd like
I want to be able to do this:

retriever = ChromaEmbeddingRetriever(document_store = docstore, top_k=5)
embedder = OpenAITextEmbedder(model = "text-embedding-3-small")

retrieval = Pipeline()
retrieval.add_component("embedder", embedder)
retrieval.add_component("retriever", retriever)
retrieval.connect("embedder.embedding", "retriever.query_embedding")
retrieval.run({
    "embedder": {"text": "What do they say about their families"},
    "retriever": {"filters": {"doc_id": 4}}
    })
@anakin87
Copy link
Member

anakin87 commented Feb 7, 2024

done in #342.

@anakin87 anakin87 closed this as completed Feb 7, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
feature request Ideas to improve an integration
Projects
None yet
Development

No branches or pull requests

2 participants