truefoundry 0.5.10rc1__py3-none-any.whl → 0.5.11rc1__py3-none-any.whl

truefoundry/ml/{autogen → _autogen}/client/api/run_artifacts_api.py

@@ -1,26 +1,27 @@
 # 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.
 """  # noqa: E501
 
 import re  # noqa: F401
+import warnings
 from typing import Optional
 
-from truefoundry.ml.autogen.client.api_client import ApiClient
-from truefoundry.ml.autogen.client.api_response import ApiResponse
-from truefoundry.ml.autogen.client.exceptions import (  # noqa: F401
+from truefoundry.ml._autogen.client.api_client import ApiClient
+from truefoundry.ml._autogen.client.api_response import ApiResponse
+from truefoundry.ml._autogen.client.exceptions import (  # noqa: F401
     ApiTypeError,
     ApiValueError,
 )
-from truefoundry.ml.autogen.client.models.list_run_artifacts_response_dto import (
+from truefoundry.ml._autogen.client.models.list_run_artifacts_response_dto import (
     ListRunArtifactsResponseDto,
 )
 from truefoundry.pydantic_v1 import (
@@ -52,7 +53,7 @@ class RunArtifactsApi:
         max_results: Optional[StrictInt] = None,
         **kwargs,
     ) -> ListRunArtifactsResponseDto:  # noqa: E501
-        """List Run Artifacts  # noqa: E501
+        """(Deprecated) List Run Artifacts  # noqa: E501
 
         List legacy run level artifacts (files and directories)  # noqa: E501
         This method makes a synchronous HTTP request by default. To make an
@@ -100,7 +101,7 @@ class RunArtifactsApi:
         max_results: Optional[StrictInt] = None,
         **kwargs,
     ) -> ApiResponse:  # noqa: E501
-        """List Run Artifacts  # noqa: E501
+        """(Deprecated) List Run Artifacts  # noqa: E501
 
         List legacy run level artifacts (files and directories)  # noqa: E501
         This method makes a synchronous HTTP request by default. To make an
@@ -144,6 +145,10 @@ class RunArtifactsApi:
         :rtype: tuple(ListRunArtifactsResponseDto, status_code(int), headers(HTTPHeaderDict))
         """
 
+        warnings.warn(
+            "GET /api/2.0/mlflow/artifacts/list is deprecated.", DeprecationWarning
+        )
+
         _params = locals()
 
         _all_params = ["run_id", "run_uuid", "path", "page_token", "max_results"]
@@ -204,7 +209,7 @@ class RunArtifactsApi:
         )  # noqa: E501
 
         # authentication setting
