truefoundry 0.5.0rc6__py3-none-any.whl → 0.5.1rc1__py3-none-any.whl

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

@@ -0,0 +1,97 @@
+# 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 ModelVersionEnvironment(BaseModel):
+    """
+    +label=Environment  # noqa: E501
+    """
+
+    python_version: Optional[constr(strict=True)] = Field(
+        default=None,
+        description="+label=Python Version +usage=Python version for the model version",
+    )
+    pip_packages: Optional[conlist(StrictStr)] = Field(
+        default=None,
+        description="+label=PIP Packages +usage=PIP packages for the model version",
+    )
+    __properties = ["python_version", "pip_packages"]
+
+    @validator("python_version")
+    def python_version_validate_regular_expression(cls, value):
+        """Validates the regular expression"""
+        if value is None:
+            return value
+
+        if not re.match(r"^\d+(\.\d+){1,2}([\-\.a-z0-9]+)?$", value):
+            raise ValueError(
+                r"must validate the regular expression /^\d+(\.\d+){1,2}([\-\.a-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) -> ModelVersionEnvironment:
+        """Create an instance of ModelVersionEnvironment 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) -> ModelVersionEnvironment:
+        """Create an instance of ModelVersionEnvironment from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return ModelVersionEnvironment.parse_obj(obj)
+
+        _obj = ModelVersionEnvironment.parse_obj(
+            {
+                "python_version": obj.get("python_version"),
+                "pip_packages": obj.get("pip_packages"),
+            }
+        )
+        return _obj

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

@@ -19,6 +19,9 @@ import re  # noqa: F401
 from typing import Any, Dict, Optional
 
 from truefoundry.ml.autogen.client.models.framework import Framework
+from truefoundry.ml.autogen.client.models.model_version_environment import (
+    ModelVersionEnvironment,
+)
 from truefoundry.ml.autogen.client.models.source1 import Source1
 from truefoundry.pydantic_v1 import (
     BaseModel,
@@ -37,21 +40,22 @@ class ModelVersionManifest(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] = "model-version"
     source: Source1 = Field(...)
     framework: Optional[Framework] = None
+    environment: Optional[ModelVersionEnvironment] = None
     step: Optional[conint(strict=True, ge=0)] = Field(
         default=0, description="+label=Step"
     )
     model_schema: Optional[Dict[str, Any]] = Field(
         default=None,
-        description="+label=Model Schema +usage=Schema of the model +uiType=Hidden",
+        description="+label=Model Schema +usage=Schema of the model +uiType=JsonInput",
     )
     __properties = [
         "description",
@@ -59,6 +63,7 @@ class ModelVersionManifest(BaseModel):
         "type",
         "source",
         "framework",
+        "environment",
         "step",
         "model_schema",
     ]
@@ -101,6 +106,9 @@ class ModelVersionManifest(BaseModel):
         # override the default output from truefoundry.pydantic_v1 by calling `to_dict()` of framework
         if self.framework:
             _dict["framework"] = self.framework.to_dict()
+        # override the default output from truefoundry.pydantic_v1 by calling `to_dict()` of environment
+        if self.environment:
+            _dict["environment"] = self.environment.to_dict()
         return _dict
 
     @classmethod
@@ -125,6 +133,9 @@ class ModelVersionManifest(BaseModel):
                 "framework": Framework.from_dict(obj.get("framework"))
                 if obj.get("framework") is not None
                 else None,
+                "environment": ModelVersionEnvironment.from_dict(obj.get("environment"))
+                if obj.get("environment") is not None
+                else None,
                 "step": obj.get("step") if obj.get("step") is not None else 0,
                 "model_schema": obj.get("model_schema"),
             }

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

@@ -0,0 +1,35 @@
+# 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 SerializationFormat(str, Enum):
+    """
+    +label=Serialization format +usage=Serialization format used for the model
+    """
+
+    """
+    allowed enum values
+    """
+    CLOUDPICKLE = "cloudpickle"
+    JOBLIB = "joblib"
+    PICKLE = "pickle"
+
+    @classmethod
+    def from_json(cls, json_str: str) -> SerializationFormat:
+        """Create an instance of SerializationFormat from a JSON string"""
+        return SerializationFormat(json.loads(json_str))

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

@@ -16,7 +16,12 @@ from __future__ import annotations
 import json
 import pprint
 import re  # noqa: F401
+from typing import Optional
 
+from truefoundry.ml.autogen.client.models.infer_method_name import InferMethodName
+from truefoundry.ml.autogen.client.models.serialization_format import (
+    SerializationFormat,
+)
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
 
 
@@ -29,7 +34,24 @@ class SklearnFramework(BaseModel):
         default=...,
         description="+label=Type +usage=Type of the framework +value=sklearn",
     )
