truefoundry 0.6.6__py3-none-any.whl → 0.7.0__py3-none-any.whl

truefoundry/_client.py

@@ -33,4 +33,4 @@ class _LazyTrueFoundry(truefoundry_sdk.TrueFoundry):
         )
 
 
-client = _LazyTrueFoundry()
+client = _LazyTrueFoundry().v1

truefoundry/cli/__main__.py

@@ -14,6 +14,7 @@ from truefoundry.deploy.cli.commands import (
     get_delete_command,
     get_deploy_command,
     get_deploy_init_command,
+    get_get_command,
     get_login_command,
     get_logout_command,
     get_patch_application_command,
@@ -88,6 +89,7 @@ def create_truefoundry_cli() -> click.Group:
     cli.add_command(get_trigger_command())
     cli.add_command(get_terminate_command())
     cli.add_command(get_ml_cli())
+    cli.add_command(get_get_command())
 
     if not (sys.platform.startswith("win32") or sys.platform.startswith("cygwin")):
         cli.add_command(get_patch_command())

truefoundry/common/entities.py

@@ -58,10 +58,14 @@ class Token(BaseModel):
         assert self.decoded_value is not None
         return self.decoded_value["tenantName"]
 
+    @property
+    def exp(self) -> int:
+        assert self.decoded_value is not None
+        return self.decoded_value["exp"]
+
     def is_going_to_be_expired(self, buffer_in_seconds: int = 120) -> bool:
         assert self.decoded_value is not None
-        exp = int(self.decoded_value["exp"])
-        return (exp - time.time()) < buffer_in_seconds
+        return (self.exp - time.time()) < buffer_in_seconds
 
     def to_user_info(self) -> UserInfo:
         assert self.decoded_value is not None

truefoundry/common/session.py

@@ -8,7 +8,7 @@ from truefoundry.common.credential_provider import (
     EnvCredentialProvider,
     FileCredentialProvider,
 )
-from truefoundry.common.entities import UserInfo
+from truefoundry.common.entities import Token, UserInfo
 from truefoundry.common.utils import relogin_error_message
 from truefoundry.logger import logger
 
@@ -20,7 +20,7 @@ class Session:
     def __init__(self) -> None:
         self._closed = False
         self._cred_provider: Optional[CredentialProvider] = self._get_cred_provider()
-        self._user_info: Optional[UserInfo] = self._cred_provider.token.to_user_info()
+        self._user_info: Optional[UserInfo] = self.token.to_user_info()
 
     @staticmethod
     def _get_cred_provider() -> CredentialProvider:
@@ -69,7 +69,12 @@ class Session:
     @property
     def access_token(self) -> str:
         assert self._cred_provider is not None
-        return self._cred_provider.token.access_token
+        return self.token.access_token
+
+    @property
+    def token(self) -> Token:
+        assert self._cred_provider is not None
+        return self._cred_provider.token
 
     @property
     def tfy_host(self) -> str:

truefoundry/deploy/__init__.py

@@ -111,6 +111,7 @@ from truefoundry.deploy.v2.lib.patched_models import (
     Schedule,
     SecretMount,
     ServiceAutoscaling,
+    SlackBot,
     SlackWebhook,
     SparkJobJavaEntrypoint,
     SparkJobPythonEntrypoint,

truefoundry/deploy/_autogen/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  application.json
-#   timestamp: 2025-04-07T11:22:45+00:00
+#   timestamp: 2025-04-21T06:26:10+00:00
 
 from __future__ import annotations
 
@@ -360,6 +360,13 @@ class Image(BaseModel):
     )
 
 
+class JobEvent(str, Enum):
+    START = "START"
+    SUCCEEDED = "SUCCEEDED"
+    FAILED = "FAILED"
+    TERMINATED = "TERMINATED"
+
+
 class Claim(BaseModel):
     key: str
     values: List[str]
@@ -466,10 +473,6 @@ class NodeSelector(BaseModel):
     """
 
     type: Literal["node_selector"] = Field(..., description="")
-    gpu_type: Optional[str] = Field(
-        None,
-        description="Name of the Nvidia GPU. One of [P4, P100, V100, T4, A10G, A100_40GB, A100_80GB]\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",
-    )
     instance_families: Optional[List[str]] = Field(
         None,
         description="Instance family of the underlying machine to use. Multiple instance families can be supplied.\nThe workload is guaranteed to be scheduled on one of them.",
@@ -699,10 +702,6 @@ class Resources(BaseModel):
         2000,
         description="Disk storage limit. The unit of memory is in megabytes(MB). Exceeding this limit will result in eviction.\nIt should be greater than the request. This is ephemeral storage and will be wiped out on pod restarts or eviction",
     )
-    gpu_count: Optional[conint(ge=0, le=16)] = Field(
-        None,
-        description="Count of GPUs to provide to the application\nNote the exact count and max count available for a given GPU type depends on cloud provider and cluster type.",
-    )
     shared_memory_size: Optional[conint(ge=64, le=2000000)] = Field(
         None,
         description="Define the shared memory requirements for your workload. Machine learning libraries like Pytorch can use Shared Memory\nfor inter-process communication. If you use this, we will mount a `tmpfs` backed volume at the `/dev/shm` directory.\nAny usage will also count against the workload's memory limit (`resources.memory_limit`) along with your workload's memory usage.\nIf the overall usage goes above `resources.memory_limit` the user process may get killed.\nShared Memory Size cannot be more than the defined Memory Limit for the workload.",
@@ -829,6 +828,16 @@ class ServiceAutoscaling(BaseAutoscaling):
     )
 
 
+class SlackBot(BaseModel):
+    type: Literal["slack-bot"] = Field(..., description="")
+    notification_channel: constr(min_length=1) = Field(
+        ..., description="Specify the notification channel to send alerts to"
+    )
+    channels: List[constr(regex=r"^#[a-z0-9\-_]{2,80}$")] = Field(
+        ..., description="List of channels to send messages to."
+    )
+
+
 class SlackWebhook(BaseModel):
     type: Literal["slack-webhook"] = Field(..., description="")
     notification_channel: constr(min_length=1) = Field(
@@ -856,6 +865,26 @@ class SparkExecutorFixedInstances(BaseModel):
     count: conint(ge=0, le=500) = Field(1, description="Number of instances to start")
 
 
+class SparkImage(BaseModel):
+    """
+    Describes that we are using a pre-built image stored in a Docker Image registry
+    """
+
+    type: Literal["image"] = Field(..., description="")
+    spark_version: str = Field(
+        "3.5.2",
+        description="Spark version should match the spark version installed in the image.",
+    )
+    image_uri: constr(regex=r"^\S*$") = Field(
+        ...,
+        description="The image URI. Specify the name of the image and the tag.\nIf the image is in Dockerhub, you can skip registry-url (for e.g. `tensorflow/tensorflow`).\nYou can use an image from a private registry using Advanced fields",
+    )
+    docker_registry: Optional[str] = Field(
+        None,
+        description="FQN of the container registry. If you can't find your registry here,\nadd it through the [Integrations](/integrations?tab=docker-registry) page",
+    )
+
+
 class SparkJobJavaEntrypoint(BaseModel):
     type: Literal["java"] = Field(..., description="")
     main_application_file: str = Field(
@@ -1031,6 +1060,13 @@ class WorkbenchImage(BaseModel):
     )
 
 
+class WorkflowEvent(str, Enum):
+    SUCCEEDED = "SUCCEEDED"
+    FAILED = "FAILED"
+    ABORTED = "ABORTED"
+    TIMED_OUT = "TIMED_OUT"
+
+
 class ArtifactsDownload(BaseModel):
     """
     Download and cache models in a volume to enhance loading speeds and reduce costs by avoiding repeated downloads. [Docs](https://docs.truefoundry.com/docs/download-and-cache-models)
@@ -1385,14 +1421,10 @@ class SparkJob(BaseModel):
     name: constr(regex=r"^[a-z](?:[a-z0-9]|-(?!-)){1,30}[a-z0-9]$") = Field(
         ..., description="Name of the job"
     )
+    image: SparkImage
     entrypoint: Union[
         SparkJobPythonEntrypoint, SparkJobScalaEntrypoint, SparkJobJavaEntrypoint
     ] = Field(..., description="")
-    image: Image
-    spark_version: str = Field(
-        "3.5.2",
-        description="Spark version should match the spark version installed in the image.",
-    )
     driver_config: SparkDriverConfig
     executor_config: SparkExecutorConfig
     env: Optional[Dict[str, Any]] = Field(
@@ -1452,7 +1484,7 @@ class WorkflowAlert(BaseModel):
     Describes the configuration for the workflow alerts
     """
 
-    notification_target: Optional[Union[Email, SlackWebhook]] = None
+    notification_target: Optional[Union[Email, SlackWebhook, SlackBot]] = None
     on_completion: bool = Field(
         False, description="Send an alert when the job completes"
     )
@@ -1519,7 +1551,7 @@ class JobAlert(BaseModel):
         None,
         description="List of recipients' email addresses if the notification channel is Email.",
     )
-    notification_target: Optional[Union[Email, SlackWebhook]] = None
+    notification_target: Optional[Union[Email, SlackWebhook, SlackBot]] = None
     on_start: bool = Field(False, description="Send an alert when the job starts")
     on_completion: bool = Field(
         False, description="Send an alert when the job completes"

truefoundry/deploy/cli/commands/get_command.py

@@ -1,19 +1,11 @@
-import contextlib
-import sys
-from enum import Enum
-
 import rich_click as click
-import yaml
-from rich.pretty import pprint
 
-from truefoundry.cli.config import CliConfig
-from truefoundry.cli.console import console
-from truefoundry.cli.const import COMMAND_CLS, GROUP_CLS
-from truefoundry.cli.display_util import print_entity_obj, print_json
-from truefoundry.cli.util import handle_exception_wrapper
-from truefoundry.deploy.lib.dao import application as application_lib
-from truefoundry.deploy.lib.dao import version as version_lib
-from truefoundry.deploy.lib.dao import workspace as workspace_lib
+from truefoundry.cli.const import GROUP_CLS
+from truefoundry.common.utils import is_internal_env_set
+from truefoundry.deploy.cli.commands.k8s_exec_credential_command import (
+    k8s_exec_credential_command,
+)
+from truefoundry.deploy.cli.commands.kubeconfig_command import kubeconfig_command
 
 # TODO (chiragjn): --json should disable all non json console prints
 
@@ -26,191 +18,13 @@ def get_command():
 
     \b
     Supported resources:
-    - Workspace
-    - Application
-    - Application Version
+    - Kubeconfig
     """
     pass
 
 
-@click.command(name="workspace", cls=COMMAND_CLS, help="Get Workspace details")
-@click.option(
-    "-w",
-    "--workspace-fqn",
-    "--workspace_fqn",
-    type=click.STRING,
-    default=None,
-    help="FQN of the Workspace",
-    required=True,
-)
-@handle_exception_wrapper
-def get_workspace(workspace_fqn):
-    workspace = workspace_lib.get_workspace_by_fqn(workspace_fqn=workspace_fqn)
-    if CliConfig.get("json"):
-        print_json(data=workspace.dict())
-    else:
-        print_entity_obj("Workspace", workspace)
-
-
-@click.command(name="application", cls=COMMAND_CLS, help="Get Application details")
-@click.option(
-    "--application-fqn",
-    "--application_fqn",
-    type=click.STRING,
-    default=None,
-    help="FQN of the application",
-    required=True,
-)
-@handle_exception_wrapper
-def get_application(application_fqn):
-    application = application_lib.get_application(application_fqn=application_fqn)
-    if CliConfig.get("json"):
-        print_json(data=application.dict())
-    else:
-        print_entity_obj(
-            "Application",
-            application,
-        )
-
-
-@click.command(
-    name="application-version", cls=COMMAND_CLS, help="Get Application Version details"
-)
-@click.option(
-    "--application-fqn",
-    "--application_fqn",
-    type=click.STRING,
-    default=None,
-    help="FQN of the application",
-    required=True,
-)
-@click.option(
-    "--version",
-    type=click.STRING,
-    default=None,
-    help="Version number of the application deployment",
-    required=True,
-)
-@handle_exception_wrapper
-def get_version(application_fqn, version):
-    version = version_lib.get_version(application_fqn=application_fqn, version=version)
-    if CliConfig.get("json"):
-        print_json(data=version.dict())
-    else:
-        print_entity_obj("Version", version)
-
-
-@click.command(
-    name="spec", cls=COMMAND_CLS, help="Get YAML/Python Spec for an Application Version"
-)
-@click.option(
-    "--application-fqn",
-    "--application_fqn",
-    type=click.STRING,
-    default=None,
-    help="FQN of the application",
-    required=True,
-)
-@click.option(
-    "--version",
-    type=click.STRING,
-    default=None,
-    help="Version number of the application deployment",
-    required=True,
-)
-@click.option(
-    "-o",
-    "--output",
-    type=click.Choice(
-        [
-            "yml",
-            "yaml",
-            "json",
-            "py",
-            "python",
-        ]
-    ),
-    default="yaml",
-    help="Output format for the spec",
-    required=False,
-)
-@handle_exception_wrapper
-def get_spec(application_fqn, version, output):
-    version = version_lib.get_version(application_fqn=application_fqn, version=version)
-    if output in ["yml", "yaml"]:
-        yaml.safe_dump(version.manifest.dict(), sys.stdout, indent=2)
-    elif output in ["json"]:
-        print_json(version.manifest.dict())
-    elif output in ["py", "python"]:
-        from truefoundry.deploy.v2.lib.deployable_patched_models import Application
-
-        manifest = version.manifest.dict()
-        instance = Application.parse_obj(manifest).__root__
-
-        # TODO (chiragjn): Can we somehow just enable `use_enum_values` on all Pydantic classes?
-
-        @contextlib.contextmanager
-        def _monkey_patch_enum_repr():
-            def new_repr(self):
-                # return "%r" % (self._value_)
-                return "%s.%s" % (self.__class__.__name__, self._name_)
-
-            enum_subclasses = [
-                es
-                for es in Enum.__subclasses__()
-                if es.__module__.startswith("truefoundry.")
-            ]
-            original_reprs = []
-            for es in enum_subclasses:
-                original_reprs.append(es)
-                es.__repr__ = new_repr
-
-            yield
-
-            for es, og_repr in zip(enum_subclasses, original_reprs):
-                es.__repr__ = og_repr
-
-        with _monkey_patch_enum_repr():
-            pprint(
-                instance,
-                indent_guides=False,
-                max_length=88,
-                expand_all=True,
-                console=console,
-            )
-
-
-@click.command(name="job-run", cls=COMMAND_CLS, help="Get Job Run")
-@click.option(
-    "--application-fqn",
-    "--application_fqn",
-    type=click.STRING,
-    default=None,
-    help="FQN of the application",
-    required=True,
-)
-@click.option(
-    "--job-run-name",
-    "--job_run_name",
-    type=click.STRING,
-    default=None,
-    help="Run name of the job",
-    required=True,
-)
-@handle_exception_wrapper
-def get_job_run(application_fqn, job_run_name):
-    job_run = application_lib.get_job_run(
-        application_fqn=application_fqn, job_run_name=job_run_name
-    )
-    if CliConfig.get("json"):
-        print_json(data=job_run.dict())
-    else:
-        print_entity_obj("Job Run", job_run)
-
-
 def get_get_command():
-    get_command.add_command(get_workspace)
-    get_command.add_command(get_application)
-    get_command.add_command(get_version)
-    get_command.add_command(get_job_run)
+    get_command.add_command(kubeconfig_command)
+    if is_internal_env_set():
+        get_command.add_command(k8s_exec_credential_command)
     return get_command

truefoundry/deploy/cli/commands/k8s_exec_credential_command.py

@@ -0,0 +1,64 @@
+import json
+from datetime import datetime, timezone
+from typing import Any, Dict, Optional
+from urllib.parse import urlparse
+
+import rich_click as click
+
+from truefoundry.cli.const import COMMAND_CLS
+from truefoundry.cli.util import handle_exception_wrapper
+from truefoundry.common.session import Session
+from truefoundry.deploy.cli.commands.utils import (
+    CONTEXT_NAME_FORMAT,
+    get_cluster_server_url,
+    get_kubeconfig_content,
+    get_kubeconfig_path,
+)
+from truefoundry.deploy.io.no_output_callback import NoOutputCallBack
+from truefoundry.deploy.lib.session import login
+
+
+@click.command(
+    name="k8s-exec-credential",
+    cls=COMMAND_CLS,
+    help="Generate a Kubernetes exec credential for the specified cluster user",
+)
+@click.option(
+    "-c",
+    "--cluster",
+    type=str,
+    required=True,
+    help="The cluster id from TrueFoundry",
+)
+@handle_exception_wrapper
+def k8s_exec_credential_command(cluster: str) -> None:
+    """
+    Generate a Kubernetes exec credential for the specified cluster.
+    This command retrieves the cluster server URL from the kubeconfig file,
+    """
+    path = get_kubeconfig_path()
+    kubeconfig: Dict[str, Any] = get_kubeconfig_content(path=path)
+    server_url: Optional[str] = get_cluster_server_url(kubeconfig, cluster)
+    if not server_url:
+        raise click.ClickException(
+            f"\nContext {CONTEXT_NAME_FORMAT.format(cluster=cluster)!r} for cluster {cluster!r} not found in kubeconfig. \n\nPlease run 'tfy get kubeconfig --cluster {cluster}' first."
+        )
+    host: str = f"{urlparse(server_url).scheme}://{urlparse(server_url).netloc}"
+    login(host=host, output_hook=NoOutputCallBack())
+
+    session = Session.new()
+    token: str = session.access_token
+
+    exec_credential: Dict[str, Any] = {
+        "kind": "ExecCredential",
+        "apiVersion": "client.authentication.k8s.io/v1beta1",
+        "spec": {},
+        "status": {
+            "expirationTimestamp": datetime.fromtimestamp(
+                session.token.exp, tz=timezone.utc
+            ).strftime("%Y-%m-%dT%H:%M:%SZ"),
+            "token": token,
+        },
+    }
+
+    print(json.dumps(exec_credential, indent=4))

truefoundry/deploy/cli/commands/kubeconfig_command.py

@@ -0,0 +1,129 @@
+from typing import Any, Dict, Optional
+from urllib.parse import urljoin
+
+import questionary
+import rich_click as click
+from rich.console import Console
+
+from truefoundry.cli.const import COMMAND_CLS
+from truefoundry.cli.util import handle_exception_wrapper
+from truefoundry.common.session import Session
+from truefoundry.deploy.cli.commands.utils import (
+    CONTEXT_NAME_FORMAT,
+    add_update_cluster_context,
+    get_cluster_server_url,
+    get_kubeconfig_content,
+    get_kubeconfig_path,
+    save_kubeconfig,
+)
+
+console = Console()
+
+
+def _select_cluster(cluster: Optional[str] = None) -> str:
+    """
+    Retrieve available clusters and either return the specified one after validation
+    or allow the user to interactively select from the list.
+    """
+    from truefoundry.deploy.lib.clients.servicefoundry_client import (
+        ServiceFoundryServiceClient,
+    )
+
+    clusters = ServiceFoundryServiceClient().list_clusters()
+
+    if not clusters:
+        raise click.ClickException("No clusters found in your account.")
+
+    if cluster:
+        if not any(c.id == cluster for c in clusters):
+            raise click.ClickException(
+                f"Cluster {cluster} not found. Either it does not exist or you might not be autthorized to access it"
+            )
+        return cluster
+
+    choices = {cluster.id: cluster for cluster in clusters}
+    cluster = questionary.select(
+        "Available Clusters:", choices=list(choices.keys())
+    ).ask()
+    if not cluster:
+        raise click.ClickException("No cluster selected.")
+    return cluster
+
+
+def _construct_k8s_proxy_server(host: str, cluster: str) -> str:
+    """
+    Construct the Kubernetes proxy server URL.
+    """
+    return urljoin(host, f"api/svc/v1/k8s/proxy/{cluster}")
+
+
+def _should_update_existing_context(cluster: str, kubeconfig: Dict[str, Any]) -> bool:
+    """
+    Prompt the user whether to overwrite an existing kubeconfig context.
+    """
+    server_url = get_cluster_server_url(kubeconfig, cluster)
+    if server_url is not None:
+        console.print(
+            f"\nContext {CONTEXT_NAME_FORMAT.format(cluster=cluster)!r} for cluster {cluster!r} already exists in kubeconfig.\n"
+        )
+        return click.confirm(
+            text="Do you want to update the context?", default=False, err=True
+        )
+    return True
+
+
+@click.command(name="kubeconfig", cls=COMMAND_CLS)
+@click.option(
+    "-c",
+    "--cluster",
+    type=str,
+    required=False,
+    help="The cluster id from TrueFoundry. If not provided, an interactive prompt will list available clusters",
+)
+@click.option(
+    "--overwrite",
+    is_flag=True,
+    default=False,
+    show_default=True,
+    help="Overwrites existing cluster entry without prompting",
+)
+@handle_exception_wrapper
+def kubeconfig_command(cluster: Optional[str] = None, overwrite: bool = False) -> None:
+    """
+    Update kubeconfig file to access cluster attached to TrueFoundry Control Plane.
+
+    By default, credentials are written to ~/.kube/config. You can provide an alternate path by setting the KUBECONFIG environment variable. If KUBECONFIG contains multiple paths, the first one is used.
+    """
+    session = Session.new()
+    cluster = _select_cluster(cluster)
+
+    path = get_kubeconfig_path()
+    kubeconfig = get_kubeconfig_content(path=path)
+
+    if not overwrite and not _should_update_existing_context(cluster, kubeconfig):
+        console.print(
+            "Existing context found. Use '--overwrite' to force update the context."
+        )
+        return
+
+    k8s_proxy_server = _construct_k8s_proxy_server(session.tfy_host, cluster)
+    context_name = add_update_cluster_context(
+        kubeconfig,
+        cluster,
+        k8s_proxy_server,
+        exec_command=[
+            "tfy",
+            "--json",
+            "get",
+            "k8s-exec-credential",
+            "--cluster",
+            cluster,
+        ],
+        envs={"TFY_INTERNAL": "1"},
+    )
+
+    save_kubeconfig(kubeconfig, path=path)
+    console.print(
+        f"\nUpdated kubeconfig at {str(path)!r} with context {context_name!r} for cluster {cluster!r}\n\n"
+        f"Run 'kubectl config use-context {context_name}' to use this context.\n"
+    )

truefoundry/deploy/cli/commands/utils.py

@@ -0,0 +1,118 @@
+import os
+from pathlib import Path
+from typing import Any, Dict, List, Optional
+
+import yaml
+
+DEFAULT_KUBECONFIG_PATH: Path = Path.home() / ".kube" / "config"
+CLUSTER_NAME_FORMAT: str = "tfy-{cluster}-cluster"
+USER_NAME_FORMAT: str = "tfy-{cluster}-user"
+CONTEXT_NAME_FORMAT: str = "tfy-{cluster}-context"
+KUBE_CONFIG_CONTENT = {
+    "apiVersion": "v1",
+    "kind": "Config",
+    "clusters": [],
+    "users": [],
+    "contexts": [],
+}
+
+
+def get_kubeconfig_path() -> Path:
+    """
+    Returns the kubeconfig path to use.
+    If KUBECONFIG is set, returns the first path from the environment variable.
+    Otherwise, returns the default kubeconfig path.
+    """
+    kubeconfig_env = os.environ.get("KUBECONFIG")
+    if kubeconfig_env:
+        # Use the first path in KUBECONFIG if multiple are provided
+        first_path = kubeconfig_env.split(os.pathsep)[0]
+        return Path(first_path)
+    return DEFAULT_KUBECONFIG_PATH
+
+
+def get_kubeconfig_content(path: Path = DEFAULT_KUBECONFIG_PATH) -> Dict[str, Any]:
+    if path.exists():
+        with open(path, "r") as f:
+            return yaml.safe_load(f) or KUBE_CONFIG_CONTENT
+    else:
+        return KUBE_CONFIG_CONTENT
+
+
+def save_kubeconfig(config: Dict[str, Any], path: Path) -> None:
+    config["apiVersion"] = "v1"
+    config["kind"] = "Config"
+    path.parent.mkdir(parents=True, exist_ok=True)
+    with open(path, "w") as f:
+        yaml.safe_dump(config, f, default_flow_style=False)
+
+
+def add_update_cluster_context(
+    config: Dict[str, Any],
+    cluster: str,
+    server_url: str,
+    exec_command: List[str],
+    envs: Optional[Dict[str, str]] = None,
+) -> str:
+    """
+    Adds a new cluster context to the given kubeconfig dictionary using exec-based authentication.
+    """
+    cluster_name: str = CLUSTER_NAME_FORMAT.format(cluster=cluster)
+    user_name: str = USER_NAME_FORMAT.format(cluster=cluster)
+    context_name: str = CONTEXT_NAME_FORMAT.format(cluster=cluster)
+
+    # Add or update cluster
+    config["clusters"] = [
+        c for c in config.get("clusters", []) if c["name"] != cluster_name
+    ]
+    config["clusters"].append(
+        {
+            "name": cluster_name,
+            "cluster": {"server": server_url},
+        }
+    )
+
+    # Add or update user with exec command
+    config["users"] = [u for u in config.get("users", []) if u["name"] != user_name]
+    config["users"].append(
+        {
+            "name": user_name,
+            "user": {
+                "exec": {
+                    "apiVersion": "client.authentication.k8s.io/v1beta1",
+                    "command": exec_command[0],
+                    "args": exec_command[1:],
+                    "env": (
+                        [{"name": k, "value": v} for k, v in envs.items()]
+                        if envs
+                        else []
+                    ),
+                }
+            },
+        }
+    )
+
+    # Add or update context
+    config["contexts"] = [
+        c for c in config.get("contexts", []) if c["name"] != context_name
+    ]
+    config["contexts"].append(
+        {"name": context_name, "context": {"cluster": cluster_name, "user": user_name}}
+    )
+    return context_name
+
+
+def get_cluster_context(
+    config: Dict[str, Any], cluster: str
+) -> Optional[Dict[str, Any]]:
+    cluster_name: str = CLUSTER_NAME_FORMAT.format(cluster=cluster)
+    return next(
+        (c for c in config.get("clusters", []) if c["name"] == cluster_name), None
+    )
+
+
+def get_cluster_server_url(config: Dict[str, Any], cluster: str) -> Optional[str]:
+    cluster: Optional[Dict[str, Any]] = get_cluster_context(config, cluster)
+    if cluster:
+        return cluster["cluster"].get("server")
+    return None

truefoundry/deploy/io/no_output_callback.py

@@ -0,0 +1,27 @@
+from typing import Any, List, Optional
+
+from truefoundry.deploy.io.output_callback import OutputCallBack
+
+
+class NoOutputCallBack(OutputCallBack):
+    def print_header(self, line: Any) -> None:
+        pass
+
+    def _print_separator(self) -> None:
+        pass
+
+    def print_line(self, line: str) -> None:
+        pass
+
+    def print_lines_in_panel(
+        self, lines: List[str], header: Optional[str] = None
+    ) -> None:
+        pass
+
+    def print_code_in_panel(
+        self, lines: List[str], header: Optional[str] = None
+    ) -> None:
+        pass
+
+    def print(self, line: Any) -> None:
+        pass

truefoundry/deploy/lib/clients/servicefoundry_client.py

@@ -32,6 +32,7 @@ from truefoundry.deploy._autogen import models as autogen_models
 from truefoundry.deploy.io.output_callback import OutputCallBack
 from truefoundry.deploy.lib.model.entity import (
     Application,
+    Cluster,
     CreateDockerRepositoryResponse,
     Deployment,
     DockerRegistryCredentials,
@@ -110,6 +111,13 @@ class ServiceFoundryServiceClient(BaseServiceFoundryServiceClient):
         )
         return request_handling(response)
 
+    @check_min_cli_version
+    def list_clusters(self) -> List[Cluster]:
+        url = f"{self._api_server_url}/{VERSION_PREFIX}/clusters"
+        response = session_with_retries().get(url, headers=self._get_headers())
+        response = request_handling(response)
+        return parse_obj_as(List[Cluster], response["data"])
+
     @check_min_cli_version
     def list_workspaces(
         self,

truefoundry/deploy/lib/model/entity.py

@@ -34,6 +34,11 @@ class Entity(Base):
     updatedAt: datetime.datetime = Field(repr=False)
 
 
+class Cluster(Entity):
+    id: str
+    cloudProvider: str
+
+
 class Workspace(Entity):
     id: str = Field(repr=False)
     name: str

truefoundry/deploy/lib/session.py

@@ -70,7 +70,9 @@ def login(
         else:
             auth_service = AuthServiceClient.from_tfy_host(tfy_host=host)
             # interactive login
-            token = _login_with_device_code(base_url=host, auth_service=auth_service)
+            token = _login_with_device_code(
+                base_url=host, auth_service=auth_service, output_hook=output_hook
+            )
 
         cred_file_content = CredentialsFileContent(
             access_token=token.access_token,
@@ -97,7 +99,7 @@ def logout(
 
 
 def get_access_token():
-    # Create a new session which handles token refresh
+    # Get the access token from the session
     session = Session.new()
     return session.access_token
 

truefoundry/deploy/v2/lib/deploy.py

@@ -189,30 +189,6 @@ def _deploy_wait_handler(  # noqa: C901
     return last_status_printed
 
 
-def _warn_when_gpu_selected_without_cuda(component: Component):
-    is_python_build_without_cuda = (
-        hasattr(component, "image")
-        and isinstance(component.image, autogen_models.Build)
-        and isinstance(component.image.build_spec, autogen_models.PythonBuild)
-        and not component.image.build_spec.cuda_version
-    )
-    uses_gpu = (
-        hasattr(component, "resources")
-        and isinstance(component, autogen_models.Resources)
-        and component.resources.gpu_count > 0
-    )
-    if is_python_build_without_cuda and uses_gpu:
-        logger.warning(
-            "Warning: `gpu_count` is greater than 0 in `Resources` (i.e. `resources.gpu_count`) "
-            "but no `cuda_version` was passed to `PythonBuild` "
-            "(i.e. `image.build_spec.cuda_version`). "
-            "Your application might optionally need CUDA toolkit installed "
-            "to utilize the GPU. You can choose one by passing one of "
-            "`truefoundry.deploy.CUDAVersion` in `PythonBuild` instance. "
-            "\n\nE.g.\n```\nPythonBuild(..., cuda_version=CUDAVersion.CUDA_11_3_CUDNN8)\n```"
-        )
-
-
 def _resolve_workspace_fqn(
     component: Component, workspace_fqn: Optional[str] = None
 ) -> str:
@@ -266,7 +242,6 @@ def deploy_component(
     wait: bool = True,
     force: bool = False,
 ) -> Deployment:
-    _warn_when_gpu_selected_without_cuda(component=component)
     workspace_fqn = _resolve_workspace_fqn(
         component=component, workspace_fqn=workspace_fqn
     )
@@ -274,10 +249,11 @@ def deploy_component(
     workspace_id = get_workspace_by_fqn(workspace_fqn).id
     if isinstance(component, autogen_models.ApplicationSet):
         updated_component = component.copy(deep=True)
-        for i in range(len(updated_component.components)):
-            updated_component.components[i] = _handle_if_local_source(
-                component=component.components[i], workspace_fqn=workspace_fqn
-            )
+        if updated_component.components:
+            for i, subcomponent in enumerate(updated_component.components):
+                updated_component.components[i] = _handle_if_local_source(
+                    component=subcomponent, workspace_fqn=workspace_fqn
+                )
     else:
         updated_component = _handle_if_local_source(
             component=component, workspace_fqn=workspace_fqn

truefoundry/deploy/v2/lib/patched_models.py

@@ -2,7 +2,6 @@ import enum
 import os
 import re
 import warnings
-from collections.abc import Mapping
 from typing import Literal, Optional, Union
 
 from truefoundry.common.warnings import TrueFoundryDeprecationWarning
@@ -15,63 +14,6 @@ from truefoundry.pydantic_v1 import (
     validator,
 )
 
-LEGACY_GPU_TYPE_COUNT_WARNING_MESSAGE_TEMPLATE = """
----------
-The `gpu_count` and `gpu_type` fields are deprecated. Please remove these fields
-from your deployment Python script or YAML Spec.
-
-If you are using Python SDK, add GPUs in the following way:
-
-```
-from truefoundry.deploy import NvidiaGPU, Resources
-...
-
-  resources=Resources(
-    ...
-    devices=[NvidiaGPU(name="{gpu_type}", count={gpu_count})],
-  )
-```
-
-If you are using YAML Spec to deploy, add GPUs in the following way:
-
-```
-resources:
-  devices:
-    - type: nvidia_gpu
-      name: {gpu_type}
-      count: {gpu_count}
-```
----------
-"""
-
-LEGACY_GPU_COUNT_WARNING_MESSAGE_TEMPLATE = """
----------
-The `gpu_count` field is deprecated. Please remove this field from your
-deployment Python script or YAML Spec.
-
-If you are using Python SDK, add GPUs in the following way:
-
-```
-from truefoundry.deploy import NvidiaGPU, Resources
-...
-
-  resources=Resources(
-    ...
-    devices=[NvidiaGPU(count={gpu_count})],
-  )
-```
-
-If you are using YAML Spec to deploy, add GPUs in the following way:
-
-```
-resources:
-  devices:
-    - type: nvidia_gpu
-      count: {gpu_count}
-```
----------
-"""
-
 AUTO_DISCOVERED_REQUIREMENTS_TXT_WARNING_MESSAGE_TEMPLATE = """\
 Using automatically discovered {requirements_txt_path} as the requirements file.
 This auto discovery behavior is deprecated and will be removed in a future release.
@@ -304,35 +246,7 @@ class Port(models.Port, PatchedModelBase):
 
 
 class Resources(models.Resources, PatchedModelBase):
-    @root_validator(pre=False)
-    def warn_gpu_count_type_depreciation(cls, values):
-        gpu_count = values.get("gpu_count")
-        gpu_type = None
-        node = values.get("node")
-        if node:
-            if isinstance(node, NodeSelector):
-                gpu_type = node.gpu_type
-            elif isinstance(node, Mapping):
-                gpu_count = node.get("gpu_type")
-
-        if gpu_count and gpu_type:
-            warnings.warn(
-                LEGACY_GPU_TYPE_COUNT_WARNING_MESSAGE_TEMPLATE.format(
-                    gpu_type=gpu_type,
-                    gpu_count=gpu_count,
-                ),
-                category=TrueFoundryDeprecationWarning,
-                stacklevel=2,
-            )
-        elif gpu_count:
-            warnings.warn(
-                LEGACY_GPU_COUNT_WARNING_MESSAGE_TEMPLATE.format(
-                    gpu_count=gpu_count,
-                ),
-                category=TrueFoundryDeprecationWarning,
-                stacklevel=2,
-            )
-        return values
+    pass
 
 
 class Param(models.Param, PatchedModelBase):
@@ -397,7 +311,6 @@ class VolumeMount(models.VolumeMount, PatchedModelBase):
 
 class NodeSelector(models.NodeSelector, PatchedModelBase):
     type: Literal["node_selector"] = "node_selector"
-    gpu_type: Optional[Union[GPUType, str]] = None
 
 
 class NodepoolSelector(models.NodepoolSelector, PatchedModelBase):
@@ -573,6 +486,10 @@ class SlackWebhook(models.SlackWebhook, PatchedModelBase):
     type: Literal["slack-webhook"] = "slack-webhook"
 
 
+class SlackBot(models.SlackBot, PatchedModelBase):
+    type: Literal["slack-bot"] = "slack-bot"
+
+
 class SparkJobScalaEntrypoint(models.SparkJobScalaEntrypoint, PatchedModelBase):
     type: Literal["scala"] = "scala"
 

truefoundry/ml/prompt_utils.py

@@ -49,17 +49,22 @@ def render_prompt(
 
     # Merge parameters from model_configuration and extra_parameters
     model_configuration = prompt_template.model_configuration
-    parameters = {
-        **(
-            model_configuration.parameters.dict()
-            if model_configuration.parameters
-            else {}
-        ),
-        **(model_configuration.extra_parameters or {}),
-    }
+    if model_configuration:
+        model = model_configuration.model
+        parameters = {
+            **(
+                model_configuration.parameters.dict()
+                if model_configuration.parameters
+                else {}
+            ),
+            **(model_configuration.extra_parameters or {}),
+        }
+    else:
+        model = None
+        parameters = {}
 
     return {
         "messages": rendered_messages,
-        "model": model_configuration.model,
+        "model": model,
         "parameters": {k: v for k, v in parameters.items() if v is not None},
     }

truefoundry/workflow/__init__.py

@@ -2,7 +2,10 @@ try:
     import fsspec
     from flytekit import task as _
 except ImportError:
-    print("To use workflows, please run 'pip install truefoundry[workflow]'.")
+    print(
+        "To use workflows, please run 'pip install truefoundry[workflow]'. "
+        "Note: The `workflow` feature is only available for Python 3.9 to 3.12"
+    )
 
 from flytekit import conditional
 from flytekit.types.directory import FlyteDirectory

{truefoundry-0.6.6.dist-info → truefoundry-0.7.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: truefoundry
-Version: 0.6.6
+Version: 0.7.0
 Summary: TrueFoundry CLI
 Author-email: TrueFoundry Team <abhishek@truefoundry.com>
 Requires-Python: <3.14,>=3.8.1
@@ -30,12 +30,12 @@ Requires-Dist: requirements-parser<0.12.0,>=0.11.0
 Requires-Dist: rich-click<2.0.0,>=1.2.1
 Requires-Dist: rich<14.0.0,>=13.7.1
 Requires-Dist: tqdm<5.0.0,>=4.0.0
-Requires-Dist: truefoundry-sdk==0.0.10
+Requires-Dist: truefoundry-sdk==0.0.15
 Requires-Dist: typing-extensions>=4.0
 Requires-Dist: urllib3<3,>=1.26.18
 Requires-Dist: yq<4.0.0,>=3.1.0
 Provides-Extra: workflow
-Requires-Dist: flytekit==1.13.13; extra == 'workflow'
+Requires-Dist: flytekit==1.15.3; (python_version >= '3.9' and python_version < '3.13') and extra == 'workflow'
 Description-Content-Type: text/markdown
 
 # TrueFoundry

{truefoundry-0.6.6.dist-info → truefoundry-0.7.0.dist-info}/RECORD

@@ -1,5 +1,5 @@
 truefoundry/__init__.py,sha256=VVpO-Awh1v93VOURe7hank8QpeSPc0dCykwr14GOFsw,967
-truefoundry/_client.py,sha256=Y3qHi_Lg4Sx6GNvsjAHIoAfFr8PJnqgCrXmpNAI3ECg,1417
+truefoundry/_client.py,sha256=VQEfRvPE7nuqq--q28cpmnIYPG3RH52RSifIFOzzvTg,1420
 truefoundry/logger.py,sha256=u-YCNjg5HBwE70uQcpjIG64Ghos-K2ulTWaxC03BSj4,714
 truefoundry/pydantic_v1.py,sha256=jSuhGtz0Mbk1qYu8jJ1AcnIDK4oxUsdhALc4spqstmM,345
 truefoundry/version.py,sha256=bqiT4Q-VWrTC6P4qfK43mez-Ppf-smWfrl6DcwV7mrw,137
@@ -28,7 +28,7 @@ truefoundry/autodeploy/utils/client.py,sha256=PvbSkfgAjAogGjisinqmh4mP4svowxAC0I
 truefoundry/autodeploy/utils/diff.py,sha256=Ef8Y-VffDKel_-q-GxRam6gqiv8qTLMcqVg6iifXfcA,5358
 truefoundry/autodeploy/utils/pydantic_compat.py,sha256=hEAUy5kLjhPdzw7yGZ2iXGMXbbMVXVlGzIofmyHafXQ,412
 truefoundry/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/cli/__main__.py,sha256=oPtY1egJ21nN46me7E-xLfU0m_Zy5Oz2jj571os8Yxg,3465
+truefoundry/cli/__main__.py,sha256=oJgbKUQbT5SDhqdtvhGw-WG6s0Kg3jiMiO0oz7bgOrc,3525
 truefoundry/cli/config.py,sha256=tf8w4UfVzcC6eYkENvuuCPYt_V3sqVpO1bclORV9tAk,206
 truefoundry/cli/console.py,sha256=9-dMy4YPisCJQziRKTg8Qa0UJnOGl1soiUnJjsnLDvE,242
 truefoundry/cli/const.py,sha256=dVHPo1uAiDSSMXwXoT2mR5kNQjExT98QNVRz98Hz_Ts,510
@@ -39,18 +39,18 @@ truefoundry/common/auth_service_client.py,sha256=N3YxKlx63r6cPZqbgb2lqBOPI69ShB7
 truefoundry/common/constants.py,sha256=rXWVNhy2j8UVmkPxpjaftL6mWRYILLzKLyCAAXKJy6Q,3247
 truefoundry/common/credential_file_manager.py,sha256=1yEk1Zm2xS4G0VDFwKSZ4w0VUrcPWQ1nJnoBaz9xyKA,4251
 truefoundry/common/credential_provider.py,sha256=_OhJ2XFlDaVsrUO-FyywxctcGGqDdC2pgcvwEKqQD0Q,4071
-truefoundry/common/entities.py,sha256=Gx52dIwUARwGLE2hCSe9PJWfXh-ARCmdWwX6KtCxJng,3805
+truefoundry/common/entities.py,sha256=ko33kesGy3vI9NJ5Ganq8HpnaURTOHictr6h75764no,3893
 truefoundry/common/exceptions.py,sha256=jkU0N7hV_P-EhXeud4I5vuB9glXXZSWPf8LcH04mSbw,459
 truefoundry/common/request_utils.py,sha256=e9qrAQ1MutU7JALDKcucmNd0KQEVBqgW3yx0w1zeHIU,5700
 truefoundry/common/servicefoundry_client.py,sha256=2fYhdVPSvLXz5C5tosOq86JD8WM3IRUIy1VO9deDxZI,3340
-truefoundry/common/session.py,sha256=xeBAPUNEJv2XVFQCRUGeBDTePh5zrKNSok8vmSxBjPw,2813
+truefoundry/common/session.py,sha256=d9l3TEBpqVP4mr4mTGY1qVxc815skzMlNNdw14otg34,2923
 truefoundry/common/storage_provider_utils.py,sha256=yURhMw8k0FLFvaviRHDiifhvc6GnuQwGMC9Qd2uM440,10934
 truefoundry/common/types.py,sha256=BMJFCsR1lPJAw66IQBSvLyV4I6o_x5oj78gVsUa9si8,188
 truefoundry/common/utils.py,sha256=yEQtJW2fT9xbNpRhfRoD9hvhGw-FgGS3agh1oZptmjg,6379
 truefoundry/common/warnings.py,sha256=rs6BHwk7imQYedo07iwh3TWEOywAR3Lqhj0AY4khByg,504
-truefoundry/deploy/__init__.py,sha256=SkkhHRP1qOeQb1XYRiTKKjA0uDm8YEbdDEPUN4rvY1w,2732
+truefoundry/deploy/__init__.py,sha256=6D22iiCgd5xlzBaG34q9Cx4rGgwf5qIAKQrOCgaCXYY,2746
 truefoundry/deploy/python_deploy_codegen.py,sha256=AainOFR20XvhNeztJkLPWGZ40lAT_nwc-ZmG77Kum4o,6525
-truefoundry/deploy/_autogen/models.py,sha256=rypFsppVuXJJOhFwYAS65Ef3OyIPTkBIod9DXoDwOwY,70918
+truefoundry/deploy/_autogen/models.py,sha256=yIGmwEptVUIM1R5DOL8JYp2-8T-YDDumADS4ZAFoEcI,71727
 truefoundry/deploy/builder/__init__.py,sha256=nGQiR3r16iumRy7xbVQ6q-k0EApmijspsfVpXDE-9po,4953
 truefoundry/deploy/builder/constants.py,sha256=amUkHoHvVKzGv0v_knfiioRuKiJM0V0xW0diERgWiI0,508
 truefoundry/deploy/builder/docker_service.py,sha256=sm7GWeIqyrKaZpxskdLejZlsxcZnM3BTDJr6orvPN4E,3948
@@ -69,7 +69,9 @@ truefoundry/deploy/cli/commands/create_command.py,sha256=rCajvQvAfZU10nDZOYpRACb
 truefoundry/deploy/cli/commands/delete_command.py,sha256=8SriRwg5mEHL4zP0mdDjJ0cKWJTU_1Pjq3450zMr9tk,3889
 truefoundry/deploy/cli/commands/deploy_command.py,sha256=8aTBvzPaT9xg6KPmpcpqJlmdj4yXzWUfAy6slcoPN74,4123
 truefoundry/deploy/cli/commands/deploy_init_command.py,sha256=g-jBfrEmhZ0TDWsyqPDn4K6q33EqJSGmBTt1eMYig-w,600
-truefoundry/deploy/cli/commands/get_command.py,sha256=HZQGWib-qrS0RdzctRDwDLfMGu0wqWkAM1u26cTZlsc,5944
+truefoundry/deploy/cli/commands/get_command.py,sha256=bR8tAjQQhimzaTQ57L6BPJwcxQ_SGWCF5CqHDpxgG90,837
+truefoundry/deploy/cli/commands/k8s_exec_credential_command.py,sha256=EknpdufMAEnjSGMG7a-Jj7tkoiS5zmbJRREafb14Alw,2160
+truefoundry/deploy/cli/commands/kubeconfig_command.py,sha256=3u3A_scsdd1UnqxKQgGg8ivZArZKQYVV0Z-ATotbPPc,4176
 truefoundry/deploy/cli/commands/list_command.py,sha256=zKu_JWY35eMIzgSNFYtmwi2uezZ4k-8yk3C1Vqsvshc,4470
 truefoundry/deploy/cli/commands/login_command.py,sha256=EHZH3cDQdbVWMkuFxd7JfB0B0Dr89sUmt214ICBmspY,1024
 truefoundry/deploy/cli/commands/logout_command.py,sha256=u3kfrEp0ETbrz40KjD4GCC3XEZ5YRAlrca_Df4U_mk0,536
@@ -79,21 +81,23 @@ truefoundry/deploy/cli/commands/patch_command.py,sha256=wA95khMO9uVz8SaJlgYMUwaX
 truefoundry/deploy/cli/commands/redeploy_command.py,sha256=JNeSK3itsirrKYJh3q-Cc10pCtYR9bojsvI8up6_0wI,1018
 truefoundry/deploy/cli/commands/terminate_comand.py,sha256=UKhOdbAej8ubX3q44vpLrOotAcvH4vHpRZJQrRf_AfM,1077
 truefoundry/deploy/cli/commands/trigger_command.py,sha256=_qSl-AShepZpbGUGTfLfJGd74VJJ_wd3eXYt2DfxIFo,4716
+truefoundry/deploy/cli/commands/utils.py,sha256=Fftlayq5PbLV7lqFVIoMy_uBoIKe2_XpX4BEyQnliC4,3670
 truefoundry/deploy/core/__init__.py,sha256=j61bMWj4BkWihdssKMSFhieo7afJDtpc7qO7zk1rDB4,140
 truefoundry/deploy/core/login.py,sha256=N2VrW3nlBzoyoYulkipxwQvCpjBhi3sfsmhxK1ktWhg,236
 truefoundry/deploy/core/logout.py,sha256=TpWLq4_DsxYS5GX2OJQGDhekNOfiOLb-vO5khQueHXw,80
 truefoundry/deploy/io/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+truefoundry/deploy/io/no_output_callback.py,sha256=aXzVwDb-LrSA-2ApsQb5ZVvS6kC0phzqmXrnVfu-xoU,613
 truefoundry/deploy/io/output_callback.py,sha256=_q79-dpFxnU762VPM9Ryy2gnuJnIotZ2_dylgv__ti8,777
 truefoundry/deploy/io/rich_output_callback.py,sha256=m99RodkILXCgy_LJujEcojbpW1tL0H5Fjb0lqe6X_PQ,958
 truefoundry/deploy/lib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/deploy/lib/const.py,sha256=Wg0GDnfFu-g1fJr4lU80NH2ULw0R0dYjV7LnW-PbOeM,173
 truefoundry/deploy/lib/logs_utils.py,sha256=SQxRv3jDDmgHdOUMhlMaAPGYskybnBUMpst7QU_i_sc,1469
 truefoundry/deploy/lib/messages.py,sha256=8424kj3kqCyDCX5Nr2WJZZ_UEutPoaSs_y2f9-O4yy8,1001
-truefoundry/deploy/lib/session.py,sha256=-FX4gOtiGlc6Jk56JPVZpDqXR9xQza77AIlBvNJpYqQ,4919
+truefoundry/deploy/lib/session.py,sha256=fLdgR6ZDp8-hFl5NTON4ngnWLsMzGxvKtfpDOOw_7lo,4963
 truefoundry/deploy/lib/util.py,sha256=J7r8San2wKo48A7-BlH2-OKTlBO67zlPjLEhMsL8os0,1059
 truefoundry/deploy/lib/win32.py,sha256=1RcvPTdlOAJ48rt8rCbE2Ufha2ztRqBAE9dueNXArrY,5009
 truefoundry/deploy/lib/clients/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/deploy/lib/clients/servicefoundry_client.py,sha256=NjNGmBsmrIfDZFfqJz12B4TVDHMZeAZDf-5eKSnS9zs,26697
+truefoundry/deploy/lib/clients/servicefoundry_client.py,sha256=fmRlPYCimk1ZLbMgdzfJVCbcKRCVnFYL5T3j2uJA0Tc,27037
 truefoundry/deploy/lib/dao/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/deploy/lib/dao/application.py,sha256=oMszpueXPUfTUuN_XdKwoRjQyqAgWHhZ-10cbprCVdM,9226
 truefoundry/deploy/lib/dao/apply.py,sha256=5IFERe5sLmZGlavaKTIxL4xPHAme4ZS2Ww0a2rKTyT0,3029
@@ -101,14 +105,14 @@ truefoundry/deploy/lib/dao/delete.py,sha256=uPL2psqWNw2O0oDikXJOlVxmG8n5d3Z0Ia9q
 truefoundry/deploy/lib/dao/version.py,sha256=AtdW_4O1DPUKdfv2qy6iUJsZ_95vM6z0AqeEy3WDKs8,1130
 truefoundry/deploy/lib/dao/workspace.py,sha256=6YvfCgWDzAULI3Q6JswyZmP1CwJ5rM-ANsIFkbQia0Q,2349
 truefoundry/deploy/lib/model/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/deploy/lib/model/entity.py,sha256=Up-DDOezkwM2tdqibfLdZO6jmT2pVq6SShB5sobBIGI,8531
+truefoundry/deploy/lib/model/entity.py,sha256=Bp9sLB-M5INCpw5lPmFdygHWS1zvnLicnSiSCi2iqhQ,8591
 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=FWmqX4N6RUN4-S68i-7xaIifYWnf0wc-zfINdpiFu-4,12901
+truefoundry/deploy/v2/lib/deploy.py,sha256=EOqVQTCMrvs_Kwp8Ft0FZYlRErPuFq4eDBDR1qEZ7_E,11767
 truefoundry/deploy/v2/lib/deploy_workflow.py,sha256=G5BzMIbap8pgDX1eY-TITruUxQdkKhYtBmRwLL6lDeY,14342
 truefoundry/deploy/v2/lib/deployable_patched_models.py,sha256=xbHFD3pURflvCm8EODPvjfvRrv67mlSrjPUknY8SMB8,4060
 truefoundry/deploy/v2/lib/models.py,sha256=ogc1UYs1Z2nBdGSKCrde9sk8d0GxFKMkem99uqO5CmM,1148
-truefoundry/deploy/v2/lib/patched_models.py,sha256=xlgFust2rSnPfQcSe2SW0Sl4gShvZnZNf1UFKjUkPI0,16917
+truefoundry/deploy/v2/lib/patched_models.py,sha256=8ib9Y7b4-DoEml2zCv3V7QIqh4tLJUjzPj1AWomwvag,14775
 truefoundry/deploy/v2/lib/source.py,sha256=d6-8_6Zn5koBglqrBrY6ZLG_7yyPuLdyEmK4iZTw6xY,9405
 truefoundry/ml/__init__.py,sha256=EEEHV7w58Krpo_W9Chd8Y3TdItfFO3LI6j6Izqc4-P8,2219
 truefoundry/ml/constants.py,sha256=vDq72d4C9FSWqr9MMdjgTF4TuyNFApvo_6RVsSeAjB4,2837
@@ -121,7 +125,7 @@ truefoundry/ml/logger.py,sha256=VT-BF3BnBYTWVq87O58F0c8uXMu94gYzsiFlGY3_7Ao,458
 truefoundry/ml/mlfoundry_api.py,sha256=WiIVpJRylBn8NrcAmXClQnavffqI9fhY5h-8Vwx4RKo,59872
 truefoundry/ml/mlfoundry_run.py,sha256=34yyQqgpG6EfrAJd37vkbCjrFoHkhvbOAxxSQcSWPtY,44320
 truefoundry/ml/model_framework.py,sha256=nVbKTtKDRBdLzt7Wrg5_vJKZ-awHbISGvL73s-V9egU,18975
-truefoundry/ml/prompt_utils.py,sha256=muN0P9OTKtm9SLr25e0U0W1vx_v9vWxV3tvAUUC-8HE,2514
+truefoundry/ml/prompt_utils.py,sha256=8FueyElVTXLnLtC3O6hKsW_snocArr_B8KG3Qv6eFIQ,2651
 truefoundry/ml/run_utils.py,sha256=0W208wSLUrbdfk2pjNcZlkUi9bNxG2JORqoe-5rVqHI,2423
 truefoundry/ml/session.py,sha256=2v8JjFyrcKzKQ8qsTaZaQTqlxl-in9odIMyWWMEqWNY,4436
 truefoundry/ml/validation_utils.py,sha256=rmIBNcpmnwq2jeje6uFg8uY-rL4JG_RV6mc6BlujVWg,12150
@@ -362,7 +366,7 @@ truefoundry/ml/log_types/image/constants.py,sha256=wLtGEOA4T5fZHSlOXPuNDLX3lpbCt
 truefoundry/ml/log_types/image/image.py,sha256=sa0tBHdyluC8bELXY16E0HgFrUDnDBxHrteix4BFXcs,12479
 truefoundry/ml/log_types/image/image_normalizer.py,sha256=vrzfuSpVGgIxw_Q2sbFe7kQ_JpAndX0bMwC7wtfi41g,3104
 truefoundry/ml/log_types/image/types.py,sha256=inFQlyAyDvZtfliFpENirNCm1XO9beyZ8DNn97DoDKs,1568
-truefoundry/workflow/__init__.py,sha256=XY83vqtLAclI82atZXyBtF9ZgLROXaaXO5p60XH5hJA,1493
+truefoundry/workflow/__init__.py,sha256=MNxnOh5fzAmDaK-cJy9qwtA-zH2CdOEP2h0q9lEiLgY,1588
 truefoundry/workflow/container_task.py,sha256=8arieePsX4__OnG337hOtCiNgJwtKJJCsZcmFmCBJtk,402
 truefoundry/workflow/map_task.py,sha256=f9vcAPRQy0Ttw6bvdZBKUVJMSm4eGQrbE1GHWhepHIU,1864
 truefoundry/workflow/python_task.py,sha256=SRXRLC4vdBqGjhkwuaY39LEWN6iPCpJAuW17URRdWTY,1128
@@ -372,7 +376,7 @@ truefoundry/workflow/remote_filesystem/__init__.py,sha256=LQ95ViEjJ7Ts4JcCGOxMPs
 truefoundry/workflow/remote_filesystem/logger.py,sha256=em2l7D6sw7xTLDP0kQSLpgfRRCLpN14Qw85TN7ujQcE,1022
 truefoundry/workflow/remote_filesystem/tfy_signed_url_client.py,sha256=xcT0wQmQlgzcj0nP3tJopyFSVWT1uv3nhiTIuwfXYeg,12342
 truefoundry/workflow/remote_filesystem/tfy_signed_url_fs.py,sha256=nSGPZu0Gyd_jz0KsEE-7w_BmnTD8CVF1S8cUJoxaCbc,13305
-truefoundry-0.6.6.dist-info/METADATA,sha256=AnPnseWxHjSlnew5DfTaYf0YkkNlHpB8EN7Oynckakg,2350
-truefoundry-0.6.6.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-truefoundry-0.6.6.dist-info/entry_points.txt,sha256=xVjn7RMN-MW2-9f7YU-bBdlZSvvrwzhpX1zmmRmsNPU,98
-truefoundry-0.6.6.dist-info/RECORD,,
+truefoundry-0.7.0.dist-info/METADATA,sha256=pn3-G0IdxJF6mi-kH-SL8UsdgoGnKo6oOVvgFJJDRao,2407
+truefoundry-0.7.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+truefoundry-0.7.0.dist-info/entry_points.txt,sha256=xVjn7RMN-MW2-9f7YU-bBdlZSvvrwzhpX1zmmRmsNPU,98
+truefoundry-0.7.0.dist-info/RECORD,,