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

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

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

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

{dyff_schema-0.38.2 → dyff_schema-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'
@@ -2832,6 +2853,7 @@ class PipelineBase(DyffSchemaBaseModel):
         for k, v in nodes.items():
             if k != v.name:
                 raise ValueError(f"nodes[{k}]: dict key must match value.name")
+        return nodes
 
 
 class Pipeline(DyffEntity, PipelineBase):

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

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

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