truefoundry 0.5.0rc7__py3-none-any.whl → 0.5.1__py3-none-any.whl

truefoundry/ml/autogen/client/__init__.py

@@ -172,6 +172,9 @@ 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.export_deployment_files_request_dto import (
+    ExportDeploymentFilesRequestDto,
+)
 from truefoundry.ml.autogen.client.models.external_artifact_source import (
     ExternalArtifactSource,
 )
@@ -224,6 +227,7 @@ from truefoundry.ml.autogen.client.models.http_validation_error import (
 )
 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.infer_method_name import InferMethodName
 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
@@ -311,6 +315,7 @@ 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_server import ModelServer
 from truefoundry.ml.autogen.client.models.model_version_dto import ModelVersionDto
 from truefoundry.ml.autogen.client.models.model_version_environment import (
     ModelVersionEnvironment,
@@ -358,15 +363,16 @@ 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.sklearn_model_schema import SklearnModelSchema
+from truefoundry.ml.autogen.client.models.sklearn_serialization_format import (
+    SklearnSerializationFormat,
+)
 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
@@ -417,8 +423,20 @@ from truefoundry.ml.autogen.client.models.update_run_response_dto import (
 )
 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.validate_external_storage_root_request_dto import (
+    ValidateExternalStorageRootRequestDto,
+)
+from truefoundry.ml.autogen.client.models.validate_external_storage_root_response_dto import (
+    ValidateExternalStorageRootResponseDto,
+)
 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
+from truefoundry.ml.autogen.client.models.xg_boost_model_schema import (
+    XGBoostModelSchema,
+)
+from truefoundry.ml.autogen.client.models.xg_boost_serialization_format import (
+    XGBoostSerializationFormat,
+)

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

@@ -73,6 +73,9 @@ from truefoundry.ml.autogen.client.models.delete_files_for_dataset_request_dto i
 from truefoundry.ml.autogen.client.models.delete_model_version_request_dto import (
     DeleteModelVersionRequestDto,
 )
+from truefoundry.ml.autogen.client.models.export_deployment_files_request_dto import (
+    ExportDeploymentFilesRequestDto,
+)
 from truefoundry.ml.autogen.client.models.finalize_artifact_version_request_dto import (
     FinalizeArtifactVersionRequestDto,
 )
@@ -164,6 +167,12 @@ from truefoundry.ml.autogen.client.models.update_dataset_request_dto import (
 from truefoundry.ml.autogen.client.models.update_model_version_request_dto import (
     UpdateModelVersionRequestDto,
 )
+from truefoundry.ml.autogen.client.models.validate_external_storage_root_request_dto import (
+    ValidateExternalStorageRootRequestDto,
+)
+from truefoundry.ml.autogen.client.models.validate_external_storage_root_response_dto import (
+    ValidateExternalStorageRootResponseDto,
+)
 from truefoundry.pydantic_v1 import (
     StrictInt,
     StrictStr,
@@ -2277,6 +2286,164 @@ class MlfoundryArtifactsApi:
             _request_auth=_params.get("_request_auth"),
         )
 
+    @validate_arguments
+    def export_deployment_files_by_fqn_post(
+        self,
+        export_deployment_files_request_dto: ExportDeploymentFilesRequestDto,
+        **kwargs,
+    ) -> object:  # noqa: E501
+        """Export Deployment Files By Fqn  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+
+        >>> thread = api.export_deployment_files_by_fqn_post(export_deployment_files_request_dto, async_req=True)
+        >>> result = thread.get()
+
+        :param export_deployment_files_request_dto: (required)
+        :type export_deployment_files_request_dto: ExportDeploymentFilesRequestDto
+        :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: object
+        """
+        kwargs["_return_http_data_only"] = True
+        if "_preload_content" in kwargs:
+            message = "Error! Please call the export_deployment_files_by_fqn_post_with_http_info method with `_preload_content` instead and obtain raw data from ApiResponse.raw_data"  # noqa: E501
+            raise ValueError(message)
+        return self.export_deployment_files_by_fqn_post_with_http_info(
+            export_deployment_files_request_dto, **kwargs
+        )  # noqa: E501
+
+    @validate_arguments
+    def export_deployment_files_by_fqn_post_with_http_info(
+        self,
+        export_deployment_files_request_dto: ExportDeploymentFilesRequestDto,
+        **kwargs,
+    ) -> ApiResponse:  # noqa: E501
+        """Export Deployment Files By Fqn  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+
+        >>> thread = api.export_deployment_files_by_fqn_post_with_http_info(export_deployment_files_request_dto, async_req=True)
+        >>> result = thread.get()
+
+        :param export_deployment_files_request_dto: (required)
+        :type export_deployment_files_request_dto: ExportDeploymentFilesRequestDto
+        :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(object, status_code(int), headers(HTTPHeaderDict))
+        """
+
+        _params = locals()
+
+        _all_params = ["export_deployment_files_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 export_deployment_files_by_fqn_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["export_deployment_files_request_dto"] is not None:
+            _body_params = _params["export_deployment_files_request_dto"]
+
+        # set the HTTP header `Accept`
+        _header_params["Accept"] = self.api_client.select_header_accept(
+            ["application/json", "application/zip"]
+        )  # 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": "object",
+            "422": "HTTPValidationError",
+        }
+
+        return self.api_client.call_api(
+            "/api/2.0/mlflow/mlfoundry-artifacts/model-versions/export-deployment-files-by-fqn",
+            "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 finalize_artifact_version_post(
         self,
@@ -7054,3 +7221,161 @@ class MlfoundryArtifactsApi:
             collection_formats=_collection_formats,
             _request_auth=_params.get("_request_auth"),
         )
+
+    @validate_arguments
+    def validate_external_storage_root_path_post(
+        self,
+        validate_external_storage_root_request_dto: ValidateExternalStorageRootRequestDto,
+        **kwargs,
+    ) -> ValidateExternalStorageRootResponseDto:  # noqa: E501
+        """Validate External Storage Root Path  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+
+        >>> thread = api.validate_external_storage_root_path_post(validate_external_storage_root_request_dto, async_req=True)
+        >>> result = thread.get()
+
+        :param validate_external_storage_root_request_dto: (required)
+        :type validate_external_storage_root_request_dto: ValidateExternalStorageRootRequestDto
+        :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: ValidateExternalStorageRootResponseDto
+        """
+        kwargs["_return_http_data_only"] = True
+        if "_preload_content" in kwargs:
+            message = "Error! Please call the validate_external_storage_root_path_post_with_http_info method with `_preload_content` instead and obtain raw data from ApiResponse.raw_data"  # noqa: E501
+            raise ValueError(message)
+        return self.validate_external_storage_root_path_post_with_http_info(
+            validate_external_storage_root_request_dto, **kwargs
+        )  # noqa: E501
+
+    @validate_arguments
+    def validate_external_storage_root_path_post_with_http_info(
+        self,
+        validate_external_storage_root_request_dto: ValidateExternalStorageRootRequestDto,
+        **kwargs,
+    ) -> ApiResponse:  # noqa: E501
+        """Validate External Storage Root Path  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+
+        >>> thread = api.validate_external_storage_root_path_post_with_http_info(validate_external_storage_root_request_dto, async_req=True)
+        >>> result = thread.get()
+
+        :param validate_external_storage_root_request_dto: (required)
+        :type validate_external_storage_root_request_dto: ValidateExternalStorageRootRequestDto
+        :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(ValidateExternalStorageRootResponseDto, status_code(int), headers(HTTPHeaderDict))
+        """
+
+        _params = locals()
+
+        _all_params = ["validate_external_storage_root_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 validate_external_storage_root_path_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["validate_external_storage_root_request_dto"] is not None:
+            _body_params = _params["validate_external_storage_root_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": "ValidateExternalStorageRootResponseDto",
+            "422": "HTTPValidationError",
+        }
+
+        return self.api_client.call_api(
+            "/api/2.0/mlflow/mlfoundry-artifacts/artifact-versions/validate-storage-root",
+            "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"),
+        )

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

@@ -143,6 +143,9 @@ 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.export_deployment_files_request_dto import (
+    ExportDeploymentFilesRequestDto,
+)
 from truefoundry.ml.autogen.client.models.external_artifact_source import (
     ExternalArtifactSource,
 )
@@ -195,6 +198,7 @@ from truefoundry.ml.autogen.client.models.http_validation_error import (
 )
 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.infer_method_name import InferMethodName
 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
@@ -282,6 +286,7 @@ 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_server import ModelServer
 from truefoundry.ml.autogen.client.models.model_version_dto import ModelVersionDto
 from truefoundry.ml.autogen.client.models.model_version_environment import (
     ModelVersionEnvironment,
@@ -329,15 +334,16 @@ 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.sklearn_model_schema import SklearnModelSchema
+from truefoundry.ml.autogen.client.models.sklearn_serialization_format import (
+    SklearnSerializationFormat,
+)
 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
@@ -388,8 +394,20 @@ from truefoundry.ml.autogen.client.models.update_run_response_dto import (
 )
 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.validate_external_storage_root_request_dto import (
+    ValidateExternalStorageRootRequestDto,
+)
+from truefoundry.ml.autogen.client.models.validate_external_storage_root_response_dto import (
+    ValidateExternalStorageRootResponseDto,
+)
 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
+from truefoundry.ml.autogen.client.models.xg_boost_model_schema import (
+    XGBoostModelSchema,
+)
+from truefoundry.ml.autogen.client.models.xg_boost_serialization_format import (
+    XGBoostSerializationFormat,
+)

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

@@ -36,11 +36,11 @@ class ArtifactVersionManifest(BaseModel):
 
     description: Optional[constr(strict=True, max_length=512)] = Field(
         default=None,
-        description="+label=Description +docs=Description of the artifact version",
+        description="+label=Description +usage=Description of the artifact or model version +docs=Description of the artifact or model version",
     )
     metadata: Dict[str, Any] = Field(
         default=...,
-        description="+label=Metadata +docs=Metadata for the model version +usage=Metadata for the model version +uiType=JsonInput",
+        description="+label=Metadata +docs=Metadata for the artifact or model version +usage=Metadata for the artifact or model version +uiType=JsonInput",
     )
     type: Optional[StrictStr] = "artifact-version"
     source: Source = Field(...)

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

@@ -0,0 +1,82 @@
+# 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.ml.autogen.client.models.model_server import ModelServer
+from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr
+
+
+class ExportDeploymentFilesRequestDto(BaseModel):
+    """
+    ExportDeploymentFilesRequestDto
+    """
+
+    model_version_fqn: StrictStr = Field(...)
+    workspace_fqn: StrictStr = Field(...)
+    service_name: StrictStr = Field(...)
+    model_server: Optional[ModelServer] = None
+    __properties = [
+        "model_version_fqn",
+        "workspace_fqn",
+        "service_name",
+        "model_server",
+    ]
+
+    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) -> ExportDeploymentFilesRequestDto:
+        """Create an instance of ExportDeploymentFilesRequestDto 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) -> ExportDeploymentFilesRequestDto:
+        """Create an instance of ExportDeploymentFilesRequestDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return ExportDeploymentFilesRequestDto.parse_obj(obj)
+
+        _obj = ExportDeploymentFilesRequestDto.parse_obj(
+            {
+                "model_version_fqn": obj.get("model_version_fqn"),
+                "workspace_fqn": obj.get("workspace_fqn"),
+                "service_name": obj.get("service_name"),
+                "model_server": obj.get("model_server"),
+            }
+        )
+        return _obj

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

@@ -0,0 +1,34 @@
+# 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
+
+import json
+import re  # noqa: F401
+
+from aenum import Enum
+
+
+class InferMethodName(str, Enum):
+    """
+    +label=Inference Method Name +usage=Name of the method used for inference
+    """
+
+    """
+    allowed enum values
+    """
+    PREDICT = "predict"
+    PREDICT_PROBA = "predict_proba"
+
+    @classmethod
+    def from_json(cls, json_str: str) -> InferMethodName:
+        """Create an instance of InferMethodName from a JSON string"""
+        return InferMethodName(json.loads(json_str))

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

@@ -0,0 +1,34 @@
+# 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
+
+import json
+import re  # noqa: F401
+
+from aenum import Enum
+
+
+class ModelServer(str, Enum):
+    """
+    An enumeration.
+    """
+
+    """
+    allowed enum values
+    """
+    TRITON = "triton"
+    FASTAPI = "fastapi"
+
+    @classmethod
+    def from_json(cls, json_str: str) -> ModelServer:
+        """Create an instance of ModelServer from a JSON string"""
+        return ModelServer(json.loads(json_str))

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

@@ -30,7 +30,7 @@ from truefoundry.pydantic_v1 import (
 
 class ModelVersionEnvironment(BaseModel):
     """
-    +label=ModelVersionEnvironment  # noqa: E501
+    +label=Environment  # noqa: E501
     """
 
     python_version: Optional[constr(strict=True)] = Field(