truefoundry 0.3.4rc1__py3-none-any.whl → 0.4.0__py3-none-any.whl

truefoundry/ml/autogen/client/models/__init__.py

@@ -0,0 +1,341 @@
+# coding: utf-8
+
+# flake8: noqa
+"""
+FastAPI
+
+No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+The version of the OpenAPI document: 0.1.0
+Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+Do not edit the class manually.
+"""  # noqa: E501
+
+# import models into model package
+from truefoundry.ml.autogen.client.models.add_custom_metrics_to_model_version_request_dto import (
+    AddCustomMetricsToModelVersionRequestDto,
+)
+from truefoundry.ml.autogen.client.models.add_features_to_model_version_request_dto import (
+    AddFeaturesToModelVersionRequestDto,
+)
+from truefoundry.ml.autogen.client.models.agent import Agent
+from truefoundry.ml.autogen.client.models.agent_app import AgentApp
+from truefoundry.ml.autogen.client.models.agent_open_api_tool import AgentOpenAPITool
+from truefoundry.ml.autogen.client.models.agent_open_api_tool_with_fqn import (
+    AgentOpenAPIToolWithFQN,
+)
+from truefoundry.ml.autogen.client.models.agent_with_fqn import AgentWithFQN
+from truefoundry.ml.autogen.client.models.artifact_dto import ArtifactDto
+from truefoundry.ml.autogen.client.models.artifact_response_dto import (
+    ArtifactResponseDto,
+)
+from truefoundry.ml.autogen.client.models.artifact_type import ArtifactType
+from truefoundry.ml.autogen.client.models.artifact_version_dto import ArtifactVersionDto
+from truefoundry.ml.autogen.client.models.artifact_version_response_dto import (
+    ArtifactVersionResponseDto,
+)
+from truefoundry.ml.autogen.client.models.artifact_version_status import (
+    ArtifactVersionStatus,
+)
+from truefoundry.ml.autogen.client.models.assistant_message import AssistantMessage
+from truefoundry.ml.autogen.client.models.authorize_user_for_model_request_dto import (
+    AuthorizeUserForModelRequestDto,
+)
+from truefoundry.ml.autogen.client.models.authorize_user_for_model_version_request_dto import (
+    AuthorizeUserForModelVersionRequestDto,
+)
+from truefoundry.ml.autogen.client.models.blob_storage_reference import (
+    BlobStorageReference,
+)
+from truefoundry.ml.autogen.client.models.body_get_search_runs_get import (
+    BodyGetSearchRunsGet,
+)
+from truefoundry.ml.autogen.client.models.chat_prompt import ChatPrompt
+from truefoundry.ml.autogen.client.models.chat_prompt_messages_inner import (
+    ChatPromptMessagesInner,
+)
+from truefoundry.ml.autogen.client.models.columns_dto import ColumnsDto
+from truefoundry.ml.autogen.client.models.content import Content
+from truefoundry.ml.autogen.client.models.content1 import Content1
+from truefoundry.ml.autogen.client.models.content2 import Content2
+from truefoundry.ml.autogen.client.models.content2_any_of_inner import (
+    Content2AnyOfInner,
+)
+from truefoundry.ml.autogen.client.models.create_artifact_request_dto import (
+    CreateArtifactRequestDto,
+)
+from truefoundry.ml.autogen.client.models.create_artifact_response_dto import (
+    CreateArtifactResponseDto,
+)
+from truefoundry.ml.autogen.client.models.create_artifact_version_request_dto import (
+    CreateArtifactVersionRequestDto,
+)
+from truefoundry.ml.autogen.client.models.create_artifact_version_response_dto import (
+    CreateArtifactVersionResponseDto,
+)
+from truefoundry.ml.autogen.client.models.create_dataset_request_dto import (
+    CreateDatasetRequestDto,
+)
+from truefoundry.ml.autogen.client.models.create_experiment_request_dto import (
+    CreateExperimentRequestDto,
+)
+from truefoundry.ml.autogen.client.models.create_experiment_response_dto import (
+    CreateExperimentResponseDto,
+)
+from truefoundry.ml.autogen.client.models.create_model_version_request_dto import (
+    CreateModelVersionRequestDto,
+)
+from truefoundry.ml.autogen.client.models.create_multi_part_upload_for_dataset_request_dto import (
+    CreateMultiPartUploadForDatasetRequestDto,
+)
+from truefoundry.ml.autogen.client.models.create_multi_part_upload_for_dataset_response_dto import (
+    CreateMultiPartUploadForDatasetResponseDto,
+)
+from truefoundry.ml.autogen.client.models.create_multi_part_upload_request_dto import (
+    CreateMultiPartUploadRequestDto,
+)
+from truefoundry.ml.autogen.client.models.create_python_deployment_config_request_dto import (
+    CreatePythonDeploymentConfigRequestDto,
+)
+from truefoundry.ml.autogen.client.models.create_python_deployment_config_response_dto import (
+    CreatePythonDeploymentConfigResponseDto,
+)
+from truefoundry.ml.autogen.client.models.create_run_request_dto import (
+    CreateRunRequestDto,
+)
+from truefoundry.ml.autogen.client.models.create_run_response_dto import (
+    CreateRunResponseDto,
+)
+from truefoundry.ml.autogen.client.models.dataset_dto import DatasetDto
+from truefoundry.ml.autogen.client.models.dataset_response_dto import DatasetResponseDto
+from truefoundry.ml.autogen.client.models.delete_artifact_versions_request_dto import (
+    DeleteArtifactVersionsRequestDto,
+)
+from truefoundry.ml.autogen.client.models.delete_dataset_request_dto import (
+    DeleteDatasetRequestDto,
+)
+from truefoundry.ml.autogen.client.models.delete_model_version_request_dto import (
+    DeleteModelVersionRequestDto,
+)
+from truefoundry.ml.autogen.client.models.delete_run_request import DeleteRunRequest
+from truefoundry.ml.autogen.client.models.delete_tag_request_dto import (
+    DeleteTagRequestDto,
+)
+from truefoundry.ml.autogen.client.models.experiment_dto import ExperimentDto
+from truefoundry.ml.autogen.client.models.experiment_id_request_dto import (
+    ExperimentIdRequestDto,
+)
+from truefoundry.ml.autogen.client.models.experiment_response_dto import (
+    ExperimentResponseDto,
+)
+from truefoundry.ml.autogen.client.models.experiment_tag_dto import ExperimentTagDto
+from truefoundry.ml.autogen.client.models.feature_dto import FeatureDto
+from truefoundry.ml.autogen.client.models.feature_value_type import FeatureValueType
+from truefoundry.ml.autogen.client.models.file_info_dto import FileInfoDto
+from truefoundry.ml.autogen.client.models.finalize_artifact_version_request_dto import (
+    FinalizeArtifactVersionRequestDto,
+)
+from truefoundry.ml.autogen.client.models.get_experiment_response_dto import (
+    GetExperimentResponseDto,
+)
+from truefoundry.ml.autogen.client.models.get_latest_run_log_response_dto import (
+    GetLatestRunLogResponseDto,
+)
+from truefoundry.ml.autogen.client.models.get_metric_history_response import (
+    GetMetricHistoryResponse,
+)
+from truefoundry.ml.autogen.client.models.get_signed_url_for_dataset_write_request_dto import (
+    GetSignedURLForDatasetWriteRequestDto,
+)
+from truefoundry.ml.autogen.client.models.get_signed_urls_for_artifact_version_read_request_dto import (
+    GetSignedURLsForArtifactVersionReadRequestDto,
+)
+from truefoundry.ml.autogen.client.models.get_signed_urls_for_artifact_version_read_response_dto import (
+    GetSignedURLsForArtifactVersionReadResponseDto,
+)
+from truefoundry.ml.autogen.client.models.get_signed_urls_for_artifact_version_write_request_dto import (
+    GetSignedURLsForArtifactVersionWriteRequestDto,
+)
+from truefoundry.ml.autogen.client.models.get_signed_urls_for_artifact_version_write_response_dto import (
+    GetSignedURLsForArtifactVersionWriteResponseDto,
+)
+from truefoundry.ml.autogen.client.models.get_signed_urls_for_dataset_read_request_dto import (
+    GetSignedURLsForDatasetReadRequestDto,
+)
+from truefoundry.ml.autogen.client.models.get_signed_urls_for_dataset_read_response_dto import (
+    GetSignedURLsForDatasetReadResponseDto,
+)
+from truefoundry.ml.autogen.client.models.get_signed_urls_for_dataset_write_response_dto import (
+    GetSignedURLsForDatasetWriteResponseDto,
+)
+from truefoundry.ml.autogen.client.models.get_tenant_id_response_dto import (
+    GetTenantIdResponseDto,
+)
+from truefoundry.ml.autogen.client.models.http_validation_error import (
+    HTTPValidationError,
+)
+from truefoundry.ml.autogen.client.models.image_content_part import ImageContentPart
+from truefoundry.ml.autogen.client.models.image_url import ImageUrl
+from truefoundry.ml.autogen.client.models.internal_metadata import InternalMetadata
+from truefoundry.ml.autogen.client.models.latest_run_log_dto import LatestRunLogDto
+from truefoundry.ml.autogen.client.models.list_artifact_versions_request_dto import (
+    ListArtifactVersionsRequestDto,
+)
+from truefoundry.ml.autogen.client.models.list_artifact_versions_response_dto import (
+    ListArtifactVersionsResponseDto,
+)
+from truefoundry.ml.autogen.client.models.list_artifacts_request_dto import (
+    ListArtifactsRequestDto,
+)
+from truefoundry.ml.autogen.client.models.list_artifacts_response_dto import (
+    ListArtifactsResponseDto,
+)
+from truefoundry.ml.autogen.client.models.list_colums_response_dto import (
+    ListColumsResponseDto,
+)
+from truefoundry.ml.autogen.client.models.list_datasets_request_dto import (
+    ListDatasetsRequestDto,
+)
+from truefoundry.ml.autogen.client.models.list_datasets_response_dto import (
+    ListDatasetsResponseDto,
+)
+from truefoundry.ml.autogen.client.models.list_experiments_response_dto import (
+    ListExperimentsResponseDto,
+)
+from truefoundry.ml.autogen.client.models.list_files_for_artifact_version_request_dto import (
+    ListFilesForArtifactVersionRequestDto,
+)
+from truefoundry.ml.autogen.client.models.list_files_for_artifact_versions_response_dto import (
+    ListFilesForArtifactVersionsResponseDto,
+)
+from truefoundry.ml.autogen.client.models.list_files_for_dataset_request_dto import (
+    ListFilesForDatasetRequestDto,
+)
+from truefoundry.ml.autogen.client.models.list_files_for_dataset_response_dto import (
+    ListFilesForDatasetResponseDto,
+)
+from truefoundry.ml.autogen.client.models.list_latest_run_logs_response_dto import (
+    ListLatestRunLogsResponseDto,
+)
+from truefoundry.ml.autogen.client.models.list_metric_history_request_dto import (
+    ListMetricHistoryRequestDto,
+)
+from truefoundry.ml.autogen.client.models.list_metric_history_response_dto import (
+    ListMetricHistoryResponseDto,
+)
+from truefoundry.ml.autogen.client.models.list_model_version_response_dto import (
+    ListModelVersionResponseDto,
+)
+from truefoundry.ml.autogen.client.models.list_model_versions_request_dto import (
+    ListModelVersionsRequestDto,
+)
+from truefoundry.ml.autogen.client.models.list_models_request_dto import (
+    ListModelsRequestDto,
+)
+from truefoundry.ml.autogen.client.models.list_models_response_dto import (
+    ListModelsResponseDto,
+)
+from truefoundry.ml.autogen.client.models.list_run_artifacts_response_dto import (
+    ListRunArtifactsResponseDto,
+)
+from truefoundry.ml.autogen.client.models.list_run_logs_response_dto import (
+    ListRunLogsResponseDto,
+)
+from truefoundry.ml.autogen.client.models.list_seed_experiments_response_dto import (
+    ListSeedExperimentsResponseDto,
+)
+from truefoundry.ml.autogen.client.models.log_batch_request_dto import (
+    LogBatchRequestDto,
+)
+from truefoundry.ml.autogen.client.models.log_metric_request_dto import (
+    LogMetricRequestDto,
+)
+from truefoundry.ml.autogen.client.models.log_param_request_dto import (
+    LogParamRequestDto,
+)
+from truefoundry.ml.autogen.client.models.method import Method
+from truefoundry.ml.autogen.client.models.metric_collection_dto import (
+    MetricCollectionDto,
+)
+from truefoundry.ml.autogen.client.models.metric_dto import MetricDto
+from truefoundry.ml.autogen.client.models.mime_type import MimeType
+from truefoundry.ml.autogen.client.models.model_configuration import ModelConfiguration
+from truefoundry.ml.autogen.client.models.model_dto import ModelDto
+from truefoundry.ml.autogen.client.models.model_response_dto import ModelResponseDto
+from truefoundry.ml.autogen.client.models.model_schema_dto import ModelSchemaDto
+from truefoundry.ml.autogen.client.models.model_version_dto import ModelVersionDto
+from truefoundry.ml.autogen.client.models.model_version_response_dto import (
+    ModelVersionResponseDto,
+)
+from truefoundry.ml.autogen.client.models.multi_part_upload_dto import (
+    MultiPartUploadDto,
+)
+from truefoundry.ml.autogen.client.models.multi_part_upload_response_dto import (
+    MultiPartUploadResponseDto,
+)
+from truefoundry.ml.autogen.client.models.multi_part_upload_storage_provider import (
+    MultiPartUploadStorageProvider,
+)
+from truefoundry.ml.autogen.client.models.notify_artifact_version_failure_dto import (
+    NotifyArtifactVersionFailureDto,
+)
+from truefoundry.ml.autogen.client.models.openapi_spec import OpenapiSpec
+from truefoundry.ml.autogen.client.models.param_dto import ParamDto
+from truefoundry.ml.autogen.client.models.parameters import Parameters
+from truefoundry.ml.autogen.client.models.prediction_type import PredictionType
+from truefoundry.ml.autogen.client.models.resolve_agent_app_response_dto import (
+    ResolveAgentAppResponseDto,
+)
+from truefoundry.ml.autogen.client.models.restore_run_request_dto import (
+    RestoreRunRequestDto,
+)
+from truefoundry.ml.autogen.client.models.run_data_dto import RunDataDto
+from truefoundry.ml.autogen.client.models.run_dto import RunDto
+from truefoundry.ml.autogen.client.models.run_info_dto import RunInfoDto
+from truefoundry.ml.autogen.client.models.run_log_dto import RunLogDto
+from truefoundry.ml.autogen.client.models.run_log_input_dto import RunLogInputDto
+from truefoundry.ml.autogen.client.models.run_response_dto import RunResponseDto
+from truefoundry.ml.autogen.client.models.run_tag_dto import RunTagDto
+from truefoundry.ml.autogen.client.models.search_runs_request_dto import (
+    SearchRunsRequestDto,
+)
+from truefoundry.ml.autogen.client.models.search_runs_response_dto import (
+    SearchRunsResponseDto,
+)
+from truefoundry.ml.autogen.client.models.set_experiment_tag_request_dto import (
+    SetExperimentTagRequestDto,
+)
+from truefoundry.ml.autogen.client.models.set_tag_request_dto import SetTagRequestDto
+from truefoundry.ml.autogen.client.models.signed_url_dto import SignedURLDto
+from truefoundry.ml.autogen.client.models.stop import Stop
+from truefoundry.ml.autogen.client.models.store_run_logs_request_dto import (
+    StoreRunLogsRequestDto,
+)
+from truefoundry.ml.autogen.client.models.system_message import SystemMessage
+from truefoundry.ml.autogen.client.models.text import Text
+from truefoundry.ml.autogen.client.models.text_content_part import TextContentPart
+from truefoundry.ml.autogen.client.models.update_artifact_version_request_dto import (
+    UpdateArtifactVersionRequestDto,
+)
+from truefoundry.ml.autogen.client.models.update_dataset_request_dto import (
+    UpdateDatasetRequestDto,
+)
+from truefoundry.ml.autogen.client.models.update_experiment_request_dto import (
+    UpdateExperimentRequestDto,
+)
+from truefoundry.ml.autogen.client.models.update_model_version_request_dto import (
+    UpdateModelVersionRequestDto,
+)
+from truefoundry.ml.autogen.client.models.update_run_request_dto import (
+    UpdateRunRequestDto,
+)
+from truefoundry.ml.autogen.client.models.update_run_response_dto import (
+    UpdateRunResponseDto,
+)
+from truefoundry.ml.autogen.client.models.url import Url
+from truefoundry.ml.autogen.client.models.user_message import UserMessage
+from truefoundry.ml.autogen.client.models.validation_error import ValidationError
+from truefoundry.ml.autogen.client.models.validation_error_loc_inner import (
+    ValidationErrorLocInner,
+)

