acceldata-sdk-python 26.7.0__py3-none-any.whl

acceldata/models/api/pipeline/node.py

@@ -0,0 +1,201 @@
+# coding: utf-8
+
+"""
+    Pipeline Control Plane API
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+    The version of the OpenAPI document: 1.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from pydantic import BaseModel, ConfigDict, Field, StrictFloat, StrictInt, StrictStr
+from typing import Any, ClassVar, Dict, List, Optional, Union
+from acceldata.models.api.pipeline.meta import Meta
+from acceldata.models.api.pipeline.metric import Metric
+from acceldata.models.api.pipeline.policy_summary import PolicySummary
+from typing import Optional, Set
+from typing_extensions import Self
+from pydantic_core import to_jsonable_python
+
+class Node(BaseModel):
+    """
+    Node
+    """ # noqa: E501
+    alerts_count: Optional[StrictInt] = Field(default=None, alias="alertsCount")
+    asset_id: Optional[StrictInt] = Field(default=None, alias="assetId")
+    asset_type: Optional[StrictStr] = Field(default=None, alias="assetType")
+    context: Optional[Dict[str, Any]] = None
+    created_via: List[StrictStr] = Field(alias="createdVia")
+    data_type: Optional[StrictStr] = Field(default=None, alias="dataType")
+    description: Optional[StrictStr] = None
+    id: Optional[StrictInt] = None
+    job_node_metric: Optional[List[Metric]] = Field(default=None, alias="jobNodeMetric")
+    job_status: Optional[StrictStr] = Field(default=None, description="Allowed values at generation time: SUCCESS, ERROR, ABORTED, RUNNING, SKIPPED.", alias="jobStatus")
+    linked_pipeline_run_id: Optional[StrictInt] = Field(default=None, alias="linkedPipelineRunId")
+    meta: Optional[Meta] = None
+    name: Optional[StrictStr] = None
+    node_properties: Optional[Dict[str, Any]] = Field(default=None, alias="nodeProperties")
+    parent_id: Optional[StrictInt] = Field(default=None, alias="parentId")
+    parent_node_id: Optional[StrictInt] = Field(default=None, alias="parentNodeId")
+    pipeline_id: Optional[StrictInt] = Field(default=None, alias="pipelineId")
+    related_pipeline_count: Optional[StrictInt] = Field(default=None, alias="relatedPipelineCount")
+    reliability_policy_summary: Optional[List[PolicySummary]] = Field(default=None, alias="reliabilityPolicySummary")
+    reliability_score: Optional[Union[StrictFloat, StrictInt]] = Field(default=None, alias="reliabilityScore")
+    source_type: Optional[StrictStr] = Field(default=None, alias="sourceType")
+    status: StrictStr = Field(description="Allowed values at generation time: ACTIVE, STALE, DELETED.")
+    type: StrictStr = Field(description="Allowed values at generation time: ASSET, FUNCTIONAL.")
+    uid: Optional[StrictStr] = None
+    version_id: Optional[StrictInt] = Field(default=None, alias="versionId")
+    __properties: ClassVar[List[str]] = ["alertsCount", "assetId", "assetType", "context", "createdVia", "dataType", "description", "id", "jobNodeMetric", "jobStatus", "linkedPipelineRunId", "meta", "name", "nodeProperties", "parentId", "parentNodeId", "pipelineId", "relatedPipelineCount", "reliabilityPolicySummary", "reliabilityScore", "sourceType", "status", "type", "uid", "versionId"]
+
+    model_config = ConfigDict(
+        validate_by_name=True,
+        validate_by_alias=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return json.dumps(to_jsonable_python(self.to_dict()))
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of Node from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        # override the default output from pydantic by calling `to_dict()` of each item in job_node_metric (list)
+        _items = []
+        if self.job_node_metric:
+            for _item_job_node_metric in self.job_node_metric:
+                if _item_job_node_metric:
+                    _items.append(_item_job_node_metric.to_dict())
+            _dict['jobNodeMetric'] = _items
+        # override the default output from pydantic by calling `to_dict()` of meta
+        if self.meta:
+            _dict['meta'] = self.meta.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of each item in reliability_policy_summary (list)
+        _items = []
+        if self.reliability_policy_summary:
+            for _item_reliability_policy_summary in self.reliability_policy_summary:
+                if _item_reliability_policy_summary:
+                    _items.append(_item_reliability_policy_summary.to_dict())
+            _dict['reliabilityPolicySummary'] = _items
+        # set to None if asset_type (nullable) is None
+        # and model_fields_set contains the field
+        if self.asset_type is None and "asset_type" in self.model_fields_set:
+            _dict['assetType'] = None
+
+        # set to None if data_type (nullable) is None
+        # and model_fields_set contains the field
+        if self.data_type is None and "data_type" in self.model_fields_set:
+            _dict['dataType'] = None
+
+        # set to None if description (nullable) is None
+        # and model_fields_set contains the field
+        if self.description is None and "description" in self.model_fields_set:
+            _dict['description'] = None
+
+        # set to None if job_node_metric (nullable) is None
+        # and model_fields_set contains the field
+        if self.job_node_metric is None and "job_node_metric" in self.model_fields_set:
+            _dict['jobNodeMetric'] = None
+
+        # set to None if job_status (nullable) is None
+        # and model_fields_set contains the field
+        if self.job_status is None and "job_status" in self.model_fields_set:
+            _dict['jobStatus'] = None
+
+        # set to None if name (nullable) is None
+        # and model_fields_set contains the field
+        if self.name is None and "name" in self.model_fields_set:
+            _dict['name'] = None
+
+        # set to None if reliability_policy_summary (nullable) is None
+        # and model_fields_set contains the field
+        if self.reliability_policy_summary is None and "reliability_policy_summary" in self.model_fields_set:
+            _dict['reliabilityPolicySummary'] = None
+
+        # set to None if source_type (nullable) is None
+        # and model_fields_set contains the field
+        if self.source_type is None and "source_type" in self.model_fields_set:
+            _dict['sourceType'] = None
+
+        # set to None if uid (nullable) is None
+        # and model_fields_set contains the field
+        if self.uid is None and "uid" in self.model_fields_set:
+            _dict['uid'] = None
+
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of Node from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "alertsCount": obj.get("alertsCount"),
+            "assetId": obj.get("assetId"),
+            "assetType": obj.get("assetType"),
+            "context": obj.get("context"),
+            "createdVia": obj.get("createdVia"),
+            "dataType": obj.get("dataType"),
+            "description": obj.get("description"),
+            "id": obj.get("id"),
+            "jobNodeMetric": [Metric.from_dict(_item) for _item in obj["jobNodeMetric"]] if obj.get("jobNodeMetric") is not None else None,
+            "jobStatus": obj.get("jobStatus"),
+            "linkedPipelineRunId": obj.get("linkedPipelineRunId"),
+            "meta": Meta.from_dict(obj["meta"]) if obj.get("meta") is not None else None,
+            "name": obj.get("name"),
+            "nodeProperties": obj.get("nodeProperties"),
+            "parentId": obj.get("parentId"),
+            "parentNodeId": obj.get("parentNodeId"),
+            "pipelineId": obj.get("pipelineId"),
+            "relatedPipelineCount": obj.get("relatedPipelineCount"),
+            "reliabilityPolicySummary": [PolicySummary.from_dict(_item) for _item in obj["reliabilityPolicySummary"]] if obj.get("reliabilityPolicySummary") is not None else None,
+            "reliabilityScore": obj.get("reliabilityScore"),
+            "sourceType": obj.get("sourceType"),
+            "status": obj.get("status") if obj.get("status") is not None else 'ACTIVE',
+            "type": obj.get("type"),
+            "uid": obj.get("uid"),
+            "versionId": obj.get("versionId")
+        })
+        return _obj
+
+

acceldata/models/api/pipeline/node_response.py

@@ -0,0 +1,92 @@
+# coding: utf-8
+
+"""
+    Pipeline Control Plane API
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+    The version of the OpenAPI document: 1.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from pydantic import BaseModel, ConfigDict
+from typing import Any, ClassVar, Dict, List
+from acceldata.models.api.pipeline.node import Node
+from typing import Optional, Set
+from typing_extensions import Self
+from pydantic_core import to_jsonable_python
+
+class NodeResponse(BaseModel):
+    """
+    NodeResponse
+    """ # noqa: E501
+    node: Node
+    __properties: ClassVar[List[str]] = ["node"]
+
+    model_config = ConfigDict(
+        validate_by_name=True,
+        validate_by_alias=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return json.dumps(to_jsonable_python(self.to_dict()))
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of NodeResponse from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        # override the default output from pydantic by calling `to_dict()` of node
+        if self.node:
+            _dict['node'] = self.node.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of NodeResponse from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "node": Node.from_dict(obj["node"]) if obj.get("node") is not None else None
+        })
+        return _obj
+
+

