truefoundry 0.5.3rc5__py3-none-any.whl → 0.5.5__py3-none-any.whl

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/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/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/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/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/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/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/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/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/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/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,

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

@@ -24,7 +24,6 @@ from truefoundry.pydantic_v1 import (
     Field,
     StrictStr,
     conint,
-    conlist,
     constr,
     validator,
 )
@@ -43,16 +42,35 @@ class ArtifactVersionManifest(BaseModel):
         default=...,
         description="+label=Metadata +docs=Metadata for the artifact or model version +usage=Metadata for the artifact or model version +uiType=JsonInput",
     )
-    tags: Optional[conlist(constr(strict=True, max_length=128))] = Field(
+    version_alias: Optional[constr(strict=True, max_length=128)] = Field(
         default=None,
-        description="+label=Tags +usage=Tags for the artifact or model version +docs=Tags for the artifact or model version",
+        description="+label=Version Alias +usage=The version alias for artifact or model version which should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v-prod, v-dev, etc) +docs=The version alias for artifact or model version which should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v-prod, v-dev, etc) +message=The version alias should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v-prod, v-dev, etc)",
     )
     type: Optional[StrictStr] = "artifact-version"
     source: Source = Field(...)
     step: Optional[conint(strict=True, ge=0)] = Field(
         default=0, description="+label=Step"
     )
-    __properties = ["description", "metadata", "tags", "type", "source", "step"]
+    __properties = [
+        "description",
+        "metadata",
+        "version_alias",
+        "type",
+        "source",
+        "step",
+    ]
+
+    @validator("version_alias")
+    def version_alias_validate_regular_expression(cls, value):
+        """Validates the regular expression"""
+        if value is None:
+            return value
+
+        if not re.match(r"^v[a-zA-Z0-9.-]*([a-zA-Z0-9]+)$", value):
+            raise ValueError(
+                r"must validate the regular expression /^v[a-zA-Z0-9.-]*([a-zA-Z0-9]+)$/"
+            )
+        return value
 
     @validator("type")
     def type_validate_enum(cls, value):
@@ -104,7 +122,7 @@ class ArtifactVersionManifest(BaseModel):
             {
                 "description": obj.get("description"),
                 "metadata": obj.get("metadata"),
-                "tags": obj.get("tags"),
+                "version_alias": obj.get("version_alias"),
                 "type": obj.get("type")
                 if obj.get("type") is not None
                 else "artifact-version",

truefoundry/ml/autogen/client/models/{get_artifact_tags_response_dto.py → get_artifact_version_aliases_response_dto.py}

@@ -20,13 +20,13 @@ import re  # noqa: F401
 from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, conlist
 
 
-class GetArtifactTagsResponseDto(BaseModel):
+class GetArtifactVersionAliasesResponseDto(BaseModel):
     """
-    GetArtifactTagsResponseDto
+    GetArtifactVersionAliasesResponseDto
     """
 
-    tags: conlist(StrictStr) = Field(...)
-    __properties = ["tags"]
+    version_aliases: conlist(StrictStr) = Field(...)
+    __properties = ["version_aliases"]
 
     class Config:
         """Pydantic configuration"""
@@ -43,8 +43,8 @@ class GetArtifactTagsResponseDto(BaseModel):
         return json.dumps(self.to_dict())
 
     @classmethod
-    def from_json(cls, json_str: str) -> GetArtifactTagsResponseDto:
-        """Create an instance of GetArtifactTagsResponseDto from a JSON string"""
+    def from_json(cls, json_str: str) -> GetArtifactVersionAliasesResponseDto:
+        """Create an instance of GetArtifactVersionAliasesResponseDto from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self):
@@ -53,13 +53,15 @@ class GetArtifactTagsResponseDto(BaseModel):
         return _dict
 
     @classmethod
-    def from_dict(cls, obj: dict) -> GetArtifactTagsResponseDto:
-        """Create an instance of GetArtifactTagsResponseDto from a dict"""
+    def from_dict(cls, obj: dict) -> GetArtifactVersionAliasesResponseDto:
+        """Create an instance of GetArtifactVersionAliasesResponseDto from a dict"""
         if obj is None:
             return None
 
         if not isinstance(obj, dict):
-            return GetArtifactTagsResponseDto.parse_obj(obj)
+            return GetArtifactVersionAliasesResponseDto.parse_obj(obj)
 
-        _obj = GetArtifactTagsResponseDto.parse_obj({"tags": obj.get("tags")})
+        _obj = GetArtifactVersionAliasesResponseDto.parse_obj(
+            {"version_aliases": obj.get("version_aliases")}
+        )
         return _obj

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

@@ -28,7 +28,6 @@ from truefoundry.pydantic_v1 import (
     Field,
     StrictStr,
     conint,
-    conlist,
     constr,
     validator,
 )
@@ -47,9 +46,9 @@ class ModelVersionManifest(BaseModel):
         default=...,
         description="+label=Metadata +docs=Metadata for the artifact or model version +usage=Metadata for the artifact or model version +uiType=JsonInput",
     )
-    tags: Optional[conlist(constr(strict=True, max_length=128))] = Field(
+    version_alias: Optional[constr(strict=True, max_length=128)] = Field(
         default=None,
-        description="+label=Tags +usage=Tags for the artifact or model version +docs=Tags for the artifact or model version",
+        description="+label=Version Alias +usage=The version alias for artifact or model version which should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v-prod, v-dev, etc) +docs=The version alias for artifact or model version which should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v-prod, v-dev, etc) +message=The version alias should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v-prod, v-dev, etc)",
     )
     type: Optional[StrictStr] = "model-version"
     source: Source1 = Field(...)
@@ -61,7 +60,7 @@ class ModelVersionManifest(BaseModel):
     __properties = [
         "description",
         "metadata",
-        "tags",
+        "version_alias",
         "type",
         "source",
         "framework",
@@ -69,6 +68,18 @@ class ModelVersionManifest(BaseModel):
         "step",
     ]
 
+    @validator("version_alias")
+    def version_alias_validate_regular_expression(cls, value):
+        """Validates the regular expression"""
+        if value is None:
+            return value
+
+        if not re.match(r"^v[a-zA-Z0-9.-]*([a-zA-Z0-9]+)$", value):
+            raise ValueError(
+                r"must validate the regular expression /^v[a-zA-Z0-9.-]*([a-zA-Z0-9]+)$/"
+            )
+        return value
+
     @validator("type")
     def type_validate_enum(cls, value):
         """Validates the enum"""
@@ -125,7 +136,7 @@ class ModelVersionManifest(BaseModel):
             {
                 "description": obj.get("description"),
                 "metadata": obj.get("metadata"),
-                "tags": obj.get("tags"),
+                "version_alias": obj.get("version_alias"),
                 "type": obj.get("type")
                 if obj.get("type") is not None
                 else "model-version",

truefoundry/ml/autogen/client_README.md

@@ -121,7 +121,7 @@ Class | Method | HTTP request | Description
 *MlfoundryArtifactsApi* | [**get_signed_urls_for_dataset_write_post**](truefoundry/ml/autogen/client/docs/MlfoundryArtifactsApi.md#get_signed_urls_for_dataset_write_post) | **POST** /api/2.0/mlflow/mlfoundry-artifacts/datasets/get-signed-urls-for-write | Get Signed Urls For Dataset Write
 *MlfoundryArtifactsApi* | [**get_signed_urls_for_read_post**](truefoundry/ml/autogen/client/docs/MlfoundryArtifactsApi.md#get_signed_urls_for_read_post) | **POST** /api/2.0/mlflow/mlfoundry-artifacts/artifact-versions/get-signed-urls-for-read | Get Signed Urls For Read
 *MlfoundryArtifactsApi* | [**get_signed_urls_for_write_post**](truefoundry/ml/autogen/client/docs/MlfoundryArtifactsApi.md#get_signed_urls_for_write_post) | **POST** /api/2.0/mlflow/mlfoundry-artifacts/artifact-versions/get-signed-urls-for-write | Get Signed Urls For Write
-*MlfoundryArtifactsApi* | [**get_tags_for_artifact_get**](truefoundry/ml/autogen/client/docs/MlfoundryArtifactsApi.md#get_tags_for_artifact_get) | **GET** /api/2.0/mlflow/mlfoundry-artifacts/artifacts/get-tags | Get Tags For Artifact
+*MlfoundryArtifactsApi* | [**get_version_aliases_for_artifact_get**](truefoundry/ml/autogen/client/docs/MlfoundryArtifactsApi.md#get_version_aliases_for_artifact_get) | **GET** /api/2.0/mlflow/mlfoundry-artifacts/artifacts/get-version-aliases | Get Version Aliases For Artifact
 *MlfoundryArtifactsApi* | [**list_artifact_versions_post**](truefoundry/ml/autogen/client/docs/MlfoundryArtifactsApi.md#list_artifact_versions_post) | **POST** /api/2.0/mlflow/mlfoundry-artifacts/artifact-versions/list | List Artifact Versions
 *MlfoundryArtifactsApi* | [**list_artifacts_post**](truefoundry/ml/autogen/client/docs/MlfoundryArtifactsApi.md#list_artifacts_post) | **POST** /api/2.0/mlflow/mlfoundry-artifacts/artifacts/list | List Artifacts
 *MlfoundryArtifactsApi* | [**list_datasets_post**](truefoundry/ml/autogen/client/docs/MlfoundryArtifactsApi.md#list_datasets_post) | **POST** /api/2.0/mlflow/mlfoundry-artifacts/datasets/list | List Datasets
@@ -220,7 +220,7 @@ Class | Method | HTTP request | Description
  - [FileInfoDto](truefoundry/ml/autogen/client/docs/FileInfoDto.md)
  - [FinalizeArtifactVersionRequestDto](truefoundry/ml/autogen/client/docs/FinalizeArtifactVersionRequestDto.md)
  - [Framework](truefoundry/ml/autogen/client/docs/Framework.md)
- - [GetArtifactTagsResponseDto](truefoundry/ml/autogen/client/docs/GetArtifactTagsResponseDto.md)
+ - [GetArtifactVersionAliasesResponseDto](truefoundry/ml/autogen/client/docs/GetArtifactVersionAliasesResponseDto.md)
  - [GetExperimentResponseDto](truefoundry/ml/autogen/client/docs/GetExperimentResponseDto.md)
  - [GetLatestRunLogResponseDto](truefoundry/ml/autogen/client/docs/GetLatestRunLogResponseDto.md)
  - [GetMetricHistoryResponse](truefoundry/ml/autogen/client/docs/GetMetricHistoryResponse.md)

truefoundry/ml/autogen/entities/artifacts.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  artifacts.json
-#   timestamp: 2024-12-16T06:57:32+00:00
+#   timestamp: 2025-01-03T09:11:54+00:00
 
 from __future__ import annotations
 
@@ -58,16 +58,11 @@ class BaseArtifactVersion(BaseModel):
         ...,
         description="+label=Metadata\n+docs=Metadata for the artifact or model version\n+usage=Metadata for the artifact or model version\n+uiType=JsonInput",
     )
-    tags: Optional[
-        List[
-            constr(
-                regex=r"^(?!^\d+$)(?!.*[._-]{2,})[a-zA-Z0-9]+([._-][a-zA-Z0-9]+)*$",
-                max_length=128,
-            )
-        ]
+    version_alias: Optional[
+        constr(regex=r"^v[a-zA-Z0-9.-]*([a-zA-Z0-9]+)$", max_length=128)
     ] = Field(
         None,
-        description="+label=Tags\n+usage=Tags for the artifact or model version\n+docs=Tags for the artifact or model version",
+        description="+label=Version Alias\n+usage=The version alias for artifact or model version which should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v-prod, v-dev, etc)\n+docs=The version alias for artifact or model version which should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v-prod, v-dev, etc)\n+message=The version alias should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v-prod, v-dev, etc)",
     )
 
 

truefoundry/ml/log_types/artifacts/artifact.py

@@ -84,7 +84,7 @@ class ArtifactVersion:
         self._deleted = False
         self._description: str = ""
         self._metadata: Dict[str, Any] = {}
-        self._tags: List[str] = []
+        self._version_alias: Optional[str] = None
         self._set_mutable_attrs()
 
     @classmethod
@@ -132,13 +132,13 @@ class ArtifactVersion:
             manifest = self._artifact_version.manifest.actual_instance
             self._description = manifest.description or ""
             self._metadata = copy.deepcopy(manifest.metadata)
-            self._tags = copy.deepcopy(manifest.tags or [])
+            self._version_alias = manifest.version_alias or None
         else:
             self._description = self._artifact_version.description or ""
             self._metadata = copy.deepcopy(
                 self._artifact_version.artifact_metadata or {}
             )
-            self._tags = []
+            self._version_alias = None
 
     def _refetch_artifact_version(self, reset_mutable_attrs: bool = True):
         _artifact_version = (
@@ -211,31 +211,31 @@ class ArtifactVersion:
         self._metadata = copy.deepcopy(value)
 
     @property
-    def tags(self) -> List[str]:
+    def version_alias(self) -> Optional[str]:
         """
-        Get tags for the current model version
+        Get version alias for the current artifact version
         """
         if not self._artifact_version.manifest:
             warnings.warn(
-                message="This model version was created using an older serialization format. tags do not exist, returning empty list",
+                message="This artifact version was created using an older serialization format. version alias does not exist",
                 category=TrueFoundryDeprecationWarning,
                 stacklevel=2,
             )
-        return self._tags
+        return self._version_alias
 
-    @tags.setter
-    def tags(self, value: List[str]):
+    @version_alias.setter
+    def version_alias(self, value: Optional[str]):
         """
-        Set the tags for current model version
+        Set the version alias for current artifact version
         """
         if not self._artifact_version.manifest:
             warnings.warn(
-                message="This model version was created using an older serialization format. Tags will not be updated",
+                message="This artifact version was created using an older serialization format. version alias will not be updated",
                 category=TrueFoundryDeprecationWarning,
                 stacklevel=2,
             )
             return
-        self._tags = copy.deepcopy(value)
+        self._version_alias = value
 
     @property
     def created_at(self) -> Optional[datetime.datetime]:
@@ -399,7 +399,7 @@ class ArtifactVersion:
             assert isinstance(manifest, ArtifactVersionManifest)
             manifest.description = self.description
             manifest.metadata = self.metadata
-            manifest.tags = self.tags
+            manifest.version_alias = self.version_alias
         else:
             manifest = None
         try:

truefoundry/ml/log_types/artifacts/model.py

@@ -8,7 +8,7 @@ import typing
 import uuid
 import warnings
 from pathlib import Path
-from typing import TYPE_CHECKING, Any, Dict, List, Optional, Union
+from typing import TYPE_CHECKING, Any, Dict, Optional, Union
 
 from truefoundry.common.warnings import TrueFoundryDeprecationWarning
 from truefoundry.ml.artifact.truefoundry_artifact_repo import (
@@ -109,7 +109,7 @@ class ModelVersion:
         self._metadata: Dict[str, Any] = {}
         self._environment: Optional[ModelVersionEnvironment] = None
         self._framework: Optional[ModelFrameworkType] = None
-        self._tags: List[str] = []
+        self._version_alias: Optional[str] = None
         self._set_mutable_attrs()
 
     @classmethod
@@ -160,7 +160,7 @@ class ModelVersion:
                 if manifest.framework
                 else None
             )
-            self._tags = copy.deepcopy(self._model_version.manifest.tags) or []
+            self._version_alias = self._model_version.manifest.version_alias or None
         else:
             self._description = self._model_version.description or ""
             self._metadata = copy.deepcopy(self._model_version.artifact_metadata or {})
@@ -168,7 +168,7 @@ class ModelVersion:
             self._framework = _ModelFramework.to_model_framework_type(
                 self._model_version.model_framework
             )
-            self._tags = []
+            self._version_alias = None
 
     def _refetch_model_version(self, reset_mutable_attrs: bool = True):
         _model_version = self._mlfoundry_artifacts_api.get_model_version_get(
@@ -239,31 +239,31 @@ class ModelVersion:
         self._metadata = copy.deepcopy(value)
 
     @property
-    def tags(self) -> List[str]:
+    def version_alias(self) -> Optional[str]:
         """
-        Get tags for the current artifact version
+        Get version alias for the current model version
         """
         if not self._model_version.manifest:
             warnings.warn(
-                message="This model version was created using an older serialization format. tags do not exist, returning empty list",
+                message="This model version was created using an older serialization format. version alias does not exist",
                 category=TrueFoundryDeprecationWarning,
                 stacklevel=2,
             )
-        return self._tags
+        return self._version_alias
 
-    @tags.setter
-    def tags(self, value: List[str]):
+    @version_alias.setter
+    def version_alias(self, value: Optional[str]):
         """
-        Set the tags for current artifact version
+        Set the version alias for current artifact version
         """
         if not self._model_version.manifest:
             warnings.warn(
-                message="This model version was created using an older serialization format. Tags will not be updated",
+                message="This model version was created using an older serialization format. version alias will not be updated",
                 category=TrueFoundryDeprecationWarning,
                 stacklevel=2,
             )
             return
-        self._tags = copy.deepcopy(value)
+        self._version_alias = value
 
     @property
     def environment(self) -> Optional[ModelVersionEnvironment]:
@@ -489,7 +489,7 @@ class ModelVersion:
             self._model_version.manifest.framework = (
                 Framework.from_dict(self.framework.dict()) if self.framework else None
             )
-            self._model_version.manifest.tags = self.tags
+            self._model_version.manifest.version_alias = self.version_alias
         try:
             _model_version = self._mlfoundry_artifacts_api.update_model_version_post(
                 update_model_version_request_dto=UpdateModelVersionRequestDto(

{truefoundry-0.5.3rc5.dist-info → truefoundry-0.5.5.dist-info}/METADATA

@@ -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.dist-info → truefoundry-0.5.5.dist-info}/RECORD

@@ -25,7 +25,6 @@ truefoundry/autodeploy/utils/diff.py,sha256=Ef8Y-VffDKel_-q-GxRam6gqiv8qTLMcqVg6
 truefoundry/autodeploy/utils/pydantic_compat.py,sha256=hEAUy5kLjhPdzw7yGZ2iXGMXbbMVXVlGzIofmyHafXQ,412
 truefoundry/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/cli/__main__.py,sha256=0nEwE0UkFagCO5ctGcqEThdmBT2fN8MVcr0ERItnsNQ,3483
-truefoundry/cli/commands/pat.py,sha256=0rm95ny1b3KfeBJSRWh1M4lR46ERnp96ovUZKu-P5Ss,602
 truefoundry/cli/config.py,sha256=tf8w4UfVzcC6eYkENvuuCPYt_V3sqVpO1bclORV9tAk,206
 truefoundry/cli/console.py,sha256=9-dMy4YPisCJQziRKTg8Qa0UJnOGl1soiUnJjsnLDvE,242
 truefoundry/cli/const.py,sha256=dVHPo1uAiDSSMXwXoT2mR5kNQjExT98QNVRz98Hz_Ts,510
@@ -44,8 +43,8 @@ truefoundry/common/session.py,sha256=BiTJ2WEzLkTvzpqy67xX3fXgiKH_ltzzmQdFG7xUtwI
 truefoundry/common/storage_provider_utils.py,sha256=yURhMw8k0FLFvaviRHDiifhvc6GnuQwGMC9Qd2uM440,10934
 truefoundry/common/utils.py,sha256=sZyg-vpShVXvnrwnINrdsyWJR3FHMUBiOB1CslYdQTo,6072
 truefoundry/common/warnings.py,sha256=rs6BHwk7imQYedo07iwh3TWEOywAR3Lqhj0AY4khByg,504
-truefoundry/deploy/__init__.py,sha256=p14_yIJw2bcaKW3c1DTaoM7UwfHvt2T_MfeFlZIz1g4,2359
-truefoundry/deploy/auto_gen/models.py,sha256=3_GTdaYE95piMzhbM4SvMSQhP4UGWDGpJnLxERjYHGw,82555
+truefoundry/deploy/__init__.py,sha256=e0EgG-fVJ6N9PNIlBINO6QtIMEehQgmz3ONPPjYlcSY,2596
+truefoundry/deploy/auto_gen/models.py,sha256=B1Ze90HKsMTh0telC_nyPaep1guos-hqfZr6ztAZ9LM,87574
 truefoundry/deploy/builder/__init__.py,sha256=1qjHMNBE1poRCZW0WrG46dFM1f1IlivD5352qzsioMU,4953
 truefoundry/deploy/builder/builders/__init__.py,sha256=tlFLXqyDaKLd4iZbo4Hcu_8gOmgtL6drnXpbmQ6x1P8,636
 truefoundry/deploy/builder/builders/dockerfile.py,sha256=AXXTziCkaqIhuM_bwyD1vT1znOwemN1TKgU7eyo-KuM,1522
@@ -83,7 +82,7 @@ truefoundry/deploy/io/output_callback.py,sha256=V2YwUFec4G4a67lM4r-x_64AqdOVNo_9
 truefoundry/deploy/io/rich_output_callback.py,sha256=TJLiRD-EnFVwgcepxR7WN0koKqW1X2DevETPhNPi_nU,829
 truefoundry/deploy/lib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/deploy/lib/clients/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/deploy/lib/clients/servicefoundry_client.py,sha256=l4KaHJmmTC3pP5hbnHuPhDH-MPs7KQBtTtfmmfyyBDI,27020
+truefoundry/deploy/lib/clients/servicefoundry_client.py,sha256=E1AZo3KmvvXoCQV4Z7A8Iy_iPH8HethHPR-xnAImEEE,26096
 truefoundry/deploy/lib/const.py,sha256=Wg0GDnfFu-g1fJr4lU80NH2ULw0R0dYjV7LnW-PbOeM,173
 truefoundry/deploy/lib/dao/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/deploy/lib/dao/application.py,sha256=oMszpueXPUfTUuN_XdKwoRjQyqAgWHhZ-10cbprCVdM,9226
@@ -94,28 +93,24 @@ truefoundry/deploy/lib/logs_utils.py,sha256=SQxRv3jDDmgHdOUMhlMaAPGYskybnBUMpst7
 truefoundry/deploy/lib/messages.py,sha256=nhp0bCYf_XpUM68hTq5lBY-__vtEyV2uP7NgnJXJ_Vg,925
 truefoundry/deploy/lib/model/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/deploy/lib/model/entity.py,sha256=3YRdyvMYI1aUXUu2tB5b5lZWwpxKdAhjpT79gzTCZGo,8373
-truefoundry/deploy/lib/session.py,sha256=rNMkdzZejrxq5gTYTqJ-nGm869JzdlhHzaJQvINTQ4U,4953
+truefoundry/deploy/lib/session.py,sha256=FBSq3bxf3my_613NzgtPoeeaAYLGZEoF0n7XccTFNf4,5141
 truefoundry/deploy/lib/util.py,sha256=J7r8San2wKo48A7-BlH2-OKTlBO67zlPjLEhMsL8os0,1059
 truefoundry/deploy/lib/win32.py,sha256=1RcvPTdlOAJ48rt8rCbE2Ufha2ztRqBAE9dueNXArrY,5009
 truefoundry/deploy/python_deploy_codegen.py,sha256=qJHH1BJQII9e6PhkcRFYiE_3De7_VMMm8nM4AX5Eq1o,6513
 truefoundry/deploy/v2/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/deploy/v2/lib/__init__.py,sha256=WEiVMZXOVljzEE3tpGJil14liIn_PCDoACJ6b3tZ6sI,188
-truefoundry/deploy/v2/lib/deploy.py,sha256=oK5yTlrhzKNTGZckwuBzu8Q_pxQYgBVcdA-KB-SbzCI,11990
+truefoundry/deploy/v2/lib/deploy.py,sha256=Q4CL8SKnFLQ1FRqFyIwOAf77j_rP-fv58e-F-s3GqBw,12105
 truefoundry/deploy/v2/lib/deploy_workflow.py,sha256=6Y4PNhd5PCc_4TjpcIZPkr6FCm3BNIQ_SsJTvpUu7a0,12579
-truefoundry/deploy/v2/lib/deployable_patched_models.py,sha256=MROgMxhn9hDEAKwJSWl3iz12tUVvRKzEtqF2QUT6dAk,3343
+truefoundry/deploy/v2/lib/deployable_patched_models.py,sha256=FmpZWpSbBTHe4yt0XMIPgdCNy_Rw1aXQCt8icv8TcWE,3701
 truefoundry/deploy/v2/lib/models.py,sha256=pSolLMTArDuYpeNsmeeS5DWliloN_iCDfZSpRllMHUg,1120
-truefoundry/deploy/v2/lib/patched_models.py,sha256=YDM75pdUPdMFgH2G8ZNbztalLIiMcHVHBHt7cLWmiDg,16050
-truefoundry/deploy/v2/lib/source.py,sha256=tzMSOTrTH3dJzICjo8gI8iY6n0-ckktP3Dh7bVK-nTI,9344
-truefoundry/gateway/__init__.py,sha256=HvAMuk60-fuZcMmdOtozHrUaH_JcbEHxpMSlS9ci4ls,41
-truefoundry/gateway/cli/cli.py,sha256=9raS6Nq_CNzagEPpbKICwUZ01zMoqmZINKMB9m_gJRE,908
-truefoundry/gateway/lib/entities.py,sha256=1P9YjrMoHbg58M57Pq-HlsT1ISHRyyX4JxK-o4of6cY,695
-truefoundry/gateway/lib/models.py,sha256=TXOur51tuYkq4Xil-icNVaVH-7xZbdOUyqS8N9295dg,876
+truefoundry/deploy/v2/lib/patched_models.py,sha256=VkDzYzlKpFu7yCsqdMUL44YS-pTIxHegBO87-jjttTM,16370
+truefoundry/deploy/v2/lib/source.py,sha256=wArU8AwSy7U9EUGLEMmB4Ke7ZwgZWmh8fFU-PACogIk,9347
 truefoundry/logger.py,sha256=u-YCNjg5HBwE70uQcpjIG64Ghos-K2ulTWaxC03BSj4,714
 truefoundry/ml/__init__.py,sha256=ssUEIs8BixPWxynKoeSh-dkRl6AtLXG0PBGYnUR5Az8,2217
 truefoundry/ml/artifact/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/ml/artifact/truefoundry_artifact_repo.py,sha256=myDpr3RZzEkd_UG9C3rZ4UKEFxXopO5HMVJNpNDiuAo,37737
 truefoundry/ml/autogen/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/ml/autogen/client/__init__.py,sha256=42Y8v6C1oDlngKCaDODi9k3q-9MPMiqO8jGe39y5QJM,20097
+truefoundry/ml/autogen/client/__init__.py,sha256=28MIn6pjqHZlnjuljt-kgrR7Dw07It7khbMZm3dlQlY,20118
 truefoundry/ml/autogen/client/api/__init__.py,sha256=NyMBxBmIzB1o5LzZZwz9LiydHB3-hPqc_sevsnY6Jrw,746
 truefoundry/ml/autogen/client/api/auth_api.py,sha256=zpWzJhUmW6HHMY_atlUf0B25k77E1kue2hmix5I5Ih0,7017
 truefoundry/ml/autogen/client/api/deprecated_api.py,sha256=mu5x_skNcwz8v1Tr6VP72-tVP7pmBV9muyKy_2NH3F0,38824
@@ -123,14 +118,14 @@ truefoundry/ml/autogen/client/api/experiments_api.py,sha256=mRKS8qGzcFJUpTWjfQoF
 truefoundry/ml/autogen/client/api/generate_code_snippet_api.py,sha256=C8iABP-pYIyIoTlL8PFi3ZBG_6B_dkQf09eErNeoj7A,22316
 truefoundry/ml/autogen/client/api/health_api.py,sha256=IAPhRAo9CLUT5ipVR1fCf-qDx57UR0wg5ekhtUl8lug,11554
 truefoundry/ml/autogen/client/api/metrics_api.py,sha256=q3L38eD-2hu4_9YvcSdnWDYXD2V8il-X9reinOAABek,14908
-truefoundry/ml/autogen/client/api/mlfoundry_artifacts_api.py,sha256=Ze2mSkXH1IxUB6Zs3C8ISYwzpKfPXgGvdKgjmEa-XRA,319720
+truefoundry/ml/autogen/client/api/mlfoundry_artifacts_api.py,sha256=i3OaGysy3Th-ZVqJ6HenSn7W7cQkRUDS_G5D28tk6e4,319913
 truefoundry/ml/autogen/client/api/run_artifacts_api.py,sha256=x-vVnY2LEFChZxiiFauswRWwFz6Qqh30PKXjzuTvxmc,8799
 truefoundry/ml/autogen/client/api/runs_api.py,sha256=-aghrZ2VYuZOw_vBtOzWDsnK7Ji29oZQxK2CLRgyo2w,119232
 truefoundry/ml/autogen/client/api_client.py,sha256=M31IycWorn10FYS8WbO_Whaipr0tVu2pVWFCfbS7XCo,29167
 truefoundry/ml/autogen/client/api_response.py,sha256=KRyvecPMXF05PaxILHZ8JHoP4rgKBjKONMgG83aU-rM,844
 truefoundry/ml/autogen/client/configuration.py,sha256=V1oaEnxt-NfpaNmp-EZpf2glovzVhM2coWYt8HBNB4M,15723
 truefoundry/ml/autogen/client/exceptions.py,sha256=XbCbDHhYT3BVejdoGNPgEa4oS56ypkwFdxk1iOc_tFY,5355
-truefoundry/ml/autogen/client/models/__init__.py,sha256=ZtmxIyFDWPGywDB8p-VZO73Cl_WkGcl4fVzEB9UHsTc,18719
+truefoundry/ml/autogen/client/models/__init__.py,sha256=m2SztLLPzl_aifXj8-aFFl_gqkArLijw0ZLabOFOdCM,18740
 truefoundry/ml/autogen/client/models/add_custom_metrics_to_model_version_request_dto.py,sha256=_ISDspicTGjBCYYXubKfRYYSSQVyW3AvG-jFh47-Zfc,2163
 truefoundry/ml/autogen/client/models/add_features_to_model_version_request_dto.py,sha256=OT1-98DyWNfAHz_EgD2gMX2XkrGQ4Re945fhoAl8qSE,2099
 truefoundry/ml/autogen/client/models/agent.py,sha256=lUAbs092yo9hZmqzX6atKebDNf1uEwTf5jLvjOdvoeM,3872
@@ -142,7 +137,7 @@ truefoundry/ml/autogen/client/models/artifact_dto.py,sha256=JWL8_sITJJXbxywcfRj4
 truefoundry/ml/autogen/client/models/artifact_response_dto.py,sha256=T_HLjkHnqiO1bnSXiCUOlwDEHGz89mlUNhBq0SJTapo,2226
 truefoundry/ml/autogen/client/models/artifact_type.py,sha256=LBXGYUoesmL5J-0xzCR8jBST3bjb6HY86RmqSGvytRs,828
 truefoundry/ml/autogen/client/models/artifact_version_dto.py,sha256=GXjjKycpsEgJ60V2hAfL4NMxM2gZKLP2qV2yBb9WgdM,6174
-truefoundry/ml/autogen/client/models/artifact_version_manifest.py,sha256=lzoIGTRplPIcXlO5Brq6DoCn1HCnzmxJmd9dPvZ3Nq4,3907
+truefoundry/ml/autogen/client/models/artifact_version_manifest.py,sha256=AOg17pPxbOuqmSLCBj_us4_heijpr_k-CZWnl87RExw,4829
 truefoundry/ml/autogen/client/models/artifact_version_response_dto.py,sha256=JmiSYdPYG-ki6LCBeF7NtihdQ2LPiHld8cSPivY2GmQ,2421
 truefoundry/ml/autogen/client/models/artifact_version_serialization_format.py,sha256=2Qfgq1nykn0_mKNEJzOxhQwHm_95WIr9wJcHLeFWXQM,774
 truefoundry/ml/autogen/client/models/artifact_version_status.py,sha256=iIcomqhuaJkZH2rT3e6IjQutM79t9hEshW-RCIMKHEY,780
@@ -193,7 +188,7 @@ truefoundry/ml/autogen/client/models/fast_ai_framework.py,sha256=GwVoMbIzYhr4srF
 truefoundry/ml/autogen/client/models/file_info_dto.py,sha256=7oc8venicsFVk8zT9wHNhHnZGtFkFlqimFnS7ozGL9k,2156
 truefoundry/ml/autogen/client/models/finalize_artifact_version_request_dto.py,sha256=tkPcNjI0_qhX-D3m6AXqLzdmo9_K0tFBtbPDeg0Zo7U,3818
 truefoundry/ml/autogen/client/models/framework.py,sha256=W_KrqH1qKRvlrssCCWFIlmPaeAw0eiM6VqS3pbdlMXU,14720
-truefoundry/ml/autogen/client/models/get_artifact_tags_response_dto.py,sha256=zS5eBxBqzm-f5dt4GspLJvVmq_RWTtZmZTXFV8rqbX0,1883
+truefoundry/ml/autogen/client/models/get_artifact_version_aliases_response_dto.py,sha256=W064ced8EQCeyJLpKikEtkCzuiqUDwNjS56sh7Mc3yg,2029
 truefoundry/ml/autogen/client/models/get_experiment_response_dto.py,sha256=FhJpoUeRbZV1s1KR3Y07Kv6U4enLbbgdpGbguT9us0A,2923
 truefoundry/ml/autogen/client/models/get_latest_run_log_response_dto.py,sha256=g5NG252VzjGHiA-w5nhxmZxGrowJO9XCHDQTjf8y2KE,2266
 truefoundry/ml/autogen/client/models/get_metric_history_response.py,sha256=rTchO8o4sjGsNaZj8TW-VGtGGxsOBjUGzgaAWNtBZ-k,2428
@@ -252,7 +247,7 @@ truefoundry/ml/autogen/client/models/model_response_dto.py,sha256=osrTxfygkuhxWj
 truefoundry/ml/autogen/client/models/model_server.py,sha256=N5mcWmnX0KzTMtTH8A9h_FmiQcMld3B8CXD_UYeDKaA,700
 truefoundry/ml/autogen/client/models/model_version_dto.py,sha256=9Mh4d8V33gEGYr4sys5lYYDA3bHKajR3oAZLZMAy_io,7094
 truefoundry/ml/autogen/client/models/model_version_environment.py,sha256=wTayQjMen24eJCqXQugzoXLANY5NXdVsAvZSh-QIhJA,2827
-truefoundry/ml/autogen/client/models/model_version_manifest.py,sha256=pZysfaTZmew4MxhtH4MTnwCQWRt43JvqU3ii5gBgm9w,4955
+truefoundry/ml/autogen/client/models/model_version_manifest.py,sha256=UrusTkY39LBGpVfjIt0xG-zbnSIa6zVfqzxMXOQGOPI,5822
 truefoundry/ml/autogen/client/models/model_version_response_dto.py,sha256=D6XOCyggxqTkbePuypqYSHYh1PYeDv7R_J32q61sDM0,2320
 truefoundry/ml/autogen/client/models/multi_part_upload_dto.py,sha256=Ckq405vud8RQmMyKCJQJBlW5iXO7Y2mlgo8eVkiMvxg,3738
 truefoundry/ml/autogen/client/models/multi_part_upload_response_dto.py,sha256=VjH0kvl7rMjgDHjYGHnsh7KsZ5-qn-k3ksdGLJ49nIM,2431
@@ -313,8 +308,8 @@ truefoundry/ml/autogen/client/models/xg_boost_framework.py,sha256=a0EHzncxe_hSFU
 truefoundry/ml/autogen/client/models/xg_boost_model_schema.py,sha256=QKJkiERNqF8UiolX2FjvhkUkUlUgDoo3CpTYPAdCxV8,2735
 truefoundry/ml/autogen/client/models/xg_boost_serialization_format.py,sha256=wFFSH0Z9MP8eT_fysHfUu4iqdjEZART-HxTH4e4gNI8,872
 truefoundry/ml/autogen/client/rest.py,sha256=9goba8qHjQuVx5O_yRaTKu7PvBnb7r7swfy3dwuTEgk,14281
-truefoundry/ml/autogen/client_README.md,sha256=Dr3x2Lm10JIWsLlRBZuDsE4Y4T1XzIpsychLjZfr89Q,37262
-truefoundry/ml/autogen/entities/artifacts.py,sha256=COkzHpmqzXpEttH2UstIO8p7WN3zz_tyAG63r66i-9w,20652
+truefoundry/ml/autogen/client_README.md,sha256=qrAp9e6D4rKZETpvJry0U8Yn2ZkxDHMA9NUNsfmO_JY,37326
+truefoundry/ml/autogen/entities/artifacts.py,sha256=jLBQk5OYpdhXwwLEIAQLkT-ttZbg522FBWHvgwOCQGQ,20995
 truefoundry/ml/autogen/models/__init__.py,sha256=--TGRea9SQBWFfwtcl3ekb1XGfFTdEkQGSG8a2SJ60I,187
 truefoundry/ml/autogen/models/exceptions.py,sha256=q3n7FGBrg_hUy1UyoefhMnhcXUAaqXlLIGHoOVzn_d8,1390
 truefoundry/ml/autogen/models/schema.py,sha256=IhpO9qbygLqEamP3NIt3m90SseJXCOm1ZTqNbNbW-M0,55772
@@ -335,11 +330,11 @@ truefoundry/ml/exceptions.py,sha256=QpDJSWmF7dIsByS0qOQbQZ_jytdNTzkHDDO3BxhTSo0,
 truefoundry/ml/git_info.py,sha256=jvAVm9ilqivnGq8qJdUvYdd8Siv0PLtqurB-PXsS5ho,2023
 truefoundry/ml/internal_namespace.py,sha256=QcqMHp6-C2im2H_02hlhi01EIcr1HhNaZprszs13EMU,1790
 truefoundry/ml/log_types/__init__.py,sha256=g4u4D4Jaj0aBK5GtrLV88-qThKZR9pSZ17vFEkN-LmM,125
-truefoundry/ml/log_types/artifacts/artifact.py,sha256=qz657N5WosBNBqqEtZ9hWatvIoa9_0MipudYH9r3Bq0,21657
+truefoundry/ml/log_types/artifacts/artifact.py,sha256=b7GXYQx6MVI5qnPIS96xGqwwWVKGY4LYrEU9pCKkU_Y,21772
 truefoundry/ml/log_types/artifacts/constants.py,sha256=qKxQ5mMvJE4j83BvGW3qNTKunxCiBg_EEjTdgbgJtyE,1036
 truefoundry/ml/log_types/artifacts/dataset.py,sha256=a4dxd2EN8p7Ci-cLGGiDOboN3t0395_XhWE1dmTw1Q4,13112
 truefoundry/ml/log_types/artifacts/general_artifact.py,sha256=B4XErLr-m6RmQWtxMTu3wlFRFcqSwPYp6J0OL4Ng6L0,3179
-truefoundry/ml/log_types/artifacts/model.py,sha256=J-_v_ertSKQPThx6I9ivLUlCJUHPZJ71X8V0VgaIBAU,24319
+truefoundry/ml/log_types/artifacts/model.py,sha256=UfGKJA9ywDtXEIH6E-VfdjvlQNdaKO-Kv76KaTZOVkA,24413
 truefoundry/ml/log_types/artifacts/utils.py,sha256=MuPk6by8jIt7LepB1fqUC8emQBSxrTGJmIel7rbMBAc,7533
 truefoundry/ml/log_types/image/__init__.py,sha256=fcOq8yQnNj1rkLcPeIjLXBpdA1WIeiPsXOlAAvMxx7M,76
 truefoundry/ml/log_types/image/constants.py,sha256=wLtGEOA4T5fZHSlOXPuNDLX3lpbCtwlvGKPFk_1fah0,255
@@ -373,7 +368,7 @@ truefoundry/workflow/remote_filesystem/tfy_signed_url_client.py,sha256=xcT0wQmQl
 truefoundry/workflow/remote_filesystem/tfy_signed_url_fs.py,sha256=nSGPZu0Gyd_jz0KsEE-7w_BmnTD8CVF1S8cUJoxaCbc,13305
 truefoundry/workflow/task.py,sha256=ToitYiKcNzFCtOVQwz1W8sRjbR97eVS7vQBdbgUQtKg,1779
 truefoundry/workflow/workflow.py,sha256=WaTqUjhwfAXDWu4E5ehuwAxrCbDJkoAf1oWmR2E9Qy0,4575
-truefoundry-0.5.3rc5.dist-info/METADATA,sha256=pJtlnztVvgBPT53hHtnRD-mp0EzV6iTNPFKlEDvH2oc,2887
-truefoundry-0.5.3rc5.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-truefoundry-0.5.3rc5.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
-truefoundry-0.5.3rc5.dist-info/RECORD,,
+truefoundry-0.5.5.dist-info/METADATA,sha256=tAB2DHOeD0JgYGfDPp48HY1IU0b5zX1yOzMoYzrdbh0,2884
+truefoundry-0.5.5.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+truefoundry-0.5.5.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
+truefoundry-0.5.5.dist-info/RECORD,,

{truefoundry-0.5.3rc5.dist-info → truefoundry-0.5.5.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 1.9.1
+Generator: poetry-core 2.0.1
 Root-Is-Purelib: true
 Tag: py3-none-any

truefoundry/cli/commands/pat.py

@@ -1,24 +0,0 @@
-import click
-
-from truefoundry.cli.console import console
-from truefoundry.cli.const import COMMAND_CLS
-from truefoundry.deploy.lib.clients.servicefoundry_client import (
-    ServiceFoundryServiceClient,
-)
-
-
-@click.command(cls=COMMAND_CLS, name="generate-api-key")
-@click.option(
-    "--name",
-    required=True,
-    help="Name for the API key",
-)
-def generate_pat(name: str):
-    """Generate a new Personal Access Token with specified name"""
-    client = ServiceFoundryServiceClient()
-    pat = client.create_pat(name)
-    console.print(pat)
-
-
-def get_generate_pat_command():
-    return generate_pat

truefoundry/gateway/__init__.py

@@ -1 +0,0 @@
-"""Gateway module for Truefoundry CLI"""

truefoundry/gateway/cli/cli.py

@@ -1,30 +0,0 @@
-import click
-
-from truefoundry.cli.const import COMMAND_CLS, GROUP_CLS
-from truefoundry.cli.display_util import print_entity_list
-from truefoundry.gateway.lib.models import list_models
-
-
-def get_gateway_cli():
-    @click.group(cls=GROUP_CLS, help="Commands to interact with TrueFoundry Gateway")
-    def gateway(): ...
-
-    @gateway.group("list", cls=GROUP_CLS, help="List gateway resources")
-    def list_group():
-        """List gateway resources"""
-        pass
-
-    @list_group.command(
-        "models", cls=COMMAND_CLS, help="List available models filtered by type"
-    )
-    @click.option(
-        "--type",
-        "model_type",
-        type=click.Choice(["chat", "completion", "embedding"]),
-        help="Filter models by type",
-    )
-    def list_models_cli(model_type: str):
-        enabled_models = list_models(model_type)
-        print_entity_list("Models", enabled_models)
-
-    return gateway

truefoundry/gateway/lib/entities.py

@@ -1,28 +0,0 @@
-from typing import Any, Dict, List, Optional, Union
-
-from truefoundry.pydantic_v1 import BaseModel
-
-
-class GatewayModel(BaseModel):
-    id: str
-    name: str
-    provider: str
-    model_id: Optional[str]
-    provider_account_name: str
-    tfy_application_id: Optional[str] = None
-    enabled: bool = True
-    types: Union[str, List[str]] = ""
-    created_by: str
-    tenant_name: str
-    model_fqn: str
-
-    def list_row_data(self) -> Dict[str, Any]:
-        return {
-            "model_id": self.model_fqn,
-            "provider": self.provider,
-            "provider_model_id": self.model_id,
-        }
-
-
-class ProviderModels(BaseModel):
-    __root__: Dict[str, Dict[str, List[GatewayModel]]]

truefoundry/gateway/lib/models.py

@@ -1,29 +0,0 @@
-from typing import List, Literal, Optional
-
-from truefoundry.deploy.lib.clients.servicefoundry_client import (
-    ServiceFoundryServiceClient,
-)
-from truefoundry.gateway.lib.entities import GatewayModel
-
-
-def list_models(
-    model_type: Optional[Literal["chat", "completion", "embedding"]] = None,
-) -> List[GatewayModel]:
-    """List available models filtered by type
-
-    Args:
-        model_type (Optional[str], optional): Filter models by type ('chat' or 'completion'). Defaults to None.
-
-    Returns:
-        List: List of enabled models
-    """
-    client = ServiceFoundryServiceClient()
-    models = client.get_gateway_models(model_type)
-
-    enabled_models = []
-    for _, accounts in models.__root__.items():
-        for _, model_list in accounts.items():
-            for model in model_list:
-                enabled_models.append(model)
-
-    return enabled_models