endee 0.1.9__py3-none-any.whl → 0.1.10__py3-none-any.whl

endee/endee.py

@@ -569,7 +569,7 @@ class Endee:
                 token=self.token,
                 url=self.base_url,
                 version=self.version,
-                params=metadata.model_dump(by_alias=True),
+                params=metadata.dict(by_alias=True),
                 session_client_manager=self.session_manager,
             )
         else:
@@ -578,7 +578,7 @@ class Endee:
                 token=self.token,
                 url=self.base_url,
                 version=self.version,
-                params=metadata.model_dump(by_alias=True),
+                params=metadata.dict(by_alias=True),
                 session_client_manager=self.client_manager,
             )
 

endee/schema.py

@@ -2,7 +2,7 @@ import re
 import sys
 from typing import Any, Dict, List, Optional, Union
 
-from pydantic import BaseModel, ConfigDict, Field, field_validator, model_validator
+from pydantic import BaseModel, Field, root_validator, validator
 
 from .constants import (
     DEFAULT_EF_SEARCH,
@@ -20,24 +20,33 @@ from .constants import (
 class VectorItem(BaseModel):
     """Model for a single vector item in an upsert operation."""
 
-    id: str = Field(..., min_length=1)
+    id: str
     vector: List[float]
     meta: Optional[Dict[str, Any]] = Field(default_factory=dict)
     filter: Optional[Dict[str, Any]] = Field(default_factory=dict)
     sparse_indices: Optional[List[int]] = None
     sparse_values: Optional[List[float]] = None
 
-    @model_validator(mode="after")
-    def validate_sparse_data(self) -> "VectorItem":
-        if (self.sparse_indices is None) != (self.sparse_values is None):
+    @validator("id")
+    @classmethod
+    def validate_id(cls, v: str) -> str:
+        if len(v) < 1:
+            raise ValueError("id must not be empty")
+        return v
+
+    @root_validator
+    @classmethod
+    def validate_sparse_data(cls, values: Dict[str, Any]) -> Dict[str, Any]:
+        sparse_indices = values.get("sparse_indices")
+        sparse_values = values.get("sparse_values")
+
+        if (sparse_indices is None) != (sparse_values is None):
             raise ValueError(
                 "Both sparse_indices and sparse_values must be provided together"
             )
-        if self.sparse_indices is not None and len(self.sparse_indices) != len(
-            self.sparse_values
-        ):
+        if sparse_indices is not None and len(sparse_indices) != len(sparse_values):
             raise ValueError("sparse_indices and sparse_values must match in length")
-        return self
+        return values
 
 
 class QueryRequest(BaseModel):
@@ -51,10 +60,15 @@ class QueryRequest(BaseModel):
     sparse_indices: Optional[List[int]] = None
     sparse_values: Optional[List[float]] = None
 
-    @model_validator(mode="after")
-    def validate_query_type(self) -> "QueryRequest":
-        has_dense = self.vector is not None
-        has_sparse = self.sparse_indices is not None or self.sparse_values is not None
+    @root_validator
+    @classmethod
+    def validate_query_type(cls, values: Dict[str, Any]) -> Dict[str, Any]:
+        vector = values.get("vector")
+        sparse_indices = values.get("sparse_indices")
+        sparse_values = values.get("sparse_values")
+
+        has_dense = vector is not None
+        has_sparse = sparse_indices is not None or sparse_values is not None
 
         if not has_dense and not has_sparse:
             raise ValueError(
@@ -62,17 +76,15 @@ class QueryRequest(BaseModel):
                 " must be provided."
             )
 
-        if (self.sparse_indices is None) != (self.sparse_values is None):
+        if (sparse_indices is None) != (sparse_values is None):
             raise ValueError(
                 "Both sparse_indices and sparse_values must be provided together"
             )
 
-        if self.sparse_indices is not None and len(self.sparse_indices) != len(
-            self.sparse_values
-        ):
+        if sparse_indices is not None and len(sparse_indices) != len(sparse_values):
             raise ValueError("sparse_indices and sparse_values must match in length")
 
-        return self
+        return values
 
 
 class IndexCreateRequest(BaseModel):
@@ -87,7 +99,7 @@ class IndexCreateRequest(BaseModel):
     version: Optional[int] = None
     sparse_dim: int = Field(default=0, ge=0, le=sys.maxsize)
 
-    @field_validator("name")
+    @validator("name")
     @classmethod
     def validate_name(cls, v: str) -> str:
         if not re.match(r"^[a-zA-Z0-9_]+$", v):
@@ -101,7 +113,7 @@ class IndexCreateRequest(BaseModel):
             )
         return v
 
-    @field_validator("space_type")
+    @validator("space_type")
     @classmethod
     def validate_space_type(cls, v: str) -> str:
         v = v.lower()
@@ -111,7 +123,7 @@ class IndexCreateRequest(BaseModel):
             )
         return v
 
