manually parse special attributes that have no associated value

Author: justin-cechmanek

redisvl/redis/connection.py

@@ -98,7 +98,25 @@ def parse_vector_attrs(attrs):
         return vector_attrs
 
     def parse_attrs(attrs):
-        return {attrs[i].lower(): attrs[i + 1] for i in range(6, len(attrs), 2)}
+        # 'SORTABLE', 'UNF', 'NOSTEM' don't have corresponding values.
+        # Their presence indicates boolean True
+        original = attrs.copy()
+        parsed_attrs = {}
+        if "NOSTEM" in attrs:
+            parsed_attrs["no_stem"] = True
+            attrs.remove("NOSTEM")
+        for special_attr in ["SORTABLE", "UNF"]:
+            if special_attr in attrs:
+                parsed_attrs[special_attr.lower()] = True
+                attrs.remove(special_attr)
+
+        try:
+            parsed_attrs.update(
+                {attrs[i].lower(): attrs[i + 1] for i in range(6, len(attrs), 2)}
+            )
+        except IndexError as e:
+            raise IndexError(f"Error parsing index attributes {original}, {str(e)}")
+        return parsed_attrs
 
     schema_fields = []
 

tests/integration/test_search_index.py

@@ -6,7 +6,15 @@
 from redisvl.redis.utils import convert_bytes
 from redisvl.schema import IndexSchema, StorageType
 
-fields = [{"name": "test", "type": "tag"}]
+fields = [
+    {"name": "test", "type": "tag"},
+    {"name": "test_text", "type": "text"},
+    {
+        "name": "test_text_attrs",
+        "type": "text",
+        "attrs": {"no_stem": True, "sortable": True},
+    },
+]
 
 
 @pytest.fixture
@@ -87,7 +95,7 @@ def test_search_index_from_existing_complex(client):
                 "name": "age",
                 "type": "numeric",
                 "path": "$.metadata.age",
-                "attrs": {"sortable": False},
+                "attrs": {"sortable": True},
             },
             {
                 "name": "user_embedding",