Downgrade pydantic to 1.x (#50)

As the title states. This is for compatibility with Langchain.

Author: Sam Partee

redisvl/index.py

@@ -95,7 +95,7 @@ def from_yaml(cls, schema_path: str):
             SearchIndex: A SearchIndex object.
         """
         schema = read_schema(schema_path)
-        return cls(fields=schema.index_fields, **schema.index.model_dump())
+        return cls(fields=schema.index_fields, **schema.index.dict())
 
     @classmethod
     def from_dict(cls, schema_dict: Dict[str, Any]):
@@ -108,7 +108,7 @@ def from_dict(cls, schema_dict: Dict[str, Any]):
             SearchIndex: A SearchIndex object.
         """
         schema = SchemaModel(**schema_dict)
-        return cls(fields=schema.index_fields, **schema.index.model_dump())
+        return cls(fields=schema.index_fields, **schema.index.dict())
 
     @classmethod
     def from_existing(

redisvl/schema.py

@@ -3,7 +3,7 @@
 from uuid import uuid4
 
 import yaml
-from pydantic import BaseModel, Field, field_validator
+from pydantic import BaseModel, Field, validator
 from redis.commands.search.field import (
     GeoField,
     NumericField,
@@ -66,8 +66,7 @@ class BaseVectorField(BaseModel):
     distance_metric: str = Field(default="COSINE")
     initial_cap: int = Field(default=20000)
 
-    @field_validator("algorithm", "datatype", "distance_metric", mode="before")
-    @classmethod
+    @validator("algorithm", "datatype", "distance_metric", pre=True, each_item=True)
     def uppercase_strings(cls, v):
         return v.upper()
 
@@ -132,8 +131,7 @@ class SchemaModel(BaseModel):
     index: IndexModel = Field(...)
     fields: FieldsModel = Field(...)
 
-    @field_validator("index")
-    @classmethod
+    @validator("index")
     def validate_index(cls, v):
         if v.storage_type not in ["hash", "json"]:
             raise ValueError(f"Storage type {v.storage_type} not supported")
@@ -142,7 +140,7 @@ def validate_index(cls, v):
     @property
     def index_fields(self):
         redis_fields = []
-        for field_name in self.fields.model_fields.keys():
+        for field_name in self.fields.__fields__.keys():
             field_group = getattr(self.fields, field_name)
             if field_group is not None:
                 for field in field_group:
@@ -158,4 +156,4 @@ def read_schema(file_path: str):
     with open(fp, "r") as f:
         schema = yaml.safe_load(f)
 
-    return SchemaModel(**schema)
+    return SchemaModel(**schema)

requirements.txt

@@ -2,6 +2,6 @@ numpy
 redis>=4.3.4
 pyyaml
 coloredlogs
-pydantic>=2.0.0
+pydantic<2.0.0
 tenacity==8.2.2
 tabulate==0.9.0

tests/integration/test_simple_async.py

@@ -102,4 +102,4 @@ async def test_simple(async_client):
         print("Score:", doc.vector_distance)
         pprint(doc)
 
-    await index.delete()
+    await index.delete()