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

truefoundry/ml/autogen/client/__init__.py

@@ -3,11 +3,11 @@
 # flake8: noqa
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.
@@ -42,20 +42,19 @@ from truefoundry.ml.autogen.client.exceptions import ApiAttributeError
 from truefoundry.ml.autogen.client.exceptions import ApiException
 
 # import models into sdk package
-from truefoundry.ml.autogen.client.models.add_custom_metrics_to_model_version_request_dto import (
-    AddCustomMetricsToModelVersionRequestDto,
-)
-from truefoundry.ml.autogen.client.models.add_features_to_model_version_request_dto import (
-    AddFeaturesToModelVersionRequestDto,
-)
-from truefoundry.ml.autogen.client.models.agent import Agent
 from truefoundry.ml.autogen.client.models.agent_app import AgentApp
-from truefoundry.ml.autogen.client.models.agent_open_api_tool import AgentOpenAPITool
+from truefoundry.ml.autogen.client.models.agent_manifest import AgentManifest
+from truefoundry.ml.autogen.client.models.agent_open_api_tool_manifest import (
+    AgentOpenAPIToolManifest,
+)
 from truefoundry.ml.autogen.client.models.agent_open_api_tool_with_fqn import (
     AgentOpenAPIToolWithFQN,
 )
 from truefoundry.ml.autogen.client.models.agent_with_fqn import AgentWithFQN
+from truefoundry.ml.autogen.client.models.apply_request_dto import ApplyRequestDto
+from truefoundry.ml.autogen.client.models.apply_response_dto import ApplyResponseDto
 from truefoundry.ml.autogen.client.models.artifact_dto import ArtifactDto
+from truefoundry.ml.autogen.client.models.artifact_path import ArtifactPath
 from truefoundry.ml.autogen.client.models.artifact_response_dto import (
     ArtifactResponseDto,
 )
