Source code for openapi_client.models.agent_details

# 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, StrictBool, field_validator
from typing import Any, ClassVar, Dict, List, Optional
from typing_extensions import Annotated
from openapi_client.models.agent_tool_object import AgentToolObject
from typing import Optional, Set
from typing_extensions import Self

[docs] class AgentDetails(BaseModel): """ Details for a single AI agent. """ # noqa: E501 agent_id: Optional[Annotated[str, Field(strict=True)]] = Field(default=None, description="Unique identifier of the agent.", alias="AgentId") agent_name: Optional[Annotated[str, Field(strict=True)]] = Field(default=None, description="Name of the agent.", alias="AgentName") agent_status: Optional[Annotated[str, Field(strict=True)]] = Field(default=None, description="Current status of the agent.", alias="AgentStatus") description: Optional[Annotated[str, Field(strict=True)]] = Field(default=None, description="Description of the agent.", alias="Description") agent_type: Optional[Annotated[str, Field(strict=True)]] = Field(default=None, description="Type of the agent (e.g., 'standard', 'custom').", alias="AgentType") agent_version: Optional[Annotated[str, Field(strict=True)]] = Field(default=None, description="Version of the agent.", alias="AgentVersion") model_id: Optional[Annotated[str, Field(strict=True)]] = Field(default=None, description="ID of the AI model used by the agent.", alias="ModelId") reference_id: Optional[Annotated[str, Field(strict=True)]] = Field(default=None, description="Reference ID associated with the agent.", alias="ReferenceId") agent_alias_id: Optional[Annotated[str, Field(strict=True)]] = Field(default=None, description="Alias ID for the agent.", alias="AgentAliasId") message: Optional[Annotated[str, Field(strict=True)]] = Field(default=None, description="Additional message or information about the agent.", alias="Message") access_type: Optional[Annotated[str, Field(strict=True)]] = Field(default=None, description="Access type of the agent (e.g., 'owner', 'shared').", alias="AccessType") created_by: Optional[Annotated[str, Field(strict=True)]] = Field(default=None, description="ID of the user who created the agent.", alias="CreatedBy") creation_time: Optional[Annotated[str, Field(strict=True)]] = Field(default=None, description="Timestamp when the agent was created.", alias="CreationTime") last_modified_by: Optional[Annotated[str, Field(strict=True)]] = Field(default=None, description="ID of the user who last modified the agent.", alias="LastModifiedBy") last_modified_time: Optional[Annotated[str, Field(strict=True)]] = Field(default=None, description="Timestamp when the agent was last modified.", alias="LastModifiedTime") is_enabled: Optional[StrictBool] = Field(default=None, description="Indicates whether the agent is enabled.", alias="IsEnabled") agent_response_type: Optional[Annotated[str, Field(strict=True)]] = Field(default=None, description="Type of response format used by the agent.", alias="AgentResponseType") action_group_id: Optional[Annotated[str, Field(strict=True)]] = Field(default=None, description="Action group ID associated with the agent.", alias="ActionGroupId") tools: Optional[Annotated[List[AgentToolObject], Field(max_length=10)]] = Field(default=None, description="List of tools available to the agent.", alias="Tools") __properties: ClassVar[List[str]] = ["AgentId", "AgentName", "AgentStatus", "Description", "AgentType", "AgentVersion", "ModelId", "ReferenceId", "AgentAliasId", "Message", "AccessType", "CreatedBy", "CreationTime", "LastModifiedBy", "LastModifiedTime", "IsEnabled", "AgentResponseType", "ActionGroupId", "Tools"]
[docs] @field_validator('agent_id') def agent_id_validate_regular_expression(cls, value): """Validates the regular expression""" if value is None: return value if not re.match(r"^[A-Za-z0-9:_\-]+$", value): raise ValueError(r"must validate the regular expression /^[A-Za-z0-9:_\-]+$/") return value
[docs] @field_validator('agent_name') def agent_name_validate_regular_expression(cls, value): """Validates the regular expression""" if value is None: return value if not re.match(r".*", value): raise ValueError(r"must validate the regular expression /.*/") return value
[docs] @field_validator('agent_status') def agent_status_validate_regular_expression(cls, value): """Validates the regular expression""" if value is None: return value if not re.match(r".*", value): raise ValueError(r"must validate the regular expression /.*/") return value
[docs] @field_validator('description') def description_validate_regular_expression(cls, value): """Validates the regular expression""" if value is None: return value if not re.match(r".*", value): raise ValueError(r"must validate the regular expression /.*/") return value
[docs] @field_validator('agent_type') def agent_type_validate_regular_expression(cls, value): """Validates the regular expression""" if value is None: return value if not re.match(r".*", value): raise ValueError(r"must validate the regular expression /.*/") return value
[docs] @field_validator('agent_version') def agent_version_validate_regular_expression(cls, value): """Validates the regular expression""" if value is None: return value if not re.match(r".*", value): raise ValueError(r"must validate the regular expression /.*/") return value
[docs] @field_validator('model_id') def model_id_validate_regular_expression(cls, value): """Validates the regular expression""" if value is None: return value if not re.match(r".*", value): raise ValueError(r"must validate the regular expression /.*/") return value
[docs] @field_validator('reference_id') def reference_id_validate_regular_expression(cls, value): """Validates the regular expression""" if value is None: return value if not re.match(r".*", value): raise ValueError(r"must validate the regular expression /.*/") return value
[docs] @field_validator('agent_alias_id') def agent_alias_id_validate_regular_expression(cls, value): """Validates the regular expression""" if value is None: return value if not re.match(r".*", value): raise ValueError(r"must validate the regular expression /.*/") return value
[docs] @field_validator('message') def message_validate_regular_expression(cls, value): """Validates the regular expression""" if value is None: return value if not re.match(r".*", value): raise ValueError(r"must validate the regular expression /.*/") return value
[docs] @field_validator('access_type') def access_type_validate_regular_expression(cls, value): """Validates the regular expression""" if value is None: return value if not re.match(r".*", value): raise ValueError(r"must validate the regular expression /.*/") return value
[docs] @field_validator('created_by') def created_by_validate_regular_expression(cls, value): """Validates the regular expression""" if value is None: return value if not re.match(r".*", value): raise ValueError(r"must validate the regular expression /.*/") return value
[docs] @field_validator('creation_time') def creation_time_validate_regular_expression(cls, value): """Validates the regular expression""" if value is None: return value if not re.match(r".*", value): raise ValueError(r"must validate the regular expression /.*/") return value
[docs] @field_validator('last_modified_by') def last_modified_by_validate_regular_expression(cls, value): """Validates the regular expression""" if value is None: return value if not re.match(r".*", value): raise ValueError(r"must validate the regular expression /.*/") return value
[docs] @field_validator('last_modified_time') def last_modified_time_validate_regular_expression(cls, value): """Validates the regular expression""" if value is None: return value if not re.match(r".*", value): raise ValueError(r"must validate the regular expression /.*/") return value
[docs] @field_validator('agent_response_type') def agent_response_type_validate_regular_expression(cls, value): """Validates the regular expression""" if value is None: return value if not re.match(r".*", value): raise ValueError(r"must validate the regular expression /.*/") return value
[docs] @field_validator('action_group_id') def action_group_id_validate_regular_expression(cls, value): """Validates the regular expression""" if value is None: return value if not re.match(r".*", value): raise ValueError(r"must validate the regular expression /.*/") 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 AgentDetails 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 tools (list) _items = [] if self.tools: for _item_tools in self.tools: if _item_tools: _items.append(_item_tools.to_dict()) _dict['Tools'] = _items return _dict
[docs] @classmethod def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]: """Create an instance of AgentDetails from a dict""" if obj is None: return None if not isinstance(obj, dict): return cls.model_validate(obj) _obj = cls.model_validate({ "AgentId": obj.get("AgentId"), "AgentName": obj.get("AgentName"), "AgentStatus": obj.get("AgentStatus"), "Description": obj.get("Description"), "AgentType": obj.get("AgentType"), "AgentVersion": obj.get("AgentVersion"), "ModelId": obj.get("ModelId"), "ReferenceId": obj.get("ReferenceId"), "AgentAliasId": obj.get("AgentAliasId"), "Message": obj.get("Message"), "AccessType": obj.get("AccessType"), "CreatedBy": obj.get("CreatedBy"), "CreationTime": obj.get("CreationTime"), "LastModifiedBy": obj.get("LastModifiedBy"), "LastModifiedTime": obj.get("LastModifiedTime"), "IsEnabled": obj.get("IsEnabled"), "AgentResponseType": obj.get("AgentResponseType"), "ActionGroupId": obj.get("ActionGroupId"), "Tools": [AgentToolObject.from_dict(_item) for _item in obj["Tools"]] if obj.get("Tools") is not None else None }) return _obj