truefoundry 0.4.0rc1__py3-none-any.whl → 0.4.0rc3__py3-none-any.whl

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

@@ -20,9 +20,6 @@ from truefoundry.ml.autogen.client.exceptions import (  # noqa: F401
     ApiTypeError,
     ApiValueError,
 )
-from truefoundry.ml.autogen.client.models.backfill_default_storage_integration_id_request_dto import (
-    BackfillDefaultStorageIntegrationIdRequestDto,
-)
 from truefoundry.ml.autogen.client.models.create_experiment_request_dto import (
     CreateExperimentRequestDto,
 )
@@ -72,168 +69,6 @@ class ExperimentsApi:
             api_client = ApiClient.get_default()
         self.api_client = api_client
 
-    @validate_arguments
-    def backfill_storage_integration_post(
-        self,
-        backfill_default_storage_integration_id_request_dto: BackfillDefaultStorageIntegrationIdRequestDto,
-        **kwargs,
-    ) -> object:  # noqa: E501
-        """Backfill Storage Integration  # noqa: E501
-
-        Backfill all existing experiments to have a storage_integration_id  # noqa: E501
-        This method makes a synchronous HTTP request by default. To make an
-        asynchronous HTTP request, please pass async_req=True
-
-        >>> thread = api.backfill_storage_integration_post(backfill_default_storage_integration_id_request_dto, async_req=True)
-        >>> result = thread.get()
-
-        :param backfill_default_storage_integration_id_request_dto: (required)
-        :type backfill_default_storage_integration_id_request_dto: BackfillDefaultStorageIntegrationIdRequestDto
-        :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: object
-        """
-        kwargs["_return_http_data_only"] = True
-        if "_preload_content" in kwargs:
-            message = "Error! Please call the backfill_storage_integration_post_with_http_info method with `_preload_content` instead and obtain raw data from ApiResponse.raw_data"  # noqa: E501
-            raise ValueError(message)
-        return self.backfill_storage_integration_post_with_http_info(
-            backfill_default_storage_integration_id_request_dto, **kwargs
-        )  # noqa: E501
-
-    @validate_arguments
-    def backfill_storage_integration_post_with_http_info(
-        self,
-        backfill_default_storage_integration_id_request_dto: BackfillDefaultStorageIntegrationIdRequestDto,
-        **kwargs,
-    ) -> ApiResponse:  # noqa: E501
-        """Backfill Storage Integration  # noqa: E501
-
-        Backfill all existing experiments to have a storage_integration_id  # noqa: E501
-        This method makes a synchronous HTTP request by default. To make an
-        asynchronous HTTP request, please pass async_req=True
-
-        >>> thread = api.backfill_storage_integration_post_with_http_info(backfill_default_storage_integration_id_request_dto, async_req=True)
-        >>> result = thread.get()
-
-        :param backfill_default_storage_integration_id_request_dto: (required)
-        :type backfill_default_storage_integration_id_request_dto: BackfillDefaultStorageIntegrationIdRequestDto
-        :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(object, status_code(int), headers(HTTPHeaderDict))
-        """
-
-        _params = locals()
-
-        _all_params = ["backfill_default_storage_integration_id_request_dto"]
-        _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 backfill_storage_integration_post" % _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
-        if _params["backfill_default_storage_integration_id_request_dto"] is not None:
-            _body_params = _params[
-                "backfill_default_storage_integration_id_request_dto"
-            ]
-
-        # set the HTTP header `Accept`
-        _header_params["Accept"] = self.api_client.select_header_accept(
-            ["application/json"]
-        )  # noqa: E501
-
-        # set the HTTP header `Content-Type`
-        _content_types_list = _params.get(
-            "_content_type",
-            self.api_client.select_header_content_type(["application/json"]),
-        )
-        if _content_types_list:
-            _header_params["Content-Type"] = _content_types_list
-
-        # authentication setting
-        _auth_settings = ["HTTPBearer", "APIKeyCookie"]  # noqa: E501
-
-        _response_types_map = {
-            "200": "object",
-            "422": "HTTPValidationError",
-        }
-
-        return self.api_client.call_api(
-            "/api/2.0/mlflow/experiments/storage-integration/backfill",
-            "POST",
-            _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 create_experiment_post(
         self, create_experiment_request_dto: CreateExperimentRequestDto, **kwargs

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

@@ -45,9 +45,6 @@ from truefoundry.ml.autogen.client.models.authorize_user_for_model_request_dto i
 from truefoundry.ml.autogen.client.models.authorize_user_for_model_version_request_dto import (
     AuthorizeUserForModelVersionRequestDto,
 )
-from truefoundry.ml.autogen.client.models.backfill_default_storage_integration_id_request_dto import (
-    BackfillDefaultStorageIntegrationIdRequestDto,
-)
 from truefoundry.ml.autogen.client.models.blob_storage_reference import (
     BlobStorageReference,
 )

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

@@ -21,7 +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.pydantic_v1 import BaseModel, StrictInt, StrictStr, conlist
+from truefoundry.pydantic_v1 import BaseModel, Field, StrictInt, StrictStr, conlist
 
 
 class ArtifactDto(BaseModel):
@@ -29,11 +29,11 @@ class ArtifactDto(BaseModel):
     ArtifactDto
     """
 
-    id: Optional[StrictStr] = None
-    experiment_id: Optional[StrictStr] = None
-    type: Optional[ArtifactType] = None
-    name: Optional[StrictStr] = None
-    fqn: Optional[StrictStr] = None
+    id: StrictStr = Field(...)
+    experiment_id: StrictStr = Field(...)
+    type: ArtifactType = Field(...)
+    name: StrictStr = Field(...)
+    fqn: StrictStr = Field(...)
     description: Optional[StrictStr] = None
     created_by: Optional[StrictStr] = None
     created_at: Optional[datetime] = None

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

@@ -23,7 +23,7 @@ from truefoundry.ml.autogen.client.models.artifact_version_status import (
     ArtifactVersionStatus,
 )
 from truefoundry.ml.autogen.client.models.internal_metadata import InternalMetadata
-from truefoundry.pydantic_v1 import BaseModel, StrictInt, StrictStr
+from truefoundry.pydantic_v1 import BaseModel, Field, StrictInt, StrictStr
 
 
 class ArtifactVersionDto(BaseModel):
@@ -31,10 +31,11 @@ class ArtifactVersionDto(BaseModel):
     ArtifactVersionDto
     """
 
-    id: Optional[StrictStr] = None
-    artifact_id: Optional[StrictStr] = None
-    version: Optional[StrictInt] = None
-    fqn: Optional[StrictStr] = None
+    id: StrictStr = Field(...)
+    artifact_id: StrictStr = Field(...)
+    version: StrictInt = Field(...)
+    fqn: StrictStr = Field(...)
+    experiment_id: StrictStr = Field(...)
     artifact_storage_root: Optional[StrictStr] = None
     artifact_metadata: Optional[Dict[str, Any]] = None
     description: Optional[StrictStr] = None
@@ -46,7 +47,6 @@ class ArtifactVersionDto(BaseModel):
     artifact_fqn: Optional[StrictStr] = None
     data_path: Optional[StrictStr] = None
     usage_code_snippet: Optional[StrictStr] = None
-    experiment_id: Optional[StrictStr] = None
     run_id: Optional[StrictStr] = None
     artifact_name: Optional[StrictStr] = None
     internal_metadata: Optional[InternalMetadata] = None
@@ -56,6 +56,7 @@ class ArtifactVersionDto(BaseModel):
         "artifact_id",
         "version",
         "fqn",
+        "experiment_id",
         "artifact_storage_root",
         "artifact_metadata",
         "description",
@@ -67,7 +68,6 @@ class ArtifactVersionDto(BaseModel):
         "artifact_fqn",
         "data_path",
         "usage_code_snippet",
-        "experiment_id",
         "run_id",
         "artifact_name",
         "internal_metadata",
@@ -116,6 +116,7 @@ class ArtifactVersionDto(BaseModel):
                 "artifact_id": obj.get("artifact_id"),
                 "version": obj.get("version"),
                 "fqn": obj.get("fqn"),
+                "experiment_id": obj.get("experiment_id"),
                 "artifact_storage_root": obj.get("artifact_storage_root"),
                 "artifact_metadata": obj.get("artifact_metadata"),
                 "description": obj.get("description"),
@@ -127,7 +128,6 @@ class ArtifactVersionDto(BaseModel):
                 "artifact_fqn": obj.get("artifact_fqn"),
                 "data_path": obj.get("data_path"),
                 "usage_code_snippet": obj.get("usage_code_snippet"),
-                "experiment_id": obj.get("experiment_id"),
                 "run_id": obj.get("run_id"),
                 "artifact_name": obj.get("artifact_name"),
                 "internal_metadata": InternalMetadata.from_dict(

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

@@ -16,9 +16,8 @@ from __future__ import annotations
 import json
 import pprint
 import re  # noqa: F401
-from typing import Optional
 
-from truefoundry.pydantic_v1 import BaseModel, StrictStr
+from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr
 
 
 class CreateArtifactResponseDto(BaseModel):
@@ -26,7 +25,7 @@ class CreateArtifactResponseDto(BaseModel):
     CreateArtifactResponseDto
     """
 
-    id: Optional[StrictStr] = None
+    id: StrictStr = Field(...)
     __properties = ["id"]
 
     class Config:

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

@@ -16,9 +16,8 @@ from __future__ import annotations
 import json
 import pprint
 import re  # noqa: F401
-from typing import Optional
 
-from truefoundry.pydantic_v1 import BaseModel, StrictStr
+from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr
 
 
 class CreateArtifactVersionResponseDto(BaseModel):
@@ -26,7 +25,7 @@ class CreateArtifactVersionResponseDto(BaseModel):
     CreateArtifactVersionResponseDto
     """
 
-    id: Optional[StrictStr] = None
+    id: StrictStr = Field(...)
     __properties = ["id"]
 
     class Config:

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

@@ -18,7 +18,7 @@ import pprint
 import re  # noqa: F401
 from typing import Any, Dict, Optional
 
-from truefoundry.pydantic_v1 import BaseModel, StrictStr
+from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr
 
 
 class CreatePythonDeploymentConfigRequestDto(BaseModel):
@@ -26,7 +26,7 @@ class CreatePythonDeploymentConfigRequestDto(BaseModel):
     CreatePythonDeploymentConfigRequestDto
     """
 
-    workspace_fqn: Optional[StrictStr] = None
+    workspace_fqn: StrictStr = Field(...)
     deployment_config: Optional[Dict[str, Any]] = None
     __properties = ["workspace_fqn", "deployment_config"]
 

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

@@ -16,9 +16,8 @@ from __future__ import annotations
 import json
 import pprint
 import re  # noqa: F401
-from typing import Optional
 
-from truefoundry.pydantic_v1 import BaseModel, StrictStr
+from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr
 
 
 class CreatePythonDeploymentConfigResponseDto(BaseModel):
@@ -26,7 +25,7 @@ class CreatePythonDeploymentConfigResponseDto(BaseModel):
     CreatePythonDeploymentConfigResponseDto
     """
 
-    deploy_file_content: Optional[StrictStr] = None
+    deploy_file_content: StrictStr = Field(...)
     __properties = ["deploy_file_content"]
 
     class Config:

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

@@ -19,7 +19,7 @@ import re  # noqa: F401
 from typing import Optional
 
 from truefoundry.ml.autogen.client.models.run_tag_dto import RunTagDto
-from truefoundry.pydantic_v1 import BaseModel, StrictInt, StrictStr, conlist
+from truefoundry.pydantic_v1 import BaseModel, Field, StrictInt, StrictStr, conlist
 
 
 class CreateRunRequestDto(BaseModel):
@@ -27,18 +27,18 @@ class CreateRunRequestDto(BaseModel):
     CreateRunRequestDto
     """
 
-    experiment_id: Optional[StrictStr] = None
+    experiment_id: StrictStr = Field(...)
+    name: StrictStr = Field(...)
     user_id: Optional[StrictStr] = None
     start_time: Optional[StrictInt] = None
     tags: Optional[conlist(RunTagDto)] = None
-    name: Optional[StrictStr] = None
     description: Optional[StrictStr] = None
     __properties = [
         "experiment_id",
+        "name",
         "user_id",
         "start_time",
         "tags",
-        "name",
         "description",
     ]
 
@@ -85,12 +85,12 @@ class CreateRunRequestDto(BaseModel):
         _obj = CreateRunRequestDto.parse_obj(
             {
                 "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
                 else None,
-                "name": obj.get("name"),
                 "description": obj.get("description"),
             }
         )

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

@@ -16,10 +16,9 @@ from __future__ import annotations
 import json
 import pprint
 import re  # noqa: F401
-from typing import Optional
 
 from truefoundry.ml.autogen.client.models.run_dto import RunDto
-from truefoundry.pydantic_v1 import BaseModel
+from truefoundry.pydantic_v1 import BaseModel, Field
 
 
 class CreateRunResponseDto(BaseModel):
@@ -27,7 +26,7 @@ class CreateRunResponseDto(BaseModel):
     CreateRunResponseDto
     """
 
-    run: Optional[RunDto] = None
+    run: RunDto = Field(...)
     __properties = ["run"]
 
     class Config:

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

@@ -20,7 +20,7 @@ from datetime import datetime
 from typing import Any, Dict, Optional
 
 from truefoundry.ml.autogen.client.models.experiment_dto import ExperimentDto
-from truefoundry.pydantic_v1 import BaseModel, StrictStr
+from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr
 
 
 class DatasetDto(BaseModel):
@@ -28,11 +28,10 @@ class DatasetDto(BaseModel):
     DatasetDto
     """
 
-    id: Optional[StrictStr] = None
-    experiment_id: Optional[StrictStr] = None
-    experiment: Optional[ExperimentDto] = None
-    name: Optional[StrictStr] = None
-    fqn: Optional[StrictStr] = None
+    id: StrictStr = Field(...)
+    experiment_id: StrictStr = Field(...)
+    name: StrictStr = Field(...)
+    fqn: StrictStr = Field(...)
     description: Optional[StrictStr] = None
     created_by: Optional[StrictStr] = None
     created_at: Optional[datetime] = None
@@ -40,10 +39,10 @@ class DatasetDto(BaseModel):
     storage_root: Optional[StrictStr] = None
     dataset_metadata: Optional[Dict[str, Any]] = None
     usage_code_snippet: Optional[StrictStr] = None
+    experiment: Optional[ExperimentDto] = None
     __properties = [
         "id",
         "experiment_id",
-        "experiment",
         "name",
         "fqn",
         "description",
@@ -53,6 +52,7 @@ class DatasetDto(BaseModel):
         "storage_root",
         "dataset_metadata",
         "usage_code_snippet",
+        "experiment",
     ]
 
     class Config:
@@ -95,9 +95,6 @@ class DatasetDto(BaseModel):
             {
                 "id": obj.get("id"),
                 "experiment_id": obj.get("experiment_id"),
-                "experiment": ExperimentDto.from_dict(obj.get("experiment"))
-                if obj.get("experiment") is not None
-                else None,
                 "name": obj.get("name"),
                 "fqn": obj.get("fqn"),
                 "description": obj.get("description"),
@@ -107,6 +104,9 @@ class DatasetDto(BaseModel):
                 "storage_root": obj.get("storage_root"),
                 "dataset_metadata": obj.get("dataset_metadata"),
                 "usage_code_snippet": obj.get("usage_code_snippet"),
+                "experiment": ExperimentDto.from_dict(obj.get("experiment"))
+                if obj.get("experiment") is not None
+                else None,
             }
         )
         return _obj

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

@@ -19,7 +19,7 @@ import re  # noqa: F401
 from typing import Any, Dict, Optional
 
 from truefoundry.ml.autogen.client.models.experiment_tag_dto import ExperimentTagDto
-from truefoundry.pydantic_v1 import BaseModel, StrictInt, StrictStr, conlist
+from truefoundry.pydantic_v1 import BaseModel, Field, StrictInt, StrictStr, conlist
 
 
 class ExperimentDto(BaseModel):
@@ -27,39 +27,39 @@ class ExperimentDto(BaseModel):
     ExperimentDto
     """
 
-    experiment_id: Optional[StrictStr] = None
-    name: Optional[StrictStr] = None
+    experiment_id: StrictStr = Field(...)
+    name: StrictStr = Field(...)
+    storage_integration_id: Optional[StrictStr] = None
     artifact_location: Optional[StrictStr] = None
     lifecycle_stage: Optional[StrictStr] = None
     tags: Optional[conlist(ExperimentTagDto)] = None
     creator_user_id: Optional[StrictStr] = None
-    num_runs: Optional[StrictInt] = None
-    num_user_collaborators: Optional[StrictInt] = None
     description: Optional[StrictStr] = None
     privacy_type: Optional[StrictStr] = None
     created_at: Optional[StrictInt] = None
     tenant_name: Optional[StrictStr] = None
-    total_models_count: Optional[StrictInt] = None
-    total_artifacts_count: Optional[StrictInt] = None
-    storage_integration_id: Optional[StrictStr] = None
     manifest: Optional[Dict[str, Any]] = None
+    num_runs: Optional[StrictInt] = None
+    artifact_type_counts: Optional[Dict[str, StrictInt]] = None
+    datasets_count: Optional[StrictInt] = None
+    num_user_collaborators: Optional[StrictInt] = None
     __properties = [
         "experiment_id",
         "name",
+        "storage_integration_id",
         "artifact_location",
         "lifecycle_stage",
         "tags",
         "creator_user_id",
-        "num_runs",
-        "num_user_collaborators",
         "description",
         "privacy_type",
         "created_at",
         "tenant_name",
-        "total_models_count",
-        "total_artifacts_count",
-        "storage_integration_id",
         "manifest",
+        "num_runs",
+        "artifact_type_counts",
+        "datasets_count",
+        "num_user_collaborators",
     ]
 
     class Config:
@@ -106,22 +106,22 @@ class ExperimentDto(BaseModel):
             {
                 "experiment_id": obj.get("experiment_id"),
                 "name": obj.get("name"),
+                "storage_integration_id": obj.get("storage_integration_id"),
                 "artifact_location": obj.get("artifact_location"),
                 "lifecycle_stage": obj.get("lifecycle_stage"),
                 "tags": [ExperimentTagDto.from_dict(_item) for _item in obj.get("tags")]
                 if obj.get("tags") is not None
                 else None,
                 "creator_user_id": obj.get("creator_user_id"),
-                "num_runs": obj.get("num_runs"),
-                "num_user_collaborators": obj.get("num_user_collaborators"),
                 "description": obj.get("description"),
                 "privacy_type": obj.get("privacy_type"),
                 "created_at": obj.get("created_at"),
                 "tenant_name": obj.get("tenant_name"),
-                "total_models_count": obj.get("total_models_count"),
-                "total_artifacts_count": obj.get("total_artifacts_count"),
-                "storage_integration_id": obj.get("storage_integration_id"),
                 "manifest": obj.get("manifest"),
+                "num_runs": obj.get("num_runs"),
+                "artifact_type_counts": obj.get("artifact_type_counts"),
+                "datasets_count": obj.get("datasets_count"),
+                "num_user_collaborators": obj.get("num_user_collaborators"),
             }
         )
         return _obj

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

@@ -16,10 +16,9 @@ from __future__ import annotations
 import json
 import pprint
 import re  # noqa: F401
-from typing import Optional
 
 from truefoundry.ml.autogen.client.models.run_log_dto import RunLogDto
-from truefoundry.pydantic_v1 import BaseModel
+from truefoundry.pydantic_v1 import BaseModel, Field
 
 
 class GetLatestRunLogResponseDto(BaseModel):
@@ -27,7 +26,7 @@ class GetLatestRunLogResponseDto(BaseModel):
     GetLatestRunLogResponseDto
     """
 
-    run_log: Optional[RunLogDto] = None
+    run_log: RunLogDto = Field(...)
     __properties = ["run_log"]
 
     class Config:

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

@@ -16,9 +16,8 @@ from __future__ import annotations
 import json
 import pprint
 import re  # noqa: F401
-from typing import Optional
 
-from truefoundry.pydantic_v1 import BaseModel, StrictStr
+from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr
 
 
 class GetTenantIdResponseDto(BaseModel):
@@ -26,9 +25,9 @@ class GetTenantIdResponseDto(BaseModel):
     GetTenantIdResponseDto
     """
 
-    tenant_id: Optional[StrictStr] = None
-    tenant_name: Optional[StrictStr] = None
-    auth_server_url: Optional[StrictStr] = None
+    tenant_id: StrictStr = Field(...)
+    tenant_name: StrictStr = Field(...)
+    auth_server_url: StrictStr = Field(...)
     __properties = ["tenant_id", "tenant_name", "auth_server_url"]
 
     class Config:

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

@@ -22,7 +22,7 @@ from typing import Optional
 from truefoundry.ml.autogen.client.models.artifact_type import ArtifactType
 from truefoundry.ml.autogen.client.models.experiment_dto import ExperimentDto
 from truefoundry.ml.autogen.client.models.model_version_dto import ModelVersionDto
-from truefoundry.pydantic_v1 import BaseModel, StrictBool, StrictStr
+from truefoundry.pydantic_v1 import BaseModel, Field, StrictBool, StrictStr
 
 
 class ModelDto(BaseModel):
@@ -30,11 +30,11 @@ class ModelDto(BaseModel):
     ModelDto
     """
 
-    id: Optional[StrictStr] = None
-    experiment_id: Optional[StrictStr] = None
-    type: Optional[ArtifactType] = None
-    name: Optional[StrictStr] = None
-    fqn: Optional[StrictStr] = None
+    id: StrictStr = Field(...)
+    experiment_id: StrictStr = Field(...)
+    type: ArtifactType = Field(...)
+    name: StrictStr = Field(...)
+    fqn: StrictStr = Field(...)
     description: Optional[StrictStr] = None
     created_by: Optional[StrictStr] = None
     created_at: Optional[datetime] = None