diff --git a/libs/community/langchain_google_community/bq_storage_vectorstores/featurestore.py b/libs/community/langchain_google_community/bq_storage_vectorstores/featurestore.py index 3a9469b2..84653ef4 100644 --- a/libs/community/langchain_google_community/bq_storage_vectorstores/featurestore.py +++ b/libs/community/langchain_google_community/bq_storage_vectorstores/featurestore.py @@ -385,6 +385,7 @@ def _search_embedding( entity_id: Optional[str] = None, k: int = 5, string_filters: Optional[List[dict]] = None, + numeric_filters: Optional[List[dict]] = None, per_crowding_attribute_neighbor_count: Optional[int] = None, approximate_neighbor_candidates: Optional[int] = None, leaf_nodes_search_fraction: Optional[float] = None, @@ -402,6 +403,7 @@ def _search_embedding( embedding=embedding, neighbor_count=k, string_filters=string_filters, + numeric_filters=numeric_filters, per_crowding_attribute_neighbor_count=per_crowding_attribute_neighbor_count, parameters={ "approximate_neighbor_candidates": approximate_neighbor_candidates,