truefoundry 0.5.10rc2__py3-none-any.whl → 0.6.1__py3-none-any.whl

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

@@ -35,49 +35,40 @@ class AgentWithFQN(BaseModel):
     """
 
     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",
-    )
-    type: StrictStr = Field(
-        default=..., description="+value=agent +usage=Type of the entity"
+        default=None, description="Version of the entity"
     )
+    type: StrictStr = Field(default=..., description="Type of the entity")
     available_tools: conlist(constr(strict=True, min_length=1)) = Field(
-        default=...,
-        description='+sort=30 +uiType=Hidden +label=Tools +usage=Tools available to the agent +uiProps={"descriptionInline":true}',
+        default=..., description="Tools available to the agent"
     )
     goal: constr(strict=True, max_length=128, min_length=1) = Field(
         default=...,
-        description='+sort=10 +uiType=TextArea +label=Goal +usage=Short form description. Will be used as `description` when this agent is used as a tool. +uiProps={"descriptionInline":true}',
+        description="Short form description. Will be used as `description` when this agent is used as a tool.",
     )
     instruction: constr(strict=True, max_length=2620, min_length=1) = Field(
         default=...,
-        description='`instruction` is the system prompt for now. (2.5 * 1024) +sort=20 +uiType=AgentInstructions +uiProps={"helpText":"Use the syntax ${Tool FQN} to reference a tool, and ${AGENT FQN} to reference another agent"} +label=Instructions +usage=Instructions for the agent to follow to achieve the goal +uiProps={"descriptionInline":true}',
+        description="Instructions for the agent to follow to achieve the goal",
     )
     model_id: constr(strict=True, min_length=1) = Field(
-        default=...,
-        description='+sort=40 +uiType=EnabledModelSelector +uiProps={"searchable":true,"modelType":"chat","providerType":"openai"} +label=Model +usage=Model to use when running the agent',
+        default=..., description="Model to use when running the agent"
     )
-    id: StrictStr = Field(default=..., description="+usage=ID of the agent")
-    fqn: StrictStr = Field(default=..., description="+usage=FQN of the agent")
+    id: StrictStr = Field(default=..., description="ID of the agent")
+    fqn: StrictStr = Field(default=..., description="FQN of the agent")
     __properties = [
         "name",
         "description",

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

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

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

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

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

@@ -19,9 +19,11 @@ 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.artifact_version_dto import ArtifactVersionDto
-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.artifact_version_dto import (
+    ArtifactVersionDto,
+)
+from truefoundry.ml._autogen.client.models.subject import Subject
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictInt, StrictStr, conlist
 
 

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

@@ -18,7 +18,7 @@ import pprint
 import re  # noqa: F401
 from typing import Any, Dict, Optional
 
-from truefoundry.ml.autogen.client.models.source import Source
+from truefoundry.ml._autogen.client.models.source import Source
 from truefoundry.pydantic_v1 import (
     BaseModel,
     Field,
@@ -29,43 +29,38 @@ from truefoundry.pydantic_v1 import (
 )
 
 
-class ArtifactVersionManifest(BaseModel):
+class ArtifactManifest(BaseModel):
     """
     Artifact Version manifest.  # 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] = "artifact-version"
     source: Source = Field(...)
     step: Optional[conint(strict=True, ge=0)] = Field(
         default=0,
-        description='+label=Step +usage=Step/Epoch number in an iterative training loop the artifact 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 artifact 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 artifact version +uiType=Hidden",
+        description="ID of the MLRepo Run that generated the artifact version",
     )
     __properties = [
         "name",
@@ -141,8 +136,8 @@ class ArtifactVersionManifest(BaseModel):
         return json.dumps(self.to_dict())
 
     @classmethod
-    def from_json(cls, json_str: str) -> ArtifactVersionManifest:
-        """Create an instance of ArtifactVersionManifest from a JSON string"""
+    def from_json(cls, json_str: str) -> ArtifactManifest:
+        """Create an instance of ArtifactManifest from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self):
@@ -154,15 +149,15 @@ class ArtifactVersionManifest(BaseModel):
         return _dict
 
     @classmethod
-    def from_dict(cls, obj: dict) -> ArtifactVersionManifest:
-        """Create an instance of ArtifactVersionManifest from a dict"""
+    def from_dict(cls, obj: dict) -> ArtifactManifest:
+        """Create an instance of ArtifactManifest from a dict"""
         if obj is None:
             return None
 
         if not isinstance(obj, dict):
-            return ArtifactVersionManifest.parse_obj(obj)
+            return ArtifactManifest.parse_obj(obj)
 
-        _obj = ArtifactVersionManifest.parse_obj(
+        _obj = ArtifactManifest.parse_obj(
             {
                 "name": obj.get("name"),
                 "description": obj.get("description"),

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

@@ -26,12 +26,10 @@ class ArtifactPath(BaseModel):
     ArtifactPath
     """
 
-    src: StrictStr = Field(
-        default=..., description="+label=Source path +usage=Local file or folder path"
-    )
+    src: StrictStr = Field(default=..., description="Local file or folder path")
     dest: Optional[StrictStr] = Field(
         default=None,
-        description="+label=Destination path +usage=Relative path where the file or folder will be uploaded to in the artifact",
+        description="Relative path where the file or folder will be uploaded to in the artifact",
     )
     __properties = ["src", "dest"]
 

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

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

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

@@ -19,14 +19,14 @@ 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.manifest1 import Manifest1
-from truefoundry.ml.autogen.client.models.subject import Subject
+from truefoundry.ml._autogen.client.models.manifest1 import Manifest1
+from truefoundry.ml._autogen.client.models.subject import Subject
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictInt, StrictStr
 
 

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

@@ -17,7 +17,9 @@ import json
 import pprint
 import re  # noqa: F401
 
-from truefoundry.ml.autogen.client.models.artifact_version_dto import ArtifactVersionDto
+from truefoundry.ml._autogen.client.models.artifact_version_dto import (
+    ArtifactVersionDto,
+)
 from truefoundry.pydantic_v1 import BaseModel, Field
 
 

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

@@ -18,23 +18,19 @@ import pprint
 import re  # noqa: F401
 from typing import Optional
 
-from truefoundry.ml.autogen.client.models.content import Content
+from truefoundry.ml._autogen.client.models.content import Content
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
 
 
 class AssistantMessage(BaseModel):
     """
-    +label=Assistant Message +usage=Assistant message turn  # noqa: E501
+    Assistant message turn  # noqa: E501
     """
 
-    role: StrictStr = Field(
-        default=...,
-        description="+value=assistant +label=Role +usage=Role of the message +uiType=Ignore",
-    )
+    role: StrictStr = Field(default=..., description="Role of the message")
     content: Content = Field(...)
     name: Optional[StrictStr] = Field(
-        default=None,
-        description="+label=Name +usage=Name of the assistant this message is from",
+        default=None, description="Name of the assistant this message is from"
     )
     __properties = ["role", "content", "name"]
 

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

@@ -17,23 +17,20 @@ import json
 import pprint
 import re  # noqa: F401
 
-from truefoundry.ml.autogen.client.models.mime_type import MimeType
+from truefoundry.ml._autogen.client.models.mime_type import MimeType
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, constr, validator
 
 
 class BlobStorageReference(BaseModel):
     """
-    +label=Blob Storage Reference +usage=A reference to content offloaded to blob storage +docs=Defines the structure for blob storage content references, including type, path, and mime_type  # noqa: E501
+    A reference to content offloaded to blob storage  # noqa: E501
     """
 
-    type: StrictStr = Field(default=..., description="+value=blob-storage")
+    type: StrictStr = Field(...)
     path: constr(strict=True) = Field(
-        default=...,
-        description="+label=Path +usage=Path to the content in blob storage",
-    )
-    mime_type: MimeType = Field(
-        default=..., description="+label=MIME Type +usage=MIME type of the content"
+        default=..., description="Path to the content in blob storage"
     )
+    mime_type: MimeType = Field(default=..., description="MIME type of the content")
     __properties = ["type", "path", "mime_type"]
 
     @validator("type")

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

@@ -18,10 +18,10 @@ import pprint
 import re  # noqa: F401
 from typing import Any, Dict, Optional
 
-from truefoundry.ml.autogen.client.models.chat_prompt_manifest_messages_inner import (
+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,
@@ -39,36 +39,31 @@ class ChatPromptManifest(BaseModel):
     """
 
     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] = "chat_prompt"
     messages: conlist(ChatPromptManifestMessagesInner) = Field(
         default=...,
-        description="+sort=4000 +label=Messages in the chat conversation +usage=List of messages in the chat conversation, must be non-empty",
+        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 referenced in messages and that can be replaced when running generation +sort=5000",
+        description="Variables referenced in messages and that can be replaced when running generation",
     )
     model_configuration: ModelConfiguration = Field(...)
     __properties = [

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

@@ -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,

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

@@ -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 +uiType=TextArea
+    Text content for the assistant message
     """
 
     # data type: str

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

@@ -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 +uiType=TextArea
+    Text content for the system message
     """
 
     # data type: str

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

@@ -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=Content of the user message. can be a mix of text and images +uiType=TextArea
+    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

@@ -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

@@ -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_version_request_dto.py

@@ -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_dataset_request_dto.py

@@ -18,7 +18,7 @@ import pprint
 import re  # noqa: F401
 from typing import Any, Dict, Optional
 
-from truefoundry.ml.autogen.client.models.data_directory_manifest import (
+from truefoundry.ml._autogen.client.models.data_directory_manifest import (
     DataDirectoryManifest,
 )
 from truefoundry.pydantic_v1 import BaseModel, StrictStr

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

@@ -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_multi_part_upload_for_dataset_response_dto.py

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

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

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

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

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

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

@@ -18,7 +18,7 @@ import pprint
 import re  # noqa: F401
 from typing import Any, Dict, Optional
 
-from truefoundry.ml.autogen.client.models.source1 import Source1
+from truefoundry.ml._autogen.client.models.source1 import Source1
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, constr, validator
 
 
@@ -29,20 +29,18 @@ class DataDirectoryManifest(BaseModel):
 
     type: Optional[StrictStr] = "data-dir"
     name: constr(strict=True, max_length=256) = Field(
-        default=...,
-        description="+sort=1 +label=Name +usage=Name of the data directory +message=The data directory name should start with lowercase alphabets and can contain alphanumeric and can include '-' in between",
+        default=..., description="Name of the data directory"
     )
     ml_repo: constr(strict=True) = Field(
         default=...,
-        description="+label=MLRepo +usage=Name of the ML Repo to create the data directory under +uiType=Hidden",
+        description="Name of the ML Repo to create the data directory under",
     )
     description: Optional[constr(strict=True, max_length=512)] = Field(
-        default=None,
-        description="+sort=2 +label=Description +usage=Description for the data directory",
+        default=None, description="Description for the data directory"
     )
     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',
+        description='Key Value metadata. Should be valid JSON. For e.g. `{"business-unit": "sales", "quality": "good", "rating": 4.5}`',
     )
     source: Source1 = Field(...)
     __properties = ["type", "name", "ml_repo", "description", "metadata", "source"]

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

@@ -19,11 +19,11 @@ import re  # noqa: F401
 from datetime import datetime
 from typing import Any, Dict, Optional
 
-from truefoundry.ml.autogen.client.models.data_directory_manifest import (
+from truefoundry.ml._autogen.client.models.data_directory_manifest import (
     DataDirectoryManifest,
 )
-from truefoundry.ml.autogen.client.models.experiment_dto import ExperimentDto
-from truefoundry.ml.autogen.client.models.subject import Subject
+from truefoundry.ml._autogen.client.models.experiment_dto import ExperimentDto
+from truefoundry.ml._autogen.client.models.subject import Subject
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr
 
 

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

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

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

@@ -18,8 +18,8 @@ import pprint
 import re  # noqa: F401
 from typing import Any, Dict, Optional
 
-from truefoundry.ml.autogen.client.models.experiment_tag_dto import ExperimentTagDto
-from truefoundry.ml.autogen.client.models.subject import Subject
+from truefoundry.ml._autogen.client.models.experiment_tag_dto import ExperimentTagDto
+from truefoundry.ml._autogen.client.models.subject import Subject
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictInt, StrictStr, conlist
 
 

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

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

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

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