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()
- apache_beam.ml.rag.enrichment.milvus_search module
SearchStrategyKeywordSearchMetricsVectorSearchMetricsMilvusBaseRankerMilvusConnectionParametersBaseSearchParametersVectorSearchParametersKeywordSearchParametersHybridSearchParametersMilvusSearchParametersMilvusCollectionLoadParametersMilvusSearchResultMilvusSearchEnrichmentHandlerMilvusSearchEnrichmentHandler.convert_sparse_embedding_to_milvus_format()MilvusSearchEnrichmentHandler.collection_nameMilvusSearchEnrichmentHandler.search_strategyMilvusSearchEnrichmentHandler.partition_namesMilvusSearchEnrichmentHandler.output_fieldsMilvusSearchEnrichmentHandler.timeoutMilvusSearchEnrichmentHandler.round_decimalMilvusSearchEnrichmentHandler.batch_elements_kwargs()
join_fn()unpack_dataclass_with_kwargs()