truefoundry 0.4.3__py3-none-any.whl → 0.4.4rc1__py3-none-any.whl

truefoundry/common/constants.py

@@ -1,14 +1,32 @@
+import os
 from pathlib import Path
+from typing import Optional
+
+from truefoundry.pydantic_v1 import BaseSettings, Field, SecretStr
 
 TFY_CONFIG_DIR = Path.home() / ".truefoundry"
 CREDENTIAL_FILEPATH = TFY_CONFIG_DIR / "credentials.json"
 
+# These keys are kept separately because we use them in error messages and some checks
 TFY_HOST_ENV_KEY = "TFY_HOST"
 TFY_API_KEY_ENV_KEY = "TFY_API_KEY"
-TFY_CLI_LOCAL_DEV_MODE_ENV_KEY = "TFY_CLI_LOCAL_DEV_MODE"
-TFY_CLI_PYTHONBUILD_PYTHON_IMAGE_REPO_ENV_KEY = "TFY_CLI_PYTHONBUILD_PYTHON_IMAGE_REPO"
 
 
+class TrueFoundrySdkEnv(BaseSettings):
+    # Note: Every field in this class should have a default value
+    # Never expect the user to set these values
+    TFY_HOST: Optional[str] = Field(default=None, env=TFY_HOST_ENV_KEY)
+    TFY_API_KEY: Optional[SecretStr] = Field(default=None, env=TFY_API_KEY_ENV_KEY)
+    TFY_ARTIFACTS_DOWNLOAD_CHUNK_SIZE_BYTES: int = 100 * 1000 * 1000
+    TFY_ARTIFACTS_DOWNLOAD_MAX_WORKERS: int = max(min(32, (os.cpu_count() or 2) * 2), 4)
+    TFY_ARTIFACTS_UPLOAD_MAX_WORKERS: int = max(min(32, (os.cpu_count() or 2) * 2), 4)
+    TFY_ARTIFACTS_DISABLE_MULTIPART_UPLOAD: bool = False
+    TFY_ARTIFACTS_DOWNLOAD_FSYNC_CHUNKS: bool = False
+    TFY_CLI_LOCAL_DEV_MODE: bool = False
+    TFY_PYTHONBUILD_PYTHON_IMAGE_REPO: str = "public.ecr.aws/docker/library/python"
+
+
+ENV_VARS = TrueFoundrySdkEnv()
 API_SERVER_RELATIVE_PATH = "api/svc"
 MLFOUNDRY_SERVER_RELATIVE_PATH = "api/ml"
 VERSION_PREFIX = "v1"

truefoundry/common/credential_provider.py

@@ -3,7 +3,7 @@ import threading
 from abc import ABC, abstractmethod
 
 from truefoundry.common.auth_service_client import AuthServiceClient
-from truefoundry.common.constants import TFY_API_KEY_ENV_KEY
+from truefoundry.common.constants import ENV_VARS, TFY_API_KEY_ENV_KEY
 from truefoundry.common.credential_file_manager import CredentialsFileManager
 from truefoundry.common.entities import CredentialsFileContent, Token
 from truefoundry.common.utils import resolve_tfy_host
@@ -30,7 +30,9 @@ class CredentialProvider(ABC):
 class EnvCredentialProvider(CredentialProvider):
     def __init__(self) -> None:
         logger.debug("Using env var credential provider")
