nucliadb-models 6.6.1.post4617__py3-none-any.whl → 6.6.1.post4623__py3-none-any.whl

nucliadb_models/common.py

@@ -88,7 +88,7 @@ class FieldID(BaseModel):
 class File(BaseModel):
     filename: Optional[str] = None
     content_type: str = "application/octet-stream"
-    payload: Optional[str] = Field(None, description="Base64 encoded file content")
+    payload: Optional[str] = Field(default=None, description="Base64 encoded file content")
     md5: Optional[str] = None
     # These are to be used for external files
     uri: Optional[str] = None
@@ -176,7 +176,7 @@ class CloudLink(BaseModel):
         return DOWNLOAD_URI.format(**url_params).rstrip("/")
 
     @field_serializer("uri")
-    def serialize_uri(uri):
+    def serialize_uri(self, uri: str):
         return CloudLink.format_reader_download_uri(uri)
 
 

nucliadb_models/configuration.py

@@ -14,7 +14,7 @@
 #
 
 import warnings
-from typing import Annotated, Literal, Optional, Union
+from typing import Annotated, Any, Literal, Optional, Union
 
 from pydantic import BaseModel, Field, create_model
 
@@ -38,16 +38,14 @@ class KBConfiguration(BaseModel):
 #
 # Search configurations
 #
+def _model_fields(model: type[BaseModel], skip: list[str]) -> dict[str, Any]:
+    return {
+        name: (field.annotation, field) for name, field in model.model_fields.items() if name not in skip
+    }
+
 
 # FindConfig is a FindConfig without `search_configuration`
-FindConfig = create_model(
-    "FindConfig",
-    **{
-        name: (field.annotation, field)
-        for name, field in FindRequest.model_fields.items()
-        if name not in ("search_configuration")
-    },
-)  # type: ignore[call-overload]
+FindConfig = create_model("FindConfig", **_model_fields(FindRequest, skip=["search_configuration"]))
 
 
 class FindSearchConfiguration(BaseModel):
@@ -58,13 +56,9 @@ class FindSearchConfiguration(BaseModel):
 # AskConfig is an AskRequest where `query` is not mandatory and without `search_configuration`
 AskConfig = create_model(
     "AskConfig",
-    **{
-        name: (field.annotation, field)
-        for name, field in AskRequest.model_fields.items()
-        if name not in ("query", "search_configuration")
-    },
+    **_model_fields(AskRequest, skip=["query", "search_configuration"]),
     query=(Optional[str], None),
-)  # type: ignore[call-overload]
+)
 
 
 class AskSearchConfiguration(BaseModel):

nucliadb_models/conversation.py

@@ -18,7 +18,7 @@ from typing import List, Optional
 
 from pydantic import BaseModel, Field, field_validator
 
-from nucliadb_models import CloudLink, FieldRef, FileB64
+from nucliadb_models.common import CloudLink, FieldRef, FileB64
 from nucliadb_models.utils import DateTime
 
 # Shared classes

nucliadb_models/file.py

@@ -17,7 +17,7 @@ from typing import Optional
 
 from pydantic import BaseModel, Field
 
-from nucliadb_models import CloudLink, File
+from nucliadb_models.common import CloudLink, File
 
 # Shared classes
 # - Nothing to see here

nucliadb_models/filters.py

@@ -66,7 +66,7 @@ class Not(BaseModel, Generic[F], extra="forbid"):
 
 
 class FilterProp(BaseModel):
-    prop: str
+    prop: Any
 
     @model_validator(mode="after")
     def set_discriminator(self) -> Self:

nucliadb_models/graph/requests.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 #
 from enum import Enum
-from typing import Annotated, Literal, Optional, Union
+from typing import Annotated, Any, Literal, Optional, Union
 
 from pydantic import BaseModel, Discriminator, Field, Tag, model_validator
 from typing_extensions import Self
@@ -26,7 +26,7 @@ from nucliadb_models.security import RequestSecurity
 
 
 class GraphProp(BaseModel):
-    prop: str
+    prop: Any
 
     @model_validator(mode="after")
     def set_discriminator(self) -> Self:

nucliadb_models/metadata.py

@@ -73,7 +73,7 @@ class Relation(BaseModel):
     label: Optional[str] = None
     metadata: Optional[RelationMetadata] = None
 
-    from_: Optional[RelationEntity] = Field(None, alias="from")
+    from_: Optional[RelationEntity] = Field(default=None, alias="from")
     to: RelationEntity
 
     @model_validator(mode="after")

