vertebrae.extractors.hosted

Hosted embedding API extractor.

Classes

HostedEmbeddingExtractor

Wrap a hosted embedding API behind a batch-oriented callable.

Module Contents

class vertebrae.extractors.hosted.HostedEmbeddingExtractor(name, provider, model, embed_fn, batch_size=64, modality='text', input_fn=None, output_fn=None, request_metadata=None, recipe_data=None, max_retries=2, retry_backoff_seconds=0.0, cache_embeddings=False, cache_identity=None)[source]

Wrap a hosted embedding API behind a batch-oriented callable.

Parameters:

  • name (str)

  • provider (str)

  • model (str)

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

  • batch_size (int)

  • modality (str)

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

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

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

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

  • max_retries (int)

  • retry_backoff_seconds (float)

  • cache_embeddings (bool)

  • cache_identity (Optional[str])