truefoundry/ml/autogen/client/models/add_custom_metrics_to_model_version_request_dto.py

@@ -0,0 +1,69 @@
+# coding: utf-8
+
+"""
+FastAPI
+
+No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+The version of the OpenAPI document: 0.1.0
+Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+Do not edit the class manually.
+"""  # noqa: E501
+
+from __future__ import annotations
+
+import json
+import pprint
+import re  # noqa: F401
+from typing import Any, Dict, Optional
+
+from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, conlist
+
+
+class AddCustomMetricsToModelVersionRequestDto(BaseModel):
+    """
+    AddCustomMetricsToModelVersionRequestDto
+    """
+
+    id: StrictStr = Field(...)
+    custom_metrics: Optional[conlist(Dict[str, Any])] = None
+    __properties = ["id", "custom_metrics"]
+
+    class Config:
+        """Pydantic configuration"""
+
+        allow_population_by_field_name = True
+        validate_assignment = True
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.dict(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> AddCustomMetricsToModelVersionRequestDto:
+        """Create an instance of AddCustomMetricsToModelVersionRequestDto from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self):
+        """Returns the dictionary representation of the model using alias"""
+        _dict = self.dict(by_alias=True, exclude={}, exclude_none=True)
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> AddCustomMetricsToModelVersionRequestDto:
+        """Create an instance of AddCustomMetricsToModelVersionRequestDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return AddCustomMetricsToModelVersionRequestDto.parse_obj(obj)
+
+        _obj = AddCustomMetricsToModelVersionRequestDto.parse_obj(
+            {"id": obj.get("id"), "custom_metrics": obj.get("custom_metrics")}
+        )
+        return _obj

truefoundry/ml/autogen/client/models/add_features_to_model_version_request_dto.py

@@ -0,0 +1,83 @@
+# coding: utf-8
+
+"""
+FastAPI
+
+No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+The version of the OpenAPI document: 0.1.0
+Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+Do not edit the class manually.
+"""  # noqa: E501
+
+from __future__ import annotations
+
+import json
+import pprint
+import re  # noqa: F401
+
+from truefoundry.ml.autogen.client.models.feature_dto import FeatureDto
+from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, conlist
+
+
+class AddFeaturesToModelVersionRequestDto(BaseModel):
+    """
+    AddFeaturesToModelVersionRequestDto
+    """
+
+    id: StrictStr = Field(...)
+    features: conlist(FeatureDto) = Field(...)
+    __properties = ["id", "features"]
+
+    class Config:
+        """Pydantic configuration"""
+
+        allow_population_by_field_name = True
+        validate_assignment = True
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.dict(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> AddFeaturesToModelVersionRequestDto:
+        """Create an instance of AddFeaturesToModelVersionRequestDto from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self):
+        """Returns the dictionary representation of the model using alias"""
+        _dict = self.dict(by_alias=True, exclude={}, exclude_none=True)
+        # override the default output from truefoundry.pydantic_v1 by calling `to_dict()` of each item in features (list)
+        _items = []
+        if self.features:
+            for _item in self.features:
+                if _item:
+                    _items.append(_item.to_dict())
+            _dict["features"] = _items
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> AddFeaturesToModelVersionRequestDto:
+        """Create an instance of AddFeaturesToModelVersionRequestDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return AddFeaturesToModelVersionRequestDto.parse_obj(obj)
+
+        _obj = AddFeaturesToModelVersionRequestDto.parse_obj(
+            {
+                "id": obj.get("id"),
+                "features": [
+                    FeatureDto.from_dict(_item) for _item in obj.get("features")
+                ]
+                if obj.get("features") is not None
+                else None,
+            }
+        )
+        return _obj

truefoundry/ml/autogen/client/models/agent.py

@@ -0,0 +1,125 @@
+# coding: utf-8
+
+"""
+FastAPI
+
+No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+The version of the OpenAPI document: 0.1.0
+Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+Do not edit the class manually.
+"""  # noqa: E501
+
+from __future__ import annotations
+
+import json
+import pprint
+import re  # noqa: F401
+from typing import Optional
+
+from truefoundry.pydantic_v1 import (
+    BaseModel,
+    Field,
+    StrictStr,
+    conlist,
+    constr,
+    validator,
+)
+
+
+class Agent(BaseModel):
+    """
+    Agent artifact.  # noqa: E501
+    """
+
+    type: Optional[StrictStr] = "agent"
+    name: constr(strict=True) = Field(
+        default=..., description="Names cannot repeat accross agent +uiType=Hidden"
+    )
+    available_tools: conlist(constr(strict=True, min_length=1)) = Field(
+        default=..., description="+sort=30 +uiType=Hidden"
+    )
+    goal: constr(strict=True, max_length=128, min_length=1) = Field(
+        default=..., description="+sort=10 +uiType=TextArea"
+    )
+    instruction: constr(strict=True, max_length=2620, min_length=1) = Field(
+        default=...,
+        description='`instruction` is the system prompt for now. (2.5 * 1024) +sort=20 +uiType=AgentInstructions +uiProps={"helpText":"Use the syntax ${Tool FQN} to reference a tool, and ${AGENT FQN} to reference another agent"}',
+    )
+    model_id: constr(strict=True, min_length=1) = Field(
+        default=...,
+        description='+sort=40 +uiType=EnabledModelSelector +uiProps={"searchable":true,"modelType":"chat","providerType":"openai"}',
+    )
+    __properties = [
+        "type",
+        "name",
+        "available_tools",
+        "goal",
+        "instruction",
+        "model_id",
+    ]
+
+    @validator("type")
+    def type_validate_enum(cls, value):
+        """Validates the enum"""
+        if value is None:
+            return value
+
+        if value not in ("agent"):
+            raise ValueError("must be one of enum values ('agent')")
+        return value
+
+    @validator("name")
+    def name_validate_regular_expression(cls, value):
+        """Validates the regular expression"""
+        if not re.match(r"^[a-zA-Z][a-zA-Z0-9\-]{1,30}[a-zA-Z0-9]$", value):
+            raise ValueError(
+                r"must validate the regular expression /^[a-zA-Z][a-zA-Z0-9\-]{1,30}[a-zA-Z0-9]$/"
+            )
+        return value
+
+    class Config:
+        """Pydantic configuration"""
+
+        allow_population_by_field_name = True
+        validate_assignment = True
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.dict(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Agent:
+        """Create an instance of Agent from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self):
+        """Returns the dictionary representation of the model using alias"""
+        _dict = self.dict(by_alias=True, exclude={}, exclude_none=True)
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> Agent:
+        """Create an instance of Agent from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return Agent.parse_obj(obj)
+
+        _obj = Agent.parse_obj(
+            {
+                "type": obj.get("type") if obj.get("type") is not None else "agent",
+                "name": obj.get("name"),
+                "available_tools": obj.get("available_tools"),
+                "goal": obj.get("goal"),
+                "instruction": obj.get("instruction"),
+                "model_id": obj.get("model_id"),
+            }
+        )
+        return _obj