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

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

@@ -1,17 +1,18 @@
 # 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
@@ -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/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.
@@ -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/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.

truefoundry/ml/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
@@ -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/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.

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

@@ -2,31 +2,30 @@
 
 # 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.
 """  # noqa: E501
 
 # import models into model 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,
 )
@@ -57,9 +56,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
@@ -117,6 +116,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 (
@@ -202,7 +204,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
@@ -270,6 +271,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,
 )
@@ -280,6 +285,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,
@@ -349,6 +356,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/models/agent_app.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.
@@ -38,9 +38,17 @@ class AgentApp(BaseModel):
     """
 
     type: Optional[StrictStr] = "agent-app"
-    tools: conlist(AgentOpenAPIToolWithFQN) = Field(...)
-    agents: conlist(AgentWithFQN) = Field(...)
-    root_agent: constr(strict=True, min_length=1) = Field(...)
+    tools: conlist(AgentOpenAPIToolWithFQN) = Field(
+        default=..., description="+label=Tools +usage=Tools available to the Agent app"
+    )
+    agents: conlist(AgentWithFQN) = Field(
+        default=...,
+        description="+label=Agents +usage=Agents available to the Agent app",
+    )
+    root_agent: constr(strict=True, min_length=1) = Field(
+        default=...,
+        description="+label=Root Agent +usage=Root Agent for the app. This will be the first agent invoked",
+    )
     __properties = ["type", "tools", "agents", "root_agent"]
 
     @validator("type")