-    __properties = ["type"]
+    serialization_format: Optional[SerializationFormat] = Field(
+        default=None,
+        description="+label=Serialization format +usage=Serialization format used for the model",
+    )
+    model_filepath: Optional[StrictStr] = Field(
+        default=None,
+        description="+label=Model file path +usage=Relative path to the model file",
+    )
+    infer_method_name: Optional[InferMethodName] = Field(
+        default=None,
+        description="+label=Inference method name +usage=Name of the method used for inference",
+    )
+    __properties = [
+        "type",
+        "serialization_format",
+        "model_filepath",
+        "infer_method_name",
+    ]
 
     @validator("type")
     def type_validate_enum(cls, value):
@@ -71,5 +93,12 @@ class SklearnFramework(BaseModel):
         if not isinstance(obj, dict):
             return SklearnFramework.parse_obj(obj)
 
-        _obj = SklearnFramework.parse_obj({"type": obj.get("type")})
+        _obj = SklearnFramework.parse_obj(
+            {
+                "type": obj.get("type"),
+                "serialization_format": obj.get("serialization_format"),
+                "model_filepath": obj.get("model_filepath"),
+                "infer_method_name": obj.get("infer_method_name"),
+            }
+        )
         return _obj

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

@@ -37,11 +37,11 @@ class TransformersFramework(BaseModel):
     )
     pipeline_tag: Optional[StrictStr] = Field(
         default=None,
-        description="+label=Pipeline Tag +usage=Pipeline tag +docs=Pipeline tag for the framework",
+        description="+label=Pipeline Tag +usage=The `pipeline()` task this model can be used with e.g. `text-generation`. See [huggingface docs](https://huggingface.co/docs/transformers/main/en/main_classes/pipelines#transformers.pipeline.task) for all possible values +docs=Pipeline tag for the framework",
     )
     base_model: Optional[StrictStr] = Field(
         default=None,
-        description="+label=Base Model +usage=Base model +docs=Base model Id. If this is a finetuned model, this points to the base model used for finetuning",
+        description="+label=Base Model +usage=Base model Id. If this is a finetuned model, this points to the base model used for finetuning +docs=Base model Id. If this is a finetuned model, this points to the base model used for finetuning",
     )
     __properties = ["type", "library_name", "pipeline_tag", "base_model"]
 

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

@@ -16,7 +16,11 @@ from __future__ import annotations
 import json
 import pprint
 import re  # noqa: F401
+from typing import Optional
 
+from truefoundry.ml.autogen.client.models.serialization_format import (
+    SerializationFormat,
+)
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
 
 
@@ -29,7 +33,15 @@ class XGBoostFramework(BaseModel):
         default=...,
         description="+label=Type +usage=Type of the framework +value=xgboost",
     )
-    __properties = ["type"]
+    serialization_format: Optional[SerializationFormat] = Field(
+        default=None,
+        description="+label=Serialization format +usage=Serialization format used for the model",
+    )
+    model_filepath: Optional[StrictStr] = Field(
+        default=None,
+        description="+label=Model file path +usage=Relative path to the model file",
+    )
+    __properties = ["type", "serialization_format", "model_filepath"]
 
     @validator("type")
     def type_validate_enum(cls, value):
