truefoundry 0.4.0rc2__py3-none-any.whl → 0.4.0rc3__py3-none-any.whl

truefoundry/ml/session.py

@@ -1,131 +1,29 @@
-import abc
 import atexit
-import os
 import threading
 import weakref
 from typing import TYPE_CHECKING, Dict, Optional
 
+from truefoundry.common.credential_provider import (
+    CredentialProvider,
+    EnvCredentialProvider,
+    FileCredentialProvider,
+)
+from truefoundry.common.entities import Token, UserInfo
 from truefoundry.common.request_utils import urllib3_retry
-from truefoundry.ml import env_vars
 from truefoundry.ml.autogen.client import (  # type: ignore[attr-defined]
     ApiClient,
     Configuration,
 )
+from truefoundry.ml.clients.entities import HostCreds
 from truefoundry.ml.exceptions import MlFoundryException
 from truefoundry.ml.logger import logger
-from truefoundry.ml.login import CredentialsFileContent, CredentialsFileManager
 from truefoundry.ml.run_utils import (
     append_path_to_rest_tracking_uri,
-    resolve_tracking_uri,
 )
-from truefoundry.ml.services.auth_service import get_auth_service
-from truefoundry.ml.services.entities import HostCreds, Token, UserInfo
 
 if TYPE_CHECKING:
     from truefoundry.ml.mlfoundry_run import MlFoundryRun
 
-TOKEN_REFRESH_LOCK = threading.RLock()
-
-
-class CredentialProvider(abc.ABC):
-    @property
-    @abc.abstractmethod
-    def token(self) -> Token: ...
-
-    @staticmethod
-    @abc.abstractmethod
-    def can_provide() -> bool: ...
-
-    @property
-    @abc.abstractmethod
-    def tracking_uri(self) -> str: ...
-
-
-class EnvCredentialProvider(CredentialProvider):
-    def __init__(self):
-        logger.debug("Using env var credential provider")
-        self._tracking_uri = resolve_tracking_uri(tracking_uri=None)
-        self._auth_service = get_auth_service(tracking_uri=self._tracking_uri)
-        api_key = os.getenv(env_vars.API_KEY_GLOBAL)
-        if not api_key:
-            raise MlFoundryException(
-                f"Value of {env_vars.API_KEY_GLOBAL} env var should be non-empty string"
-            )
-        self._token: Token = Token(access_token=api_key, refresh_token=None)  # type: ignore[call-arg]
-
-    @staticmethod
-    def can_provide() -> bool:
-        return env_vars.API_KEY_GLOBAL in os.environ
-
-    @property
-    def token(self) -> Token:
-        with TOKEN_REFRESH_LOCK:
-            if self._token.is_going_to_be_expired():
-                logger.info("Refreshing access token")
-                self._token = self._auth_service.refresh_token(self._token)
-            return self._token
-
-    @property
-    def tracking_uri(self) -> str:
-        return self._tracking_uri
-
-
-class FileCredentialProvider(CredentialProvider):
-    def __init__(self):
-        logger.debug("Using file credential provider")
-        self._cred_file = CredentialsFileManager()
-
-        with self._cred_file:
-            self._last_cred_file_content = self._cred_file.read()
-            self._tracking_uri = self._last_cred_file_content.host
-            self._token = self._last_cred_file_content.to_token()
-            self._auth_service = get_auth_service(tracking_uri=self._tracking_uri)
-
-    @staticmethod
-    def can_provide() -> bool:
-        with CredentialsFileManager() as cred_file:
-            return cred_file.exists()
-
-    @property
-    def token(self) -> Token:
-        with TOKEN_REFRESH_LOCK:
-            if not self._token.is_going_to_be_expired():
-                return self._token
-
-            logger.info("Refreshing access token")
-            with self._cred_file:
-                new_cred_file_content = self._cred_file.read()
-                new_token = new_cred_file_content.to_token()
-                new_tracking_uri = new_cred_file_content.host
-
-                if new_cred_file_content == self._last_cred_file_content:
-                    self._token = self._auth_service.refresh_token(self._token)
-                    self._last_cred_file_content = CredentialsFileContent(
-                        host=self._tracking_uri,
-                        access_token=self._token.access_token,
-                        refresh_token=self._token.refresh_token,
-                    )
-                    self._cred_file.write(self._last_cred_file_content)
-                    return self._token
-
-                if (
-                    new_tracking_uri == self._tracking_uri
-                    and new_token.to_user_info() == self._token.to_user_info()
-                ):
-                    self._last_cred_file_content = new_cred_file_content
-                    self._token = new_token
-                    # recursive
-                    return self.token
-
-                raise MlFoundryException(
-                    "Credentials on disk changed while mlfoundry was running."
-                )
-
-    @property
-    def tracking_uri(self) -> str:
-        return self._tracking_uri
-
-
 SESSION_LOCK = threading.RLock()
 
 
@@ -159,8 +57,8 @@ class Session:
     def __init__(self, cred_provider: CredentialProvider):
         # Note: Whenever a new session is initialized all the active runs are ended
         self._closed = False
-        self._cred_provider: CredentialProvider = cred_provider
-        self._user_info: UserInfo = self._cred_provider.token.to_user_info()
+        self._cred_provider: Optional[CredentialProvider] = cred_provider
+        self._user_info: Optional[UserInfo] = self._cred_provider.token.to_user_info()
 
     def close(self):
         logger.debug("Closing existing session")
@@ -187,7 +85,7 @@ class Session:
 
     @property
     def tracking_uri(self) -> str:
-        return self._cred_provider.tracking_uri
+        return self._cred_provider.base_url
 
     def __eq__(self, other: object) -> bool:
         if not isinstance(other, Session):
@@ -199,9 +97,7 @@ class Session:
         )
 
     def get_host_creds(self) -> HostCreds:
-        tracking_uri = append_path_to_rest_tracking_uri(
-            self._cred_provider.tracking_uri
-        )
+        tracking_uri = append_path_to_rest_tracking_uri(self._cred_provider.base_url)
         return HostCreds(
             host=tracking_uri, token=self._cred_provider.token.access_token
         )
@@ -235,6 +131,7 @@ def _get_api_client(
     )
     configuration.retries = urllib3_retry(retries=2)
     api_client = ApiClient(configuration=configuration)
+    api_client.user_agent = "truefoundry-cli"
     return api_client
 
 
@@ -247,8 +144,8 @@ def init_session() -> Session:
                 break
         if final_cred_provider is None:
             raise MlFoundryException(
-                "Please login using `mlfoundry login` command "
-                "or `truefoundry.ml.login()` function call"
+                "Please login using `tfy login` command "
+                "or `truefoundry.login()` function call"
             )
         new_session = Session(cred_provider=final_cred_provider)
 

truefoundry/pydantic_v1.py

@@ -5,7 +5,7 @@ try:
 except ImportError:
     # pydantic <=1.10.17
     from pydantic import *  # noqa: F403
-    from pydantic import utils  # noqa: F401
+    from pydantic import ConstrainedStr, utils  # noqa: F401
 
 
 class NonEmptyStr(ConstrainedStr):

truefoundry/workflow/__init__.py

@@ -1,5 +1,5 @@
 try:
-    from flytekit import task
+    from flytekit import task as _
 except ImportError:
     print("To use workflows, please run 'pip install truefoundry[workflow]'.")
 
@@ -17,3 +17,18 @@ from truefoundry.workflow.map_task import map_task
 from truefoundry.workflow.python_task import PythonFunctionTask
 from truefoundry.workflow.task import task
 from truefoundry.workflow.workflow import ExecutionConfig, workflow
+
+__all__ = [
+    "task",
+    "ContainerTask",
+    "PythonFunctionTask",
+    "map_task",
+    "workflow",
+    "conditional",
+    "FlyteDirectory",
+    "TaskDockerFileBuild",
+    "TaskPythonBuild",
+    "ContainerTaskConfig",
+    "PythonTaskConfig",
+    "ExecutionConfig",
+]