acceldata/models/api/pipeline/notification_payload.py

@@ -0,0 +1,114 @@
+# coding: utf-8
+
+"""
+    Pipeline Control Plane API
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+    The version of the OpenAPI document: 1.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictInt, StrictStr
+from typing import Any, ClassVar, Dict, List, Optional
+from typing import Optional, Set
+from typing_extensions import Self
+from pydantic_core import to_jsonable_python
+
+class NotificationPayload(BaseModel):
+    """
+    NotificationPayload
+    """ # noqa: E501
+    alerts_enabled: StrictBool = Field(alias="alertsEnabled")
+    configured_notification_group_ids: List[StrictInt] = Field(alias="configuredNotificationGroupIds")
+    notify_on: List[StrictStr] = Field(alias="notifyOn")
+    notify_on_success: StrictBool = Field(alias="notifyOnSuccess")
+    notify_on_warning: StrictBool = Field(alias="notifyOnWarning")
+    re_notify_factor: Optional[StrictInt] = Field(default=None, alias="reNotifyFactor")
+    re_notify_mode: StrictStr = Field(description="Allowed values at generation time: INTERVAL, RULE_CHANGE.", alias="reNotifyMode")
+    severity: Optional[StrictStr] = Field(default=None, description="Allowed values at generation time: LOW, MEDIUM, HIGH, CRITICAL.")
+    tags: Optional[List[StrictStr]] = None
+    __properties: ClassVar[List[str]] = ["alertsEnabled", "configuredNotificationGroupIds", "notifyOn", "notifyOnSuccess", "notifyOnWarning", "reNotifyFactor", "reNotifyMode", "severity", "tags"]
+
+    model_config = ConfigDict(
+        validate_by_name=True,
+        validate_by_alias=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return json.dumps(to_jsonable_python(self.to_dict()))
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of NotificationPayload from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        # set to None if severity (nullable) is None
+        # and model_fields_set contains the field
+        if self.severity is None and "severity" in self.model_fields_set:
+            _dict['severity'] = None
+
+        # set to None if tags (nullable) is None
+        # and model_fields_set contains the field
+        if self.tags is None and "tags" in self.model_fields_set:
+            _dict['tags'] = None
+
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of NotificationPayload from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "alertsEnabled": obj.get("alertsEnabled"),
+            "configuredNotificationGroupIds": obj.get("configuredNotificationGroupIds"),
+            "notifyOn": obj.get("notifyOn"),
+            "notifyOnSuccess": obj.get("notifyOnSuccess"),
+            "notifyOnWarning": obj.get("notifyOnWarning"),
+            "reNotifyFactor": obj.get("reNotifyFactor"),
+            "reNotifyMode": obj.get("reNotifyMode"),
+            "severity": obj.get("severity"),
+            "tags": obj.get("tags")
+        })
+        return _obj
+
+

acceldata/models/api/pipeline/pipeline.py

@@ -0,0 +1,152 @@
+# coding: utf-8
+
+"""
+    Pipeline Control Plane API
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+    The version of the OpenAPI document: 1.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from datetime import datetime
+from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictInt, StrictStr
+from typing import Any, ClassVar, Dict, List, Optional
+from acceldata.models.api.pipeline.meta import Meta
+from acceldata.models.api.pipeline.notification_payload import NotificationPayload
+from acceldata.models.api.pipeline.pipeline_baseline_metric import PipelineBaselineMetric
+from acceldata.models.api.pipeline.tag import Tag
+from typing import Optional, Set
+from typing_extensions import Self
+from pydantic_core import to_jsonable_python
+
+class Pipeline(BaseModel):
+    """
+    Pipeline
+    """ # noqa: E501
+    context: Optional[Dict[str, Any]] = None
+    created_at: Optional[datetime] = Field(default=None, alias="createdAt")
+    description: Optional[StrictStr] = None
+    enabled: StrictBool
+    id: Optional[StrictInt] = None
+    interrupt: StrictBool
+    latest_run_id: Optional[StrictInt] = Field(default=None, alias="latestRunId")
+    meta: Optional[Meta] = None
+    name: StrictStr
+    notification_channels: Optional[NotificationPayload] = Field(default=None, alias="notificationChannels")
+    pipeline_baseline_metric: Optional[PipelineBaselineMetric] = Field(default=None, alias="pipelineBaselineMetric")
+    schedule: Optional[StrictStr] = None
+    scheduled: StrictBool
+    scheduler_type: StrictStr = Field(description="Allowed values at generation time: INTERNAL, EXTERNAL.", alias="schedulerType")
+    source_type: StrictStr = Field(description="Allowed values at generation time: AIRFLOW, AZURE_DATA_FACTORY, TRINO, OTHERS, AUTOSYS, SPARK, DBT, FIVETRAN, SNAPLOGIC.", alias="sourceType")
+    tags: List[Tag]
+    uid: StrictStr = Field(description="Unique id for the pipeline. When omitted on create, the server assigns a new random UUID.")
+    updated_at: Optional[datetime] = Field(default=None, alias="updatedAt")
+    __properties: ClassVar[List[str]] = ["context", "createdAt", "description", "enabled", "id", "interrupt", "latestRunId", "meta", "name", "notificationChannels", "pipelineBaselineMetric", "schedule", "scheduled", "schedulerType", "sourceType", "tags", "uid", "updatedAt"]
+
+    model_config = ConfigDict(
+        validate_by_name=True,
+        validate_by_alias=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return json.dumps(to_jsonable_python(self.to_dict()))
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of Pipeline from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        # override the default output from pydantic by calling `to_dict()` of meta
+        if self.meta:
+            _dict['meta'] = self.meta.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of notification_channels
+        if self.notification_channels:
+            _dict['notificationChannels'] = self.notification_channels.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of pipeline_baseline_metric
+        if self.pipeline_baseline_metric:
+            _dict['pipelineBaselineMetric'] = self.pipeline_baseline_metric.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of each item in tags (list)
+        _items = []
+        for _item_tags in self.tags or []:
+            if _item_tags:
+                _items.append(_item_tags.to_dict())
+        _dict['tags'] = _items
+        # set to None if description (nullable) is None
+        # and model_fields_set contains the field
+        if self.description is None and "description" in self.model_fields_set:
+            _dict['description'] = None
+
+        # set to None if schedule (nullable) is None
+        # and model_fields_set contains the field
+        if self.schedule is None and "schedule" in self.model_fields_set:
+            _dict['schedule'] = None
+
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of Pipeline from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "context": obj.get("context"),
+            "createdAt": obj.get("createdAt"),
+            "description": obj.get("description"),
+            "enabled": obj.get("enabled") if obj.get("enabled") is not None else True,
+            "id": obj.get("id"),
+            "interrupt": obj.get("interrupt") if obj.get("interrupt") is not None else False,
+            "latestRunId": obj.get("latestRunId"),
+            "meta": Meta.from_dict(obj["meta"]) if obj.get("meta") is not None else None,
+            "name": obj.get("name"),
+            "notificationChannels": NotificationPayload.from_dict(obj["notificationChannels"]) if obj.get("notificationChannels") is not None else None,
+            "pipelineBaselineMetric": PipelineBaselineMetric.from_dict(obj["pipelineBaselineMetric"]) if obj.get("pipelineBaselineMetric") is not None else None,
+            "schedule": obj.get("schedule"),
+            "scheduled": obj.get("scheduled") if obj.get("scheduled") is not None else False,
+            "schedulerType": obj.get("schedulerType") if obj.get("schedulerType") is not None else 'INTERNAL',
+            "sourceType": obj.get("sourceType") if obj.get("sourceType") is not None else 'OTHERS',
+            "tags": [Tag.from_dict(_item) for _item in obj["tags"]] if obj.get("tags") is not None else None,
+            "uid": obj.get("uid"),
+            "updatedAt": obj.get("updatedAt")
+        })
+        return _obj
+
+

