Implement Geographic Filter

docs/api/filter.rst

@@ -57,3 +57,20 @@ NumericFilter
     :show-inheritance:
     :members:
     :inherited-members:
+
+
+GeoFilter
+=========
+
+.. currentmodule:: redisvl.query
+
+.. autosummary::
+
+    GeoFilter.__init__
+    GeoFilter.to_string
+
+
+.. autoclass:: GeoFilter
+    :show-inheritance:
+    :members:
+    :inherited-members:

docs/user_guide/jupyterutils.py

@@ -0,0 +1,42 @@
+from IPython.display import display, HTML
+
+def table_print(dict_list):
+    # If there's nothing in the list, there's nothing to print
+    if len(dict_list) == 0:
+        return
+
+    # Getting column names (dictionary keys) using the first dictionary
+    columns = dict_list[0].keys()
+
+    # HTML table header
+    html = '<table><tr><th>'
+    html += '</th><th>'.join(columns)
+    html += '</th></tr>'
+
+    # HTML table content
+    for dictionary in dict_list:
+        html += '<tr><td>'
+        html += '</td><td>'.join(str(dictionary[column]) for column in columns)
+        html += '</td></tr>'
+
+    # HTML table footer
+    html += '</table>'
+
+    # Displaying the table
+    display(HTML(html))
+
+
+def result_print(results):
+    # If there's nothing in the list, there's nothing to print
+    if len(results.docs) == 0:
+        return
+
+    data = [doc.__dict__ for doc in results.docs]
+
+    to_remove = ["id", "payload"]
+    for doc in data:
+        for key in to_remove:
+            if key in doc:
+                del doc[key]
+
+    table_print(data)

redisvl/query.py

@@ -66,6 +66,61 @@ def to_string(self) -> str:
         )
 
 
+class GeoFilter(Filter):
+    GEO_UNITS = ["m", "km", "mi", "ft"]
+
+    def __init__(self, field, longitude, latitude, radius, unit="km"):
+        """Filter for Geo fields.
+
+        Args:
+            field (str): The field to filter on.
+            longitude (float): The longitude.
+            latitude (float): The latitude.
+            radius (float): The radius.
+            unit (str, optional): The unit of the radius. Defaults to "km".
+
+        Raises:
+            ValueError: If the unit is not one of ["m", "km", "mi", "ft"].
+
+        Examples:
+            >>> # looking for Chinese restaurants near San Francisco
+            >>> # (within a 5km radius) would be
+            >>> #
+            >>> from redisvl.query import GeoFilter
+            >>> gf = GeoFilter("location", -122.4194, 37.7749, 5)
+        """
+        super().__init__(field)
+        self._longitude = longitude
+        self._latitude = latitude
+        self._radius = radius
+        self._unit = self._set_unit(unit)
+
+    def _set_unit(self, unit):
+        if unit.lower() not in self.GEO_UNITS:
+            raise ValueError(f"Unit must be one of {self.GEO_UNITS}")
+        return unit.lower()
+
+    def to_string(self) -> str:
+        """Converts the geo filter to a string.
+
+        Returns:
+            str: The geo filter as a string.
+        """
+        return (
+            "@"
+            + self._field
+            + ":["
+            + str(self._longitude)
+            + " "
+            + str(self._latitude)
+            + " "
+            + str(self._radius)
+            + " "
+            + self._unit
+            + "]"
+        )
+
+
 class NumericFilter(Filter):
     def __init__(self, field, minval, maxval, min_exclusive=False, max_exclusive=False):
         """Filter for Numeric fields.

tests/test_filter.py

@@ -1,6 +1,13 @@
 import pytest
 
-from redisvl.query import Filter, NumericFilter, TagFilter, TextFilter, VectorQuery
+from redisvl.query import (
+    Filter,
+    GeoFilter,
+    NumericFilter,
+    TagFilter,
+    TextFilter,
+    VectorQuery,
+)
 from redisvl.utils.utils import TokenEscaper
 
 
@@ -25,6 +32,13 @@ def test_text_filter(self):
         txt_f = TextFilter("text_field", "text")
         assert txt_f.to_string() == "@text_field:text"
 
+    def test_geo_filter(self):
+        geo_f = GeoFilter("geo_field", 1, 2, 3)
+        assert geo_f.to_string() == "@geo_field:[1 2 3 km]"
+
+        geo_f = GeoFilter("geo_field", 1, 2, 3, unit="m")
+        assert geo_f.to_string() == "@geo_field:[1 2 3 m]"
+
     def test_filters_combination(self):
         tf1 = TagFilter("tag_field", ["tag1", "tag2"])
         tf2 = TagFilter("tag_field", ["tag3"])