zenml-nightly 0.66.0.dev20240910__py3-none-any.whl → 0.66.0.dev20240913__py3-none-any.whl

zenml/VERSION

@@ -1 +1 @@
-0.66.0.dev20240910
+0.66.0.dev20240913

zenml/cli/service_accounts.py

@@ -35,6 +35,7 @@ def _create_api_key(
     name: str,
     description: Optional[str],
     set_key: bool = False,
+    output_file: Optional[str] = None,
 ) -> None:
     """Create an API key.
 
@@ -44,6 +45,7 @@ def _create_api_key(
         name: Name of the API key
         description: The API key description.
         set_key: Configure the local client with the generated key.
+        output_file: Output file to write the API key to.
     """
     client = Client()
     zen_store = client.zen_store
@@ -74,13 +76,19 @@ def _create_api_key(
             )
             return
 
-    cli_utils.declare(
-        f"The API key value is: '{api_key.key}'\nPlease store it safely as "
-        "it will not be shown again.\nTo configure a ZenML client to use "
-        "this API key, run:\n\n"
-        f"zenml connect --url {zen_store.config.url} --api-key \\\n"
-        f"    '{api_key.key}'\n"
-    )
+    if output_file and api_key.key:
+        with open(output_file, "w") as f:
+            f.write(api_key.key)
+
+        cli_utils.declare(f"Wrote API key value to {output_file}")
+    else:
+        cli_utils.declare(
+            f"The API key value is: '{api_key.key}'\nPlease store it safely as "
+            "it will not be shown again.\nTo configure a ZenML client to use "
+            "this API key, run:\n\n"
+            f"zenml connect --url {zen_store.config.url} --api-key \\\n"
+            f"    '{api_key.key}'\n"
+        )
 
 
 @cli.group(cls=TagGroup, tag=CliCategories.IDENTITY_AND_SECURITY)
@@ -111,11 +119,18 @@ def service_account() -> None:
     help=("Configure the local client to use the generated API key."),
     is_flag=True,
 )
+@click.option(
+    "--output-file",
+    type=str,
+    required=False,
+    help="File to write the API key to.",
+)
 def create_service_account(
     service_account_name: str,
     description: str = "",
     create_api_key: bool = True,
     set_api_key: bool = False,
+    output_file: Optional[str] = None,
 ) -> None:
     """Create a new service account.
 
@@ -124,6 +139,7 @@ def create_service_account(
         description: The API key description.
         create_api_key: Create an API key for the service account.
         set_api_key: Configure the local client to use the generated API key.
+        output_file: Output file to write the API key to.
     """
     client = Client()
     try:
@@ -142,6 +158,7 @@ def create_service_account(
             name="default",
             description="Default API key.",
             set_key=set_api_key,
+            output_file=output_file,
         )
 
 
@@ -302,12 +319,19 @@ def api_key(
     is_flag=True,
     help="Configure the local client with the generated key.",
 )
+@click.option(
+    "--output-file",
+    type=str,
+    required=False,
+    help="File to write the API key to.",
+)
 @click.pass_obj
 def create_api_key(
     service_account_name_or_id: str,
     name: str,
     description: Optional[str],
     set_key: bool = False,
+    output_file: Optional[str] = None,
 ) -> None:
     """Create an API key.
 
@@ -317,12 +341,14 @@ def create_api_key(
         name: Name of the API key
         description: The API key description.
         set_key: Configure the local client with the generated key.
+        output_file: Output file to write the API key to.
     """
     _create_api_key(
         service_account_name_or_id=service_account_name_or_id,
         name=name,
         description=description,
         set_key=set_key,
+        output_file=output_file,
     )
 
 
@@ -450,12 +476,19 @@ def update_api_key(
     is_flag=True,
     help="Configure the local client with the generated key.",
 )
+@click.option(
+    "--output-file",
+    type=str,
+    required=False,
+    help="File to write the API key to.",
+)
 @click.pass_obj
 def rotate_api_key(
     service_account_name_or_id: str,
     name_or_id: str,
     retain: int = 0,
     set_key: bool = False,
+    output_file: Optional[str] = None,
 ) -> None:
     """Rotate an API key.
 
@@ -466,6 +499,7 @@ def rotate_api_key(
         retain: Number of minutes for which the previous key is still valid
             after it has been rotated.
         set_key: Configure the local client with the newly generated key.
+        output_file: Output file to write the API key to.
     """
     client = Client()
     zen_store = client.zen_store
@@ -499,13 +533,19 @@ def rotate_api_key(
             )
             return
 
