dyff-schema 0.38.4__tar.gz → 0.38.5__tar.gz

{dyff_schema-0.38.4/dyff_schema.egg-info → dyff_schema-0.38.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dyff-schema
-Version: 0.38.4
+Version: 0.38.5
 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.5/dyff/schema/_version.py

@@ -0,0 +1,2 @@
+__version__ = version = "0.38.5"
+__version_tuple__ = version_tuple = (0, 38, 5)

{dyff_schema-0.38.4 → dyff_schema-0.38.5}/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-0.38.4 → dyff_schema-0.38.5/dyff_schema.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dyff-schema
-Version: 0.38.4
+Version: 0.38.5
 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/dyff/schema/_version.py

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