Update featurestore.py (#624)

Updated to include numeric_filters.
langchain-ai · Dec 16, 2024 · 8c569ed · 8c569ed
1 parent dbad3b8
commit 8c569ed
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/libs/community/langchain_google_community/bq_storage_vectorstores/featurestore.py b/libs/community/langchain_google_community/bq_storage_vectorstores/featurestore.py
@@ -421,6 +421,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,
@@ -438,6 +439,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,