truefoundry 0.4.10__py3-none-any.whl → 0.5.0__py3-none-any.whl

truefoundry/ml/autogen/client/api/mlfoundry_artifacts_api.py

@@ -20,12 +20,6 @@ from truefoundry.ml.autogen.client.exceptions import (  # noqa: F401
     ApiTypeError,
     ApiValueError,
 )
-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.artifact_response_dto import (
     ArtifactResponseDto,
 )
@@ -189,322 +183,6 @@ class MlfoundryArtifactsApi:
             api_client = ApiClient.get_default()
         self.api_client = api_client
 
-    @validate_arguments
-    def add_custom_metrics_to_model_version_post(
-        self,
-        add_custom_metrics_to_model_version_request_dto: AddCustomMetricsToModelVersionRequestDto,
-        **kwargs,
-    ) -> ModelVersionResponseDto:  # noqa: E501
-        """Add Custom Metrics To Model Version  # noqa: E501
-
-        This method makes a synchronous HTTP request by default. To make an
-        asynchronous HTTP request, please pass async_req=True
-
-        >>> thread = api.add_custom_metrics_to_model_version_post(add_custom_metrics_to_model_version_request_dto, async_req=True)
-        >>> result = thread.get()
-
-        :param add_custom_metrics_to_model_version_request_dto: (required)
-        :type add_custom_metrics_to_model_version_request_dto: AddCustomMetricsToModelVersionRequestDto
-        :param async_req: Whether to execute the request asynchronously.
-        :type async_req: bool, optional
-        :param _request_timeout: timeout setting for this request.
-               If one number provided, it will be total request
-               timeout. It can also be a pair (tuple) of
-               (connection, read) timeouts.
-        :return: Returns the result object.
-                 If the method is called asynchronously,
-                 returns the request thread.
-        :rtype: ModelVersionResponseDto
-        """
-        kwargs["_return_http_data_only"] = True
-        if "_preload_content" in kwargs:
-            message = "Error! Please call the add_custom_metrics_to_model_version_post_with_http_info method with `_preload_content` instead and obtain raw data from ApiResponse.raw_data"  # noqa: E501
-            raise ValueError(message)
-        return self.add_custom_metrics_to_model_version_post_with_http_info(
-            add_custom_metrics_to_model_version_request_dto, **kwargs
-        )  # noqa: E501
-
-    @validate_arguments
-    def add_custom_metrics_to_model_version_post_with_http_info(
-        self,
-        add_custom_metrics_to_model_version_request_dto: AddCustomMetricsToModelVersionRequestDto,
-        **kwargs,
-    ) -> ApiResponse:  # noqa: E501
-        """Add Custom Metrics To Model Version  # noqa: E501
-
-        This method makes a synchronous HTTP request by default. To make an
-        asynchronous HTTP request, please pass async_req=True
-
-        >>> thread = api.add_custom_metrics_to_model_version_post_with_http_info(add_custom_metrics_to_model_version_request_dto, async_req=True)
-        >>> result = thread.get()
-
-        :param add_custom_metrics_to_model_version_request_dto: (required)
-        :type add_custom_metrics_to_model_version_request_dto: AddCustomMetricsToModelVersionRequestDto
-        :param async_req: Whether to execute the request asynchronously.
-        :type async_req: bool, optional
-        :param _preload_content: if False, the ApiResponse.data will
-                                 be set to none and raw_data will store the
-                                 HTTP response body without reading/decoding.
-                                 Default is True.
-        :type _preload_content: bool, optional
-        :param _return_http_data_only: response data instead of ApiResponse
-                                       object with status code, headers, etc
-        :type _return_http_data_only: bool, optional
-        :param _request_timeout: timeout setting for this request. If one
-                                 number provided, it will be total request
-                                 timeout. It can also be a pair (tuple) of
-                                 (connection, read) timeouts.
-        :param _request_auth: set to override the auth_settings for an a single
-                              request; this effectively ignores the authentication
-                              in the spec for a single request.
-        :type _request_auth: dict, optional
-        :type _content_type: string, optional: force content-type for the request
-        :return: Returns the result object.
-                 If the method is called asynchronously,
-                 returns the request thread.
-        :rtype: tuple(ModelVersionResponseDto, status_code(int), headers(HTTPHeaderDict))
-        """
-
-        _params = locals()
-
-        _all_params = ["add_custom_metrics_to_model_version_request_dto"]
-        _all_params.extend(
-            [
-                "async_req",
-                "_return_http_data_only",
-                "_preload_content",
-                "_request_timeout",
-                "_request_auth",
-                "_content_type",
-                "_headers",
-            ]
-        )
-
-        # validate the arguments
-        for _key, _val in _params["kwargs"].items():
-            if _key not in _all_params:
-                raise ApiTypeError(
-                    "Got an unexpected keyword argument '%s'"
-                    " to method add_custom_metrics_to_model_version_post" % _key
-                )
-            _params[_key] = _val
-        del _params["kwargs"]
-
-        _collection_formats = {}
-
-        # process the path parameters
-        _path_params = {}
-
-        # process the query parameters
-        _query_params = []
-        # process the header parameters
-        _header_params = dict(_params.get("_headers", {}))
-        # process the form parameters
-        _form_params = []
-        _files = {}
-        # process the body parameter
-        _body_params = None
-        if _params["add_custom_metrics_to_model_version_request_dto"] is not None:
-            _body_params = _params["add_custom_metrics_to_model_version_request_dto"]
-
-        # set the HTTP header `Accept`
-        _header_params["Accept"] = self.api_client.select_header_accept(
-            ["application/json"]
-        )  # noqa: E501
-
-        # set the HTTP header `Content-Type`
-        _content_types_list = _params.get(
-            "_content_type",
-            self.api_client.select_header_content_type(["application/json"]),
-        )
-        if _content_types_list:
-            _header_params["Content-Type"] = _content_types_list
-
-        # authentication setting
-        _auth_settings = ["HTTPBearer", "APIKeyCookie"]  # noqa: E501
-
-        _response_types_map = {
-            "200": "ModelVersionResponseDto",
-            "422": "HTTPValidationError",
-        }
-
-        return self.api_client.call_api(
-            "/api/2.0/mlflow/mlfoundry-artifacts/model-versions/custom-metrics/add",
-            "POST",
-            _path_params,
-            _query_params,
-            _header_params,
-            body=_body_params,
-            post_params=_form_params,
-            files=_files,
-            response_types_map=_response_types_map,
-            auth_settings=_auth_settings,
-            async_req=_params.get("async_req"),
-            _return_http_data_only=_params.get("_return_http_data_only"),  # noqa: E501
-            _preload_content=_params.get("_preload_content", True),
-            _request_timeout=_params.get("_request_timeout"),
-            collection_formats=_collection_formats,
-            _request_auth=_params.get("_request_auth"),
-        )
-
-    @validate_arguments
-    def add_features_to_model_version_post(
-        self,
-        add_features_to_model_version_request_dto: AddFeaturesToModelVersionRequestDto,
-        **kwargs,
-    ) -> ModelVersionResponseDto:  # noqa: E501
-        """Add Features To Model Version  # noqa: E501
-
-        This method makes a synchronous HTTP request by default. To make an
-        asynchronous HTTP request, please pass async_req=True
-
-        >>> thread = api.add_features_to_model_version_post(add_features_to_model_version_request_dto, async_req=True)
-        >>> result = thread.get()
-
-        :param add_features_to_model_version_request_dto: (required)
-        :type add_features_to_model_version_request_dto: AddFeaturesToModelVersionRequestDto
-        :param async_req: Whether to execute the request asynchronously.
-        :type async_req: bool, optional
-        :param _request_timeout: timeout setting for this request.
-               If one number provided, it will be total request
-               timeout. It can also be a pair (tuple) of
-               (connection, read) timeouts.
-        :return: Returns the result object.
-                 If the method is called asynchronously,
-                 returns the request thread.
-        :rtype: ModelVersionResponseDto
-        """
-        kwargs["_return_http_data_only"] = True
-        if "_preload_content" in kwargs:
-            message = "Error! Please call the add_features_to_model_version_post_with_http_info method with `_preload_content` instead and obtain raw data from ApiResponse.raw_data"  # noqa: E501
-            raise ValueError(message)
-        return self.add_features_to_model_version_post_with_http_info(
-            add_features_to_model_version_request_dto, **kwargs
-        )  # noqa: E501
-
-    @validate_arguments
-    def add_features_to_model_version_post_with_http_info(
-        self,
-        add_features_to_model_version_request_dto: AddFeaturesToModelVersionRequestDto,
-        **kwargs,
-    ) -> ApiResponse:  # noqa: E501
-        """Add Features To Model Version  # noqa: E501
-
-        This method makes a synchronous HTTP request by default. To make an
-        asynchronous HTTP request, please pass async_req=True
-
-        >>> thread = api.add_features_to_model_version_post_with_http_info(add_features_to_model_version_request_dto, async_req=True)
-        >>> result = thread.get()
-
-        :param add_features_to_model_version_request_dto: (required)
-        :type add_features_to_model_version_request_dto: AddFeaturesToModelVersionRequestDto
-        :param async_req: Whether to execute the request asynchronously.
-        :type async_req: bool, optional
-        :param _preload_content: if False, the ApiResponse.data will
-                                 be set to none and raw_data will store the
-                                 HTTP response body without reading/decoding.
-                                 Default is True.
-        :type _preload_content: bool, optional
-        :param _return_http_data_only: response data instead of ApiResponse
-                                       object with status code, headers, etc
-        :type _return_http_data_only: bool, optional
-        :param _request_timeout: timeout setting for this request. If one
-                                 number provided, it will be total request
-                                 timeout. It can also be a pair (tuple) of
-                                 (connection, read) timeouts.
-        :param _request_auth: set to override the auth_settings for an a single
-                              request; this effectively ignores the authentication
-                              in the spec for a single request.
-        :type _request_auth: dict, optional
-        :type _content_type: string, optional: force content-type for the request
-        :return: Returns the result object.
-                 If the method is called asynchronously,
-                 returns the request thread.
-        :rtype: tuple(ModelVersionResponseDto, status_code(int), headers(HTTPHeaderDict))
-        """
-
-        _params = locals()
-
-        _all_params = ["add_features_to_model_version_request_dto"]
-        _all_params.extend(
-            [
-                "async_req",
-                "_return_http_data_only",
-                "_preload_content",
-                "_request_timeout",
-                "_request_auth",
-                "_content_type",
-                "_headers",
-            ]
-        )
-
-        # validate the arguments
-        for _key, _val in _params["kwargs"].items():
-            if _key not in _all_params:
-                raise ApiTypeError(
-                    "Got an unexpected keyword argument '%s'"
-                    " to method add_features_to_model_version_post" % _key
-                )
-            _params[_key] = _val
-        del _params["kwargs"]
-
-        _collection_formats = {}
-
-        # process the path parameters
-        _path_params = {}
-
-        # process the query parameters
-        _query_params = []
-        # process the header parameters
-        _header_params = dict(_params.get("_headers", {}))
-        # process the form parameters
-        _form_params = []
-        _files = {}
-        # process the body parameter
-        _body_params = None
-        if _params["add_features_to_model_version_request_dto"] is not None:
-            _body_params = _params["add_features_to_model_version_request_dto"]
-
-        # set the HTTP header `Accept`
-        _header_params["Accept"] = self.api_client.select_header_accept(
-            ["application/json"]
-        )  # noqa: E501
-
-        # set the HTTP header `Content-Type`
-        _content_types_list = _params.get(
-            "_content_type",
-            self.api_client.select_header_content_type(["application/json"]),
-        )
-        if _content_types_list:
-            _header_params["Content-Type"] = _content_types_list
-
-        # authentication setting
-        _auth_settings = ["HTTPBearer", "APIKeyCookie"]  # noqa: E501
-
-        _response_types_map = {
-            "200": "ModelVersionResponseDto",
-            "422": "HTTPValidationError",
-        }
-
-        return self.api_client.call_api(
-            "/api/2.0/mlflow/mlfoundry-artifacts/model-versions/features/add",
-            "POST",
-            _path_params,
-            _query_params,
-            _header_params,
-            body=_body_params,
-            post_params=_form_params,
-            files=_files,
-            response_types_map=_response_types_map,
-            auth_settings=_auth_settings,
-            async_req=_params.get("async_req"),
-            _return_http_data_only=_params.get("_return_http_data_only"),  # noqa: E501
-            _preload_content=_params.get("_preload_content", True),
-            _request_timeout=_params.get("_request_timeout"),
-            collection_formats=_collection_formats,
-            _request_auth=_params.get("_request_auth"),
-        )
-
     @validate_arguments
     def authorize_user_for_model_post(
         self,

truefoundry/ml/autogen/client/api_client.py

@@ -28,6 +28,7 @@ from truefoundry.ml.autogen.client import rest
 from truefoundry.ml.autogen.client.api_response import ApiResponse
 from truefoundry.ml.autogen.client.configuration import Configuration
 from truefoundry.ml.autogen.client.exceptions import ApiException, ApiValueError
+from truefoundry.pydantic_v1 import SecretStr
 
 
 class ApiClient:
@@ -288,6 +289,7 @@ class ApiClient:
         """Builds a JSON POST object.
 
         If obj is None, return None.
+        If obj is SecretStr, return obj.get_secret_value()
         If obj is str, int, long, float, bool, return directly.
         If obj is datetime.datetime, datetime.date
             convert to string in iso8601 format.
@@ -300,6 +302,8 @@ class ApiClient:
         """
         if obj is None:
             return None
+        elif isinstance(obj, SecretStr):
+            return obj.get_secret_value()
         elif isinstance(obj, self.PRIMITIVE_TYPES):
             return obj
         elif isinstance(obj, list):
@@ -317,7 +321,10 @@ class ApiClient:
             # and attributes which value is not None.
             # Convert attribute name to json key in
             # model definition for request.
-            obj_dict = obj.to_dict()
+            if hasattr(obj, "to_dict") and callable(obj.to_dict):
+                obj_dict = obj.to_dict()
+            else:
+                obj_dict = obj.__dict__
 
         return {
             key: self.sanitize_for_serialization(val) for key, val in obj_dict.items()

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

@@ -32,6 +32,9 @@ from truefoundry.ml.autogen.client.models.artifact_response_dto import (
 )
 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_manifest import (
+    ArtifactVersionManifest,
+)
 from truefoundry.ml.autogen.client.models.artifact_version_response_dto import (
     ArtifactVersionResponseDto,
 )
@@ -59,6 +62,7 @@ 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.command import Command
 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
@@ -110,6 +114,9 @@ from truefoundry.ml.autogen.client.models.create_run_request_dto import (
 from truefoundry.ml.autogen.client.models.create_run_response_dto import (
     CreateRunResponseDto,
 )
+from truefoundry.ml.autogen.client.models.create_workflow_task_config_request_dto import (
+    CreateWorkflowTaskConfigRequestDto,
+)
 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 (
@@ -136,11 +143,10 @@ 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.external_model_source import (
-    ExternalModelSource,
+from truefoundry.ml.autogen.client.models.external_artifact_source import (
+    ExternalArtifactSource,
 )
-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.fast_ai_framework import FastAIFramework
 from truefoundry.ml.autogen.client.models.file_info_dto import FileInfoDto
 from truefoundry.ml.autogen.client.models.finalize_artifact_version_request_dto import (
     FinalizeArtifactVersionRequestDto,
@@ -182,14 +188,18 @@ from truefoundry.ml.autogen.client.models.get_signed_urls_for_dataset_write_resp
 from truefoundry.ml.autogen.client.models.get_tenant_id_response_dto import (
     GetTenantIdResponseDto,
 )
+from truefoundry.ml.autogen.client.models.gluon_framework import GluonFramework
+from truefoundry.ml.autogen.client.models.h2_o_framework import H2OFramework
 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.keras_framework import KerasFramework
 from truefoundry.ml.autogen.client.models.latest_run_log_dto import LatestRunLogDto
 from truefoundry.ml.autogen.client.models.library_name import LibraryName
+from truefoundry.ml.autogen.client.models.light_gbm_framework import LightGBMFramework
 from truefoundry.ml.autogen.client.models.list_artifact_versions_request_dto import (
     ListArtifactVersionsRequestDto,
 )
@@ -262,6 +272,7 @@ from truefoundry.ml.autogen.client.models.log_metric_request_dto import (
 from truefoundry.ml.autogen.client.models.log_param_request_dto import (
     LogParamRequestDto,
 )
+from truefoundry.ml.autogen.client.models.manifest import Manifest
 from truefoundry.ml.autogen.client.models.method import Method
 from truefoundry.ml.autogen.client.models.metric_collection_dto import (
     MetricCollectionDto,
@@ -271,8 +282,10 @@ 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_environment import (
+    ModelVersionEnvironment,
+)
 from truefoundry.ml.autogen.client.models.model_version_manifest import (
     ModelVersionManifest,
 )
@@ -291,10 +304,12 @@ from truefoundry.ml.autogen.client.models.multi_part_upload_storage_provider imp
 from truefoundry.ml.autogen.client.models.notify_artifact_version_failure_dto import (
     NotifyArtifactVersionFailureDto,
 )
+from truefoundry.ml.autogen.client.models.onnx_framework import ONNXFramework
 from truefoundry.ml.autogen.client.models.openapi_spec import OpenapiSpec
+from truefoundry.ml.autogen.client.models.paddle_framework import PaddleFramework
 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.py_torch_framework import PyTorchFramework
 from truefoundry.ml.autogen.client.models.resolve_agent_app_response_dto import (
     ResolveAgentAppResponseDto,
 )
@@ -314,12 +329,21 @@ from truefoundry.ml.autogen.client.models.search_runs_request_dto import (
 from truefoundry.ml.autogen.client.models.search_runs_response_dto import (
     SearchRunsResponseDto,
 )
+from truefoundry.ml.autogen.client.models.serialization_format import (
+    SerializationFormat,
+)
 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.sklearn_framework import SklearnFramework
 from truefoundry.ml.autogen.client.models.source import Source
+from truefoundry.ml.autogen.client.models.source1 import Source1
+from truefoundry.ml.autogen.client.models.spa_cy_framework import SpaCyFramework
+from truefoundry.ml.autogen.client.models.stats_models_framework import (
+    StatsModelsFramework,
+)
 from truefoundry.ml.autogen.client.models.stop import Stop
 from truefoundry.ml.autogen.client.models.store_run_logs_request_dto import (
     StoreRunLogsRequestDto,
@@ -327,16 +351,22 @@ from truefoundry.ml.autogen.client.models.store_run_logs_request_dto import (
 from truefoundry.ml.autogen.client.models.subject import Subject
 from truefoundry.ml.autogen.client.models.subject_type import SubjectType
 from truefoundry.ml.autogen.client.models.system_message import SystemMessage
-from truefoundry.ml.autogen.client.models.tensorflow_framework import (
-    TensorflowFramework,
+from truefoundry.ml.autogen.client.models.tensor_flow_framework import (
+    TensorFlowFramework,
 )
 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.transformers_framework import (
     TransformersFramework,
 )
-from truefoundry.ml.autogen.client.models.truefoundry_model_source import (
-    TruefoundryModelSource,
+from truefoundry.ml.autogen.client.models.trigger_job_run_config_request_dto import (
+    TriggerJobRunConfigRequestDto,
+)
+from truefoundry.ml.autogen.client.models.trigger_job_run_config_response_dto import (
+    TriggerJobRunConfigResponseDto,
+)
+from truefoundry.ml.autogen.client.models.true_foundry_artifact_source import (
+    TrueFoundryArtifactSource,
 )
 from truefoundry.ml.autogen.client.models.update_artifact_version_request_dto import (
     UpdateArtifactVersionRequestDto,
@@ -356,10 +386,10 @@ from truefoundry.ml.autogen.client.models.update_run_request_dto import (
 from truefoundry.ml.autogen.client.models.update_run_response_dto import (
     UpdateRunResponseDto,
 )
-from truefoundry.ml.autogen.client.models.upload_model_source import UploadModelSource
 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,
 )
+from truefoundry.ml.autogen.client.models.xg_boost_framework import XGBoostFramework

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

@@ -16,8 +16,8 @@ from __future__ import annotations
 import json
 import pprint
 import re  # noqa: F401
+from typing import Any, Dict, Optional
 
-from truefoundry.ml.autogen.client.models.feature_dto import FeatureDto
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, conlist
 
 
@@ -27,7 +27,7 @@ class AddFeaturesToModelVersionRequestDto(BaseModel):
     """
 
     id: StrictStr = Field(...)
-    features: conlist(FeatureDto) = Field(...)
+    features: Optional[conlist(Dict[str, Any])] = None
     __properties = ["id", "features"]
 
     class Config:
@@ -52,13 +52,6 @@ class AddFeaturesToModelVersionRequestDto(BaseModel):
     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
@@ -71,13 +64,6 @@ class AddFeaturesToModelVersionRequestDto(BaseModel):
             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,
-            }
+            {"id": obj.get("id"), "features": obj.get("features")}
         )
         return _obj

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

@@ -66,7 +66,7 @@ class Agent(BaseModel):
         if value is None:
             return value
 
-        if value not in ("agent"):
+        if value not in ("agent",):
             raise ValueError("must be one of enum values ('agent')")
         return value
 

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

@@ -49,7 +49,7 @@ class AgentApp(BaseModel):
         if value is None:
             return value
 
-        if value not in ("agent-app"):
+        if value not in ("agent-app",):
             raise ValueError("must be one of enum values ('agent-app')")
         return value
 

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

@@ -63,7 +63,7 @@ class AgentOpenAPITool(BaseModel):
         if value is None:
             return value
 
-        if value not in ("openapi-tool"):
+        if value not in ("openapi-tool",):
             raise ValueError("must be one of enum values ('openapi-tool')")
         return value
 

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

@@ -64,7 +64,7 @@ class AgentOpenAPIToolWithFQN(BaseModel):
     @validator("type")
     def type_validate_enum(cls, value):
         """Validates the enum"""
-        if value not in ("openapi-tool"):
+        if value not in ("openapi-tool",):
             raise ValueError("must be one of enum values ('openapi-tool')")
         return value
 

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

@@ -66,7 +66,7 @@ class AgentWithFQN(BaseModel):
     @validator("type")
     def type_validate_enum(cls, value):
         """Validates the enum"""
-        if value not in ("agent"):
+        if value not in ("agent",):
             raise ValueError("must be one of enum values ('agent')")
         return value
 

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

@@ -26,9 +26,7 @@ from truefoundry.ml.autogen.client.models.artifact_version_status import (
     ArtifactVersionStatus,
 )
 from truefoundry.ml.autogen.client.models.internal_metadata import InternalMetadata
-from truefoundry.ml.autogen.client.models.model_version_manifest import (
-    ModelVersionManifest,
-)
+from truefoundry.ml.autogen.client.models.manifest import Manifest
 from truefoundry.ml.autogen.client.models.subject import Subject
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictInt, StrictStr
 
@@ -59,7 +57,7 @@ class ArtifactVersionDto(BaseModel):
     artifact_name: Optional[StrictStr] = None
     internal_metadata: Optional[InternalMetadata] = None
     artifact_size: Optional[StrictInt] = None
-    manifest: Optional[ModelVersionManifest] = None
+    manifest: Optional[Manifest] = None
     serialization_format: ArtifactVersionSerializationFormat = Field(...)
     __properties = [
         "id",
@@ -158,7 +156,7 @@ class ArtifactVersionDto(BaseModel):
                 if obj.get("internal_metadata") is not None
                 else None,
                 "artifact_size": obj.get("artifact_size"),
-                "manifest": ModelVersionManifest.from_dict(obj.get("manifest"))
+                "manifest": Manifest.from_dict(obj.get("manifest"))
                 if obj.get("manifest") is not None
                 else None,
                 "serialization_format": obj.get("serialization_format"),