Adapters
Adapters are an optional feature to transform data before adding to or querying from a collection. Adapters provide a customizable and modular way to express data transformations and make interacting with collections more ergonomic.
Additionally, adapter transformations are applied lazily and can internally batch operations which can make them more memory and CPU efficient compared to manually executing transforms.
Example:
As an example, we’ll create a collection with an adapter that chunks text into paragraphs and converts each chunk into an embedding vector using the all-MiniLM-L6-v2
model.
First, install vecs
with optional dependencies for text embeddings:
Then create a collection with an adapter to chunk text into paragraphs and embed each paragraph using the all-MiniLM-L6-v2
384 dimensional text embedding model.
With the adapter registered against the collection, we can upsert records into the collection passing in text rather than vectors.
Similarly, we can query the collection using text.
In summary, Adapter
s allow you to work with a collection as though they store your prefered data type natively.
Built-in Adapters
vecs provides several built-in Adapters.
Have an idea for a useful adapter? Open an issue requesting it.
ParagraphChunker
The ParagraphChunker
AdapterStep
splits text media into paragraphs and yields each paragraph as a separate record. That can be a useful preprocessing step when upserting large documents that contain multiple paragraphs. The ParagraphChunker
delimits paragraphs by two consecutive line breaks \n\n
.
ParagrphChunker
is a pre-preocessing step and must be used in combination with another adapter step like TextEmbedding
to transform the chunked text into a vector.
When querying the collection, you probably do not want to chunk the text. To skip text chunking during queries, set the skip_during_query
argument to True
. Setting skip_during_query
to False
will raise an exception if the input text contains more than one paragraph.
TextEmbedding
The TextEmbedding
AdapterStep
accepts text and converts it into a vector that can be consumed by the Collection
. TextEmbedding
supports all models available in the sentence_transformers
package. A complete list of supported models is available in vecs.adapter.TextEmbeddingModel
.
Interface
Adapters are objects that take in data in the form of Iterable[Tuple[str, Any, Optional[Dict]]]
where Tuple[str, Any, Optional[Dict]]]
represents records of (id, media, metadata)
.
The main use of Adapters is to transform the media part of the records into a form that is ready to be ingested into the collection (like converting text into embeddings). However, Adapters can also modify the id
or metadata
if required.
Due to the common interface, adapters may be comprised of multiple adapter steps to create multi-stage preprocessing pipelines. For example, a multi-step adapter might first convert text into chunks and then convert each text chunk into an embedding vector.