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

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

@@ -22,12 +22,10 @@ from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
 
 class ONNXFramework(BaseModel):
     """
-    +label=ONNX +icon=onnx  # noqa: E501
+    ONNXFramework
     """
 
-    type: StrictStr = Field(
-        default=..., description="+label=Type +usage=Type of the framework +value=onnx"
-    )
+    type: StrictStr = Field(default=..., description="Type of the framework")
     __properties = ["type"]
 
     @validator("type")

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

@@ -18,7 +18,7 @@ import pprint
 import re  # noqa: F401
 from typing import TYPE_CHECKING, Any, Dict, 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 (
@@ -33,7 +33,7 @@ OPENAPISPEC_ANY_OF_SCHEMAS = ["BlobStorageReference", "object"]
 
 class OpenapiSpec(BaseModel):
     """
-    +sort=20 +uiType=OpenapiSchema +label=OpenAPI Spec +usage=OpenAPI Spec for the tool describing the API, endpoints and parameters. [Sample OpenAPI Spec Link](https://assets.production.truefoundry.com/sample-openapi.json) +uiProps={\"descriptionInline\":true} +placeholder={\"openapi\":\"3.0.0\",\"info\":{\"title\":\"LLM Agent Tools API\",\"version\":\"1.0.0\"},\"servers\":[{\"url\":\"https://api.example.com/v1\"}],\"paths\":{\"/weather\":{\"get\":{\"summary\":\"Get current weather\",\"description\":\"Fetches the current weather for a given location.\",\"parameters\":[{\"name\":\"location\",\"in\":\"query\",\"required\":true,\"schema\":{\"type\":\"string\"}}],\"responses\":{\"200\":{\"description\":\"Successful response\",\"content\":{\"application/json\":{\"schema\":{\"type\":\"object\",\"properties\":{\"location\":{\"type\":\"string\"},\"temperature\":{\"type\":\"number\"},\"condition\":{\"type\":\"string\"}}}}}}}}}}}
+    OpenAPI Spec for the tool describing the API, endpoints and parameters. [Sample OpenAPI Spec Link](https://assets.production.truefoundry.com/sample-openapi.json)
     """
 
     # data type: BlobStorageReference

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

@@ -22,13 +22,10 @@ from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
 
 class PaddleFramework(BaseModel):
     """
-    +label=Paddle +icon=paddle  # noqa: E501
+    PaddleFramework
     """
 
-    type: StrictStr = Field(
-        default=...,
-        description="+label=Type +usage=Type of the framework +value=paddle",
-    )
+    type: StrictStr = Field(default=..., description="Type of the framework")
     __properties = ["type"]
 
     @validator("type")

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

@@ -18,13 +18,13 @@ import pprint
 import re  # noqa: F401
 from typing import Optional, Union
 
-from truefoundry.ml.autogen.client.models.stop import Stop
+from truefoundry.ml._autogen.client.models.stop import Stop
 from truefoundry.pydantic_v1 import BaseModel, StrictFloat, StrictInt
 
 
 class Parameters(BaseModel):
     """
-    +label=Parameters +usage=Parameters to pass to the model when generating  # noqa: E501
+    Parameters to pass to the model when generating  # noqa: E501
     """
 
     max_tokens: Optional[StrictInt] = None

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

@@ -22,13 +22,10 @@ from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
 
 class PyTorchFramework(BaseModel):
     """
-    +label=PyTorch +icon=pytorch  # noqa: E501
+    PyTorchFramework
     """
 
-    type: StrictStr = Field(
-        default=...,
-        description="+label=Type +usage=Type of the framework +value=pytorch",
-    )
+    type: StrictStr = Field(default=..., description="Type of the framework")
     __properties = ["type"]
 
     @validator("type")

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

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

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

@@ -18,9 +18,9 @@ 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.param_dto import ParamDto
-from truefoundry.ml.autogen.client.models.run_tag_dto import RunTagDto
+from truefoundry.ml._autogen.client.models.metric_dto import MetricDto
+from truefoundry.ml._autogen.client.models.param_dto import ParamDto
+from truefoundry.ml._autogen.client.models.run_tag_dto import RunTagDto
 from truefoundry.pydantic_v1 import BaseModel, conlist
 
 

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

@@ -18,8 +18,8 @@ import pprint
 import re  # noqa: F401
 from typing import Optional
 
-from truefoundry.ml.autogen.client.models.run_data_dto import RunDataDto
-from truefoundry.ml.autogen.client.models.run_info_dto import RunInfoDto
+from truefoundry.ml._autogen.client.models.run_data_dto import RunDataDto
+from truefoundry.ml._autogen.client.models.run_info_dto import RunInfoDto
 from truefoundry.pydantic_v1 import BaseModel, Field
 
 

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

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

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

@@ -18,7 +18,7 @@ import pprint
 import re  # noqa: F401
 from typing import Optional
 
-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, StrictInt, StrictStr, conlist
 
 

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

@@ -16,7 +16,6 @@ from __future__ import annotations
 import json
 import pprint
 import re  # noqa: F401
-from typing import Optional
 
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr
 
@@ -27,7 +26,7 @@ class SignedURLDto(BaseModel):
     """
 
     signed_url: StrictStr = Field(...)
-    path: Optional[StrictStr] = None
+    path: StrictStr = Field(...)
     __properties = ["signed_url", "path"]
 
     class Config:

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

@@ -18,8 +18,10 @@ import pprint
 import re  # noqa: F401
 from typing import Optional
 
-from truefoundry.ml.autogen.client.models.sklearn_model_schema import SklearnModelSchema
-from truefoundry.ml.autogen.client.models.sklearn_serialization_format import (
+from truefoundry.ml._autogen.client.models.sklearn_model_schema import (
+    SklearnModelSchema,
+)
+from truefoundry.ml._autogen.client.models.sklearn_serialization_format import (
     SklearnSerializationFormat,
 )
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
@@ -27,16 +29,13 @@ from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
 
 class SklearnFramework(BaseModel):
     """
-    +label=Sklearn +icon=sklearn  # noqa: E501
+    SklearnFramework
     """
 
-    type: StrictStr = Field(
-        default=...,
-        description="+label=Type +usage=Type of the framework +value=sklearn",
-    )
+    type: StrictStr = Field(default=..., description="Type of the framework")
     model_filepath: Optional[StrictStr] = Field(
         default=None,
-        description='+label=Model file path +usage=Relative path to the model file in the model version contents +uiProps={"descriptionInline":true}',
+        description="Relative path to the model file in the model version contents",
     )
     serialization_format: Optional[SklearnSerializationFormat] = None
     model_schema: Optional[SklearnModelSchema] = None

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

@@ -18,26 +18,23 @@ import pprint
 import re  # noqa: F401
 from typing import Any, Dict
 
-from truefoundry.ml.autogen.client.models.infer_method_name import InferMethodName
+from truefoundry.ml._autogen.client.models.infer_method_name import InferMethodName
 from truefoundry.pydantic_v1 import BaseModel, Field, conlist
 
 
 class SklearnModelSchema(BaseModel):
     """
-    +label=Scikit Learn Model Schema  # noqa: E501
+    SklearnModelSchema
     """
 
     infer_method_name: InferMethodName = Field(
-        default=...,
-        description="+label=Inference Method Name +usage=Name of the method used for inference",
+        default=..., description="Name of the method used for inference"
     )
     inputs: conlist(Dict[str, Any]) = Field(
-        default=...,
-        description='+label= Input Schema +usage=Schema of the input +uiProps={"descriptionInline":true}',
+        default=..., description="Schema of the input"
     )
     outputs: conlist(Dict[str, Any]) = Field(
-        default=...,
-        description='+label= Output Schema +usage=Schema of the output +uiProps={"descriptionInline":true}',
+        default=..., description="Schema of the output"
     )
     __properties = ["infer_method_name", "inputs", "outputs"]
 

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

@@ -19,7 +19,7 @@ from aenum import Enum
 
 class SklearnSerializationFormat(str, Enum):
     """
-    +label=Scikit Learn Serialization Format +usage=Serialization format used to save the sklearn model
+    Serialization format used to save the sklearn model
     """
 
     """

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

@@ -18,13 +18,13 @@ import pprint
 import re  # noqa: F401
 from typing import TYPE_CHECKING, Any, List, Optional, Union
 
-from truefoundry.ml.autogen.client.models.external_blob_storage_source import (
+from truefoundry.ml._autogen.client.models.external_blob_storage_source import (
     ExternalBlobStorageSource,
 )
-from truefoundry.ml.autogen.client.models.local_artifact_source import (
+from truefoundry.ml._autogen.client.models.local_artifact_source import (
     LocalArtifactSource,
 )
-from truefoundry.ml.autogen.client.models.true_foundry_managed_source import (
+from truefoundry.ml._autogen.client.models.true_foundry_managed_source import (
     TrueFoundryManagedSource,
 )
 from truefoundry.pydantic_v1 import (
@@ -43,7 +43,7 @@ SOURCE_ANY_OF_SCHEMAS = [
 
 class Source(BaseModel):
     """
-    +label=Artifact Source +uiType=Group
+    Source
     """
 
     # data type: TrueFoundryManagedSource

truefoundry/ml/{autogen → _autogen}/client/models/source1.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.external_blob_storage_source import (
+from truefoundry.ml._autogen.client.models.external_blob_storage_source import (
     ExternalBlobStorageSource,
 )
-from truefoundry.ml.autogen.client.models.true_foundry_managed_source import (
+from truefoundry.ml._autogen.client.models.true_foundry_managed_source import (
     TrueFoundryManagedSource,
 )
 from truefoundry.pydantic_v1 import (
@@ -36,7 +36,7 @@ SOURCE1_ANY_OF_SCHEMAS = ["ExternalBlobStorageSource", "TrueFoundryManagedSource
 
 class Source1(BaseModel):
     """
-    +label=Data Directory Source +uiType=Group
+    Source1
     """
 
     # data type: TrueFoundryManagedSource

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

@@ -18,11 +18,11 @@ import pprint
 import re  # noqa: F401
 from typing import TYPE_CHECKING, Any, List, Optional, Union
 
-from truefoundry.ml.autogen.client.models.external_blob_storage_source import (
+from truefoundry.ml._autogen.client.models.external_blob_storage_source import (
     ExternalBlobStorageSource,
 )
-from truefoundry.ml.autogen.client.models.local_model_source import LocalModelSource
-from truefoundry.ml.autogen.client.models.true_foundry_managed_source import (
+from truefoundry.ml._autogen.client.models.local_model_source import LocalModelSource
+from truefoundry.ml._autogen.client.models.true_foundry_managed_source import (
     TrueFoundryManagedSource,
 )
 from truefoundry.pydantic_v1 import (
@@ -41,7 +41,7 @@ SOURCE2_ANY_OF_SCHEMAS = [
 
 class Source2(BaseModel):
     """
-    +label=Model Source +uiType=Group
+    Source2
     """
 
     # data type: TrueFoundryManagedSource

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

@@ -22,12 +22,10 @@ from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
 
 class SpaCyFramework(BaseModel):
     """
-    +label=SpaCy +icon=spacy  # noqa: E501
+    SpaCyFramework
     """
 
-    type: StrictStr = Field(
-        default=..., description="+label=Type +usage=Type of the framework +value=spacy"
-    )
+    type: StrictStr = Field(default=..., description="Type of the framework")
     __properties = ["type"]
 
     @validator("type")

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

@@ -22,13 +22,10 @@ from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
 
 class StatsModelsFramework(BaseModel):
     """
-    +label=StatsModels +icon=statsmodel  # noqa: E501
+    StatsModelsFramework
     """
 
-    type: StrictStr = Field(
-        default=...,
-        description="+label=Type +usage=Type of the framework +value=statsmodels",
-    )
+    type: StrictStr = Field(default=..., description="Type of the framework")
     __properties = ["type"]
 
     @validator("type")

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

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

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

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

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

@@ -18,23 +18,18 @@ import pprint
 import re  # noqa: F401
 from typing import Optional
 
-from truefoundry.ml.autogen.client.models.content1 import Content1
+from truefoundry.ml._autogen.client.models.content1 import Content1
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
 
 
 class SystemMessage(BaseModel):
     """
-    +label=System Message +usage=System message for the chat  # noqa: E501
+    System message for the chat  # noqa: E501
     """
 
-    role: StrictStr = Field(
-        default=...,
-        description="+value=system +label=Role +usage=Role of the message +uiType=Ignore",
-    )
+    role: StrictStr = Field(default=..., description="Role of the message")
     content: Content1 = Field(...)
-    name: Optional[StrictStr] = Field(
-        default=None, description="+label=Name +usage=Name of the system"
-    )
+    name: Optional[StrictStr] = Field(default=None, description="Name of the system")
     __properties = ["role", "content", "name"]
 
     @validator("role")

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

@@ -22,13 +22,10 @@ from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
 
 class TensorFlowFramework(BaseModel):
     """
-    +label=Tensorflow +icon=tensorflow  # noqa: E501
+    TensorFlowFramework
     """
 
-    type: StrictStr = Field(
-        default=...,
-        description="+label=Type +usage=Type of the framework +value=tensorflow",
-    )
+    type: StrictStr = Field(default=..., description="Type of the framework")
     __properties = ["type"]
 
     @validator("type")

truefoundry/ml/{autogen → _autogen}/client/models/text.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 @@ TEXT_ANY_OF_SCHEMAS = ["BlobStorageReference", "str"]
 
 class Text(BaseModel):
     """
-    +label=Text +usage=Text content for the message
+    Text content for the message
     """
 
     # data type: str

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

@@ -17,18 +17,16 @@ import json
 import pprint
 import re  # noqa: F401
 
-from truefoundry.ml.autogen.client.models.text import Text
+from truefoundry.ml._autogen.client.models.text import Text
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
 
 
 class TextContentPart(BaseModel):
     """
-    +label=Text Content +usage=Text content for the message  # noqa: E501
+    Text content for the message  # noqa: E501
     """
 
-    type: StrictStr = Field(
-        default=..., description="+value=text +usage=Type of the content part"
-    )
+    type: StrictStr = Field(default=..., description="Type of the content part")
     text: Text = Field(...)
     __properties = ["type", "text"]
 

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

@@ -18,30 +18,26 @@ import pprint
 import re  # noqa: F401
 from typing import Optional
 
-from truefoundry.ml.autogen.client.models.library_name import LibraryName
+from truefoundry.ml._autogen.client.models.library_name import LibraryName
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
 
 
 class TransformersFramework(BaseModel):
     """
-    +label=Transformers +icon=transformers  # noqa: E501
+    TransformersFramework
     """
 
-    type: StrictStr = Field(
-        default=...,
-        description="+label=Type +usage=Type of the framework +value=transformers",
-    )
+    type: StrictStr = Field(default=..., description="Type of the framework")
     library_name: Optional[LibraryName] = Field(
-        default=None,
-        description="+label=Library Name +usage=Name of the library for the framework",
+        default=None, description="Name of the library for the framework"
     )
     pipeline_tag: Optional[StrictStr] = Field(
         default=None,
-        description='+label=Pipeline Tag +usage=The `pipeline()` task this model can be used with e.g. `text-generation`. See [huggingface docs](https://huggingface.co/docs/transformers/main/en/main_classes/pipelines#transformers.pipeline.task) for all possible values +uiProps={"descriptionInline":true}',
+        description="The `pipeline()` task this model can be used with e.g. `text-generation`. See [huggingface docs](https://huggingface.co/docs/transformers/main/en/main_classes/pipelines#transformers.pipeline.task) for all possible values",
     )
     base_model: Optional[StrictStr] = Field(
         default=None,
-        description='+label=Base Model +usage=Base model Id from HuggingFace Hub. If this is a finetuned model, this points to the base model id used for finetuning. +uiProps={"descriptionInline":true}',
+        description="Base model Id from HuggingFace Hub. If this is a finetuned model, this points to the base model id used for finetuning.",
     )
     __properties = ["type", "library_name", "pipeline_tag", "base_model"]
 

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

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

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

@@ -18,22 +18,16 @@ import pprint
 import re  # noqa: F401
 from typing import Optional
 
-from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
+from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, constr, validator
 
 
 class TrueFoundryManagedSource(BaseModel):
     """
-    +label=TrueFoundry Managed Source  # noqa: E501
+    TrueFoundryManagedSource
     """
 
-    type: StrictStr = Field(
-        default=...,
-        description="+label=Type +usage=Type of the source +value=truefoundry",
-    )
-    uri: Optional[StrictStr] = Field(
-        default=None,
-        description='+label=URI +uiType=ModelUploadInput +uiProps={"hideClear":true}',
-    )
+    type: StrictStr = Field(default=..., description="Type of the source")
+    uri: Optional[constr(strict=True)] = None
     __properties = ["type", "uri"]
 
     @validator("type")
@@ -43,6 +37,18 @@ class TrueFoundryManagedSource(BaseModel):
             raise ValueError("must be one of enum values ('truefoundry')")
         return value
 
+    @validator("uri")
+    def uri_validate_regular_expression(cls, value):
+        """Validates the regular expression"""
+        if value is None:
+            return value
+
+        if not re.match(r"^(s3|gs|wasbs):\/\/.*", value):
+            raise ValueError(
+                r"must validate the regular expression /^(s3|gs|wasbs):\/\/.*/"
+            )
+        return value
+
     class Config:
         """Pydantic configuration"""
 

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

truefoundry/ml/{autogen → _autogen}/client/models/update_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, Field, StrictStr

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

@@ -18,9 +18,7 @@ import pprint
 import re  # noqa: F401
 from typing import Any, Dict, Optional
 
-from truefoundry.ml.autogen.client.models.model_version_manifest import (
-    ModelVersionManifest,
-)
+from truefoundry.ml._autogen.client.models.model_manifest import ModelManifest
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr
 
 
@@ -33,7 +31,7 @@ class UpdateModelVersionRequestDto(BaseModel):
     description: Optional[StrictStr] = None
     artifact_metadata: Optional[Dict[str, Any]] = None
     model_framework: Optional[StrictStr] = None
-    manifest: Optional[ModelVersionManifest] = None
+    manifest: Optional[ModelManifest] = None
     __properties = [
         "id",
         "description",
@@ -84,7 +82,7 @@ class UpdateModelVersionRequestDto(BaseModel):
                 "description": obj.get("description"),
                 "artifact_metadata": obj.get("artifact_metadata"),
                 "model_framework": obj.get("model_framework"),
-                "manifest": ModelVersionManifest.from_dict(obj.get("manifest"))
+                "manifest": ModelManifest.from_dict(obj.get("manifest"))
                 if obj.get("manifest") is not None
                 else None,
             }

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

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

truefoundry/ml/{autogen → _autogen}/client/models/url.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 @@ URL_ANY_OF_SCHEMAS = ["BlobStorageReference", "str"]
 
 class Url(BaseModel):
     """
-    +label=URL +usage=Image URL linking to the image
+    Image URL linking to the image
     """
 
     # data type: str