dyff-schema 0.6.0__tar.gz → 0.7.1__tar.gz

{dyff_schema-0.6.0 → dyff_schema-0.7.1}/.gitlab-ci.yml

@@ -20,7 +20,7 @@ include:
     file:
       - prettier.yml
   - project: buildgarden/pipelines/python
-    ref: 0.10.0
+    ref: 0.12.0
     file:
       - python-autoflake.yml
       - python-black.yml

{dyff_schema-0.6.0/dyff_schema.egg-info → dyff_schema-0.7.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dyff-schema
-Version: 0.6.0
+Version: 0.7.1
 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.6.0 → dyff_schema-0.7.1}/dyff/schema/v0/r1/base.py

@@ -1,6 +1,7 @@
 # SPDX-FileCopyrightText: 2024 UL Research Institutes
 # SPDX-License-Identifier: Apache-2.0
 
+from datetime import datetime
 from typing import Any, Generic, NamedTuple, Optional, Type, TypeVar
 
 import pydantic
@@ -563,6 +564,14 @@ class DyffSchemaBaseModel(pydantic.BaseModel):
     Python reserved words like 'bytes' as field names.
     """
 
+    @pydantic.root_validator
+    def _require_datetime_timezone_aware(cls, values):
+        for k, v in values.items():
+            if isinstance(v, datetime):
+                if v.tzinfo is None:
+                    raise ValueError(f"{cls.__qualname__}.{k}: timezone not set")
+        return values
+
     def dict(self, *, by_alias: bool = True, **kwargs) -> dict[str, Any]:
         return super().dict(by_alias=by_alias, **kwargs)
 

{dyff_schema-0.6.0 → dyff_schema-0.7.1}/dyff/schema/v0/r1/platform.py

@@ -298,7 +298,7 @@ class Status(DyffSchemaBaseModel):
     )
 
 
-class Documentation(DyffSchemaBaseModel):
+class DocumentationBase(DyffSchemaBaseModel):
     title: Optional[str] = pydantic.Field(
         default=None,
         description='A short plain string suitable as a title or "headline".',
@@ -318,9 +318,13 @@ class Documentation(DyffSchemaBaseModel):
     )
 
 
+class Documentation(SchemaVersion, DocumentationBase):
+    pass
+
+
 class Documented(DyffSchemaBaseModel):
-    documentation: Documentation = pydantic.Field(
-        default_factory=Documentation,
+    documentation: DocumentationBase = pydantic.Field(
+        default_factory=DocumentationBase,
         description="Documentation of the resource. The content is used to"
         " populate various views in the web UI.",
     )
@@ -380,8 +384,8 @@ class Family(Labeled, SchemaVersion, DyffModelWithID):
         description="Tags mapping interpretable names to resource IDs.",
     )
 
-    documentation: Documentation = pydantic.Field(
-        default_factory=Documentation,
+    documentation: DocumentationBase = pydantic.Field(
+        default_factory=DocumentationBase,
         description="Documentation of the resource family. The content is used"
         " to populate various views in the web UI.",
     )
@@ -1861,6 +1865,7 @@ __all__ = [
     "DataView",
     "Digest",
     "Documentation",
+    "DocumentationBase",
     "DyffDataSchema",
     "DyffEntity",
     "DyffModelWithID",

{dyff_schema-0.6.0 → dyff_schema-0.7.1}/dyff/schema/v0/r1/requests.py

@@ -13,6 +13,7 @@ in response.
 """
 
 
+from datetime import datetime
 from typing import Any, Optional, Union
 
 import pydantic
@@ -22,7 +23,7 @@ from .platform import (
     AnalysisBase,
     DatasetBase,
     DataView,
-    Documentation,
+    DocumentationBase,
     EvaluationBase,
     InferenceServiceBase,
     InferenceSessionBase,
@@ -51,7 +52,7 @@ class DatasetCreateRequest(DyffEntityCreateRequest, DatasetBase):
     pass
 
 
-class DocumentationEditRequest(Documentation):
+class DocumentationEditRequest(SchemaVersion, DocumentationBase):
     pass
 
 
@@ -65,6 +66,14 @@ class InferenceSessionCreateRequest(DyffEntityCreateRequest, InferenceSessionBas
     inferenceService: str = pydantic.Field(description="InferenceService ID")
 
 
+class InferenceSessionTokenCreateRequest(SchemaVersion, DyffSchemaBaseModel):
+    expires: Optional[datetime] = pydantic.Field(
+        default=None,
+        description="Expiration time of the token. Must be <= expiration time"
+        " of session. Default: expiration time of session.",
+    )
+
+
 class EvaluationInferenceSessionRequest(InferenceSessionBase):
     inferenceService: str = pydantic.Field(description="InferenceService ID")
 
@@ -128,11 +137,11 @@ class ReportCreateRequest(DyffEntityCreateRequest, ReportBase):
     )
 
 
-class TagCreateRequest(TagBase):
+class TagCreateRequest(SchemaVersion, TagBase):
     pass
 
 
-class LabelUpdateRequest(Labeled):
+class LabelUpdateRequest(SchemaVersion, Labeled):
     pass
 
 
@@ -241,6 +250,7 @@ __all__ = [
     "InferenceServiceQueryRequest",
     "InferenceSessionCreateRequest",
     "InferenceSessionQueryRequest",
+    "InferenceSessionTokenCreateRequest",
     "LabelUpdateRequest",
     "MeasurementQueryRequest",
     "MethodCreateRequest",

{dyff_schema-0.6.0 → dyff_schema-0.7.1/dyff_schema.egg-info}/PKG-INFO

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