dyff-schema 0.38.0__py3-none-any.whl → 0.38.2__py3-none-any.whl

dyff/schema/_version.py

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

dyff/schema/v0/r1/platform.py

@@ -1732,8 +1732,6 @@ class EvaluationRequestBase(EvaluationBase):
     """A description of how to run an InferenceService on a Dataset to obtain a set of
     evaluation results."""
 
-    kind: Literal["EvaluationRequest"] = "EvaluationRequest"
-
     inferenceSession: Optional[EvaluationInferenceSessionRequest] = pydantic.Field(
         default=None,
         description="Specification of the InferenceSession that will perform inference for the evaluation.",
@@ -2255,8 +2253,6 @@ class AnalysisBase(DyffSchemaBaseModel):
 
 
 class AnalysisRequestBase(AnalysisBase):
-    kind: Literal["AnalysisRequest"] = "AnalysisRequest"
-
     method: EntityIDField = pydantic.Field(description="Method ID")
 
 
@@ -2765,13 +2761,34 @@ class ChallengeSubmission(DyffEntity):
 # ---------------------------------------------------------------------------
 # Pipelines
 
+
+class PipelineEvaluationRequest(EvaluationRequestBase):
+    kind: Literal["PipelineEvaluationRequest"] = "PipelineEvaluationRequest"
+
+
+class PipelineMeasurementRequest(AnalysisRequestBase):
+    kind: Literal["PipelineMeasurementRequest"] = "PipelineMeasurementRequest"
+
+
+class PipelineSafetyCaseRequest(AnalysisRequestBase):
+    kind: Literal["PipelineSafetyCaseRequest"] = "PipelineSafetyCaseRequest"
+
+
 PipelineNodeRequest: TypeAlias = Union[
-    AnalysisRequestBase,
-    EvaluationRequestBase,
+    PipelineEvaluationRequest,
+    PipelineMeasurementRequest,
+    PipelineSafetyCaseRequest,
 ]
 
 
 class PipelineNode(DyffSchemaBaseModel):
+    """A node in the graph that defines the pipeline.
+
+    Each node contains a Dyff API request that might depend on the outcome of other
+    requests in the pipeline graph. When the pipeline runs, the requests are executed in
+    an order that respects these dependencies.
+    """
+
     name: str = pydantic.Field(
         description="The name of the node. Must be unique in the context of the pipeline."
     )
@@ -2779,7 +2796,7 @@ class PipelineNode(DyffSchemaBaseModel):
     request: PipelineNodeRequest = pydantic.Field(  # type: ignore[valid-type]
         discriminator="kind",
         description="The request template that will be executed when this node"
-        ' executes. You can use the syntax ``"$(node_name)"`` in request fields'
+        ' executes. You can use the syntax "$(node_name)" in request fields'
         " that reference another entity to indicate that the request depends"
         " on another node in the pipeline. The placeholder will be substituted"
         " with the ID of the created entity once it is known. Dyff infers the"
@@ -2797,7 +2814,9 @@ class PipelineParameter(DyffSchemaBaseModel):
         " parameter value. Should be a string like 'node_name.field1.field2'."
     )
     description: Optional[str] = pydantic.Field(
-        default=None, description="A description of the argument."
+        default=None,
+        description="A description of the argument.",
+        max_length=summary_maxlen(),
     )
 
 
@@ -2818,11 +2837,16 @@ class PipelineBase(DyffSchemaBaseModel):
 class Pipeline(DyffEntity, PipelineBase):
     """A set of Dyff workflows that can be executed as a group.
 
-    The pipeline
-    is a directed acyclic graph representing data dependencies between
-    workflows. For example, a simple pipeline might run an Evaluation and
-    then create a SafetyCase from the evaluation output. This pipeline would
+    The pipeline is a directed acyclic graph representing data dependencies
+    between workflows. For example, a simple pipeline might run an Evaluation
+    and then create a SafetyCase from the evaluation output. This pipeline would
     have a graph structure like ``evaluation -> safetycase``.
+
+    Each node in the pipeline contains the specification of a Dyff API request.
+    The request specifications may contain placeholders that reference other
+    nodes in the pipeline graph. When the pipeline is run, the nodes execute
+    in an order that respects their dependencies, and the placeholders are
+    replaced with concrete values once they are known.
     """
 
     kind: Literal["Pipeline"] = Entities.Pipeline.value
@@ -2859,6 +2883,12 @@ class PipelineRun(DyffEntity, PipelineRunBase):
         return None
 
 
+class PipelineRunStatus(Status):
+    nodes: dict[str, Status] = pydantic.Field(
+        default_factory=dict, description="The status of each pipeline node."
+    )
+
+
 # ---------------------------------------------------------------------------
 # Status enumerations
 
@@ -3099,34 +3129,6 @@ def is_status_success(status: str) -> bool:
     return status in [EntityStatus.complete, EntityStatus.ready]
 
 
-_ENTITY_CLASS = {
-    Entities.Analysis: Analysis,
-    Entities.Artifact: OCIArtifact,
-    Entities.Audit: Audit,
-    Entities.AuditProcedure: AuditProcedure,
-    Entities.Challenge: Challenge,
-    Entities.Dataset: Dataset,
-    Entities.DataSource: DataSource,
-    Entities.Evaluation: Evaluation,
-    Entities.Family: Family,
-    Entities.Hazard: Hazard,
-    Entities.InferenceService: InferenceService,
-    Entities.InferenceSession: InferenceSession,
-    Entities.Measurement: Measurement,
-    Entities.Method: Method,
-    Entities.Model: Model,
-    Entities.Module: Module,
-    Entities.Report: Report,
-    Entities.SafetyCase: SafetyCase,
-    Entities.Team: Team,
-    Entities.UseCase: UseCase,
-}
-
-
-def entity_class(kind: Entities):
-    return _ENTITY_CLASS[kind]
-
-
 # A Union type containing all Dyff top-level schema types, except the ones that
 # should not have Revisions generated for them when they change
 _DyffEntityTypeRevisable = Union[
@@ -3206,6 +3208,37 @@ class Revision(DyffEntity, RevisionMetadata):
 DyffEntityType = Union[_DyffEntityTypeRevisable, History, Revision]
 
 
+_ENTITY_CLASS: dict[Entities, type[DyffEntityType]] = {
+    Entities.Artifact: OCIArtifact,
+    Entities.Audit: Audit,
+    Entities.AuditProcedure: AuditProcedure,
+    Entities.Challenge: Challenge,
+    Entities.Dataset: Dataset,
+    Entities.DataSource: DataSource,
+    Entities.Evaluation: Evaluation,
+    Entities.Family: Family,
+    Entities.Hazard: Hazard,
+    Entities.History: History,
+    Entities.InferenceService: InferenceService,
+    Entities.InferenceSession: InferenceSession,
+    Entities.Measurement: Measurement,
+    Entities.Method: Method,
+    Entities.Model: Model,
+    Entities.Module: Module,
+    Entities.Pipeline: Pipeline,
+    Entities.PipelineRun: PipelineRun,
+    Entities.Report: Report,
+    Entities.Revision: Revision,
+    Entities.SafetyCase: SafetyCase,
+    Entities.Team: Team,
+    Entities.UseCase: UseCase,
+}
+
+
+def entity_class(kind: Entities) -> type[DyffEntityType]:
+    return _ENTITY_CLASS[kind]
+
+
 # A TypeVar that matches any Dyff top-level schema type
 DyffEntityT = TypeVar(
     "DyffEntityT",
@@ -3371,10 +3404,14 @@ __all__ = [
     "OCIArtifact",
     "Pipeline",
     "PipelineBase",
+    "PipelineEvaluationRequest",
+    "PipelineMeasurementRequest",
     "PipelineNode",
     "PipelineParameter",
     "PipelineRun",
     "PipelineRunBase",
+    "PipelineRunStatus",
+    "PipelineSafetyCaseRequest",
     "QueryableDyffEntity",
     "Report",
     "ReportBase",

dyff/schema/v0/r1/requests.py

@@ -19,11 +19,10 @@ from datetime import datetime
 from typing import Any, Literal, Optional, Union
 
 import pydantic
-from typing_extensions import TypeAlias
 
 from ... import upcast
 from . import commands
-from .base import DyffBaseModel, DyffSchemaBaseModel, JsonMergePatchSemantics
+from .base import DyffBaseModel, JsonMergePatchSemantics
 from .platform import (
     AnalysisRequestBase,
     AnalysisScope,

{dyff_schema-0.38.0.dist-info → dyff_schema-0.38.2.dist-info}/METADATA

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

@@ -1,5 +1,5 @@
 dyff/schema/__init__.py,sha256=w7OWDFuyGKd6xt_yllNtKzHahPgywrfU4Ue02psYaMA,2244
-dyff/schema/_version.py,sha256=c6iSoBC38zb_RKO51qx5EpHTYKsovdOH6wTmfXhQoZs,80
+dyff/schema/_version.py,sha256=vzg_BxADQGKoiaI8AUhHfE78RbKIhGvRvLFeN3aEmoc,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=QuZXp_PSI7hlXPiAn0TJwevHYIq_z4w3QQTB4PGm2s8,109389
-dyff/schema/v0/r1/requests.py,sha256=06ah2sDYzcLdokSrlqQyTCUoT4NKoqFvvjlfUU1DR8s,18382
+dyff/schema/v0/r1/platform.py,sha256=jWZXCnf0MaZb7k3pWUSVJ9rf8833AFSCZXU99Ddyyec,110859
+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
 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.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-dyff_schema-0.38.0.dist-info/licenses/NOTICE,sha256=YONACu0s_Ui6jNi-wtEsVQbTU1JIkh8wvLH6d1-Ni_w,43
-dyff_schema-0.38.0.dist-info/METADATA,sha256=WHoFpnO42oCa4VlL90M85Rw2l_ZmY0VDLN3P3nP5Kqc,3734
-dyff_schema-0.38.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-dyff_schema-0.38.0.dist-info/top_level.txt,sha256=9e3VVdeX73t_sUJOPQPCcGtYO1JhoErhHIi3WoWGcFI,5
-dyff_schema-0.38.0.dist-info/RECORD,,
+dyff_schema-0.38.2.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+dyff_schema-0.38.2.dist-info/licenses/NOTICE,sha256=YONACu0s_Ui6jNi-wtEsVQbTU1JIkh8wvLH6d1-Ni_w,43
+dyff_schema-0.38.2.dist-info/METADATA,sha256=Npn-CVL-k0hByX8LQ5crFv23EDtLnsUDX0FQHvePF4s,3734
+dyff_schema-0.38.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+dyff_schema-0.38.2.dist-info/top_level.txt,sha256=9e3VVdeX73t_sUJOPQPCcGtYO1JhoErhHIi3WoWGcFI,5
+dyff_schema-0.38.2.dist-info/RECORD,,