truefoundry 0.11.0__tar.gz → 0.11.1rc1__tar.gz

{truefoundry-0.11.0 → truefoundry-0.11.1rc1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: truefoundry
-Version: 0.11.0
+Version: 0.11.1rc1
 Summary: TrueFoundry CLI
 Author-email: TrueFoundry Team <abhishek@truefoundry.com>
 Requires-Python: <3.14,>=3.8.1
@@ -30,7 +30,7 @@ Requires-Dist: requirements-parser<0.12.0,>=0.11.0
 Requires-Dist: rich-click<2.0.0,>=1.2.1
 Requires-Dist: rich<14.0.0,>=13.7.1
 Requires-Dist: tqdm<5.0.0,>=4.0.0
-Requires-Dist: truefoundry-sdk<0.2.0,>=0.1.5
+Requires-Dist: truefoundry-sdk<0.2.0,>=0.1.7
 Requires-Dist: typing-extensions>=4.0
 Requires-Dist: urllib3<3,>=1.26.18
 Requires-Dist: yq<4.0.0,>=3.1.0

{truefoundry-0.11.0 → truefoundry-0.11.1rc1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "truefoundry"
-version = "0.11.0"  # This is a placeholder version, it will be replaced when releasing
+version = "0.11.1rc1"  # This is a placeholder version, it will be replaced when releasing
 requires-python = ">=3.8.1,<3.14"
 description = "TrueFoundry CLI"
 authors = [{ name = "TrueFoundry Team", email = "abhishek@truefoundry.com" }]
@@ -31,7 +31,7 @@ dependencies = [
   "rich-click>=1.2.1,<2.0.0",
   "rich>=13.7.1,<14.0.0",
   "tqdm>=4.0.0,<5.0.0",
-  "truefoundry-sdk>=0.1.5,<0.2.0",
+  "truefoundry-sdk>=0.1.7,<0.2.0",
   "typing-extensions>=4.0",
   "urllib3>=1.26.18,<3",
   "yq>=3.1.0,<4.0.0",

{truefoundry-0.11.0 → truefoundry-0.11.1rc1}/truefoundry/ml/mlfoundry_api.py

@@ -46,7 +46,6 @@ from truefoundry.ml._autogen.client.exceptions import (
     ApiException,
     NotFoundException,
 )
-from truefoundry.ml.clients.servicefoundry_client import ServiceFoundryServiceClient
 from truefoundry.ml.enums import ModelFramework, ViewType
 from truefoundry.ml.exceptions import MlFoundryException
 from truefoundry.ml.internal_namespace import NAMESPACE
@@ -139,7 +138,8 @@ class MlFoundry:
         except (IndexError, NotFoundError):
             err_msg = (
                 f"ML Repo Does Not Exist for name: {ml_repo}. You may either "
-                "create it from the dashboard or using client.create_ml_repo('<ml_repo_name>')"
+                f"create it from the dashboard or using "
+                f"`client.create_ml_repo(ml_repo='{ml_repo}', storage_integration_fqn='<storage_integration_fqn>')`"
             )
             raise MlFoundryException(err_msg) from None
         except ApiError as e:
@@ -191,7 +191,12 @@ class MlFoundry:
 
             client = get_client()
 
-            client.create_ml_repo(ml_repo="my-repo")
+            client.create_ml_repo(
+                ml_repo="my-repo",
+                # This controls which bucket is used.
+                # You can get this from Platform > Integrations > Copy FQN of any Blob Storage type integration.
+                storage_integration_fqn="..."
+            )
             ```
         """
         _validate_ml_repo_name(ml_repo_name=name)
@@ -235,28 +240,11 @@ class MlFoundry:
                 raise MlFoundryException(err_msg) from e
             return
 
-        servicefoundry_client = ServiceFoundryServiceClient(
-            tfy_host=self._api_client.tfy_host,
-            token=self._api_client.access_token,
-        )
-
-        assert existing_ml_repo.storage_integration_id is not None
-        try:
-            existing_storage_integration = (
-                servicefoundry_client.get_integration_from_id(
-                    existing_ml_repo.storage_integration_id
-                )
-            )
-        except Exception as e:
-            raise MlFoundryException(
-                "Error in getting storage integration for ML Repo"
-            ) from e
-
-        if existing_storage_integration["fqn"] != storage_integration_fqn:
+        if existing_ml_repo.manifest.storage_integration_fqn != storage_integration_fqn:
             raise MlFoundryException(
                 f"ML Repo with same name already exists with storage integration:"
-                f"{existing_storage_integration['fqn']}. Cannot update the storage integration to: "
-                f"{storage_integration_fqn}"
+                f"{existing_ml_repo.manifest.storage_integration_fqn!r}. "
+                f"Cannot update the storage integration to: {storage_integration_fqn!r}"
             )
 
     def create_run(
@@ -562,7 +550,8 @@ class MlFoundry:
         except (IndexError, NotFoundError):
             raise MlFoundryException(
                 f"ML Repo with name {ml_repo} does not exist. "
-                "You may either create it from the dashboard or using client.create_ml_repo('<ml_repo_name>')"
+                f"You may either create it from the dashboard or using "
+                f"`client.create_ml_repo(ml_repo='{ml_repo}', storage_integration_fqn='<storage_integration_fqn>')`"
             ) from None
         except ApiError as e:
             raise MlFoundryException(
@@ -1202,7 +1191,12 @@ class MlFoundry:
             client = get_client()
             ml_repo = "sample-repo"
 
-            client.create_ml_repo(ml_repo=ml_repo)
+            client.create_ml_repo(
+                ml_repo=ml_repo,
+                # This controls which bucket is used.
+                # You can get this from Platform > Integrations > Copy FQN of any Blob Storage type integration.
+                storage_integration_fqn="..."
+            )
             client.log_artifact(
                 ml_repo=ml_repo,
                 name="hello-world-file",
@@ -1301,8 +1295,8 @@ class MlFoundry:
             client.create_ml_repo(  # This is only required once
                 ml_repo="my-classification-project",
                 # This controls which bucket is used.
-                # You can get this from Integrations > Blob Storage. `None` picks the default
-                storage_integration_fqn=None
+                # You can get this from Platform > Integrations > Copy FQN of any Blob Storage type integration.
+                storage_integration_fqn="..."
             )
             model_version = client.log_model(
                 ml_repo="my-classification-project",

truefoundry-0.11.0/truefoundry/ml/clients/servicefoundry_client.py

@@ -1,65 +0,0 @@
-import functools
-
-from truefoundry.common.constants import (
-    SERVICEFOUNDRY_CLIENT_MAX_RETRIES,
-    VERSION_PREFIX,
-)
-from truefoundry.common.exceptions import HttpRequestException
-from truefoundry.common.request_utils import (
-    http_request,
-    request_handling,
-    requests_retry_session,
-)
-from truefoundry.common.servicefoundry_client import (
-    ServiceFoundryServiceClient as BaseServiceFoundryServiceClient,
-)
-from truefoundry.common.utils import get_user_agent
-from truefoundry.ml.exceptions import MlFoundryException
-
-
-class ServiceFoundryServiceClient(BaseServiceFoundryServiceClient):
-    def __init__(self, tfy_host: str, token: str):
-        super().__init__(tfy_host=tfy_host)
-        self._token = token
-
-    @functools.cached_property
-    def _min_cli_version_required(self) -> str:
-        # TODO (chiragjn): read the mlfoundry min cli version from the config?
-        return self.python_sdk_config.truefoundry_cli_min_version
-
-    def get_integration_from_id(self, integration_id: str):
-        integration_id = integration_id or ""
-        session = requests_retry_session(retries=SERVICEFOUNDRY_CLIENT_MAX_RETRIES)
-        response = http_request(
-            method="get",
-            url=f"{self._api_server_url}/{VERSION_PREFIX}/provider-accounts/provider-integrations",
-            token=self._token,
-            timeout=3,
-            params={"id": integration_id, "type": "blob-storage"},
-            session=session,
-            headers={
-                "User-Agent": get_user_agent(),
-            },
-        )
-
-        try:
-            result = request_handling(response)
-            assert isinstance(result, dict)
-        except HttpRequestException as he:
-            raise MlFoundryException(
-                f"Failed to get storage integration from id: {integration_id}. Error: {he.message}",
-                status_code=he.status_code,
-            ) from None
-        except Exception as e:
-            raise MlFoundryException(
-                f"Failed to get storage integration from id: {integration_id}. Error: {str(e)}"
-            ) from None
-
-        data = result.get("data", result.get("providerAccounts"))
-        # TODO (chiragjn): Parse this using Pydantic
-        if data and len(data) > 0 and data[0]:
-            return data[0]
-        else:
-            raise MlFoundryException(
-                f"Invalid storage integration id: {integration_id}"
-            )