acceldata/models/api/pipeline/pipeline_baseline_metric.py

@@ -0,0 +1,92 @@
+# coding: utf-8
+
+"""
+    Pipeline Control Plane API
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+    The version of the OpenAPI document: 1.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictInt, StrictStr
+from typing import Any, ClassVar, Dict, List
+from typing import Optional, Set
+from typing_extensions import Self
+from pydantic_core import to_jsonable_python
+
+class PipelineBaselineMetric(BaseModel):
+    """
+    PipelineBaselineMetric
+    """ # noqa: E501
+    include_successful_runs_only: StrictBool = Field(alias="includeSuccessfulRunsOnly")
+    metrics: StrictInt
+    unit: StrictStr = Field(description="Allowed values at generation time: DAYS, RUNS.")
+    __properties: ClassVar[List[str]] = ["includeSuccessfulRunsOnly", "metrics", "unit"]
+
+    model_config = ConfigDict(
+        validate_by_name=True,
+        validate_by_alias=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return json.dumps(to_jsonable_python(self.to_dict()))
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of PipelineBaselineMetric from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of PipelineBaselineMetric from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "includeSuccessfulRunsOnly": obj.get("includeSuccessfulRunsOnly"),
+            "metrics": obj.get("metrics"),
+            "unit": obj.get("unit")
+        })
+        return _obj
+
+