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

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.
@@ -16,11 +16,16 @@ from __future__ import annotations
 import json
 import pprint
 import re  # noqa: F401
-from typing import TYPE_CHECKING, Any, Dict, List, Optional, Union
+from typing import TYPE_CHECKING, Any, List, Optional, Union
 
-from truefoundry.ml.autogen.client.models.agent import Agent
-from truefoundry.ml.autogen.client.models.agent_open_api_tool import AgentOpenAPITool
-from truefoundry.ml.autogen.client.models.chat_prompt import ChatPrompt
+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_manifest import ArtifactManifest
+from truefoundry.ml._autogen.client.models.chat_prompt_manifest import (
+    ChatPromptManifest,
+)
 from truefoundry.pydantic_v1 import (
     BaseModel,
     Field,
@@ -28,27 +33,37 @@ from truefoundry.pydantic_v1 import (
     validator,
 )
 
-INTERNALMETADATA_ANY_OF_SCHEMAS = ["Agent", "AgentOpenAPITool", "ChatPrompt", "object"]
+MANIFEST2_ANY_OF_SCHEMAS = [
+    "AgentManifest",
+    "AgentOpenAPIToolManifest",
+    "ArtifactManifest",
+    "ChatPromptManifest",
+]
 
 
-class InternalMetadata(BaseModel):
+class Manifest2(BaseModel):
     """
-    InternalMetadata
+    Manifest2
     """
 
-    # data type: ChatPrompt
-    anyof_schema_1_validator: Optional[ChatPrompt] = None
-    # data type: AgentOpenAPITool
-    anyof_schema_2_validator: Optional[AgentOpenAPITool] = None
-    # data type: Agent
-    anyof_schema_3_validator: Optional[Agent] = None
-    # data type: object
-    anyof_schema_4_validator: Optional[Dict[str, Any]] = None
+    # data type: ChatPromptManifest
+    anyof_schema_1_validator: Optional[ChatPromptManifest] = None
+    # data type: AgentOpenAPIToolManifest
+    anyof_schema_2_validator: Optional[AgentOpenAPIToolManifest] = None
+    # data type: AgentManifest
+    anyof_schema_3_validator: Optional[AgentManifest] = None
+    # data type: ArtifactManifest
+    anyof_schema_4_validator: Optional[ArtifactManifest] = None
     if TYPE_CHECKING:
-        actual_instance: Union[Agent, AgentOpenAPITool, ChatPrompt, object]
+        actual_instance: Union[
+            AgentManifest,
+            AgentOpenAPIToolManifest,
+            ArtifactManifest,
+            ChatPromptManifest,
+        ]
     else:
         actual_instance: Any
-    any_of_schemas: List[str] = Field(INTERNALMETADATA_ANY_OF_SCHEMAS, const=True)
+    any_of_schemas: List[str] = Field(MANIFEST2_ANY_OF_SCHEMAS, const=True)
 
     class Config:
         validate_assignment = True
@@ -69,76 +84,79 @@ class InternalMetadata(BaseModel):
 
     @validator("actual_instance")
     def actual_instance_must_validate_anyof(cls, v):
-        instance = InternalMetadata.construct()
+        instance = Manifest2.construct()
         error_messages = []
-        # validate data type: ChatPrompt
-        if not isinstance(v, ChatPrompt):
-            error_messages.append(f"Error! Input type `{type(v)}` is not `ChatPrompt`")
+        # validate data type: ChatPromptManifest
+        if not isinstance(v, ChatPromptManifest):
+            error_messages.append(
+                f"Error! Input type `{type(v)}` is not `ChatPromptManifest`"
+            )
         else:
             return v
 
-        # validate data type: AgentOpenAPITool
-        if not isinstance(v, AgentOpenAPITool):
+        # validate data type: AgentOpenAPIToolManifest
+        if not isinstance(v, AgentOpenAPIToolManifest):
             error_messages.append(
-                f"Error! Input type `{type(v)}` is not `AgentOpenAPITool`"
+                f"Error! Input type `{type(v)}` is not `AgentOpenAPIToolManifest`"
             )
         else:
             return v
 
-        # validate data type: Agent
-        if not isinstance(v, Agent):
-            error_messages.append(f"Error! Input type `{type(v)}` is not `Agent`")
+        # validate data type: AgentManifest
+        if not isinstance(v, AgentManifest):
+            error_messages.append(
+                f"Error! Input type `{type(v)}` is not `AgentManifest`"
+            )
         else:
             return v
 
-        # validate data type: object
-        try:
-            instance.anyof_schema_4_validator = v
+        # validate data type: ArtifactManifest
+        if not isinstance(v, ArtifactManifest):
+            error_messages.append(
+                f"Error! Input type `{type(v)}` is not `ArtifactManifest`"
+            )
+        else:
             return v
-        except (ValidationError, ValueError) as e:
-            error_messages.append(str(e))
+
         if error_messages:
             # no match
             raise ValueError(
-                "No match found when setting the actual_instance in InternalMetadata with anyOf schemas: Agent, AgentOpenAPITool, ChatPrompt, object. Details: "
+                "No match found when setting the actual_instance in Manifest2 with anyOf schemas: AgentManifest, AgentOpenAPIToolManifest, ArtifactManifest, ChatPromptManifest. Details: "
                 + ", ".join(error_messages)
             )
         else:
             return v
 
     @classmethod
-    def from_dict(cls, obj: dict) -> InternalMetadata:
+    def from_dict(cls, obj: dict) -> Manifest2:
         return cls.from_json(json.dumps(obj))
 
     @classmethod
-    def from_json(cls, json_str: str) -> InternalMetadata:
+    def from_json(cls, json_str: str) -> Manifest2:
         """Returns the object represented by the json string"""
-        instance = InternalMetadata.construct()
+        instance = Manifest2.construct()
         error_messages = []
-        # anyof_schema_1_validator: Optional[ChatPrompt] = None
+        # anyof_schema_1_validator: Optional[ChatPromptManifest] = None
         try:
-            instance.actual_instance = ChatPrompt.from_json(json_str)
+            instance.actual_instance = ChatPromptManifest.from_json(json_str)
             return instance
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
-        # anyof_schema_2_validator: Optional[AgentOpenAPITool] = None
+        # anyof_schema_2_validator: Optional[AgentOpenAPIToolManifest] = None
         try:
-            instance.actual_instance = AgentOpenAPITool.from_json(json_str)
+            instance.actual_instance = AgentOpenAPIToolManifest.from_json(json_str)
             return instance
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
-        # anyof_schema_3_validator: Optional[Agent] = None
+        # anyof_schema_3_validator: Optional[AgentManifest] = None
         try:
-            instance.actual_instance = Agent.from_json(json_str)
+            instance.actual_instance = AgentManifest.from_json(json_str)
             return instance
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
-        # deserialize data into object
+        # anyof_schema_4_validator: Optional[ArtifactManifest] = None
         try:
-            # validation
-            instance.anyof_schema_4_validator = json.loads(json_str)
-            # assign value to actual_instance
-            instance.actual_instance = instance.anyof_schema_4_validator
+            instance.actual_instance = ArtifactManifest.from_json(json_str)
             return instance
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
@@ -146,7 +164,7 @@ class InternalMetadata(BaseModel):
         if error_messages:
             # no match
             raise ValueError(
-                "No match found when deserializing the JSON string into InternalMetadata with anyOf schemas: Agent, AgentOpenAPITool, ChatPrompt, object. 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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.
@@ -19,7 +19,7 @@ from aenum import Enum
 
 class Method(str, Enum):
     """
-    +sort=50 +uiType=Hidden
+    HTTP Method for the tool call HTTP request
     """
 
     """

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.
@@ -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/metric_dto.py

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.
@@ -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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.
@@ -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 +message=Configuration details for the provider and corresponding config +usage=Define the provider and its configuration +docs=Configuration settings specific to the provider, including model settings and other parameters.  # noqa: E501
+    Add a model from Gateway along with parameters to be used for chat completion  # noqa: E501
     """
 
     provider: StrictStr = Field(
         default=...,
-        description='+label=Provider Name +usage=Name of the provider, must be non-empty, e.g., "openai", "google_gemini"',
+        description="Name of the provider, must be one of the integration providers configured for the Gateway",
     )
     model: constr(strict=True) = Field(
         default=...,
-        description="+label=Model name +usage=Model name, must be non-empty",
+        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="+usage=Parameters for the provider +label=Parameters +docs=Key-value pairs to store additional parameters for the provider",
+        default=None, description="Parameters to pass to the model when generating"
     )
     extra_parameters: Optional[Dict[str, Any]] = Field(
         default=None,
-        description="+usage=Extra parameters for the provider +label=Extra Parameters +docs=Additional parameters for the provider",
+        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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.
@@ -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}

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.
@@ -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.source1 import Source1
+from truefoundry.ml._autogen.client.models.source2 import Source2
 from truefoundry.pydantic_v1 import (
     BaseModel,
     Field,
@@ -33,71 +33,77 @@ from truefoundry.pydantic_v1 import (
 )
 
 
-class ModelVersionManifest(BaseModel):
+class ModelManifest(BaseModel):
     """
     Model Version artifact.  # noqa: E501
     """
 
-    description: Optional[constr(strict=True, max_length=512)] = Field(
-        default=None, description="+sort=2 +label=Description"
+    name: Optional[constr(strict=True, max_length=256)] = Field(
+        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 +docs=Key value pairs to store additional metadata +usage=Key value pairs to store additional metadata +uiType=JsonInput",
+        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=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="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 +docs=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="Version of the entity"
     )
     type: Optional[StrictStr] = "model-version"
-    source: Source1 = Field(...)
+    source: Source2 = Field(...)
     framework: Optional[Framework] = None
     environment: Optional[ModelVersionEnvironment] = None
     step: Optional[conint(strict=True, ge=0)] = Field(
-        default=0, description="+label=Step"
+        default=0,
+        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="ID of the MLRepo Run that generated the model version",
     )
     __properties = [
+        "name",
         "description",
         "metadata",
         "version_alias",
-        "name",
         "ml_repo",
+        "version",
         "type",
         "source",
         "framework",
         "environment",
         "step",
+        "run_id",
     ]
 
-    @validator("version_alias")
-    def version_alias_validate_regular_expression(cls, 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"^v[a-zA-Z0-9.-]*([a-zA-Z0-9]+)$", value):
+        if not re.match(r"^[A-Za-z0-9_\-]+$", value):
             raise ValueError(
-                r"must validate the regular expression /^v[a-zA-Z0-9.-]*([a-zA-Z0-9]+)$/"
+                r"must validate the regular expression /^[A-Za-z0-9_\-]+$/"
             )
         return value
 
-    @validator("name")
-    def name_validate_regular_expression(cls, value):
+    @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"^[A-Za-z0-9_\-]+$", value):
+        if not re.match(r"^v[a-zA-Z0-9.-]*([a-zA-Z0-9]+)$", value):
             raise ValueError(
-                r"must validate the regular expression /^[A-Za-z0-9_\-]+$/"
+                r"must validate the regular expression /^v[a-zA-Z0-9.-]*([a-zA-Z0-9]+)$/"
             )
         return value
 
@@ -138,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):
@@ -157,25 +163,26 @@ 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"),
                 "metadata": obj.get("metadata"),
                 "version_alias": obj.get("version_alias"),
-                "name": obj.get("name"),
                 "ml_repo": obj.get("ml_repo"),
+                "version": obj.get("version"),
                 "type": obj.get("type")
                 if obj.get("type") is not None
                 else "model-version",
-                "source": Source1.from_dict(obj.get("source"))
+                "source": Source2.from_dict(obj.get("source"))
                 if obj.get("source") is not None
                 else None,
                 "framework": Framework.from_dict(obj.get("framework"))
@@ -185,6 +192,7 @@ class ModelVersionManifest(BaseModel):
                 if obj.get("environment") is not None
                 else None,
                 "step": obj.get("step") if obj.get("step") is not None else 0,
+                "run_id": obj.get("run_id"),
             }
         )
         return _obj

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.
@@ -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_server.py

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.
@@ -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"),