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

dyff/schema/_version.py

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

dyff/schema/v0/r1/platform.py

@@ -2765,20 +2765,37 @@ 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"
 
-
-PipelineNodeRequest: TypeAlias = Union[
-    PipelineEvaluationRequest,
-    PipelineMeasurementRequest,
-    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
 
 
 class PipelineNode(DyffSchemaBaseModel):
@@ -2793,7 +2810,11 @@ class PipelineNode(DyffSchemaBaseModel):
         description="The name of the node. Must be unique in the context of the pipeline."
     )
 
-    request: PipelineNodeRequest = pydantic.Field(  # type: ignore[valid-type]
+    request: Union[
+        PipelineEvaluationRequest,
+        PipelineMeasurementRequest,
+        PipelineSafetyCaseRequest,
+    ] = pydantic.Field(
         discriminator="kind",
         description="The request template that will be executed when this node"
         ' executes. You can use the syntax "$(node_name)" in request fields'

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

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

@@ -1,5 +1,5 @@
 dyff/schema/__init__.py,sha256=w7OWDFuyGKd6xt_yllNtKzHahPgywrfU4Ue02psYaMA,2244
-dyff/schema/_version.py,sha256=-CCWFr62vN0w7SrCBH-IeTHMoX8ScRHffpLPQafa5pY,80
+dyff/schema/_version.py,sha256=kw__pY_ddHPtQSY-UfuqtSZD-1FO7lt7GDT52FWQGjA,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,7 +29,7 @@ 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=bQO3gQ6G-7Cl_FHHNLW1ozAG-ShYPOTIf-NfB6uvnps,110880
+dyff/schema/v0/r1/platform.py,sha256=aBLDIkObSjJdlEt1EhU0UkfLE55ptUFWuHJWg2kMjMI,111593
 dyff/schema/v0/r1/requests.py,sha256=JMpdx9WBlUpePshacG85XMmlGPBkkaK5453VaK3kH9s,18321
 dyff/schema/v0/r1/responses.py,sha256=nxy7FPtfw2B_bljz5UGGuSE79HTkDQxKH56AJVmd4Qo,1287
 dyff/schema/v0/r1/test.py,sha256=X6dUyVd5svcPCI-PBMOAqEfK9jv3bRDvkQTJzwS96c0,10720
@@ -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.3.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-dyff_schema-0.38.3.dist-info/licenses/NOTICE,sha256=YONACu0s_Ui6jNi-wtEsVQbTU1JIkh8wvLH6d1-Ni_w,43
-dyff_schema-0.38.3.dist-info/METADATA,sha256=LRDYIEaUA4AMUGCTGCLb1ABwrCh6rT28AU8ywiYU60o,3734
-dyff_schema-0.38.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-dyff_schema-0.38.3.dist-info/top_level.txt,sha256=9e3VVdeX73t_sUJOPQPCcGtYO1JhoErhHIi3WoWGcFI,5
-dyff_schema-0.38.3.dist-info/RECORD,,
+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,,