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

truefoundry/ml/{autogen/client/models/chat_prompt.py → _autogen/client/models/chat_prompt_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,87 +18,88 @@ import pprint
 import re  # noqa: F401
 from typing import Any, Dict, Optional
 
-from truefoundry.ml.autogen.client.models.chat_prompt_messages_inner import (
-    ChatPromptMessagesInner,
+from truefoundry.ml._autogen.client.models.chat_prompt_manifest_messages_inner import (
+    ChatPromptManifestMessagesInner,
 )
-from truefoundry.ml.autogen.client.models.model_configuration import ModelConfiguration
+from truefoundry.ml._autogen.client.models.model_configuration import ModelConfiguration
 from truefoundry.pydantic_v1 import (
     BaseModel,
     Field,
     StrictStr,
+    conint,
     conlist,
     constr,
     validator,
 )
 
 
-class ChatPrompt(BaseModel):
+class ChatPromptManifest(BaseModel):
     """
-    Chat Prompt artifact.  # noqa: E501
+    Chat Prompt manifest.  # 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] = "chat_prompt"
-    messages: conlist(ChatPromptMessagesInner) = Field(
+    messages: conlist(ChatPromptManifestMessagesInner) = Field(
         default=...,
-        description="+sort=4000 +usage=Chat completion messages +label=Messages in the chat conversation +message=Chat completion messages +usage=List of messages in the chat conversation, must be non-empty +docs=Messages that define the chat conversation, including system, assistant, and user messages.",
+        description="List of messages in the chat conversation, must be non-empty",
     )
     variables: Optional[Dict[str, StrictStr]] = Field(
         default=None,
-        description="+label=Variables +usage=Variables for the chat completion messages to be used in the prompt messages +sort=5000",
+        description="Variables referenced in messages and that can be replaced when running generation",
     )
     model_configuration: ModelConfiguration = Field(...)
     __properties = [
+        "name",
         "description",
         "metadata",
         "version_alias",
-        "name",
         "ml_repo",
+        "version",
         "type",
         "messages",
         "variables",
         "model_configuration",
     ]
 
-    @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
 
@@ -139,8 +140,8 @@ class ChatPrompt(BaseModel):
         return json.dumps(self.to_dict())
 
     @classmethod
-    def from_json(cls, json_str: str) -> ChatPrompt:
-        """Create an instance of ChatPrompt from a JSON string"""
+    def from_json(cls, json_str: str) -> ChatPromptManifest:
+        """Create an instance of ChatPromptManifest from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self):
@@ -159,26 +160,27 @@ class ChatPrompt(BaseModel):
         return _dict
 
     @classmethod
-    def from_dict(cls, obj: dict) -> ChatPrompt:
-        """Create an instance of ChatPrompt from a dict"""
+    def from_dict(cls, obj: dict) -> ChatPromptManifest:
+        """Create an instance of ChatPromptManifest from a dict"""
         if obj is None:
             return None
 
         if not isinstance(obj, dict):
-            return ChatPrompt.parse_obj(obj)
+            return ChatPromptManifest.parse_obj(obj)
 
-        _obj = ChatPrompt.parse_obj(
+        _obj = ChatPromptManifest.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 "chat_prompt",
                 "messages": [
-                    ChatPromptMessagesInner.from_dict(_item)
+                    ChatPromptManifestMessagesInner.from_dict(_item)
                     for _item in obj.get("messages")
                 ]
                 if obj.get("messages") is not None

truefoundry/ml/{autogen/client/models/chat_prompt_messages_inner.py → _autogen/client/models/chat_prompt_manifest_messages_inner.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,9 +18,9 @@ import pprint
 import re  # noqa: F401
 from typing import TYPE_CHECKING, Any, List, Optional, Union
 
-from truefoundry.ml.autogen.client.models.assistant_message import AssistantMessage
-from truefoundry.ml.autogen.client.models.system_message import SystemMessage
-from truefoundry.ml.autogen.client.models.user_message import UserMessage
+from truefoundry.ml._autogen.client.models.assistant_message import AssistantMessage
+from truefoundry.ml._autogen.client.models.system_message import SystemMessage
+from truefoundry.ml._autogen.client.models.user_message import UserMessage
 from truefoundry.pydantic_v1 import (
     BaseModel,
     Field,
@@ -28,16 +28,16 @@ from truefoundry.pydantic_v1 import (
     validator,
 )
 
-CHATPROMPTMESSAGESINNER_ANY_OF_SCHEMAS = [
+CHATPROMPTMANIFESTMESSAGESINNER_ANY_OF_SCHEMAS = [
     "AssistantMessage",
     "SystemMessage",
     "UserMessage",
 ]
 
 
-class ChatPromptMessagesInner(BaseModel):
+class ChatPromptManifestMessagesInner(BaseModel):
     """
-    ChatPromptMessagesInner
+    ChatPromptManifestMessagesInner
     """
 
     # data type: SystemMessage
@@ -51,7 +51,7 @@ class ChatPromptMessagesInner(BaseModel):
     else:
         actual_instance: Any
     any_of_schemas: List[str] = Field(
-        CHATPROMPTMESSAGESINNER_ANY_OF_SCHEMAS, const=True
+        CHATPROMPTMANIFESTMESSAGESINNER_ANY_OF_SCHEMAS, const=True
     )
 
     class Config:
@@ -73,7 +73,7 @@ class ChatPromptMessagesInner(BaseModel):
 
     @validator("actual_instance")
     def actual_instance_must_validate_anyof(cls, v):
-        instance = ChatPromptMessagesInner.construct()
+        instance = ChatPromptManifestMessagesInner.construct()
         error_messages = []
         # validate data type: SystemMessage
         if not isinstance(v, SystemMessage):
@@ -100,20 +100,20 @@ class ChatPromptMessagesInner(BaseModel):
         if error_messages:
             # no match
             raise ValueError(
-                "No match found when setting the actual_instance in ChatPromptMessagesInner with anyOf schemas: AssistantMessage, SystemMessage, UserMessage. Details: "
+                "No match found when setting the actual_instance in ChatPromptManifestMessagesInner with anyOf schemas: AssistantMessage, SystemMessage, UserMessage. Details: "
                 + ", ".join(error_messages)
             )
         else:
             return v
 
     @classmethod
-    def from_dict(cls, obj: dict) -> ChatPromptMessagesInner:
+    def from_dict(cls, obj: dict) -> ChatPromptManifestMessagesInner:
         return cls.from_json(json.dumps(obj))
 
     @classmethod
-    def from_json(cls, json_str: str) -> ChatPromptMessagesInner:
+    def from_json(cls, json_str: str) -> ChatPromptManifestMessagesInner:
         """Returns the object represented by the json string"""
-        instance = ChatPromptMessagesInner.construct()
+        instance = ChatPromptManifestMessagesInner.construct()
         error_messages = []
         # anyof_schema_1_validator: Optional[SystemMessage] = None
         try:
@@ -137,7 +137,7 @@ class ChatPromptMessagesInner(BaseModel):
         if error_messages:
             # no match
             raise ValueError(
-                "No match found when deserializing the JSON string into ChatPromptMessagesInner with anyOf schemas: AssistantMessage, SystemMessage, UserMessage. Details: "
+                "No match found when deserializing the JSON string into ChatPromptManifestMessagesInner with anyOf schemas: AssistantMessage, SystemMessage, UserMessage. Details: "
                 + ", ".join(error_messages)
             )
         else:

truefoundry/ml/{autogen → _autogen}/client/models/columns_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/command.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/content.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 TYPE_CHECKING, Any, List, Optional, Union
 
-from truefoundry.ml.autogen.client.models.blob_storage_reference import (
+from truefoundry.ml._autogen.client.models.blob_storage_reference import (
     BlobStorageReference,
 )
 from truefoundry.pydantic_v1 import (
@@ -34,7 +34,7 @@ CONTENT_ANY_OF_SCHEMAS = ["BlobStorageReference", "str"]
 
 class Content(BaseModel):
     """
-    +label=Content +usage=Text content for the assistant message
+    Text content for the assistant message
     """
 
     # data type: str

truefoundry/ml/{autogen → _autogen}/client/models/content1.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 TYPE_CHECKING, Any, List, Optional, Union
 
-from truefoundry.ml.autogen.client.models.blob_storage_reference import (
+from truefoundry.ml._autogen.client.models.blob_storage_reference import (
     BlobStorageReference,
 )
 from truefoundry.pydantic_v1 import (
@@ -34,7 +34,7 @@ CONTENT1_ANY_OF_SCHEMAS = ["BlobStorageReference", "str"]
 
 class Content1(BaseModel):
     """
-    +label=Content +usage=Text content for the system message
+    Text content for the system message
     """
 
     # data type: str

truefoundry/ml/{autogen → _autogen}/client/models/content2.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,10 +18,10 @@ import pprint
 import re  # noqa: F401
 from typing import TYPE_CHECKING, Any, List, Optional, Union
 
-from truefoundry.ml.autogen.client.models.blob_storage_reference import (
+from truefoundry.ml._autogen.client.models.blob_storage_reference import (
     BlobStorageReference,
 )
-from truefoundry.ml.autogen.client.models.content2_any_of_inner import (
+from truefoundry.ml._autogen.client.models.content2_any_of_inner import (
     Content2AnyOfInner,
 )
 from truefoundry.pydantic_v1 import (
@@ -38,7 +38,7 @@ CONTENT2_ANY_OF_SCHEMAS = ["BlobStorageReference", "List[Content2AnyOfInner]", "
 
 class Content2(BaseModel):
     """
-    +label=Content +usage=Text content for the user message
+    Content of the user message. can be a mix of text and images
     """
 
     # data type: str

truefoundry/ml/{autogen → _autogen}/client/models/content2_any_of_inner.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,8 +18,8 @@ import pprint
 import re  # noqa: F401
 from typing import TYPE_CHECKING, Any, List, Optional, Union
 
-from truefoundry.ml.autogen.client.models.image_content_part import ImageContentPart
-from truefoundry.ml.autogen.client.models.text_content_part import TextContentPart
+from truefoundry.ml._autogen.client.models.image_content_part import ImageContentPart
+from truefoundry.ml._autogen.client.models.text_content_part import TextContentPart
 from truefoundry.pydantic_v1 import (
     BaseModel,
     Field,

truefoundry/ml/{autogen → _autogen}/client/models/create_artifact_request_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.artifact_type import ArtifactType
+from truefoundry.ml._autogen.client.models.artifact_type import ArtifactType
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr
 
 

truefoundry/ml/{autogen → _autogen}/client/models/create_artifact_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.

truefoundry/ml/{autogen → _autogen}/client/models/create_artifact_version_request_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.artifact_type import ArtifactType
+from truefoundry.ml._autogen.client.models.artifact_type import ArtifactType
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictInt, StrictStr
 
 

truefoundry/ml/{autogen → _autogen}/client/models/create_artifact_version_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.

truefoundry/ml/{autogen → _autogen}/client/models/create_dataset_request_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,10 @@ import pprint
 import re  # noqa: F401
 from typing import Any, Dict, Optional
 
-from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr
+from truefoundry.ml._autogen.client.models.data_directory_manifest import (
+    DataDirectoryManifest,
+)
+from truefoundry.pydantic_v1 import BaseModel, StrictStr
 
 
 class CreateDatasetRequestDto(BaseModel):
@@ -26,11 +29,18 @@ class CreateDatasetRequestDto(BaseModel):
     CreateDatasetRequestDto
     """
 
-    name: StrictStr = Field(...)
+    name: Optional[StrictStr] = None
     description: Optional[StrictStr] = None
     experiment_id: Optional[StrictStr] = None
     dataset_metadata: Optional[Dict[str, Any]] = None
-    __properties = ["name", "description", "experiment_id", "dataset_metadata"]
+    manifest: Optional[DataDirectoryManifest] = None
+    __properties = [
+        "name",
+        "description",
+        "experiment_id",
+        "dataset_metadata",
+        "manifest",
+    ]
 
     class Config:
         """Pydantic configuration"""
@@ -54,6 +64,9 @@ class CreateDatasetRequestDto(BaseModel):
     def to_dict(self):
         """Returns the dictionary representation of the model using alias"""
         _dict = self.dict(by_alias=True, exclude={}, exclude_none=True)
+        # override the default output from truefoundry.pydantic_v1 by calling `to_dict()` of manifest
+        if self.manifest:
+            _dict["manifest"] = self.manifest.to_dict()
         return _dict
 
     @classmethod
@@ -71,6 +84,9 @@ class CreateDatasetRequestDto(BaseModel):
                 "description": obj.get("description"),
                 "experiment_id": obj.get("experiment_id"),
                 "dataset_metadata": obj.get("dataset_metadata"),
+                "manifest": DataDirectoryManifest.from_dict(obj.get("manifest"))
+                if obj.get("manifest") is not None
+                else None,
             }
         )
         return _obj

truefoundry/ml/{autogen → _autogen}/client/models/create_experiment_request_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.experiment_tag_dto import ExperimentTagDto
+from truefoundry.ml._autogen.client.models.experiment_tag_dto import ExperimentTagDto
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, conlist
 
 

truefoundry/ml/{autogen → _autogen}/client/models/create_experiment_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.

truefoundry/ml/{autogen → _autogen}/client/models/create_model_version_request_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,6 @@ import pprint
 import re  # noqa: F401
 from typing import Any, Dict, Optional
 
-from truefoundry.ml.autogen.client.models.internal_metadata import InternalMetadata
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictInt, StrictStr
 
 
@@ -32,7 +31,7 @@ class CreateModelVersionRequestDto(BaseModel):
     artifact_metadata: Optional[Dict[str, Any]] = None
     data_path: Optional[StrictStr] = None
     step: Optional[StrictInt] = None
-    internal_metadata: Optional[InternalMetadata] = None
+    internal_metadata: Optional[Dict[str, Any]] = None
     __properties = [
         "artifact_version_id",
         "description",
@@ -64,9 +63,6 @@ class CreateModelVersionRequestDto(BaseModel):
     def to_dict(self):
         """Returns the dictionary representation of the model using alias"""
         _dict = self.dict(by_alias=True, exclude={}, exclude_none=True)
-        # override the default output from truefoundry.pydantic_v1 by calling `to_dict()` of internal_metadata
-        if self.internal_metadata:
-            _dict["internal_metadata"] = self.internal_metadata.to_dict()
         return _dict
 
     @classmethod
@@ -85,11 +81,7 @@ class CreateModelVersionRequestDto(BaseModel):
                 "artifact_metadata": obj.get("artifact_metadata"),
                 "data_path": obj.get("data_path"),
                 "step": obj.get("step"),
-                "internal_metadata": InternalMetadata.from_dict(
-                    obj.get("internal_metadata")
-                )
-                if obj.get("internal_metadata") is not None
-                else None,
+                "internal_metadata": obj.get("internal_metadata"),
             }
         )
         return _obj