nucliadb-utils 5.0.0.post809__py3-none-any.whl → 5.0.0.post821__py3-none-any.whl

nucliadb_utils/aiopynecone/client.py

@@ -35,6 +35,7 @@ from nucliadb_utils.aiopynecone.exceptions import (
     raise_for_status,
 )
 from nucliadb_utils.aiopynecone.models import (
+    CreateIndexRequest,
     CreateIndexResponse,
     ListResponse,
     QueryResponse,
@@ -100,14 +101,16 @@ class ControlPlane:
         Returns:
         - The index host to be used for data plane operations.
         """
-        payload = {
-            "name": name,
-            "dimension": dimension,
-            "metric": metric,
-            "spec": {"serverless": {"cloud": "aws", "region": "us-east-1"}},
-        }
+        payload = CreateIndexRequest(
+            name=name,
+            dimension=dimension,
+            metric=metric,
+            spec={"serverless": {"cloud": "aws", "region": "us-east-1"}},
+        )
         headers = {"Api-Key": self.api_key}
-        http_response = await self.http_session.post("/indexes", json=payload, headers=headers)
+        http_response = await self.http_session.post(
+            "/indexes", json=payload.model_dump(), headers=headers
+        )
         raise_for_status("create_index", http_response)
         response = CreateIndexResponse.model_validate(http_response.json())
         return response.host

nucliadb_utils/aiopynecone/exceptions.py

@@ -70,10 +70,11 @@ def raise_for_status(operation: str, response: httpx.Response):
         details = None
         try:
             resp_json = response.json()
-            code = resp_json.get("code")
-            message = resp_json.get("message")
-            details = resp_json.get("details")
-        except Exception:
+            error = resp_json.get("error") or {}
+            code = error.get("code")
+            message = error.get("message")
+            details = error.get("details")
+        except Exception:  # pragma: no cover
             message = response.text
         if response.status_code == 429:
             raise PineconeRateLimitError(

nucliadb_utils/aiopynecone/models.py

@@ -20,14 +20,47 @@
 import json
 from typing import Any, Optional
 
+import pydantic
 from pydantic import BaseModel, Field, field_validator
+from typing_extensions import Annotated
 
 KILO_BYTE = 1024
 MAX_METADATA_SIZE = 40 * KILO_BYTE
+MAX_INDEX_NAME_LENGTH = 45
 
 
 # Requests
 
+IndexNamePattern = r"^[a-z0-9-]+$"
+
+
+def validate_index_name(value, handler, info):
+    try:
+        return handler(value)
+    except pydantic.ValidationError as e:
+        if any(x["type"] == "string_pattern_mismatch" for x in e.errors()):
+            raise ValueError(
+                f"Invalid field_id: '{value}'. Pinecone index names must be a string with only "
+                "lowercase letters, numbers and dashes."
+            )
+        else:
+            raise e
+
+
+IndexNameStr = Annotated[
+    str,
+    pydantic.StringConstraints(pattern=IndexNamePattern),
+    pydantic.StringConstraints(min_length=1, max_length=MAX_INDEX_NAME_LENGTH),
+    pydantic.WrapValidator(validate_index_name),
+]
+
+
+class CreateIndexRequest(BaseModel):
+    name: IndexNameStr
+    dimension: int
+    metric: str
+    spec: dict[str, Any] = {}
+
 
 class Vector(BaseModel):
     id: str = Field(max_length=512)

{nucliadb_utils-5.0.0.post809.dist-info → nucliadb_utils-5.0.0.post821.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nucliadb_utils
-Version: 5.0.0.post809
+Version: 5.0.0.post821
 Home-page: https://nuclia.com
 License: BSD
 Classifier: Development Status :: 4 - Beta
@@ -23,8 +23,8 @@ Requires-Dist: PyNaCl
 Requires-Dist: pyjwt >=2.4.0
 Requires-Dist: memorylru >=1.1.2
 Requires-Dist: mrflagly
-Requires-Dist: nucliadb-protos >=5.0.0.post809
-Requires-Dist: nucliadb-telemetry >=5.0.0.post809
+Requires-Dist: nucliadb-protos >=5.0.0.post821
+Requires-Dist: nucliadb-telemetry >=5.0.0.post821
 Provides-Extra: cache
 Requires-Dist: redis >=4.3.4 ; extra == 'cache'
 Requires-Dist: orjson >=3.6.7 ; extra == 'cache'

{nucliadb_utils-5.0.0.post809.dist-info → nucliadb_utils-5.0.0.post821.dist-info}/RECORD

@@ -18,9 +18,9 @@ nucliadb_utils/store.py,sha256=kQ35HemE0v4_Qg6xVqNIJi8vSFAYQtwI3rDtMsNy62Y,890
 nucliadb_utils/transaction.py,sha256=mwcI3aIHAvU5KOGqd_Uz_d1XQzXhk_-NWY8NqU1lfb0,7307
 nucliadb_utils/utilities.py,sha256=oz3tEODG2g3todnyvA-nW1Ou6xXDveL_tMKTDGdWXM4,15287
 nucliadb_utils/aiopynecone/__init__.py,sha256=cp15ZcFnHvpcu_5-aK2A4uUyvuZVV_MJn4bIXMa20ks,835
-nucliadb_utils/aiopynecone/client.py,sha256=T0S6aoVefnjyKgJAGzBJGnrhBDlvSAj5DJBqdGRuzZw,18407
-nucliadb_utils/aiopynecone/exceptions.py,sha256=JJ-Ui9TSoHRe8WfkHR04S-t7NsPCy1xxluPH5R7fDl8,2816
-nucliadb_utils/aiopynecone/models.py,sha256=DVlCVrinHAaDxuii3fzdciFn4dqS5HTTyBavnoDlR2U,2024
+nucliadb_utils/aiopynecone/client.py,sha256=wUQIUZHKvhMhmLVfwrOF_nMBcf9l-4mXCvlSI0l0H24,18472
+nucliadb_utils/aiopynecone/exceptions.py,sha256=hFhq-UEY4slqNWjObXr_LPnRf_AQ1vpcG4SF2XRFd1E,2873
+nucliadb_utils/aiopynecone/models.py,sha256=sEmifzQ6rvqIB8nbkJbh8-hrCW4j8J9lJ_xLRaTAqro,2934
 nucliadb_utils/audit/__init__.py,sha256=cp15ZcFnHvpcu_5-aK2A4uUyvuZVV_MJn4bIXMa20ks,835
 nucliadb_utils/audit/audit.py,sha256=dn5ZnCVQUlCcvdjzaORghbrjk9QgVGrtkfIftq30Bp8,2819
 nucliadb_utils/audit/basic.py,sha256=NViey6mKbCXqRTLDBX2xNTcCg9I-2e4oB2xkekuhDvM,3392
@@ -64,8 +64,8 @@ nucliadb_utils/tests/indexing.py,sha256=YW2QhkhO9Q_8A4kKWJaWSvXvyQ_AiAwY1VylcfVQ
 nucliadb_utils/tests/local.py,sha256=c3gZJJWmvOftruJkIQIwB3q_hh3uxEhqGIAVWim1Bbk,1343
 nucliadb_utils/tests/nats.py,sha256=Tosonm9A9cusImyji80G4pgdXEHNVPaCLT5TbFK_ra0,7543
 nucliadb_utils/tests/s3.py,sha256=YB8QqDaBXxyhHonEHmeBbRRDmvB7sTOaKBSi8KBGokg,2330
-nucliadb_utils-5.0.0.post809.dist-info/METADATA,sha256=j1oedwxyH4rR_jpUuyUgFWnAcNImYZtGXSGvaIiXtnE,2073
-nucliadb_utils-5.0.0.post809.dist-info/WHEEL,sha256=Z4pYXqR_rTB7OWNDYFOm1qRk0RX6GFP2o8LgvP453Hk,91
-nucliadb_utils-5.0.0.post809.dist-info/top_level.txt,sha256=fE3vJtALTfgh7bcAWcNhcfXkNPp_eVVpbKK-2IYua3E,15
-nucliadb_utils-5.0.0.post809.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-nucliadb_utils-5.0.0.post809.dist-info/RECORD,,
+nucliadb_utils-5.0.0.post821.dist-info/METADATA,sha256=54y8hOk8DjtRFOBXlvQnqvcvOOgLvdV3LKexpyS-iMM,2073
+nucliadb_utils-5.0.0.post821.dist-info/WHEEL,sha256=Z4pYXqR_rTB7OWNDYFOm1qRk0RX6GFP2o8LgvP453Hk,91
+nucliadb_utils-5.0.0.post821.dist-info/top_level.txt,sha256=fE3vJtALTfgh7bcAWcNhcfXkNPp_eVVpbKK-2IYua3E,15
+nucliadb_utils-5.0.0.post821.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+nucliadb_utils-5.0.0.post821.dist-info/RECORD,,