apache_beam.ml.rag.embeddings.base module

RAG-specific embedding adapters.

This module provides adapters for extracting content from EmbeddableItem instances and mapping embeddings back. Adapters are used by EmbeddingsManager to support various input types.

apache_beam.ml.rag.embeddings.base.create_text_adapter() → EmbeddingTypeAdapter[EmbeddableItem, EmbeddableItem]

Creates adapter for text content embedding.

Works with any EmbeddableItem that has text content (content.text). Extracts text for embedding and maps results back as Embedding objects.

Returns:

EmbeddingTypeAdapter configured for text embedding

apache_beam.ml.rag.embeddings.base.create_rag_adapter() → EmbeddingTypeAdapter[EmbeddableItem, EmbeddableItem]

Creates adapter for text content embedding.

Works with any EmbeddableItem that has text content (content.text). Extracts text for embedding and maps results back as Embedding objects.

Returns:

EmbeddingTypeAdapter configured for text embedding