truefoundry 0.4.5__py3-none-any.whl → 0.5.0rc1__py3-none-any.whl

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

@@ -0,0 +1,74 @@
+# 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
+
+from __future__ import annotations
+
+import json
+import pprint
+import re  # noqa: F401
+
+from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
+
+
+class SpaCyFramework(BaseModel):
+    """
+    +docs=spaCy framework for the model version +label=SpaCy  # noqa: E501
+    """
+
+    type: StrictStr = Field(
+        default=..., description="+label=Type +usage=Type of the framework +value=spacy"
+    )
+    __properties = ["type"]
+
+    @validator("type")
+    def type_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in ("spacy"):
+            raise ValueError("must be one of enum values ('spacy')")
+        return value
+
+    class Config:
+        """Pydantic configuration"""
+
+        allow_population_by_field_name = True
+        validate_assignment = True
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.dict(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> SpaCyFramework:
+        """Create an instance of SpaCyFramework from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self):
+        """Returns the dictionary representation of the model using alias"""
+        _dict = self.dict(by_alias=True, exclude={}, exclude_none=True)
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> SpaCyFramework:
+        """Create an instance of SpaCyFramework from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return SpaCyFramework.parse_obj(obj)
+
+        _obj = SpaCyFramework.parse_obj({"type": obj.get("type")})
+        return _obj

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

@@ -0,0 +1,75 @@
+# 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
+
+from __future__ import annotations
+
+import json
+import pprint
+import re  # noqa: F401
+
+from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
+
+
+class StatsModelsFramework(BaseModel):
+    """
+    +docs=StatsModels framework for the model version +label=StatsModels  # noqa: E501
+    """
+
+    type: StrictStr = Field(
+        default=...,
+        description="+label=Type +usage=Type of the framework +value=statsmodels",
+    )
+    __properties = ["type"]
+
+    @validator("type")
+    def type_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in ("statsmodels"):
+            raise ValueError("must be one of enum values ('statsmodels')")
+        return value
+
+    class Config:
+        """Pydantic configuration"""
+
+        allow_population_by_field_name = True
+        validate_assignment = True
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.dict(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> StatsModelsFramework:
+        """Create an instance of StatsModelsFramework from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self):
+        """Returns the dictionary representation of the model using alias"""
+        _dict = self.dict(by_alias=True, exclude={}, exclude_none=True)
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> StatsModelsFramework:
+        """Create an instance of StatsModelsFramework from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return StatsModelsFramework.parse_obj(obj)
+
+        _obj = StatsModelsFramework.parse_obj({"type": obj.get("type")})
+        return _obj

truefoundry/ml/autogen/client/models/{tensorflow_framework.py → tensor_flow_framework.py}

@@ -20,13 +20,14 @@ import re  # noqa: F401
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
 
 
-class TensorflowFramework(BaseModel):
+class TensorFlowFramework(BaseModel):
     """
-    +docs=Tensorflow framework for the model version +usage=Tensorflow framework for the model version +label=Tensorflow framework  # noqa: E501
+    +docs=TensorFlow framework for the model version +label=TensorFlow  # noqa: E501
     """
 
     type: StrictStr = Field(
-        default=..., description="+label=Type +usage=Type of the framework"
+        default=...,
+        description="+label=Type +usage=Type of the framework +value=tensorflow",
     )
     __properties = ["type"]
 
@@ -52,8 +53,8 @@ class TensorflowFramework(BaseModel):
         return json.dumps(self.to_dict())
 
     @classmethod
-    def from_json(cls, json_str: str) -> TensorflowFramework:
-        """Create an instance of TensorflowFramework from a JSON string"""
+    def from_json(cls, json_str: str) -> TensorFlowFramework:
+        """Create an instance of TensorFlowFramework from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self):
@@ -62,13 +63,13 @@ class TensorflowFramework(BaseModel):
         return _dict
 
     @classmethod
-    def from_dict(cls, obj: dict) -> TensorflowFramework:
-        """Create an instance of TensorflowFramework from a dict"""
+    def from_dict(cls, obj: dict) -> TensorFlowFramework:
+        """Create an instance of TensorFlowFramework from a dict"""
         if obj is None:
             return None
 
         if not isinstance(obj, dict):
-            return TensorflowFramework.parse_obj(obj)
+            return TensorFlowFramework.parse_obj(obj)
 
-        _obj = TensorflowFramework.parse_obj({"type": obj.get("type")})
+        _obj = TensorFlowFramework.parse_obj({"type": obj.get("type")})
         return _obj

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

@@ -24,11 +24,12 @@ from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
 
 class TransformersFramework(BaseModel):
     """
-    +docs=Transformers framework for the model version +usage=Transformers framework for the model version +label=Transformers framework  # noqa: E501
+    +docs=Transformers framework for the model version +label=Transformers +value=transformers  # noqa: E501
     """
 
     type: StrictStr = Field(
-        default=..., description="+label=Type +usage=Type of the framework"
+        default=...,
+        description="+label=Type +usage=Type of the framework +value=transformers",
     )
     library_name: Optional[LibraryName] = Field(
         default=None,

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

@@ -0,0 +1,90 @@
+# 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
+
+from __future__ import annotations
+
+import json
+import pprint
+import re  # noqa: F401
+from typing import Dict, Optional
+
+from truefoundry.ml.autogen.client.models.command import Command
+from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr
+
+
+class TriggerJobRunConfigRequestDto(BaseModel):
+    """
+    TriggerJobRunConfigRequestDto
+    """
+
+    control_plane_url: StrictStr = Field(...)
+    application_fqn: StrictStr = Field(...)
+    application_id: StrictStr = Field(...)
+    command: Optional[Command] = None
+    params: Optional[Dict[str, StrictStr]] = None
+    __properties = [
+        "control_plane_url",
+        "application_fqn",
+        "application_id",
+        "command",
+        "params",
+    ]
+
+    class Config:
+        """Pydantic configuration"""
+
+        allow_population_by_field_name = True
+        validate_assignment = True
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.dict(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> TriggerJobRunConfigRequestDto:
+        """Create an instance of TriggerJobRunConfigRequestDto from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self):
+        """Returns the dictionary representation of the model using alias"""
+        _dict = self.dict(by_alias=True, exclude={}, exclude_none=True)
+        # override the default output from truefoundry.pydantic_v1 by calling `to_dict()` of command
+        if self.command:
+            _dict["command"] = self.command.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> TriggerJobRunConfigRequestDto:
+        """Create an instance of TriggerJobRunConfigRequestDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return TriggerJobRunConfigRequestDto.parse_obj(obj)
+
+        _obj = TriggerJobRunConfigRequestDto.parse_obj(
+            {
+                "control_plane_url": obj.get("control_plane_url"),
+                "application_fqn": obj.get("application_fqn"),
+                "application_id": obj.get("application_id"),
+                "command": Command.from_dict(obj.get("command"))
+                if obj.get("command") is not None
+                else None,
+                "params": obj.get("params"),
+            }
+        )
+        return _obj

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

@@ -0,0 +1,71 @@
+# 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
+
+from __future__ import annotations
+
+import json
+import pprint
+import re  # noqa: F401
+
+from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr
+
+
+class TriggerJobRunConfigResponseDto(BaseModel):
+    """
+    TriggerJobRunConfigResponseDto
+    """
+
+    python_code_snippet: StrictStr = Field(...)
+    curl_code_snippet: StrictStr = Field(...)
+    __properties = ["python_code_snippet", "curl_code_snippet"]
+
+    class Config:
+        """Pydantic configuration"""
+
+        allow_population_by_field_name = True
+        validate_assignment = True
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.dict(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> TriggerJobRunConfigResponseDto:
+        """Create an instance of TriggerJobRunConfigResponseDto from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self):
+        """Returns the dictionary representation of the model using alias"""
+        _dict = self.dict(by_alias=True, exclude={}, exclude_none=True)
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> TriggerJobRunConfigResponseDto:
+        """Create an instance of TriggerJobRunConfigResponseDto from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return TriggerJobRunConfigResponseDto.parse_obj(obj)
+
+        _obj = TriggerJobRunConfigResponseDto.parse_obj(
+            {
+                "python_code_snippet": obj.get("python_code_snippet"),
+                "curl_code_snippet": obj.get("curl_code_snippet"),
+            }
+        )
+        return _obj

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

@@ -22,14 +22,16 @@ from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
 
 class TruefoundryModelSource(BaseModel):
     """
