Skip to content
This repository was archived by the owner on Mar 2, 2026. It is now read-only.

Commit 643f014

Browse files
daniel-sanchedaniel-sanche
authored
feat: Additional Pipeline Expressions (#1115)
* fixed tests * added vector expressions * added new math expressions * added string manipulation expressions * added not_nan, not_null, and is_absent * added new Array type * added map and related expressions * remove dict and list from constant types * Fixed lint * added count_if and count_distinct * added misc expressions * added error functions * fixed lint * fixed typos
1 parent fed7af2 commit 643f014

File tree

9 files changed

+2546
-208
lines changed

9 files changed

+2546
-208
lines changed

‎google/cloud/firestore_v1/_pipeline_stages.py‎

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -274,13 +274,17 @@ def __init__(
274274
self,
275275
field: str | Expr,
276276
vector: Sequence[float] | Vector,
277-
distance_measure: "DistanceMeasure",
277+
distance_measure: "DistanceMeasure" | str,
278278
options: Optional["FindNearestOptions"] = None,
279279
):
280280
super().__init__("find_nearest")
281281
self.field: Expr = Field(field) if isinstance(field, str) else field
282282
self.vector: Vector = vector if isinstance(vector, Vector) else Vector(vector)
283-
self.distance_measure = distance_measure
283+
self.distance_measure = (
284+
distance_measure
285+
if isinstance(distance_measure, DistanceMeasure)
286+
else DistanceMeasure[distance_measure.upper()]
287+
)
284288
self.options = options or FindNearestOptions()
285289

286290
def _pb_args(self):

0 commit comments

Comments
 (0)