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
BigQueryVectorSearchParametersBigQueryVectorSearchParameters.projectBigQueryVectorSearchParameters.table_nameBigQueryVectorSearchParameters.embedding_columnBigQueryVectorSearchParameters.columnsBigQueryVectorSearchParameters.neighbor_countBigQueryVectorSearchParameters.metadata_restriction_templateBigQueryVectorSearchParameters.distance_typeBigQueryVectorSearchParameters.optionsBigQueryVectorSearchParameters.include_distanceBigQueryVectorSearchParameters.format_query()
BigQueryVectorSearchEnrichmentHandlerjoin_fn()