truefoundry 0.5.8rc2__py3-none-any.whl → 0.5.8rc4__py3-none-any.whl

truefoundry/common/entities.py

@@ -1,5 +1,4 @@
 import time
-from enum import Enum
 from typing import Optional
 
 import jwt
@@ -8,18 +7,12 @@ from typing_extensions import NotRequired, TypedDict
 from truefoundry.pydantic_v1 import BaseModel, Field, NonEmptyStr, validator
 
 
-class UserType(Enum):
-    user = "user"
-    serviceaccount = "serviceaccount"
-
-
 class UserInfo(BaseModel):
     class Config:
         allow_population_by_field_name = True
         allow_mutation = False
 
     user_id: NonEmptyStr
-    user_type: UserType = UserType.user
     email: Optional[str] = None
     tenant_name: NonEmptyStr = Field(alias="tenantName")
 
@@ -27,9 +20,16 @@ class UserInfo(BaseModel):
 class _DecodedToken(TypedDict):
     tenantName: str
     exp: int
-    username: str
+    username: NotRequired[str]
     email: NotRequired[str]
-    userType: UserType
+
+
+def _user_slug(decoded_token: _DecodedToken) -> str:
+    return (
+        decoded_token.get("username")
+        or decoded_token.get("email")
+        or "--user-slug-missing--"
+    )
 
 
 class Token(BaseModel):
@@ -55,19 +55,21 @@ class Token(BaseModel):
 
     @property
     def tenant_name(self) -> str:
+        assert self.decoded_value is not None
         return self.decoded_value["tenantName"]
 
     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
 
     def to_user_info(self) -> UserInfo:
+        assert self.decoded_value is not None
         return UserInfo(
-            user_id=self.decoded_value["username"],
+            user_id=_user_slug(self.decoded_value),
             email=self.decoded_value["email"]
             if "email" in self.decoded_value
             else None,
-            user_type=UserType(self.decoded_value.get("userType", UserType.user.value)),
             tenant_name=self.tenant_name,
         )
 

truefoundry/common/session.py

@@ -50,10 +50,9 @@ class Session:
 
             ACTIVE_SESSION = new_session
             logger.info(
-                "Logged in to %r as %r (%s)",
+                "Logged in to %r as %r",
                 new_session.tfy_host,
                 new_session.user_info.user_id,
-                new_session.user_info.email or new_session.user_info.user_type.value,
             )
 
             return ACTIVE_SESSION

truefoundry/deploy/auto_gen/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  application.json
-#   timestamp: 2025-01-29T10:40:43+00:00
+#   timestamp: 2025-02-19T09:48:05+00:00
 
 from __future__ import annotations
 
@@ -1565,7 +1565,7 @@ class Notebook(BaseWorkbenchInput):
     image: WorkbenchImage
     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",
+        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\n+uiProps={"descriptionInline":true}',
     )
 
 
@@ -1656,7 +1656,7 @@ class SSHServer(BaseWorkbenchInput):
     )
     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",
+        description='+label=Stop after (minutes of inactivity)\n+usage=Stop the SSH Server instance after this much time in minutes of inactivity. The instance is considered active if there is at least one active SSH connection (a client connected to the SSH server), or if a background job is running using tmux or screen, or if the pod has restarted.\n+sort=5\n+uiProps={"descriptionInline":true, "warningMessage":"Please note that stop after inactivity is only available for images with tag(including custom images) >= v0.3.10"}',
     )
 
 
@@ -1696,6 +1696,10 @@ class SparkJob(BaseModel):
         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",
     )
+    conf: Optional[Dict[str, Any]] = Field(
+        None,
+        description="+label=Spark Config Properties\n+usage=Extra configuration properties to be passed to the spark job. [Docs](https://spark.apache.org/docs/latest/configuration.html)\n+icon=fa-gear:#68BBE3\n+sort=21500",
+    )
     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",
@@ -1888,6 +1892,10 @@ class ApplicationSet(BaseModel):
         None,
         description="+label=Workspace FQN\n+docs=Fully qualified name of the workspace\n+uiType=Hidden",
     )
+    convert_template_manifest: Optional[bool] = Field(
+        None,
+        description="+label=Convert Template Manifest\n+docs=Flag to indicate if the template manifest should be converted to TrueFoundry manifest\n+uiType=Hidden",
+    )
 
 
 class Application(BaseModel):

