# 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, 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 openapi_client.models.create_agent_request_body_resource_access_metadata import CreateAgentRequestBodyResourceAccessMetadata
from openapi_client.models.file_add_delete_file_level_tags_inner import FileAddDeleteFileLevelTagsInner
from typing import Optional, Set
from typing_extensions import Self
[docs]
class CreateAgentRequestBody(BaseModel):
"""
CreateAgentRequestBody
""" # noqa: E501
agent_id: Annotated[str, Field(strict=True)] = Field(description="Unique identifier for the agent", alias="AgentId")
agent_name: Annotated[str, Field(strict=True)] = Field(description="Name of the agent", alias="AgentName")
description: Annotated[str, Field(strict=True)] = Field(description="Description of the agent", alias="Description")
instruction: Annotated[str, Field(strict=True)] = Field(description="Instructions for the agent", alias="Instruction")
model_id: Annotated[str, Field(strict=True)] = Field(description="ID of the AI model to use", alias="ModelId")
agent_response_type: Optional[Annotated[str, Field(strict=True)]] = Field(default=None, description="Type of response format (optional)", alias="AgentResponseType")
tools: Annotated[List[AgentToolObject], Field(max_length=10)] = Field(description="List of tools available to the agent", alias="Tools")
cost_tags: Optional[List[FileAddDeleteFileLevelTagsInner]] = Field(default=None, alias="CostTags")
resource_access_metadata: Optional[CreateAgentRequestBodyResourceAccessMetadata] = Field(default=None, alias="ResourceAccessMetadata")
__properties: ClassVar[List[str]] = ["AgentId", "AgentName", "Description", "Instruction", "ModelId", "AgentResponseType", "Tools", "CostTags", "ResourceAccessMetadata"]
[docs]
@field_validator('agent_id')
def agent_id_validate_regular_expression(cls, value):
"""Validates the regular expression"""
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 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 not re.match(r".*", value):
raise ValueError(r"must validate the regular expression /.*/")
return value
[docs]
@field_validator('instruction')
def instruction_validate_regular_expression(cls, value):
"""Validates the regular expression"""
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 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
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 CreateAgentRequestBody 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
# override the default output from pydantic by calling `to_dict()` of each item in cost_tags (list)
_items = []
if self.cost_tags:
for _item_cost_tags in self.cost_tags:
if _item_cost_tags:
_items.append(_item_cost_tags.to_dict())
_dict['CostTags'] = _items
# override the default output from pydantic by calling `to_dict()` of resource_access_metadata
if self.resource_access_metadata:
_dict['ResourceAccessMetadata'] = self.resource_access_metadata.to_dict()
return _dict
[docs]
@classmethod
def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
"""Create an instance of CreateAgentRequestBody 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"),
"Description": obj.get("Description"),
"Instruction": obj.get("Instruction"),
"ModelId": obj.get("ModelId"),
"AgentResponseType": obj.get("AgentResponseType"),
"Tools": [AgentToolObject.from_dict(_item) for _item in obj["Tools"]] if obj.get("Tools") is not None else None,
"CostTags": [FileAddDeleteFileLevelTagsInner.from_dict(_item) for _item in obj["CostTags"]] if obj.get("CostTags") is not None else None,
"ResourceAccessMetadata": CreateAgentRequestBodyResourceAccessMetadata.from_dict(obj["ResourceAccessMetadata"]) if obj.get("ResourceAccessMetadata") is not None else None
})
return _obj