-        api_key = os.getenv(TFY_API_KEY_ENV_KEY)
+        api_key = (
+            ENV_VARS.TFY_API_KEY.get_secret_value() if ENV_VARS.TFY_API_KEY else None
+        )
         if not api_key:
             raise Exception(
                 f"Value of {TFY_API_KEY_ENV_KEY} env var should be non-empty string"

truefoundry/common/request_utils.py

@@ -51,7 +51,7 @@ def urllib3_retry(
 def requests_retry_session(
     retries: int = 2,
     backoff_factor: float = 0.3,
-    status_forcelist=(408, 429, 500, 502, 503, 504, 524),
+    status_forcelist=(408, 429, 417, 500, 502, 503, 504, 524),
     method_whitelist=frozenset({"GET", "POST"}),
     raise_on_status: bool = False,
     session: Optional[requests.Session] = None,

truefoundry/common/servicefoundry_client.py

@@ -43,8 +43,10 @@ def check_min_cli_version(fn):
     return inner
 
 
-def session_with_retries() -> requests.Session:
-    return requests_retry_session(retries=SERVICEFOUNDRY_CLIENT_MAX_RETRIES)
+def session_with_retries(
+    retries: int = SERVICEFOUNDRY_CLIENT_MAX_RETRIES,
+) -> requests.Session:
+    return requests_retry_session(retries=retries)
 
 
 @timed_lru_cache(seconds=30 * 60)

truefoundry/common/utils.py

@@ -1,4 +1,3 @@
-import os
 import time
 from functools import lru_cache, wraps
 from time import monotonic_ns
@@ -7,8 +6,8 @@ from urllib.parse import urljoin, urlparse
 
 from truefoundry.common.constants import (
     API_SERVER_RELATIVE_PATH,
+    ENV_VARS,
     MLFOUNDRY_SERVER_RELATIVE_PATH,
-    TFY_CLI_LOCAL_DEV_MODE_ENV_KEY,
     TFY_HOST_ENV_KEY,
 )
 from truefoundry.pydantic_v1 import BaseSettings
@@ -41,7 +40,7 @@ _tfy_servers_config = None
 def get_tfy_servers_config(base_url: str) -> _TFYServersConfig:
     global _tfy_servers_config
     if _tfy_servers_config is None:
-        if os.getenv(TFY_CLI_LOCAL_DEV_MODE_ENV_KEY):
+        if ENV_VARS.TFY_CLI_LOCAL_DEV_MODE:
             _tfy_servers_config = _TFYServersConfig()  # type: ignore[call-arg]
         else:
             _tfy_servers_config = _TFYServersConfig.from_base_url(base_url)
@@ -95,11 +94,11 @@ def validate_tfy_host(tfy_host: str) -> None:
 
 
 def resolve_tfy_host(tfy_host: Optional[str] = None) -> str:
-    if not tfy_host and not os.getenv(TFY_HOST_ENV_KEY):
+    if not tfy_host and not ENV_VARS.TFY_HOST:
         raise ValueError(
             f"Either `host` should be provided using `--host <value>`, or `{TFY_HOST_ENV_KEY}` env must be set"
         )
-    tfy_host = tfy_host or os.getenv(TFY_HOST_ENV_KEY)
+    tfy_host = tfy_host or ENV_VARS.TFY_HOST
     tfy_host = tfy_host.strip("/")
     validate_tfy_host(tfy_host)
     return tfy_host

truefoundry/deploy/auto_gen/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  application.json
-#   timestamp: 2024-10-09T13:00:44+00:00
+#   timestamp: 2024-10-16T17:06:03+00:00
 
 from __future__ import annotations
 
@@ -362,13 +362,17 @@ class HelmRepo(BaseModel):
         ...,
         description="+label=Helm repository URL\n+sort=1\n+message=Needs to be a valid URL.",
     )
+    integration_fqn: Optional[str] = Field(
+        None,
+        description='+docs=FQN of the helm repo integration. You can use the FQN of your desired helm integration (or add one)\nin the  Integrations page[Integrations](https://app.truefoundry.tech/integrations) page\n+label=Helm repo integration\n+sort=2\n+usage=FQN of the helm repo integration. If you can\'t find your integration here,\nadd it through the [Integrations](/integrations) page\n+uiType=IntegrationSelect\n+uiProps={"integrationType":"helm-repo"}',
+    )
     chart: str = Field(
         ...,
-        description='+label=Chart name\n+sort=2\n+usage=The helm chart name\n+uiType=InputSelect\n+uiProps={"creatable":true, "searchable":true}',
+        description='+label=Chart name\n+sort=3\n+usage=The helm chart name\n+uiType=InputSelect\n+uiProps={"creatable":true, "searchable":true}',
     )
     version: str = Field(
         ...,
-        description='+label=Version\n+sort=3\n+usage=Helm chart version\n+uiType=InputSelect\n+uiProps={"creatable":true, "searchable":true}',
+        description='+label=Version\n+sort=4\n+usage=Helm chart version\n+uiType=InputSelect\n+uiProps={"creatable":true, "searchable":true}',
     )
 
 
@@ -472,6 +476,31 @@ class JobAlert(BaseModel):
     )
 
 
+class Claim(BaseModel):
+    key: str
+    values: List[str]
+
+
+class JwtAuthCreds(BaseModel):
+    """
+    +label=JWT Authentication
+    +usage=Configure JWT-based authentication using JWKS
+    """
+
+    type: Literal["jwt_auth"] = Field(..., description="+value=jwt_auth")
+    issuer: str = Field(
+        ..., description="+label=Issuer\n+usage=The issuer of the JWT tokens"
+    )
+    jwksUri: str = Field(
+        ...,
+        description="+label=JWKS URI\n+usage=The URI of the JSON Web Key Set (JWKS) containing the public keys",
+    )
+    claims: Optional[List[Claim]] = Field(
+        None,
+        description="+label=Claims\n+usage=List of key-value pairs of claims to verify in the JWT token",
+    )
+
+
 class KafkaMetricConfig(BaseModel):
     type: Literal["kafka"] = Field(..., description="+value=kafka")
     lag_threshold: conint(ge=1) = Field(
@@ -655,6 +684,10 @@ class OCIRepo(BaseModel):
     oci_chart_url: constr(
         regex=r"^(((oci):\/\/)?(?:www\.)?[-a-zA-Z0-9@:%._\+~#=]{1,256}\.[a-zA-Z0-9()]{1,6}(?:[-a-zA-Z0-9()@:%_\+.~#?&\/=]*))$"
     ) = Field(..., description="+label=OCI chart URL\n+message=Need to be a valid URL.")
+    integration_fqn: Optional[str] = Field(
+        None,
+        description='+docs=FQN of the container registry. You can use the FQN of your desired container registry (or add one)\nin the  Integrations page[Integrations](https://app.truefoundry.tech/integrations) page\n+label=Container Registry\n+usage=FQN of the container registry. If you can\'t find your registry here,\nadd it through the [Integrations](/integrations) page\n+uiType=IntegrationSelect\n+uiProps={"integrationType":"docker-registry"}',
+    )
     version: str = Field(..., description="+label=Version\n+usage=Helm chart version")
 
 
@@ -736,7 +769,9 @@ class Port(BaseModel):
         None,
         description="+label=Rewrite Path to\n+usage=Rewrite the path prefix to a different path.\nIf `path` is `/v1/api` and `rewrite_path_to` is `/api`. The URI in the HTTP request `http://0.0.0.0:8080/v1/api/houses` will be rewritten to `http://0.0.0.0:8080/api/houses` before the request is forwarded your service.\nDefaults to `/`.\nThis is only applicable if `path` is given.\n+message=Should begin and end with a forward slash (/). Each part can can contain alphabets, digits and hypen, must begin and end with an alphanumeric characters. Parts should be separated by forward slashes (/)",
     )
-    auth: Optional[BasicAuthCreds] = None
+    auth: Optional[Union[BasicAuthCreds, JwtAuthCreds]] = Field(
+        None, description="+usage=Username and Password for service auth"
+    )
 
 
 class PythonBuild(BaseModel):

truefoundry/deploy/builder/builders/tfy_python_buildpack/dockerfile_template.py

@@ -3,7 +3,7 @@ from typing import Dict, List, Optional
 
 from mako.template import Template
 
-from truefoundry.common.constants import TFY_CLI_PYTHONBUILD_PYTHON_IMAGE_REPO_ENV_KEY
+from truefoundry.common.constants import ENV_VARS
 from truefoundry.deploy.auto_gen.models import PythonBuild
 from truefoundry.deploy.builder.constants import (
     PIP_CONF_BUILDKIT_SECRET_MOUNT,
@@ -156,9 +156,7 @@ def generate_dockerfile_content(
         apt_packages=build_configuration.apt_packages
     )
     template_args = {
-        "python_image_repo": os.getenv(
-            TFY_CLI_PYTHONBUILD_PYTHON_IMAGE_REPO_ENV_KEY, DEFAULT_PYTHON_IMAGE_REPO
-        ),
+        "python_image_repo": ENV_VARS.TFY_PYTHONBUILD_PYTHON_IMAGE_REPO,
         "python_version": build_configuration.python_version,
         "apt_install_command": apt_install_command,
         "requirements_path": requirements_path,

truefoundry/deploy/lib/clients/servicefoundry_client.py

@@ -48,6 +48,7 @@ from truefoundry.pydantic_v1 import parse_obj_as
 
 DEPLOYMENT_LOGS_SUBSCRIBE_MESSAGE = "DEPLOYMENT_LOGS"
 BUILD_LOGS_SUBSCRIBE_MESSAGE = "BUILD_LOGS"
+MAX_RETRIES_WORKFLOW_TRIGGER = 3
 
 if TYPE_CHECKING:
     from truefoundry.deploy.auto_gen.models import Application
@@ -597,7 +598,7 @@ class ServiceFoundryServiceClient(BaseServiceFoundryServiceClient):
     def trigger_workflow(self, application_id: str, inputs: Dict[str, Any]):
         url = f"{self._api_server_url}/{VERSION_PREFIX}/workflow/{application_id}/executions"
         body = {"inputs": inputs}
-        response = session_with_retries().post(
+        response = session_with_retries(retries=MAX_RETRIES_WORKFLOW_TRIGGER).post(
             url, json=body, headers=self._get_header()
         )
         response = request_handling(response)

truefoundry/deploy/lib/model/entity.py

@@ -32,7 +32,6 @@ class Base(BaseModel):
 class Entity(Base):
     createdAt: datetime.datetime = Field(repr=False)
     updatedAt: datetime.datetime = Field(repr=False)
-    createdBy: str = Field(repr=False)
 
 
 class Workspace(Entity):
@@ -56,7 +55,6 @@ class Workspace(Entity):
             "cluster_fqn": self.clusterId,
             "created_at": self.createdAt,
             "updated_at": self.updatedAt,
-            "created_by": self.createdBy,
         }
 
 
@@ -161,7 +159,6 @@ class Deployment(Entity):
             "cluster_fqn": cluster_fqn,
             "created_at": self.createdAt,
             "updated_at": self.updatedAt,
-            "created_by": self.createdBy,
         }
 
     class Config:
@@ -208,7 +205,6 @@ class Application(Entity):
             "cluster_fqn": self.workspace.clusterId,
             "created_at": self.createdAt,
             "updated_at": self.updatedAt,
-            "created_by": self.createdBy,
         }
 
 

truefoundry/ml/artifact/truefoundry_artifact_repo.py

@@ -34,6 +34,7 @@ from rich.progress import (
 )
 from tqdm.utils import CallbackIOWrapper
 
+from truefoundry.common.constants import ENV_VARS
 from truefoundry.ml.autogen.client import (  # type: ignore[attr-defined]
     ApiClient,
     CreateMultiPartUploadForDatasetRequestDto,
@@ -58,14 +59,12 @@ from truefoundry.ml.clients.utils import (
     augmented_raise_for_status,
     cloud_storage_http_request,
 )
-from truefoundry.ml.env_vars import DISABLE_MULTIPART_UPLOAD
 from truefoundry.ml.exceptions import MlFoundryException
 from truefoundry.ml.logger import logger
 from truefoundry.ml.session import _get_api_client
 from truefoundry.pydantic_v1 import BaseModel, root_validator
 
 _MIN_BYTES_REQUIRED_FOR_MULTIPART = 100 * 1024 * 1024
-_MULTIPART_DISABLED = os.getenv(DISABLE_MULTIPART_UPLOAD, "").lower() == "true"
 # GCP/S3 Maximum number of parts per upload	10,000
 # Maximum number of blocks in a block blob 50,000 blocks
 # TODO: This number is artificially limited now. Later
@@ -84,9 +83,6 @@ _MAX_NUM_PARTS_FOR_MULTIPART = 1000
 # Azure Maximum size of a block in a block blob	4000 MiB
 # GCP/S3 Maximum size of an individual part in a multipart upload 5 GiB
 _MAX_PART_SIZE_BYTES_FOR_MULTIPART = 4 * 1024 * 1024 * 1000
-_cpu_count = os.cpu_count() or 2
-_MAX_WORKERS_FOR_UPLOAD = max(min(32, _cpu_count * 2), 4)
-_MAX_WORKERS_FOR_DOWNLOAD = max(min(32, _cpu_count * 2), 4)
 _LIST_FILES_PAGE_SIZE = 500
 _GENERATE_SIGNED_URL_BATCH_SIZE = 50
 DEFAULT_PRESIGNED_URL_EXPIRY_TIME = 3600
@@ -199,7 +195,10 @@ class _FileMultiPartInfo(NamedTuple):
 
 def _decide_file_parts(file_path: str) -> _FileMultiPartInfo:
     file_size = os.path.getsize(file_path)
-    if file_size < _MIN_BYTES_REQUIRED_FOR_MULTIPART or _MULTIPART_DISABLED:
+    if (
+        file_size < _MIN_BYTES_REQUIRED_FOR_MULTIPART
+        or ENV_VARS.TFY_ARTIFACTS_DISABLE_MULTIPART_UPLOAD
+    ):
         return _FileMultiPartInfo(1, part_size=file_size, file_size=file_size)
 
     ideal_num_parts = math.ceil(file_size / _PART_SIZE_BYTES_FOR_MULTIPART)
@@ -262,7 +261,7 @@ def _signed_url_upload_file(
 def _download_file_using_http_uri(
     http_uri,
     download_path,
-    chunk_size=100000000,
+    chunk_size=ENV_VARS.TFY_ARTIFACTS_DOWNLOAD_CHUNK_SIZE_BYTES,
     callback: Optional[Callable[[int, int], Any]] = None,
 ):
     """
@@ -283,6 +282,9 @@ def _download_file_using_http_uri(
                 if not chunk:
                     break
                 output_file.write(chunk)
+                if ENV_VARS.TFY_ARTIFACTS_DOWNLOAD_FSYNC_CHUNKS:
+                    output_file.flush()
+                    os.fsync(output_file.fileno())
 
 
 class _CallbackIOWrapperForMultiPartUpload(CallbackIOWrapper):
@@ -608,7 +610,7 @@ class MlFoundryArtifactsRepository:
             disable=not show_progress,
             expand=True,
         ) as progress_bar, ThreadPoolExecutor(
-            max_workers=_MAX_WORKERS_FOR_UPLOAD
+            max_workers=ENV_VARS.TFY_ARTIFACTS_UPLOAD_MAX_WORKERS
         ) as executor:
             futures: List[Future] = []
             # Note: While this batching is beneficial when there is a large number of files, there is also
@@ -763,7 +765,9 @@ class MlFoundryArtifactsRepository:
             )
 
         if not executor_for_multipart_upload:
-            with ThreadPoolExecutor(max_workers=_MAX_WORKERS_FOR_UPLOAD) as executor:
+            with ThreadPoolExecutor(
+                max_workers=ENV_VARS.TFY_ARTIFACTS_UPLOAD_MAX_WORKERS
+            ) as executor:
                 return self._multipart_upload(
                     local_file=local_file,
                     artifact_path=artifact_path,
@@ -899,7 +903,9 @@ class MlFoundryArtifactsRepository:
                         )
                     file_paths.append((file_path, download_dest_path))
 
-                with ThreadPoolExecutor(_MAX_WORKERS_FOR_DOWNLOAD) as executor:
+                with ThreadPoolExecutor(
+                    max_workers=ENV_VARS.TFY_ARTIFACTS_DOWNLOAD_MAX_WORKERS
+                ) as executor:
                     # Note: While this batching is beneficial when there is a large number of files, there is also
                     # a rare case risk of the signed url expiring before a request is made to it
                     batch_size = _GENERATE_SIGNED_URL_BATCH_SIZE

truefoundry/ml/log_types/artifacts/artifact.py

@@ -184,11 +184,6 @@ class ArtifactVersion:
         _validate_artifact_metadata(value)
         self._metadata = value
 
-    @property
-    def created_by(self) -> str:
-        """Get the information about who created the artifact"""
-        return self._artifact_version.created_by
-
     @property
     def created_at(self) -> datetime.datetime:
         """Get the time at which artifact was created"""

truefoundry/ml/log_types/artifacts/dataset.py

@@ -124,11 +124,6 @@ class DataDirectory:
         _validate_artifact_metadata(value)
         self._metadata = value
 
-    @property
-    def created_by(self) -> str:
-        """Get the information about who created the DataDirectory"""
-        return self._dataset.created_by
-
     @property
     def created_at(self) -> datetime.datetime:
         """Get the time at which DataDirectory was created"""

truefoundry/ml/log_types/artifacts/model.py

@@ -219,11 +219,6 @@ class ModelVersion:
             metrics_as_kv[metric.key] = metric.value
         return metrics_as_kv
 
-    @property
-    def created_by(self) -> str:
-        """Get the information about who created the model version"""
-        return self._model_version.created_by
-
     @property
     def created_at(self) -> datetime.datetime:
         """Get the time at which model version was created"""

truefoundry/ml/mlfoundry_api.py

@@ -8,7 +8,7 @@ import coolname
 import pandas as pd
 
 from truefoundry.common.utils import relogin_error_message
-from truefoundry.ml import constants, env_vars
+from truefoundry.ml import constants
 from truefoundry.ml.autogen.client import (  # type: ignore[attr-defined]
     ArtifactDto,
     ArtifactType,
@@ -59,10 +59,15 @@ from truefoundry.ml.validation_utils import (
 
 _SEARCH_MAX_RESULTS_DEFAULT = 1000
 
+_INTERNAL_ENV_VARS = [
+    "TFY_INTERNAL_APPLICATION_ID",
+    "TFY_INTERNAL_JOB_RUN_NAME",
+]
+
 
 def _get_internal_env_vars_values() -> Dict[str, str]:
     env = {}
-    for env_var_name in env_vars.INTERNAL_ENV_VARS:
+    for env_var_name in _INTERNAL_ENV_VARS:
         value = os.getenv(env_var_name)
         if value:
             env[env_var_name] = value

truefoundry/ml/session.py

@@ -113,6 +113,8 @@ def _get_api_client(
     session: Optional[Session] = None,
     allow_anonymous: bool = False,
 ) -> ApiClient:
+    from truefoundry.version import __version__
+
     session = session or get_active_session()
     if session is None:
         if allow_anonymous:
@@ -131,7 +133,7 @@ def _get_api_client(
     )
     configuration.retries = urllib3_retry(retries=2)
     api_client = ApiClient(configuration=configuration)
-    api_client.user_agent = "truefoundry-cli"
+    api_client.user_agent = f"truefoundry-cli/{__version__}"
     return api_client
 
 

{truefoundry-0.4.3.dist-info → truefoundry-0.4.4rc1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: truefoundry
-Version: 0.4.3
+Version: 0.4.4rc1
 Summary: Truefoundry CLI
 Author: Abhishek Choudhary
 Author-email: abhishek@truefoundry.com

{truefoundry-0.4.3.dist-info → truefoundry-0.4.4rc1.dist-info}/RECORD

@@ -27,23 +27,23 @@ 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/auth_service_client.py,sha256=tZOa0NdATnItsMeTnEnUeTZQIgUJtpU-nvLdWtB4Px8,7978
-truefoundry/common/constants.py,sha256=gnswB-aK5atSRY-GE1Cji6ZCEs2FoK9n1lCX5JuQGqI,484
+truefoundry/common/constants.py,sha256=zpU9PD3B_cRVev6u9A37gKMxkPrcD6knJbKpwUCsaQg,1354
 truefoundry/common/credential_file_manager.py,sha256=1yEk1Zm2xS4G0VDFwKSZ4w0VUrcPWQ1nJnoBaz9xyKA,4251
-truefoundry/common/credential_provider.py,sha256=YQ6HKl8ZZFTg48vBZMauEAnM6IrEO3oOzM2DA47N-P0,4071
+truefoundry/common/credential_provider.py,sha256=Aht7hFLsnyRgMR34dRbzln7dor0WYSeA8ej8ApNmnKM,4148
 truefoundry/common/entities.py,sha256=8O-EGPk4PKqnyoFMKUTxISCU19rz0KBnfRDJU695DhY,3797
 truefoundry/common/exceptions.py,sha256=ePpiQ_zmWe4e94gOgeMiyP_AZnKwjEBfyXsB5ScGYcI,329
-truefoundry/common/request_utils.py,sha256=-ss8033PClJOMJuS5Ue1zCFRVK7ZW9vjtva1b5G9uok,2849
-truefoundry/common/servicefoundry_client.py,sha256=klohrELTFTJX-jvl9LGtdIoJiZMZRuQQB4Ho7PJ7u3s,3091
-truefoundry/common/utils.py,sha256=lFiDEVoQDJoZPEN4Zun4GQz2TsfdbXfNHI41BErETq8,3320
+truefoundry/common/request_utils.py,sha256=5xw4YGUcMf71Ncal3OfFCa-PoWDIvG3hYGCDa4Da4OI,2854
+truefoundry/common/servicefoundry_client.py,sha256=2fxmgCM-ckFHpnm6n_mL-5Z8RWN_q-dYVvFC29bkYSg,3120
+truefoundry/common/utils.py,sha256=MYFjNtHGqauqhj9tmbdErCJR49AfXDwg-5kYbBh8HpI,3258
 truefoundry/deploy/__init__.py,sha256=ugawKF2G02EmEXX35oZ2tec12d9oWN28Sf6mtGGIERY,2281
-truefoundry/deploy/auto_gen/models.py,sha256=AjbjNFwWDTIOjN_E1fSUgP-ej9nsb2OpZPhsnIRAVGs,80621
+truefoundry/deploy/auto_gen/models.py,sha256=4MaxkG2_5Wg6avaZRlK0D4JiVEM5rk3NU0BCiTx8VyU,82477
 truefoundry/deploy/builder/__init__.py,sha256=1qjHMNBE1poRCZW0WrG46dFM1f1IlivD5352qzsioMU,4953
 truefoundry/deploy/builder/builders/__init__.py,sha256=tlFLXqyDaKLd4iZbo4Hcu_8gOmgtL6drnXpbmQ6x1P8,636
 truefoundry/deploy/builder/builders/dockerfile.py,sha256=AXXTziCkaqIhuM_bwyD1vT1znOwemN1TKgU7eyo-KuM,1522
 truefoundry/deploy/builder/builders/tfy_notebook_buildpack/__init__.py,sha256=UmMcTY-8MrLY3H5owpn6ax-VePQl4MiMTmHlQ9qEtQw,1742
 truefoundry/deploy/builder/builders/tfy_notebook_buildpack/dockerfile_template.py,sha256=IZbXsSj86SxWRZNQXepEnsOL_eYwWaOq5oh-WiyvL2c,2485
 truefoundry/deploy/builder/builders/tfy_python_buildpack/__init__.py,sha256=XNJ3MKWqyWIbFNnUQMpB8oVC5Pt5Wsm_bRKbvkXRIG8,1696
-truefoundry/deploy/builder/builders/tfy_python_buildpack/dockerfile_template.py,sha256=cJlQK_kgNELULcp2Q_kMWyPNmUuZdhOELDnI0rMiqRc,7473
+truefoundry/deploy/builder/builders/tfy_python_buildpack/dockerfile_template.py,sha256=nwTpeIrDOZC8_m_VcbQ5Vp_rEO2ioC8FtzOEo17JY7A,7373
 truefoundry/deploy/builder/constants.py,sha256=eIukBjD6I4KvEmAPpdbPlPPr76yhS-uNr3RVFkzEdgs,257
 truefoundry/deploy/builder/docker_service.py,sha256=OI8efqK0Gnoii8bcHihpA2StwHVzsMREfBk7NvMR4hY,3950
 truefoundry/deploy/builder/utils.py,sha256=9RZnkhoHFTRUt_x3nck0aVz7cLpzA3jiwQH-ZZZrjf8,938
@@ -92,7 +92,7 @@ truefoundry/deploy/json_util.py,sha256=x_-7YYQ4_HUIJ8ofOcclAp9JWhgTWjR9Th6Q0FuRq
 truefoundry/deploy/lib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/deploy/lib/auth/servicefoundry_session.py,sha256=5TCYPunAygtn5mb0mp_VcWKEalKMKPbyWMWer-Vty2g,1916
 truefoundry/deploy/lib/clients/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truefoundry/deploy/lib/clients/servicefoundry_client.py,sha256=vFrd5yeStjT_uWL3JxHrx0iAmNpuv26K7IEnsF1RMM8,26101
+truefoundry/deploy/lib/clients/servicefoundry_client.py,sha256=tkuPNbo9vDTgXPHD4p3d-t6HigBvKNfML1pTQXBwJfY,26170
 truefoundry/deploy/lib/const.py,sha256=repGJLuoMqtzeq5tCjjkN4bH187FVHVKI30BricOlvc,244
 truefoundry/deploy/lib/dao/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/deploy/lib/dao/application.py,sha256=uUTFSQkLUrFCtQQgS2Jm9BpyHyhMkN4GI1yx9oJo4_E,9161
@@ -103,7 +103,7 @@ truefoundry/deploy/lib/exceptions.py,sha256=ZT2o3mar3BYtnjKHn2wf4rBGEaFIgf-zkQmz
 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=8J8yd98iWtSy8giShdDRNyzbN1UgSXx4XtmZLljdWnE,8552
+truefoundry/deploy/lib/model/entity.py,sha256=fq8hvdJQgQn4uZqxpKrzmaoJhQG53_EbDoDxUPPmOZ0,8387
 truefoundry/deploy/lib/session.py,sha256=Vg6rCA315T0yS0xG4ayJ84Ia_9ZfibH8utOSwPBMAmw,4953
 truefoundry/deploy/lib/util.py,sha256=3TapV7yczkheC1MMMfmJDGGzTl2l6e4jCYd_Rr5aoQ8,1330
 truefoundry/deploy/lib/win32.py,sha256=1RcvPTdlOAJ48rt8rCbE2Ufha2ztRqBAE9dueNXArrY,5009
@@ -125,7 +125,7 @@ truefoundry/langchain/utils.py,sha256=PGLDe9chZ3BuUjakexOGpIqZRFoHEgu-zJ9yKdpLLm
 truefoundry/logger.py,sha256=7dLqW_Q2rEgo-_z1WZnQbGHaoy1L1MP3NqGgssaSS6o,685
 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=dA_QYAyNCcCuOw5eVxPbU5OeMK8YBNh42vBnGf6I0Sw,44911
+truefoundry/ml/artifact/truefoundry_artifact_repo.py,sha256=uS4FZq4wUhxB8-BDx--fLYTcsLYL_V9clNtb32G0Wo0,45050
 truefoundry/ml/autogen/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truefoundry/ml/autogen/client/__init__.py,sha256=M68r1z8Mi6dEFFKDsWPJ9NGdNOYdm6pcdH7_srCwkCI,16651
 truefoundry/ml/autogen/client/api/__init__.py,sha256=3sMSljMIS3UHYeF0BcNvrPPx6VbBSRt_1IfDn-13Kyc,752
@@ -302,16 +302,15 @@ truefoundry/ml/clients/utils.py,sha256=c0LdC8moejs-Zm30hu1sCqifLEmqhdq4SfZ_m0nUI
 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=0CqhoV6RgxgPju71AJYvo7crHxVr0vJNzWTiSWKnLcI,212
 truefoundry/ml/exceptions.py,sha256=8aJm2NYtAWWsRLu4MbzaoOqHsQZ6RjOFwBWQWqb6qrc,244
 truefoundry/ml/git_info.py,sha256=jvAVm9ilqivnGq8qJdUvYdd8Siv0PLtqurB-PXsS5ho,2023
 truefoundry/ml/internal_namespace.py,sha256=QcqMHp6-C2im2H_02hlhi01EIcr1HhNaZprszs13EMU,1790
 truefoundry/ml/log_types/__init__.py,sha256=g4u4D4Jaj0aBK5GtrLV88-qThKZR9pSZ17vFEkN-LmM,125
-truefoundry/ml/log_types/artifacts/artifact.py,sha256=pNgLc6GaQ0UVJ6gaSRzpAW5CmGpJMC4iBgBY5FLlofc,17640
+truefoundry/ml/log_types/artifacts/artifact.py,sha256=ltHLkJLLeeRO1d6BqoSq3VtizicvyjdG_Hf52wqig7w,17478
 truefoundry/ml/log_types/artifacts/constants.py,sha256=qKxQ5mMvJE4j83BvGW3qNTKunxCiBg_EEjTdgbgJtyE,1036
-truefoundry/ml/log_types/artifacts/dataset.py,sha256=oXi7itXOe3PrnN4_KY7s1GdhWV1lyvWXRWgY6_vPTAs,12981
+truefoundry/ml/log_types/artifacts/dataset.py,sha256=ou6hF0RnwQ8clgpWW8SJPOu2YCG-xya51b91pGY-PjY,12823
 truefoundry/ml/log_types/artifacts/general_artifact.py,sha256=fY74EUeCGdHdlkrp_GMoaZPvl7EQctt8OlRM6yFgLF8,3772
-truefoundry/ml/log_types/artifacts/model.py,sha256=NRIvGX0yCLGnLvZkuCRJGGEuONoUhXZSY2jmSUM3JvI,22162
+truefoundry/ml/log_types/artifacts/model.py,sha256=Qb0wq2qCbPCk4os74xsJwOXUQc-4sdGsBwKkV9emDsI,21998
 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
@@ -323,10 +322,10 @@ truefoundry/ml/log_types/plot.py,sha256=oFnXNb2o5fVF0zsnRjvqjSjLaphQWUnQCdw72e2u
 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/mlfoundry_api.py,sha256=I2T8tXeAIWpD8EH05fm80mNyX6cs2S1ORI4qoo0HTpQ,60847
+truefoundry/ml/mlfoundry_api.py,sha256=UV42BLXf3R9Ql7TBJ4zGUoHq2B4ENEU2jK9aaEniGQc,60923
 truefoundry/ml/mlfoundry_run.py,sha256=rNJl130iJkpjW3MNoe5-d_J9VJJQBqWHEJCfYeiZCbE,45123
 truefoundry/ml/run_utils.py,sha256=0W208wSLUrbdfk2pjNcZlkUi9bNxG2JORqoe-5rVqHI,2423
-truefoundry/ml/session.py,sha256=Rvq419hdLNBt0KUzKn7jGvKKg9LI_gafOHcw-lFceAE,5330
+truefoundry/ml/session.py,sha256=F83GTC5WwGBjnJ69Ct8MqMnlutYc56JCc6YhEY1Wl-A,5394
 truefoundry/ml/validation_utils.py,sha256=XBSUd9OoyriWJpT3M5LKz17iWY3yVMr3hM5vdaVjtf0,12082
 truefoundry/pydantic_v1.py,sha256=jSuhGtz0Mbk1qYu8jJ1AcnIDK4oxUsdhALc4spqstmM,345
 truefoundry/version.py,sha256=bqiT4Q-VWrTC6P4qfK43mez-Ppf-smWfrl6DcwV7mrw,137
@@ -341,7 +340,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.3.dist-info/METADATA,sha256=GWyg6SdKw3D3tmYLGzlQ6tYsosmIVMAC0NuUMJ3qsJs,3098
-truefoundry-0.4.3.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-truefoundry-0.4.3.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
-truefoundry-0.4.3.dist-info/RECORD,,
+truefoundry-0.4.4rc1.dist-info/METADATA,sha256=1fqRs-fBIXRc1DI21OJSATulj9ZPmnUtuNKVZML2y-U,3101
+truefoundry-0.4.4rc1.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+truefoundry-0.4.4rc1.dist-info/entry_points.txt,sha256=TXvUxQkI6zmqJuycPsyxEIMr3oqfDjgrWj0m_9X12x4,95
+truefoundry-0.4.4rc1.dist-info/RECORD,,

truefoundry/ml/env_vars.py

@@ -1,9 +0,0 @@
-"""
-Define all the env variable names that users can use
-"""
-
-DISABLE_MULTIPART_UPLOAD = "MLF_DISABLE_MULTIPART_UPLOAD"
-INTERNAL_ENV_VARS = [
-    "TFY_INTERNAL_APPLICATION_ID",
-    "TFY_INTERNAL_JOB_RUN_NAME",
-]