-    +docs=Describes the source for the Model +usage=Source for the Model +label=Model Source  # noqa: E501
+    +usage=Source for the Model +label=Upload  # noqa: E501
     """
 
     type: StrictStr = Field(
-        default=..., description="+label=Type +usage=Type of the source"
+        default=...,
+        description="+usage=Type of the source +type=Upload +value=truefoundry +label=Upload",
     )
     uri: StrictStr = Field(
-        default=..., description="+label=URI +usage=URI of the model source"
+        default=...,
+        description='+label=URI +uiType=UploadInput +uiProps={"hideClear":true}',
     )
     __properties = ["type", "uri"]
 

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

@@ -0,0 +1,75 @@
+# 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
+
+from __future__ import annotations
+
+import json
+import pprint
+import re  # noqa: F401
+
+from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
+
+
+class XGBoostFramework(BaseModel):
+    """
+    +docs=XGBoost framework for the model version +label=XGBoost  # noqa: E501
+    """
+
+    type: StrictStr = Field(
+        default=...,
+        description="+label=Type +usage=Type of the framework +value=xgboost",
+    )
+    __properties = ["type"]
+
+    @validator("type")
+    def type_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in ("xgboost"):
+            raise ValueError("must be one of enum values ('xgboost')")
+        return value
+
+    class Config:
+        """Pydantic configuration"""
+
+        allow_population_by_field_name = True
+        validate_assignment = True
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.dict(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> XGBoostFramework:
+        """Create an instance of XGBoostFramework from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self):
+        """Returns the dictionary representation of the model using alias"""
+        _dict = self.dict(by_alias=True, exclude={}, exclude_none=True)
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: dict) -> XGBoostFramework:
+        """Create an instance of XGBoostFramework from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return XGBoostFramework.parse_obj(obj)
+
+        _obj = XGBoostFramework.parse_obj({"type": obj.get("type")})
+        return _obj

truefoundry/ml/autogen/client_README.md

@@ -5,7 +5,7 @@ The `truefoundry.ml.autogen.client` package is automatically generated by the [O
 
 - API version: 0.1.0
 - Package version: 0.1.0
-- Generator version: 7.9.0-SNAPSHOT
+- Generator version: 7.9.0
 - Build package: org.openapitools.codegen.languages.PythonPydanticV1ClientCodegen
 
 ## Requirements.
@@ -66,7 +66,7 @@ All URIs are relative to *http://localhost*
 Class | Method | HTTP request | Description
 ------------ | ------------- | ------------- | -------------
 *AuthApi* | [**get_tenant_id_get**](truefoundry/ml/autogen/client/docs/AuthApi.md#get_tenant_id_get) | **GET** /api/2.0/mlflow/tenant-id | Get Tenant Id
-*DeprecatedApi* | [**get_run_by_name_get**](truefoundry/ml/autogen/client/docs/DeprecatedApi.md#get_run_by_name_get) | **GET** /api/2.0mlflow/runs/get-by-name | Get Run By Name
+*DeprecatedApi* | [**get_run_by_name_get_0**](truefoundry/ml/autogen/client/docs/DeprecatedApi.md#get_run_by_name_get_0) | **GET** /api/2.0mlflow/runs/get-by-name | Get Run By Name
 *DeprecatedApi* | [**get_search_runs_get**](truefoundry/ml/autogen/client/docs/DeprecatedApi.md#get_search_runs_get) | **GET** /api/2.0/preview/mlflow/runs/search | Get Search Runs
 *DeprecatedApi* | [**get_signed_urls_for_dataset_write_deprecated_get**](truefoundry/ml/autogen/client/docs/DeprecatedApi.md#get_signed_urls_for_dataset_write_deprecated_get) | **GET** /api/2.0/mlflow/mlfoundry-artifacts/datasets/get-signed-urls-for-write | Get Signed Urls For Dataset Write Deprecated
 *ExperimentsApi* | [**create_experiment_post**](truefoundry/ml/autogen/client/docs/ExperimentsApi.md#create_experiment_post) | **POST** /api/2.0/mlflow/experiments/create | Create Experiment
@@ -80,6 +80,9 @@ Class | Method | HTTP request | Description
 *ExperimentsApi* | [**restore_experiment_post**](truefoundry/ml/autogen/client/docs/ExperimentsApi.md#restore_experiment_post) | **POST** /api/2.0/mlflow/experiments/restore | Restore Experiment
 *ExperimentsApi* | [**set_experiment_tag_post**](truefoundry/ml/autogen/client/docs/ExperimentsApi.md#set_experiment_tag_post) | **POST** /api/2.0/mlflow/experiments/set-experiment-tag | Set Experiment Tag
 *ExperimentsApi* | [**update_experiment_post**](truefoundry/ml/autogen/client/docs/ExperimentsApi.md#update_experiment_post) | **POST** /api/2.0/mlflow/experiments/update | Update Experiment
+*GenerateCodeSnippetApi* | [**generate_py_development_config_post**](truefoundry/ml/autogen/client/docs/GenerateCodeSnippetApi.md#generate_py_development_config_post) | **POST** /api/2.0/mlflow/generate-code-snippet/application | Generate Py Development Config
+*GenerateCodeSnippetApi* | [**trigger_job_run_post**](truefoundry/ml/autogen/client/docs/GenerateCodeSnippetApi.md#trigger_job_run_post) | **POST** /api/2.0/mlflow/generate-code-snippet/trigger-job-run | Trigger Job Run
+*GenerateCodeSnippetApi* | [**trigger_workflow_task_config_post**](truefoundry/ml/autogen/client/docs/GenerateCodeSnippetApi.md#trigger_workflow_task_config_post) | **POST** /api/2.0/mlflow/generate-code-snippet/workflow-task-config | Trigger Workflow Task Config
 *HealthApi* | [**health_get**](truefoundry/ml/autogen/client/docs/HealthApi.md#health_get) | **GET** /health | Health
 *HealthApi* | [**serve_get**](truefoundry/ml/autogen/client/docs/HealthApi.md#serve_get) | **GET** / | Serve
 *MetricsApi* | [**get_metric_history_get**](truefoundry/ml/autogen/client/docs/MetricsApi.md#get_metric_history_get) | **GET** /api/2.0/mlflow/metrics/get-history | Get Metric History
@@ -129,7 +132,6 @@ Class | Method | HTTP request | Description
 *MlfoundryArtifactsApi* | [**update_artifact_version_post**](truefoundry/ml/autogen/client/docs/MlfoundryArtifactsApi.md#update_artifact_version_post) | **POST** /api/2.0/mlflow/mlfoundry-artifacts/artifact-versions/update | Update Artifact Version
 *MlfoundryArtifactsApi* | [**update_dataset_post**](truefoundry/ml/autogen/client/docs/MlfoundryArtifactsApi.md#update_dataset_post) | **POST** /api/2.0/mlflow/mlfoundry-artifacts/datasets/update | Update Dataset
 *MlfoundryArtifactsApi* | [**update_model_version_post**](truefoundry/ml/autogen/client/docs/MlfoundryArtifactsApi.md#update_model_version_post) | **POST** /api/2.0/mlflow/mlfoundry-artifacts/model-versions/update | Update Model Version
-*PythonDeploymentConfigApi* | [**generate_py_development_config_post**](truefoundry/ml/autogen/client/docs/PythonDeploymentConfigApi.md#generate_py_development_config_post) | **POST** /api/2.0/mlflow/python-deployment-config/generate | Generate Py Development Config
 *RunArtifactsApi* | [**list_run_artifacts_get**](truefoundry/ml/autogen/client/docs/RunArtifactsApi.md#list_run_artifacts_get) | **GET** /api/2.0/mlflow/artifacts/list | List Run Artifacts
 *RunsApi* | [**create_run_post**](truefoundry/ml/autogen/client/docs/RunsApi.md#create_run_post) | **POST** /api/2.0/mlflow/runs/create | Create Run
 *RunsApi* | [**delete_run_post**](truefoundry/ml/autogen/client/docs/RunsApi.md#delete_run_post) | **POST** /api/2.0/mlflow/runs/delete | Delete Run
@@ -175,6 +177,7 @@ Class | Method | HTTP request | Description
  - [ChatPrompt](truefoundry/ml/autogen/client/docs/ChatPrompt.md)
  - [ChatPromptMessagesInner](truefoundry/ml/autogen/client/docs/ChatPromptMessagesInner.md)
  - [ColumnsDto](truefoundry/ml/autogen/client/docs/ColumnsDto.md)
+ - [Command](truefoundry/ml/autogen/client/docs/Command.md)
  - [Content](truefoundry/ml/autogen/client/docs/Content.md)
  - [Content1](truefoundry/ml/autogen/client/docs/Content1.md)
  - [Content2](truefoundry/ml/autogen/client/docs/Content2.md)
@@ -194,6 +197,7 @@ Class | Method | HTTP request | Description
  - [CreatePythonDeploymentConfigResponseDto](truefoundry/ml/autogen/client/docs/CreatePythonDeploymentConfigResponseDto.md)
  - [CreateRunRequestDto](truefoundry/ml/autogen/client/docs/CreateRunRequestDto.md)
  - [CreateRunResponseDto](truefoundry/ml/autogen/client/docs/CreateRunResponseDto.md)
+ - [CreateWorkflowTaskConfigRequestDto](truefoundry/ml/autogen/client/docs/CreateWorkflowTaskConfigRequestDto.md)
  - [DatasetDto](truefoundry/ml/autogen/client/docs/DatasetDto.md)
  - [DatasetResponseDto](truefoundry/ml/autogen/client/docs/DatasetResponseDto.md)
  - [DeleteArtifactVersionsRequestDto](truefoundry/ml/autogen/client/docs/DeleteArtifactVersionsRequestDto.md)
@@ -207,6 +211,7 @@ Class | Method | HTTP request | Description
  - [ExperimentResponseDto](truefoundry/ml/autogen/client/docs/ExperimentResponseDto.md)
  - [ExperimentTagDto](truefoundry/ml/autogen/client/docs/ExperimentTagDto.md)
  - [ExternalModelSource](truefoundry/ml/autogen/client/docs/ExternalModelSource.md)
+ - [FastAIFramework](truefoundry/ml/autogen/client/docs/FastAIFramework.md)
  - [FeatureDto](truefoundry/ml/autogen/client/docs/FeatureDto.md)
  - [FeatureValueType](truefoundry/ml/autogen/client/docs/FeatureValueType.md)
  - [FileInfoDto](truefoundry/ml/autogen/client/docs/FileInfoDto.md)
@@ -224,12 +229,16 @@ Class | Method | HTTP request | Description
  - [GetSignedURLsForDatasetReadResponseDto](truefoundry/ml/autogen/client/docs/GetSignedURLsForDatasetReadResponseDto.md)
  - [GetSignedURLsForDatasetWriteResponseDto](truefoundry/ml/autogen/client/docs/GetSignedURLsForDatasetWriteResponseDto.md)
  - [GetTenantIdResponseDto](truefoundry/ml/autogen/client/docs/GetTenantIdResponseDto.md)
+ - [GluonFramework](truefoundry/ml/autogen/client/docs/GluonFramework.md)
+ - [H2OFramework](truefoundry/ml/autogen/client/docs/H2OFramework.md)
  - [HTTPValidationError](truefoundry/ml/autogen/client/docs/HTTPValidationError.md)
  - [ImageContentPart](truefoundry/ml/autogen/client/docs/ImageContentPart.md)
  - [ImageUrl](truefoundry/ml/autogen/client/docs/ImageUrl.md)
  - [InternalMetadata](truefoundry/ml/autogen/client/docs/InternalMetadata.md)
+ - [KerasFramework](truefoundry/ml/autogen/client/docs/KerasFramework.md)
  - [LatestRunLogDto](truefoundry/ml/autogen/client/docs/LatestRunLogDto.md)
  - [LibraryName](truefoundry/ml/autogen/client/docs/LibraryName.md)
+ - [LightGBMFramework](truefoundry/ml/autogen/client/docs/LightGBMFramework.md)
  - [ListArtifactVersionsRequestDto](truefoundry/ml/autogen/client/docs/ListArtifactVersionsRequestDto.md)
  - [ListArtifactVersionsResponseDto](truefoundry/ml/autogen/client/docs/ListArtifactVersionsResponseDto.md)
  - [ListArtifactsRequestDto](truefoundry/ml/autogen/client/docs/ListArtifactsRequestDto.md)
@@ -269,10 +278,13 @@ Class | Method | HTTP request | Description
  - [MultiPartUploadResponseDto](truefoundry/ml/autogen/client/docs/MultiPartUploadResponseDto.md)
  - [MultiPartUploadStorageProvider](truefoundry/ml/autogen/client/docs/MultiPartUploadStorageProvider.md)
  - [NotifyArtifactVersionFailureDto](truefoundry/ml/autogen/client/docs/NotifyArtifactVersionFailureDto.md)
+ - [ONNXFramework](truefoundry/ml/autogen/client/docs/ONNXFramework.md)
  - [OpenapiSpec](truefoundry/ml/autogen/client/docs/OpenapiSpec.md)
+ - [PaddleFramework](truefoundry/ml/autogen/client/docs/PaddleFramework.md)
  - [ParamDto](truefoundry/ml/autogen/client/docs/ParamDto.md)
  - [Parameters](truefoundry/ml/autogen/client/docs/Parameters.md)
  - [PredictionType](truefoundry/ml/autogen/client/docs/PredictionType.md)
+ - [PyTorchFramework](truefoundry/ml/autogen/client/docs/PyTorchFramework.md)
  - [ResolveAgentAppResponseDto](truefoundry/ml/autogen/client/docs/ResolveAgentAppResponseDto.md)
  - [RestoreRunRequestDto](truefoundry/ml/autogen/client/docs/RestoreRunRequestDto.md)
  - [RunDataDto](truefoundry/ml/autogen/client/docs/RunDataDto.md)
@@ -287,16 +299,21 @@ Class | Method | HTTP request | Description
  - [SetExperimentTagRequestDto](truefoundry/ml/autogen/client/docs/SetExperimentTagRequestDto.md)
  - [SetTagRequestDto](truefoundry/ml/autogen/client/docs/SetTagRequestDto.md)
  - [SignedURLDto](truefoundry/ml/autogen/client/docs/SignedURLDto.md)
+ - [SklearnFramework](truefoundry/ml/autogen/client/docs/SklearnFramework.md)
  - [Source](truefoundry/ml/autogen/client/docs/Source.md)
+ - [SpaCyFramework](truefoundry/ml/autogen/client/docs/SpaCyFramework.md)
+ - [StatsModelsFramework](truefoundry/ml/autogen/client/docs/StatsModelsFramework.md)
  - [Stop](truefoundry/ml/autogen/client/docs/Stop.md)
  - [StoreRunLogsRequestDto](truefoundry/ml/autogen/client/docs/StoreRunLogsRequestDto.md)
  - [Subject](truefoundry/ml/autogen/client/docs/Subject.md)
  - [SubjectType](truefoundry/ml/autogen/client/docs/SubjectType.md)
  - [SystemMessage](truefoundry/ml/autogen/client/docs/SystemMessage.md)
- - [TensorflowFramework](truefoundry/ml/autogen/client/docs/TensorflowFramework.md)
+ - [TensorFlowFramework](truefoundry/ml/autogen/client/docs/TensorFlowFramework.md)
  - [Text](truefoundry/ml/autogen/client/docs/Text.md)
  - [TextContentPart](truefoundry/ml/autogen/client/docs/TextContentPart.md)
  - [TransformersFramework](truefoundry/ml/autogen/client/docs/TransformersFramework.md)
+ - [TriggerJobRunConfigRequestDto](truefoundry/ml/autogen/client/docs/TriggerJobRunConfigRequestDto.md)
+ - [TriggerJobRunConfigResponseDto](truefoundry/ml/autogen/client/docs/TriggerJobRunConfigResponseDto.md)
  - [TruefoundryModelSource](truefoundry/ml/autogen/client/docs/TruefoundryModelSource.md)
  - [UpdateArtifactVersionRequestDto](truefoundry/ml/autogen/client/docs/UpdateArtifactVersionRequestDto.md)
  - [UpdateDatasetRequestDto](truefoundry/ml/autogen/client/docs/UpdateDatasetRequestDto.md)
@@ -304,11 +321,11 @@ Class | Method | HTTP request | Description
  - [UpdateModelVersionRequestDto](truefoundry/ml/autogen/client/docs/UpdateModelVersionRequestDto.md)
  - [UpdateRunRequestDto](truefoundry/ml/autogen/client/docs/UpdateRunRequestDto.md)
  - [UpdateRunResponseDto](truefoundry/ml/autogen/client/docs/UpdateRunResponseDto.md)
- - [UploadModelSource](truefoundry/ml/autogen/client/docs/UploadModelSource.md)
  - [Url](truefoundry/ml/autogen/client/docs/Url.md)
  - [UserMessage](truefoundry/ml/autogen/client/docs/UserMessage.md)
  - [ValidationError](truefoundry/ml/autogen/client/docs/ValidationError.md)
  - [ValidationErrorLocInner](truefoundry/ml/autogen/client/docs/ValidationErrorLocInner.md)
+ - [XGBoostFramework](truefoundry/ml/autogen/client/docs/XGBoostFramework.md)
 
 
 <a id="documentation-for-authorization"></a>

truefoundry/ml/autogen/entities/artifacts.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  artifacts.json
-#   timestamp: 2024-10-25T06:52:16+00:00
+#   timestamp: 2024-10-30T07:36:31+00:00
 
 from __future__ import annotations
 
@@ -226,6 +226,17 @@ class ModelConfiguration(BaseModel):
     )
 
 
+class ONNXFramework(BaseModel):
+    """
+    +docs=ONNX framework for the model version
+    +label=ONNX
+    """
+
+    type: Literal["onnx"] = Field(
+        ..., description="+label=Type\n+usage=Type of the framework\n+value=onnx"
+    )
+
+
 class PaddleFramework(BaseModel):
     """
     +docs=PaddlePaddle framework for the model version
@@ -348,6 +359,10 @@ class TransformersFramework(BaseModel):
         None,
         description="+label=Pipeline Tag\n+usage=Pipeline tag\n+docs=Pipeline tag for the framework",
     )
+    base_model: Optional[str] = Field(
+        None,
+        description="+label=Base Model\n+usage=Base model\n+docs=Base model Id. If this is a finetuned model, this points to the base model used for finetuning",
+    )
 
 
 class TruefoundryModelSource(BaseModel):
@@ -361,7 +376,8 @@ class TruefoundryModelSource(BaseModel):
         description="+usage=Type of the source\n+type=Upload\n+value=truefoundry\n+label=Upload",
     )
     uri: str = Field(
-        ..., description='+label=URI\n+uiType=UploadInput\n+uiProps={"hideClear":true}'
+        ...,
+        description='+label=URI\n+uiType=ModelUploadInput\n+uiProps={"hideClear":true}',
     )
 
 
@@ -496,6 +512,7 @@ class ModelVersion(BaseArtifactVersion):
             LightGBMFramework,
             FastAIFramework,
             H2OFramework,
+            ONNXFramework,
             SpaCyFramework,
             StatsModelsFramework,
             GluonFramework,