# coding: utf-8
"""
Amorphic Data Platform
Amorphic Data Platform - API Definition documentation
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, StrictStr, field_validator
from typing import Any, ClassVar, Dict, List, Optional
from openapi_client.models.allowed_guard_rail import AllowedGuardRail
from typing import Optional, Set
from typing_extensions import Self
[docs]
class AIServiceConfiguration(BaseModel):
"""
AIServiceConfiguration
""" # noqa: E501
service_name: StrictStr = Field(description="Component/service identifier (e.g., nl2sql, knowledgebases, agents, chats, datasets, jobs, datalabs, data-pipelines).", alias="ServiceName")
status: StrictStr = Field(description="Enablement status based on associated service toggle.", alias="Status")
assigned_models: Optional[List[StrictStr]] = Field(default=None, description="List of assigned/allowed model IDs for this component/service.", alias="AssignedModels")
default_model: Optional[StrictStr] = Field(default=None, description="Default model ID or 'N/A' when not applicable.", alias="DefaultModel")
assigned_guard_rails: Optional[List[AllowedGuardRail]] = Field(default=None, description="List of guard rails assigned to this component/service.", alias="AssignedGuardRails")
__properties: ClassVar[List[str]] = ["ServiceName", "Status", "AssignedModels", "DefaultModel", "AssignedGuardRails"]
[docs]
@field_validator('status')
def status_validate_enum(cls, value):
"""Validates the enum"""
if value not in set(['Enabled', 'Disabled']):
raise ValueError("must be one of enum values ('Enabled', 'Disabled')")
return value
model_config = ConfigDict(
populate_by_name=True,
validate_assignment=True,
protected_namespaces=(),
)
[docs]
def to_str(self) -> str:
"""Returns the string representation of the model using alias"""
return pprint.pformat(self.model_dump(by_alias=True))
[docs]
def to_json(self) -> str:
"""Returns the JSON representation of the model using alias"""
# TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
return json.dumps(self.to_dict())
[docs]
@classmethod
def from_json(cls, json_str: str) -> Optional[Self]:
"""Create an instance of AIServiceConfiguration from a JSON string"""
return cls.from_dict(json.loads(json_str))
[docs]
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 assigned_guard_rails (list)
_items = []
if self.assigned_guard_rails:
for _item_assigned_guard_rails in self.assigned_guard_rails:
if _item_assigned_guard_rails:
_items.append(_item_assigned_guard_rails.to_dict())
_dict['AssignedGuardRails'] = _items
return _dict
[docs]
@classmethod
def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
"""Create an instance of AIServiceConfiguration from a dict"""
if obj is None:
return None
if not isinstance(obj, dict):
return cls.model_validate(obj)
_obj = cls.model_validate({
"ServiceName": obj.get("ServiceName"),
"Status": obj.get("Status"),
"AssignedModels": obj.get("AssignedModels"),
"DefaultModel": obj.get("DefaultModel"),
"AssignedGuardRails": [AllowedGuardRail.from_dict(_item) for _item in obj["AssignedGuardRails"]] if obj.get("AssignedGuardRails") is not None else None
})
return _obj