apache_beam.ml.rag.embeddings package

Embedding components for RAG pipelines. This module provides components for generating embeddings in RAG pipelines.

Submodules

• apache_beam.ml.rag.embeddings.base module
  • create_rag_adapter()
• apache_beam.ml.rag.embeddings.huggingface module
  • HuggingfaceTextEmbeddings
    • HuggingfaceTextEmbeddings.get_model_handler()
    • HuggingfaceTextEmbeddings.get_ptransform_for_processing()
• apache_beam.ml.rag.embeddings.vertex_ai module
  • VertexAITextEmbeddings
    • VertexAITextEmbeddings.get_model_handler()
    • VertexAITextEmbeddings.get_ptransform_for_processing()