dyff-schema 0.38.4__py3-none-any.whl → 0.38.6__py3-none-any.whl

dyff/schema/_version.py

@@ -1,2 +1,2 @@
-__version__ = version = "0.38.4"
-__version_tuple__ = version_tuple = (0, 38, 4)
+__version__ = version = "0.38.6"
+__version_tuple__ = version_tuple = (0, 38, 6)

dyff/schema/v0/r1/platform.py

@@ -2765,37 +2765,20 @@ class ChallengeSubmission(DyffEntity):
 class PipelineEvaluationRequest(EvaluationRequestBase):
     kind: Literal["PipelineEvaluationRequest"] = "PipelineEvaluationRequest"
 
-    @pydantic.model_serializer(mode="wrap")
-    def serialize_model(
-        self, handler: pydantic.SerializerFunctionWrapHandler
-    ) -> dict[str, object]:
-        serialized = handler(self)
-        serialized["kind"] = self.kind
-        return serialized
-
 
 class PipelineMeasurementRequest(AnalysisRequestBase):
     kind: Literal["PipelineMeasurementRequest"] = "PipelineMeasurementRequest"
 
-    @pydantic.model_serializer(mode="wrap")
-    def serialize_model(
-        self, handler: pydantic.SerializerFunctionWrapHandler
-    ) -> dict[str, object]:
-        serialized = handler(self)
-        serialized["kind"] = self.kind
-        return serialized
-
 
 class PipelineSafetyCaseRequest(AnalysisRequestBase):
     kind: Literal["PipelineSafetyCaseRequest"] = "PipelineSafetyCaseRequest"
 
-    @pydantic.model_serializer(mode="wrap")
-    def serialize_model(
-        self, handler: pydantic.SerializerFunctionWrapHandler
-    ) -> dict[str, object]:
-        serialized = handler(self)
-        serialized["kind"] = self.kind
-        return serialized
+
+PipelineNodeRequest = Union[
+    PipelineEvaluationRequest,
+    PipelineMeasurementRequest,
+    PipelineSafetyCaseRequest,
+]
 
 
 class PipelineNode(DyffSchemaBaseModel):
@@ -2810,12 +2793,7 @@ class PipelineNode(DyffSchemaBaseModel):
         description="The name of the node. Must be unique in the context of the pipeline."
     )
 
