truefoundry 0.4.4rc9__py3-none-any.whl → 0.4.4rc11__py3-none-any.whl

truefoundry/common/tfy_signed_url_client.py

@@ -92,7 +92,7 @@ class SignedURLClient:
             retries=self.max_retries, backoff_factor=self.retry_backoff_factor
         )
 
-    @log_time
+    @log_time(prefix=LOG_PREFIX)
     def _make_request(
         self,
         endpoint: str,
@@ -111,6 +111,7 @@ class SignedURLClient:
         except RequestException as e:
             raise RuntimeError(f"Error during request to {url}: {e}") from e
 
+    @log_time(prefix=LOG_PREFIX)
     def _make_server_api_call(
         self,
         endpoint: SignedURLServerEndpoint,
@@ -124,7 +125,7 @@ class SignedURLClient:
             endpoint=endpoint_with_params, method="GET", payload=None, headers=headers
         )
 
-    @log_time
+    @log_time(prefix=LOG_PREFIX)
     def _upload_data(self, signed_url: str, data: Any) -> None:
         """
         Upload data to the specified storage path using a signed URL.
@@ -144,6 +145,7 @@ class SignedURLClient:
         except Exception as e:
             raise RuntimeError(f"Failed to upload data: {e}") from e
 
+    @log_time(prefix=LOG_PREFIX)
     def upload_from_bytes(self, data: bytes, storage_uri: str) -> str:
         """Upload bytes to the specified storage path using a signed URL."""
         signed_object = self._make_server_api_call(
@@ -155,7 +157,7 @@ class SignedURLClient:
         self._upload_data(pre_signed_object_dto.signed_url, data)
         return storage_uri
 
-    @log_time
+    @log_time(prefix=LOG_PREFIX)
     def upload(self, file_path: str, storage_uri: str) -> str:
         """Upload a file to the specified storage path using a signed URL."""
         logger.info(f"{LOG_PREFIX} Uploading {file_path} to {storage_uri}")
@@ -169,7 +171,7 @@ class SignedURLClient:
             self._upload_data(pre_signed_object_dto.signed_url, file)
         return storage_uri
 
-    @log_time
+    @log_time(prefix=LOG_PREFIX)
     def _download_file(
         self, signed_url: str, local_path: Optional[str] = None
     ) -> Optional[bytes]:
@@ -191,7 +193,7 @@ class SignedURLClient:
         except RequestException as e:
             raise RuntimeError(f"Failed to download file from {signed_url}: {e}") from e
 
-    @log_time
+    @log_time(prefix=LOG_PREFIX)
     def download(self, storage_uri: str, local_path: str) -> Optional[str]:
         """Download a file from the specified storage path to a local path using a signed URL."""
         logger.info(f"{LOG_PREFIX} Downloading {storage_uri} to {local_path}")
@@ -204,6 +206,7 @@ class SignedURLClient:
         self._download_file(presigned_object.signed_url, local_path)
         return local_path
 
+    @log_time(prefix=LOG_PREFIX)
     def download_to_bytes(self, storage_uri: str) -> bytes:
         """Download a file from the specified storage path and return it as bytes."""
         response = self._make_server_api_call(
@@ -214,6 +217,7 @@ class SignedURLClient:
         presigned_object = SignedURLAPIResponseDto.parse_obj(response)
         return self._download_file(presigned_object.signed_url)
 
+    @log_time(prefix=LOG_PREFIX)
     def exists(self, uri: str) -> bool:
         """Check if a file exists at the specified path."""
         response = self._make_server_api_call(
@@ -223,6 +227,7 @@ class SignedURLClient:
         )
         return SignedURLExistsAPIResponseDto.parse_obj(response).exists
 
+    @log_time(prefix=LOG_PREFIX)
     def is_directory(self, uri: str) -> bool:
         """Check if the specified URI is a directory."""
         response = self._make_server_api_call(
@@ -236,7 +241,7 @@ class SignedURLClient:
         logger.info(f"{LOG_PREFIX} Path {uri} is a directory: {is_directory}")
         return is_directory
 
-    @log_time
+    @log_time(prefix=LOG_PREFIX)
     def list_files(
         self, path: str, detail: bool = False, max_results: int = 1000
     ) -> Union[List[FileInfo], List[str]]:

truefoundry/common/utils.py

@@ -10,6 +10,7 @@ from truefoundry.common.constants import (
     MLFOUNDRY_SERVER_RELATIVE_PATH,
     TFY_HOST_ENV_KEY,
 )
+from truefoundry.logger import logger
 from truefoundry.pydantic_v1 import BaseSettings
 
 T = TypeVar("T")
@@ -104,7 +105,7 @@ def resolve_tfy_host(tfy_host: Optional[str] = None) -> str:
     return tfy_host
 
 
-def log_time(func):
+def log_time(func, prefix: str = ""):
     """Decorator to log the time taken by I/O operations."""
 
     @wraps(func)
@@ -112,7 +113,7 @@ def log_time(func):
         start_time = time.time()
         result = func(*args, **kwargs)
         elapsed_time = time.time() - start_time
-        print(f"{func.__name__} took {elapsed_time:.2f} seconds")
+        logger.info(f"{prefix} {func.__name__} took {elapsed_time:.2f} seconds")
         return result
 
     return wrapper

truefoundry/ml/autogen/client/__init__.py

@@ -64,6 +64,9 @@ from truefoundry.ml.autogen.client.models.artifact_version_dto import ArtifactVe
 from truefoundry.ml.autogen.client.models.artifact_version_response_dto import (
     ArtifactVersionResponseDto,
 )
+from truefoundry.ml.autogen.client.models.artifact_version_serialization_format import (
+    ArtifactVersionSerializationFormat,
+)
 from truefoundry.ml.autogen.client.models.artifact_version_status import (
     ArtifactVersionStatus,
 )
@@ -162,12 +165,16 @@ from truefoundry.ml.autogen.client.models.experiment_response_dto import (
     ExperimentResponseDto,
 )
 from truefoundry.ml.autogen.client.models.experiment_tag_dto import ExperimentTagDto
+from truefoundry.ml.autogen.client.models.external_model_source import (
+    ExternalModelSource,
+)
 from truefoundry.ml.autogen.client.models.feature_dto import FeatureDto
 from truefoundry.ml.autogen.client.models.feature_value_type import FeatureValueType
 from truefoundry.ml.autogen.client.models.file_info_dto import FileInfoDto
 from truefoundry.ml.autogen.client.models.finalize_artifact_version_request_dto import (
     FinalizeArtifactVersionRequestDto,
 )
+from truefoundry.ml.autogen.client.models.framework import Framework
 from truefoundry.ml.autogen.client.models.get_experiment_response_dto import (
     GetExperimentResponseDto,
 )
@@ -211,6 +218,7 @@ from truefoundry.ml.autogen.client.models.image_content_part import ImageContent
 from truefoundry.ml.autogen.client.models.image_url import ImageUrl
 from truefoundry.ml.autogen.client.models.internal_metadata import InternalMetadata
 from truefoundry.ml.autogen.client.models.latest_run_log_dto import LatestRunLogDto
+from truefoundry.ml.autogen.client.models.library_name import LibraryName
 from truefoundry.ml.autogen.client.models.list_artifact_versions_request_dto import (
     ListArtifactVersionsRequestDto,
 )
@@ -274,9 +282,6 @@ from truefoundry.ml.autogen.client.models.list_run_artifacts_response_dto import
 from truefoundry.ml.autogen.client.models.list_run_logs_response_dto import (
     ListRunLogsResponseDto,
 )
-from truefoundry.ml.autogen.client.models.list_seed_experiments_response_dto import (
-    ListSeedExperimentsResponseDto,
-)
 from truefoundry.ml.autogen.client.models.log_batch_request_dto import (
     LogBatchRequestDto,
 )
@@ -297,6 +302,9 @@ from truefoundry.ml.autogen.client.models.model_dto import ModelDto
 from truefoundry.ml.autogen.client.models.model_response_dto import ModelResponseDto
 from truefoundry.ml.autogen.client.models.model_schema_dto import ModelSchemaDto
 from truefoundry.ml.autogen.client.models.model_version_dto import ModelVersionDto
+from truefoundry.ml.autogen.client.models.model_version_manifest import (
+    ModelVersionManifest,
+)
 from truefoundry.ml.autogen.client.models.model_version_response_dto import (
     ModelVersionResponseDto,
 )
@@ -340,13 +348,25 @@ from truefoundry.ml.autogen.client.models.set_experiment_tag_request_dto import
 )
 from truefoundry.ml.autogen.client.models.set_tag_request_dto import SetTagRequestDto
 from truefoundry.ml.autogen.client.models.signed_url_dto import SignedURLDto
+from truefoundry.ml.autogen.client.models.source import Source
 from truefoundry.ml.autogen.client.models.stop import Stop
 from truefoundry.ml.autogen.client.models.store_run_logs_request_dto import (
     StoreRunLogsRequestDto,
 )
+from truefoundry.ml.autogen.client.models.subject import Subject
+from truefoundry.ml.autogen.client.models.subject_type import SubjectType
 from truefoundry.ml.autogen.client.models.system_message import SystemMessage
+from truefoundry.ml.autogen.client.models.tensorflow_framework import (
+    TensorflowFramework,
+)
 from truefoundry.ml.autogen.client.models.text import Text
 from truefoundry.ml.autogen.client.models.text_content_part import TextContentPart
+from truefoundry.ml.autogen.client.models.transformers_framework import (
+    TransformersFramework,
+)
+from truefoundry.ml.autogen.client.models.truefoundry_model_source import (
+    TruefoundryModelSource,
+)
 from truefoundry.ml.autogen.client.models.update_artifact_version_request_dto import (
     UpdateArtifactVersionRequestDto,
 )
@@ -365,6 +385,7 @@ from truefoundry.ml.autogen.client.models.update_run_request_dto import (
 from truefoundry.ml.autogen.client.models.update_run_response_dto import (
     UpdateRunResponseDto,
 )
+from truefoundry.ml.autogen.client.models.upload_model_source import UploadModelSource
 from truefoundry.ml.autogen.client.models.url import Url
 from truefoundry.ml.autogen.client.models.user_message import UserMessage
 from truefoundry.ml.autogen.client.models.validation_error import ValidationError

truefoundry/ml/autogen/client/api/experiments_api.py

@@ -41,9 +41,6 @@ from truefoundry.ml.autogen.client.models.list_colums_response_dto import (
 from truefoundry.ml.autogen.client.models.list_experiments_response_dto import (
     ListExperimentsResponseDto,
 )
-from truefoundry.ml.autogen.client.models.list_seed_experiments_response_dto import (
-    ListSeedExperimentsResponseDto,
-)
 from truefoundry.ml.autogen.client.models.set_experiment_tag_request_dto import (
     SetExperimentTagRequestDto,
 )
@@ -1497,140 +1494,6 @@ class ExperimentsApi:
             _request_auth=_params.get("_request_auth"),
         )
 
-    @validate_arguments
-    def seed_list_experiment_get(self, **kwargs) -> ListSeedExperimentsResponseDto:  # noqa: E501
-        """Seed List Experiment  # noqa: E501
-
-        List Seed Experiments  # noqa: E501
-        This method makes a synchronous HTTP request by default. To make an
-        asynchronous HTTP request, please pass async_req=True
-
-        >>> thread = api.seed_list_experiment_get(async_req=True)
-        >>> result = thread.get()
-
-        :param async_req: Whether to execute the request asynchronously.
-        :type async_req: bool, optional
-        :param _request_timeout: timeout setting for this request.
-               If one number provided, it will be total request
-               timeout. It can also be a pair (tuple) of
-               (connection, read) timeouts.
-        :return: Returns the result object.
-                 If the method is called asynchronously,
-                 returns the request thread.
-        :rtype: ListSeedExperimentsResponseDto
-        """
-        kwargs["_return_http_data_only"] = True
-        if "_preload_content" in kwargs:
-            message = "Error! Please call the seed_list_experiment_get_with_http_info method with `_preload_content` instead and obtain raw data from ApiResponse.raw_data"  # noqa: E501
-            raise ValueError(message)
-        return self.seed_list_experiment_get_with_http_info(**kwargs)  # noqa: E501
-
-    @validate_arguments
-    def seed_list_experiment_get_with_http_info(self, **kwargs) -> ApiResponse:  # noqa: E501
-        """Seed List Experiment  # noqa: E501
-
-        List Seed Experiments  # noqa: E501
-        This method makes a synchronous HTTP request by default. To make an
-        asynchronous HTTP request, please pass async_req=True
-
-        >>> thread = api.seed_list_experiment_get_with_http_info(async_req=True)
-        >>> result = thread.get()
-
-        :param async_req: Whether to execute the request asynchronously.
-        :type async_req: bool, optional
-        :param _preload_content: if False, the ApiResponse.data will
-                                 be set to none and raw_data will store the
-                                 HTTP response body without reading/decoding.
-                                 Default is True.
-        :type _preload_content: bool, optional
-        :param _return_http_data_only: response data instead of ApiResponse
-                                       object with status code, headers, etc
-        :type _return_http_data_only: bool, optional
-        :param _request_timeout: timeout setting for this request. If one
-                                 number provided, it will be total request
-                                 timeout. It can also be a pair (tuple) of
-                                 (connection, read) timeouts.
-        :param _request_auth: set to override the auth_settings for an a single
-                              request; this effectively ignores the authentication
-                              in the spec for a single request.
-        :type _request_auth: dict, optional
-        :type _content_type: string, optional: force content-type for the request
-        :return: Returns the result object.
-                 If the method is called asynchronously,
-                 returns the request thread.
-        :rtype: tuple(ListSeedExperimentsResponseDto, status_code(int), headers(HTTPHeaderDict))
-        """
-
-        _params = locals()
-
-        _all_params = []
-        _all_params.extend(
-            [
-                "async_req",
-                "_return_http_data_only",
-                "_preload_content",
-                "_request_timeout",
-                "_request_auth",
-                "_content_type",
-                "_headers",
-            ]
-        )
-
-        # validate the arguments
-        for _key, _val in _params["kwargs"].items():
-            if _key not in _all_params:
-                raise ApiTypeError(
-                    "Got an unexpected keyword argument '%s'"
-                    " to method seed_list_experiment_get" % _key
-                )
-            _params[_key] = _val
-        del _params["kwargs"]
-
-        _collection_formats = {}
-
-        # process the path parameters
-        _path_params = {}
-
-        # process the query parameters
-        _query_params = []
-        # process the header parameters
-        _header_params = dict(_params.get("_headers", {}))
-        # process the form parameters
-        _form_params = []
-        _files = {}
-        # process the body parameter
-        _body_params = None
-        # set the HTTP header `Accept`
-        _header_params["Accept"] = self.api_client.select_header_accept(
-            ["application/json"]
-        )  # noqa: E501
-
-        # authentication setting
-        _auth_settings = ["HTTPBearer", "APIKeyCookie"]  # noqa: E501
-
-        _response_types_map = {
-            "200": "ListSeedExperimentsResponseDto",
-        }
-
-        return self.api_client.call_api(
-            "/api/2.0/mlflow/experiments/seed/list",
-            "GET",
-            _path_params,
-            _query_params,
-            _header_params,
-            body=_body_params,
-            post_params=_form_params,
-            files=_files,
-            response_types_map=_response_types_map,
-            auth_settings=_auth_settings,
-            async_req=_params.get("async_req"),
-            _return_http_data_only=_params.get("_return_http_data_only"),  # noqa: E501
-            _preload_content=_params.get("_preload_content", True),
-            _request_timeout=_params.get("_request_timeout"),
-            collection_formats=_collection_formats,
-            _request_auth=_params.get("_request_auth"),
-        )
-
     @validate_arguments
     def set_experiment_tag_post(
         self, set_experiment_tag_request_dto: SetExperimentTagRequestDto, **kwargs

truefoundry/ml/autogen/client/api/mlfoundry_artifacts_api.py

@@ -1016,6 +1016,7 @@ class MlfoundryArtifactsApi:
     ) -> CreateArtifactVersionResponseDto:  # noqa: E501
         """Create Artifact Version  # noqa: E501
 
+        Create artifact version API  # noqa: E501
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
 
@@ -1060,6 +1061,7 @@ class MlfoundryArtifactsApi:
     ) -> ApiResponse:  # noqa: E501
         """Create Artifact Version  # noqa: E501
 
+        Create artifact version API  # noqa: E501
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
 

truefoundry/ml/autogen/client/models/__init__.py

@@ -35,6 +35,9 @@ from truefoundry.ml.autogen.client.models.artifact_version_dto import ArtifactVe
 from truefoundry.ml.autogen.client.models.artifact_version_response_dto import (
     ArtifactVersionResponseDto,
 )
+from truefoundry.ml.autogen.client.models.artifact_version_serialization_format import (
+    ArtifactVersionSerializationFormat,
+)
 from truefoundry.ml.autogen.client.models.artifact_version_status import (
     ArtifactVersionStatus,
 )
@@ -133,12 +136,16 @@ from truefoundry.ml.autogen.client.models.experiment_response_dto import (
     ExperimentResponseDto,
 )
 from truefoundry.ml.autogen.client.models.experiment_tag_dto import ExperimentTagDto
+from truefoundry.ml.autogen.client.models.external_model_source import (
+    ExternalModelSource,
+)
 from truefoundry.ml.autogen.client.models.feature_dto import FeatureDto
 from truefoundry.ml.autogen.client.models.feature_value_type import FeatureValueType
 from truefoundry.ml.autogen.client.models.file_info_dto import FileInfoDto
 from truefoundry.ml.autogen.client.models.finalize_artifact_version_request_dto import (
     FinalizeArtifactVersionRequestDto,
 )
+from truefoundry.ml.autogen.client.models.framework import Framework
 from truefoundry.ml.autogen.client.models.get_experiment_response_dto import (
     GetExperimentResponseDto,
 )
@@ -182,6 +189,7 @@ from truefoundry.ml.autogen.client.models.image_content_part import ImageContent
 from truefoundry.ml.autogen.client.models.image_url import ImageUrl
 from truefoundry.ml.autogen.client.models.internal_metadata import InternalMetadata
 from truefoundry.ml.autogen.client.models.latest_run_log_dto import LatestRunLogDto
+from truefoundry.ml.autogen.client.models.library_name import LibraryName
 from truefoundry.ml.autogen.client.models.list_artifact_versions_request_dto import (
     ListArtifactVersionsRequestDto,
 )
@@ -245,9 +253,6 @@ from truefoundry.ml.autogen.client.models.list_run_artifacts_response_dto import
 from truefoundry.ml.autogen.client.models.list_run_logs_response_dto import (
     ListRunLogsResponseDto,
 )
-from truefoundry.ml.autogen.client.models.list_seed_experiments_response_dto import (
-    ListSeedExperimentsResponseDto,
-)
 from truefoundry.ml.autogen.client.models.log_batch_request_dto import (
     LogBatchRequestDto,
 )
@@ -268,6 +273,9 @@ from truefoundry.ml.autogen.client.models.model_dto import ModelDto
 from truefoundry.ml.autogen.client.models.model_response_dto import ModelResponseDto
 from truefoundry.ml.autogen.client.models.model_schema_dto import ModelSchemaDto
 from truefoundry.ml.autogen.client.models.model_version_dto import ModelVersionDto
+from truefoundry.ml.autogen.client.models.model_version_manifest import (
+    ModelVersionManifest,
+)
 from truefoundry.ml.autogen.client.models.model_version_response_dto import (
     ModelVersionResponseDto,
 )
@@ -311,13 +319,25 @@ from truefoundry.ml.autogen.client.models.set_experiment_tag_request_dto import
 )
 from truefoundry.ml.autogen.client.models.set_tag_request_dto import SetTagRequestDto
 from truefoundry.ml.autogen.client.models.signed_url_dto import SignedURLDto
+from truefoundry.ml.autogen.client.models.source import Source
 from truefoundry.ml.autogen.client.models.stop import Stop
 from truefoundry.ml.autogen.client.models.store_run_logs_request_dto import (
     StoreRunLogsRequestDto,
 )
+from truefoundry.ml.autogen.client.models.subject import Subject
+from truefoundry.ml.autogen.client.models.subject_type import SubjectType
 from truefoundry.ml.autogen.client.models.system_message import SystemMessage
+from truefoundry.ml.autogen.client.models.tensorflow_framework import (
+    TensorflowFramework,
+)
 from truefoundry.ml.autogen.client.models.text import Text
 from truefoundry.ml.autogen.client.models.text_content_part import TextContentPart
+from truefoundry.ml.autogen.client.models.transformers_framework import (
+    TransformersFramework,
+)
+from truefoundry.ml.autogen.client.models.truefoundry_model_source import (
+    TruefoundryModelSource,
+)
 from truefoundry.ml.autogen.client.models.update_artifact_version_request_dto import (
     UpdateArtifactVersionRequestDto,
 )
@@ -336,6 +356,7 @@ from truefoundry.ml.autogen.client.models.update_run_request_dto import (
 from truefoundry.ml.autogen.client.models.update_run_response_dto import (
     UpdateRunResponseDto,
 )
+from truefoundry.ml.autogen.client.models.upload_model_source import UploadModelSource
 from truefoundry.ml.autogen.client.models.url import Url
 from truefoundry.ml.autogen.client.models.user_message import UserMessage
 from truefoundry.ml.autogen.client.models.validation_error import ValidationError

truefoundry/ml/autogen/client/models/artifact_dto.py

@@ -21,6 +21,7 @@ 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.pydantic_v1 import BaseModel, Field, StrictInt, StrictStr, conlist
 
 
@@ -36,6 +37,7 @@ class ArtifactDto(BaseModel):
     fqn: StrictStr = Field(...)
     description: Optional[StrictStr] = None
     created_by: Optional[StrictStr] = None
+    created_by_subject: Subject = Field(...)
     created_at: Optional[datetime] = None
     updated_at: Optional[datetime] = None
     artifact_storage_root: Optional[StrictStr] = None
@@ -49,6 +51,7 @@ class ArtifactDto(BaseModel):
         "fqn",
         "description",
         "created_by",
+        "created_by_subject",
         "created_at",
         "updated_at",
         "artifact_storage_root",
@@ -78,6 +81,9 @@ class ArtifactDto(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 created_by_subject
+        if self.created_by_subject:
+            _dict["created_by_subject"] = self.created_by_subject.to_dict()
         # override the default output from truefoundry.pydantic_v1 by calling `to_dict()` of latest_version
         if self.latest_version:
             _dict["latest_version"] = self.latest_version.to_dict()
@@ -101,6 +107,9 @@ class ArtifactDto(BaseModel):
                 "fqn": obj.get("fqn"),
                 "description": obj.get("description"),
                 "created_by": obj.get("created_by"),
+                "created_by_subject": Subject.from_dict(obj.get("created_by_subject"))
+                if obj.get("created_by_subject") is not None
+                else None,
                 "created_at": obj.get("created_at"),
                 "updated_at": obj.get("updated_at"),
                 "artifact_storage_root": obj.get("artifact_storage_root"),

truefoundry/ml/autogen/client/models/artifact_version_dto.py

@@ -19,10 +19,17 @@ 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 (
+    ArtifactVersionSerializationFormat,
+)
 from truefoundry.ml.autogen.client.models.artifact_version_status import (
     ArtifactVersionStatus,
 )
 from truefoundry.ml.autogen.client.models.internal_metadata import InternalMetadata
+from truefoundry.ml.autogen.client.models.model_version_manifest import (
+    ModelVersionManifest,
+)
+from truefoundry.ml.autogen.client.models.subject import Subject
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictInt, StrictStr
 
 
@@ -42,6 +49,7 @@ class ArtifactVersionDto(BaseModel):
     status: Optional[ArtifactVersionStatus] = None
     step: Optional[StrictInt] = 0
     created_by: Optional[StrictStr] = None
+    created_by_subject: Subject = Field(...)
     created_at: Optional[datetime] = None
     updated_at: Optional[datetime] = None
     artifact_fqn: Optional[StrictStr] = None
@@ -51,6 +59,8 @@ class ArtifactVersionDto(BaseModel):
     artifact_name: Optional[StrictStr] = None
     internal_metadata: Optional[InternalMetadata] = None
     artifact_size: Optional[StrictInt] = None
+    manifest: Optional[ModelVersionManifest] = None
+    serialization_format: ArtifactVersionSerializationFormat = Field(...)
     __properties = [
         "id",
         "artifact_id",
@@ -63,6 +73,7 @@ class ArtifactVersionDto(BaseModel):
         "status",
         "step",
         "created_by",
+        "created_by_subject",
         "created_at",
         "updated_at",
         "artifact_fqn",
@@ -72,6 +83,8 @@ class ArtifactVersionDto(BaseModel):
         "artifact_name",
         "internal_metadata",
         "artifact_size",
+        "manifest",
+        "serialization_format",
     ]
 
     class Config:
@@ -96,9 +109,15 @@ class ArtifactVersionDto(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 created_by_subject
+        if self.created_by_subject:
+            _dict["created_by_subject"] = self.created_by_subject.to_dict()
         # 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()
+        # 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
@@ -123,6 +142,9 @@ class ArtifactVersionDto(BaseModel):
                 "status": obj.get("status"),
                 "step": obj.get("step") if obj.get("step") is not None else 0,
                 "created_by": obj.get("created_by"),
+                "created_by_subject": Subject.from_dict(obj.get("created_by_subject"))
+                if obj.get("created_by_subject") is not None
+                else None,
                 "created_at": obj.get("created_at"),
                 "updated_at": obj.get("updated_at"),
                 "artifact_fqn": obj.get("artifact_fqn"),
@@ -136,6 +158,10 @@ class ArtifactVersionDto(BaseModel):
                 if obj.get("internal_metadata") is not None
                 else None,
                 "artifact_size": obj.get("artifact_size"),
+                "manifest": ModelVersionManifest.from_dict(obj.get("manifest"))
+                if obj.get("manifest") is not None
+                else None,
+                "serialization_format": obj.get("serialization_format"),
             }
         )
         return _obj

truefoundry/ml/autogen/client/models/artifact_version_serialization_format.py

@@ -0,0 +1,34 @@
+# coding: utf-8
+
+"""
+FastAPI
+
+No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+The version of the OpenAPI document: 0.1.0
+Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+Do not edit the class manually.
+"""  # noqa: E501
+
+import json
+import re  # noqa: F401
+
+from aenum import Enum
+
+
+class ArtifactVersionSerializationFormat(str, Enum):
+    """
+    An enumeration.
+    """
+
+    """
+    allowed enum values
+    """
+    V1 = "V1"
+    V2 = "V2"
+
+    @classmethod
+    def from_json(cls, json_str: str) -> ArtifactVersionSerializationFormat:
+        """Create an instance of ArtifactVersionSerializationFormat from a JSON string"""
+        return ArtifactVersionSerializationFormat(json.loads(json_str))

truefoundry/ml/autogen/client/models/create_artifact_version_response_dto.py

@@ -26,7 +26,8 @@ class CreateArtifactVersionResponseDto(BaseModel):
     """
 
     id: StrictStr = Field(...)
