apache_beam.ml.rag.enrichment package

Enrichment components for RAG pipelines. This module provides components for vector search enrichment in RAG pipelines.

Submodules

• apache_beam.ml.rag.enrichment.bigquery_vector_search module
  • BigQueryVectorSearchParameters
    • BigQueryVectorSearchParameters.project
    • BigQueryVectorSearchParameters.table_name
    • BigQueryVectorSearchParameters.embedding_column
    • BigQueryVectorSearchParameters.columns
    • BigQueryVectorSearchParameters.neighbor_count
    • BigQueryVectorSearchParameters.metadata_restriction_template
    • BigQueryVectorSearchParameters.distance_type
    • BigQueryVectorSearchParameters.options
    • BigQueryVectorSearchParameters.format_query()
  • BigQueryVectorSearchEnrichmentHandler
    • BigQueryVectorSearchEnrichmentHandler.batch_elements_kwargs()
  • join_fn()