truefoundry 0.5.10rc2__py3-none-any.whl → 0.5.11rc1__py3-none-any.whl

truefoundry/ml/{autogen → _autogen}/client/models/manifest.py

@@ -18,17 +18,15 @@ import pprint
 import re  # noqa: F401
 from typing import TYPE_CHECKING, Any, List, Optional, Union
 
-from truefoundry.ml.autogen.client.models.agent_manifest import AgentManifest
-from truefoundry.ml.autogen.client.models.agent_open_api_tool_manifest import (
+from truefoundry.ml._autogen.client.models.agent_manifest import AgentManifest
+from truefoundry.ml._autogen.client.models.agent_open_api_tool_manifest import (
     AgentOpenAPIToolManifest,
 )
-from truefoundry.ml.autogen.client.models.artifact_version_manifest import (
-    ArtifactVersionManifest,
-)
-from truefoundry.ml.autogen.client.models.chat_prompt_manifest import ChatPromptManifest
-from truefoundry.ml.autogen.client.models.model_version_manifest import (
-    ModelVersionManifest,
+from truefoundry.ml._autogen.client.models.artifact_manifest import ArtifactManifest
+from truefoundry.ml._autogen.client.models.chat_prompt_manifest import (
+    ChatPromptManifest,
 )
+from truefoundry.ml._autogen.client.models.model_manifest import ModelManifest
 from truefoundry.pydantic_v1 import (
     BaseModel,
     Field,
@@ -39,9 +37,9 @@ from truefoundry.pydantic_v1 import (
 MANIFEST_ONE_OF_SCHEMAS = [
     "AgentManifest",
     "AgentOpenAPIToolManifest",
-    "ArtifactVersionManifest",
+    "ArtifactManifest",
     "ChatPromptManifest",
-    "ModelVersionManifest",
+    "ModelManifest",
 ]
 
 
@@ -50,23 +48,23 @@ class Manifest(BaseModel):
     Manifest
     """
 
-    # data type: ModelVersionManifest
-    oneof_schema_1_validator: Optional[ModelVersionManifest] = None
+    # data type: ModelManifest
+    oneof_schema_1_validator: Optional[ModelManifest] = None
     # data type: ChatPromptManifest
     oneof_schema_2_validator: Optional[ChatPromptManifest] = None
     # data type: AgentOpenAPIToolManifest
     oneof_schema_3_validator: Optional[AgentOpenAPIToolManifest] = None
     # data type: AgentManifest
     oneof_schema_4_validator: Optional[AgentManifest] = None
-    # data type: ArtifactVersionManifest
-    oneof_schema_5_validator: Optional[ArtifactVersionManifest] = None
+    # data type: ArtifactManifest
+    oneof_schema_5_validator: Optional[ArtifactManifest] = None
     if TYPE_CHECKING:
         actual_instance: Union[
             AgentManifest,
             AgentOpenAPIToolManifest,
-            ArtifactVersionManifest,
+            ArtifactManifest,
             ChatPromptManifest,
-            ModelVersionManifest,
+            ModelManifest,
         ]
     else:
         actual_instance: Any
@@ -96,10 +94,10 @@ class Manifest(BaseModel):
         instance = Manifest.construct()
         error_messages = []
         match = 0
-        # validate data type: ModelVersionManifest
-        if not isinstance(v, ModelVersionManifest):
+        # validate data type: ModelManifest
+        if not isinstance(v, ModelManifest):
             error_messages.append(
-                f"Error! Input type `{type(v)}` is not `ModelVersionManifest`"
+                f"Error! Input type `{type(v)}` is not `ModelManifest`"
             )
         else:
             match += 1
@@ -124,23 +122,23 @@ class Manifest(BaseModel):
             )
         else:
             match += 1
-        # validate data type: ArtifactVersionManifest
-        if not isinstance(v, ArtifactVersionManifest):
+        # validate data type: ArtifactManifest
+        if not isinstance(v, ArtifactManifest):
             error_messages.append(
-                f"Error! Input type `{type(v)}` is not `ArtifactVersionManifest`"
+                f"Error! Input type `{type(v)}` is not `ArtifactManifest`"
             )
         else:
             match += 1
         if match > 1:
             # more than 1 match
             raise ValueError(
-                "Multiple matches found when setting `actual_instance` in Manifest with oneOf schemas: AgentManifest, AgentOpenAPIToolManifest, ArtifactVersionManifest, ChatPromptManifest, ModelVersionManifest. Details: "
+                "Multiple matches found when setting `actual_instance` in Manifest with oneOf schemas: AgentManifest, AgentOpenAPIToolManifest, ArtifactManifest, ChatPromptManifest, ModelManifest. Details: "
                 + ", ".join(error_messages)
             )
         elif match == 0:
             # no match
             raise ValueError(
-                "No match found when setting `actual_instance` in Manifest with oneOf schemas: AgentManifest, AgentOpenAPIToolManifest, ArtifactVersionManifest, ChatPromptManifest, ModelVersionManifest. Details: "
+                "No match found when setting `actual_instance` in Manifest with oneOf schemas: AgentManifest, AgentOpenAPIToolManifest, ArtifactManifest, ChatPromptManifest, ModelManifest. Details: "
                 + ", ".join(error_messages)
             )
         else:
@@ -157,9 +155,9 @@ class Manifest(BaseModel):
         error_messages = []
         match = 0
 
-        # deserialize data into ModelVersionManifest
+        # deserialize data into ModelManifest
         try:
-            instance.actual_instance = ModelVersionManifest.from_json(json_str)
+            instance.actual_instance = ModelManifest.from_json(json_str)
             match += 1
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
@@ -181,9 +179,9 @@ class Manifest(BaseModel):
             match += 1
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
-        # deserialize data into ArtifactVersionManifest
+        # deserialize data into ArtifactManifest
         try:
-            instance.actual_instance = ArtifactVersionManifest.from_json(json_str)
+            instance.actual_instance = ArtifactManifest.from_json(json_str)
             match += 1
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
@@ -191,13 +189,13 @@ class Manifest(BaseModel):
         if match > 1:
             # more than 1 match
             raise ValueError(
-                "Multiple matches found when deserializing the JSON string into Manifest with oneOf schemas: AgentManifest, AgentOpenAPIToolManifest, ArtifactVersionManifest, ChatPromptManifest, ModelVersionManifest. Details: "
+                "Multiple matches found when deserializing the JSON string into Manifest with oneOf schemas: AgentManifest, AgentOpenAPIToolManifest, ArtifactManifest, ChatPromptManifest, ModelManifest. Details: "
                 + ", ".join(error_messages)
             )
         elif match == 0:
             # no match
             raise ValueError(
-                "No match found when deserializing the JSON string into Manifest with oneOf schemas: AgentManifest, AgentOpenAPIToolManifest, ArtifactVersionManifest, ChatPromptManifest, ModelVersionManifest. Details: "
+                "No match found when deserializing the JSON string into Manifest with oneOf schemas: AgentManifest, AgentOpenAPIToolManifest, ArtifactManifest, ChatPromptManifest, ModelManifest. Details: "
                 + ", ".join(error_messages)
             )
         else:

truefoundry/ml/{autogen → _autogen}/client/models/manifest1.py

@@ -18,17 +18,15 @@ import pprint
 import re  # noqa: F401
 from typing import TYPE_CHECKING, Any, List, Optional, Union
 
-from truefoundry.ml.autogen.client.models.agent_manifest import AgentManifest
-from truefoundry.ml.autogen.client.models.agent_open_api_tool_manifest import (
+from truefoundry.ml._autogen.client.models.agent_manifest import AgentManifest
+from truefoundry.ml._autogen.client.models.agent_open_api_tool_manifest import (
     AgentOpenAPIToolManifest,
 )
-from truefoundry.ml.autogen.client.models.artifact_version_manifest import (
-    ArtifactVersionManifest,
-)
-from truefoundry.ml.autogen.client.models.chat_prompt_manifest import ChatPromptManifest
-from truefoundry.ml.autogen.client.models.model_version_manifest import (
-    ModelVersionManifest,
+from truefoundry.ml._autogen.client.models.artifact_manifest import ArtifactManifest
+from truefoundry.ml._autogen.client.models.chat_prompt_manifest import (
+    ChatPromptManifest,
 )
+from truefoundry.ml._autogen.client.models.model_manifest import ModelManifest
 from truefoundry.pydantic_v1 import (
     BaseModel,
     Field,
@@ -39,9 +37,9 @@ from truefoundry.pydantic_v1 import (
 MANIFEST1_ANY_OF_SCHEMAS = [
     "AgentManifest",
     "AgentOpenAPIToolManifest",
-    "ArtifactVersionManifest",
+    "ArtifactManifest",
     "ChatPromptManifest",
-    "ModelVersionManifest",
+    "ModelManifest",
 ]
 
 
@@ -50,23 +48,23 @@ class Manifest1(BaseModel):
     Manifest1
     """
 
-    # data type: ModelVersionManifest
-    anyof_schema_1_validator: Optional[ModelVersionManifest] = None
+    # data type: ModelManifest
+    anyof_schema_1_validator: Optional[ModelManifest] = None
     # data type: ChatPromptManifest
     anyof_schema_2_validator: Optional[ChatPromptManifest] = None
     # data type: AgentOpenAPIToolManifest
     anyof_schema_3_validator: Optional[AgentOpenAPIToolManifest] = None
     # data type: AgentManifest
     anyof_schema_4_validator: Optional[AgentManifest] = None
-    # data type: ArtifactVersionManifest
-    anyof_schema_5_validator: Optional[ArtifactVersionManifest] = None
+    # data type: ArtifactManifest
+    anyof_schema_5_validator: Optional[ArtifactManifest] = None
     if TYPE_CHECKING:
         actual_instance: Union[
             AgentManifest,
             AgentOpenAPIToolManifest,
-            ArtifactVersionManifest,
+            ArtifactManifest,
             ChatPromptManifest,
-            ModelVersionManifest,
+            ModelManifest,
         ]
     else:
         actual_instance: Any
@@ -93,10 +91,10 @@ class Manifest1(BaseModel):
     def actual_instance_must_validate_anyof(cls, v):
         instance = Manifest1.construct()
         error_messages = []
-        # validate data type: ModelVersionManifest
-        if not isinstance(v, ModelVersionManifest):
+        # validate data type: ModelManifest
+        if not isinstance(v, ModelManifest):
             error_messages.append(
-                f"Error! Input type `{type(v)}` is not `ModelVersionManifest`"
+                f"Error! Input type `{type(v)}` is not `ModelManifest`"
             )
         else:
             return v
@@ -125,10 +123,10 @@ class Manifest1(BaseModel):
         else:
             return v
 
-        # validate data type: ArtifactVersionManifest
-        if not isinstance(v, ArtifactVersionManifest):
+        # validate data type: ArtifactManifest
+        if not isinstance(v, ArtifactManifest):
             error_messages.append(
-                f"Error! Input type `{type(v)}` is not `ArtifactVersionManifest`"
+                f"Error! Input type `{type(v)}` is not `ArtifactManifest`"
             )
         else:
             return v
@@ -136,7 +134,7 @@ class Manifest1(BaseModel):
         if error_messages:
             # no match
             raise ValueError(
-                "No match found when setting the actual_instance in Manifest1 with anyOf schemas: AgentManifest, AgentOpenAPIToolManifest, ArtifactVersionManifest, ChatPromptManifest, ModelVersionManifest. Details: "
+                "No match found when setting the actual_instance in Manifest1 with anyOf schemas: AgentManifest, AgentOpenAPIToolManifest, ArtifactManifest, ChatPromptManifest, ModelManifest. Details: "
                 + ", ".join(error_messages)
             )
         else:
@@ -151,9 +149,9 @@ class Manifest1(BaseModel):
         """Returns the object represented by the json string"""
         instance = Manifest1.construct()
         error_messages = []
-        # anyof_schema_1_validator: Optional[ModelVersionManifest] = None
+        # anyof_schema_1_validator: Optional[ModelManifest] = None
         try:
-            instance.actual_instance = ModelVersionManifest.from_json(json_str)
+            instance.actual_instance = ModelManifest.from_json(json_str)
             return instance
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
@@ -175,9 +173,9 @@ class Manifest1(BaseModel):
             return instance
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
-        # anyof_schema_5_validator: Optional[ArtifactVersionManifest] = None
+        # anyof_schema_5_validator: Optional[ArtifactManifest] = None
         try:
-            instance.actual_instance = ArtifactVersionManifest.from_json(json_str)
+            instance.actual_instance = ArtifactManifest.from_json(json_str)
             return instance
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
@@ -185,7 +183,7 @@ class Manifest1(BaseModel):
         if error_messages:
             # no match
             raise ValueError(
-                "No match found when deserializing the JSON string into Manifest1 with anyOf schemas: AgentManifest, AgentOpenAPIToolManifest, ArtifactVersionManifest, ChatPromptManifest, ModelVersionManifest. Details: "
+                "No match found when deserializing the JSON string into Manifest1 with anyOf schemas: AgentManifest, AgentOpenAPIToolManifest, ArtifactManifest, ChatPromptManifest, ModelManifest. Details: "
                 + ", ".join(error_messages)
             )
         else:

truefoundry/ml/{autogen → _autogen}/client/models/manifest2.py

@@ -18,14 +18,14 @@ import pprint
 import re  # noqa: F401
 from typing import TYPE_CHECKING, Any, List, Optional, Union
 
-from truefoundry.ml.autogen.client.models.agent_manifest import AgentManifest
-from truefoundry.ml.autogen.client.models.agent_open_api_tool_manifest import (
+from truefoundry.ml._autogen.client.models.agent_manifest import AgentManifest
+from truefoundry.ml._autogen.client.models.agent_open_api_tool_manifest import (
     AgentOpenAPIToolManifest,
 )
-from truefoundry.ml.autogen.client.models.artifact_version_manifest import (
-    ArtifactVersionManifest,
+from truefoundry.ml._autogen.client.models.artifact_manifest import ArtifactManifest
+from truefoundry.ml._autogen.client.models.chat_prompt_manifest import (
+    ChatPromptManifest,
 )
-from truefoundry.ml.autogen.client.models.chat_prompt_manifest import ChatPromptManifest
 from truefoundry.pydantic_v1 import (
     BaseModel,
     Field,
@@ -36,7 +36,7 @@ from truefoundry.pydantic_v1 import (
 MANIFEST2_ANY_OF_SCHEMAS = [
     "AgentManifest",
     "AgentOpenAPIToolManifest",
-    "ArtifactVersionManifest",
+    "ArtifactManifest",
     "ChatPromptManifest",
 ]
 
@@ -52,13 +52,13 @@ class Manifest2(BaseModel):
     anyof_schema_2_validator: Optional[AgentOpenAPIToolManifest] = None
     # data type: AgentManifest
     anyof_schema_3_validator: Optional[AgentManifest] = None
-    # data type: ArtifactVersionManifest
-    anyof_schema_4_validator: Optional[ArtifactVersionManifest] = None
+    # data type: ArtifactManifest
+    anyof_schema_4_validator: Optional[ArtifactManifest] = None
     if TYPE_CHECKING:
         actual_instance: Union[
             AgentManifest,
             AgentOpenAPIToolManifest,
-            ArtifactVersionManifest,
+            ArtifactManifest,
             ChatPromptManifest,
         ]
     else:
@@ -110,10 +110,10 @@ class Manifest2(BaseModel):
         else:
             return v
 
-        # validate data type: ArtifactVersionManifest
-        if not isinstance(v, ArtifactVersionManifest):
+        # validate data type: ArtifactManifest
+        if not isinstance(v, ArtifactManifest):
             error_messages.append(
-                f"Error! Input type `{type(v)}` is not `ArtifactVersionManifest`"
+                f"Error! Input type `{type(v)}` is not `ArtifactManifest`"
             )
         else:
             return v
@@ -121,7 +121,7 @@ class Manifest2(BaseModel):
         if error_messages:
             # no match
             raise ValueError(
-                "No match found when setting the actual_instance in Manifest2 with anyOf schemas: AgentManifest, AgentOpenAPIToolManifest, ArtifactVersionManifest, ChatPromptManifest. Details: "
+                "No match found when setting the actual_instance in Manifest2 with anyOf schemas: AgentManifest, AgentOpenAPIToolManifest, ArtifactManifest, ChatPromptManifest. Details: "
                 + ", ".join(error_messages)
             )
         else:
@@ -154,9 +154,9 @@ class Manifest2(BaseModel):
             return instance
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
-        # anyof_schema_4_validator: Optional[ArtifactVersionManifest] = None
+        # anyof_schema_4_validator: Optional[ArtifactManifest] = None
         try:
-            instance.actual_instance = ArtifactVersionManifest.from_json(json_str)
+            instance.actual_instance = ArtifactManifest.from_json(json_str)
             return instance
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
@@ -164,7 +164,7 @@ class Manifest2(BaseModel):
         if error_messages:
             # no match
             raise ValueError(
-                "No match found when deserializing the JSON string into Manifest2 with anyOf schemas: AgentManifest, AgentOpenAPIToolManifest, ArtifactVersionManifest, ChatPromptManifest. Details: "
+                "No match found when deserializing the JSON string into Manifest2 with anyOf schemas: AgentManifest, AgentOpenAPIToolManifest, ArtifactManifest, ChatPromptManifest. Details: "
                 + ", ".join(error_messages)
             )
         else:

truefoundry/ml/{autogen → _autogen}/client/models/method.py

@@ -19,7 +19,7 @@ from aenum import Enum
 
 class Method(str, Enum):
     """
-    +sort=50 +uiType=Hidden +label=API HTTP Method +usage=HTTP Method for the tool call HTTP request
+    HTTP Method for the tool call HTTP request
     """
 
     """

truefoundry/ml/{autogen → _autogen}/client/models/metric_collection_dto.py

@@ -18,7 +18,7 @@ import pprint
 import re  # noqa: F401
 from typing import Optional
 
-from truefoundry.ml.autogen.client.models.metric_dto import MetricDto
+from truefoundry.ml._autogen.client.models.metric_dto import MetricDto
 from truefoundry.pydantic_v1 import BaseModel, StrictStr, conlist
 
 

truefoundry/ml/{autogen → _autogen}/client/models/mime_type.py

@@ -19,7 +19,7 @@ from aenum import Enum
 
 class MimeType(str, Enum):
     """
-    +label=MIME Type +usage=MIME type of the content
+    MIME type of the content
     """
 
     """

truefoundry/ml/{autogen → _autogen}/client/models/model_configuration.py

@@ -18,30 +18,29 @@ import pprint
 import re  # noqa: F401
 from typing import Any, Dict, Optional
 
-from truefoundry.ml.autogen.client.models.parameters import Parameters
+from truefoundry.ml._autogen.client.models.parameters import Parameters
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, constr, validator
 
 
 class ModelConfiguration(BaseModel):
     """
-    +label=Model Configuration +icon=fa-cogs:#326ce5 +usage=Add a model from Gateway along with parameters to be used for chat completion  # noqa: E501
+    Add a model from Gateway along with parameters to be used for chat completion  # noqa: E501
     """
 
     provider: StrictStr = Field(
         default=...,
-        description='+sort=1000 +uiType=SelectModelAndProvider +uiProps={"isModel": false,"providerJsonKey": "model_configuration.provider","searchable": true} +label=Provider Name +usage=Name of the provider, must be one of the integration providers configured for the Gateway',
+        description="Name of the provider, must be one of the integration providers configured for the Gateway",
     )
     model: constr(strict=True) = Field(
         default=...,
-        description='+sort=1001 +uiType=SelectModelAndProvider +uiProps={"isModel": true,"providerJsonKey": "model_configuration.provider","searchable": true} +label=Model Name +usage=Name of the model to be used for generations. This model should be available in the provider',
+        description="Name of the model to be used for generations. This model should be available in the provider",
     )
     parameters: Optional[Parameters] = Field(
-        default=None,
-        description="+label=Parameters +usage=Parameters to pass to the model when generating",
+        default=None, description="Parameters to pass to the model when generating"
     )
     extra_parameters: Optional[Dict[str, Any]] = Field(
         default=None,
-        description="+label=Extra Parameters +usage=Arbitrary extra parameters to pass to the model when generating",
+        description="Arbitrary extra parameters to pass to the model when generating",
     )
     __properties = ["provider", "model", "parameters", "extra_parameters"]
 

truefoundry/ml/{autogen → _autogen}/client/models/model_dto.py

@@ -19,10 +19,10 @@ import re  # noqa: F401
 from datetime import datetime
 from typing import Optional
 
-from truefoundry.ml.autogen.client.models.artifact_type import ArtifactType
-from truefoundry.ml.autogen.client.models.experiment_dto import ExperimentDto
-from truefoundry.ml.autogen.client.models.model_version_dto import ModelVersionDto
-from truefoundry.ml.autogen.client.models.subject import Subject
+from truefoundry.ml._autogen.client.models.artifact_type import ArtifactType
+from truefoundry.ml._autogen.client.models.experiment_dto import ExperimentDto
+from truefoundry.ml._autogen.client.models.model_version_dto import ModelVersionDto
+from truefoundry.ml._autogen.client.models.subject import Subject
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictBool, StrictStr
 
 

truefoundry/ml/{autogen/client/models/model_version_manifest.py → _autogen/client/models/model_manifest.py}

@@ -18,11 +18,11 @@ import pprint
 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 (
+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.source2 import Source2
+from truefoundry.ml._autogen.client.models.source2 import Source2
 from truefoundry.pydantic_v1 import (
     BaseModel,
     Field,
@@ -33,33 +33,28 @@ from truefoundry.pydantic_v1 import (
 )
 
 
-class ModelVersionManifest(BaseModel):
+class ModelManifest(BaseModel):
     """
     Model Version artifact.  # noqa: E501
     """
 
     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 +usage=Name of the entity",
-    )
-    description: Optional[constr(strict=True, max_length=512)] = Field(
-        default=None, description="+sort=2 +label=Description"
+        default=None, description="Name of the entity"
     )
+    description: Optional[constr(strict=True, max_length=512)] = None
     metadata: Dict[str, Any] = Field(
         default=...,
-        description='+label=Metadata +usage=Key value metadata. Should be valid JSON. For e.g. `{"business-unit": "sales", "quality": "good", "rating": 4.5}` +uiType=JsonInput +uiProps={"descriptionInline":true}',
+        description='Key value metadata. Should be valid JSON. For e.g. `{"business-unit": "sales", "quality": "good", "rating": 4.5}`',
     )
     version_alias: Optional[constr(strict=True, max_length=128)] = Field(
         default=None,
-        description='+label=Version Alias +usage=Version alias is alternate, ideally human readable, version string to reference an artifact version. It 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 can include `.` and `-` in between (e.g. `v1.0.0`, `v1-prod`, `v3-dev`, etc) +uiProps={"descriptionInline":true} +placeholder=v1.0.0',
+        description="Version alias is alternate, ideally human readable, version string to reference an artifact version. It should start with `v` followed by alphanumeric and it can include `.` and `-` in between (e.g. `v1.0.0`, `v1-prod`, `v3-dev`, etc)",
     )
     ml_repo: Optional[constr(strict=True)] = Field(
-        default=None,
-        description="+label=ML Repo +usage=Name of the ML Repo +uiType=Hidden",
+        default=None, description="Name of the ML Repo"
     )
     version: Optional[conint(strict=True, ge=1)] = Field(
-        default=None,
-        description="+label=Version of the entity +usage=Version of the entity +uiType=Hidden",
+        default=None, description="Version of the entity"
     )
     type: Optional[StrictStr] = "model-version"
     source: Source2 = Field(...)
@@ -67,11 +62,11 @@ class ModelVersionManifest(BaseModel):
     environment: Optional[ModelVersionEnvironment] = None
     step: Optional[conint(strict=True, ge=0)] = Field(
         default=0,
-        description='+label=Step +usage=Step/Epoch number in an iterative training loop the model version was created. Generally useful when logging a model version from a MLRepo Run +uiProps={"descriptionInline":true}',
+        description="Step/Epoch number in an iterative training loop the model version was created. Generally useful when logging a model version from a MLRepo Run",
     )
     run_id: Optional[StrictStr] = Field(
         default=None,
-        description="+label=Run ID +usage=ID of the MLRepo Run that generated the model version +uiType=Hidden",
+        description="ID of the MLRepo Run that generated the model version",
     )
     __properties = [
         "name",
@@ -149,8 +144,8 @@ class ModelVersionManifest(BaseModel):
         return json.dumps(self.to_dict())
 
     @classmethod
-    def from_json(cls, json_str: str) -> ModelVersionManifest:
-        """Create an instance of ModelVersionManifest from a JSON string"""
+    def from_json(cls, json_str: str) -> ModelManifest:
+        """Create an instance of ModelManifest from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self):
@@ -168,15 +163,15 @@ class ModelVersionManifest(BaseModel):
         return _dict
 
     @classmethod
-    def from_dict(cls, obj: dict) -> ModelVersionManifest:
-        """Create an instance of ModelVersionManifest from a dict"""
+    def from_dict(cls, obj: dict) -> ModelManifest:
+        """Create an instance of ModelManifest from a dict"""
         if obj is None:
             return None
 
         if not isinstance(obj, dict):
-            return ModelVersionManifest.parse_obj(obj)
+            return ModelManifest.parse_obj(obj)
 
-        _obj = ModelVersionManifest.parse_obj(
+        _obj = ModelManifest.parse_obj(
             {
                 "name": obj.get("name"),
                 "description": obj.get("description"),

truefoundry/ml/{autogen → _autogen}/client/models/model_response_dto.py

@@ -17,7 +17,7 @@ import json
 import pprint
 import re  # noqa: F401
 
-from truefoundry.ml.autogen.client.models.model_dto import ModelDto
+from truefoundry.ml._autogen.client.models.model_dto import ModelDto
 from truefoundry.pydantic_v1 import BaseModel, Field
 
 

truefoundry/ml/{autogen → _autogen}/client/models/model_version_dto.py

@@ -19,17 +19,15 @@ import re  # noqa: F401
 from datetime import datetime
 from typing import Any, Dict, Optional
 
-from truefoundry.ml.autogen.client.models.artifact_version_serialization_format import (
+from truefoundry.ml._autogen.client.models.artifact_version_serialization_format import (
     ArtifactVersionSerializationFormat,
 )
-from truefoundry.ml.autogen.client.models.artifact_version_status import (
+from truefoundry.ml._autogen.client.models.artifact_version_status import (
     ArtifactVersionStatus,
 )
-from truefoundry.ml.autogen.client.models.metric_dto import MetricDto
-from truefoundry.ml.autogen.client.models.model_version_manifest import (
-    ModelVersionManifest,
-)
-from truefoundry.ml.autogen.client.models.subject import Subject
+from truefoundry.ml._autogen.client.models.metric_dto import MetricDto
+from truefoundry.ml._autogen.client.models.model_manifest import ModelManifest
+from truefoundry.ml._autogen.client.models.subject import Subject
 from truefoundry.pydantic_v1 import (
     BaseModel,
     Field,
@@ -70,7 +68,7 @@ class ModelVersionDto(BaseModel):
     model_name: Optional[StrictStr] = None
     model_framework: Optional[StrictStr] = None
     artifact_size: Optional[StrictInt] = None
-    manifest: Optional[ModelVersionManifest] = None
+    manifest: Optional[ModelManifest] = None
     serialization_format: ArtifactVersionSerializationFormat = Field(...)
     deployable: Optional[StrictBool] = False
     __properties = [
@@ -183,7 +181,7 @@ class ModelVersionDto(BaseModel):
                 "model_name": obj.get("model_name"),
                 "model_framework": obj.get("model_framework"),
                 "artifact_size": obj.get("artifact_size"),
-                "manifest": ModelVersionManifest.from_dict(obj.get("manifest"))
+                "manifest": ModelManifest.from_dict(obj.get("manifest"))
                 if obj.get("manifest") is not None
                 else None,
                 "serialization_format": obj.get("serialization_format"),

truefoundry/ml/{autogen → _autogen}/client/models/model_version_environment.py

@@ -30,16 +30,14 @@ from truefoundry.pydantic_v1 import (
 
 class ModelVersionEnvironment(BaseModel):
     """
-    +label=Environment +usage=Runtime environment describing python version and dependencies for running the model +uiProps={\"descriptionInline\":true}  # noqa: E501
+    Runtime environment describing python version and dependencies for running the model  # noqa: E501
     """
 
     python_version: Optional[constr(strict=True)] = Field(
-        default=None,
-        description="+label=Python Version +usage=Python version to use when running the model",
+        default=None, description="Python version to use when running the model"
     )
     pip_packages: Optional[conlist(StrictStr)] = Field(
-        default=None,
-        description="+label=PIP Packages +usage=pip dependencies needed for running the model",
+        default=None, description="pip dependencies needed for running the model"
     )
     __properties = ["python_version", "pip_packages"]
 

truefoundry/ml/{autogen → _autogen}/client/models/model_version_response_dto.py

@@ -17,7 +17,7 @@ import json
 import pprint
 import re  # noqa: F401
 
-from truefoundry.ml.autogen.client.models.model_version_dto import ModelVersionDto
+from truefoundry.ml._autogen.client.models.model_version_dto import ModelVersionDto
 from truefoundry.pydantic_v1 import BaseModel, Field
 
 

truefoundry/ml/{autogen → _autogen}/client/models/multi_part_upload_dto.py

@@ -18,10 +18,10 @@ import pprint
 import re  # noqa: F401
 from typing import Optional
 
-from truefoundry.ml.autogen.client.models.multi_part_upload_storage_provider import (
+from truefoundry.ml._autogen.client.models.multi_part_upload_storage_provider import (
     MultiPartUploadStorageProvider,
 )
-from truefoundry.ml.autogen.client.models.signed_url_dto import SignedURLDto
+from truefoundry.ml._autogen.client.models.signed_url_dto import SignedURLDto
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, conlist