{truefoundry-0.4.0rc2.dist-info → truefoundry-0.4.0rc3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: truefoundry
-Version: 0.4.0rc2
+Version: 0.4.0rc3
 Summary: Truefoundry CLI
 Author: Abhishek Choudhary
 Author-email: abhishek@truefoundry.com

{truefoundry-0.4.0rc2.dist-info → truefoundry-0.4.0rc3.dist-info}/RECORD

@@ -1,11 +1,11 @@
-truefoundry/__init__.py,sha256=WFspBK4kl1ApJGf0mmHE4ab4-pi9iPa69sZyRE6kaD4,50
+truefoundry/__init__.py,sha256=jynqfY0pDmcaXlxewDnMQC5CYAIW9JFjptBrw7y76rY,81
 truefoundry/autodeploy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/autodeploy/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/autodeploy/agents/base.py,sha256=woCry229x7WX0IZmBopW_e_NB76aCLQRLXZBs2QBklk,6405
-truefoundry/autodeploy/agents/developer.py,sha256=u1BfjaMDX6Ad32vWYOEFZPWFK33ux0Ro8JztrRAFviY,4126
-truefoundry/autodeploy/agents/project_identifier.py,sha256=JVaBUI_u4IVuohzYyEqojYUzKBxKkSXyqPnzzh-prJA,4504
-truefoundry/autodeploy/agents/tester.py,sha256=9WSVU2MNiamdUNJYrHXDcEq_Q544LSElJ6Nt3XkJ8TU,2402
-truefoundry/autodeploy/cli.py,sha256=nReMRVwVkE1jqtZe4E5cs-fk3S-rleLvsThYtxsvHsI,14223
+truefoundry/autodeploy/agents/developer.py,sha256=Fh-EJDT3wPwCPYsrxRNtanyKMiZxhK73LHb-5Kpmh0A,4127
+truefoundry/autodeploy/agents/project_identifier.py,sha256=dm1A6MHNA-0Sc48y7dYpSA7e4uvw3Wi8AADSs-JcOZs,4505
+truefoundry/autodeploy/agents/tester.py,sha256=Hp1bul8Xd6qZPdUDUPqQwwHF7bsPQ5tK7x6gaBuX0NI,2403
+truefoundry/autodeploy/cli.py,sha256=GkUuiMqtg8U8qachhA3-YgQVUH1PxM6m6yjIwoXuHWg,14222
 truefoundry/autodeploy/constants.py,sha256=vTh2nA7cjqghqbW2rNh3FbtcIk2scdAWZuuQCmVBO80,1273
 truefoundry/autodeploy/exception.py,sha256=fa_ZyTDUKiMKG2Uayynk1yWcEMsuVluqk2GtV4tfTPU,158
 truefoundry/autodeploy/logger.py,sha256=tkV2UKcOTFl5nz0cn4eRbzxF-2CZd8b7MK9vnhaflYw,325
@@ -16,7 +16,7 @@ truefoundry/autodeploy/tools/commit.py,sha256=bB5kqAY5rEvfc7ZHtdueXusGznAbk58xqv
 truefoundry/autodeploy/tools/docker_build.py,sha256=TwSvmxrjBkI7PcPhjIzsc8oTA6fWzPIIpkMv2zMpb-E,4000
 truefoundry/autodeploy/tools/docker_run.py,sha256=8skvqF9zmZSIBbWkmJsmOyQRVz8ttw1SVjPhjGwDq24,5061
 truefoundry/autodeploy/tools/file_type_counts.py,sha256=RAiArWx2q1WBALJq6LKN8XMrtwecZiQrD6VtAdmTuvE,2354
-truefoundry/autodeploy/tools/list_files.py,sha256=p7vvLMWtNu9X1K29Hy4i3i_UzW4cOlu7OkQeY67QLSw,2643
+truefoundry/autodeploy/tools/list_files.py,sha256=mBhpnL0zY3Va2RM8Uuap-2G5Z-QPelSKp6O38K4DbGI,2642
 truefoundry/autodeploy/tools/read_file.py,sha256=uejLuCB9Tjki0-Ud_mk5fuo9nmKRHYHYqqzCTE1nuyo,2254
 truefoundry/autodeploy/tools/send_request.py,sha256=0Si6tdNEK152GUUkLPKKfV6K5vaS8kPw2FKIstRJKAA,1688
 truefoundry/autodeploy/tools/write_file.py,sha256=EcYetnJ4X-NWsd_CdYUKfOnAdz2XXiRukTzJfb_GzFs,3130
@@ -26,9 +26,17 @@ truefoundry/autodeploy/utils/pydantic_compat.py,sha256=hEAUy5kLjhPdzw7yGZ2iXGMXb
 truefoundry/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/cli/__main__.py,sha256=-NkhYlT3mC5MhtekueKAvCw-sWvguj0LJRpXWzvvFjc,727
 truefoundry/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/common/request_utils.py,sha256=UTeEbNzSswelwqnV3mAFRh8CAuKVGnKhSvQ8chN3Bv0,1827
+truefoundry/common/auth_service_client.py,sha256=upUTo6VODAaOKtszCQZNWnPpRT8CEhpCq9RLBJ0RrRY,7840
+truefoundry/common/constants.py,sha256=kHP5ebp7EMfEYaOC-K8T_XPfYi7mXttYgrXYS8bOgkg,359
+truefoundry/common/credential_file_manager.py,sha256=MRn5h9Gm4pxZEF_XT-_b2agMBiqyw0o2SLpq9ty_1lY,4255
+truefoundry/common/credential_provider.py,sha256=UHr3tH39wGjhC3XdRUdgW7IZeh0cHSIEYJMVO8B4eyw,4107
+truefoundry/common/entities.py,sha256=FB7tcOl3oBKX_q5IkvQJYLAIv5WmRrgUSubDx_6fxzc,3771
+truefoundry/common/exceptions.py,sha256=ePpiQ_zmWe4e94gOgeMiyP_AZnKwjEBfyXsB5ScGYcI,329
+truefoundry/common/request_utils.py,sha256=-ss8033PClJOMJuS5Ue1zCFRVK7ZW9vjtva1b5G9uok,2849
+truefoundry/common/servicefoundry_client.py,sha256=ornmmhjSY2y-NJeBcoHPKxX3q9Z9XilcHERFSFk-dSU,2886
+truefoundry/common/utils.py,sha256=DbyLQM4WiKxzIStg2Ekv7fLs8amZR0HtsF75uqy8tMc,1741
 truefoundry/deploy/__init__.py,sha256=ugawKF2G02EmEXX35oZ2tec12d9oWN28Sf6mtGGIERY,2281
-truefoundry/deploy/auto_gen/models.py,sha256=YhvSvYruchLu4jx9UgKUUWJRekgbdEfVHLhdAkbvHMw,79990
+truefoundry/deploy/auto_gen/models.py,sha256=I4-tV2V74mljN7lH606pCD5LArK6MvND_i7Vt_aRI5Q,79952
 truefoundry/deploy/builder/__init__.py,sha256=a1qR6nicHGcxRaeNTxWRsmDs8zsmXc7j13-I8q0qqVk,4938
 truefoundry/deploy/builder/builders/__init__.py,sha256=tlFLXqyDaKLd4iZbo4Hcu_8gOmgtL6drnXpbmQ6x1P8,636
 truefoundry/deploy/builder/builders/dockerfile.py,sha256=AXXTziCkaqIhuM_bwyD1vT1znOwemN1TKgU7eyo-KuM,1522
@@ -38,29 +46,29 @@ truefoundry/deploy/builder/builders/tfy_python_buildpack/__init__.py,sha256=n7Mw
 truefoundry/deploy/builder/builders/tfy_python_buildpack/dockerfile_template.py,sha256=vFmFeK38-t8booJEGREapEjrIL8xnyOQeRSJQq7d3ZQ,6183
 truefoundry/deploy/builder/docker_service.py,sha256=vQS15790njzlFJZ3JW6txYLBdT11ltxqqpf78ZFL_Ng,5208
 truefoundry/deploy/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/deploy/cli/cli.py,sha256=3kECntNXNIYFIAud68THiqRFbm6_rR2qz7F-URlaWPQ,2739
+truefoundry/deploy/cli/cli.py,sha256=o19n7MsMzXXXG3GlpRYaxX3l9Xvtof-iPK4-y31Dw9k,2811
 truefoundry/deploy/cli/commands/__init__.py,sha256=oOstv608ueOJTxR3qlQbZBIdo-FsFe2O0NYNOXi01-s,1276
-truefoundry/deploy/cli/commands/apply_command.py,sha256=Xzky2csXwLARTpQBcPj87imnDu8tIDRN0GnqY8ebEuw,1863
-truefoundry/deploy/cli/commands/build_command.py,sha256=QwKkZ3nVecPMzs-R57YP-_ih9rjTSnhwBWX5tqHVsvU,1255
+truefoundry/deploy/cli/commands/apply_command.py,sha256=FdTeuyOPEoPSU6GP_aQGNOJV0S0rm7PkxrAlXiNvoX8,1864
+truefoundry/deploy/cli/commands/build_command.py,sha256=DQ7NARgkIgV4z0Zdnl3zMDKU_fSkN5-FkjCQypgmWpo,1255
 truefoundry/deploy/cli/commands/build_logs_command.py,sha256=WrPOlFec_wwuzdJmKZ8mjca-oFVvxgfblcqj2LlhWJA,2804
 truefoundry/deploy/cli/commands/create_command.py,sha256=ZjA4EP1jHYuVE1zx0kN-giBr3y0sEiXnu8xMsNyD2Rg,1850
 truefoundry/deploy/cli/commands/delete_command.py,sha256=4tyIameA1pVu9uZZNJPK6rqdV-cJogf51iCCrG-9noI,2390
-truefoundry/deploy/cli/commands/deploy_command.py,sha256=SpCd_1OxA8ZEUyJ6Jkf5Ay8UNakCCLGR--svMHp9EEM,2858
+truefoundry/deploy/cli/commands/deploy_command.py,sha256=f3Bd0EClT4y6Eahw0pX2MFKiRD29j5kHlWVK5rbvK20,2858
 truefoundry/deploy/cli/commands/get_command.py,sha256=w7h5C4bJpmHJ99rgiGg9J_X0xi8aZUeB6Q-SoZUV1tg,5979
 truefoundry/deploy/cli/commands/list_command.py,sha256=cFARY22L5xspBX7TWsx41IF4RiRMok7KBwv7hQRFXNs,4498
-truefoundry/deploy/cli/commands/login_command.py,sha256=Kin6ZdmeADvXOGA08M5_t1MwMnu3QEUfjz1seJurstw,1032
+truefoundry/deploy/cli/commands/login_command.py,sha256=WV4Ad8PJ2_yNfCJi1VWW2GNwU6khZ2sWZbKebxJ7oVM,1038
 truefoundry/deploy/cli/commands/logout_command.py,sha256=hgw6LhJtF3xJ6YyQ_VrBuUWY0WmT8tPYbOEOhMu1eXU,550
 truefoundry/deploy/cli/commands/logs_command.py,sha256=fQSY4o_B78WEJDD5qULWPFQJsCXLJLTr0RI1QjmX1WE,4154
-truefoundry/deploy/cli/commands/patch_application_command.py,sha256=Zuswj36o-qniklOsOjEPUKnd9f5mHv_U7rOPPwALrZ8,2457
-truefoundry/deploy/cli/commands/patch_command.py,sha256=OQCmxcn1Qd2iZfBUaqs48oUKjHVZ13SoMsWLhksPvks,1682
+truefoundry/deploy/cli/commands/patch_application_command.py,sha256=M2PR2nk9ToUGwXP7i1DcGKCvPyKdH6BQpowXT2yXkj0,2457
+truefoundry/deploy/cli/commands/patch_command.py,sha256=bwZdBDmLq9tMOGh5Zq2HJZs2QhfxMRJNFRMxwXfBozY,1679
 truefoundry/deploy/cli/commands/redeploy_command.py,sha256=-wMQLeMcBdJsQBMgTv7rS6hCqH1EJsSKgPrAXU9HjxU,1032
-truefoundry/deploy/cli/commands/terminate_comand.py,sha256=RTNuykw5DBvid44u2nCka_Eu0XSK3Qrzwob9fybHLgQ,1097
+truefoundry/deploy/cli/commands/terminate_comand.py,sha256=G1XFUtSebhCB1EOb4bM5v7CaVZs4SYS9X4yLsOt6Ffs,1098
 truefoundry/deploy/cli/commands/trigger_command.py,sha256=Js6x7-TEuQQN1fTBoILTJiCt76TdH9qpAnkCnei6vYo,4730
 truefoundry/deploy/cli/config.py,sha256=tf8w4UfVzcC6eYkENvuuCPYt_V3sqVpO1bclORV9tAk,206
 truefoundry/deploy/cli/console.py,sha256=9-dMy4YPisCJQziRKTg8Qa0UJnOGl1soiUnJjsnLDvE,242
 truefoundry/deploy/cli/const.py,sha256=dVHPo1uAiDSSMXwXoT2mR5kNQjExT98QNVRz98Hz_Ts,510
 truefoundry/deploy/cli/display_util.py,sha256=gq8EBdpBMHUzYQp_hxOg9EOYi08FIHgOVFCqXqC2tuo,3044
-truefoundry/deploy/cli/util.py,sha256=0bc0jUjfnxLfAblDt4D4DroCw7aTgnxvIvkElBZscDU,3550
+truefoundry/deploy/cli/util.py,sha256=n2-YzdYkyJV1w7XzzWCLtksoaY35I6tM58T_qNpm4vY,3587
 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
@@ -71,7 +79,7 @@ truefoundry/deploy/function_service/build.py,sha256=XWB4bIIii76Cwq5UBupt5WUNs6In
 truefoundry/deploy/function_service/remote/__init__.py,sha256=X1zEW6SSuemydmparLolvQuQSmwXkWUnx516xm3IUaA,199
 truefoundry/deploy/function_service/remote/context.py,sha256=g6HaHM-w-dcfswJFcu6muYhbyFCJID64XnuPN_c0fXo,67
 truefoundry/deploy/function_service/remote/method.py,sha256=DLLgtdQcnTmTiAfeX6pvqM57P1UPbO7kF9B-hPugAGo,1867
-truefoundry/deploy/function_service/remote/remote.py,sha256=wnwF_9lyfeoO5F_buEFFR1gaDYom3UgHd3IeHcl0M5Y,4202
+truefoundry/deploy/function_service/remote/remote.py,sha256=iz3Rts03atxlD4fmDOuHeGIt8HlXLF1h8OYD7Lp6XZg,4203
 truefoundry/deploy/function_service/route.py,sha256=Wcn5YacWZXA2XKw98cXbwZLL-AaknLEskybN852LSkk,4513
 truefoundry/deploy/function_service/service.py,sha256=2RnwTSxVa1f12SSqkWD4Ttj6A9hoWchPIscvfXeX3IQ,4169
 truefoundry/deploy/function_service/utils.py,sha256=fSF8g19L-fpbaqf91SEB3HNtaJ93SwKC7txqYePhES0,1332
@@ -80,32 +88,28 @@ 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/json_util.py,sha256=x_-7YYQ4_HUIJ8ofOcclAp9JWhgTWjR9Th6Q0FuRqGk,175
 truefoundry/deploy/lib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/deploy/lib/auth/auth_service_client.py,sha256=C3sgoDZ2RUijJY_nStqt-BmSwUAdjVa8yAhazOGz710,7386
-truefoundry/deploy/lib/auth/credential_file_manager.py,sha256=DXeXWoVakfZI2Geu8Futwef_eVMHrZ1et7d9MIXOhhY,4233
-truefoundry/deploy/lib/auth/credential_provider.py,sha256=vgVRsm5TOr0MVxGMNFY9XVZHqnSorygwOk4YgOXe0v8,4120
-truefoundry/deploy/lib/auth/servicefoundry_session.py,sha256=2OahwRg-8l3QUwpC2iLA8lHavkeTxKXUP_0AN27HPS8,1859
+truefoundry/deploy/lib/auth/servicefoundry_session.py,sha256=d4M56O_ZftFeDRiyRci11QABU9B1HW2eVOeVIrFvevE,1842
 truefoundry/deploy/lib/clients/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/deploy/lib/clients/servicefoundry_client.py,sha256=nIRaMKw6sqYcwLFFMIRgy6PqT5eru6BlJQqE9evJjL0,26616
+truefoundry/deploy/lib/clients/servicefoundry_client.py,sha256=fgou4wjuWJzqQgW7iatbSCvmstSjY0iUJldQK5vsruE,26018
 truefoundry/deploy/lib/clients/shell_client.py,sha256=tMrc0Ha1DmGtUCJrZD8eusOzfe8R_WIe6AAH7nxL0xA,461
-truefoundry/deploy/lib/clients/utils.py,sha256=rK7DrvA71kSTjy23-Bk4LTQjgViBWeHtcV_SlBLZw6M,1282
-truefoundry/deploy/lib/const.py,sha256=Yk_nXeZWzwKs-6hEXDjVDyjwEGh35T5TWaBxyJgP-Zw,1395
+truefoundry/deploy/lib/const.py,sha256=FCQfnO7IecB1ikQHdLGNvvubq_iF900C9l5TJtDfvFc,314
 truefoundry/deploy/lib/dao/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/deploy/lib/dao/application.py,sha256=uUTFSQkLUrFCtQQgS2Jm9BpyHyhMkN4GI1yx9oJo4_E,9161
 truefoundry/deploy/lib/dao/apply.py,sha256=sXnQY6RVzLVm1fX2BKuWHAoKlKISirrcByHEhY3x4zo,2570
 truefoundry/deploy/lib/dao/version.py,sha256=AtdW_4O1DPUKdfv2qy6iUJsZ_95vM6z0AqeEy3WDKs8,1130
 truefoundry/deploy/lib/dao/workspace.py,sha256=jm8UWytwVajVcrYyHSTCwWYDYl-RHuk0zAf9Caj4GzQ,2356
-truefoundry/deploy/lib/exceptions.py,sha256=XmXzPWDInmsTY5VRh5pDtIRLZlD-7dfsd8M20pI58ao,537
+truefoundry/deploy/lib/exceptions.py,sha256=ZT2o3mar3BYtnjKHn2wf4rBGEaFIgf-zkQmzDicg6Ps,236
 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/model/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/deploy/lib/model/entity.py,sha256=aNiQ7kK_V1kH7C_VC6HP7ibas_9Ohkoqzs-WdTfrrf4,12037
-truefoundry/deploy/lib/session.py,sha256=6z_3UYHmJ0TZ6MJgR9WOPg1zZRIn1URdaliTxzHOidg,5324
-truefoundry/deploy/lib/util.py,sha256=sNMv9peDU7RTCDaAfyWPpPpCs_WcZGqOvc13Jqx9y6w,2809
+truefoundry/deploy/lib/model/entity.py,sha256=8J8yd98iWtSy8giShdDRNyzbN1UgSXx4XtmZLljdWnE,8552
+truefoundry/deploy/lib/session.py,sha256=e0iA-fWPBzxm5JqT4hnfv4NOpYexbrtRLSWLqqVtFzs,4986
+truefoundry/deploy/lib/util.py,sha256=RlL3bjZu5Z0LU_OKYaMVfcMU8k7_rmkAp89_0CrZDLk,1520
 truefoundry/deploy/lib/win32.py,sha256=1RcvPTdlOAJ48rt8rCbE2Ufha2ztRqBAE9dueNXArrY,5009
 truefoundry/deploy/python_deploy_codegen.py,sha256=SoXIyPKLCA-VxUlUKt9ZmSrNPVTzHwZISOSZQNf-nUc,4022
 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=o0bOpXuYV9-AvyMaTYnPc3Gy3vjeN3ZP7F8r3nFzxjk,12001
+truefoundry/deploy/v2/lib/deploy.py,sha256=HIcY3SzQ5lWl7avuuKi3J0Z-PBES6Sf4hgMK-m6_53U,11990
 truefoundry/deploy/v2/lib/deploy_workflow.py,sha256=WhlrBuRf7r83qoQdTZSQzHt635fw9D4_qQIqusFWGag,12372
 truefoundry/deploy/v2/lib/deployable_patched_models.py,sha256=MROgMxhn9hDEAKwJSWl3iz12tUVvRKzEtqF2QUT6dAk,3343
 truefoundry/deploy/v2/lib/models.py,sha256=pSolLMTArDuYpeNsmeeS5DWliloN_iCDfZSpRllMHUg,1120
@@ -118,15 +122,15 @@ truefoundry/langchain/truefoundry_embeddings.py,sha256=8nRaZ7W1ao1WF0LHk6nNel1Lu
 truefoundry/langchain/truefoundry_llm.py,sha256=CJXyCgXIMbDsVRuuvEA5PKJsf6aRyVlYuG7zC4qtZXE,3802
 truefoundry/langchain/utils.py,sha256=PGLDe9chZ3BuUjakexOGpIqZRFoHEgu-zJ9yKdpLLmM,1329
 truefoundry/logger.py,sha256=7dLqW_Q2rEgo-_z1WZnQbGHaoy1L1MP3NqGgssaSS6o,685
-truefoundry/ml/__init__.py,sha256=473v7D2cBNsgww_JHKInbNUKzX_SSHE3uS3Yu0mlqqM,994
+truefoundry/ml/__init__.py,sha256=2A1l7pgqbVRt3cRW_0Lxg92hyJEkMxkCUh1EFprrmc0,942
 truefoundry/ml/artifact/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/ml/artifact/truefoundry_artifact_repo.py,sha256=bxvGoGxRgiVz3RTM8ac_623e1x_lI795Wbv2Qhhy07I,43548
+truefoundry/ml/artifact/truefoundry_artifact_repo.py,sha256=aJxvYtzBExUM2m-o3KY7Gt6TnGkAmElfNXleftiCL8k,44930
 truefoundry/ml/autogen/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/ml/autogen/client/__init__.py,sha256=n6QhfYFhzAkRXRecGz7ubTGKQ3_5Hj15qyVBwAGdwiE,16680
+truefoundry/ml/autogen/client/__init__.py,sha256=9tmMg8M1td9h0zbAAdl8Lla3A33iU1XNMKx6pZj1yzA,16524
 truefoundry/ml/autogen/client/api/__init__.py,sha256=3sMSljMIS3UHYeF0BcNvrPPx6VbBSRt_1IfDn-13Kyc,752
 truefoundry/ml/autogen/client/api/auth_api.py,sha256=zpWzJhUmW6HHMY_atlUf0B25k77E1kue2hmix5I5Ih0,7017
 truefoundry/ml/autogen/client/api/deprecated_api.py,sha256=JCQ39Y3VHdgJ1zM4XVabdTl6QpOHtQFjh04XUb6mN9A,24218
-truefoundry/ml/autogen/client/api/experiments_api.py,sha256=7CHkpjbUlixsf5BKu__2OLys3D42cABbOs5icFazbKc,88042
+truefoundry/ml/autogen/client/api/experiments_api.py,sha256=vzlPMzclwk7oD2ekUBInIAHLNO3qCAPF38AAvxfDBSY,80682
 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=CzhvDh_ZrsKmVVeX2fqNbfbWkCvosUQhkTEHx2pLGCo,306479
@@ -137,7 +141,7 @@ truefoundry/ml/autogen/client/api_client.py,sha256=8qg-WpadDuKgbRt5yABJ4wVS4IRxd
 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=y1KPHnofaOk8a2yI-_yKqJWXDgI-W8fqZghzakQUaRg,15296
+truefoundry/ml/autogen/client/models/__init__.py,sha256=IS04fIw-KPC87XbVETy7dFS2Exrpm3IMupfBf8E5ym8,15140
 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=rU0h96pEE8K1Ukw2pzDSjq0e6BgtDEuOctI-aZMrpUY,2653
 truefoundry/ml/autogen/client/models/agent.py,sha256=fnMWdEPe5Iw50WKydtu7QAxU419j3ju1IukpChUnGqY,3871
@@ -145,16 +149,15 @@ truefoundry/ml/autogen/client/models/agent_app.py,sha256=h-0xdTMDE6FPzvksnbhVSXO
 truefoundry/ml/autogen/client/models/agent_open_api_tool.py,sha256=QotUMmGqAMpNsRFwVoNJHg9OeobimS-Wc9Eb85VRD24,5023
 truefoundry/ml/autogen/client/models/agent_open_api_tool_with_fqn.py,sha256=8iHIEpGtQJO8QZS66T0FysqKg1VaxJT44hPu_a6zuwE,5121
 truefoundry/ml/autogen/client/models/agent_with_fqn.py,sha256=_LFRmdT0KNe1CRNBfEt9gFbBvGMiwPInbesHSKrLH-g,3976
-truefoundry/ml/autogen/client/models/artifact_dto.py,sha256=jsmnNl9ScDjiky71YIVJpzLAOUNyyGhqVqYOgqMO274,3717
+truefoundry/ml/autogen/client/models/artifact_dto.py,sha256=I-5m316b3wiSxGrgx7dEymTr--Q8u0jIkl53EbeO4ms,3704
 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=gU5etTFBO2QAHBUHrMf9c-pPjqE9RYmQ93wySlhQBOg,4881
+truefoundry/ml/autogen/client/models/artifact_version_dto.py,sha256=MdpHGU69R7_ej6_2ppp1a1WWkYcPCdXRyovpQN-WsDs,4868
 truefoundry/ml/autogen/client/models/artifact_version_response_dto.py,sha256=JmiSYdPYG-ki6LCBeF7NtihdQ2LPiHld8cSPivY2GmQ,2421
 truefoundry/ml/autogen/client/models/artifact_version_status.py,sha256=iIcomqhuaJkZH2rT3e6IjQutM79t9hEshW-RCIMKHEY,780
 truefoundry/ml/autogen/client/models/assistant_message.py,sha256=rmCQ4OAbVpjbX3rXiNtv2ePxV-7kq6Eg1Bs_tm_rmNE,2845
 truefoundry/ml/autogen/client/models/authorize_user_for_model_request_dto.py,sha256=G00T6eroxZrDw782-UBCQxlkwkxyAPfmvd9QNcMiZDI,2017
 truefoundry/ml/autogen/client/models/authorize_user_for_model_version_request_dto.py,sha256=Cy_7YVK47WMgweWbrSzR8-pb3B0WotriQWrKYdcxhvk,2070
-truefoundry/ml/autogen/client/models/backfill_default_storage_integration_id_request_dto.py,sha256=2ffg3yokxnJjlZy2s4D5a57Dnz92-CrjRvzqrzLx5GI,2111
 truefoundry/ml/autogen/client/models/blob_storage_reference.py,sha256=UcfYHxRyJyDMge90Ln9Agm9ynbddVjOvGonX6VeKFKo,2995
 truefoundry/ml/autogen/client/models/body_get_search_runs_get.py,sha256=aGu5Ah2QB-fnke4joOSAnhD5aO0BFGP3dzZcvSNSe-s,2064
 truefoundry/ml/autogen/client/models/chat_prompt.py,sha256=PFpvfDGWLSUWYTHEFAOtScJeTYjUUziQD_Meii2D0gw,5974
@@ -165,9 +168,9 @@ truefoundry/ml/autogen/client/models/content1.py,sha256=Sh22V-dSQr8zz2N2BgqeTtau
 truefoundry/ml/autogen/client/models/content2.py,sha256=T55JENSS5E9HRkefccw24L7PB-rNp3VP1ppXfV7BNHk,5920
 truefoundry/ml/autogen/client/models/content2_any_of_inner.py,sha256=yeGDgvaMMnymUkQpw_3v-DL5Pjt_FWvCl13na91bSEs,5004
 truefoundry/ml/autogen/client/models/create_artifact_request_dto.py,sha256=U9oPwVw_G_3IIQzV_OqjuL_6ytvmfUwOMAyrYi_UhRo,2217
-truefoundry/ml/autogen/client/models/create_artifact_response_dto.py,sha256=5b4WjMXpJ_7JngrINzK2pUVqzmpGPN-0DtnuYHvWYWI,1874
+truefoundry/ml/autogen/client/models/create_artifact_response_dto.py,sha256=mRRGlXQXW2d3oWv9YOqKG55GHuIpWslSzosUxaNXYXo,1849
 truefoundry/ml/autogen/client/models/create_artifact_version_request_dto.py,sha256=DTmslB75W6Cr_ec1ddEt40VaKowaZVv2ypf7sRH3elI,2284
-truefoundry/ml/autogen/client/models/create_artifact_version_response_dto.py,sha256=iWrfppDAA4IP2K-eofcFoeKXZc4F_ZAFOqIf5vYzq0c,1930
+truefoundry/ml/autogen/client/models/create_artifact_version_response_dto.py,sha256=iLPOAM7bw-t6hlKL4_mtw8m0QfWj4rRT8vxLdSHE9-8,1905
 truefoundry/ml/autogen/client/models/create_dataset_request_dto.py,sha256=6Ke3swxo9DG-mbOZOTXosepZ6toHPmYc4GOCtahUs2A,2308
 truefoundry/ml/autogen/client/models/create_experiment_request_dto.py,sha256=yD3u5li_ShKbX1RGomEOkSEXQPZKxqhJGYXAMZTKXDU,3013
 truefoundry/ml/autogen/client/models/create_experiment_response_dto.py,sha256=GmN5gDzieW8xT2RXOhf0QJ5_AL7JzCSwFYJMnys1pgA,1931
@@ -175,18 +178,18 @@ truefoundry/ml/autogen/client/models/create_model_version_request_dto.py,sha256=
 truefoundry/ml/autogen/client/models/create_multi_part_upload_for_dataset_request_dto.py,sha256=2-51iZCBmMNnl8v_3sQYyOJwK2RsnzM5v860axbFpS4,2261
 truefoundry/ml/autogen/client/models/create_multi_part_upload_for_dataset_response_dto.py,sha256=VDgQ4gZ6xcrWun3fP-mln8lGwYGOo640GH_TQQ5ceKE,2559
 truefoundry/ml/autogen/client/models/create_multi_part_upload_request_dto.py,sha256=nGrPRw5R44APVGeyegxFLxfi8KAHX-NDORtS1iBw5Xc,2213
-truefoundry/ml/autogen/client/models/create_python_deployment_config_request_dto.py,sha256=6FueO8APWdFJZycz2fK6K4YE7Usi8cUxtXyDWzAC1Fk,2229
-truefoundry/ml/autogen/client/models/create_python_deployment_config_response_dto.py,sha256=mhOMdwInQRks8r5Nd2Q-6Xlv9U4V0L3Kz_DaKoyZ1oA,2076
-truefoundry/ml/autogen/client/models/create_run_request_dto.py,sha256=hi18iz7aL-t_WgaCIbd2GvfDoPihhJ4DjnPdb3DHOaA,2995
-truefoundry/ml/autogen/client/models/create_run_response_dto.py,sha256=wg5iqhaHhd0g5r_zdPtC96vh2fW14u4gapYbZND1sp0,2194
-truefoundry/ml/autogen/client/models/dataset_dto.py,sha256=4BTbfg4BIizRq-PRJrpyn5ph7YuzJ_Qo5zOAdoDYkNQ,3570
+truefoundry/ml/autogen/client/models/create_python_deployment_config_request_dto.py,sha256=1xmhL67Pz0Ckgfky3SIwtD0ayPJA-Vvv51Qec0LVDAA,2232
+truefoundry/ml/autogen/client/models/create_python_deployment_config_response_dto.py,sha256=5fnCJ7ybEcRA03s8ZLGdt1w8_Xz5nt6-PPmvA6Yxt60,2051
+truefoundry/ml/autogen/client/models/create_run_request_dto.py,sha256=6lwK8CLGgAnNtvBn3_0jcKicsWI3DMUfxj1OUctzDwA,2994
+truefoundry/ml/autogen/client/models/create_run_response_dto.py,sha256=cFzIkhbiB_IEmQSx1asNAzPMN-MzCBtqWj2PJ3oTLTM,2169
+truefoundry/ml/autogen/client/models/dataset_dto.py,sha256=2GqBLyIFXjNJ8stQRQgB1GoC9eqqArKlBafIGtDOie0,3561
 truefoundry/ml/autogen/client/models/dataset_response_dto.py,sha256=5h3eHpiG4Zzqn5Hm0WpZgyXGjxujKem42_8YnGq5KLo,2205
 truefoundry/ml/autogen/client/models/delete_artifact_versions_request_dto.py,sha256=JF2dQuEe_W1c2ox9zZDER_pZSTbKwSnns8hIfG5o7oA,1905
 truefoundry/ml/autogen/client/models/delete_dataset_request_dto.py,sha256=bfzzgBoekOBcGg2Uzzel80rf2Z5fDgggShkYd0ZlbSM,2143
 truefoundry/ml/autogen/client/models/delete_model_version_request_dto.py,sha256=6abYuueylSgv7TLkIHq8X0NWeFr-J5k72caAyOnn6fw,1873
 truefoundry/ml/autogen/client/models/delete_run_request.py,sha256=jXQmROKfkPA8idoxkUXT-2O6QJ4OgxgFX-CEmjryIpY,1793
 truefoundry/ml/autogen/client/models/delete_tag_request_dto.py,sha256=eQSPMamCJbsehcwWoMxQxlzZKQfX5GEwm3KHGF2Ajts,1901
-truefoundry/ml/autogen/client/models/experiment_dto.py,sha256=oKUWAqXf_Lhs2urevBW0bEYDdJmtg4bSwlEzwdHlYV8,4419
+truefoundry/ml/autogen/client/models/experiment_dto.py,sha256=LBR3FkOw5j_2w1vrY3PYdqFWJwVXAE-Len8L8ua0XzI,4409
 truefoundry/ml/autogen/client/models/experiment_id_request_dto.py,sha256=ZMH827n_UTpDI30UnkuOam-4ANBKCDgocIzI8StxFR8,1891
 truefoundry/ml/autogen/client/models/experiment_response_dto.py,sha256=wuflV6_f8PQq061-wU2GzNY4BZi8SG8ARCIbSQN1oT4,2268
 truefoundry/ml/autogen/client/models/experiment_tag_dto.py,sha256=nEpCkeZ9ficIDkjmmLfkJeNNokd-Rhgr-cepPWG6L3M,1902
@@ -195,7 +198,7 @@ truefoundry/ml/autogen/client/models/feature_value_type.py,sha256=3ZKxxEfrrWk8eP
 truefoundry/ml/autogen/client/models/file_info_dto.py,sha256=7oc8venicsFVk8zT9wHNhHnZGtFkFlqimFnS7ozGL9k,2156
 truefoundry/ml/autogen/client/models/finalize_artifact_version_request_dto.py,sha256=P20UWglW6pzhNJVchC37ypZSHUxzzxf9tfKJkLgxoaY,3361
 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=DSX3nsb_wIloHrk-Vw5U4EORIyWeOCOplgTSqazeymE,2291
+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
 truefoundry/ml/autogen/client/models/get_signed_url_for_dataset_write_request_dto.py,sha256=liRpV-VFuM2DnF5jLVp90BvI-5Lg4Bx5RpZijt1sBqA,2091
 truefoundry/ml/autogen/client/models/get_signed_urls_for_artifact_version_read_request_dto.py,sha256=Ijc6WXjiHQARfjvgRAo9E4LEh1N-BRrIktXhTUrzPa0,2119
@@ -205,7 +208,7 @@ truefoundry/ml/autogen/client/models/get_signed_urls_for_artifact_version_write_
 truefoundry/ml/autogen/client/models/get_signed_urls_for_dataset_read_request_dto.py,sha256=CgXGykv4D1SkYFk5OexhD20-6v7mOlAOD5ciaUcQ5_s,2091
 truefoundry/ml/autogen/client/models/get_signed_urls_for_dataset_read_response_dto.py,sha256=1EQ33wOq7VpAlt7o-5XCBZ2mpn5GXKrOUfULbp4gA2g,2627
 truefoundry/ml/autogen/client/models/get_signed_urls_for_dataset_write_response_dto.py,sha256=tRud_0Qrr4DtXD1SMSnouVlxrV6WaL5FG2mG626CFZ0,2635
-truefoundry/ml/autogen/client/models/get_tenant_id_response_dto.py,sha256=aKj1Hy7lP9wH8tQ4VE4bYHBkUGXKFK9Lu4TQ1T0iKYo,2175
+truefoundry/ml/autogen/client/models/get_tenant_id_response_dto.py,sha256=zEG9rogifl100cOqshYDNJE-faxGF3ayRb1RNWCtaXo,2142
 truefoundry/ml/autogen/client/models/http_validation_error.py,sha256=bU-uUlz_qeFba3iHdBT7cHjLu6WZ-JmImYqgZ-kCvxM,2466
 truefoundry/ml/autogen/client/models/image_content_part.py,sha256=_DXwAxeuagwRWMkDxaz--i_OitlyNkPd3puewVV_fwM,2716
 truefoundry/ml/autogen/client/models/image_url.py,sha256=TN8iEdBUmtKdTFlApUodaC4o6X_Ggd3OSOKVPv7laJE,2170
@@ -241,10 +244,10 @@ truefoundry/ml/autogen/client/models/metric_collection_dto.py,sha256=02ZWKm2_8Dj
 truefoundry/ml/autogen/client/models/metric_dto.py,sha256=Xrcc8XRY0sZeg_cnb6gyf-AUUIS0FPxSGTCbDzcBzXQ,2174
 truefoundry/ml/autogen/client/models/mime_type.py,sha256=A-N5-Mof_IyHWguXmG67k9wuaS5xwLPcj1J11jrEGyc,897
 truefoundry/ml/autogen/client/models/model_configuration.py,sha256=XKhwqO8caQ8w9XRCdb_pAl-84cd_PWkEutjLNSby6hQ,3861
-truefoundry/ml/autogen/client/models/model_dto.py,sha256=PU2SKPP8gD8edxUapYIzVjXXp-pXN_IBeFPPAkxBXDY,4159
+truefoundry/ml/autogen/client/models/model_dto.py,sha256=iz9RAxADkeWwe0RmsxiLcYHZ5rZhkRF0PRcy004RT_I,4146
 truefoundry/ml/autogen/client/models/model_response_dto.py,sha256=osrTxfygkuhxWj6SkRBALrSnFVPH4LSK6qTufTeZuJg,2163
 truefoundry/ml/autogen/client/models/model_schema_dto.py,sha256=ElEPK7fwuf3eiohNBBHJyl-nhNqzGnDLzZVzuQl-xC0,2620
-truefoundry/ml/autogen/client/models/model_version_dto.py,sha256=RAuitVL3211XjjzoM_7aVefJA1aEPY2j87svcc4Zy4o,5901
+truefoundry/ml/autogen/client/models/model_version_dto.py,sha256=1NqfoCKN-MWeWi8_ih4EHfknJnOyUuBbhCu-Kg5WVDE,5917
 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
@@ -258,7 +261,7 @@ truefoundry/ml/autogen/client/models/resolve_agent_app_response_dto.py,sha256=sg
 truefoundry/ml/autogen/client/models/restore_run_request_dto.py,sha256=7iBbil4YfaOHJwP0ifPHp-D1Kt-ZiSRjAaVuU7_idhw,1825
 truefoundry/ml/autogen/client/models/run_data_dto.py,sha256=iPFm5f4x2ymRqea6EDZlcjWYN_-f3t9JZckwskXSTKY,3533
 truefoundry/ml/autogen/client/models/run_dto.py,sha256=inAYRSKnHSkF3cdf2eQYp5wVnOgB47ZPdavxCHXh9zY,2535
-truefoundry/ml/autogen/client/models/run_info_dto.py,sha256=mhB1cIdOcbBFwx2f5fKHjWaao2Yvo7_IsER3KpOqU1A,3095
+truefoundry/ml/autogen/client/models/run_info_dto.py,sha256=HJqhChLit5ub3LDQarqDQxgd9z2YCZyO06tIt-s-RQ8,3082
 truefoundry/ml/autogen/client/models/run_log_dto.py,sha256=VEIXYILD4AW0QAmokyzS9mo4JmiAV7NgU5TRgymQMek,2529
 truefoundry/ml/autogen/client/models/run_log_input_dto.py,sha256=xQ854-3vcocDzcNYejYVlo2BmEpDLlHh1MbiP2OHen4,2390
 truefoundry/ml/autogen/client/models/run_response_dto.py,sha256=iayBnhwmByVC3iiw_ZUbB_V6FTadEQu9_U2JOs5kvkw,2121
@@ -278,55 +281,53 @@ truefoundry/ml/autogen/client/models/update_dataset_request_dto.py,sha256=MF-rjJ
 truefoundry/ml/autogen/client/models/update_experiment_request_dto.py,sha256=QE_kZzTDdiXKtQ2U-zbI_gb5vxS5Yv9YupYvD22RkFs,2198
 truefoundry/ml/autogen/client/models/update_model_version_request_dto.py,sha256=dVP9WW0bQV0Bolj_hVbsQQuUT-8lhJ9ocDWsThfARlI,3073
 truefoundry/ml/autogen/client/models/update_run_request_dto.py,sha256=fn4VB4Zlllr6Bt_l3aYkkHxAkU9w7vtcqu4iw8BvQWY,2318
-truefoundry/ml/autogen/client/models/update_run_response_dto.py,sha256=IprygJC8plnEIwahAZU2ilESae8WuAESkejMM-hQtn0,2256
+truefoundry/ml/autogen/client/models/update_run_response_dto.py,sha256=wolIFiihGZCBBkuj7mJEYCmGAOqDLFo_bNLcH4LeXSo,2231
 truefoundry/ml/autogen/client/models/url.py,sha256=zMyOmdVkp1ANnQnc9GrHt42xlVwES7FTjrpOp_OtHqo,4953
 truefoundry/ml/autogen/client/models/user_message.py,sha256=6QOmKT_SDfAVZA8V3Bzofm7_c-Qznn4KsE12NmHG4FM,2783
 truefoundry/ml/autogen/client/models/validation_error.py,sha256=mFjwoc8g2-Usu1HXZhOQKQ4TGvLy4lwCzk8dHrJ69aA,2597
 truefoundry/ml/autogen/client/models/validation_error_loc_inner.py,sha256=nThJ5Gmy8W2Wok-ZOI4sK7uRe1BAkLS0qzq-XZbq8zs,4915
 truefoundry/ml/autogen/client/rest.py,sha256=9goba8qHjQuVx5O_yRaTKu7PvBnb7r7swfy3dwuTEgk,14281
-truefoundry/ml/autogen/client_README.md,sha256=GkgNes2w7a3z1Cn4oGMWIum9WS_w4TWJ_erRCAqklx8,33258
+truefoundry/ml/autogen/client_README.md,sha256=lobq-OlEcetj90WhMC1EmFQ85Bivmg2OPY2_CT1zc94,32876
 truefoundry/ml/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/ml/cli/cli.py,sha256=ckBcjUpqfhgrPE1okqT_G2iouOLt-0KjpLhHp2YdVFU,256
 truefoundry/ml/cli/commands/__init__.py,sha256=diDUiRUX4l6TtNLI4iF-ZblczkELM7FRViJ-8gGNJQY,82
 truefoundry/ml/cli/commands/download.py,sha256=cbz9KijiLKXj4-twlig3xZLTVRNm4fnjwpy0leZr31w,2342
+truefoundry/ml/clients/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+truefoundry/ml/clients/entities.py,sha256=sNP4DnAVdQoMfc06s0r3VTzKHTo7jmxAOuTlQOVmsMs,151
+truefoundry/ml/clients/servicefoundry_client.py,sha256=lK7mepLNRHHS65Eebynk5HyCMSxqmduXRHO2k8pTIK0,1680
+truefoundry/ml/clients/utils.py,sha256=LX2_p8WhKJYdDjg08395nwW-L1Yx7YumRsFn9PCDdNI,4385
 truefoundry/ml/constants.py,sha256=vDq72d4C9FSWqr9MMdjgTF4TuyNFApvo_6RVsSeAjB4,2837
 truefoundry/ml/entities.py,sha256=si5GAqZsWzKu5MPrU4Hk6se7bebHOYhTiNw69ai-Uk8,1485
 truefoundry/ml/enums.py,sha256=arqDkF8duU_oVLFeYMhcfWYbF6Nq5mmjwupJMIheyXM,1790
-truefoundry/ml/env_vars.py,sha256=ehrKLWCs3A407zrfTzRQP-X3PzjgUvKpheGByEa2rPE,339
+truefoundry/ml/env_vars.py,sha256=0CqhoV6RgxgPju71AJYvo7crHxVr0vJNzWTiSWKnLcI,212
 truefoundry/ml/exceptions.py,sha256=8aJm2NYtAWWsRLu4MbzaoOqHsQZ6RjOFwBWQWqb6qrc,244
 truefoundry/ml/git_info.py,sha256=jvAVm9ilqivnGq8qJdUvYdd8Siv0PLtqurB-PXsS5ho,2023
-truefoundry/ml/internal_namespace.py,sha256=ROJ9e_qMR50RBk1OgZBaXH8H352M6GwSYBf8xC-82I4,1790
+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=amqZMaC094jPxr26QiS_37c_LVagSnukgEZlarDmHzg,15377
+truefoundry/ml/log_types/artifacts/artifact.py,sha256=KrQhROx82rT9Trzoj_-Lq6Ya5vuIzoRVIbjBiPz_sNc,15511
 truefoundry/ml/log_types/artifacts/constants.py,sha256=qKxQ5mMvJE4j83BvGW3qNTKunxCiBg_EEjTdgbgJtyE,1036
-truefoundry/ml/log_types/artifacts/dataset.py,sha256=ys6UOOXeftsddJ5w6QNuWQX79xg6mBJhQwfRVbCzFvA,12985
+truefoundry/ml/log_types/artifacts/dataset.py,sha256=oXi7itXOe3PrnN4_KY7s1GdhWV1lyvWXRWgY6_vPTAs,12981
 truefoundry/ml/log_types/artifacts/general_artifact.py,sha256=fY74EUeCGdHdlkrp_GMoaZPvl7EQctt8OlRM6yFgLF8,3772
-truefoundry/ml/log_types/artifacts/model.py,sha256=iFPKDOtgYCy1PuALohPC6L1s8PwQq66dpPMVprr7uco,22146
+truefoundry/ml/log_types/artifacts/model.py,sha256=NRIvGX0yCLGnLvZkuCRJGGEuONoUhXZSY2jmSUM3JvI,22162
 truefoundry/ml/log_types/artifacts/model_extras.py,sha256=TIE73bLKfwIVzNiVcjmaZ841A70BHBwu4XAM6ZAQRFI,1045
 truefoundry/ml/log_types/artifacts/utils.py,sha256=dwmRBrEmwIktVcA1a3lrsUfmROO7THTLgYwJ4MpSUWU,5885
 truefoundry/ml/log_types/image/__init__.py,sha256=fcOq8yQnNj1rkLcPeIjLXBpdA1WIeiPsXOlAAvMxx7M,76
 truefoundry/ml/log_types/image/constants.py,sha256=wLtGEOA4T5fZHSlOXPuNDLX3lpbCtwlvGKPFk_1fah0,255
-truefoundry/ml/log_types/image/image.py,sha256=qxGKCywCWIP8HZpoaF3haEmh8NwAIH-cTaZvYCOJE1I,12270
-truefoundry/ml/log_types/image/image_normalizer.py,sha256=axhjP-yVA0mq97u9ewEqGsH2FYzz8eTN69qXJFDb14c,3044
+truefoundry/ml/log_types/image/image.py,sha256=y2FOUVTNGtjzGW5AcaXObIxKQDGr1GoiSYBoRAB7ZaM,12310
+truefoundry/ml/log_types/image/image_normalizer.py,sha256=vrzfuSpVGgIxw_Q2sbFe7kQ_JpAndX0bMwC7wtfi41g,3104
 truefoundry/ml/log_types/image/types.py,sha256=inFQlyAyDvZtfliFpENirNCm1XO9beyZ8DNn97DoDKs,1568
 truefoundry/ml/log_types/plot.py,sha256=oFnXNb2o5fVF0zsnRjvqjSjLaphQWUnQCdw72e2uiMY,7347
 truefoundry/ml/log_types/pydantic_base.py,sha256=eBlw_AEyAz4iJKDP4zgJOCFWcldwQqpf7FADW1jzIQY,272
 truefoundry/ml/log_types/utils.py,sha256=xjJ21jdPScvFmw3TbVh5NCzbzJwaqiXJyiiT4xxX1EI,335
 truefoundry/ml/logger.py,sha256=VT-BF3BnBYTWVq87O58F0c8uXMu94gYzsiFlGY3_7Ao,458
-truefoundry/ml/login.py,sha256=Yp9Xeb1p1FcxPK3nmScJyCg3ZIMHrfuN334rtbm6vcY,9221
-truefoundry/ml/mlfoundry_api.py,sha256=11Ql5s8UILLRIqmUdDhPCGuYPtx4tcqEQlOJWYUQp3E,59683
-truefoundry/ml/mlfoundry_run.py,sha256=FQdILe17Yt-2dO_l7ETMRmywwm7qgJf-_dnCILb0Cvo,44578
-truefoundry/ml/run_utils.py,sha256=9VUVakz92kA4hQXnAGRi3V7o9CMRdUbCG8yyZ9eByyU,3289
-truefoundry/ml/services/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/ml/services/auth_service.py,sha256=RgxVz_CCKlBrIVw3sR6CrnAtjna3jSw5UfTHBwsdkkA,4022
-truefoundry/ml/services/entities.py,sha256=x2_ldqxBMXIRfxtu3HHS28jvaqr5IBv4V21MwJkkxB0,3005
-truefoundry/ml/services/servicefoundry_service.py,sha256=jopsRSdNqfan0AtRO0UPiPOWEgwfzRP7qLamzMDkz-0,1369
-truefoundry/ml/services/utils.py,sha256=VPRa78tZk5wzNLBAFARXaDPwvOgCPeJZIDNoTjWXp0Q,4384
-truefoundry/ml/session.py,sha256=0vIznzgjqVT8sNayXIa_O6APLdDJ3NH6mKb1yTspgmI,8888
+truefoundry/ml/mlfoundry_api.py,sha256=7RZZaMd6Z1f-qekpEFhjgqgDAFUZ3_7iZbzrRAhVj58,59379
+truefoundry/ml/mlfoundry_run.py,sha256=CTUgf2-kPANie9Kog_rJRvpkDfotVzWTHvBg2ofv_FA,44341
+truefoundry/ml/run_utils.py,sha256=dzbQ_TGkSCYO-gg8tlr5tglR0p2SIQdy0wl4IEQE5JA,2899
+truefoundry/ml/session.py,sha256=gASORQ7TZ3kqAyqG64YPfn5Z8Uo2BjOvfhMmTX-X7uM,5263
 truefoundry/ml/validation_utils.py,sha256=XBSUd9OoyriWJpT3M5LKz17iWY3yVMr3hM5vdaVjtf0,12082
-truefoundry/pydantic_v1.py,sha256=UKv05OZx5WC4E0J-25xLiTbhJOYdDVp8SZC4lT7iEFI,329
+truefoundry/pydantic_v1.py,sha256=jSuhGtz0Mbk1qYu8jJ1AcnIDK4oxUsdhALc4spqstmM,345
 truefoundry/version.py,sha256=bqiT4Q-VWrTC6P4qfK43mez-Ppf-smWfrl6DcwV7mrw,137
-truefoundry/workflow/__init__.py,sha256=m0puY8RJ3GBvdAXW0A9PLuCpmDdLqh6eF0t3DCEpZkg,659
+truefoundry/workflow/__init__.py,sha256=hkDVrnG95duO-9XEmdjgrsvyhjQUVqyeQ1-dADB1QuE,935
 truefoundry/workflow/container_task.py,sha256=8arieePsX4__OnG337hOtCiNgJwtKJJCsZcmFmCBJtk,402
 truefoundry/workflow/example/deploy.sh,sha256=wfbPRrCi04WYRqCf4g-Xo12uWbcqPD6G_Tz0lV0jU_U,60
 truefoundry/workflow/example/hello_world_package/workflow.py,sha256=ACWJJWTY5NhYyU_dAEfdSqXgyLuNXPRDpOBuUkTEzhU,459
@@ -337,7 +338,7 @@ truefoundry/workflow/map_task.py,sha256=2m3qGXQ90k9LdS45q8dqCCECc3qr8t2m_LMCVd1m
 truefoundry/workflow/python_task.py,sha256=SRXRLC4vdBqGjhkwuaY39LEWN6iPCpJAuW17URRdWTY,1128
 truefoundry/workflow/task.py,sha256=ToitYiKcNzFCtOVQwz1W8sRjbR97eVS7vQBdbgUQtKg,1779
 truefoundry/workflow/workflow.py,sha256=WaTqUjhwfAXDWu4E5ehuwAxrCbDJkoAf1oWmR2E9Qy0,4575
-truefoundry-0.4.0rc2.dist-info/METADATA,sha256=YUDcmF33_NekQH307XOSUfx_FsLPU3da85heTvQKf-o,3135
-truefoundry-0.4.0rc2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-truefoundry-0.4.0rc2.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
-truefoundry-0.4.0rc2.dist-info/RECORD,,
+truefoundry-0.4.0rc3.dist-info/METADATA,sha256=1cuqkHPKYi-kxkBz_6Xqm0BLsiFHPghCOw8QpqpHHnk,3135
+truefoundry-0.4.0rc3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+truefoundry-0.4.0rc3.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
+truefoundry-0.4.0rc3.dist-info/RECORD,,