-        _auth_settings = ["HTTPBearer", "APIKeyCookie"]  # noqa: E501
+        _auth_settings = ["APIKey"]  # noqa: E501
 
         _response_types_map = {
             "200": "ListRunArtifactsResponseDto",

truefoundry/ml/{autogen → _autogen}/client/api/runs_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.
@@ -14,58 +14,58 @@ Do not edit the class manually.
 import re  # noqa: F401
 from typing import Optional
 
-from truefoundry.ml.autogen.client.api_client import ApiClient
-from truefoundry.ml.autogen.client.api_response import ApiResponse
-from truefoundry.ml.autogen.client.exceptions import (  # noqa: F401
+from truefoundry.ml._autogen.client.api_client import ApiClient
+from truefoundry.ml._autogen.client.api_response import ApiResponse
+from truefoundry.ml._autogen.client.exceptions import (  # noqa: F401
     ApiTypeError,
     ApiValueError,
 )
-from truefoundry.ml.autogen.client.models.create_run_request_dto import (
+from truefoundry.ml._autogen.client.models.create_run_request_dto import (
     CreateRunRequestDto,
 )
-from truefoundry.ml.autogen.client.models.create_run_response_dto import (
+from truefoundry.ml._autogen.client.models.create_run_response_dto import (
     CreateRunResponseDto,
 )
-from truefoundry.ml.autogen.client.models.delete_run_request import DeleteRunRequest
-from truefoundry.ml.autogen.client.models.delete_tag_request_dto import (
+from truefoundry.ml._autogen.client.models.delete_run_request import DeleteRunRequest
+from truefoundry.ml._autogen.client.models.delete_tag_request_dto import (
     DeleteTagRequestDto,
 )
-from truefoundry.ml.autogen.client.models.get_latest_run_log_response_dto import (
+from truefoundry.ml._autogen.client.models.get_latest_run_log_response_dto import (
     GetLatestRunLogResponseDto,
 )
-from truefoundry.ml.autogen.client.models.list_latest_run_logs_response_dto import (
+from truefoundry.ml._autogen.client.models.list_latest_run_logs_response_dto import (
     ListLatestRunLogsResponseDto,
 )
-from truefoundry.ml.autogen.client.models.list_run_logs_response_dto import (
+from truefoundry.ml._autogen.client.models.list_run_logs_response_dto import (
     ListRunLogsResponseDto,
 )
-from truefoundry.ml.autogen.client.models.log_batch_request_dto import (
+from truefoundry.ml._autogen.client.models.log_batch_request_dto import (
     LogBatchRequestDto,
 )
-from truefoundry.ml.autogen.client.models.log_metric_request_dto import (
+from truefoundry.ml._autogen.client.models.log_metric_request_dto import (
     LogMetricRequestDto,
 )
-from truefoundry.ml.autogen.client.models.log_param_request_dto import (
+from truefoundry.ml._autogen.client.models.log_param_request_dto import (
     LogParamRequestDto,
 )
-from truefoundry.ml.autogen.client.models.restore_run_request_dto import (
+from truefoundry.ml._autogen.client.models.restore_run_request_dto import (
     RestoreRunRequestDto,
 )
-from truefoundry.ml.autogen.client.models.run_response_dto import RunResponseDto
-from truefoundry.ml.autogen.client.models.search_runs_request_dto import (
+from truefoundry.ml._autogen.client.models.run_response_dto import RunResponseDto
+from truefoundry.ml._autogen.client.models.search_runs_request_dto import (
     SearchRunsRequestDto,
 )
-from truefoundry.ml.autogen.client.models.search_runs_response_dto import (
+from truefoundry.ml._autogen.client.models.search_runs_response_dto import (
     SearchRunsResponseDto,
 )
-from truefoundry.ml.autogen.client.models.set_tag_request_dto import SetTagRequestDto
-from truefoundry.ml.autogen.client.models.store_run_logs_request_dto import (
+from truefoundry.ml._autogen.client.models.set_tag_request_dto import SetTagRequestDto
+from truefoundry.ml._autogen.client.models.store_run_logs_request_dto import (
     StoreRunLogsRequestDto,
 )
-from truefoundry.ml.autogen.client.models.update_run_request_dto import (
+from truefoundry.ml._autogen.client.models.update_run_request_dto import (
     UpdateRunRequestDto,
 )
-from truefoundry.ml.autogen.client.models.update_run_response_dto import (
+from truefoundry.ml._autogen.client.models.update_run_response_dto import (
     UpdateRunResponseDto,
 )
 from truefoundry.pydantic_v1 import (
@@ -216,7 +216,7 @@ class RunsApi:
             _header_params["Content-Type"] = _content_types_list
 
         # authentication setting
-        _auth_settings = ["HTTPBearer", "APIKeyCookie"]  # noqa: E501
+        _auth_settings = ["APIKey"]  # noqa: E501
 
         _response_types_map = {
             "200": "CreateRunResponseDto",
@@ -368,7 +368,7 @@ class RunsApi:
             _header_params["Content-Type"] = _content_types_list
 
         # authentication setting
-        _auth_settings = ["HTTPBearer", "APIKeyCookie"]  # noqa: E501
+        _auth_settings = ["APIKey"]  # noqa: E501
 
         _response_types_map = {
             "200": "object",
@@ -522,7 +522,7 @@ class RunsApi:
             _header_params["Content-Type"] = _content_types_list
 
         # authentication setting
-        _auth_settings = ["HTTPBearer", "APIKeyCookie"]  # noqa: E501
+        _auth_settings = ["APIKey"]  # noqa: E501
 
         _response_types_map = {
             "200": "object",
@@ -666,7 +666,7 @@ class RunsApi:
         )  # noqa: E501
 
         # authentication setting
-        _auth_settings = ["HTTPBearer", "APIKeyCookie"]  # noqa: E501
+        _auth_settings = ["APIKey"]  # noqa: E501
 
         _response_types_map = {
             "200": "RunResponseDto",
@@ -836,7 +836,7 @@ class RunsApi:
         )  # noqa: E501
 
         # authentication setting
-        _auth_settings = ["HTTPBearer", "APIKeyCookie"]  # noqa: E501
+        _auth_settings = ["APIKey"]  # noqa: E501
 
         _response_types_map = {
             "200": "RunResponseDto",
@@ -995,7 +995,7 @@ class RunsApi:
         )  # noqa: E501
 
         # authentication setting
-        _auth_settings = ["HTTPBearer", "APIKeyCookie"]  # noqa: E501
+        _auth_settings = ["APIKey"]  # noqa: E501
 
         _response_types_map = {
             "200": "RunResponseDto",
@@ -1149,7 +1149,7 @@ class RunsApi:
             _header_params["Content-Type"] = _content_types_list
 
         # authentication setting
-        _auth_settings = ["HTTPBearer", "APIKeyCookie"]  # noqa: E501
+        _auth_settings = ["APIKey"]  # noqa: E501
 
         _response_types_map = {
             "200": "object",
@@ -1309,7 +1309,7 @@ class RunsApi:
         )  # noqa: E501
 
         # authentication setting
-        _auth_settings = ["HTTPBearer", "APIKeyCookie"]  # noqa: E501
+        _auth_settings = ["APIKey"]  # noqa: E501
 
         _response_types_map = {
             "200": "GetLatestRunLogResponseDto",
@@ -1477,7 +1477,7 @@ class RunsApi:
         )  # noqa: E501
 
         # authentication setting
-        _auth_settings = ["HTTPBearer", "APIKeyCookie"]  # noqa: E501
+        _auth_settings = ["APIKey"]  # noqa: E501
 
         _response_types_map = {
             "200": "ListLatestRunLogsResponseDto",
@@ -1664,7 +1664,7 @@ class RunsApi:
             _header_params["Content-Type"] = _content_types_list
 
         # authentication setting
-        _auth_settings = ["HTTPBearer", "APIKeyCookie"]  # noqa: E501
+        _auth_settings = ["APIKey"]  # noqa: E501
 
         _response_types_map = {
             "200": "ListRunLogsResponseDto",
@@ -1818,7 +1818,7 @@ class RunsApi:
             _header_params["Content-Type"] = _content_types_list
 
         # authentication setting
-        _auth_settings = ["HTTPBearer", "APIKeyCookie"]  # noqa: E501
+        _auth_settings = ["APIKey"]  # noqa: E501
 
         _response_types_map = {
             "200": "object",
@@ -1972,7 +1972,7 @@ class RunsApi:
             _header_params["Content-Type"] = _content_types_list
 
         # authentication setting
-        _auth_settings = ["HTTPBearer", "APIKeyCookie"]  # noqa: E501
+        _auth_settings = ["APIKey"]  # noqa: E501
 
         _response_types_map = {
             "200": "object",
@@ -2126,7 +2126,7 @@ class RunsApi:
             _header_params["Content-Type"] = _content_types_list
 
         # authentication setting
-        _auth_settings = ["HTTPBearer", "APIKeyCookie"]  # noqa: E501
+        _auth_settings = ["APIKey"]  # noqa: E501
 
         _response_types_map = {
             "200": "object",
@@ -2278,7 +2278,7 @@ class RunsApi:
             _header_params["Content-Type"] = _content_types_list
 
         # authentication setting
-        _auth_settings = ["HTTPBearer", "APIKeyCookie"]  # noqa: E501
+        _auth_settings = ["APIKey"]  # noqa: E501
 
         _response_types_map = {
             "200": "object",
@@ -2432,7 +2432,7 @@ class RunsApi:
             _header_params["Content-Type"] = _content_types_list
 
         # authentication setting
-        _auth_settings = ["HTTPBearer", "APIKeyCookie"]  # noqa: E501
+        _auth_settings = ["APIKey"]  # noqa: E501
 
         _response_types_map = {
             "200": "object",
@@ -2584,7 +2584,7 @@ class RunsApi:
             _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",
@@ -2738,7 +2738,7 @@ class RunsApi:
             _header_params["Content-Type"] = _content_types_list
 
         # authentication setting
-        _auth_settings = ["HTTPBearer", "APIKeyCookie"]  # noqa: E501
+        _auth_settings = ["APIKey"]  # noqa: E501
 
         _response_types_map = {
             "200": "object",
@@ -2892,7 +2892,7 @@ class RunsApi:
             _header_params["Content-Type"] = _content_types_list
 
         # authentication setting
-        _auth_settings = ["HTTPBearer", "APIKeyCookie"]  # noqa: E501
+        _auth_settings = ["APIKey"]  # noqa: E501
 
         _response_types_map = {
             "200": "UpdateRunResponseDto",

truefoundry/ml/{autogen → _autogen}/client/api_client.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.
@@ -23,11 +23,11 @@ from urllib.parse import quote
 
 from dateutil.parser import parse
 
-import truefoundry.ml.autogen.client.models
-from truefoundry.ml.autogen.client import rest
-from truefoundry.ml.autogen.client.api_response import ApiResponse
-from truefoundry.ml.autogen.client.configuration import Configuration
-from truefoundry.ml.autogen.client.exceptions import ApiException, ApiValueError
+import truefoundry.ml._autogen.client.models
+from truefoundry.ml._autogen.client import rest
+from truefoundry.ml._autogen.client.api_response import ApiResponse
+from truefoundry.ml._autogen.client.configuration import Configuration
+from truefoundry.ml._autogen.client.exceptions import ApiException, ApiValueError
 from truefoundry.pydantic_v1 import SecretStr
 
 
@@ -376,7 +376,7 @@ class ApiClient:
             if klass in self.NATIVE_TYPES_MAPPING:
                 klass = self.NATIVE_TYPES_MAPPING[klass]
             else:
-                klass = getattr(truefoundry.ml.autogen.client.models, klass)
+                klass = getattr(truefoundry.ml._autogen.client.models, klass)
 
         if klass in self.PRIMITIVE_TYPES:
             return self.__deserialize_primitive(data, klass)

truefoundry/ml/{autogen → _autogen}/client/configuration.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.
@@ -36,50 +36,31 @@ JSON_SCHEMA_VALIDATION_KEYWORDS = {
 class Configuration:
     """This class contains various settings of the API client.
 
-        :param host: Base url.
-        :param api_key: Dict to store API key(s).
-          Each entry in the dict specifies an API key.
-          The dict key is the name of the security scheme in the OAS specification.
-          The dict value is the API key secret.
-        :param api_key_prefix: Dict to store API prefix (e.g. Bearer).
-          The dict key is the name of the security scheme in the OAS specification.
-          The dict value is an API key prefix when generating the auth data.
-        :param username: Username for HTTP basic authentication.
-        :param password: Password for HTTP basic authentication.
-        :param access_token: Access token.
-        :param server_index: Index to servers configuration.
-        :param server_variables: Mapping with string values to replace variables in
-          templated server configuration. The validation of enums is performed for
-          variables with defined enum values before.
-        :param server_operation_index: Mapping from operation ID to an index to server
-          configuration.
-        :param server_operation_variables: Mapping from operation ID to a mapping with
-          string values to replace variables in templated server configuration.
-          The validation of enums is performed for variables with defined enum
-          values before.
-        :param ssl_ca_cert: str - the path to a file of concatenated CA certificates
-          in PEM format.
-
-        :Example:
-
-        API Key Authentication Example.
-        Given the following security scheme in the OpenAPI specification:
-          components:
-            securitySchemes:
-              cookieAuth:         # name for the security scheme
-                type: apiKey
-                in: cookie
-                name: JSESSIONID  # cookie name
-
-        You can programmatically set the cookie:
-
-    conf = truefoundry.ml.autogen.client.Configuration(
-        api_key={'cookieAuth': 'abc123'}
-        api_key_prefix={'cookieAuth': 'JSESSIONID'}
-    )
-
-        The following cookie will be added to the HTTP request:
-           Cookie: JSESSIONID abc123
+    :param host: Base url.
+    :param api_key: Dict to store API key(s).
+      Each entry in the dict specifies an API key.
+      The dict key is the name of the security scheme in the OAS specification.
+      The dict value is the API key secret.
+    :param api_key_prefix: Dict to store API prefix (e.g. Bearer).
+      The dict key is the name of the security scheme in the OAS specification.
+      The dict value is an API key prefix when generating the auth data.
+    :param username: Username for HTTP basic authentication.
+    :param password: Password for HTTP basic authentication.
+    :param access_token: Access token.
+    :param server_index: Index to servers configuration.
+    :param server_variables: Mapping with string values to replace variables in
+      templated server configuration. The validation of enums is performed for
+      variables with defined enum values before.
+    :param server_operation_index: Mapping from operation ID to an index to server
+      configuration.
+    :param server_operation_variables: Mapping from operation ID to a mapping with
+      string values to replace variables in templated server configuration.
+      The validation of enums is performed for variables with defined enum
+      values before.
+    :param ssl_ca_cert: str - the path to a file of concatenated CA certificates
+      in PEM format.
+
+    :Example:
     """
 
     _default = None
@@ -140,7 +121,7 @@ class Configuration:
         """Logging Settings
         """
         self.logger["package_logger"] = logging.getLogger(
-            "truefoundry.ml.autogen.client"
+            "truefoundry.ml._autogen.client"
         )
         self.logger["urllib3_logger"] = logging.getLogger("urllib3")
         self.logger_format = "%(asctime)s %(levelname)s %(message)s"
@@ -394,21 +375,12 @@ class Configuration:
         """
         auth = {}
         if self.access_token is not None:
-            auth["HTTPBearer"] = {
+            auth["APIKey"] = {
                 "type": "bearer",
                 "in": "header",
                 "key": "Authorization",
                 "value": "Bearer " + self.access_token,
             }
-        if "APIKeyCookie" in self.api_key:
-            auth["APIKeyCookie"] = {
-                "type": "api_key",
-                "in": "cookie",
-                "key": "accessToken",
-                "value": self.get_api_key_with_prefix(
-                    "APIKeyCookie",
-                ),
-            }
         return auth
 
     def to_debug_report(self):
@@ -420,7 +392,7 @@ class Configuration:
             "Python SDK Debug Report:\n"
             "OS: {env}\n"
             "Python Version: {pyversion}\n"
-            "Version of the API: 0.1.0\n"
+            "Version of the API: 1.0.0\n"
             "SDK Package Version: 0.1.0".format(env=sys.platform, pyversion=sys.version)
         )
 

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