zenml-nightly 0.75.0.dev20250312__py3-none-any.whl → 0.75.0.dev20250314__py3-none-any.whl

zenml/cli/service_connectors.py

@@ -1131,7 +1131,6 @@ def describe_service_connector(
             )
 
         connector = connector_client.to_response_model(
-            workspace=client.active_workspace,
             user=client.active_user,
         )
     else:
@@ -1903,29 +1902,28 @@ def login_service_connector(
     "list-resources",
     help="""List all resources accessible by service connectors.
 
-This command can be used to list all resources that can be accessed by service
-connectors configured in your workspace. You can filter the list by connector
+This command can be used to list all resources that can be accessed by the
+currently registered service connectors. You can filter the list by connector
 type and/or resource type.
 
 Use this command to answer questions like:
 
 - show a list of all Kubernetes clusters that can be accessed by way of service
-connectors configured in my workspace
-- show a list of all connectors configured for my workspace along with all the
-resources they can access or the error state they are in, if any
+connectors
+- show a list of all connectors along with all the resources they can access or
+the error state they are in, if any
 
-NOTE: since this command exercises all service connectors in your workspace, it
-may take a while to complete.
+NOTE: since this command exercises all service connectors currently registered
+with ZenML, it may take a while to complete.
 
 Examples:
 
-- show a list of all S3 buckets that can be accessed by service connectors
-configured in your workspace:
+- show a list of all S3 buckets that can be accessed by service connectors:
 
     $ zenml service-connector list-resources --resource-type s3-bucket
 
-- show a list of all resources that the AWS connectors in your workspace can
-access:
+- show a list of all resources that the AWS connectors currently registered
+with ZenML can access:
 
     $ zenml service-connector list-resources --connector-type aws
     
@@ -1983,10 +1981,10 @@ def list_service_connector_resources(
     if not resource_type and not resource_id:
         cli_utils.warning(
             "Fetching all service connector resources can take a long time, "
-            "depending on the number of connectors configured in your "
-            "workspace. Consider using the '--connector-type', "
-            "'--resource-type' and '--resource-id' options to narrow down the "
-            "list of resources to fetch."
+            "depending on the number of connectors currently registered with "
+            "ZenML. Consider using the '--connector-type', '--resource-type' "
+            "and '--resource-id' options to narrow down the list of resources "
+            "to fetch."
         )
 
     with console.status(
@@ -2031,7 +2029,7 @@ def list_service_connector_resources(
 
     click.echo(
         f"The {resource_str} can be accessed by"
-        f"{connector_str} service connectors configured in your workspace:"
+        f"{connector_str} service connectors:"
     )
 
     cli_utils.print_service_connector_resource_table(

zenml/cli/stack.py

@@ -503,8 +503,6 @@ def register_stack(
         try:
             created_stack = client.zen_store.create_stack(
                 stack=StackRequest(
-                    user=client.active_user.id,
-                    workspace=client.active_workspace.id,
                     name=stack_name,
                     components=components,
                     service_connectors=[service_connector]
@@ -1914,7 +1912,6 @@ def _get_stack_component_info(
                 "Enter the subscription ID:"
             )
             config["resource_group"] = Prompt.ask("Enter the resource group:")
-            config["workspace"] = Prompt.ask("Enter the workspace name:")
         elif flavor == "vertex":
             config["location"] = query_region(
                 StackDeploymentProvider.GCP, "Vertex AI job"

zenml/cli/stack_components.py

@@ -1424,8 +1424,8 @@ def connect_stack_component_with_service_connector(
 
             cli_utils.error(
                 f"No compatible valid resources were found for the "
-                f"'{component_model.name}' {display_name} in your "
-                f"workspace. {additional_info}You can create a new "
+                f"'{component_model.name}' {display_name}. "
+                f"{additional_info}You can create a new "
                 "connector using the 'zenml service-connector register' "
                 "command or list the compatible resources using the "
                 f"'zenml service-connector list-resources{command_args}' "

zenml/cli/tag.py

@@ -26,8 +26,6 @@ from zenml.exceptions import EntityExistsError
 from zenml.logger import get_logger
 from zenml.models import (
     TagFilter,
-    TagRequest,
-    TagUpdate,
 )
 from zenml.utils.dict_utils import remove_none_values
 
@@ -47,7 +45,7 @@ def list_tags(**kwargs: Any) -> None:
     Args:
         **kwargs: Keyword arguments to filter models.
     """
-    tags = Client().list_tags(TagFilter(**kwargs))
+    tags = Client().list_tags(**kwargs)
 
     if not tags:
         cli_utils.declare("No tags found.")
@@ -83,7 +81,7 @@ def register_tag(name: str, color: Optional[ColorVariants]) -> None:
     """
     request_dict = remove_none_values(dict(name=name, color=color))
     try:
-        tag = Client().create_tag(TagRequest(**request_dict))
+        tag = Client().create_tag(**request_dict)
     except (EntityExistsError, ValueError) as e:
         cli_utils.error(str(e))
 
@@ -126,7 +124,7 @@ def update_tag(
 
     tag = Client().update_tag(
         tag_name_or_id=tag_name_or_id,
-        tag_update_model=TagUpdate(**update_dict),
+        **update_dict,
     )
 
     cli_utils.print_pydantic_models(

zenml/cli/utils.py

@@ -81,6 +81,7 @@ from zenml.stack.flavor import Flavor
 from zenml.stack.stack_component import StackComponentConfig
 from zenml.utils import secret_utils
 from zenml.utils.time_utils import expires_in
+from zenml.utils.typing_utils import get_origin, is_union
 
 if TYPE_CHECKING:
     from uuid import UUID
@@ -1778,7 +1779,6 @@ def print_service_connector_configuration(
             "AUTH METHOD": connector.auth_method,
             "RESOURCE TYPES": ", ".join(connector.emojified_resource_types),
             "RESOURCE NAME": connector.resource_id or "<multiple>",
-            "SECRET ID": connector.secret_id or "",
             "SESSION DURATION": expiration,
             "EXPIRES IN": (
                 expires_in(
@@ -1795,7 +1795,6 @@ def print_service_connector_configuration(
                 else "N/A"
             ),
             "OWNER": user_name,
-            "WORKSPACE": connector.workspace.name,
             "CREATED_AT": connector.created,
             "UPDATED_AT": connector.updated,
         }
@@ -2161,9 +2160,6 @@ def print_debug_stack() -> None:
     console.print(f"ID: {str(stack.id)}")
     if stack.user and stack.user.name and stack.user.id:  # mypy check
         console.print(f"User: {stack.user.name} / {str(stack.user.id)}")
-    console.print(
-        f"Workspace: {stack.workspace.name} / {str(stack.workspace.id)}"
-    )
 
     for component_type, components in stack.components.items():
         component = components[0]
@@ -2190,9 +2186,6 @@ def print_debug_stack() -> None:
             console.print(
                 f"User: {component_response.user.name} / {str(component_response.user.id)}"
             )
-        console.print(
-            f"Workspace: {component_response.workspace.name} / {str(component_response.workspace.id)}"
-        )
 
 
 def _component_display_name(
@@ -2275,22 +2268,13 @@ def print_pipeline_runs_table(
     print_table(runs_dicts)
 
 
-def warn_unsupported_non_default_workspace() -> None:
-    """Warning for unsupported non-default workspace."""
-    from zenml.constants import (
-        ENV_ZENML_DISABLE_WORKSPACE_WARNINGS,
-        handle_bool_env_var,
-    )
-
-    disable_warnings = handle_bool_env_var(
-        ENV_ZENML_DISABLE_WORKSPACE_WARNINGS, False
-    )
-    if not disable_warnings:
+def check_zenml_pro_project_availability() -> None:
+    """Check if the ZenML Pro project feature is available."""
+    client = Client()
+    if not client.zen_store.get_store_info().is_pro_server():
         warning(
-            "Currently the concept of `workspace` is not supported "
-            "within the Dashboard. The Project functionality will be "
-            "completed in the coming weeks. For the time being it "
-            "is recommended to stay within the `default` workspace."
+            "The ZenML projects feature is available only on ZenML Pro. "
+            "Please visit https://zenml.io/pro to learn more."
         )
 
 
@@ -2402,6 +2386,23 @@ def create_data_type_help_text(
         return f"{field}"
 
 
+def _is_list_field(field_info: Any) -> bool:
+    """Check if a field is a list field.
+
+    Args:
+        field_info: The field info to check.
+
+    Returns:
+        True if the field is a list field, False otherwise.
+    """
+    field_type = field_info.annotation
+    origin = get_origin(field_type)
+    return origin is list or (
+        is_union(origin)
+        and any(get_origin(arg) is list for arg in field_type.__args__)
+    )
+
+
 def list_options(filter_model: Type[BaseFilter]) -> Callable[[F], F]:
     """Create a decorator to generate the correct list of filter parameters.
 
@@ -2430,6 +2431,7 @@ def list_options(filter_model: Type[BaseFilter]) -> Callable[[F], F]:
                         type=str,
                         default=v.default,
                         required=False,
+                        multiple=_is_list_field(v),
                         help=create_filter_help_text(filter_model, k),
                     )
                 )