dyff-schema 0.19.0__tar.gz → 0.20.0__tar.gz

{dyff_schema-0.19.0 → dyff_schema-0.20.0}/.gitlab-ci.yml

@@ -12,7 +12,7 @@ include:
     file:
       - gitlab-release.yml
   - project: buildgarden/pipelines/detect-secrets
-    ref: 0.2.1
+    ref: 0.2.2
     file:
       - detect-secrets.yml
   - project: buildgarden/pipelines/prettier

{dyff_schema-0.19.0/dyff_schema.egg-info → dyff_schema-0.20.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dyff-schema
-Version: 0.19.0
+Version: 0.20.0
 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.19.0 → dyff_schema-0.20.0}/dyff/schema/v0/r1/platform.py

@@ -194,11 +194,13 @@ class Entities(str, enum.Enum):
     Analysis = "Analysis"
     Audit = "Audit"
     AuditProcedure = "AuditProcedure"
+    Concern = "Concern"
     DataSource = "DataSource"
     Dataset = "Dataset"
     Documentation = "Documentation"
     Evaluation = "Evaluation"
     Family = "Family"
+    Hazard = "Hazard"
     History = "History"
     InferenceService = "InferenceService"
     InferenceSession = "InferenceSession"
@@ -210,6 +212,7 @@ class Entities(str, enum.Enum):
     Revision = "Revision"
     SafetyCase = "SafetyCase"
     Score = "Score"
+    UseCase = "UseCase"
 
 
 class Resources(str, enum.Enum):
@@ -218,11 +221,14 @@ class Resources(str, enum.Enum):
     Analysis = "analyses"
     Audit = "audits"
     AuditProcedure = "auditprocedures"
+    Concern = "concerns"
     Dataset = "datasets"
     DataSource = "datasources"
+    Descriptor = "descriptors"
     Documentation = "documentation"
     Evaluation = "evaluations"
     Family = "families"
+    Hazard = "hazards"
     History = "histories"
     InferenceService = "inferenceservices"
     InferenceSession = "inferencesessions"
@@ -234,6 +240,7 @@ class Resources(str, enum.Enum):
     Revision = "revisions"
     SafetyCase = "safetycases"
     Score = "scores"
+    UseCase = "usecases"
 
     Task = "tasks"
     """
@@ -396,6 +403,15 @@ class Documented(DyffSchemaBaseModel):
     )
 
 
+class Metadata(DyffSchemaBaseModel):
+    pass
+
+
+class Mutable(DyffSchemaBaseModel):
+    etag: str
+    lastModificationTime: datetime
+
+
 class DyffEntity(Status, Labeled, SchemaVersion, DyffModelWithID):
     kind: Literal[
         "Analysis",
@@ -405,6 +421,7 @@ class DyffEntity(Status, Labeled, SchemaVersion, DyffModelWithID):
         "Dataset",
         "Evaluation",
         "Family",
+        "Hazard",
         "History",
         "InferenceService",
         "InferenceSession",
@@ -415,6 +432,7 @@ class DyffEntity(Status, Labeled, SchemaVersion, DyffModelWithID):
         "Report",
         "Revision",
         "SafetyCase",
+        "UseCase",
     ]
 
     annotations: list[Annotation] = pydantic.Field(
@@ -683,6 +701,32 @@ class History(DyffEntity):
     )
 
 
+class ConcernBase(Documented):
+    pass
+
+
+class Concern(ConcernBase, DyffEntity):
+    def label_key(self) -> str:
+        return f"{Resources[self.kind].value}.dyff.io/{self.id}"
+
+    def label_value(self) -> str:
+        return "1"
+
+    def dependencies(self) -> list[str]:
+        return []
+
+    def resource_allocation(self) -> Optional[ResourceAllocation]:
+        return None
+
+
+class Hazard(Concern):
+    kind: Literal["Hazard"] = Entities.Hazard.value
+
+
+class UseCase(Concern):
+    kind: Literal["UseCase"] = Entities.UseCase.value
+
+
 # ----------------------------------------------------------------------------
 
 
@@ -1727,6 +1771,16 @@ class MethodBase(DyffSchemaBaseModel):
         return scores
 
 
+class MethodMetadata(Metadata):
+    concerns: list[Concern] = pydantic.Field(
+        default_factory=list, description="The Concerns applicable to this Method."
+    )
+
+
+class ModelMetadata(Metadata):
+    pass
+
+
 class Method(DyffEntity, MethodBase):
     kind: Literal["Method"] = Entities.Method.value
 
@@ -2169,6 +2223,7 @@ DyffEntityTypeExceptRevision = Union[
     Dataset,
     Evaluation,
     Family,
+    Hazard,
     History,
     InferenceService,
     InferenceSession,
@@ -2178,6 +2233,7 @@ DyffEntityTypeExceptRevision = Union[
     Module,
     Report,
     SafetyCase,
+    UseCase,
 ]
 
 
@@ -2232,6 +2288,8 @@ __all__ = [
     "Audit",
     "AuditProcedure",
     "AuditRequirement",
+    "Concern",
+    "ConcernBase",
     "ContainerImageSource",
     "DataSchema",
     "Dataset",
@@ -2243,6 +2301,7 @@ __all__ = [
     "Digest",
     "Documentation",
     "DocumentationBase",
+    "Documented",
     "DyffDataSchema",
     "DyffEntity",
     "DyffEntityType",
@@ -2263,6 +2322,7 @@ __all__ = [
     "ForeignMethod",
     "ForeignModel",
     "Frameworks",
+    "Hazard",
     "History",
     "Identity",
     "InferenceInterface",
@@ -2333,6 +2393,7 @@ __all__ = [
     "StorageSignedURL",
     "TagName",
     "TaskSchema",
+    "UseCase",
     "entity_class",
     "JobStatus",
     "EntityStatus",

{dyff_schema-0.19.0 → dyff_schema-0.20.0}/dyff/schema/v0/r1/requests.py

@@ -24,6 +24,7 @@ from .base import DyffBaseModel
 from .platform import (
     AnalysisBase,
     AnalysisScope,
+    ConcernBase,
     DatasetBase,
     DataView,
     DocumentationBase,
@@ -92,6 +93,10 @@ class AnalysisCreateRequest(DyffEntityCreateRequest, AnalysisBase):
         return scope
 
 
+class ConcernCreateRequest(DyffEntityCreateRequest, ConcernBase):
+    pass
+
+
 class DatasetCreateRequest(DyffEntityCreateRequest, DatasetBase):
     pass
 
@@ -330,9 +335,14 @@ class ScoreQueryRequest(DyffRequestDefaultValidators):
     model: Optional[str] = pydantic.Field(default=None)
 
 
+class UseCaseQueryRequest(DyffEntityQueryRequest):
+    pass
+
+
 __all__ = [
     "AnalysisCreateRequest",
     "AuditQueryRequest",
+    "ConcernCreateRequest",
     "DyffEntityCreateRequest",
     "DyffEntityQueryRequest",
     "DyffRequestBase",
@@ -364,4 +374,5 @@ __all__ = [
     "ReportQueryRequest",
     "SafetyCaseQueryRequest",
     "ScoreQueryRequest",
+    "UseCaseQueryRequest",
 ]

{dyff_schema-0.19.0 → dyff_schema-0.20.0/dyff_schema.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dyff-schema
-Version: 0.19.0
+Version: 0.20.0
 Summary: Data models for the Dyff AI auditing platform.
 Author-email: Digital Safety Research Institute <contact@dsri.org>
 License: Apache-2.0