@@ -71,5 +83,11 @@ class XGBoostFramework(BaseModel):
         if not isinstance(obj, dict):
             return XGBoostFramework.parse_obj(obj)
 
-        _obj = XGBoostFramework.parse_obj({"type": obj.get("type")})
+        _obj = XGBoostFramework.parse_obj(
+            {
+                "type": obj.get("type"),
+                "serialization_format": obj.get("serialization_format"),
+                "model_filepath": obj.get("model_filepath"),
+            }
+        )
         return _obj

truefoundry/ml/autogen/client_README.md

@@ -102,6 +102,7 @@ Class | Method | HTTP request | Description
 *MlfoundryArtifactsApi* | [**delete_dataset_post**](truefoundry/ml/autogen/client/docs/MlfoundryArtifactsApi.md#delete_dataset_post) | **POST** /api/2.0/mlflow/mlfoundry-artifacts/datasets/delete | Delete Dataset
 *MlfoundryArtifactsApi* | [**delete_files_for_dataset_delete**](truefoundry/ml/autogen/client/docs/MlfoundryArtifactsApi.md#delete_files_for_dataset_delete) | **DELETE** /api/2.0/mlflow/mlfoundry-artifacts/datasets/files/ | Delete Files For Dataset
 *MlfoundryArtifactsApi* | [**delete_model_version_post**](truefoundry/ml/autogen/client/docs/MlfoundryArtifactsApi.md#delete_model_version_post) | **POST** /api/2.0/mlflow/mlfoundry-artifacts/model-versions/delete | Delete Model Version
+*MlfoundryArtifactsApi* | [**export_deployment_files_by_fqn_post**](truefoundry/ml/autogen/client/docs/MlfoundryArtifactsApi.md#export_deployment_files_by_fqn_post) | **POST** /api/2.0/mlflow/mlfoundry-artifacts/model-versions/export-deployment-files-by-fqn | Export Deployment Files By Fqn
 *MlfoundryArtifactsApi* | [**finalize_artifact_version_post**](truefoundry/ml/autogen/client/docs/MlfoundryArtifactsApi.md#finalize_artifact_version_post) | **POST** /api/2.0/mlflow/mlfoundry-artifacts/artifact-versions/finalize | Finalize Artifact Version
 *MlfoundryArtifactsApi* | [**get_artifact_by_fqn_get**](truefoundry/ml/autogen/client/docs/MlfoundryArtifactsApi.md#get_artifact_by_fqn_get) | **GET** /api/2.0/mlflow/mlfoundry-artifacts/artifacts/get-by-fqn | Get Artifact By Fqn
 *MlfoundryArtifactsApi* | [**get_artifact_by_id_get**](truefoundry/ml/autogen/client/docs/MlfoundryArtifactsApi.md#get_artifact_by_id_get) | **GET** /api/2.0/mlflow/mlfoundry-artifacts/artifacts/get | Get Artifact By Id
@@ -211,6 +212,7 @@ Class | Method | HTTP request | Description
  - [ExperimentIdRequestDto](truefoundry/ml/autogen/client/docs/ExperimentIdRequestDto.md)
  - [ExperimentResponseDto](truefoundry/ml/autogen/client/docs/ExperimentResponseDto.md)
  - [ExperimentTagDto](truefoundry/ml/autogen/client/docs/ExperimentTagDto.md)
+ - [ExportDeploymentFilesRequestDto](truefoundry/ml/autogen/client/docs/ExportDeploymentFilesRequestDto.md)
  - [ExternalArtifactSource](truefoundry/ml/autogen/client/docs/ExternalArtifactSource.md)
  - [FastAIFramework](truefoundry/ml/autogen/client/docs/FastAIFramework.md)
  - [FileInfoDto](truefoundry/ml/autogen/client/docs/FileInfoDto.md)
@@ -233,6 +235,7 @@ Class | Method | HTTP request | Description
  - [HTTPValidationError](truefoundry/ml/autogen/client/docs/HTTPValidationError.md)
  - [ImageContentPart](truefoundry/ml/autogen/client/docs/ImageContentPart.md)
  - [ImageUrl](truefoundry/ml/autogen/client/docs/ImageUrl.md)
+ - [InferMethodName](truefoundry/ml/autogen/client/docs/InferMethodName.md)
  - [InternalMetadata](truefoundry/ml/autogen/client/docs/InternalMetadata.md)
  - [KerasFramework](truefoundry/ml/autogen/client/docs/KerasFramework.md)
  - [LatestRunLogDto](truefoundry/ml/autogen/client/docs/LatestRunLogDto.md)
@@ -270,7 +273,9 @@ Class | Method | HTTP request | Description
  - [ModelConfiguration](truefoundry/ml/autogen/client/docs/ModelConfiguration.md)
  - [ModelDto](truefoundry/ml/autogen/client/docs/ModelDto.md)
  - [ModelResponseDto](truefoundry/ml/autogen/client/docs/ModelResponseDto.md)
+ - [ModelServer](truefoundry/ml/autogen/client/docs/ModelServer.md)
  - [ModelVersionDto](truefoundry/ml/autogen/client/docs/ModelVersionDto.md)
+ - [ModelVersionEnvironment](truefoundry/ml/autogen/client/docs/ModelVersionEnvironment.md)
  - [ModelVersionManifest](truefoundry/ml/autogen/client/docs/ModelVersionManifest.md)
  - [ModelVersionResponseDto](truefoundry/ml/autogen/client/docs/ModelVersionResponseDto.md)
  - [MultiPartUploadDto](truefoundry/ml/autogen/client/docs/MultiPartUploadDto.md)
@@ -294,6 +299,7 @@ Class | Method | HTTP request | Description
  - [RunTagDto](truefoundry/ml/autogen/client/docs/RunTagDto.md)
  - [SearchRunsRequestDto](truefoundry/ml/autogen/client/docs/SearchRunsRequestDto.md)
  - [SearchRunsResponseDto](truefoundry/ml/autogen/client/docs/SearchRunsResponseDto.md)
+ - [SerializationFormat](truefoundry/ml/autogen/client/docs/SerializationFormat.md)
  - [SetExperimentTagRequestDto](truefoundry/ml/autogen/client/docs/SetExperimentTagRequestDto.md)
  - [SetTagRequestDto](truefoundry/ml/autogen/client/docs/SetTagRequestDto.md)
  - [SignedURLDto](truefoundry/ml/autogen/client/docs/SignedURLDto.md)

truefoundry/ml/autogen/entities/artifacts.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  artifacts.json
-#   timestamp: 2024-11-22T19:31:48+00:00
+#   timestamp: 2024-12-04T12:00:28+00:00
 
 from __future__ import annotations
 
@@ -52,11 +52,11 @@ class AgentWithFQN(Agent):
 class BaseArtifactVersion(BaseModel):
     description: Optional[constr(max_length=512)] = Field(
         None,
-        description="+label=Description\n+docs=Description of the artifact version",
+        description="+label=Description\n+usage=Description of the artifact or model version\n+docs=Description of the artifact or model version",
     )
     metadata: Dict[str, Any] = Field(
         ...,
-        description="+label=Metadata\n+docs=Metadata for the model version\n+usage=Metadata for the model version\n+uiType=JsonInput",
+        description="+label=Metadata\n+docs=Metadata for the artifact or model version\n+usage=Metadata for the artifact or model version\n+uiType=JsonInput",
     )
 
 
@@ -226,6 +226,23 @@ class ModelConfiguration(BaseModel):
     )
 
 
+class ModelVersionEnvironment(BaseModel):
+    """
+    +label=Environment
+    """
+
+    python_version: Optional[constr(regex=r"^\d+(\.\d+){1,2}([\-\.a-z0-9]+)?$")] = (
+        Field(
+            None,
+            description="+label=Python Version\n+usage=Python version for the model version",
+        )
+    )
+    pip_packages: Optional[List[str]] = Field(
+        None,
+        description="+label=PIP Packages\n+usage=PIP packages for the model version",
+    )
+
+
 class ONNXFramework(BaseModel):
     """
     +docs=ONNX framework for the model version
@@ -259,6 +276,27 @@ class PyTorchFramework(BaseModel):
     )
 
 
+class SerializationFormat(str, Enum):
+    """
+    +label=Serialization format
+    +usage=Serialization format used for the model
+    """
+
+    cloudpickle = "cloudpickle"
+    joblib = "joblib"
+    pickle = "pickle"
+
+
+class InferMethodName(str, Enum):
+    """
+    +label=Inference method name
+    +usage=Name of the method used for inference
+    """
+
+    predict = "predict"
+    predict_proba = "predict_proba"
+
+
 class SklearnFramework(BaseModel):
     """
     +docs=Scikit-learn framework for the model version
@@ -268,6 +306,18 @@ class SklearnFramework(BaseModel):
     type: Literal["sklearn"] = Field(
         ..., description="+label=Type\n+usage=Type of the framework\n+value=sklearn"
     )
+    serialization_format: Optional[SerializationFormat] = Field(
+        None,
+        description="+label=Serialization format\n+usage=Serialization format used for the model",
+    )
+    model_filepath: Optional[str] = Field(
+        None,
+        description="+label=Model file path\n+usage=Relative path to the model file",
+    )
+    infer_method_name: Optional[InferMethodName] = Field(
+        None,
+        description="+label=Inference method name\n+usage=Name of the method used for inference",
+    )
 
 
 class SpaCyFramework(BaseModel):
@@ -357,11 +407,11 @@ class TransformersFramework(BaseModel):
     )
     pipeline_tag: Optional[str] = Field(
         None,
-        description="+label=Pipeline Tag\n+usage=Pipeline tag\n+docs=Pipeline tag for the framework",
+        description="+label=Pipeline Tag\n+usage=The `pipeline()` task this model can be used with e.g. `text-generation`. See [huggingface docs](https://huggingface.co/docs/transformers/main/en/main_classes/pipelines#transformers.pipeline.task) for all possible values\n+docs=Pipeline tag for the framework",
     )
     base_model: Optional[str] = Field(
         None,
-        description="+label=Base Model\n+usage=Base model\n+docs=Base model Id. If this is a finetuned model, this points to the base model used for finetuning",
+        description="+label=Base Model\n+usage=Base model Id. If this is a finetuned model, this points to the base model used for finetuning\n+docs=Base model Id. If this is a finetuned model, this points to the base model used for finetuning",
     )
 
 
@@ -410,6 +460,14 @@ class XGBoostFramework(BaseModel):
     type: Literal["xgboost"] = Field(
         ..., description="+label=Type\n+usage=Type of the framework\n+value=xgboost"
     )
+    serialization_format: Optional[SerializationFormat] = Field(
+        None,
+        description="+label=Serialization format\n+usage=Serialization format used for the model",
+    )
+    model_filepath: Optional[str] = Field(
+        None,
+        description="+label=Model file path\n+usage=Relative path to the model file",
+    )
 
 
 class AgentOpenAPITool(BaseModel):
@@ -533,10 +591,11 @@ class ModelVersion(BaseArtifactVersion):
     ] = Field(
         None, description="+label=Framework\n+usage=Framework for the model version"
     )
+    environment: Optional[ModelVersionEnvironment] = None
     step: conint(ge=0) = Field(0, description="+label=Step")
     model_schema: Optional[Dict[str, Any]] = Field(
         None,
-        description="+label=Model Schema\n+usage=Schema of the model\n+uiType=Hidden",
+        description="+label=Model Schema\n+usage=Schema of the model\n+uiType=JsonInput",
     )
 
 

truefoundry/ml/cli/commands/model_init.py

@@ -0,0 +1,97 @@
+import os
+from typing import Optional
+
+import rich_click as click
+
+from truefoundry.deploy.cli.console import console
+from truefoundry.deploy.cli.const import COMMAND_CLS
+from truefoundry.deploy.cli.util import handle_exception_wrapper
+from truefoundry.ml.autogen.client.models import ModelServer
+from truefoundry.ml.cli.utils import (
+    AppName,
+    NonEmptyString,
+)
+from truefoundry.ml.mlfoundry_api import get_client
+
+
+@click.command(
+    name="model",
+    cls=COMMAND_CLS,
+    help="Generating application code for the specified model version.",
+)
+@click.option(
+    "--name",
+    required=True,
+    type=AppName(),
+    help="Name for the model server deployment",
+    show_default=True,
+)
+@click.option(
+    "--model-version-fqn",
+    "--model_version_fqn",
+    type=NonEmptyString(),
+    required=True,
+    show_default=True,
+    help="Fully Qualified Name (FQN) of the model version to deploy, e.g., 'model:tenant_name/my-model/linear-regression:2'",
+)
+@click.option(
+    "-w",
+    "--workspace-fqn",
+    "--workspace_fqn",
+    type=NonEmptyString(),
+    required=True,
+    show_default=True,
+    help="Fully Qualified Name (FQN) of the workspace to deploy",
+)
+@click.option(
+    "--model-server",
+    "--model_server",
+    type=click.Choice(ModelServer, case_sensitive=False),
+    default=ModelServer.FASTAPI.value,
+    show_default=True,
+    help="Specify the model server (Case Insensitive).",
+)
+@click.option(
+    "--output-dir",
+    "--output_dir",
+    type=click.Path(exists=True, file_okay=False, writable=True),
+    help="Output directory for the model server code",
+    required=False,
+    show_default=True,
+    default=os.getcwd(),
+)
+@handle_exception_wrapper
+def model_init_command(
+    name: str,
+    model_version_fqn: str,
+    workspace_fqn: str,
+    model_server: ModelServer,
+    output_dir: Optional[str],
+):
+    """
+    Generates application code for the specified model version.
+    """
+    ml_client = get_client()
+    console.print(f"Generating application code for {model_version_fqn!r}")
+    output_dir = ml_client._initialize_model_server(
+        name=name,
+        model_version_fqn=model_version_fqn,
+        workspace_fqn=workspace_fqn,
+        model_server=ModelServer[model_server.upper()],
+        output_dir=output_dir,
+    )
+    message = f"""
+[bold green]Model Server code initialized successfully![/bold green]
+
+[bold]Code Location:[/bold] {output_dir}
+
+[bold]Next Steps:[/bold]
+- Navigate to the model server directory:
+[green]cd {output_dir}[/green]
+- Refer to the README file in the directory for further instructions.
+"""
+    console.print(message)
+
+
+def get_model_init_command():
+    return model_init_command

truefoundry/ml/cli/utils.py

@@ -0,0 +1,34 @@
+import rich_click as click
+
+from truefoundry.ml import MlFoundryException
+from truefoundry.ml.validation_utils import (
+    _APP_NAME_REGEX,
+)
+
+
+class AppName(click.ParamType):
+    """
+    Custom ParamType to validate application names.
+    """
+
+    name = "application-name"
+
+    def convert(self, value, param, ctx):
+        try:
+            if not value or not _APP_NAME_REGEX.match(value):
+                raise MlFoundryException(
+                    f"{value!r} must be lowercase and cannot contain spaces. It can only contain alphanumeric characters and hyphens. "
+                    f"Length must be between 1 and 30 characters."
+                )
+        except MlFoundryException as e:
+            self.fail(str(e), param, ctx)
+        return value
+
+
+class NonEmptyString(click.ParamType):
+    name = "non-empty-string"
+
+    def convert(self, value, param, ctx):
+        if isinstance(value, str) and not value.strip():
+            self.fail("Value cannot be empty or contain only spaces.", param, ctx)
+        return value