Skip to content
Merged
8 changes: 6 additions & 2 deletions google/cloud/firestore_v1/_pipeline_stages.py
Original file line number Diff line number Diff line change
Expand Up @@ -274,13 +274,17 @@ def __init__(
self,
field: str | Expr,
vector: Sequence[float] | Vector,
distance_measure: "DistanceMeasure",
distance_measure: "DistanceMeasure" | str,
options: Optional["FindNearestOptions"] = None,
):
super().__init__("find_nearest")
self.field: Expr = Field(field) if isinstance(field, str) else field
self.vector: Vector = vector if isinstance(vector, Vector) else Vector(vector)
self.distance_measure = distance_measure
self.distance_measure = (
distance_measure
if isinstance(distance_measure, DistanceMeasure)
else DistanceMeasure[distance_measure.upper()]
)
self.options = options or FindNearestOptions()

def _pb_args(self):
Expand Down
Loading