nucliadb_models/notifications.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 #
 from enum import Enum
+from typing import Any
 
 from pydantic import BaseModel, Field
 
@@ -29,7 +30,7 @@ class Notification(BaseModel):
         title="Notification Type",
         description="Type of notification.",
     )
-    data: BaseModel = Field(
+    data: Any = Field(
         ...,
         title="Notification Data",
         description="Notification data.",

nucliadb_models/search.py

@@ -1100,7 +1100,7 @@ class ImageRagStrategyName:
 
 
 class RagStrategy(BaseModel):
-    name: str
+    name: Any
 
     @model_validator(mode="after")
     def set_discriminator(self) -> Self:
@@ -1110,7 +1110,7 @@ class RagStrategy(BaseModel):
 
 
 class ImageRagStrategy(BaseModel):
-    name: str
+    name: Any
 
     @model_validator(mode="after")
     def set_discriminator(self) -> Self:

{nucliadb_models-6.6.1.post4617.dist-info → nucliadb_models-6.6.1.post4623.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nucliadb_models
-Version: 6.6.1.post4617
+Version: 6.6.1.post4623
 Author-email: Nuclia <nucliadb@nuclia.com>
 License-Expression: Apache-2.0
 Project-URL: Homepage, https://nuclia.com

{nucliadb_models-6.6.1.post4617.dist-info → nucliadb_models-6.6.1.post4623.dist-info}/RECORD

@@ -1,22 +1,22 @@
 nucliadb_models/__init__.py,sha256=3y8-htogKuCZcbhaUZdSjTeEjUSeec9aRWyL8AlKCyM,1077
-nucliadb_models/common.py,sha256=E5rYsahuRKgUX3ZASIFHjR6LPBfezh453JbfJ6hbckI,7885
-nucliadb_models/configuration.py,sha256=V1d4hhs_1r-_ik7uBpqBHYrpzpMZYI4QeieWpHc8iOM,2508
+nucliadb_models/common.py,sha256=YW84w1NAQARObs2nXw6YBgdxQJeVCmTZZr5lSqj-IdQ,7904
+nucliadb_models/configuration.py,sha256=aTV5mBwYFlwiV1_nWyVAXaCh7F6lDVTVh28Xfwy8ox8,2448
 nucliadb_models/content_types.py,sha256=eMlBhWwzfYJFlErcWsNCvBdypbv8J9eC-MXw727QiBE,3430
-nucliadb_models/conversation.py,sha256=bgePP_aLvvnLV9VK4x90Aazlkdc4ajJKFVDYEa9sOJY,3383
+nucliadb_models/conversation.py,sha256=2igXbzF7kfnOd6qjY-b0AZSAjI7O3sLpL0Cp-uSfJvA,3390
 nucliadb_models/entities.py,sha256=i-7Y8qmFRRTih5zw0ajv1U_iiXexe66M3TK8hUikQZk,2356
 nucliadb_models/export_import.py,sha256=mNm9IArOLnC6TLupkwqVFhxD5d08mpIVOVFneECv8UA,1073
 nucliadb_models/external_index_providers.py,sha256=IIKjJjLixWQC1zrbzam2FDcAo5UUxShZfueZSxqZu8Y,1535
 nucliadb_models/extracted.py,sha256=Zh79jrOcqedVYc36qM4D5qrOn5RutTuJAHusEeyIDiU,6245
-nucliadb_models/file.py,sha256=U66oSmywTWqE48mpNv9KG4X--XZtNI9i4Kl3EcYSLLM,1740
-nucliadb_models/filters.py,sha256=1Y1HwO6KBxIgQNg9MyJ3aJKnK_p6d1Y5-ZlUbX1WbvE,14759
+nucliadb_models/file.py,sha256=ZfxnXeigmMkoZ5ae22g-JvlYf8UBz4-pkx1t7aOL8Rs,1747
+nucliadb_models/filters.py,sha256=NQI2-4AFzzJuZy8NeY3jXlTbbU5wxiwMCP-5DrD-7lE,14759
 nucliadb_models/labels.py,sha256=9zqRgkpZuX3kUPwsTTgCH7JyOWK7dM5pwyuHJR86YdU,3949
 nucliadb_models/link.py,sha256=cfMOwaKDnaYSMxD5QClBu2Ab1orGfuTurFqNozp3KFk,2010
-nucliadb_models/metadata.py,sha256=7rCIpeWjvqDk4ytIVP3azR9Qmm5PTGVI1BNkNLXfHVo,8348
-nucliadb_models/notifications.py,sha256=3w1HeX9F8nuA7WupHdpXIksX7d0bHO9ofmemYp4RcmM,4019
+nucliadb_models/metadata.py,sha256=MFVYnpXMBoY4ylMg029o7yDKGxhK7NB0c0FSshzJHm4,8356
+nucliadb_models/notifications.py,sha256=mna8-AoD_29Wds0Thl0AF0zpERnJmYGLZX1w1fUopMY,4036
 nucliadb_models/processing.py,sha256=nhKuHQjqCdb9zJVkYGPTLub23tK9e_lwL5OCDVymZjY,719
 nucliadb_models/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nucliadb_models/resource.py,sha256=FkhwmbjoqQBGyKQIqa40WK2Vq3wtryZeIVWmJ4b-84g,9003
-nucliadb_models/search.py,sha256=6R-HIGANbwl_fm21UI_IGUw8L8GS7TNY678lXnLtgqk,91648
+nucliadb_models/search.py,sha256=UtENf6-osZ6TvV7EzqrBN67GfZtrBhCZEo3qBi15ejs,91648
 nucliadb_models/security.py,sha256=opxaDLfvk3aU0sjesK0jGrYLx5h4YCwlKKN0moYs_ig,1150
 nucliadb_models/synonyms.py,sha256=afbaVqSQSxGLwi2PusVaLSRpkOtA5AZmWOKd1f4nl2E,690
 nucliadb_models/text.py,sha256=kY2ub7AaGm-4vNaLX3Ju2VvRw-eKZ2LRdM9z7XCNaG0,2898
@@ -27,12 +27,12 @@ nucliadb_models/vectorsets.py,sha256=XAgg9DfdfLYpfLh9OepJ_KPH0_RqRQNpVZJr74UnNh0
 nucliadb_models/writer.py,sha256=diwrarp6DxjSUoRmdEljZb68z_ghNvpOgPUGZeKg328,8220
 nucliadb_models/agents/ingestion.py,sha256=W9cJ0dQT_1vPcjeJ4_Fjb8DylnhQ6qqZrY4v8x1RqUs,3093
 nucliadb_models/graph/__init__.py,sha256=X538kZPZnndmQeEtnzzPv1hYVGUTDe9U1O7UmAqqxXU,645
-nucliadb_models/graph/requests.py,sha256=qkIOTL1lP6tstvnbrqIt6IFhlOPZbjShds_HbSgAK-8,7164
+nucliadb_models/graph/requests.py,sha256=ppQ7cOnybvrw1wGC7qDps-182PfmicWU6-4vLRfK16w,7169
 nucliadb_models/graph/responses.py,sha256=Sdq8OgFAL1YT-1lJyLLrkqcScvj7YTEqAUwQ-kFAk9M,1399
 nucliadb_models/internal/__init__.py,sha256=zG33bUz1rHFPtvqQPWn4rDwBJt3FJodGuQYD45quiQg,583
 nucliadb_models/internal/predict.py,sha256=Pnx6MmLfK65eExe1XnVxqmSlvMwdowewwks9BOEoqMw,2029
 nucliadb_models/internal/shards.py,sha256=__y1OZtWGiNcPQEWfSFOj8yw458WGi7mM4vZe0K-L1Y,1691
-nucliadb_models-6.6.1.post4617.dist-info/METADATA,sha256=VrP9rFXhlLxxRMj9XLNNcf1dqf7aLKiji4v8YOEo6aw,794
-nucliadb_models-6.6.1.post4617.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nucliadb_models-6.6.1.post4617.dist-info/top_level.txt,sha256=UrY1I8oeovIRwkXLYplssTrxQdUjhSEFDFbnwaIV3tA,16
-nucliadb_models-6.6.1.post4617.dist-info/RECORD,,
+nucliadb_models-6.6.1.post4623.dist-info/METADATA,sha256=aECtqNWM9uJdW-wfJyqyPO2fEwCkq60IWmLNFgLqj34,794
+nucliadb_models-6.6.1.post4623.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nucliadb_models-6.6.1.post4623.dist-info/top_level.txt,sha256=UrY1I8oeovIRwkXLYplssTrxQdUjhSEFDFbnwaIV3tA,16
+nucliadb_models-6.6.1.post4623.dist-info/RECORD,,