truefoundry/deploy/cli/commands/delete_command.py

@@ -1,31 +1,63 @@
+from typing import List, Tuple
+
 import rich_click as click
 
 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_json
 from truefoundry.cli.util import handle_exception_wrapper
 from truefoundry.deploy.io.rich_output_callback import RichOutputCallBack
+from truefoundry.deploy.lib.clients.servicefoundry_client import (
+    ServiceFoundryServiceClient,
+)
 from truefoundry.deploy.lib.dao import application as application_lib
+from truefoundry.deploy.lib.dao import delete as delete_lib
 from truefoundry.deploy.lib.dao import workspace as workspace_lib
 from truefoundry.deploy.lib.messages import (
     PROMPT_DELETED_APPLICATION,
     PROMPT_DELETED_WORKSPACE,
+    PROMPT_DELETING_MANIFEST,
 )
+from truefoundry.deploy.lib.model.entity import DeleteResult
 
 # TODO (chiragjn): --json should disable all non json console prints
 
 
-@click.group(name="delete", cls=GROUP_CLS)
-def delete_command():
+@click.group(name="delete", cls=GROUP_CLS, invoke_without_command=True)
+@click.pass_context
+@click.option(
+    "-f",
+    "--file",
+    "files",
+    type=click.Path(exists=True, dir_okay=False, resolve_path=True),
+    help="Path to yaml manifest file (You can pass multiple files at once by providing multiple -f options)",
+    show_default=True,
+    required=False,
+    multiple=True,
+)
+@handle_exception_wrapper
+def delete_command(ctx, files: Tuple[str, ...]):
     """
     Delete TrueFoundry resources
-
-    \b
-    Supported resources:
-    - Workspace
-    - Application
     """
-    pass
+    if ctx.invoked_subcommand is None:
+        if not files:
+            raise click.UsageError("Missing option '-f' / '--file'")
+        delete_results: List[DeleteResult] = []
+        client = ServiceFoundryServiceClient()
+        for file in files:
+            with console.status(PROMPT_DELETING_MANIFEST.format(file), spinner="dots"):
+                for delete_result in delete_lib.delete_manifest_file(file, client):
+                    if delete_result.success:
+                        console.print(f"[green]\u2714 {delete_result.message}[/]")
+                    else:
+                        console.print(f"[red]\u2718 {delete_result.message}[/]")
+
+                    delete_results.append(delete_result)
+
+        if not all(delete_result.success for delete_result in delete_results):
+            raise Exception("Failed to delete one or more resource manifests")
 
 
 @click.command(name="workspace", cls=COMMAND_CLS, help="Delete a Workspace")

truefoundry/deploy/lib/clients/servicefoundry_client.py

@@ -693,6 +693,16 @@ class ServiceFoundryServiceClient(BaseServiceFoundryServiceClient):
         response_data = request_handling(response)
         return response_data
 
