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

truefoundry/common/utils.py

@@ -1,7 +1,14 @@
+import json
+import logging
+import os
+import sys
 import time
+from collections import namedtuple
 from functools import lru_cache, wraps
+from shutil import rmtree
+from subprocess import check_output
 from time import monotonic_ns
-from typing import Callable, Generator, Optional, TypeVar
+from typing import Callable, Generator, List, Optional, TypeVar
 from urllib.parse import urljoin, urlparse
 
 from truefoundry.common.constants import (
@@ -12,7 +19,10 @@ from truefoundry.common.constants import (
 )
 from truefoundry.pydantic_v1 import BaseSettings
 
+logger = logging.getLogger(__name__)
+
 T = TypeVar("T")
+InstalledPipPackage = namedtuple("InstalledPipPackage", ["name", "version"])
 
 
 class _TFYServersConfig(BaseSettings):
@@ -102,3 +112,65 @@ def resolve_tfy_host(tfy_host: Optional[str] = None) -> str:
     tfy_host = tfy_host.strip("/")
     validate_tfy_host(tfy_host)
     return tfy_host
+
+
+class ContextualDirectoryManager:
+    def __init__(self, dir_path: str, cleanup_on_error: bool = True):
+        self.dir_path = dir_path
+        self.cleanup_on_error = cleanup_on_error
+
+    def __enter__(self):
+        if os.path.exists(self.dir_path):
+            raise FileExistsError(
+                f"The directory {self.dir_path!r} already exists. "
+                "Please provide a path with a different name that does not already exist."
+            )
+
+        os.makedirs(self.dir_path, exist_ok=False)
+        return self.dir_path
+
+    def __exit__(self, exc_type, exc_value, traceback):
+        if self.cleanup_on_error and exc_type is not None:
+            # Only delete the directory if an exception occurred
+            if os.path.exists(self.dir_path):
+                rmtree(self.dir_path)
+
+
+def get_python_version_major_minor() -> str:
+    """
+    Returns the major.minor version of the Python interpreter
+    """
+    version_info = sys.version_info
+    return f"{version_info.major}.{version_info.minor}"
+
+
+def list_pip_packages_installed(
+    filter_package_names: Optional[List[str]] = None,
+) -> List[InstalledPipPackage]:
+    """
+    List the installed package_names, along with their versions.
+    Args:
+        filter_package_names (List[str]): A list of specific libraries to filter for.
+
+    Returns:
+        List[InstalledPipPackage]: A list of InstalledPipPackage namedtuples for each match.
+    """
+    relevant_package_names: List[InstalledPipPackage] = []
+
+    # Get the installed packages in JSON format from pip
+    try:
+        output = check_output(
+            [sys.executable, "-m", "pip", "list", "--pre", "--format=json"]
+        )
+        installed_package_names = json.loads(output.decode("utf-8"))
+    except Exception:
+        logger.exception("Failed to list installed packages using pip.")
+        return relevant_package_names
+
+    package_names_to_check = set(filter_package_names or [])
+    for package in installed_package_names:
+        if package["name"] in package_names_to_check:
+            relevant_package_names.append(
+                InstalledPipPackage(package["name"], package["version"])
+            )
+    return relevant_package_names

truefoundry/deploy/__init__.py

@@ -1,8 +1,11 @@
 from truefoundry.deploy.auto_gen.models import (
     AppProtocol,
+    Autoshutdown,
     CanaryStep,
     CapacityType,
+    Claim,
     ConcurrencyPolicy,
+    JobAlert,
     Kustomize,
     ParamType,
     Protocol,
@@ -66,6 +69,7 @@ from truefoundry.deploy.v2.lib.patched_models import (
     DynamicVolumeConfig,
     Endpoint,
     GcpTPU,
+    GitHelmRepo,
     GitSource,
     GPUType,
     HealthProbe,
@@ -73,6 +77,7 @@ from truefoundry.deploy.v2.lib.patched_models import (
     HttpProbe,
     HuggingfaceArtifactSource,
     Image,
+    JwtAuthCreds,
     KafkaInputConfig,
     KafkaMetricConfig,
     KafkaOutputConfig,

truefoundry/deploy/cli/cli.py

@@ -9,6 +9,7 @@ from truefoundry.deploy.cli.commands import (
     get_build_command,
     get_delete_command,
     get_deploy_command,
+    get_deploy_init_command,
     get_login_command,
     get_logout_command,
     get_patch_application_command,
@@ -36,6 +37,7 @@ def create_truefoundry_cli() -> click.MultiCommand:
     cli.add_command(get_logout_command())
     cli.add_command(get_apply_command())
     cli.add_command(get_deploy_command())
+    cli.add_command(get_deploy_init_command())
     cli.add_command(get_patch_application_command())
     cli.add_command(get_delete_command())
     cli.add_command(get_trigger_command())

truefoundry/deploy/cli/commands/__init__.py

@@ -4,6 +4,7 @@ from truefoundry.deploy.cli.commands.build_logs_command import get_build_logs_co
 from truefoundry.deploy.cli.commands.create_command import get_create_command
 from truefoundry.deploy.cli.commands.delete_command import get_delete_command
 from truefoundry.deploy.cli.commands.deploy_command import get_deploy_command
+from truefoundry.deploy.cli.commands.deploy_init_command import get_deploy_init_command
 from truefoundry.deploy.cli.commands.get_command import get_get_command
 from truefoundry.deploy.cli.commands.list_command import get_list_command
 from truefoundry.deploy.cli.commands.login_command import get_login_command

truefoundry/deploy/cli/commands/deploy_init_command.py

@@ -0,0 +1,22 @@
+import rich_click as click
+
+from truefoundry.deploy.cli.const import GROUP_CLS
+from truefoundry.ml.cli.commands.model_init import get_model_init_command
+
+
+@click.group(
+    name="deploy-init",
+    cls=GROUP_CLS,
+    help="Initialize the TrueFoundry deployment configuration.",
+)
+def deploy_init_command(): ...
+
+
+def get_deploy_init_command():
+    """
+    Generates the deploy-init command.
+        model: Initialize the TrueFoundry deployment configuration for a model. eg: tfy deploy-init model [--args]
+    """
+
+    deploy_init_command.add_command(get_model_init_command())
+    return deploy_init_command

truefoundry/deploy/lib/dao/application.py

@@ -179,10 +179,11 @@ def trigger_job(
         command=command_str if command_str else None,
         params=params if params else None,
     )
+    jobRunName = result.jobRunName
     previous_runs_url = f"{client.base_url.strip('/')}/deployments/{application_info.id}?tab=previousRuns"
     logger.info(
         f"{message}.\n"
-        f"You can check the status of your job run at {previous_runs_url}"
+        f"You can check the status of your job run at {previous_runs_url} with jobRunName: {jobRunName}"
     )
     return result
 

truefoundry/deploy/v2/lib/patched_models.py

@@ -186,6 +186,10 @@ class BasicAuthCreds(models.BasicAuthCreds, PatchedModelBase):
     type: Literal["basic_auth"] = "basic_auth"
 
 
+class JwtAuthCreds(models.JwtAuthCreds, PatchedModelBase):
+    type: Literal["jwt_auth"] = "jwt_auth"
+
+
 class HealthProbe(models.HealthProbe, PatchedModelBase):
     pass
 
@@ -338,6 +342,10 @@ class HelmRepo(models.HelmRepo, PatchedModelBase):
     type: Literal["helm-repo"] = "helm-repo"
 
 
+class GitHelmRepo(models.GitHelmRepo, PatchedModelBase):
+    type: Literal["git-helm-repo"] = "git-helm-repo"
+
+
 class OCIRepo(models.OCIRepo, PatchedModelBase):
     type: Literal["oci-repo"] = "oci-repo"
 

truefoundry/ml/__init__.py

@@ -16,6 +16,7 @@ from truefoundry.ml.log_types.artifacts.artifact import (
 from truefoundry.ml.log_types.artifacts.dataset import DataDirectory, DataDirectoryPath
 from truefoundry.ml.log_types.artifacts.model import (
     ModelVersion,
+    infer_signature,
 )
 from truefoundry.ml.logger import init_logger
 from truefoundry.ml.mlfoundry_api import get_client
@@ -45,33 +46,34 @@ __all__ = [
     "DataDirectory",
     "DataDirectoryPath",
     "DataSlice",
+    "FastAIFramework",
     "FileFormat",
+    "GluonFramework",
+    "H2OFramework",
     "Image",
-    "MlFoundryRun",
+    "KerasFramework",
+    "LibraryName",
+    "LightGBMFramework",
     "MlFoundryException",
-    "ModelVersionEnvironment",
+    "MlFoundryRun",
     "ModelFramework",
+    "ModelFrameworkType",
     "ModelType",
     "ModelVersion",
-    "Plot",
-    "ViewType",
-    "get_client",
-    "FastAIFramework",
-    "GluonFramework",
-    "H2OFramework",
-    "KerasFramework",
-    "LightGBMFramework",
+    "ModelVersionEnvironment",
     "ONNXFramework",
     "PaddleFramework",
+    "Plot",
     "PyTorchFramework",
     "SklearnFramework",
     "SpaCyFramework",
     "StatsModelsFramework",
     "TensorFlowFramework",
     "TransformersFramework",
+    "ViewType",
     "XGBoostFramework",
-    "LibraryName",
-    "ModelFrameworkType",
+    "get_client",
+    "infer_signature",
 ]
 
 init_logger()

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,

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,
 )
@@ -2277,6 +2280,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,

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,

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(