-    @field_validator("precision")
+    @validator("precision")
     @classmethod
     def validate_precision(cls, v: Union[str, Precision]) -> Union[str, Precision]:
         if isinstance(v, Precision):
@@ -126,7 +138,8 @@ class IndexCreateRequest(BaseModel):
 class IndexMetadata(BaseModel):
     """Model for index metadata returned by the server."""
 
-    model_config = ConfigDict(populate_by_name=True)
+    class Config:
+        allow_population_by_field_name = True
 
     name: Optional[str] = Field(None, alias="name")
     lib_token: str

{endee-0.1.9.dist-info → endee-0.1.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: endee
-Version: 0.1.9
+Version: 0.1.10
 Summary: Endee is the Next-Generation Vector Database for Scalable, High-Performance AI
 Home-page: https://endee.io
 Author: Endee Labs
@@ -29,7 +29,7 @@ Requires-Dist: httpx[http2]>=0.28.1
 Requires-Dist: numpy>=2.2.4
 Requires-Dist: msgpack>=1.1.0
 Requires-Dist: orjson>=3.11.5
-Requires-Dist: pydantic>=2.0.0
+Requires-Dist: pydantic==1.10.26
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier

endee-0.1.10.dist-info/RECORD

@@ -0,0 +1,13 @@
+endee/__init__.py,sha256=PZnr7T97fJlLfv9bMPeiSy8vcySBpFgBF2b4VJZbE1s,58
+endee/compression.py,sha256=HdNKSpdOFJVv_iGYV9UcLj-1_VMcX8tl1dH7mZB9x50,1134
+endee/constants.py,sha256=kg96MFbtbNIZbjyYS32SqBf-pWYL1TUJztlcE96PGxk,4297
+endee/endee.py,sha256=SFDLLEiBixTC389RdniEcJWhav1T1gCnZ64iakBihto,19831
+endee/exceptions.py,sha256=u1JZKvKHUyLlWlfaCXgN8vzASkbwIJeEQqtllKVr2ug,7505
+endee/index.py,sha256=0X_AU_4ROSQsklwTpo8iYjpEU3MUAslwCrpGGenyvAM,22751
+endee/schema.py,sha256=Y52cI_97hPW_7c0L8lip0fv_a2pNZWKOQ2dH4pJITUM,5026
+endee/utils.py,sha256=UGRMf_2DLDE9Ze-2HDoeAv97Mc6w8ABIX9M1MFOBwVc,1258
+endee-0.1.10.dist-info/licenses/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+endee-0.1.10.dist-info/METADATA,sha256=volhvIuSQ6qoZWPhwmhiEOOaBdHEARmC3iqyhKLwIy0,24338
+endee-0.1.10.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+endee-0.1.10.dist-info/top_level.txt,sha256=zOEvXIfzdm7vXJaVX_jq5OX3fTftKq14KzynxlAp8ZQ,6
+endee-0.1.10.dist-info/RECORD,,

endee-0.1.9.dist-info/RECORD

@@ -1,13 +0,0 @@
-endee/__init__.py,sha256=PZnr7T97fJlLfv9bMPeiSy8vcySBpFgBF2b4VJZbE1s,58
-endee/compression.py,sha256=HdNKSpdOFJVv_iGYV9UcLj-1_VMcX8tl1dH7mZB9x50,1134
-endee/constants.py,sha256=kg96MFbtbNIZbjyYS32SqBf-pWYL1TUJztlcE96PGxk,4297
-endee/endee.py,sha256=4sCNsUNqJqPP41IyUjT2MZnC6QvkBjerGVUnJhEZJXs,19843
-endee/exceptions.py,sha256=u1JZKvKHUyLlWlfaCXgN8vzASkbwIJeEQqtllKVr2ug,7505
-endee/index.py,sha256=0X_AU_4ROSQsklwTpo8iYjpEU3MUAslwCrpGGenyvAM,22751
-endee/schema.py,sha256=3Akwmtqv3oqr3-5iDXDYcOLGxY0zVSOUhoJhnJV4qVA,4707
-endee/utils.py,sha256=UGRMf_2DLDE9Ze-2HDoeAv97Mc6w8ABIX9M1MFOBwVc,1258
-endee-0.1.9.dist-info/licenses/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-endee-0.1.9.dist-info/METADATA,sha256=ATehjQEJ2m-ye-PjlnaGuelAu9z1x1SOSQxqjlSqvMQ,24335
-endee-0.1.9.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-endee-0.1.9.dist-info/top_level.txt,sha256=zOEvXIfzdm7vXJaVX_jq5OX3fTftKq14KzynxlAp8ZQ,6
-endee-0.1.9.dist-info/RECORD,,