-    request: Union[
-        PipelineEvaluationRequest,
-        PipelineMeasurementRequest,
-        PipelineSafetyCaseRequest,
-    ] = pydantic.Field(
-        discriminator="kind",
+    request: PipelineNodeRequest = pydantic.Field(
         description="The request template that will be executed when this node"
         ' executes. You can use the syntax "$(node_name)" in request fields'
         " that reference another entity to indicate that the request depends"
@@ -2824,6 +2802,15 @@ class PipelineNode(DyffSchemaBaseModel):
         " dependency graph structure from these placeholders.",
     )
 
+    # The .kind field is dropped when we serialize nodes as part of a
+    # PipelineCreateRequest because of the default exclude_unset=True
+    # behavior for requests, so we add it back.
+    @pydantic.field_serializer("request")
+    def _serialize_request(self, request: PipelineNodeRequest, _info):
+        data = request.model_dump(mode=_info.mode)
+        data["kind"] = request.kind
+        return data
+
 
 class PipelineParameter(DyffSchemaBaseModel):
     """Declares a parameter that can be passed to the pipeline to customize its

dyff/schema/v0/r1/requests.py

@@ -398,6 +398,10 @@ class AuditQueryRequest(DyffEntityQueryRequest):
     name: Optional[str] = pydantic.Field(default=None)
 
 
+class ChallengeQueryRequest(DyffEntityQueryRequest):
+    pass
+
+
 class DatasetQueryRequest(DyffEntityQueryRequest):
     name: Optional[str] = pydantic.Field(default=None)
 
@@ -492,6 +496,7 @@ __all__ = [
     "AuditQueryRequest",
     "ChallengeContentEditRequest",
     "ChallengeCreateRequest",
+    "ChallengeQueryRequest",
     "ChallengeTaskCreateRequest",
     "ChallengeTaskRulesEditRequest",
     "ChallengeTeamCreateRequest",

{dyff_schema-0.38.4.dist-info → dyff_schema-0.38.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dyff-schema
-Version: 0.38.4
+Version: 0.38.6
 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.38.4.dist-info → dyff_schema-0.38.6.dist-info}/RECORD

@@ -1,5 +1,5 @@
 dyff/schema/__init__.py,sha256=w7OWDFuyGKd6xt_yllNtKzHahPgywrfU4Ue02psYaMA,2244
-dyff/schema/_version.py,sha256=kw__pY_ddHPtQSY-UfuqtSZD-1FO7lt7GDT52FWQGjA,80
+dyff/schema/_version.py,sha256=Kj9GP_7NUacasKg6ybcdwasvyMnqhCYPGnOrKFolAWQ,80
 dyff/schema/adapters.py,sha256=YMTHv_2VlLGFp-Kqwa6H51hjffHmk8gXjZilHysIF5Q,123
 dyff/schema/annotations.py,sha256=nE6Jk1PLqlShj8uqjE_EzZC9zYnTDW5AVtQcjysiK8M,10018
 dyff/schema/base.py,sha256=jvaNtsSZyFfsdUZTcY_U-yfLY5_GyrMxSXhON2R9XR0,119
@@ -29,8 +29,8 @@ dyff/schema/v0/r1/adapters.py,sha256=hpwCSW8lkMkUKCLe0zaMUDu-VS_caSxJvPsECEi_XRA
 dyff/schema/v0/r1/base.py,sha256=1VJXVLKldOq3aH2HdVgxXXk4DJTZEIiaTa4GzMKzziU,20228
 dyff/schema/v0/r1/commands.py,sha256=jqbEvDRLpcVGWXqlbniuSKg3UIjlrpKt6_0hiwUpPfQ,20153
 dyff/schema/v0/r1/oci.py,sha256=YjHDVBJ2IIxqijll70OK6pM-qT6pq8tvU7D3YB9vGM0,6700
-dyff/schema/v0/r1/platform.py,sha256=aBLDIkObSjJdlEt1EhU0UkfLE55ptUFWuHJWg2kMjMI,111593
-dyff/schema/v0/r1/requests.py,sha256=JMpdx9WBlUpePshacG85XMmlGPBkkaK5453VaK3kH9s,18321
+dyff/schema/v0/r1/platform.py,sha256=U5t-Giixi52_aWpDzZVrXfLSAmar1i2BoFA0mSstWtg,111220
+dyff/schema/v0/r1/requests.py,sha256=9fFUTv4Zgt5Dq2ueQRGjaBU-RxacNMrDZW40RMEwsIg,18414
 dyff/schema/v0/r1/responses.py,sha256=nxy7FPtfw2B_bljz5UGGuSE79HTkDQxKH56AJVmd4Qo,1287
 dyff/schema/v0/r1/test.py,sha256=X6dUyVd5svcPCI-PBMOAqEfK9jv3bRDvkQTJzwS96c0,10720
 dyff/schema/v0/r1/version.py,sha256=NONebgcv5Thsw_ymud6PacZdGjV6ndBrmLnap-obcpo,428
@@ -43,9 +43,9 @@ dyff/schema/v0/r1/dataset/text.py,sha256=MYG5seGODDryRSCy-g0Unh5dD0HCytmZ3FeElC-
 dyff/schema/v0/r1/dataset/vision.py,sha256=tJFF4dkhHX0UXTj1sPW-G22xTSI40gbYO465FuvmvAU,443
 dyff/schema/v0/r1/io/__init__.py,sha256=L5y8UhRnojerPYHumsxQJRcHCNz8Hj9NM8b47mewMNs,92
 dyff/schema/v0/r1/io/vllm.py,sha256=vWyLg-susbg0JDfv6VExBpgFdU2GHP2a14ChOdbckvs,5321
-dyff_schema-0.38.4.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-dyff_schema-0.38.4.dist-info/licenses/NOTICE,sha256=YONACu0s_Ui6jNi-wtEsVQbTU1JIkh8wvLH6d1-Ni_w,43
-dyff_schema-0.38.4.dist-info/METADATA,sha256=TgRDIF30usn5haI6HaLucU1aYEzxq7I-RPnui_2YCwc,3734
-dyff_schema-0.38.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-dyff_schema-0.38.4.dist-info/top_level.txt,sha256=9e3VVdeX73t_sUJOPQPCcGtYO1JhoErhHIi3WoWGcFI,5
-dyff_schema-0.38.4.dist-info/RECORD,,
+dyff_schema-0.38.6.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+dyff_schema-0.38.6.dist-info/licenses/NOTICE,sha256=YONACu0s_Ui6jNi-wtEsVQbTU1JIkh8wvLH6d1-Ni_w,43
+dyff_schema-0.38.6.dist-info/METADATA,sha256=f8TUmKB1WVBTwptV_BrVgMVovyNT8WTYX_Fyci1urEQ,3734
+dyff_schema-0.38.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+dyff_schema-0.38.6.dist-info/top_level.txt,sha256=9e3VVdeX73t_sUJOPQPCcGtYO1JhoErhHIi3WoWGcFI,5
+dyff_schema-0.38.6.dist-info/RECORD,,