vertebrae.extractors.retrieval

Explicit adapter for independent query and gallery embedding branches.

Classes

CallableRetrievalExtractor

Wrap explicit frozen query/gallery encoding functions for retrieval.

Module Contents

class vertebrae.extractors.retrieval.CallableRetrievalExtractor(name, query_fn, gallery_fn=None, *, query_modality='embeddings', gallery_modality=None, metadata=None, cache_identity=None)

Wrap explicit frozen query/gallery encoding functions for retrieval.

Parameters:

• name (str)

• query_fn (Callable[[Any], Any])

• gallery_fn (Optional[Callable[[Any], Any]])

• query_modality (str)

• gallery_modality (Optional[str])

• metadata (Optional[Dict[str, Any]])

• cache_identity (Optional[str])