truefoundry 0.5.5rc1__py3-none-any.whl → 0.5.6__py3-none-any.whl

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

@@ -146,8 +146,8 @@ from truefoundry.ml.autogen.client.models.experiment_tag_dto import ExperimentTa
 from truefoundry.ml.autogen.client.models.export_deployment_files_request_dto import (
     ExportDeploymentFilesRequestDto,
 )
-from truefoundry.ml.autogen.client.models.external_artifact_source import (
-    ExternalArtifactSource,
+from truefoundry.ml.autogen.client.models.external_blob_storage_source import (
+    ExternalBlobStorageSource,
 )
 from truefoundry.ml.autogen.client.models.fast_ai_framework import FastAIFramework
 from truefoundry.ml.autogen.client.models.file_info_dto import FileInfoDto
@@ -374,8 +374,8 @@ from truefoundry.ml.autogen.client.models.trigger_job_run_config_request_dto imp
 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.true_foundry_managed_source import (
+    TrueFoundryManagedSource,
 )
 from truefoundry.ml.autogen.client.models.update_artifact_version_request_dto import (
     UpdateArtifactVersionRequestDto,

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

@@ -16,7 +16,7 @@ from __future__ import annotations
 import json
 import pprint
 import re  # noqa: F401
-from typing import Optional
+from typing import Any, Dict, Optional
 
 from truefoundry.pydantic_v1 import (
     BaseModel,
@@ -33,10 +33,26 @@ class Agent(BaseModel):
     Agent artifact.  # noqa: E501
     """
 
-    type: Optional[StrictStr] = "agent"
-    name: constr(strict=True) = Field(
-        default=..., description="Names cannot repeat accross agent +uiType=Hidden"
+    description: Optional[constr(strict=True, max_length=512)] = Field(
+        default=None, description="+sort=2 +label=Description"
+    )
+    metadata: Dict[str, Any] = Field(
+        default=...,
+        description="+label=Metadata +docs=Key value pairs to store additional metadata +usage=Key value pairs to store additional metadata +uiType=JsonInput",
+    )
+    version_alias: Optional[constr(strict=True, max_length=128)] = Field(
+        default=None,
+        description="+label=Version Alias +usage=The version alias should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v1-prod, v3-dev, etc) +docs=The version alias should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v1-prod, v3-dev, etc) +message=The version alias should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v1-prod, v3-dev, etc)",
+    )
+    name: Optional[constr(strict=True, max_length=256)] = Field(
+        default=None,
+        description="+sort=1 +label=Name +message=The name should start with lowercase alphabets  and can contain alphanumeric and can include '-' in between",
     )
+    ml_repo: Optional[constr(strict=True)] = Field(
+        default=None,
+        description="+label=ML Repo +usage=name of the ML Repo +docs=name of the ML Repo +uiType=Hidden",
+    )
+    type: Optional[StrictStr] = "agent"
     available_tools: conlist(constr(strict=True, min_length=1)) = Field(
         default=..., description="+sort=30 +uiType=Hidden"
     )
@@ -52,33 +68,64 @@ class Agent(BaseModel):
         description='+sort=40 +uiType=EnabledModelSelector +uiProps={"searchable":true,"modelType":"chat","providerType":"openai"}',
     )
     __properties = [
-        "type",
+        "description",
+        "metadata",
+        "version_alias",
         "name",
+        "ml_repo",
+        "type",
         "available_tools",
         "goal",
         "instruction",
         "model_id",
     ]
 
-    @validator("type")
-    def type_validate_enum(cls, value):
-        """Validates the enum"""
+    @validator("version_alias")
+    def version_alias_validate_regular_expression(cls, value):
+        """Validates the regular expression"""
         if value is None:
             return value
 
-        if value not in ("agent",):
-            raise ValueError("must be one of enum values ('agent')")
+        if not re.match(r"^v[a-zA-Z0-9.-]*([a-zA-Z0-9]+)$", value):
+            raise ValueError(
+                r"must validate the regular expression /^v[a-zA-Z0-9.-]*([a-zA-Z0-9]+)$/"
+            )
         return value
 
     @validator("name")
     def name_validate_regular_expression(cls, value):
         """Validates the regular expression"""
-        if not re.match(r"^[a-zA-Z][a-zA-Z0-9\-]{1,30}[a-zA-Z0-9]$", value):
+        if value is None:
+            return value
+
+        if not re.match(r"^[A-Za-z0-9_\-]+$", value):
+            raise ValueError(
+                r"must validate the regular expression /^[A-Za-z0-9_\-]+$/"
+            )
+        return value
+
+    @validator("ml_repo")
+    def ml_repo_validate_regular_expression(cls, value):
+        """Validates the regular expression"""
+        if value is None:
+            return value
+
+        if not re.match(r"^[a-zA-Z][a-zA-Z0-9\-]{1,98}[a-zA-Z0-9]$", value):
             raise ValueError(
-                r"must validate the regular expression /^[a-zA-Z][a-zA-Z0-9\-]{1,30}[a-zA-Z0-9]$/"
+                r"must validate the regular expression /^[a-zA-Z][a-zA-Z0-9\-]{1,98}[a-zA-Z0-9]$/"
             )
         return value
 
+    @validator("type")
+    def type_validate_enum(cls, value):
+        """Validates the enum"""
+        if value is None:
+            return value
+
+        if value not in ("agent",):
+            raise ValueError("must be one of enum values ('agent')")
+        return value
+
     class Config:
         """Pydantic configuration"""
 
@@ -114,8 +161,12 @@ class Agent(BaseModel):
 
         _obj = Agent.parse_obj(
             {
-                "type": obj.get("type") if obj.get("type") is not None else "agent",
+                "description": obj.get("description"),
+                "metadata": obj.get("metadata"),
+                "version_alias": obj.get("version_alias"),
                 "name": obj.get("name"),
+                "ml_repo": obj.get("ml_repo"),
+                "type": obj.get("type") if obj.get("type") is not None else "agent",
                 "available_tools": obj.get("available_tools"),
                 "goal": obj.get("goal"),
                 "instruction": obj.get("instruction"),

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

@@ -16,7 +16,7 @@ from __future__ import annotations
 import json
 import pprint
 import re  # noqa: F401
-from typing import Dict, Optional
+from typing import Any, Dict, Optional
 
 from truefoundry.ml.autogen.client.models.method import Method
 from truefoundry.ml.autogen.client.models.openapi_spec import OpenapiSpec
@@ -28,13 +28,26 @@ class AgentOpenAPITool(BaseModel):
     OpenAPI Tool artifact.  # noqa: E501
     """
 
-    type: Optional[StrictStr] = "openapi-tool"
-    name: constr(strict=True) = Field(
-        default=..., description="Names cannot repeat accross tools +uiType=Hidden"
+    description: Optional[constr(strict=True, max_length=512)] = Field(
+        default=None, description="+sort=2 +label=Description"
+    )
+    metadata: Dict[str, Any] = Field(
+        default=...,
+        description="+label=Metadata +docs=Key value pairs to store additional metadata +usage=Key value pairs to store additional metadata +uiType=JsonInput",
+    )
+    version_alias: Optional[constr(strict=True, max_length=128)] = Field(
+        default=None,
+        description="+label=Version Alias +usage=The version alias should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v1-prod, v3-dev, etc) +docs=The version alias should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v1-prod, v3-dev, etc) +message=The version alias should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v1-prod, v3-dev, etc)",
+    )
+    name: Optional[constr(strict=True, max_length=256)] = Field(
+        default=None,
+        description="+sort=1 +label=Name +message=The name should start with lowercase alphabets  and can contain alphanumeric and can include '-' in between",
     )
-    description: constr(strict=True, max_length=128, min_length=1) = Field(
-        default=..., description="+sort=10 +uiType=TextArea"
+    ml_repo: Optional[constr(strict=True)] = Field(
+        default=None,
+        description="+label=ML Repo +usage=name of the ML Repo +docs=name of the ML Repo +uiType=Hidden",
     )
+    type: Optional[StrictStr] = "openapi-tool"
     openapi_spec: OpenapiSpec = Field(...)
     base_url: constr(strict=True) = Field(default=..., description="+sort=30")
     path: constr(strict=True, min_length=1) = Field(
@@ -47,9 +60,12 @@ class AgentOpenAPITool(BaseModel):
         description='+sort=60 +uiType=KV +uiProps={"allowSecrets":true,"secretConfig":{"enableNew":true,"hideOptions":true}}',
     )
     __properties = [
-        "type",
-        "name",
         "description",
+        "metadata",
+        "version_alias",
+        "name",
+        "ml_repo",
+        "type",
         "openapi_spec",
         "base_url",
         "path",
@@ -57,25 +73,52 @@ class AgentOpenAPITool(BaseModel):
         "headers",
     ]
 
-    @validator("type")
-    def type_validate_enum(cls, value):
-        """Validates the enum"""
+    @validator("version_alias")
+    def version_alias_validate_regular_expression(cls, value):
+        """Validates the regular expression"""
         if value is None:
             return value
 
-        if value not in ("openapi-tool",):
-            raise ValueError("must be one of enum values ('openapi-tool')")
+        if not re.match(r"^v[a-zA-Z0-9.-]*([a-zA-Z0-9]+)$", value):
+            raise ValueError(
+                r"must validate the regular expression /^v[a-zA-Z0-9.-]*([a-zA-Z0-9]+)$/"
+            )
         return value
 
     @validator("name")
     def name_validate_regular_expression(cls, value):
         """Validates the regular expression"""
-        if not re.match(r"^[a-zA-Z][a-zA-Z0-9\-]{1,30}[a-zA-Z0-9]$", value):
+        if value is None:
+            return value
+
+        if not re.match(r"^[A-Za-z0-9_\-]+$", value):
+            raise ValueError(
+                r"must validate the regular expression /^[A-Za-z0-9_\-]+$/"
+            )
+        return value
+
+    @validator("ml_repo")
+    def ml_repo_validate_regular_expression(cls, value):
+        """Validates the regular expression"""
+        if value is None:
+            return value
+
+        if not re.match(r"^[a-zA-Z][a-zA-Z0-9\-]{1,98}[a-zA-Z0-9]$", value):
             raise ValueError(
-                r"must validate the regular expression /^[a-zA-Z][a-zA-Z0-9\-]{1,30}[a-zA-Z0-9]$/"
+                r"must validate the regular expression /^[a-zA-Z][a-zA-Z0-9\-]{1,98}[a-zA-Z0-9]$/"
             )
         return value
 
+    @validator("type")
+    def type_validate_enum(cls, value):
+        """Validates the enum"""
+        if value is None:
+            return value
+
+        if value not in ("openapi-tool",):
+            raise ValueError("must be one of enum values ('openapi-tool')")
+        return value
+
     @validator("base_url")
     def base_url_validate_regular_expression(cls, value):
         """Validates the regular expression"""
@@ -126,11 +169,14 @@ class AgentOpenAPITool(BaseModel):
 
         _obj = AgentOpenAPITool.parse_obj(
             {
+                "description": obj.get("description"),
+                "metadata": obj.get("metadata"),
+                "version_alias": obj.get("version_alias"),
+                "name": obj.get("name"),
+                "ml_repo": obj.get("ml_repo"),
                 "type": obj.get("type")
                 if obj.get("type") is not None
                 else "openapi-tool",
-                "name": obj.get("name"),
-                "description": obj.get("description"),
                 "openapi_spec": OpenapiSpec.from_dict(obj.get("openapi_spec"))
                 if obj.get("openapi_spec") is not None
                 else None,

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

@@ -16,7 +16,7 @@ from __future__ import annotations
 import json
 import pprint
 import re  # noqa: F401
-from typing import Dict, Optional
+from typing import Any, Dict, Optional
 
 from truefoundry.ml.autogen.client.models.method import Method
 from truefoundry.ml.autogen.client.models.openapi_spec import OpenapiSpec
@@ -28,13 +28,26 @@ class AgentOpenAPIToolWithFQN(BaseModel):
     AgentOpenAPIToolWithFQN
     """
 
-    type: StrictStr = Field(default=..., description="+value=openapi-tool")
-    name: constr(strict=True) = Field(
-        default=..., description="Names cannot repeat accross tools +uiType=Hidden"
+    description: Optional[constr(strict=True, max_length=512)] = Field(
+        default=None, description="+sort=2 +label=Description"
+    )
+    metadata: Dict[str, Any] = Field(
+        default=...,
+        description="+label=Metadata +docs=Key value pairs to store additional metadata +usage=Key value pairs to store additional metadata +uiType=JsonInput",
+    )
+    version_alias: Optional[constr(strict=True, max_length=128)] = Field(
+        default=None,
+        description="+label=Version Alias +usage=The version alias should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v1-prod, v3-dev, etc) +docs=The version alias should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v1-prod, v3-dev, etc) +message=The version alias should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v1-prod, v3-dev, etc)",
     )
-    description: constr(strict=True, max_length=128, min_length=1) = Field(
-        default=..., description="+sort=10 +uiType=TextArea"
+    name: Optional[constr(strict=True, max_length=256)] = Field(
+        default=None,
+        description="+sort=1 +label=Name +message=The name should start with lowercase alphabets  and can contain alphanumeric and can include '-' in between",
+    )
+    ml_repo: Optional[constr(strict=True)] = Field(
+        default=None,
+        description="+label=ML Repo +usage=name of the ML Repo +docs=name of the ML Repo +uiType=Hidden",
     )
+    type: StrictStr = Field(default=..., description="+value=openapi-tool")
     openapi_spec: OpenapiSpec = Field(...)
     base_url: constr(strict=True) = Field(default=..., description="+sort=30")
     path: constr(strict=True, min_length=1) = Field(
@@ -49,9 +62,12 @@ class AgentOpenAPIToolWithFQN(BaseModel):
     id: StrictStr = Field(...)
     fqn: StrictStr = Field(...)
     __properties = [
-        "type",
-        "name",
         "description",
+        "metadata",
+        "version_alias",
+        "name",
+        "ml_repo",
+        "type",
         "openapi_spec",
         "base_url",
         "path",
@@ -61,22 +77,49 @@ class AgentOpenAPIToolWithFQN(BaseModel):
         "fqn",
     ]
 
-    @validator("type")
-    def type_validate_enum(cls, value):
-        """Validates the enum"""
-        if value not in ("openapi-tool",):
-            raise ValueError("must be one of enum values ('openapi-tool')")
+    @validator("version_alias")
+    def version_alias_validate_regular_expression(cls, value):
+        """Validates the regular expression"""
+        if value is None:
+            return value
+
+        if not re.match(r"^v[a-zA-Z0-9.-]*([a-zA-Z0-9]+)$", value):
+            raise ValueError(
+                r"must validate the regular expression /^v[a-zA-Z0-9.-]*([a-zA-Z0-9]+)$/"
+            )
         return value
 
     @validator("name")
     def name_validate_regular_expression(cls, value):
         """Validates the regular expression"""
-        if not re.match(r"^[a-zA-Z][a-zA-Z0-9\-]{1,30}[a-zA-Z0-9]$", value):
+        if value is None:
+            return value
+
+        if not re.match(r"^[A-Za-z0-9_\-]+$", value):
+            raise ValueError(
+                r"must validate the regular expression /^[A-Za-z0-9_\-]+$/"
+            )
+        return value
+
+    @validator("ml_repo")
+    def ml_repo_validate_regular_expression(cls, value):
+        """Validates the regular expression"""
+        if value is None:
+            return value
+
+        if not re.match(r"^[a-zA-Z][a-zA-Z0-9\-]{1,98}[a-zA-Z0-9]$", value):
             raise ValueError(
-                r"must validate the regular expression /^[a-zA-Z][a-zA-Z0-9\-]{1,30}[a-zA-Z0-9]$/"
+                r"must validate the regular expression /^[a-zA-Z][a-zA-Z0-9\-]{1,98}[a-zA-Z0-9]$/"
             )
         return value
 
+    @validator("type")
+    def type_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in ("openapi-tool",):
+            raise ValueError("must be one of enum values ('openapi-tool')")
+        return value
+
     @validator("base_url")
     def base_url_validate_regular_expression(cls, value):
         """Validates the regular expression"""
@@ -127,9 +170,12 @@ class AgentOpenAPIToolWithFQN(BaseModel):
 
         _obj = AgentOpenAPIToolWithFQN.parse_obj(
             {
-                "type": obj.get("type"),
-                "name": obj.get("name"),
                 "description": obj.get("description"),
+                "metadata": obj.get("metadata"),
+                "version_alias": obj.get("version_alias"),
+                "name": obj.get("name"),
+                "ml_repo": obj.get("ml_repo"),
+                "type": obj.get("type"),
                 "openapi_spec": OpenapiSpec.from_dict(obj.get("openapi_spec"))
                 if obj.get("openapi_spec") is not None
                 else None,

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

@@ -16,6 +16,7 @@ from __future__ import annotations
 import json
 import pprint
 import re  # noqa: F401
+from typing import Any, Dict, Optional
 
 from truefoundry.pydantic_v1 import (
     BaseModel,
@@ -32,10 +33,26 @@ class AgentWithFQN(BaseModel):
     AgentWithFQN
     """
 
-    type: StrictStr = Field(default=..., description="+value=agent")
-    name: constr(strict=True) = Field(
-        default=..., description="Names cannot repeat accross agent +uiType=Hidden"
+    description: Optional[constr(strict=True, max_length=512)] = Field(
+        default=None, description="+sort=2 +label=Description"
+    )
+    metadata: Dict[str, Any] = Field(
+        default=...,
+        description="+label=Metadata +docs=Key value pairs to store additional metadata +usage=Key value pairs to store additional metadata +uiType=JsonInput",
+    )
+    version_alias: Optional[constr(strict=True, max_length=128)] = Field(
+        default=None,
+        description="+label=Version Alias +usage=The version alias should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v1-prod, v3-dev, etc) +docs=The version alias should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v1-prod, v3-dev, etc) +message=The version alias should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v1-prod, v3-dev, etc)",
+    )
+    name: Optional[constr(strict=True, max_length=256)] = Field(
+        default=None,
+        description="+sort=1 +label=Name +message=The name should start with lowercase alphabets  and can contain alphanumeric and can include '-' in between",
     )
+    ml_repo: Optional[constr(strict=True)] = Field(
+        default=None,
+        description="+label=ML Repo +usage=name of the ML Repo +docs=name of the ML Repo +uiType=Hidden",
+    )
+    type: StrictStr = Field(default=..., description="+value=agent")
     available_tools: conlist(constr(strict=True, min_length=1)) = Field(
         default=..., description="+sort=30 +uiType=Hidden"
     )
@@ -53,8 +70,12 @@ class AgentWithFQN(BaseModel):
     id: StrictStr = Field(...)
     fqn: StrictStr = Field(...)
     __properties = [
-        "type",
+        "description",
+        "metadata",
+        "version_alias",
         "name",
+        "ml_repo",
+        "type",
         "available_tools",
         "goal",
         "instruction",
@@ -63,22 +84,49 @@ class AgentWithFQN(BaseModel):
         "fqn",
     ]
 
-    @validator("type")
-    def type_validate_enum(cls, value):
-        """Validates the enum"""
-        if value not in ("agent",):
-            raise ValueError("must be one of enum values ('agent')")
+    @validator("version_alias")
+    def version_alias_validate_regular_expression(cls, value):
+        """Validates the regular expression"""
+        if value is None:
+            return value
+
+        if not re.match(r"^v[a-zA-Z0-9.-]*([a-zA-Z0-9]+)$", value):
+            raise ValueError(
+                r"must validate the regular expression /^v[a-zA-Z0-9.-]*([a-zA-Z0-9]+)$/"
+            )
         return value
 
     @validator("name")
     def name_validate_regular_expression(cls, value):
         """Validates the regular expression"""
-        if not re.match(r"^[a-zA-Z][a-zA-Z0-9\-]{1,30}[a-zA-Z0-9]$", value):
+        if value is None:
+            return value
+
+        if not re.match(r"^[A-Za-z0-9_\-]+$", value):
             raise ValueError(
-                r"must validate the regular expression /^[a-zA-Z][a-zA-Z0-9\-]{1,30}[a-zA-Z0-9]$/"
+                r"must validate the regular expression /^[A-Za-z0-9_\-]+$/"
             )
         return value
 
+    @validator("ml_repo")
+    def ml_repo_validate_regular_expression(cls, value):
+        """Validates the regular expression"""
+        if value is None:
+            return value
+
+        if not re.match(r"^[a-zA-Z][a-zA-Z0-9\-]{1,98}[a-zA-Z0-9]$", value):
+            raise ValueError(
+                r"must validate the regular expression /^[a-zA-Z][a-zA-Z0-9\-]{1,98}[a-zA-Z0-9]$/"
+            )
+        return value
+
+    @validator("type")
+    def type_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in ("agent",):
+            raise ValueError("must be one of enum values ('agent')")
+        return value
+
     class Config:
         """Pydantic configuration"""
 
@@ -114,8 +162,12 @@ class AgentWithFQN(BaseModel):
 
         _obj = AgentWithFQN.parse_obj(
             {
-                "type": obj.get("type"),
+                "description": obj.get("description"),
+                "metadata": obj.get("metadata"),
+                "version_alias": obj.get("version_alias"),
                 "name": obj.get("name"),
+                "ml_repo": obj.get("ml_repo"),
+                "type": obj.get("type"),
                 "available_tools": obj.get("available_tools"),
                 "goal": obj.get("goal"),
                 "instruction": obj.get("instruction"),

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

@@ -35,16 +35,23 @@ class ArtifactVersionManifest(BaseModel):
     """
 
     description: Optional[constr(strict=True, max_length=512)] = Field(
-        default=None,
-        description="+label=Description +usage=Description of the artifact or model version +docs=Description of the artifact or model version",
+        default=None, description="+sort=2 +label=Description"
     )
     metadata: Dict[str, Any] = Field(
         default=...,
-        description="+label=Metadata +docs=Metadata for the artifact or model version +usage=Metadata for the artifact or model version +uiType=JsonInput",
+        description="+label=Metadata +docs=Key value pairs to store additional metadata +usage=Key value pairs to store additional metadata +uiType=JsonInput",
     )
     version_alias: Optional[constr(strict=True, max_length=128)] = Field(
         default=None,
-        description="+label=Version Alias +usage=The version alias for artifact or model version which should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v-prod, v-dev, etc) +docs=The version alias for artifact or model version which should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v-prod, v-dev, etc) +message=The version alias should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v-prod, v-dev, etc)",
+        description="+label=Version Alias +usage=The version alias should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v1-prod, v3-dev, etc) +docs=The version alias should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v1-prod, v3-dev, etc) +message=The version alias should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v1-prod, v3-dev, etc)",
+    )
+    name: Optional[constr(strict=True, max_length=256)] = Field(
+        default=None,
+        description="+sort=1 +label=Name +message=The name should start with lowercase alphabets  and can contain alphanumeric and can include '-' in between",
+    )
+    ml_repo: Optional[constr(strict=True)] = Field(
+        default=None,
+        description="+label=ML Repo +usage=name of the ML Repo +docs=name of the ML Repo +uiType=Hidden",
     )
     type: Optional[StrictStr] = "artifact-version"
     source: Source = Field(...)
@@ -55,6 +62,8 @@ class ArtifactVersionManifest(BaseModel):
         "description",
         "metadata",
         "version_alias",
+        "name",
+        "ml_repo",
         "type",
         "source",
         "step",
@@ -72,6 +81,30 @@ class ArtifactVersionManifest(BaseModel):
             )
         return value
 
+    @validator("name")
+    def name_validate_regular_expression(cls, value):
+        """Validates the regular expression"""
+        if value is None:
+            return value
+
+        if not re.match(r"^[A-Za-z0-9_\-]+$", value):
+            raise ValueError(
+                r"must validate the regular expression /^[A-Za-z0-9_\-]+$/"
+            )
+        return value
+
+    @validator("ml_repo")
+    def ml_repo_validate_regular_expression(cls, value):
+        """Validates the regular expression"""
+        if value is None:
+            return value
+
+        if not re.match(r"^[a-zA-Z][a-zA-Z0-9\-]{1,98}[a-zA-Z0-9]$", value):
+            raise ValueError(
+                r"must validate the regular expression /^[a-zA-Z][a-zA-Z0-9\-]{1,98}[a-zA-Z0-9]$/"
+            )
+        return value
+
     @validator("type")
     def type_validate_enum(cls, value):
         """Validates the enum"""
@@ -123,6 +156,8 @@ class ArtifactVersionManifest(BaseModel):
                 "description": obj.get("description"),
                 "metadata": obj.get("metadata"),
                 "version_alias": obj.get("version_alias"),
+                "name": obj.get("name"),
+                "ml_repo": obj.get("ml_repo"),
                 "type": obj.get("type")
                 if obj.get("type") is not None
                 else "artifact-version",