-    cli_utils.declare(
-        f"The new API key value is: '{api_key.key}'\nPlease store it "
-        "safely as it will not be shown again.\nTo configure a ZenML "
-        "client to use this API key, run:\n\n"
-        f"zenml connect --url {zen_store.config.url} --api-key \\\n"
-        f"    '{api_key.key}'\n"
-    )
+    if output_file and api_key.key:
+        with open(output_file, "w") as f:
+            f.write(api_key.key)
+
+        cli_utils.declare(f"Wrote API key value to {output_file}")
+    else:
+        cli_utils.declare(
+            f"The new API key value is: '{api_key.key}'\nPlease store it "
+            "safely as it will not be shown again.\nTo configure a ZenML "
+            "client to use this API key, run:\n\n"
+            f"zenml connect --url {zen_store.config.url} --api-key \\\n"
+            f"    '{api_key.key}'\n"
+        )
 
 
 @api_key.command("delete")

zenml/stack/stack_component.py

@@ -421,23 +421,56 @@ class StackComponent:
         else:
             user_id = None
 
-        return flavor.implementation_class(
-            user=user_id,
-            workspace=component_model.workspace.id,
-            name=component_model.name,
-            id=component_model.id,
-            config=configuration,
-            labels=component_model.labels,
-            flavor=component_model.flavor,
-            type=component_model.type,
-            created=component_model.created,
-            updated=component_model.updated,
-            connector_requirements=flavor.service_connector_requirements,
-            connector=component_model.connector.id
-            if component_model.connector
-            else None,
-            connector_resource_id=component_model.connector_resource_id,
-        )
+        try:
+            return flavor.implementation_class(
+                user=user_id,
+                workspace=component_model.workspace.id,
+                name=component_model.name,
+                id=component_model.id,
+                config=configuration,
+                labels=component_model.labels,
+                flavor=component_model.flavor,
+                type=component_model.type,
+                created=component_model.created,
+                updated=component_model.updated,
+                connector_requirements=flavor.service_connector_requirements,
+                connector=component_model.connector.id
+                if component_model.connector
+                else None,
+                connector_resource_id=component_model.connector_resource_id,
+            )
+        except ImportError as e:
+            from zenml.integrations.registry import integration_registry
+
+            integration_requirements = " ".join(
+                integration_registry.select_integration_requirements(
+                    flavor_model.integration
+                )
+            )
+
+            if integration_registry.is_installed(flavor_model.integration):
+                raise ImportError(
+                    f"{e}\n\n"
+                    f"Something went wrong while trying to import from the "
+                    f"`{flavor_model.integration}` integration. Please make "
+                    "sure that all its requirements are installed properly by "
+                    "reinstalling the integration either through our CLI: "
+                    f"`zenml integration install {flavor_model.integration} "
+                    "-y` or by manually installing its requirements: "
+                    f"`pip install {integration_requirements}`. If the error"
+                    "persists, please contact the ZenML team."
+                ) from e
+            else:
+                raise ImportError(
+                    f"{e}\n\n"
+                    f"The `{flavor_model.integration}` integration that you "
+                    "are trying to use is not installed in your current "
+                    "environment. Please make sure that it is installed by "
+                    "either using our CLI: `zenml integration install "
+                    f"{flavor_model.integration}` or by manually installing "
+                    f"its requirements: `pip install "
+                    f"{integration_requirements}`"
+                ) from e
 
     @property
     def config(self) -> StackComponentConfig:

