truefoundry 0.4.6__py3-none-any.whl → 0.5.0rc2__py3-none-any.whl

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

@@ -0,0 +1,526 @@
+# coding: utf-8
+
+"""
+FastAPI
+
+No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+The version of the OpenAPI document: 0.1.0
+Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+Do not edit the class manually.
+"""  # noqa: E501
+
+import re  # 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_python_deployment_config_request_dto import (
+    CreatePythonDeploymentConfigRequestDto,
+)
+from truefoundry.ml.autogen.client.models.create_python_deployment_config_response_dto import (
+    CreatePythonDeploymentConfigResponseDto,
+)
+from truefoundry.ml.autogen.client.models.create_workflow_task_config_request_dto import (
+    CreateWorkflowTaskConfigRequestDto,
+)
+from truefoundry.ml.autogen.client.models.trigger_job_run_config_request_dto import (
+    TriggerJobRunConfigRequestDto,
+)
+from truefoundry.ml.autogen.client.models.trigger_job_run_config_response_dto import (
+    TriggerJobRunConfigResponseDto,
+)
+from truefoundry.pydantic_v1 import validate_arguments
+
+
+class GenerateCodeSnippetApi:
+    """NOTE: This class is auto generated by OpenAPI Generator
+    Ref: https://openapi-generator.tech
+
+    Do not edit the class manually.
+    """
+
+    def __init__(self, api_client=None) -> None:
+        if api_client is None:
+            api_client = ApiClient.get_default()
+        self.api_client = api_client
+
+    @validate_arguments
+    def generate_py_development_config_post(
+        self,
+        create_python_deployment_config_request_dto: CreatePythonDeploymentConfigRequestDto,
+        **kwargs,
+    ) -> CreatePythonDeploymentConfigResponseDto:  # noqa: E501
+        """Generate Py Development Config  # noqa: E501
+
+        Generate Python Deployment Config  # noqa: E501
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+
+        >>> thread = api.generate_py_development_config_post(create_python_deployment_config_request_dto, async_req=True)
+        >>> result = thread.get()
+
+        :param create_python_deployment_config_request_dto: (required)
+        :type create_python_deployment_config_request_dto: CreatePythonDeploymentConfigRequestDto
+        :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: CreatePythonDeploymentConfigResponseDto
+        """
+        kwargs["_return_http_data_only"] = True
+        if "_preload_content" in kwargs:
+            message = "Error! Please call the generate_py_development_config_post_with_http_info method with `_preload_content` instead and obtain raw data from ApiResponse.raw_data"  # noqa: E501
+            raise ValueError(message)
+        return self.generate_py_development_config_post_with_http_info(
+            create_python_deployment_config_request_dto, **kwargs
+        )  # noqa: E501
+
+    @validate_arguments
+    def generate_py_development_config_post_with_http_info(
+        self,
+        create_python_deployment_config_request_dto: CreatePythonDeploymentConfigRequestDto,
+        **kwargs,
+    ) -> ApiResponse:  # noqa: E501
+        """Generate Py Development Config  # noqa: E501
+
+        Generate Python Deployment Config  # noqa: E501
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+
+        >>> thread = api.generate_py_development_config_post_with_http_info(create_python_deployment_config_request_dto, async_req=True)
+        >>> result = thread.get()
+
+        :param create_python_deployment_config_request_dto: (required)
+        :type create_python_deployment_config_request_dto: CreatePythonDeploymentConfigRequestDto
+        :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(CreatePythonDeploymentConfigResponseDto, status_code(int), headers(HTTPHeaderDict))
+        """
+
+        _params = locals()
+
+        _all_params = ["create_python_deployment_config_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 generate_py_development_config_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["create_python_deployment_config_request_dto"] is not None:
+            _body_params = _params["create_python_deployment_config_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 = []  # noqa: E501
+
+        _response_types_map = {
+            "200": "CreatePythonDeploymentConfigResponseDto",
+            "422": "HTTPValidationError",
+        }
+
+        return self.api_client.call_api(
+            "/api/2.0/mlflow/generate-code-snippet/application",
+            "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 trigger_job_run_post(
+        self,
+        trigger_job_run_config_request_dto: TriggerJobRunConfigRequestDto,
+        **kwargs,
+    ) -> TriggerJobRunConfigResponseDto:  # noqa: E501
+        """Trigger Job Run  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+
+        >>> thread = api.trigger_job_run_post(trigger_job_run_config_request_dto, async_req=True)
+        >>> result = thread.get()
+
+        :param trigger_job_run_config_request_dto: (required)
+        :type trigger_job_run_config_request_dto: TriggerJobRunConfigRequestDto
+        :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: TriggerJobRunConfigResponseDto
+        """
+        kwargs["_return_http_data_only"] = True
+        if "_preload_content" in kwargs:
+            message = "Error! Please call the trigger_job_run_post_with_http_info method with `_preload_content` instead and obtain raw data from ApiResponse.raw_data"  # noqa: E501
+            raise ValueError(message)
+        return self.trigger_job_run_post_with_http_info(
+            trigger_job_run_config_request_dto, **kwargs
+        )  # noqa: E501
+
+    @validate_arguments
+    def trigger_job_run_post_with_http_info(
+        self,
+        trigger_job_run_config_request_dto: TriggerJobRunConfigRequestDto,
+        **kwargs,
+    ) -> ApiResponse:  # noqa: E501
+        """Trigger Job Run  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+
+        >>> thread = api.trigger_job_run_post_with_http_info(trigger_job_run_config_request_dto, async_req=True)
+        >>> result = thread.get()
+
+        :param trigger_job_run_config_request_dto: (required)
+        :type trigger_job_run_config_request_dto: TriggerJobRunConfigRequestDto
+        :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(TriggerJobRunConfigResponseDto, status_code(int), headers(HTTPHeaderDict))
+        """
+
+        _params = locals()
+
+        _all_params = ["trigger_job_run_config_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 trigger_job_run_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["trigger_job_run_config_request_dto"] is not None:
+            _body_params = _params["trigger_job_run_config_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 = []  # noqa: E501
+
+        _response_types_map = {
+            "200": "TriggerJobRunConfigResponseDto",
+            "422": "HTTPValidationError",
+        }
+
+        return self.api_client.call_api(
+            "/api/2.0/mlflow/generate-code-snippet/trigger-job-run",
+            "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 trigger_workflow_task_config_post(
+        self,
+        create_workflow_task_config_request_dto: CreateWorkflowTaskConfigRequestDto,
+        **kwargs,
+    ) -> CreatePythonDeploymentConfigResponseDto:  # noqa: E501
+        """Trigger Workflow Task Config  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+
+        >>> thread = api.trigger_workflow_task_config_post(create_workflow_task_config_request_dto, async_req=True)
+        >>> result = thread.get()
+
+        :param create_workflow_task_config_request_dto: (required)
+        :type create_workflow_task_config_request_dto: CreateWorkflowTaskConfigRequestDto
+        :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: CreatePythonDeploymentConfigResponseDto
+        """
+        kwargs["_return_http_data_only"] = True
+        if "_preload_content" in kwargs:
+            message = "Error! Please call the trigger_workflow_task_config_post_with_http_info method with `_preload_content` instead and obtain raw data from ApiResponse.raw_data"  # noqa: E501
+            raise ValueError(message)
+        return self.trigger_workflow_task_config_post_with_http_info(
+            create_workflow_task_config_request_dto, **kwargs
+        )  # noqa: E501
+
+    @validate_arguments
+    def trigger_workflow_task_config_post_with_http_info(
+        self,
+        create_workflow_task_config_request_dto: CreateWorkflowTaskConfigRequestDto,
+        **kwargs,
+    ) -> ApiResponse:  # noqa: E501
+        """Trigger Workflow Task Config  # noqa: E501
+
+        This method makes a synchronous HTTP request by default. To make an
+        asynchronous HTTP request, please pass async_req=True
+
+        >>> thread = api.trigger_workflow_task_config_post_with_http_info(create_workflow_task_config_request_dto, async_req=True)
+        >>> result = thread.get()
+
+        :param create_workflow_task_config_request_dto: (required)
+        :type create_workflow_task_config_request_dto: CreateWorkflowTaskConfigRequestDto
+        :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(CreatePythonDeploymentConfigResponseDto, status_code(int), headers(HTTPHeaderDict))
+        """
+
+        _params = locals()
+
+        _all_params = ["create_workflow_task_config_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 trigger_workflow_task_config_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["create_workflow_task_config_request_dto"] is not None:
+            _body_params = _params["create_workflow_task_config_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 = []  # noqa: E501
+
+        _response_types_map = {
+            "200": "CreatePythonDeploymentConfigResponseDto",
+            "422": "HTTPValidationError",
+        }
+
+        return self.api_client.call_api(
+            "/api/2.0/mlflow/generate-code-snippet/workflow-task-config",
+            "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"),
+        )

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

@@ -59,6 +59,7 @@ from truefoundry.ml.autogen.client.models.chat_prompt_messages_inner import (
     ChatPromptMessagesInner,
 )
 from truefoundry.ml.autogen.client.models.columns_dto import ColumnsDto
+from truefoundry.ml.autogen.client.models.command import Command
 from truefoundry.ml.autogen.client.models.content import Content
 from truefoundry.ml.autogen.client.models.content1 import Content1
 from truefoundry.ml.autogen.client.models.content2 import Content2
@@ -110,6 +111,9 @@ from truefoundry.ml.autogen.client.models.create_run_request_dto import (
 from truefoundry.ml.autogen.client.models.create_run_response_dto import (
     CreateRunResponseDto,
 )
+from truefoundry.ml.autogen.client.models.create_workflow_task_config_request_dto import (
+    CreateWorkflowTaskConfigRequestDto,
+)
 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 (
@@ -139,6 +143,7 @@ from truefoundry.ml.autogen.client.models.experiment_tag_dto import ExperimentTa
 from truefoundry.ml.autogen.client.models.external_model_source import (
     ExternalModelSource,
 )
+from truefoundry.ml.autogen.client.models.fast_ai_framework import FastAIFramework
 from truefoundry.ml.autogen.client.models.feature_dto import FeatureDto
 from truefoundry.ml.autogen.client.models.feature_value_type import FeatureValueType
 from truefoundry.ml.autogen.client.models.file_info_dto import FileInfoDto
@@ -182,14 +187,18 @@ from truefoundry.ml.autogen.client.models.get_signed_urls_for_dataset_write_resp
 from truefoundry.ml.autogen.client.models.get_tenant_id_response_dto import (
     GetTenantIdResponseDto,
 )
+from truefoundry.ml.autogen.client.models.gluon_framework import GluonFramework
+from truefoundry.ml.autogen.client.models.h2_o_framework import H2OFramework
 from truefoundry.ml.autogen.client.models.http_validation_error import (
     HTTPValidationError,
 )
 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.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
+from truefoundry.ml.autogen.client.models.light_gbm_framework import LightGBMFramework
 from truefoundry.ml.autogen.client.models.list_artifact_versions_request_dto import (
     ListArtifactVersionsRequestDto,
 )
@@ -291,10 +300,13 @@ from truefoundry.ml.autogen.client.models.multi_part_upload_storage_provider imp
 from truefoundry.ml.autogen.client.models.notify_artifact_version_failure_dto import (
     NotifyArtifactVersionFailureDto,
 )
+from truefoundry.ml.autogen.client.models.onnx_framework import ONNXFramework
 from truefoundry.ml.autogen.client.models.openapi_spec import OpenapiSpec
+from truefoundry.ml.autogen.client.models.paddle_framework import PaddleFramework
 from truefoundry.ml.autogen.client.models.param_dto import ParamDto
 from truefoundry.ml.autogen.client.models.parameters import Parameters
 from truefoundry.ml.autogen.client.models.prediction_type import PredictionType
+from truefoundry.ml.autogen.client.models.py_torch_framework import PyTorchFramework
 from truefoundry.ml.autogen.client.models.resolve_agent_app_response_dto import (
     ResolveAgentAppResponseDto,
 )
@@ -319,7 +331,12 @@ from truefoundry.ml.autogen.client.models.set_experiment_tag_request_dto import
 )
 from truefoundry.ml.autogen.client.models.set_tag_request_dto import SetTagRequestDto
 from truefoundry.ml.autogen.client.models.signed_url_dto import SignedURLDto
+from truefoundry.ml.autogen.client.models.sklearn_framework import SklearnFramework
 from truefoundry.ml.autogen.client.models.source import Source
+from truefoundry.ml.autogen.client.models.spa_cy_framework import SpaCyFramework
+from truefoundry.ml.autogen.client.models.stats_models_framework import (
+    StatsModelsFramework,
+)
 from truefoundry.ml.autogen.client.models.stop import Stop
 from truefoundry.ml.autogen.client.models.store_run_logs_request_dto import (
     StoreRunLogsRequestDto,
@@ -327,14 +344,20 @@ from truefoundry.ml.autogen.client.models.store_run_logs_request_dto import (
 from truefoundry.ml.autogen.client.models.subject import Subject
 from truefoundry.ml.autogen.client.models.subject_type import SubjectType
 from truefoundry.ml.autogen.client.models.system_message import SystemMessage
-from truefoundry.ml.autogen.client.models.tensorflow_framework import (
-    TensorflowFramework,
+from truefoundry.ml.autogen.client.models.tensor_flow_framework import (
+    TensorFlowFramework,
 )
 from truefoundry.ml.autogen.client.models.text import Text
 from truefoundry.ml.autogen.client.models.text_content_part import TextContentPart
 from truefoundry.ml.autogen.client.models.transformers_framework import (
     TransformersFramework,
 )
+from truefoundry.ml.autogen.client.models.trigger_job_run_config_request_dto import (
+    TriggerJobRunConfigRequestDto,
+)
+from truefoundry.ml.autogen.client.models.trigger_job_run_config_response_dto import (
+    TriggerJobRunConfigResponseDto,
+)
 from truefoundry.ml.autogen.client.models.truefoundry_model_source import (
     TruefoundryModelSource,
 )
@@ -356,10 +379,10 @@ from truefoundry.ml.autogen.client.models.update_run_request_dto import (
 from truefoundry.ml.autogen.client.models.update_run_response_dto import (
     UpdateRunResponseDto,
 )
-from truefoundry.ml.autogen.client.models.upload_model_source import UploadModelSource
 from truefoundry.ml.autogen.client.models.url import Url
 from truefoundry.ml.autogen.client.models.user_message import UserMessage
 from truefoundry.ml.autogen.client.models.validation_error import ValidationError
 from truefoundry.ml.autogen.client.models.validation_error_loc_inner import (
     ValidationErrorLocInner,
 )
+from truefoundry.ml.autogen.client.models.xg_boost_framework import XGBoostFramework