truefoundry 0.3.4__tar.gz → 0.4.0__tar.gz

{truefoundry-0.3.4 → truefoundry-0.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: truefoundry
-Version: 0.3.4
+Version: 0.4.0
 Summary: Truefoundry CLI
 Author: Abhishek Choudhary
 Author-email: abhishek@truefoundry.com
@@ -10,37 +10,44 @@ Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
-Provides-Extra: ml
 Provides-Extra: workflow
-Requires-Dist: GitPython (>=3.1.43,<3.2.0)
+Requires-Dist: GitPython (>=3.1.43,<4.0.0)
 Requires-Dist: Mako (>=1.1.6,<2.0.0)
 Requires-Dist: PyJWT (>=2.0.0,<3.0.0)
 Requires-Dist: PyYAML (>=6.0.0,<7.0.0)
+Requires-Dist: aenum (>=3.0.0,<4.0.0)
 Requires-Dist: click (>=7.0.0,<9.0.0)
+Requires-Dist: coolname (>=1.1.0,<2.0.0)
 Requires-Dist: docker (>=6.1.2,<8.0.0)
 Requires-Dist: fastapi (>=0.56.0,<0.200.0)
 Requires-Dist: filelock (>=3.8.0,<4.0.0)
 Requires-Dist: flytekit (==1.12.2) ; extra == "workflow"
-Requires-Dist: gitignorefile (>=1.1.2,<1.2.0)
-Requires-Dist: importlib-metadata (>=6.0.1,<8.0.0)
-Requires-Dist: importlib-resources (>=5.2.0,<6.0.0)
-Requires-Dist: mlfoundry (==0.11.5) ; extra == "ml"
+Requires-Dist: gitignorefile (>=1.1.2,<2.0.0)
+Requires-Dist: importlib-metadata (>=4.11.3,<9.0.0)
+Requires-Dist: importlib-resources (>=5.2.0,<7.0.0)
+Requires-Dist: numpy (>=1.23.0,<2.0.0) ; python_version < "3.12"
+Requires-Dist: numpy (>=1.26.0,<2.0.0) ; python_version >= "3.12"
 Requires-Dist: openai (>=1.16.2,<2.0.0)
 Requires-Dist: packaging (>=20.0,<25.0)
-Requires-Dist: pydantic (>=1.10.0,<3)
+Requires-Dist: pandas (>=1.0.0,<3.0.0) ; python_version < "3.10"
+Requires-Dist: pandas (>=1.4.0,<3.0.0) ; python_version >= "3.10"
+Requires-Dist: psutil (>=5.9.0,<7.0.0)
+Requires-Dist: pydantic (>=1.8.2,<3.0.0)
 Requires-Dist: pygments (>=2.12.0,<3.0.0)
 Requires-Dist: python-dateutil (>=2.8.2,<3.0.0)
-Requires-Dist: python-dotenv (>=1.0.1,<1.1.0)
+Requires-Dist: python-dotenv (>=1.0.1,<2.0.0)
 Requires-Dist: python-socketio[client] (>=5.5.2,<6.0.0)
 Requires-Dist: questionary (>=1.10.0,<2.0.0)
 Requires-Dist: requests (>=2.31.0,<3.0.0)
-Requires-Dist: requirements-parser (>=0.10.2,<0.11.0)
+Requires-Dist: requirements-parser (>=0.11.0,<0.12.0)
 Requires-Dist: rich (>=13.7.1,<14.0.0)
 Requires-Dist: rich-click (>=1.2.1,<2.0.0)
+Requires-Dist: scipy (>=1.12.0,<2.0.0) ; python_version >= "3.12"
+Requires-Dist: scipy (>=1.5.0,<2.0.0) ; python_version < "3.12"
 Requires-Dist: tqdm (>=4.0.0,<5.0.0)
-Requires-Dist: typing-extensions (>=3.10.0)
+Requires-Dist: typing-extensions (>=4.0)
 Requires-Dist: urllib3 (>=1.26.18,<3)
-Requires-Dist: uvicorn (>=0.13.0,<0.40.0)
+Requires-Dist: uvicorn (>=0.13.0,<1.0.0)
 Requires-Dist: yq (>=3.1.0,<4.0.0)
 Description-Content-Type: text/markdown
 
@@ -61,10 +68,10 @@ To install the `TrueFoundry CLI`, you can use pip:
 pip install truefoundry
 ```
 
-To install the `TrueFoundry CLI` with metadata/artifact tracking, you can use pip:
+To install the `TrueFoundry CLI` with Workflow deployment features, you can use pip:
 
 ```bash
-pip install truefoundry[ml]
+pip install truefoundry[workflow]
 ```
 
 ## Examples

{truefoundry-0.3.4 → truefoundry-0.4.0}/README.md

@@ -15,10 +15,10 @@ To install the `TrueFoundry CLI`, you can use pip:
 pip install truefoundry
 ```
 
-To install the `TrueFoundry CLI` with metadata/artifact tracking, you can use pip:
+To install the `TrueFoundry CLI` with Workflow deployment features, you can use pip:
 
 ```bash
-pip install truefoundry[ml]
+pip install truefoundry[workflow]
 ```
 
 ## Examples

truefoundry-0.4.0/pyproject.toml

@@ -0,0 +1,156 @@
+[tool.poetry]
+name = "truefoundry"
+version = "0.4.0"
+description = "Truefoundry CLI"
+authors = ["Abhishek Choudhary <abhishek@truefoundry.com>"]
+readme = "README.md"
+
+[tool.poetry.dependencies]
+python = ">=3.8.1,<3.13"
+aenum = ">=3.0.0,<4.0.0"
+click = ">=7.0.0,<9.0.0"
+coolname = ">=1.1.0,<2.0.0"
+docker = ">=6.1.2,<8.0.0"
+fastapi = ">=0.56.0,<0.200.0"
+filelock = ">=3.8.0,<4.0.0"
+flytekit = { version = "1.12.2", optional = true }
+gitignorefile = ">=1.1.2,<2.0.0"
+GitPython = ">=3.1.43,<4.0.0"
+importlib-metadata = ">=4.11.3,<9.0.0"
+importlib-resources = ">=5.2.0,<7.0.0"
+Mako = ">=1.1.6,<2.0.0"
+openai = ">=1.16.2,<2.0.0"
+packaging = ">=20.0,<25.0"
+psutil = ">=5.9.0,<7.0.0"
+pydantic = ">=1.8.2,<3.0.0"
+pygments = ">=2.12.0,<3.0.0"
+PyJWT = ">=2.0.0,<3.0.0"
+python-dateutil = ">=2.8.2,<3.0.0"
+python-dotenv = ">=1.0.1,<2.0.0"
+python-socketio = { version = ">=5.5.2,<6.0.0", extras = ["client"] }
+PyYAML = ">=6.0.0,<7.0.0"
+questionary = ">=1.10.0,<2.0.0"
+requests = ">=2.31.0,<3.0.0"
+requirements-parser = ">=0.11.0,<0.12.0"
+rich = ">=13.7.1,<14.0.0"
+rich-click = ">=1.2.1,<2.0.0"
+tqdm = ">=4.0.0,<5.0.0"
+typing-extensions = ">=4.0"
+urllib3 = ">=1.26.18,<3"
+uvicorn = ">=0.13.0,<1.0.0"
+yq = ">=3.1.0,<4.0.0"
+# Check and try to eliminate libs below this comment
+scipy = [
+    # This is split into two because poetry will select a version which has no wheel on python >= 3.12 :/
+    { version = ">=1.5.0,<2.0.0", python = "<3.12" },
+    { version = ">=1.12.0,<2.0.0", python = ">=3.12" },
+]
+numpy = [
+    # This is split into two because poetry will select a version which has no wheel on python >= 3.12 :/
+    { version = ">=1.23.0,<2.0.0", python = "<3.12" },
+    { version = ">=1.26.0,<2.0.0", python = ">=3.12" },
+]
+pandas = [
+    # This is split into two because poetry will select a version which has no wheel on python >= 3.10 :/
+    { version = ">=1.0.0,<3.0.0", python = "<3.10" },
+    { version = ">=1.4.0,<3.0.0", python = ">=3.10" },
+]
+
+
+[tool.poetry.extras]
+workflow = ["flytekit"]
+
+[tool.poetry.group.dev.dependencies]
+Pillow = ">=9.1.1,<11.0.0"
+ipykernel = ">=6.16.0"
+ipython = ">=7.10.0"
+langchain = ">=0.2.0,<0.3.0"
+matplotlib = ">=3.0.0,<4.0.0"
+mypy = "*"
+plotly = ">=5.8.2,<6.0.0"
+pre-commit = ">=3.5.0,<4.0.0"
+pytest = ">=8.2.2,<9.0.0"
+ruff = ">=0.4.10"
+seaborn = ">=0.13.0,<0.14.0"
+scikit-learn = ">=1.3.0,<2.0.0"
+shap =  [
+    { version = ">=0.40.0", python = "<3.12" },
+    { version = ">=0.45.0", python = ">=3.12" },
+]
+
+
+[tool.poetry-dynamic-versioning]
+enable = false
+
+[build-system]
+requires = ["poetry-core>=1.0.0", "poetry-dynamic-versioning>=1.0.0,<2.0.0"]
+build-backend = "poetry_dynamic_versioning.backend"
+
+[tool.poetry.plugins."console_scripts"]
+tfy = "truefoundry.cli.__main__:main"
+truefoundry = "truefoundry.cli.__main__:main"
+
+[tool.ruff]
+lint.select = [
+    "E",  # pycodestyle errors
+    "W",  # pycodestyle warnings
+    "F",  # pyflakes
+    "C",  # flake8-comprehensions
+    "B",  # flake8-bugbear
+    "Q",  # flake8-quotes
+    "I",
+]
+lint.ignore = ["E501"]
+exclude = ["venv", "repos"]
+
+[tool.ruff.lint.per-file-ignores]
+"__init__.py" = ["F401"]
+"truefoundry/ml/autogen/client/*" = ["B028", "B904", "C901", "C409", "F841"]
+"truefoundry/ml/autogen/client/models/*" = ["F821"]
+"truefoundry/ml/autogen/client/api_client.py" = ["E721"]
+
+[tool.mypy]
+plugins = [
+  "pydantic.mypy"
+]
+install_types = true
+non_interactive = true
+
+allow_redefinition = false
+allow_untyped_globals = false
+check_untyped_defs = true
+disallow_any_generics = true
+disallow_incomplete_defs = false
+disallow_subclassing_any = true
+disallow_untyped_calls = false
+disallow_untyped_decorators = false
+disallow_untyped_defs = false
+follow_imports = "silent"
+force_uppercase_builtins = true
+no_implicit_optional = true
+no_implicit_reexport = true
+strict_equality = true
+strict_optional = true
+warn_no_return = true
+warn_redundant_casts = true
+warn_return_any = true
+warn_unreachable = true
+warn_unused_ignores = true
+
+packages = [
+    "truefoundry.common",
+    "truefoundry.ml",
+]
+exclude = [
+    "^truefoundry/ml/autogen",
+    "^truefoundry/deploy/auto_gen",
+    "^truefoundry/pydantic_v1.py$",
+]
+disable_error_code = [
+    "import-untyped",
+]
+
+[tool.pydantic-mypy]
+init_forbid_extra = true
+init_typed = true
+warn_required_dynamic_aliases = true

{truefoundry-0.3.4 → truefoundry-0.4.0}/truefoundry/__init__.py

@@ -1 +1,3 @@
 from truefoundry.deploy.core import login, logout
+
+__all__ = ["login", "logout"]

{truefoundry-0.3.4 → truefoundry-0.4.0}/truefoundry/autodeploy/agents/developer.py

@@ -23,7 +23,7 @@ from truefoundry.autodeploy.tools import (
 class Developer(Agent):
     system_prompt = """
 You are a software engineer.
-You goal is to do whatever you have to do to succesfully run a project.
+You goal is to do whatever you have to do to successfully run a project.
 If the project already contains a dockerfile, you can use that.
 If the project does not contain a dockerfile, you have to create one.
 You need to fix any issue to ensure that the image builds succesfully.

{truefoundry-0.3.4 → truefoundry-0.4.0}/truefoundry/autodeploy/agents/project_identifier.py

@@ -38,7 +38,7 @@ Prefer using file type counts over list files.
 use the knowledge of file type counts to create the right glob pattern while listing files.
 Prefer yarn over npm if the project is using yarn.
 If there is  a lock file, the project maybe using yarn.
-Look for *.lock if there is a lock type file persent.
+Look for *.lock if there is a lock type file present.
 Always response with a function call.
     """
     max_iter = 10
@@ -50,7 +50,7 @@ Always response with a function call.
             None,
             description="""
 A Service is designed to run always and should never terminate.
-A Job is desiged to finish after sometime.
+A Job is designed to finish after sometime.
 """,
         )
         primary_programming_language: str = Field(

{truefoundry-0.3.4 → truefoundry-0.4.0}/truefoundry/autodeploy/agents/tester.py

@@ -29,7 +29,7 @@ If you find any port number in the logs. Re run the docker image by exposing tha
 
 If the image is not running a service, try to identify whether there is any issue.
 Your goal is not to fix the issue. Your goal is to create a very detailed justification and report whether
-the testing was succesful.
+the testing was successful.
 Always response with a function call.
 Return response once you are done testing.
     """

{truefoundry-0.3.4 → truefoundry-0.4.0}/truefoundry/autodeploy/cli.py

@@ -292,7 +292,7 @@ def cli(project_root_path: str, deploy: bool, workspace_fqn: str = None):
         )
         if not re.match(r"^[a-z][a-z0-9\-]{1,30}[a-z0-9]$", name):
             console.print(
-                "[bold magenta]TrueFoundry:[/] The name should be between 2-30 alphaneumaric"
+                "[bold magenta]TrueFoundry:[/] The name should be between 2-30 alphanumeric"
                 " characters and '-'. The first character should not be a digit."
             )
         else:

{truefoundry-0.3.4 → truefoundry-0.4.0}/truefoundry/autodeploy/tools/list_files.py

@@ -20,7 +20,7 @@ from truefoundry.autodeploy.tools.base import (
 class ListFiles(Tool):
     description = """
 List all files.
-If you want to find all json files recuresively under directory a/b
+If you want to find all json files recursively under directory a/b
 the subdir should be a/b and pattern will be *.json
 
 If you want to find all json files recuresively under current directory

{truefoundry-0.3.4 → truefoundry-0.4.0}/truefoundry/cli/__main__.py

@@ -1,21 +1,9 @@
 import sys
 
-import click
+import rich_click as click
 
 from truefoundry.deploy.cli.cli import create_truefoundry_cli
-
-MLFOUNDRY_INSTALLED = True
-
-
-try:
-    from mlfoundry.cli.commands import download
-except ImportError:
-    MLFOUNDRY_INSTALLED = False
-
-
-@click.group()
-def ml():
-    """MlFoundry CLI"""
+from truefoundry.ml.cli.cli import get_ml_cli
 
 
 def main():
@@ -25,9 +13,7 @@ def main():
     # If it is another kind of object, it will be printed and the system exit status will be one (i.e., failure).
     try:
         cli = create_truefoundry_cli()
-        if MLFOUNDRY_INSTALLED:
-            ml.add_command(download)
-            cli.add_command(ml)
+        cli.add_command(get_ml_cli())
     except Exception as e:
         raise click.exceptions.UsageError(message=str(e)) from e
     sys.exit(cli())

{truefoundry-0.3.4/truefoundry/deploy/lib/auth → truefoundry-0.4.0/truefoundry/common}/auth_service_client.py

@@ -1,40 +1,39 @@
 import time
 from abc import ABC, abstractmethod
+from typing import Optional
 
 import requests
 
-from truefoundry.deploy.lib.clients.utils import poll_for_function, request_handling
-from truefoundry.deploy.lib.const import VERSION_PREFIX
-from truefoundry.deploy.lib.exceptions import BadRequestException
-from truefoundry.deploy.lib.model.entity import DeviceCode, Token
+from truefoundry.common.constants import VERSION_PREFIX
+from truefoundry.common.entities import DeviceCode, Token
+from truefoundry.common.exceptions import BadRequestException
+from truefoundry.common.request_utils import request_handling, requests_retry_session
+from truefoundry.common.utils import poll_for_function
 from truefoundry.logger import logger
 
 
 class AuthServiceClient(ABC):
-    def __init__(self, base_url):
-        from truefoundry.deploy.lib.clients.servicefoundry_client import (
-            ServiceFoundryServiceClient,
-        )
-
-        client = ServiceFoundryServiceClient(init_session=False, base_url=base_url)
-
-        self._api_server_url = client._api_server_url
-        self._auth_server_url = client.tenant_info.auth_server_url
-        self._tenant_name = client.tenant_info.tenant_name
+    def __init__(self, tenant_name: str):
+        self._tenant_name = tenant_name
 
     @classmethod
     def from_base_url(cls, base_url: str) -> "AuthServiceClient":
-        from truefoundry.deploy.lib.clients.servicefoundry_client import (
+        from truefoundry.common.servicefoundry_client import (
             ServiceFoundryServiceClient,
         )
 
-        client = ServiceFoundryServiceClient(init_session=False, base_url=base_url)
+        client = ServiceFoundryServiceClient(base_url=base_url)
         if client.python_sdk_config.use_sfy_server_auth_apis:
-            return ServiceFoundryServerAuthServiceClient(base_url)
-        return AuthServerServiceClient(base_url)
+            return ServiceFoundryServerAuthServiceClient(
+                tenant_name=client.tenant_info.tenant_name, url=client._api_server_url
+            )
+        return AuthServerServiceClient(
+            tenant_name=client.tenant_info.tenant_name,
+            url=client.tenant_info.auth_server_url,
+        )
 
     @abstractmethod
-    def refresh_token(self, token: Token, host: str = None) -> Token: ...
+    def refresh_token(self, token: Token, host: Optional[str] = None) -> Token: ...
 
     @abstractmethod
     def get_device_code(self) -> DeviceCode: ...
@@ -46,10 +45,11 @@ class AuthServiceClient(ABC):
 
 
 class ServiceFoundryServerAuthServiceClient(AuthServiceClient):
-    def __init__(self, base_url):
-        super().__init__(base_url)
+    def __init__(self, tenant_name: str, url: str):
+        super().__init__(tenant_name=tenant_name)
+        self._api_server_url = url
 
-    def refresh_token(self, token: Token, host: str = None) -> Token:
+    def refresh_token(self, token: Token, host: Optional[str] = None) -> Token:
         host_arg_str = f"--host {host}" if host else "--host HOST"
         if not token.refresh_token:
             # TODO: Add a way to propagate error messages without traceback to the output interface side
@@ -63,10 +63,11 @@ class ServiceFoundryServerAuthServiceClient(AuthServiceClient):
             "grantType": "refresh_token",
             "returnJWT": True,
         }
-        res = requests.post(url, json=data)
+        session = requests_retry_session(retries=2)
+        response = session.post(url, json=data)
         try:
-            res = request_handling(res)
-            return Token.parse_obj(res)
+            response_data = request_handling(response)
+            return Token.parse_obj(response_data)
         except BadRequestException as ex:
             raise Exception(
                 f"Unable to resume login session. Please log in again using `tfy login {host_arg_str} --relogin`"
@@ -75,9 +76,10 @@ class ServiceFoundryServerAuthServiceClient(AuthServiceClient):
     def get_device_code(self) -> DeviceCode:
         url = f"{self._api_server_url}/{VERSION_PREFIX}/oauth2/device-authorize"
         data = {"tenantName": self._tenant_name}
-        res = requests.post(url, json=data)
-        res = request_handling(res)
-        return DeviceCode.parse_obj(res)
+        session = requests_retry_session(retries=2)
+        response = session.post(url, json=data)
+        response_data = request_handling(response)
+        return DeviceCode.parse_obj(response_data)
 
     def get_token_from_device_code(
         self, device_code: str, timeout: float = 60, poll_interval_seconds: int = 1
@@ -91,7 +93,6 @@ class ServiceFoundryServerAuthServiceClient(AuthServiceClient):
             "grantType": "device_code",
             "returnJWT": True,
         }
-        response = requests.post(url=url, json=data)
         start_time = time.monotonic()
 
         for response in poll_for_function(
@@ -116,10 +117,12 @@ class ServiceFoundryServerAuthServiceClient(AuthServiceClient):
 
 
 class AuthServerServiceClient(AuthServiceClient):
-    def __init__(self, base_url):
-        super().__init__(base_url)
+    def __init__(self, tenant_name: str, url: str):
+        super().__init__(tenant_name=tenant_name)
+
+        self._auth_server_url = url
 
-    def refresh_token(self, token: Token, host: str = None) -> Token:
+    def refresh_token(self, token: Token, host: Optional[str] = None) -> Token:
         host_arg_str = f"--host {host}" if host else "--host HOST"
         if not token.refresh_token:
             # TODO: Add a way to propagate error messages without traceback to the output interface side
@@ -131,10 +134,11 @@ class AuthServerServiceClient(AuthServiceClient):
             "tenantName": token.tenant_name,
             "refreshToken": token.refresh_token,
         }
-        res = requests.post(url, json=data)
+        session = requests_retry_session(retries=2)
+        response = session.post(url, json=data)
         try:
-            res = request_handling(res)
-            return Token.parse_obj(res)
+            response_data = request_handling(response)
+            return Token.parse_obj(response_data)
         except BadRequestException as ex:
             raise Exception(
                 f"Unable to resume login session. Please log in again using `tfy login {host_arg_str} --relogin`"
@@ -143,22 +147,28 @@ class AuthServerServiceClient(AuthServiceClient):
     def get_device_code(self) -> DeviceCode:
         url = f"{self._auth_server_url}/api/{VERSION_PREFIX}/oauth/device"
         data = {"tenantName": self._tenant_name}
-        res = requests.post(url, json=data)
-        res = request_handling(res)
+        session = requests_retry_session(retries=2)
+        response = session.post(url, json=data)
+        response_data = request_handling(response)
         # TODO: temporary cleanup of incorrect attributes
-        res = {"userCode": res.get("userCode"), "deviceCode": res.get("deviceCode")}
-        return DeviceCode.parse_obj(res)
+        return DeviceCode.parse_obj(
+            {
+                "userCode": response_data.get("userCode"),
+                "deviceCode": response_data.get("deviceCode"),
+            }
+        )
 
     def get_token_from_device_code(
         self, device_code: str, timeout: float = 60, poll_interval_seconds: int = 1
     ) -> Token:
+        timeout = timeout or 60
+        poll_interval_seconds = poll_interval_seconds or 1
         url = f"{self._auth_server_url}/api/{VERSION_PREFIX}/oauth/device/token"
         data = {
             "tenantName": self._tenant_name,
             "deviceCode": device_code,
         }
         start_time = time.monotonic()
-        poll_interval_seconds = 1
 
         for response in poll_for_function(
             requests.post, poll_after_secs=poll_interval_seconds, url=url, json=data

truefoundry-0.4.0/truefoundry/common/constants.py

@@ -0,0 +1,12 @@
+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"
+SERVICEFOUNDRY_SERVER_URL_ENV_KEY = "SERVICEFOUNDRY_SERVER_URL"
+
+API_SERVER_RELATIVE_PATH = "api/svc"
+VERSION_PREFIX = "v1"
+SERVICEFOUNDRY_CLIENT_MAX_RETRIES = 2

{truefoundry-0.3.4/truefoundry/deploy/lib/auth → truefoundry-0.4.0/truefoundry/common}/credential_file_manager.py

@@ -8,14 +8,14 @@ from typing import Optional
 
 from filelock import FileLock, Timeout
 
-from truefoundry.deploy.lib.const import CREDENTIAL_FILEPATH
-from truefoundry.deploy.lib.model.entity import CredentialsFileContent
+from truefoundry.common.constants import CREDENTIAL_FILEPATH
+from truefoundry.common.entities import CredentialsFileContent
 from truefoundry.logger import logger
 
 
 def _ensure_lock_taken(method):
     @wraps(method)
-    def lock_guard(self, *method_args, **method_kwargs):
+    def lock_guard(self: "CredentialsFileManager", *method_args, **method_kwargs):
         if not self.lock_taken():
             raise Exception(
                 "Trying to write to credential file without using with block"
@@ -55,17 +55,17 @@ class CredentialsFileManager:
 
     def __enter__(self) -> CredentialsFileManager:
         # The lock objects are recursive locks, which means that once acquired, they will not block on successive lock requests:
-        lock_aquired = CRED_FILE_THREAD_LOCK.acquire(timeout=self._lock_timeout)
-        if not lock_aquired:
+        lock_acquired = CRED_FILE_THREAD_LOCK.acquire(timeout=self._lock_timeout)
+        if not lock_acquired:
             raise Exception(
-                "Could not aquire CRED_FILE_THREAD_LOCK"
+                "Could not acquire CRED_FILE_THREAD_LOCK"
                 f" in {self._lock_timeout} seconds"
             )
         try:
             self._file_lock.acquire(timeout=self._lock_timeout)
         except Timeout as ex:
             raise Exception(
-                f"Failed to aquire lock on credential file within {self._lock_timeout} seconds.\n"
+                f"Failed to acquire lock on credential file within {self._lock_timeout} seconds.\n"
                 "Is any other process trying to login?"
             ) from ex
         logger.debug("Acquired file and thread lock to access credential file")

{truefoundry-0.3.4/truefoundry/deploy/lib/auth → truefoundry-0.4.0/truefoundry/common}/credential_provider.py

@@ -2,14 +2,11 @@ import os
 import threading
 from abc import ABC, abstractmethod
 
-from truefoundry.deploy.lib.auth.auth_service_client import AuthServiceClient
-from truefoundry.deploy.lib.auth.credential_file_manager import (
-    CredentialsFileContent,
-    CredentialsFileManager,
-)
-from truefoundry.deploy.lib.clients.utils import resolve_base_url
-from truefoundry.deploy.lib.const import API_KEY_ENV_NAME
-from truefoundry.deploy.lib.model.entity import Token
+from truefoundry.common.auth_service_client import AuthServiceClient
+from truefoundry.common.constants import 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_base_url
 from truefoundry.logger import logger
 
 TOKEN_REFRESH_LOCK = threading.RLock()
@@ -31,31 +28,21 @@ class CredentialProvider(ABC):
 
 class EnvCredentialProvider(CredentialProvider):
     def __init__(self) -> None:
-        from truefoundry.deploy.lib.clients.servicefoundry_client import (
-            ServiceFoundryServiceClient,
-        )
-
         logger.debug("Using env var credential provider")
-        api_key = os.getenv(API_KEY_ENV_NAME)
+        api_key = os.getenv(TFY_API_KEY_ENV_KEY)
         if not api_key:
             raise Exception(
-                f"Value of {API_KEY_ENV_NAME} env var should be non-empty string"
+                f"Value of {TFY_API_KEY_ENV_KEY} env var should be non-empty string"
             )
         # TODO: Read host from cred file as well.
         base_url = resolve_base_url().strip("/")
         self._host = base_url
         self._auth_service = AuthServiceClient.from_base_url(base_url=base_url)
-
-        servicefoundry_client = ServiceFoundryServiceClient(
-            init_session=False, base_url=base_url
-        )
-        self._token: Token = servicefoundry_client.get_token_from_api_key(
-            api_key=api_key
-        )
+        self._token: Token = Token(access_token=api_key, refresh_token=None)  # type: ignore[call-arg]
 
     @staticmethod
     def can_provide() -> bool:
-        return API_KEY_ENV_NAME in os.environ
+        return TFY_API_KEY_ENV_KEY in os.environ
 
     @property
     def token(self) -> Token:
@@ -123,7 +110,7 @@ class FileCredentialProvider(CredentialProvider):
                     return self.token
 
                 raise Exception(
-                    "Credentials on disk changed while mlfoundry was running."
+                    "Credentials on disk changed while truefoundry was running."
                 )
 
     @property