+    @check_min_cli_version
+    def delete(self, manifest: Dict[str, Any]):
+        url = f"{self._api_server_url}/{VERSION_PREFIX}/delete"
+        body = {"manifest": manifest}
+        response = session_with_retries().post(
+            url, headers=self._get_header(), json=body
+        )
+        response_data = request_handling(response)
+        return response_data
+
     def terminate_job_run(
         self,
         deployment_id: str,

truefoundry/deploy/lib/dao/delete.py

@@ -0,0 +1,88 @@
+from pathlib import Path
+from typing import Any, Dict, Iterator, Optional
+
+import yaml
+
+from truefoundry.deploy.lib.clients.servicefoundry_client import (
+    ServiceFoundryServiceClient,
+)
+from truefoundry.deploy.lib.model.entity import DeleteResult, Manifest
+from truefoundry.pydantic_v1 import ValidationError
+
+
+def _delete_manifest(
+    manifest: Dict[str, Any],
+    client: Optional[ServiceFoundryServiceClient] = None,
+    filename: Optional[str] = None,
+    index: Optional[int] = None,
+) -> DeleteResult:
+    client = client or ServiceFoundryServiceClient()
+
+    file_metadata = ""
+    if index is not None:
+        file_metadata += f" at index {index}"
+    if filename:
+        file_metadata += f" from file {filename}"
+
+    try:
+        manifest = Manifest.parse_obj(manifest)
+    except ValidationError as ex:
+        return DeleteResult(
+            success=False,
+            message=f"Failed to parse manifest{file_metadata}. Error: {ex}",
+        )
+
+    try:
+        client.delete(manifest.dict())
+
+        return DeleteResult(
+            success=True,
+            message=(
+                f"Successfully deleted resource manifest {manifest.name} of type {manifest.type}."
+            ),
+        )
+    except Exception as ex:
+        return DeleteResult(
+            success=False,
+            message=(
+                f"Failed to delete resource manifest {manifest.name} of type {manifest.type}. Error: {ex}."
+            ),
+        )
+
+
+def delete_manifest(
+    manifest: Dict[str, Any],
+    client: Optional[ServiceFoundryServiceClient] = None,
+) -> DeleteResult:
+    return _delete_manifest(manifest=manifest, client=client)
+
+
+def delete_manifest_file(
+    filepath: str,
+    client: Optional[ServiceFoundryServiceClient] = None,
+) -> Iterator[DeleteResult]:
+    client = client or ServiceFoundryServiceClient()
+    filename = Path(filepath).name
+    try:
+        with open(filepath, "r") as f:
+            manifests_it = list(yaml.safe_load_all(f))
+    except Exception as ex:
+        yield DeleteResult(
+            success=False,
+            message=f"Failed to read file {filepath} as a valid YAML file. Error: {ex}",
+        )
+    else:
+        for index, manifest in enumerate(manifests_it):
+            if not isinstance(manifest, dict):
+                yield DeleteResult(
+                    success=False,
+                    message=f"Failed to delete resource manifest at index {index} from file {filename}. Error: A manifest must be a dict, got type {type(manifest)}",
+                )
+                continue
+
+            yield _delete_manifest(
+                manifest=manifest,
+                client=client,
+                filename=filename,
+                index=index,
+            )

truefoundry/deploy/lib/messages.py

@@ -10,3 +10,4 @@ PROMPT_NO_WORKSPACES = f"""[yellow]No workspaces found. Either cluster name is w
 PROMPT_NO_APPLICATIONS = f"""[yellow]No applications found. You can create one with [bold]{TFY} deploy[/] from within your application folder"""
 PROMPT_NO_VERSIONS = """[yellow]No application versions found."""
 PROMPT_APPLYING_MANIFEST = """[yellow]Applying manifest for file {!r}[/]"""
+PROMPT_DELETING_MANIFEST = """[yellow]Deleting manifest for file {!r}[/]"""

truefoundry/deploy/lib/model/entity.py

@@ -280,6 +280,11 @@ class ApplyResult(BaseModel):
     message: str
 
 
+class DeleteResult(BaseModel):
+    success: bool
+    message: str
+
+
 class Manifest(Base):
     type: str
     name: str

truefoundry/deploy/lib/session.py

@@ -57,10 +57,9 @@ def login(
                     return login(api_key=api_key, host=host, relogin=True)
 
             user_info = cred_file_content.to_token().to_user_info()
-            user_name_display_info = user_info.email or user_info.user_type.value
             output_hook.print_line(
                 relogin_error_message(
-                    f"Already logged in to {cred_file_content.host!r} as {user_info.user_id!r} ({user_name_display_info})",
+                    f"Already logged in to {cred_file_content.host!r} as {user_info.user_id!r}",
                     host=host,
                 )
             )
@@ -81,10 +80,8 @@ def login(
         cred_file.write(cred_file_content)
 
     user_info = token.to_user_info()
-    user_name_display_info = user_info.email or user_info.user_type.value
     output_hook.print_line(
-        f"Successfully logged in to {cred_file_content.host!r} as "
-        f"{user_info.user_id!r} ({user_name_display_info})"
+        f"Successfully logged in to {cred_file_content.host!r} as {user_info.user_id!r}"
     )
     return True
 

truefoundry/deploy/python_deploy_codegen.py

@@ -27,7 +27,7 @@ logging.basicConfig(level=logging.INFO)
 
 {application_type} = {spec_repr}
 
-{application_type}.deploy(workspace_fqn="{workspace_fqn}")\
+{application_type}.deploy(workspace_fqn="{workspace_fqn}", wait=False)\
 """
     return code
 

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

@@ -247,8 +247,7 @@ class HealthApi:
         for _key, _val in _params["kwargs"].items():
             if _key not in _all_params:
                 raise ApiTypeError(
-                    "Got an unexpected keyword argument '%s'"
-                    " to method serve_get" % _key
+                    "Got an unexpected keyword argument '%s' to method serve_get" % _key
                 )
             _params[_key] = _val
         del _params["kwargs"]

truefoundry/ml/autogen/client/exceptions.py

@@ -115,7 +115,7 @@ class ApiException(OpenApiException):
 
     def __str__(self):
         """Custom error messages for exception"""
-        error_message = "({0})\n" "Reason: {1}\n".format(self.status, self.reason)
+        error_message = "({0})\nReason: {1}\n".format(self.status, self.reason)
         if self.headers:
             error_message += "HTTP response headers: {0}\n".format(self.headers)
 

truefoundry/ml/autogen/models/schema.py

@@ -767,7 +767,7 @@ class TensorInfo:
     def __init__(self, dtype: np.dtype, shape: Union[tuple, list]):
         if not isinstance(dtype, np.dtype):
             raise TypeError(
-                f"Expected `dtype` to be instance of `{np.dtype}`, received `{ dtype.__class__}`"
+                f"Expected `dtype` to be instance of `{np.dtype}`, received `{dtype.__class__}`"
             )
         # Throw if size information exists flexible numpy data types
         if dtype.char in ["U", "S"] and not dtype.name.isalpha():

truefoundry/ml/session.py

@@ -75,10 +75,9 @@ class MLFoundrySession(Session):
 
             ACTIVE_SESSION = new_session
             logger.info(
-                "Logged in to %r as %r (%s)",
+                "Logged in to %r as %r",
                 new_session.tfy_host,
                 new_session.user_info.user_id,
-                new_session.user_info.email or new_session.user_info.user_type.value,
             )
 
             return ACTIVE_SESSION

{truefoundry-0.5.8rc2.dist-info → truefoundry-0.5.8rc4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: truefoundry
-Version: 0.5.8rc2
+Version: 0.5.8rc4
 Summary: TrueFoundry CLI
 Author-email: TrueFoundry Team <abhishek@truefoundry.com>
 Requires-Python: <3.13,>=3.8.1

{truefoundry-0.5.8rc2.dist-info → truefoundry-0.5.8rc4.dist-info}/RECORD

@@ -38,18 +38,18 @@ truefoundry/common/auth_service_client.py,sha256=N3YxKlx63r6cPZqbgb2lqBOPI69ShB7
 truefoundry/common/constants.py,sha256=WQjylPv7xTVYURv4zicJUHYL73Gv3c3i6Iw4oy9Lhg4,2744
 truefoundry/common/credential_file_manager.py,sha256=1yEk1Zm2xS4G0VDFwKSZ4w0VUrcPWQ1nJnoBaz9xyKA,4251
 truefoundry/common/credential_provider.py,sha256=_OhJ2XFlDaVsrUO-FyywxctcGGqDdC2pgcvwEKqQD0Q,4071
-truefoundry/common/entities.py,sha256=12olLKzRT092Jnze5busK68cwH8wwyL6ch77ohKOFPU,3723
+truefoundry/common/entities.py,sha256=Gx52dIwUARwGLE2hCSe9PJWfXh-ARCmdWwX6KtCxJng,3805
 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=BiTJ2WEzLkTvzpqy67xX3fXgiKH_ltzzmQdFG7xUtwI,2904
+truefoundry/common/session.py,sha256=xeBAPUNEJv2XVFQCRUGeBDTePh5zrKNSok8vmSxBjPw,2813
 truefoundry/common/storage_provider_utils.py,sha256=yURhMw8k0FLFvaviRHDiifhvc6GnuQwGMC9Qd2uM440,10934
 truefoundry/common/types.py,sha256=BMJFCsR1lPJAw66IQBSvLyV4I6o_x5oj78gVsUa9si8,188
 truefoundry/common/utils.py,sha256=E1kKw_hqlVMgwyPkER2SvVLIoQ-B_4TQCuM5IJXvgig,6046
 truefoundry/common/warnings.py,sha256=rs6BHwk7imQYedo07iwh3TWEOywAR3Lqhj0AY4khByg,504
 truefoundry/deploy/__init__.py,sha256=e0EgG-fVJ6N9PNIlBINO6QtIMEehQgmz3ONPPjYlcSY,2596
-truefoundry/deploy/python_deploy_codegen.py,sha256=qJHH1BJQII9e6PhkcRFYiE_3De7_VMMm8nM4AX5Eq1o,6513
-truefoundry/deploy/auto_gen/models.py,sha256=83OoBHcu3VP8Xe75vBNaQmPGjqxhsSkHAZZZD4mROOM,87620
+truefoundry/deploy/python_deploy_codegen.py,sha256=AainOFR20XvhNeztJkLPWGZ40lAT_nwc-ZmG77Kum4o,6525
+truefoundry/deploy/auto_gen/models.py,sha256=u9gRv78pTebU3C0_DJI2UkvY-hyZGiJ80Z_AoUq1Ug4,88347
 truefoundry/deploy/builder/__init__.py,sha256=1qjHMNBE1poRCZW0WrG46dFM1f1IlivD5352qzsioMU,4953
 truefoundry/deploy/builder/constants.py,sha256=amUkHoHvVKzGv0v_knfiioRuKiJM0V0xW0diERgWiI0,508
 truefoundry/deploy/builder/docker_service.py,sha256=sm7GWeIqyrKaZpxskdLejZlsxcZnM3BTDJr6orvPN4E,3948
@@ -65,7 +65,7 @@ truefoundry/deploy/cli/commands/__init__.py,sha256=wn_TsmrImj9pER16O5aK3aW2mfqbR
 truefoundry/deploy/cli/commands/apply_command.py,sha256=Y2e_C8HVpo8CssVod-3JRz-89qStC5JRaNzJ7O2mRlY,2039
 truefoundry/deploy/cli/commands/build_command.py,sha256=zJBywMatbpUlXx5O2aqpEVmPeBIJ9RNnG9abSc8C8CE,1234
 truefoundry/deploy/cli/commands/create_command.py,sha256=rCajvQvAfZU10nDZOYpRACbAATH1zj52ihTWrhnLLUc,1829
-truefoundry/deploy/cli/commands/delete_command.py,sha256=JMibxvt1rsZ8sODa0jI1e6vavjN8Pv04ttVFURSTKTw,2362
+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
@@ -87,19 +87,20 @@ truefoundry/deploy/io/rich_output_callback.py,sha256=TJLiRD-EnFVwgcepxR7WN0koKqW
 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=nhp0bCYf_XpUM68hTq5lBY-__vtEyV2uP7NgnJXJ_Vg,925
-truefoundry/deploy/lib/session.py,sha256=FBSq3bxf3my_613NzgtPoeeaAYLGZEoF0n7XccTFNf4,5141
+truefoundry/deploy/lib/messages.py,sha256=8424kj3kqCyDCX5Nr2WJZZ_UEutPoaSs_y2f9-O4yy8,1001
+truefoundry/deploy/lib/session.py,sha256=-FX4gOtiGlc6Jk56JPVZpDqXR9xQza77AIlBvNJpYqQ,4919
 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=xW_JGOk28bVMqdRiggCYMJFIx7iPjOUCdg8DDxfpjmo,26341
+truefoundry/deploy/lib/clients/servicefoundry_client.py,sha256=RIlJnpBivhBXlDJZFaNQLIdIsFF6RpHKIDK1QMjqHBg,26712
 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=Jadfa3DcbIRz-Bb4Tlixed179HLDnmDXOlrvSxfEgQ0,3021
+truefoundry/deploy/lib/dao/delete.py,sha256=ajtnKnffhv8Uak4GhJx4npPMJPe2SHyWKIqb3vkAszw,2727
 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=3YRdyvMYI1aUXUu2tB5b5lZWwpxKdAhjpT79gzTCZGo,8373
+truefoundry/deploy/lib/model/entity.py,sha256=y6sj4QKxa5GEEu85j72bq9uTPXvJv4iSqaww65a9FKw,8441
 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=OvotIwCzluaOqX3xceV5sb86WFXYNs4fXn6FJsR9KBU,12420
@@ -120,7 +121,7 @@ truefoundry/ml/mlfoundry_api.py,sha256=PUBGDcjrZsJKQYy1PjT8NYGYxrTvhhHzQ-0jLSl7-
 truefoundry/ml/mlfoundry_run.py,sha256=w9_Tph48eTDjna-rZ1fPa6PtX4JzhF-t5PS2hmbK2rY,44319
 truefoundry/ml/model_framework.py,sha256=JGmzdG7o5P6CJr5EYTUOmuly53FfhpoNVqKrhfijAVg,18972
 truefoundry/ml/run_utils.py,sha256=0W208wSLUrbdfk2pjNcZlkUi9bNxG2JORqoe-5rVqHI,2423
-truefoundry/ml/session.py,sha256=eQo1zG5nd0-8Ok68uEBCQDPAWYeJkpzqHak-DQX1ENU,4553
+truefoundry/ml/session.py,sha256=IRXVoAAJTHik3md-pE4IpTPSGJvs5qdAyWBf0O9_TJo,4462
 truefoundry/ml/validation_utils.py,sha256=J5atNhcJLvKj64ralSV9Y5Fv1Rt4SE237ICdP9-7sP4,12149
 truefoundry/ml/artifact/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/ml/artifact/truefoundry_artifact_repo.py,sha256=myDpr3RZzEkd_UG9C3rZ4UKEFxXopO5HMVJNpNDiuAo,37737
@@ -130,14 +131,14 @@ truefoundry/ml/autogen/client/__init__.py,sha256=Kh1qP7S8L-e7ItofskXQf3y7QFfmxGP
 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/exceptions.py,sha256=VGKETfmF3kru3LQ0Ao226B1keHhyLMMnjEd7aScP45c,5352
 truefoundry/ml/autogen/client/rest.py,sha256=9goba8qHjQuVx5O_yRaTKu7PvBnb7r7swfy3dwuTEgk,14281
 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
 truefoundry/ml/autogen/client/api/experiments_api.py,sha256=VEwgHJca_mmfxYaoaJknQg_YYQOaY8Sd_Hx2wW8CSA8,68174
 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/health_api.py,sha256=BMC9_V1UYqcsvdiSC9XXQLyaosnReg0a_r8Lvnru_i4,11531
 truefoundry/ml/autogen/client/api/metrics_api.py,sha256=q3L38eD-2hu4_9YvcSdnWDYXD2V8il-X9reinOAABek,14908
 truefoundry/ml/autogen/client/api/mlfoundry_artifacts_api.py,sha256=T8_cYdrd60KzDddRxX3nDeIQe6ihMq-jD8BOAAmzPHo,309716
 truefoundry/ml/autogen/client/api/run_artifacts_api.py,sha256=x-vVnY2LEFChZxiiFauswRWwFz6Qqh30PKXjzuTvxmc,8799
@@ -327,7 +328,7 @@ truefoundry/ml/autogen/client/models/xg_boost_serialization_format.py,sha256=wFF
 truefoundry/ml/autogen/entities/artifacts.py,sha256=TOm71ZrNo36le3YyMEsE1Cs_BYyMLkFQ7hQtAaMoB9M,21251
 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
+truefoundry/ml/autogen/models/schema.py,sha256=a_bp42MMPUbwO3407m0UW2W8EOhnxZXfSx3xS4aszVs,55771
 truefoundry/ml/autogen/models/signature.py,sha256=rBjpxUIsEeWM0sIyYG5uCJB18DKHR4k5yZw8TzuoP48,4987
 truefoundry/ml/autogen/models/utils.py,sha256=c7RtSLXhOLcP8rjuUtfnMdaKVTZvvbsmw98gPAkAFrs,24371
 truefoundry/ml/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -363,7 +364,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.5.8rc2.dist-info/METADATA,sha256=6TAdw_xKTV9A3DU7jZMAJp5yCSW6NyGzDtit4D8_1tg,2523
-truefoundry-0.5.8rc2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-truefoundry-0.5.8rc2.dist-info/entry_points.txt,sha256=xVjn7RMN-MW2-9f7YU-bBdlZSvvrwzhpX1zmmRmsNPU,98
-truefoundry-0.5.8rc2.dist-info/RECORD,,
+truefoundry-0.5.8rc4.dist-info/METADATA,sha256=FmamDp6h0WeTwzoj2Nt1uhx8K6jiacb-foS695wCBOA,2523
+truefoundry-0.5.8rc4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+truefoundry-0.5.8rc4.dist-info/entry_points.txt,sha256=xVjn7RMN-MW2-9f7YU-bBdlZSvvrwzhpX1zmmRmsNPU,98
+truefoundry-0.5.8rc4.dist-info/RECORD,,