truefoundry 0.4.3__tar.gz → 0.4.4rc1__tar.gz

{truefoundry-0.4.3 → truefoundry-0.4.4rc1}/PKG-INFO

@@ -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 → truefoundry-0.4.4rc1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "truefoundry"
-version = "0.4.3"
+version = "0.4.4rc1"
 description = "Truefoundry CLI"
 authors = ["Abhishek Choudhary <abhishek@truefoundry.com>"]
 readme = "README.md"
@@ -158,5 +158,5 @@ warn_required_dynamic_aliases = true
 
 [tool.pytest.ini_options]
 env = [
-    "TFY_CLI_PYTHONBUILD_PYTHON_IMAGE_REPO=docker.io/truefoundrycloud/test/python",
+    "TFY_PYTHONBUILD_PYTHON_IMAGE_REPO=docker.io/truefoundrycloud/test/python",
 ]

truefoundry-0.4.4rc1/truefoundry/common/constants.py

@@ -0,0 +1,33 @@
+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"
+
+
+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"
+SERVICEFOUNDRY_CLIENT_MAX_RETRIES = 2

{truefoundry-0.4.3 → truefoundry-0.4.4rc1}/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-0.4.3 → truefoundry-0.4.4rc1}/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-0.4.3 → truefoundry-0.4.4rc1}/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-0.4.3 → truefoundry-0.4.4rc1}/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-0.4.3 → truefoundry-0.4.4rc1}/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-0.4.3 → truefoundry-0.4.4rc1}/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-0.4.3 → truefoundry-0.4.4rc1}/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-0.4.3 → truefoundry-0.4.4rc1}/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-0.4.3 → truefoundry-0.4.4rc1}/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-0.4.3 → truefoundry-0.4.4rc1}/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-0.4.3 → truefoundry-0.4.4rc1}/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-0.4.3 → truefoundry-0.4.4rc1}/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-0.4.3 → truefoundry-0.4.4rc1}/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-0.4.3 → truefoundry-0.4.4rc1}/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/truefoundry/common/constants.py

@@ -1,15 +0,0 @@
-from pathlib import Path
-
-TFY_CONFIG_DIR = Path.home() / ".truefoundry"
-CREDENTIAL_FILEPATH = TFY_CONFIG_DIR / "credentials.json"
-
-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"
-
-
-API_SERVER_RELATIVE_PATH = "api/svc"
-MLFOUNDRY_SERVER_RELATIVE_PATH = "api/ml"
-VERSION_PREFIX = "v1"
-SERVICEFOUNDRY_CLIENT_MAX_RETRIES = 2

truefoundry-0.4.3/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",
-]