@@ -86,9 +85,9 @@ from truefoundry.ml.autogen.client.models.blob_storage_reference import (
 from truefoundry.ml.autogen.client.models.body_get_search_runs_get import (
     BodyGetSearchRunsGet,
 )
-from truefoundry.ml.autogen.client.models.chat_prompt import ChatPrompt
-from truefoundry.ml.autogen.client.models.chat_prompt_messages_inner import (
-    ChatPromptMessagesInner,
+from truefoundry.ml.autogen.client.models.chat_prompt_manifest import ChatPromptManifest
+from truefoundry.ml.autogen.client.models.chat_prompt_manifest_messages_inner import (
+    ChatPromptManifestMessagesInner,
 )
 from truefoundry.ml.autogen.client.models.columns_dto import ColumnsDto
 from truefoundry.ml.autogen.client.models.command import Command
@@ -146,6 +145,9 @@ from truefoundry.ml.autogen.client.models.create_run_response_dto import (
 from truefoundry.ml.autogen.client.models.create_workflow_task_config_request_dto import (
     CreateWorkflowTaskConfigRequestDto,
 )
+from truefoundry.ml.autogen.client.models.data_directory_manifest import (
+    DataDirectoryManifest,
+)
 from truefoundry.ml.autogen.client.models.dataset_dto import DatasetDto
 from truefoundry.ml.autogen.client.models.dataset_response_dto import DatasetResponseDto
 from truefoundry.ml.autogen.client.models.delete_artifact_versions_request_dto import (
@@ -231,7 +233,6 @@ from truefoundry.ml.autogen.client.models.http_validation_error import (
 from truefoundry.ml.autogen.client.models.image_content_part import ImageContentPart
 from truefoundry.ml.autogen.client.models.image_url import ImageUrl
 from truefoundry.ml.autogen.client.models.infer_method_name import InferMethodName
-from truefoundry.ml.autogen.client.models.internal_metadata import InternalMetadata
 from truefoundry.ml.autogen.client.models.keras_framework import KerasFramework
 from truefoundry.ml.autogen.client.models.latest_run_log_dto import LatestRunLogDto
 from truefoundry.ml.autogen.client.models.library_name import LibraryName
@@ -299,6 +300,10 @@ 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.local_artifact_source import (
+    LocalArtifactSource,
+)
+from truefoundry.ml.autogen.client.models.local_model_source import LocalModelSource
 from truefoundry.ml.autogen.client.models.log_batch_request_dto import (
     LogBatchRequestDto,
 )
@@ -309,6 +314,8 @@ from truefoundry.ml.autogen.client.models.log_param_request_dto import (
     LogParamRequestDto,
 )
 from truefoundry.ml.autogen.client.models.manifest import Manifest
+from truefoundry.ml.autogen.client.models.manifest1 import Manifest1
+from truefoundry.ml.autogen.client.models.manifest2 import Manifest2
 from truefoundry.ml.autogen.client.models.method import Method
 from truefoundry.ml.autogen.client.models.metric_collection_dto import (
     MetricCollectionDto,
@@ -378,6 +385,7 @@ from truefoundry.ml.autogen.client.models.sklearn_serialization_format import (
 )
 from truefoundry.ml.autogen.client.models.source import Source
 from truefoundry.ml.autogen.client.models.source1 import Source1
+from truefoundry.ml.autogen.client.models.source2 import Source2
 from truefoundry.ml.autogen.client.models.spa_cy_framework import SpaCyFramework
 from truefoundry.ml.autogen.client.models.stats_models_framework import (
     StatsModelsFramework,

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.

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

@@ -1,11 +1,11 @@
 # coding: utf-8
 
 """
-FastAPI
+TrueFoundry ML API
 
 No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
-The version of the OpenAPI document: 0.1.0
+The version of the OpenAPI document: 1.0.0
 Generated by OpenAPI Generator (https://openapi-generator.tech)
 
 Do not edit the class manually.
@@ -21,21 +21,12 @@ from truefoundry.ml.autogen.client.exceptions import (  # noqa: F401
     ApiTypeError,
     ApiValueError,
 )
-from truefoundry.ml.autogen.client.models.add_custom_metrics_to_model_version_request_dto import (
-    AddCustomMetricsToModelVersionRequestDto,
-)
-from truefoundry.ml.autogen.client.models.add_features_to_model_version_request_dto import (
-    AddFeaturesToModelVersionRequestDto,
-)
 from truefoundry.ml.autogen.client.models.body_get_search_runs_get import (
     BodyGetSearchRunsGet,
 )
 from truefoundry.ml.autogen.client.models.get_signed_urls_for_dataset_write_response_dto import (
     GetSignedURLsForDatasetWriteResponseDto,
 )
-from truefoundry.ml.autogen.client.models.model_version_response_dto import (
-    ModelVersionResponseDto,
-)
 from truefoundry.ml.autogen.client.models.run_response_dto import RunResponseDto
 from truefoundry.ml.autogen.client.models.search_runs_response_dto import (
     SearchRunsResponseDto,
@@ -60,330 +51,6 @@ class DeprecatedApi:
             api_client = ApiClient.get_default()
         self.api_client = api_client
 
-    @validate_arguments
-    def add_custom_metrics_to_model_version_post(
-        self,
-        add_custom_metrics_to_model_version_request_dto: AddCustomMetricsToModelVersionRequestDto,
-        **kwargs,
-    ) -> ModelVersionResponseDto:  # noqa: E501
-        """(Deprecated) Add Custom Metrics To Model Version  # noqa: E501
-
-        This method makes a synchronous HTTP request by default. To make an
-        asynchronous HTTP request, please pass async_req=True
-
-        >>> thread = api.add_custom_metrics_to_model_version_post(add_custom_metrics_to_model_version_request_dto, async_req=True)
-        >>> result = thread.get()
-
-        :param add_custom_metrics_to_model_version_request_dto: (required)
-        :type add_custom_metrics_to_model_version_request_dto: AddCustomMetricsToModelVersionRequestDto
-        :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: ModelVersionResponseDto
-        """
-        kwargs["_return_http_data_only"] = True
-        if "_preload_content" in kwargs:
-            message = "Error! Please call the add_custom_metrics_to_model_version_post_with_http_info method with `_preload_content` instead and obtain raw data from ApiResponse.raw_data"  # noqa: E501
-            raise ValueError(message)
-        return self.add_custom_metrics_to_model_version_post_with_http_info(
-            add_custom_metrics_to_model_version_request_dto, **kwargs
-        )  # noqa: E501
-
-    @validate_arguments
-    def add_custom_metrics_to_model_version_post_with_http_info(
-        self,
-        add_custom_metrics_to_model_version_request_dto: AddCustomMetricsToModelVersionRequestDto,
-        **kwargs,
-    ) -> ApiResponse:  # noqa: E501
-        """(Deprecated) Add Custom Metrics To Model Version  # noqa: E501
-
-        This method makes a synchronous HTTP request by default. To make an
-        asynchronous HTTP request, please pass async_req=True
-
-        >>> thread = api.add_custom_metrics_to_model_version_post_with_http_info(add_custom_metrics_to_model_version_request_dto, async_req=True)
-        >>> result = thread.get()
-
-        :param add_custom_metrics_to_model_version_request_dto: (required)
-        :type add_custom_metrics_to_model_version_request_dto: AddCustomMetricsToModelVersionRequestDto
-        :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(ModelVersionResponseDto, status_code(int), headers(HTTPHeaderDict))
-        """
-
-        warnings.warn(
-            "POST /model-versions/custom-metrics/add is deprecated.", DeprecationWarning
-        )
-
-        _params = locals()
-
-        _all_params = ["add_custom_metrics_to_model_version_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 add_custom_metrics_to_model_version_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["add_custom_metrics_to_model_version_request_dto"] is not None:
-            _body_params = _params["add_custom_metrics_to_model_version_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": "ModelVersionResponseDto",
-            "422": "HTTPValidationError",
-        }
-
-        return self.api_client.call_api(
-            "/model-versions/custom-metrics/add",
-            "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 add_features_to_model_version_post(
-        self,
-        add_features_to_model_version_request_dto: AddFeaturesToModelVersionRequestDto,
-        **kwargs,
-    ) -> ModelVersionResponseDto:  # noqa: E501
-        """(Deprecated) Add Features To Model Version  # noqa: E501
-
-        This method makes a synchronous HTTP request by default. To make an
-        asynchronous HTTP request, please pass async_req=True
-
-        >>> thread = api.add_features_to_model_version_post(add_features_to_model_version_request_dto, async_req=True)
-        >>> result = thread.get()
-
-        :param add_features_to_model_version_request_dto: (required)
-        :type add_features_to_model_version_request_dto: AddFeaturesToModelVersionRequestDto
-        :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: ModelVersionResponseDto
-        """
-        kwargs["_return_http_data_only"] = True
-        if "_preload_content" in kwargs:
-            message = "Error! Please call the add_features_to_model_version_post_with_http_info method with `_preload_content` instead and obtain raw data from ApiResponse.raw_data"  # noqa: E501
-            raise ValueError(message)
-        return self.add_features_to_model_version_post_with_http_info(
-            add_features_to_model_version_request_dto, **kwargs
-        )  # noqa: E501
-
-    @validate_arguments
-    def add_features_to_model_version_post_with_http_info(
-        self,
-        add_features_to_model_version_request_dto: AddFeaturesToModelVersionRequestDto,
-        **kwargs,
-    ) -> ApiResponse:  # noqa: E501
-        """(Deprecated) Add Features To Model Version  # noqa: E501
-
-        This method makes a synchronous HTTP request by default. To make an
-        asynchronous HTTP request, please pass async_req=True
-
-        >>> thread = api.add_features_to_model_version_post_with_http_info(add_features_to_model_version_request_dto, async_req=True)
-        >>> result = thread.get()
-
-        :param add_features_to_model_version_request_dto: (required)
-        :type add_features_to_model_version_request_dto: AddFeaturesToModelVersionRequestDto
-        :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(ModelVersionResponseDto, status_code(int), headers(HTTPHeaderDict))
-        """
-
-        warnings.warn(
-            "POST /model-versions/features/add is deprecated.", DeprecationWarning
-        )
-
-        _params = locals()
-
-        _all_params = ["add_features_to_model_version_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 add_features_to_model_version_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["add_features_to_model_version_request_dto"] is not None:
-            _body_params = _params["add_features_to_model_version_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": "ModelVersionResponseDto",
-            "422": "HTTPValidationError",
-        }
-
-        return self.api_client.call_api(
-            "/model-versions/features/add",
-            "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 get_run_by_name_get(
         self,
@@ -532,7 +199,7 @@ class DeprecatedApi:
         )  # noqa: E501
 
         # authentication setting
-        _auth_settings = ["HTTPBearer", "APIKeyCookie"]  # noqa: E501
+        _auth_settings = ["APIKey"]  # noqa: E501
 
         _response_types_map = {
             "200": "RunResponseDto",
@@ -752,7 +419,7 @@ class DeprecatedApi:
             _header_params["Content-Type"] = _content_types_list
 
         # authentication setting
-        _auth_settings = ["HTTPBearer", "APIKeyCookie"]  # noqa: E501
+        _auth_settings = ["APIKey"]  # noqa: E501
 
         _response_types_map = {
             "200": "SearchRunsResponseDto",
@@ -911,7 +578,7 @@ class DeprecatedApi:
         )  # noqa: E501
 
         # authentication setting
-        _auth_settings = ["HTTPBearer", "APIKeyCookie"]  # noqa: E501
+        _auth_settings = ["APIKey"]  # noqa: E501
 
         _response_types_map = {
             "200": "GetSignedURLsForDatasetWriteResponseDto",