truefoundry 0.5.3rc5__tar.gz → 0.5.5__tar.gz

{truefoundry-0.5.3rc5 → truefoundry-0.5.5}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: truefoundry
-Version: 0.5.3rc5
+Version: 0.5.5
 Summary: Truefoundry CLI
 Author: Abhishek Choudhary
 Author-email: abhishek@truefoundry.com

{truefoundry-0.5.3rc5 → truefoundry-0.5.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "truefoundry"
-version = "0.5.3rc5"
+version = "0.5.5"
 description = "Truefoundry CLI"
 authors = ["Abhishek Choudhary <abhishek@truefoundry.com>"]
 readme = "README.md"
@@ -139,7 +139,6 @@ warn_unused_ignores = true
 packages = [
     "truefoundry.common",
     "truefoundry.ml",
-    "truefoundry.gateway",
 ]
 exclude = [
     "^truefoundry/ml/autogen",

{truefoundry-0.5.3rc5 → truefoundry-0.5.5}/truefoundry/deploy/__init__.py

@@ -9,6 +9,10 @@ from truefoundry.deploy.auto_gen.models import (
     Kustomize,
     ParamType,
     Protocol,
+    SparkDriverConfig,
+    SparkExecutorConfig,
+    SparkExecutorDynamicScaling,
+    SparkExecutorFixedInstances,
     WorkbenchImage,
 )
 from truefoundry.deploy.lib.dao.application import (
@@ -36,11 +40,12 @@ from truefoundry.deploy.v2.lib.deployable_patched_models import (
     Application,
     ApplicationSet,
     AsyncService,
-    Codeserver,
     Helm,
     Job,
     Notebook,
+    RStudio,
     Service,
+    SparkJob,
     SSHServer,
     Volume,
     Workflow,
@@ -110,7 +115,9 @@ from truefoundry.deploy.v2.lib.patched_models import (
     StaticVolumeConfig,
     StringDataMount,
     TPUType,
-    TruefoundryArtifactSource,
+    TrueFoundryArtifactSource,
+    TruefoundryArtifactSource,  # deprecated, kept for backwards compatibility
+    TrueFoundryInteractiveLogin,
     VolumeBrowser,
     VolumeMount,
     WorkerConfig,

{truefoundry-0.5.3rc5 → truefoundry-0.5.5}/truefoundry/deploy/auto_gen/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  application.json
-#   timestamp: 2024-12-16T11:59:07+00:00
+#   timestamp: 2025-01-20T07:24:00+00:00
 
 from __future__ import annotations
 
@@ -659,6 +659,10 @@ class Profile(str, Enum):
 
 class NvidiaMIGGPU(BaseModel):
     type: Literal["nvidia_mig_gpu"] = Field(..., description="+value=nvidia_mig_gpu")
+    name: Optional[str] = Field(
+        None,
+        description="+label=GPU Name\n+usage=Name of the Nvidia GPU. One of [P4, P100, V100, T4, A10G, A100_40GB, A100_80GB]\nThis field is required for Node Selector and can be ignored in Nodepool Selector.\nOne instance of the card contains the following amount of memory -\nP4: 8 GB, P100: 16 GB, V100: 16 GB, T4: 16 GB, A10G: 24 GB, A100_40GB: 40GB, A100_80GB: 80 GB",
+    )
     profile: Profile = Field(
         ...,
         description="+label=MIG Profile\n+usage=Name of the MIG profile to use. One of [1g.5gb, 2g.10gb, 3g.20gb, 1g.10gb, 2g.20gb, 3g.40gb]",
@@ -669,6 +673,10 @@ class NvidiaTimeslicingGPU(BaseModel):
     type: Literal["nvidia_timeslicing_gpu"] = Field(
         ..., description="+value=nvidia_timeslicing_gpu"
     )
+    name: Optional[str] = Field(
+        None,
+        description="+label=GPU Name\n+usage=Name of the Nvidia GPU. One of [P4, P100, V100, T4, A10G, A100_40GB, A100_80GB]\nThis field is required for Node Selector and can be ignored in Nodepool Selector.\nOne instance of the card contains the following amount of memory -\nP4: 8 GB, P100: 16 GB, V100: 16 GB, T4: 16 GB, A10G: 24 GB, A100_40GB: 40GB, A100_80GB: 80 GB",
+    )
     gpu_memory: conint(ge=1, le=200000) = Field(
         ...,
         description="+label=GPU Memory (MB)\n+usage=Amount of GPU memory (in MB) to allocate. Please note, this limit is not being enforced today but will be in future. Applications are expected to operate in co-opertative mode",
@@ -735,45 +743,6 @@ class AppProtocol(str, Enum):
     tcp = "tcp"
 
 
-class Port(BaseModel):
-    """
-    +docs=Describes the ports the service should be exposed to.
-    """
-
-    port: conint(ge=1, le=65535) = Field(
-        80, description="+usage=Port number to expose."
-    )
-    protocol: Protocol = Field("TCP", description="+usage=Protocol for the port.")
-    expose: bool = Field(True, description="+usage=Expose the port")
-    app_protocol: AppProtocol = Field(
-        "http",
-        description="+label=Application Protocol\n+usage=Application Protocol for the port.\nSelect the application protocol used by your service. For most use cases, this should be `http`(HTTP/1.1).\nIf you are running a gRPC server, select the `grpc` option.\nThis is only applicable if `expose=true`.",
-    )
-    host: Optional[
-        constr(
-            regex=r"^((([a-zA-Z0-9\-]{1,63}\.)([a-zA-Z0-9\-]{1,63}\.)*([A-Za-z]{1,63}))|(((25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.){3}(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)))$"
-        )
-    ] = Field(
-        None,
-        description="+usage=Host e.g. ai.example.com, app.truefoundry.com\n+message=Upto 253 characters, each part of host should be at most 63 characters long, can contain alphabets, digits and hypen, must begin and end with an alphanumeric characters. Parts must be separated by periods (.)",
-    )
-    path: Optional[
-        constr(regex=r"^(/([a-zA-Z0-9]|[a-zA-Z0-9][a-zA-Z0-9\-_\.]*[a-zA-Z0-9]))*/$")
-    ] = Field(
-        None,
-        description="+usage=Path e.g. /v1/api/ml/, /v2/docs/\n+message=Should begin and end with a forward slash (/). Each part can can contain alphabets, digits and hypen, must begin and end with an alphanumeric characters. Parts should be separated by forward slashes (/)",
-    )
-    rewrite_path_to: Optional[
-        constr(regex=r"^(/([a-zA-Z0-9]|[a-zA-Z0-9][a-zA-Z0-9\-_\.]*[a-zA-Z0-9]))*/$")
-    ] = Field(
-        None,
-        description="+label=Rewrite Path to\n+usage=Rewrite the path prefix to a different path.\nIf `path` is `/v1/api` and `rewrite_path_to` is `/api`. The URI in the HTTP request `http://0.0.0.0:8080/v1/api/houses` will be rewritten to `http://0.0.0.0:8080/api/houses` before the request is forwarded your service.\nDefaults to `/`.\nThis is only applicable if `path` is given.\n+message=Should begin and end with a forward slash (/). Each part can can contain alphabets, digits and hypen, must begin and end with an alphanumeric characters. Parts should be separated by forward slashes (/)",
-    )
-    auth: Optional[Union[BasicAuthCreds, JwtAuthCreds]] = Field(
-        None, description="+usage=Username and Password for service auth"
-    )
-
-
 class PythonBuild(BaseModel):
     """
     +docs=Describes that we are using python to build a container image with a specific python version and pip packages installed.
@@ -1021,6 +990,42 @@ class ServiceAutoscaling(BaseAutoscaling):
     )
 
 
+class SparkDriverConfig(BaseModel):
+    """
+    +label=Driver Config
+    """
+
+    ui_endpoint: Endpoint
+    resources: Optional[Resources] = None
+
+
+class SparkExecutorDynamicScaling(BaseModel):
+    """
+    +label=Dynamic Scaling
+    """
+
+    type: Literal["dynamic"] = Field(..., description="+value=dynamic")
+    min: conint(ge=0, le=500) = Field(
+        1,
+        description="+label=Min Instances\n+usage=Minimum number of instances to start / scale down to\n+sort=100",
+    )
+    max: conint(ge=0, le=500) = Field(
+        1,
+        description="+label=Max Instances\n+usage=Maximum number of instances to scale up to\n+sort=200",
+    )
+
+
+class SparkExecutorFixedInstances(BaseModel):
+    """
+    +label=Fixed Instances
+    """
+
+    type: Literal["fixed"] = Field(..., description="+value=fixed")
+    count: conint(ge=0, le=500) = Field(
+        1, description="+label=Instances Count\n+usage=Number of instances to start"
+    )
+
+
 class StaticVolumeConfig(BaseModel):
     """
     +label=Static Volume Config
@@ -1106,6 +1111,16 @@ class TaskPythonBuild(BaseModel):
     )
 
 
+class TrueFoundryInteractiveLogin(BaseModel):
+    """
+    +label=Login with truefoundry
+    """
+
+    type: Literal["truefoundry_oauth"] = Field(
+        ..., description="+value=truefoundry_oauth"
+    )
+
+
 class TruefoundryArtifactSource(BaseModel):
     """
     +docs=Input for Artifact from Truefoundry Artifact Registry
@@ -1279,9 +1294,8 @@ class Codeserver(BaseWorkbenchInput):
     +docs=Describes the configuration for the code server
     """
 
-    type: Literal["codeserver"] = Field(..., description="+value=Code Server")
+    type: Literal["codeserver"] = Field(..., description="+value=codeserver")
     image: WorkbenchImage
-    auth: Optional[BasicAuthCreds] = None
 
 
 class ContainerTaskConfig(BaseModel):
@@ -1549,13 +1563,51 @@ class Notebook(BaseWorkbenchInput):
 
     type: Literal["notebook"] = Field(..., description="+value=notebook")
     image: WorkbenchImage
-    auth: Optional[BasicAuthCreds] = None
     cull_timeout: conint(ge=5) = Field(
         30,
         description="+label=Stop after (minutes of inactivity)\n+usage=Stop the notebook instance after this much time in minutes of inactivity.\nThe notebook instance will be stopped even if the notebook is open in your browser, but nothing is running on the notebook.\n+sort=5",
     )
 
 
+class Port(BaseModel):
+    """
+    +docs=Describes the ports the service should be exposed to.
+    """
+
+    port: conint(ge=1, le=65535) = Field(
+        80, description="+usage=Port number to expose."
+    )
+    protocol: Protocol = Field("TCP", description="+usage=Protocol for the port.")
+    expose: bool = Field(True, description="+usage=Expose the port")
+    app_protocol: AppProtocol = Field(
+        "http",
+        description="+label=Application Protocol\n+usage=Application Protocol for the port.\nSelect the application protocol used by your service. For most use cases, this should be `http`(HTTP/1.1).\nIf you are running a gRPC server, select the `grpc` option.\nThis is only applicable if `expose=true`.",
+    )
+    host: Optional[
+        constr(
+            regex=r"^((([a-zA-Z0-9\-]{1,63}\.)([a-zA-Z0-9\-]{1,63}\.)*([A-Za-z]{1,63}))|(((25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.){3}(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)))$"
+        )
+    ] = Field(
+        None,
+        description="+usage=Host e.g. ai.example.com, app.truefoundry.com\n+message=Upto 253 characters, each part of host should be at most 63 characters long, can contain alphabets, digits and hypen, must begin and end with an alphanumeric characters. Parts must be separated by periods (.)",
+    )
+    path: Optional[
+        constr(regex=r"^(/([a-zA-Z0-9]|[a-zA-Z0-9][a-zA-Z0-9\-_\.]*[a-zA-Z0-9]))*/$")
+    ] = Field(
+        None,
+        description="+usage=Path e.g. /v1/api/ml/, /v2/docs/\n+message=Should begin and end with a forward slash (/). Each part can can contain alphabets, digits and hypen, must begin and end with an alphanumeric characters. Parts should be separated by forward slashes (/)",
+    )
+    rewrite_path_to: Optional[
+        constr(regex=r"^(/([a-zA-Z0-9]|[a-zA-Z0-9][a-zA-Z0-9\-_\.]*[a-zA-Z0-9]))*/$")
+    ] = Field(
+        None,
+        description="+label=Rewrite Path to\n+usage=Rewrite the path prefix to a different path.\nIf `path` is `/v1/api` and `rewrite_path_to` is `/api`. The URI in the HTTP request `http://0.0.0.0:8080/v1/api/houses` will be rewritten to `http://0.0.0.0:8080/api/houses` before the request is forwarded your service.\nDefaults to `/`.\nThis is only applicable if `path` is given.\n+message=Should begin and end with a forward slash (/). Each part can can contain alphabets, digits and hypen, must begin and end with an alphanumeric characters. Parts should be separated by forward slashes (/)",
+    )
+    auth: Optional[Union[BasicAuthCreds, JwtAuthCreds, TrueFoundryInteractiveLogin]] = (
+        Field(None, description="+usage=Authentication method for inbound traffic")
+    )
+
+
 class PythonTaskConfig(BaseModel):
     """
     +docs=Describes the configuration for the python function task
@@ -1582,17 +1634,83 @@ class PythonTaskConfig(BaseModel):
     )
 
 
+class RStudio(BaseWorkbenchInput):
+    """
+    +docs=Describes the configuration for the Rstudio server
+    """
+
+    type: Literal["rstudio"] = Field(..., description="+value=rstudio")
+    image: WorkbenchImage
+
+
 class SSHServer(BaseWorkbenchInput):
     """
     +docs=Describes the configuration for the ssh server
     """
 
-    type: Literal["ssh-server"] = Field(..., description="+value=SSH Server")
+    type: Literal["ssh-server"] = Field(..., description="+value=ssh-server")
     image: WorkbenchImage
     ssh_public_key: str = Field(
         ...,
         description="+label: SSH Public Key\n+usage=Add Your SSH Public Key, this will be used to authenticate you to the SSH Server.  \\\nYou can find it using `cat ~/.ssh/id_rsa.pub` in Mac/Linux or `type $home\\.ssh\\id_rsa.pub` in Windows Powershell.  \\\nYou can also generate a new SSH key pair using `ssh-keygen -t rsa` in your local terminal. (same for both Mac/Linux and Windows Powershell)\n+uiType=TextArea\n+sort=4",
     )
+    cull_timeout: Optional[conint(ge=5)] = Field(
+        None,
+        description="+label=Stop after (minutes of inactivity)\n+usage=Stop the SSH Server instance after this much time in minutes of inactivity. \\\nThe instance is considered active if there is at least one active SSH connection (a client connected to the SSH server), \\\nor if a background job is running using tmux or screen, or if the pod has restarted.\n+sort=5",
+    )
+
+
+class SparkExecutorConfig(BaseModel):
+    """
+    +label=Executor Config
+    """
+
+    instances: Union[SparkExecutorFixedInstances, SparkExecutorDynamicScaling] = Field(
+        {"type": "fixed", "count": 1}, description="+label=Executor Instances"
+    )
+    resources: Optional[Resources] = None
+
+
+class SparkJob(BaseModel):
+    type: Literal["spark-job"] = Field(..., description="+value=spark-job\n+sort=1")
+    name: constr(regex=r"^[a-z][a-z0-9\-]{1,30}[a-z0-9]$") = Field(
+        ...,
+        description="+label=Name\n+usage=Name of the job\n+message=3 to 32 lower case characters long alphanumeric word, may contain - in between, cannot start with a number\n+sort=2",
+    )
+    image: Image
+    spark_version: str = Field(
+        "3.5.2",
+        description="+label=Spark Version\n+usage=Spark version should match the spark version installed in the image.\n+sort=2000",
+    )
+    main_application_file: str = Field(
+        ...,
+        description="+label=Main Application File\n+usage=The main application file to be executed by the spark job.\n+sort=3000",
+    )
+    arguments: Optional[str] = Field(
+        None,
+        description="+label=Arguments\n+usage=Arguments to be passed to the main application file.\n+sort=4000",
+    )
+    driver_config: SparkDriverConfig
+    executor_config: SparkExecutorConfig
+    env: Optional[Dict[str, Any]] = Field(
+        None,
+        description="+label=Environment Variables\n+usage=Configure environment variables to be injected in the service either as plain text. [Docs](https://docs.truefoundry.com/docs/env-variables)\n+icon=fa-globe\n+sort=21000",
+    )
+    mounts: Optional[List[VolumeMount]] = Field(
+        None,
+        description="+label=Mounts\n+usage=Configure volumes to be mounted to driver and executors. [Docs](https://docs.truefoundry.com/docs/mounting-volumes-job)\n+sort=22000\n+uiType=Mounts",
+    )
+    retries: conint(ge=0, le=10) = Field(
+        0,
+        description="+label=Retries\n+usage=Specify the maximum number of attempts to retry a job before it is marked as failed.\n+icon=fa-repeat\n+sort=23000",
+    )
+    service_account: Optional[str] = Field(
+        None, description="+label=Service Account\n+sort=24000"
+    )
+    workspace_fqn: Optional[str] = Field(
+        None,
+        description="+label=Workspace FQN\n+docs=Fully qualified name of the workspace\n+uiType=Hidden",
+    )
 
 
 class Volume(BaseModel):
@@ -1780,8 +1898,10 @@ class Application(BaseModel):
         Notebook,
         Codeserver,
         SSHServer,
+        RStudio,
         Helm,
         Volume,
         ApplicationSet,
         Workflow,
+        SparkJob,
     ]

{truefoundry-0.5.3rc5 → truefoundry-0.5.5}/truefoundry/deploy/lib/clients/servicefoundry_client.py

@@ -4,7 +4,7 @@ import json
 import os
 import time
 from datetime import datetime, timezone
-from typing import TYPE_CHECKING, Any, Dict, List, Literal, Optional
+from typing import TYPE_CHECKING, Any, Dict, List, Optional
 from urllib.parse import urljoin
 
 import requests
@@ -43,7 +43,6 @@ from truefoundry.deploy.v2.lib.models import (
     BuildResponse,
     DeploymentFqnResponse,
 )
-from truefoundry.gateway.lib.entities import ProviderModels
 from truefoundry.logger import logger
 from truefoundry.pydantic_v1 import parse_obj_as
 
@@ -708,25 +707,3 @@ class ServiceFoundryServiceClient(BaseServiceFoundryServiceClient):
         )
         response_data = request_handling(response)
         return response_data
-
-    def get_gateway_models(
-        self, model_type: Optional[Literal["chat", "completion", "embedding"]] = None
-    ) -> ProviderModels:
-        url = f"{self._api_server_url}/{VERSION_PREFIX}/llm-gateway/model/enabled"
-        params = {}
-        if model_type:
-            params["modelType"] = model_type
-        response = session_with_retries().get(
-            url, headers=self._get_header(), params=params
-        )
-        response = request_handling(response)
-        return parse_obj_as(ProviderModels, response)
-
-    def create_pat(self, name: str):
-        url = f"{self._api_server_url}/{VERSION_PREFIX}/service-account/pat"
-        body = {"name": name}
-        response = session_with_retries().post(
-            url, headers=self._get_header(), json=body
-        )
-        response = request_handling(response)
-        return response

{truefoundry-0.5.3rc5 → truefoundry-0.5.5}/truefoundry/deploy/lib/session.py

@@ -8,6 +8,7 @@ from truefoundry.common.constants import TFY_API_KEY_ENV_KEY, TFY_HOST_ENV_KEY
 from truefoundry.common.credential_file_manager import CredentialsFileManager
 from truefoundry.common.credential_provider import EnvCredentialProvider
 from truefoundry.common.entities import CredentialsFileContent, Token
+from truefoundry.common.session import Session
 from truefoundry.common.utils import relogin_error_message, resolve_tfy_host
 from truefoundry.deploy.io.output_callback import OutputCallBack
 from truefoundry.deploy.lib.const import (
@@ -98,6 +99,12 @@ def logout(
             output_hook.print_line(PROMPT_ALREADY_LOGGED_OUT)
 
 
+def get_access_token():
+    # Create a new session which handles token refresh
+    session = Session.new()
+    return session.access_token
+
+
 def _login_with_device_code(
     base_url: str,
     auth_service: AuthServiceClient,

{truefoundry-0.5.3rc5 → truefoundry-0.5.5}/truefoundry/deploy/v2/lib/deploy.py

@@ -74,10 +74,12 @@ def _handle_if_local_source(component: Component, workspace_fqn: str) -> Compone
         else:
             # We'll build image on TrueFoundry servers, upload the source and update image.build_source
             logger.info("Uploading code for %s '%s'", component.type, component.name)
+            client = ServiceFoundryServiceClient()
             new_component.image.build_source = local_source_to_remote_source(
                 local_source=component.image.build_source,
                 workspace_fqn=workspace_fqn,
                 component_name=component.name,
+                upload_code_package=client.upload_code_package,
             )
             logger.debug("Uploaded code for %s '%s'", component.type, component.name)
         return new_component

{truefoundry-0.5.3rc5 → truefoundry-0.5.5}/truefoundry/deploy/v2/lib/deployable_patched_models.py

@@ -30,6 +30,10 @@ class Job(models.Job, DeployablePatchedModelBase):
     resources: models.Resources = Field(default_factory=models.Resources)
 
 
+class SparkJob(models.SparkJob, DeployablePatchedModelBase):
+    type: Literal["spark-job"] = "spark-job"
+
+
 class Notebook(models.Notebook, DeployablePatchedModelBase):
     type: Literal["notebook"] = "notebook"
     resources: models.Resources = Field(default_factory=models.Resources)
@@ -37,6 +41,12 @@ class Notebook(models.Notebook, DeployablePatchedModelBase):
 
 class Codeserver(models.Codeserver, DeployablePatchedModelBase):
     type: Literal["codeserver"] = "codeserver"
+    resources: models.Resources = Field(default_factory=models.Resources)
+
+
+class RStudio(models.RStudio, DeployablePatchedModelBase):
+    type: Literal["rstudio"] = "rstudio"
+    resources: models.Resources = Field(default_factory=models.Resources)
 
 
 class Helm(models.Helm, DeployablePatchedModelBase):

{truefoundry-0.5.3rc5 → truefoundry-0.5.5}/truefoundry/deploy/v2/lib/patched_models.py

@@ -256,6 +256,10 @@ class JwtAuthCreds(models.JwtAuthCreds, PatchedModelBase):
     type: Literal["jwt_auth"] = "jwt_auth"
 
 
+class TrueFoundryInteractiveLogin(models.TrueFoundryInteractiveLogin, PatchedModelBase):
+    type: Literal["truefoundry_oauth"] = "truefoundry_oauth"
+
+
 class HealthProbe(models.HealthProbe, PatchedModelBase):
     pass
 
@@ -504,6 +508,10 @@ class TruefoundryArtifactSource(models.TruefoundryArtifactSource, PatchedModelBa
     type: Literal["truefoundry-artifact"] = "truefoundry-artifact"
 
 
+# TODO: When we do the renaming in CUE, we will make TruefoundryArtifactSource the alias and mark it deprecated
+TrueFoundryArtifactSource = TruefoundryArtifactSource
+
+
 class ArtifactsDownload(models.ArtifactsDownload, PatchedModelBase):
     pass
 

{truefoundry-0.5.3rc5 → truefoundry-0.5.5}/truefoundry/deploy/v2/lib/source.py

@@ -136,6 +136,7 @@ def local_source_to_remote_source(
     local_source: models.LocalSource,
     workspace_fqn: str,
     component_name: str,
+    upload_code_package: Callable[[str, str, str], str],
 ) -> RemoteSource:
     with tempfile.TemporaryDirectory() as local_dir:
         package_local_path = os.path.join(local_dir, "build.tar.gz")
@@ -164,8 +165,7 @@ def local_source_to_remote_source(
             logger.exception("Failed to calculate code archive size")
 
         logger.debug("Uploading code archive.")
-        client = ServiceFoundryServiceClient()
-        remote_uri = client.upload_code_package(
+        remote_uri = upload_code_package(
             workspace_fqn=workspace_fqn,
             component_name=component_name,
             package_local_path=package_local_path,

{truefoundry-0.5.3rc5 → truefoundry-0.5.5}/truefoundry/ml/autogen/client/__init__.py

@@ -184,8 +184,8 @@ from truefoundry.ml.autogen.client.models.finalize_artifact_version_request_dto
     FinalizeArtifactVersionRequestDto,
 )
 from truefoundry.ml.autogen.client.models.framework import Framework
-from truefoundry.ml.autogen.client.models.get_artifact_tags_response_dto import (
-    GetArtifactTagsResponseDto,
+from truefoundry.ml.autogen.client.models.get_artifact_version_aliases_response_dto import (
+    GetArtifactVersionAliasesResponseDto,
 )
 from truefoundry.ml.autogen.client.models.get_experiment_response_dto import (
     GetExperimentResponseDto,

{truefoundry-0.5.3rc5 → truefoundry-0.5.5}/truefoundry/ml/autogen/client/api/mlfoundry_artifacts_api.py

@@ -79,8 +79,8 @@ from truefoundry.ml.autogen.client.models.export_deployment_files_request_dto im
 from truefoundry.ml.autogen.client.models.finalize_artifact_version_request_dto import (
     FinalizeArtifactVersionRequestDto,
 )
-from truefoundry.ml.autogen.client.models.get_artifact_tags_response_dto import (
-    GetArtifactTagsResponseDto,
+from truefoundry.ml.autogen.client.models.get_artifact_version_aliases_response_dto import (
+    GetArtifactVersionAliasesResponseDto,
 )
 from truefoundry.ml.autogen.client.models.get_signed_url_for_dataset_write_request_dto import (
     GetSignedURLForDatasetWriteRequestDto,
@@ -5342,15 +5342,15 @@ class MlfoundryArtifactsApi:
         )
 
     @validate_arguments
-    def get_tags_for_artifact_get(
+    def get_version_aliases_for_artifact_get(
         self, artifact_id: StrictStr, **kwargs
-    ) -> GetArtifactTagsResponseDto:  # noqa: E501
-        """Get Tags For Artifact  # noqa: E501
+    ) -> GetArtifactVersionAliasesResponseDto:  # noqa: E501
+        """Get Version Aliases For Artifact  # noqa: E501
 
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
 
-        >>> thread = api.get_tags_for_artifact_get(artifact_id, async_req=True)
+        >>> thread = api.get_version_aliases_for_artifact_get(artifact_id, async_req=True)
         >>> result = thread.get()
 
         :param artifact_id: (required)
@@ -5364,24 +5364,26 @@ class MlfoundryArtifactsApi:
         :return: Returns the result object.
                  If the method is called asynchronously,
                  returns the request thread.
-        :rtype: GetArtifactTagsResponseDto
+        :rtype: GetArtifactVersionAliasesResponseDto
         """
         kwargs["_return_http_data_only"] = True
         if "_preload_content" in kwargs:
-            message = "Error! Please call the get_tags_for_artifact_get_with_http_info method with `_preload_content` instead and obtain raw data from ApiResponse.raw_data"  # noqa: E501
+            message = "Error! Please call the get_version_aliases_for_artifact_get_with_http_info method with `_preload_content` instead and obtain raw data from ApiResponse.raw_data"  # noqa: E501
             raise ValueError(message)
-        return self.get_tags_for_artifact_get_with_http_info(artifact_id, **kwargs)  # noqa: E501
+        return self.get_version_aliases_for_artifact_get_with_http_info(
+            artifact_id, **kwargs
+        )  # noqa: E501
 
     @validate_arguments
-    def get_tags_for_artifact_get_with_http_info(
+    def get_version_aliases_for_artifact_get_with_http_info(
         self, artifact_id: StrictStr, **kwargs
     ) -> ApiResponse:  # noqa: E501
-        """Get Tags For Artifact  # noqa: E501
+        """Get Version Aliases For Artifact  # noqa: E501
 
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
 
-        >>> thread = api.get_tags_for_artifact_get_with_http_info(artifact_id, async_req=True)
+        >>> thread = api.get_version_aliases_for_artifact_get_with_http_info(artifact_id, async_req=True)
         >>> result = thread.get()
 
         :param artifact_id: (required)
@@ -5408,7 +5410,7 @@ class MlfoundryArtifactsApi:
         :return: Returns the result object.
                  If the method is called asynchronously,
                  returns the request thread.
-        :rtype: tuple(GetArtifactTagsResponseDto, status_code(int), headers(HTTPHeaderDict))
+        :rtype: tuple(GetArtifactVersionAliasesResponseDto, status_code(int), headers(HTTPHeaderDict))
         """
 
         _params = locals()
@@ -5431,7 +5433,7 @@ class MlfoundryArtifactsApi:
             if _key not in _all_params:
                 raise ApiTypeError(
                     "Got an unexpected keyword argument '%s'"
-                    " to method get_tags_for_artifact_get" % _key
+                    " to method get_version_aliases_for_artifact_get" % _key
                 )
             _params[_key] = _val
         del _params["kwargs"]
@@ -5462,12 +5464,12 @@ class MlfoundryArtifactsApi:
         _auth_settings = ["HTTPBearer", "APIKeyCookie"]  # noqa: E501
 
         _response_types_map = {
-            "200": "GetArtifactTagsResponseDto",
+            "200": "GetArtifactVersionAliasesResponseDto",
             "422": "HTTPValidationError",
         }
 
         return self.api_client.call_api(
-            "/api/2.0/mlflow/mlfoundry-artifacts/artifacts/get-tags",
+            "/api/2.0/mlflow/mlfoundry-artifacts/artifacts/get-version-aliases",
             "GET",
             _path_params,
             _query_params,

{truefoundry-0.5.3rc5 → truefoundry-0.5.5}/truefoundry/ml/autogen/client/models/__init__.py

@@ -155,8 +155,8 @@ from truefoundry.ml.autogen.client.models.finalize_artifact_version_request_dto
     FinalizeArtifactVersionRequestDto,
 )
 from truefoundry.ml.autogen.client.models.framework import Framework
-from truefoundry.ml.autogen.client.models.get_artifact_tags_response_dto import (
-    GetArtifactTagsResponseDto,
+from truefoundry.ml.autogen.client.models.get_artifact_version_aliases_response_dto import (
+    GetArtifactVersionAliasesResponseDto,
 )
 from truefoundry.ml.autogen.client.models.get_experiment_response_dto import (
     GetExperimentResponseDto,