dyff-schema 0.27.0__py3-none-any.whl → 0.27.2__py3-none-any.whl

dyff/schema/v0/r1/base.py

@@ -658,6 +658,29 @@ class DyffSchemaBaseModel(DyffBaseModel):
         return values
 
 
+class JsonMergePatchSemantics(DyffSchemaBaseModel):
+    """Explicit None values will be output as json 'null', and fields that are not set
+    explicitly are not output.
+
+    In JSON Merge Patch terms, None means "delete this field", and not setting a value
+    means "leave this field unchanged".
+    """
+
+    def dict(
+        self, *, by_alias: bool = True, exclude_unset=True, exclude_none=False, **kwargs
+    ) -> _ModelAsDict:
+        return super().dict(
+            by_alias=by_alias, exclude_unset=True, exclude_none=False, **kwargs
+        )
+
+    def json(
+        self, *, by_alias: bool = True, exclude_unset=True, exclude_none=False, **kwargs
+    ) -> str:
+        return super().json(
+            by_alias=by_alias, exclude_unset=True, exclude_none=False, **kwargs
+        )
+
+
 __all__ = [
     "DTYPE",
     "DType",
@@ -671,6 +694,7 @@ __all__ = [
     "Int16",
     "Int32",
     "Int64",
+    "JsonMergePatchSemantics",
     "Null",
     "UInt8",
     "UInt16",

dyff/schema/v0/r1/commands.py

@@ -7,13 +7,14 @@ These are used internally by the platform and users typically won't encounter th
 
 from __future__ import annotations
 
-from typing import Any, Literal, Optional, Union
+from typing import Literal, Optional, Union
 
 import pydantic
 
-from .base import DyffSchemaBaseModel, Null
+from .base import DyffSchemaBaseModel, JsonMergePatchSemantics, Null
 from .platform import (
     DyffEntityType,
+    EntityIdentifier,
     EntityKindLiteral,
     FamilyMember,
     FamilyMembers,
@@ -29,29 +30,6 @@ from .platform import (
 # ----------------------------------------------------------------------------
 
 
-# mypy gets confused because 'dict' is the name of a method in DyffBaseModel
-_ModelAsDict = dict[str, Any]
-
-
-class _JsonMergePatchSemantics(DyffSchemaBaseModel):
-    """Explicit None values will be output as json 'null', and fields that are not set
-    explicitly are not output.
-
-    In JSON Merge Patch terms, None means "delete this field", and not setting a value
-    means "leave this field unchanged".
-    """
-
-    def dict(self, *, by_alias: bool = True, **kwargs) -> _ModelAsDict:
-        return super().dict(
-            by_alias=by_alias, exclude_unset=True, exclude_none=False, **kwargs
-        )
-
-    def json(self, *, by_alias: bool = True, **kwargs) -> str:
-        return super().json(
-            by_alias=by_alias, exclude_unset=True, exclude_none=False, **kwargs
-        )
-
-
 # class _NoneMeansUndefined(DyffSchemaBaseModel):
 #     """Fields with the value None will not be emitted in the JSON output."""
 
@@ -64,23 +42,6 @@ class _JsonMergePatchSemantics(DyffSchemaBaseModel):
 #         return super().json(exclude_none=True, **kwargs)
 
 
-class EntityIdentifier(DyffSchemaBaseModel):
-    """Identifies a single entity."""
-
-    @staticmethod
-    def of(entity: DyffEntityType) -> EntityIdentifier:
-        """Create an identifier that identifies the given entity."""
-        return EntityIdentifier(kind=entity.kind, id=entity.id)
-
-    id: str = pydantic.Field(description="The .id of the entity.")
-    kind: Optional[EntityKindLiteral] = pydantic.Field(
-        default=None,
-        description="The .kind of the entity. This is optional because"
-        " sometimes you need to send a command without knowing the kind,"
-        " but it should be set whenever possible.",
-    )
-
-
 class FamilyIdentifier(EntityIdentifier):
     """Identifies a single Family entity."""
 
@@ -119,7 +80,7 @@ class CreateEntity(Command):
 # ----------------------------------------------------------------------------
 
 
-class EditEntityDocumentationPatch(_JsonMergePatchSemantics):
+class EditEntityDocumentationPatch(JsonMergePatchSemantics):
     """Same properties as DocumentationBase, but assigning None to a field is
     interpreted as a command to delete that field.
 
@@ -182,7 +143,7 @@ class EditEntityDocumentation(Command):
 # ----------------------------------------------------------------------------
 
 
-class EditEntityLabelsAttributes(_JsonMergePatchSemantics):
+class EditEntityLabelsAttributes(JsonMergePatchSemantics):
     """Attributes for the EditEntityLabels command."""
 
     labels: dict[LabelKeyType, Optional[Union[LabelValueType, Null]]] = pydantic.Field(
@@ -216,7 +177,7 @@ class EditEntityLabels(Command):
 # ----------------------------------------------------------------------------
 
 
-class EditFamilyMembersAttributes(_JsonMergePatchSemantics):
+class EditFamilyMembersAttributes(JsonMergePatchSemantics):
     """Attributes for the EditFamilyMembers command."""
 
     members: dict[TagNameType, Optional[Union[FamilyMember, Null]]] = pydantic.Field(
@@ -258,7 +219,7 @@ class ForgetEntity(Command):
 # ----------------------------------------------------------------------------
 
 
-class UpdateEntityStatusAttributes(_JsonMergePatchSemantics):
+class UpdateEntityStatusAttributes(JsonMergePatchSemantics):
     """Attributes for the UpdateEntityStatus command."""
 
     status: str = pydantic.Field(

dyff/schema/v0/r1/platform.py

@@ -291,6 +291,23 @@ class DyffModelWithID(DyffSchemaBaseModel):
     account: str = pydantic.Field(description="Account that owns the entity")
 
 
+class EntityIdentifier(DyffSchemaBaseModel):
+    """Identifies a single entity."""
+
+    @staticmethod
+    def of(entity: "DyffEntityType") -> "EntityIdentifier":
+        """Create an identifier that identifies the given entity."""
+        return EntityIdentifier(kind=entity.kind, id=entity.id)
+
+    id: str = pydantic.Field(description="The .id of the entity.")
+    kind: Optional[EntityKindLiteral] = pydantic.Field(
+        default=None,
+        description="The .kind of the entity. This is optional because"
+        " sometimes you need to send a command without knowing the kind,"
+        " but it should be set whenever possible.",
+    )
+
+
 def LabelKey() -> type[str]:
     return pydantic.constr(
         regex=_k8s_label_key_regex(), max_length=_k8s_label_key_maxlen()
@@ -682,24 +699,24 @@ class FamilyMemberKind(str, enum.Enum):
 
 
 class FamilyMemberBase(DyffSchemaBaseModel):
-    name: TagNameType = pydantic.Field(
-        description="An interpretable identifier for the member that is unique"
-        " in the context of the corresponding Family.",
-    )
-
-    resource: str = pydantic.Field(
+    entity: EntityIdentifier = pydantic.Field(
         description="ID of the resource this member references.",
     )
 
-    description: Optional[str] = pydantic.Field(
+    description: Optional[pydantic.constr(max_length=summary_maxlen())] = pydantic.Field(  # type: ignore
         default=None,
-        description="A short description of the member. Interpreted as Markdown."
-        " This should include information about how this version of the resource"
+        description="A short description of the member."
+        " This should describe how this version of the resource"
         " is different from other versions.",
     )
 
 
 class FamilyMember(FamilyMemberBase):
+    name: TagNameType = pydantic.Field(
+        description="An interpretable identifier for the member that is unique"
+        " in the context of the corresponding Family.",
+    )
+
     family: str = pydantic.Field(
         description="Identifier of the Family containing this tag."
     )
@@ -725,6 +742,12 @@ class FamilyBase(DyffSchemaBaseModel):
 class Family(DyffEntity, FamilyBase, FamilyMembers):
     kind: Literal["Family"] = "Family"
 
+    def dependencies(self) -> list[str]:
+        return []
+
+    def resource_allocation(self) -> Optional[ResourceAllocation]:
+        return None
+
 
 class RevisionMetadata(DyffSchemaBaseModel):
     previousRevision: Optional[str] = pydantic.Field(

dyff/schema/v0/r1/requests.py

@@ -21,7 +21,7 @@ from typing import Any, Literal, Optional, Union
 import pydantic
 
 from . import commands
-from .base import DyffBaseModel
+from .base import DyffBaseModel, JsonMergePatchSemantics, Null
 from .platform import (
     AnalysisBase,
     AnalysisScope,
@@ -31,12 +31,14 @@ from .platform import (
     DocumentationBase,
     EvaluationBase,
     FamilyBase,
+    FamilyMemberBase,
     InferenceServiceBase,
     InferenceSessionBase,
     MethodBase,
     ModelSpec,
     ModuleBase,
     ReportBase,
+    TagNameType,
     summary_maxlen,
     title_maxlen,
 )
@@ -242,8 +244,12 @@ class DocumentationEditRequest(
     pass
 
 
-class FamilyMembersEditRequest(DyffRequestBase, commands.EditFamilyMembersAttributes):
-    pass
+class FamilyMembersEditRequest(DyffRequestBase, JsonMergePatchSemantics):
+    members: dict[TagNameType, Optional[Union[FamilyMemberBase, Null]]] = (
+        pydantic.Field(
+            description="Mapping of names to IDs of member resources.",
+        )
+    )
 
 
 class LabelsEditRequest(DyffRequestBase, commands.EditEntityLabelsAttributes):

{dyff_schema-0.27.0.dist-info → dyff_schema-0.27.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: dyff-schema
-Version: 0.27.0
+Version: 0.27.2
 Summary: Data models for the Dyff AI auditing platform.
 Author-email: Digital Safety Research Institute <contact@dsri.org>
 License: Apache-2.0

{dyff_schema-0.27.0.dist-info → dyff_schema-0.27.2.dist-info}/RECORD

@@ -24,10 +24,10 @@ dyff/schema/io/vllm.py,sha256=2q05M_-lTzq9oywKXHPPpCFCSDVCSsRQqtmERzWTtio,123
 dyff/schema/v0/__init__.py,sha256=L5y8UhRnojerPYHumsxQJRcHCNz8Hj9NM8b47mewMNs,92
 dyff/schema/v0/r1/__init__.py,sha256=L5y8UhRnojerPYHumsxQJRcHCNz8Hj9NM8b47mewMNs,92
 dyff/schema/v0/r1/adapters.py,sha256=dmQS2ecgDX4ZvTMOW-6NzV_Oq_UpaiyFd7QnSNoOnK8,33057
-dyff/schema/v0/r1/base.py,sha256=fSWeuiArzt7MOdLGqQPPNW4IPye-6EBrHSikFllQXxQ,20273
-dyff/schema/v0/r1/commands.py,sha256=7kP8G4amtvt3pRd01PHusV3EYGrYX82pV2GFe4WcitE,9752
-dyff/schema/v0/r1/platform.py,sha256=-HbsM-cHmm7HiLiVx-1vPQ4S8SqZqHK3UXwuCBtJi-g,80753
-dyff/schema/v0/r1/requests.py,sha256=ZoaT_g7dPKIWtibWJFwlbuMJ2dXqIkS8h9mJ-7VnLr4,15789
+dyff/schema/v0/r1/base.py,sha256=Rry7GKdQyCLNXskB1CxqVlhVsgM8pw6a4xy3BFk_lRc,21095
+dyff/schema/v0/r1/commands.py,sha256=eBXoDxmRZ9s2Idj1srmIaLPCn6QVbWXwpbNDHaJZDco,8354
+dyff/schema/v0/r1/platform.py,sha256=mXI32MyJNVAYjT3xQtwlaysU6u_K6p9bb7phErayCq0,81574
+dyff/schema/v0/r1/requests.py,sha256=AIBzfOnDEMVh1sCOCM6oKMVwZTCW3v9f5BVKB2s3-X0,16025
 dyff/schema/v0/r1/test.py,sha256=X6dUyVd5svcPCI-PBMOAqEfK9jv3bRDvkQTJzwS96c0,10720
 dyff/schema/v0/r1/version.py,sha256=isKAGuGxsdru8vDaYmI4YiZdJOu_wNxXK7u6QzD6FE4,392
 dyff/schema/v0/r1/dataset/__init__.py,sha256=LbVlkO2asyGYBKk2z49xjJYTM-pu9y9e4eQDXgTDLnM,2553
@@ -39,9 +39,9 @@ dyff/schema/v0/r1/dataset/text.py,sha256=nLIn91Zlt0tNdXUklSgjJ-kEDxoPX32ISLkiv2D
 dyff/schema/v0/r1/dataset/vision.py,sha256=aIe0fbfM_g3DsrDTdg2K803YKLjZBpurM_VJcJFuZLc,369
 dyff/schema/v0/r1/io/__init__.py,sha256=L5y8UhRnojerPYHumsxQJRcHCNz8Hj9NM8b47mewMNs,92
 dyff/schema/v0/r1/io/vllm.py,sha256=CUE9y8KthtUI7sD49S875rDmPvKotSXVIRaBS79aBZs,5320
-dyff_schema-0.27.0.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-dyff_schema-0.27.0.dist-info/METADATA,sha256=sCwQZWjkkkWNhuiV0OJ9FuQa2ddUVHsVvZqXOjhRKKs,3482
-dyff_schema-0.27.0.dist-info/NOTICE,sha256=YONACu0s_Ui6jNi-wtEsVQbTU1JIkh8wvLH6d1-Ni_w,43
-dyff_schema-0.27.0.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
-dyff_schema-0.27.0.dist-info/top_level.txt,sha256=9e3VVdeX73t_sUJOPQPCcGtYO1JhoErhHIi3WoWGcFI,5
-dyff_schema-0.27.0.dist-info/RECORD,,
+dyff_schema-0.27.2.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+dyff_schema-0.27.2.dist-info/METADATA,sha256=ajCQ1ZTd0A7GA8g9h6Y-sx0SFHLJHDBqlr9ihKD8Qeg,3482
+dyff_schema-0.27.2.dist-info/NOTICE,sha256=YONACu0s_Ui6jNi-wtEsVQbTU1JIkh8wvLH6d1-Ni_w,43
+dyff_schema-0.27.2.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
+dyff_schema-0.27.2.dist-info/top_level.txt,sha256=9e3VVdeX73t_sUJOPQPCcGtYO1JhoErhHIi3WoWGcFI,5
+dyff_schema-0.27.2.dist-info/RECORD,,