# coding: utf-8
"""
Amorphic Data Platform
Amorphic Data Platform - API Definition documentation
The version of the OpenAPI document: 0.3.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, StrictFloat, StrictInt, StrictStr, field_validator
from typing import Any, ClassVar, Dict, List, Optional, Union
from openapi_client.models.custom_word_filters import CustomWordFilters
from openapi_client.models.denied_topic import DeniedTopic
from openapi_client.models.guard_rail_filter_config import GuardRailFilterConfig
from openapi_client.models.sensitive_information_filters import SensitiveInformationFilters
from typing import Optional, Set
from typing_extensions import Self
[docs]
class GuardRailDetail(BaseModel):
"""
Detailed information about a guard rail.
""" # noqa: E501
guard_rail_id: Optional[StrictStr] = Field(default=None, description="The unique identifier of the guard rail.", alias="GuardRailId")
guard_rail_version: Optional[StrictStr] = Field(default=None, description="The version of the guard rail (e.g., DRAFT, PUBLISHED).", alias="GuardRailVersion")
guard_rail_name: Optional[StrictStr] = Field(default=None, description="The name of the guard rail.", alias="GuardRailName")
description: Optional[StrictStr] = Field(default=None, description="The description of the guard rail.", alias="Description")
creation_time: Optional[datetime] = Field(default=None, description="The timestamp when the guard rail was created.", alias="CreationTime")
created_by: Optional[StrictStr] = Field(default=None, description="The user who created the guard rail.", alias="CreatedBy")
last_modified_time: Optional[datetime] = Field(default=None, description="The timestamp when the guard rail was last modified.", alias="LastModifiedTime")
last_modified_by: Optional[StrictStr] = Field(default=None, description="The user who last modified the guard rail.", alias="LastModifiedBy")
tier: Optional[StrictStr] = Field(default=None, description="The tier of the guard rail (CLASSIC or STANDARD).", alias="Tier")
is_cross_region_enabled: Optional[StrictBool] = Field(default=None, description="Whether cross-region configuration is enabled for the guard rail.", alias="IsCrossRegionEnabled")
scope: Optional[StrictStr] = Field(default=None, description="The scope of the guard rail (global or private).", alias="Scope")
owner: Optional[StrictStr] = Field(default=None, description="The owner of the guard rail (user ID for private, system user for global scope).", alias="Owner")
blocked_message: Optional[StrictStr] = Field(default=None, description="The message to display when content is blocked.", alias="BlockedMessage")
filters_config: Optional[List[GuardRailFilterConfig]] = Field(default=None, description="Configuration for content filters.", alias="FiltersConfig")
denied_topics: Optional[List[DeniedTopic]] = Field(default=None, description="List of topics that are denied.", alias="DeniedTopics")
sensitive_information_filters: Optional[SensitiveInformationFilters] = Field(default=None, alias="SensitiveInformationFilters")
custom_word_filters: Optional[CustomWordFilters] = Field(default=None, alias="CustomWordFilters")
enable_contextual_grounding_check: Optional[StrictBool] = Field(default=None, description="Whether to enable contextual grounding check.", alias="EnableContextualGroundingCheck")
grounding_threshold: Optional[Union[StrictFloat, StrictInt]] = Field(default=None, description="The threshold for grounding check.", alias="GroundingThreshold")
relevance_threshold: Optional[Union[StrictFloat, StrictInt]] = Field(default=None, description="The threshold for relevance check.", alias="RelevanceThreshold")
access_type: Optional[StrictStr] = Field(default=None, description="The access type of the guard rail.", alias="AccessType")
is_default: Optional[StrictBool] = Field(default=None, description="Whether the guard rail is the default guard rail.", alias="IsDefault")
__properties: ClassVar[List[str]] = ["GuardRailId", "GuardRailVersion", "GuardRailName", "Description", "CreationTime", "CreatedBy", "LastModifiedTime", "LastModifiedBy", "Tier", "IsCrossRegionEnabled", "Scope", "Owner", "BlockedMessage", "FiltersConfig", "DeniedTopics", "SensitiveInformationFilters", "CustomWordFilters", "EnableContextualGroundingCheck", "GroundingThreshold", "RelevanceThreshold", "AccessType", "IsDefault"]
[docs]
@field_validator('tier')
def tier_validate_enum(cls, value):
"""Validates the enum"""
if value is None:
return value
if value not in set(['CLASSIC', 'STANDARD']):
raise ValueError("must be one of enum values ('CLASSIC', 'STANDARD')")
return value
[docs]
@field_validator('scope')
def scope_validate_enum(cls, value):
"""Validates the enum"""
if value is None:
return value
if value not in set(['global', 'private']):
raise ValueError("must be one of enum values ('global', 'private')")
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 GuardRailDetail 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 filters_config (list)
_items = []
if self.filters_config:
for _item_filters_config in self.filters_config:
if _item_filters_config:
_items.append(_item_filters_config.to_dict())
_dict['FiltersConfig'] = _items
# override the default output from pydantic by calling `to_dict()` of each item in denied_topics (list)
_items = []
if self.denied_topics:
for _item_denied_topics in self.denied_topics:
if _item_denied_topics:
_items.append(_item_denied_topics.to_dict())
_dict['DeniedTopics'] = _items
# override the default output from pydantic by calling `to_dict()` of sensitive_information_filters
if self.sensitive_information_filters:
_dict['SensitiveInformationFilters'] = self.sensitive_information_filters.to_dict()
# override the default output from pydantic by calling `to_dict()` of custom_word_filters
if self.custom_word_filters:
_dict['CustomWordFilters'] = self.custom_word_filters.to_dict()
return _dict
[docs]
@classmethod
def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
"""Create an instance of GuardRailDetail from a dict"""
if obj is None:
return None
if not isinstance(obj, dict):
return cls.model_validate(obj)
_obj = cls.model_validate({
"GuardRailId": obj.get("GuardRailId"),
"GuardRailVersion": obj.get("GuardRailVersion"),
"GuardRailName": obj.get("GuardRailName"),
"Description": obj.get("Description"),
"CreationTime": obj.get("CreationTime"),
"CreatedBy": obj.get("CreatedBy"),
"LastModifiedTime": obj.get("LastModifiedTime"),
"LastModifiedBy": obj.get("LastModifiedBy"),
"Tier": obj.get("Tier"),
"IsCrossRegionEnabled": obj.get("IsCrossRegionEnabled"),
"Scope": obj.get("Scope"),
"Owner": obj.get("Owner"),
"BlockedMessage": obj.get("BlockedMessage"),
"FiltersConfig": [GuardRailFilterConfig.from_dict(_item) for _item in obj["FiltersConfig"]] if obj.get("FiltersConfig") is not None else None,
"DeniedTopics": [DeniedTopic.from_dict(_item) for _item in obj["DeniedTopics"]] if obj.get("DeniedTopics") is not None else None,
"SensitiveInformationFilters": SensitiveInformationFilters.from_dict(obj["SensitiveInformationFilters"]) if obj.get("SensitiveInformationFilters") is not None else None,
"CustomWordFilters": CustomWordFilters.from_dict(obj["CustomWordFilters"]) if obj.get("CustomWordFilters") is not None else None,
"EnableContextualGroundingCheck": obj.get("EnableContextualGroundingCheck"),
"GroundingThreshold": obj.get("GroundingThreshold"),
"RelevanceThreshold": obj.get("RelevanceThreshold"),
"AccessType": obj.get("AccessType"),
"IsDefault": obj.get("IsDefault")
})
return _obj