{zenml_nightly-0.66.0.dev20240910.dist-info → zenml_nightly-0.66.0.dev20240913.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: zenml-nightly
-Version: 0.66.0.dev20240910
+Version: 0.66.0.dev20240913
 Summary: ZenML: Write production-ready ML code.
 Home-page: https://zenml.io
 License: Apache-2.0

{zenml_nightly-0.66.0.dev20240910.dist-info → zenml_nightly-0.66.0.dev20240913.dist-info}/RECORD

@@ -6,7 +6,7 @@ RELEASE_NOTES.md,sha256=GjuHlUd-5AcepXC8ETMkIRMxKVQtBekiuY_CW0oP4XQ,365106
 ROADMAP.md,sha256=hiLSmr16BH8Dfx7SaQM4JcXCGCVl6mFZPFAwJeDTrJU,407
 SECURITY.md,sha256=9DepA8y03yvCZLHEfcXLTDH4lUyKHquAdukBsccNN7c,682
 zenml/README.md,sha256=827dekbOWAs1BpW7VF1a4d7EbwPbjwccX-2zdXBENZo,1777
-zenml/VERSION,sha256=IIehTiT1H_PJFlSTpiWfIJ2bnVYUB7OQ3S6LYuDUvKI,19
+zenml/VERSION,sha256=ROKzFghy4O4DnUbwD7E-q2OY0RGQuY1sP2kCiGOOyrI,19
 zenml/__init__.py,sha256=X_JnjAYTBG2ZYgcM2_FDPssygdj7z7VTAmU9HEi17p0,2306
 zenml/actions/__init__.py,sha256=mrt6wPo73iKRxK754_NqsGyJ3buW7RnVeIGXr1xEw8Y,681
 zenml/actions/base_action.py,sha256=UcaHev6BTuLDwuswnyaPjdA8AgUqB5xPZ-lRtuvf2FU,25553
@@ -51,7 +51,7 @@ zenml/cli/pipeline.py,sha256=8UO3ZjKZbaU3tiRLBsN-N3kK_i_bD-8Qbq_YpR8mstw,17442
 zenml/cli/secret.py,sha256=2WxDYg6iodnAVSbVJQvQEYWEgcy0oytcSBVnnyN0JOw,20135
 zenml/cli/served_model.py,sha256=lIPd6kOS9kfWFOsd3DSILtiYomINpmXrdpyP5vtQlj8,14604
 zenml/cli/server.py,sha256=N2cvu9C_z8CzwGu3a7P6fE2X8Se_rHEx30iq9nYMy9o,28682
-zenml/cli/service_accounts.py,sha256=MO8pVVjppdvIULFxpvQKlhfEDmRqvlH8aIbPTdNSHXU,16489
+zenml/cli/service_accounts.py,sha256=V6Oeomcs0tg-MIOUbYABqTcO4PsGCdM2WXJH576iqKM,17723
 zenml/cli/service_connectors.py,sha256=v6o-FpTEnDXYcQXczBELwfWPCcYBi9z7b1GC90Fvp30,74397
 zenml/cli/stack.py,sha256=vT-7Kac_Cf2g9XoIzwrxdQYgtkpuv_d-BpDvL-p9XUk,82862
 zenml/cli/stack_components.py,sha256=BxWsqeu5e2DtP5TMjzE98d8Lqfwe5o45TC2awjil5wY,74144
@@ -717,7 +717,7 @@ zenml/stack/authentication_mixin.py,sha256=sg7GkpB-Ao9Gsa7Po0jxMn-_mVYUB42etmspZ
 zenml/stack/flavor.py,sha256=rut-jFQFB6OGm25RRyMtYlJ3mm_6wrHYVI54Hmf9te4,9726
 zenml/stack/flavor_registry.py,sha256=IL0fRrxxQJ9YkCYCeADP7nwWEQo4XBElJY4owMjKGbQ,6108
 zenml/stack/stack.py,sha256=5pgR30_Xgj4Qi09zKYyleuaWwMhe0zvi4k7yF1r8_BM,38021
-zenml/stack/stack_component.py,sha256=DpWsgaqvPv9bJhACdMklL-Mjo453TjzQL4JVMzI3PQU,29887
+zenml/stack/stack_component.py,sha256=9sgAu2EGZ2yiEn_D0dq_kmytFmlt8spJ3iG0GlHIFVU,31634
 zenml/stack/stack_validator.py,sha256=hWbvvGIeWLj6NwSsF4GCc6RAxAWvxHXTcBZL9nJvcak,3111
 zenml/stack/utils.py,sha256=dcJsbLZf-hJYM9Ut8BfNzrWygmz6shHegqHPRVLoLWE,5840
 zenml/stack_deployments/__init__.py,sha256=-7593cQ_ZgRn774Ol-8AKXXQquIU4DSiaThVEr6TfWM,644
@@ -1399,8 +1399,8 @@ zenml/zen_stores/secrets_stores/sql_secrets_store.py,sha256=Bq1djrUP9saoD7vECjS7
 zenml/zen_stores/sql_zen_store.py,sha256=4z458K5ItHkAO0bEUAl6rWJDHKh6GS-sRrt2RPridnE,392435
 zenml/zen_stores/template_utils.py,sha256=UB_CBoMIfBequa8g3H7aTI6V_ke1SJr4IW1qwFCL1jc,8932
 zenml/zen_stores/zen_store_interface.py,sha256=kzR_i8vHjULld3MquSaMorcab8lJk1e9RZquw1VXjHY,93510
-zenml_nightly-0.66.0.dev20240910.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
-zenml_nightly-0.66.0.dev20240910.dist-info/METADATA,sha256=HXKwlZMRzcjTm7ssSRK6JClUhT80eOXfXTf9BmBvR-s,20961
-zenml_nightly-0.66.0.dev20240910.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-zenml_nightly-0.66.0.dev20240910.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
-zenml_nightly-0.66.0.dev20240910.dist-info/RECORD,,
+zenml_nightly-0.66.0.dev20240913.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
+zenml_nightly-0.66.0.dev20240913.dist-info/METADATA,sha256=OInj2XGlAGtuckrQScaRo6jDy68hTD-2ZV4kGNVUBhw,20961
+zenml_nightly-0.66.0.dev20240913.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+zenml_nightly-0.66.0.dev20240913.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
+zenml_nightly-0.66.0.dev20240913.dist-info/RECORD,,