-    __properties = ["id"]
+    artifact_storage_root: StrictStr = Field(...)
+    __properties = ["id", "artifact_storage_root"]
 
     class Config:
         """Pydantic configuration"""
@@ -61,5 +62,10 @@ class CreateArtifactVersionResponseDto(BaseModel):
         if not isinstance(obj, dict):
             return CreateArtifactVersionResponseDto.parse_obj(obj)
 
-        _obj = CreateArtifactVersionResponseDto.parse_obj({"id": obj.get("id")})
+        _obj = CreateArtifactVersionResponseDto.parse_obj(
+            {
+                "id": obj.get("id"),
+                "artifact_storage_root": obj.get("artifact_storage_root"),
+            }
+        )
         return _obj

truefoundry/ml/autogen/client/models/create_run_request_dto.py

@@ -29,18 +29,10 @@ class CreateRunRequestDto(BaseModel):
 
     experiment_id: StrictStr = Field(...)
     name: StrictStr = Field(...)
-    user_id: Optional[StrictStr] = None
     start_time: Optional[StrictInt] = None
     tags: Optional[conlist(RunTagDto)] = None
     description: Optional[StrictStr] = None
-    __properties = [
-        "experiment_id",
-        "name",
-        "user_id",
-        "start_time",
-        "tags",
-        "description",
-    ]
+    __properties = ["experiment_id", "name", "start_time", "tags", "description"]
 
     class Config:
         """Pydantic configuration"""
@@ -86,7 +78,6 @@ class CreateRunRequestDto(BaseModel):
             {
                 "experiment_id": obj.get("experiment_id"),
                 "name": obj.get("name"),
-                "user_id": obj.get("user_id"),
                 "start_time": obj.get("start_time"),
                 "tags": [RunTagDto.from_dict(_item) for _item in obj.get("tags")]
                 if obj.get("tags") is not None