# 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, StrictInt, StrictStr, field_validator
from typing import Any, ClassVar, Dict, List, Optional
from typing_extensions import Annotated
from openapi_client.models.agent_details import AgentDetails
from typing import Optional, Set
from typing_extensions import Self
[docs]
class ListAgentsResponse(BaseModel):
"""
Response for GET /ai/agents endpoint. Structure varies based on query parameters. See endpoint description for details.
""" # noqa: E501
agents: Optional[List[AgentDetails]] = None
message: Optional[Annotated[str, Field(strict=True)]] = Field(default=None, description="Status message from the API.", alias="Message")
presigned_url: Optional[Annotated[str, Field(strict=True)]] = Field(default=None, description="Generated presigned URL (present when action=generate_presigned_url)", alias="PresignedUrl")
agent_id: Optional[Annotated[str, Field(strict=True)]] = Field(default=None, description="ID of the created agent", alias="AgentId")
utility_tool: Optional[Annotated[str, Field(strict=True)]] = Field(default=None, description="Output for action=get_sample_templates: a JSON object where keys are template filenames and values are the script contents.", alias="utilityTool")
http_tool: Optional[Annotated[str, Field(strict=True)]] = Field(default=None, description="Output for action=get_sample_templates: a JSON object where keys are template filenames and values are the script contents.", alias="httpTool")
processor_tool: Optional[Annotated[str, Field(strict=True)]] = Field(default=None, description="Output for action=get_sample_templates: a JSON object where keys are template filenames and values are the script contents.", alias="processorTool")
count: Optional[StrictInt] = Field(default=None, description="Number of agents in current page (present when no action parameter)")
total_count: Optional[StrictInt] = Field(default=None, description="Total number of agents available (present when no action parameter)")
next_available: Optional[StrictStr] = Field(default=None, description="Whether more pages are available (present when no action parameter)")
__properties: ClassVar[List[str]] = ["agents", "Message", "PresignedUrl", "AgentId", "utilityTool", "httpTool", "processorTool", "count", "total_count", "next_available"]
[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('presigned_url')
def presigned_url_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_id')
def agent_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 ListAgentsResponse 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 agents (list)
_items = []
if self.agents:
for _item_agents in self.agents:
if _item_agents:
_items.append(_item_agents.to_dict())
_dict['agents'] = _items
return _dict
[docs]
@classmethod
def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
"""Create an instance of ListAgentsResponse from a dict"""
if obj is None:
return None
if not isinstance(obj, dict):
return cls.model_validate(obj)
_obj = cls.model_validate({
"agents": [AgentDetails.from_dict(_item) for _item in obj["agents"]] if obj.get("agents") is not None else None,
"Message": obj.get("Message"),
"PresignedUrl": obj.get("PresignedUrl"),
"AgentId": obj.get("AgentId"),
"utilityTool": obj.get("utilityTool"),
"httpTool": obj.get("httpTool"),
"processorTool": obj.get("processorTool"),
"count": obj.get("count"),
"total_count": obj.get("total_count"),
"next_available": obj.get("next_available")
})
return _obj