dyff-schema 0.38.3__py3-none-any.whl → 0.38.5__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.5"
+__version_tuple__ = version_tuple = (0, 38, 5)

dyff/schema/v0/r1/platform.py

@@ -2774,7 +2774,7 @@ class PipelineSafetyCaseRequest(AnalysisRequestBase):
     kind: Literal["PipelineSafetyCaseRequest"] = "PipelineSafetyCaseRequest"
 
 
-PipelineNodeRequest: TypeAlias = Union[
+PipelineNodeRequest = Union[
     PipelineEvaluationRequest,
     PipelineMeasurementRequest,
     PipelineSafetyCaseRequest,
@@ -2793,8 +2793,7 @@ 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]
-        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"
@@ -2803,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.3.dist-info → dyff_schema-0.38.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dyff-schema
-Version: 0.38.3
+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.3.dist-info → dyff_schema-0.38.5.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=dF25OI_Y4HFdLqtKnsCR-dWZ4t-wHRU8ZvxHTBt0VjE,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=U5t-Giixi52_aWpDzZVrXfLSAmar1i2BoFA0mSstWtg,111220
 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.5.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+dyff_schema-0.38.5.dist-info/licenses/NOTICE,sha256=YONACu0s_Ui6jNi-wtEsVQbTU1JIkh8wvLH6d1-Ni_w,43
+dyff_schema-0.38.5.dist-info/METADATA,sha256=PUJIvQg6-yRPp-gTsD-U8cIYAUDC_DzhMwphUdthxg0,3734
+dyff_schema-0.38.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+dyff_schema-0.38.5.dist-info/top_level.txt,sha256=9e3VVdeX73t_sUJOPQPCcGtYO1JhoErhHIi3WoWGcFI,5
+dyff_schema-0.38.5.dist-info/RECORD,,