truefoundry 0.5.5rc1__py3-none-any.whl → 0.5.6__py3-none-any.whl

truefoundry/deploy/v2/lib/patched_models.py

@@ -256,6 +256,10 @@ class JwtAuthCreds(models.JwtAuthCreds, PatchedModelBase):
     type: Literal["jwt_auth"] = "jwt_auth"
 
 
+class TrueFoundryInteractiveLogin(models.TrueFoundryInteractiveLogin, PatchedModelBase):
+    type: Literal["truefoundry_oauth"] = "truefoundry_oauth"
+
+
 class HealthProbe(models.HealthProbe, PatchedModelBase):
     pass
 
@@ -500,10 +504,14 @@ class HuggingfaceArtifactSource(models.HuggingfaceArtifactSource, PatchedModelBa
     type: Literal["huggingface-hub"] = "huggingface-hub"
 
 
-class TruefoundryArtifactSource(models.TruefoundryArtifactSource, PatchedModelBase):
+class TrueFoundryArtifactSource(models.TrueFoundryArtifactSource, PatchedModelBase):
     type: Literal["truefoundry-artifact"] = "truefoundry-artifact"
 
 
+# Deprecated alias, kept for backward compatibility
+TruefoundryArtifactSource = TrueFoundryArtifactSource
+
+
 class ArtifactsDownload(models.ArtifactsDownload, PatchedModelBase):
     pass
 

truefoundry/ml/autogen/client/__init__.py

@@ -175,8 +175,8 @@ from truefoundry.ml.autogen.client.models.experiment_tag_dto import ExperimentTa
 from truefoundry.ml.autogen.client.models.export_deployment_files_request_dto import (
     ExportDeploymentFilesRequestDto,
 )
-from truefoundry.ml.autogen.client.models.external_artifact_source import (
-    ExternalArtifactSource,
+from truefoundry.ml.autogen.client.models.external_blob_storage_source import (
+    ExternalBlobStorageSource,
 )
 from truefoundry.ml.autogen.client.models.fast_ai_framework import FastAIFramework
 from truefoundry.ml.autogen.client.models.file_info_dto import FileInfoDto
@@ -403,8 +403,8 @@ from truefoundry.ml.autogen.client.models.trigger_job_run_config_request_dto imp
 from truefoundry.ml.autogen.client.models.trigger_job_run_config_response_dto import (
     TriggerJobRunConfigResponseDto,
 )
-from truefoundry.ml.autogen.client.models.true_foundry_artifact_source import (
-    TrueFoundryArtifactSource,
+from truefoundry.ml.autogen.client.models.true_foundry_managed_source import (
+    TrueFoundryManagedSource,
 )
 from truefoundry.ml.autogen.client.models.update_artifact_version_request_dto import (
     UpdateArtifactVersionRequestDto,

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

@@ -1182,162 +1182,6 @@ class ExperimentsApi:
             _request_auth=_params.get("_request_auth"),
         )
 
-    @validate_arguments
-    def put_privacy_type_put(
-        self, set_experiment_tag_request_dto: SetExperimentTagRequestDto, **kwargs
-    ) -> object:  # noqa: E501
-        """Put Privacy Type  # noqa: E501
-
-        Set the privacy type of an expermient to Private or Public(publically readable)  # noqa: E501
-        This method makes a synchronous HTTP request by default. To make an
-        asynchronous HTTP request, please pass async_req=True
-
-        >>> thread = api.put_privacy_type_put(set_experiment_tag_request_dto, async_req=True)
-        >>> result = thread.get()
-
-        :param set_experiment_tag_request_dto: (required)
-        :type set_experiment_tag_request_dto: SetExperimentTagRequestDto
-        :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 put_privacy_type_put_with_http_info method with `_preload_content` instead and obtain raw data from ApiResponse.raw_data"  # noqa: E501
-            raise ValueError(message)
-        return self.put_privacy_type_put_with_http_info(
-            set_experiment_tag_request_dto, **kwargs
-        )  # noqa: E501
-
-    @validate_arguments
-    def put_privacy_type_put_with_http_info(
-        self, set_experiment_tag_request_dto: SetExperimentTagRequestDto, **kwargs
-    ) -> ApiResponse:  # noqa: E501
-        """Put Privacy Type  # noqa: E501
-
-        Set the privacy type of an expermient to Private or Public(publically readable)  # noqa: E501
-        This method makes a synchronous HTTP request by default. To make an
-        asynchronous HTTP request, please pass async_req=True
-
-        >>> thread = api.put_privacy_type_put_with_http_info(set_experiment_tag_request_dto, async_req=True)
-        >>> result = thread.get()
-
-        :param set_experiment_tag_request_dto: (required)
-        :type set_experiment_tag_request_dto: SetExperimentTagRequestDto
-        :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 = ["set_experiment_tag_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 put_privacy_type_put" % _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["set_experiment_tag_request_dto"] is not None:
-            _body_params = _params["set_experiment_tag_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/privacy-type",
-            "PUT",
-            _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 restore_experiment_post(
         self, experiment_id_request_dto: ExperimentIdRequestDto, **kwargs