zenml-nightly 0.75.0.dev20250318__py3-none-any.whl → 0.80.0.dev20250321__py3-none-any.whl

zenml/zen_server/deploy/daemon/daemon_zen_server.py

@@ -38,11 +38,11 @@ from zenml.constants import (
 from zenml.enums import AuthScheme, StoreType
 from zenml.logger import get_logger
 from zenml.models import ServerDeploymentType
+from zenml.models.v2.misc.service import ServiceType
 from zenml.services import (
     LocalDaemonService,
     LocalDaemonServiceConfig,
     LocalDaemonServiceEndpoint,
-    ServiceType,
 )
 from zenml.utils.io_utils import get_global_config_directory
 from zenml.zen_server.deploy.deployment import LocalServerDeploymentConfig

zenml/zen_server/deploy/deployment.py

@@ -17,8 +17,7 @@ from typing import Optional
 
 from pydantic import BaseModel, ConfigDict
 
-from zenml.enums import ServerProviderType
-from zenml.services.service_status import ServiceState
+from zenml.enums import ServerProviderType, ServiceState
 
 
 class LocalServerDeploymentConfig(BaseModel):

zenml/zen_server/deploy/docker/docker_zen_server.py

@@ -38,11 +38,11 @@ from zenml.constants import (
 from zenml.enums import AuthScheme, StoreType
 from zenml.logger import get_logger
 from zenml.models import ServerDeploymentType
+from zenml.models.v2.misc.service import ServiceType
 from zenml.services import (
     ContainerService,
     ContainerServiceConfig,
     ContainerServiceEndpoint,
-    ServiceType,
 )
 from zenml.services.container.container_service import (
     SERVICE_CONTAINER_GLOBAL_CONFIG_DIR,

zenml/zen_server/rbac/endpoint_utils.py

@@ -306,8 +306,7 @@ def verify_permissions_and_delete_entity(
     Returns:
         The deleted entity.
     """
-    # We don't need the hydrated version here
-    model = get_method(id, False)
+    model = get_method(id, True)
     verify_permission_for_model(model, action=Action.DELETE)
     delete_method(model.id, **delete_method_kwargs)
     delete_model_resource(model)

zenml/zen_server/routers/projects_endpoints.py

@@ -37,6 +37,9 @@ from zenml.models import (
 )
 from zenml.zen_server.auth import AuthContext, authorize
 from zenml.zen_server.exceptions import error_response
+from zenml.zen_server.feature_gate.endpoint_utils import (
+    report_decrement,
+)
 from zenml.zen_server.rbac.endpoint_utils import (
     verify_permissions_and_create_entity,
     verify_permissions_and_delete_entity,
@@ -51,6 +54,7 @@ from zenml.zen_server.rbac.utils import (
 from zenml.zen_server.utils import (
     handle_exceptions,
     make_dependable,
+    server_config,
     zen_store,
 )
 
@@ -225,11 +229,14 @@ def delete_project(
     Args:
         project_name_or_id: Name or ID of the project.
     """
-    verify_permissions_and_delete_entity(
+    project = verify_permissions_and_delete_entity(
         id=project_name_or_id,
         get_method=zen_store().get_project,
         delete_method=zen_store().delete_project,
     )
+    if server_config().feature_gate_enabled:
+        if ResourceType.PROJECT in server_config().reportable_resources:
+            report_decrement(ResourceType.PROJECT, resource_id=project.id)
 
 
 # TODO: kept for backwards compatibility only; to be removed after the migration
@@ -268,13 +275,17 @@ def get_project_statistics(
     run_filter = PipelineRunFilter(project=project.id)
     run_filter.configure_rbac(
         authenticated_user_id=user_id,
-        id=get_allowed_resource_ids(resource_type=ResourceType.PIPELINE_RUN),
+        id=get_allowed_resource_ids(
+            resource_type=ResourceType.PIPELINE_RUN, project_id=project.id
+        ),
     )
 
     pipeline_filter = PipelineFilter(project=project.id)
     pipeline_filter.configure_rbac(
         authenticated_user_id=user_id,
-        id=get_allowed_resource_ids(resource_type=ResourceType.PIPELINE),
+        id=get_allowed_resource_ids(
+            resource_type=ResourceType.PIPELINE, project_id=project.id
+        ),
     )
 
     return ProjectStatistics(

zenml/zen_server/utils.py

@@ -29,7 +29,6 @@ from typing import (
     Union,
     cast,
 )
-from urllib.parse import urlparse
 from uuid import UUID
 
 from pydantic import BaseModel, ValidationError
@@ -43,16 +42,11 @@ from zenml.constants import (
     INFO,
     VERSION_1,
 )
-from zenml.enums import StoreType
 from zenml.exceptions import IllegalOperationError, OAuthError
 from zenml.logger import get_logger
 from zenml.models.v2.base.scoped import ProjectScopedFilter
 from zenml.plugins.plugin_flavor_registry import PluginFlavorRegistry
 from zenml.zen_server.cache import MemoryCache
-from zenml.zen_server.deploy.deployment import (
-    LocalServerDeployment,
-)
-from zenml.zen_server.deploy.exceptions import ServerDeploymentNotFoundError
 from zenml.zen_server.exceptions import http_exception_from_error
 from zenml.zen_server.feature_gate.feature_gate_interface import (
     FeatureGateInterface,
@@ -66,6 +60,7 @@ from zenml.zen_stores.sql_zen_store import SqlZenStore
 if TYPE_CHECKING:
     from fastapi import Request
 
+
 logger = get_logger(__name__)
 
 _zen_store: Optional["SqlZenStore"] = None
@@ -98,6 +93,7 @@ def plugin_flavor_registry() -> PluginFlavorRegistry:
         The plugin flavor registry.
     """
     global _plugin_flavor_registry
+
     if _plugin_flavor_registry is None:
         _plugin_flavor_registry = PluginFlavorRegistry()
         _plugin_flavor_registry.initialize_plugins()
@@ -269,86 +265,6 @@ def server_config() -> ServerConfiguration:
     return _server_config
 
 
-def get_local_server() -> Optional["LocalServerDeployment"]:
-    """Get the active local server.
-
-    Call this function to retrieve the local server deployed on this machine.
-
-    Returns:
-        The local server deployment or None, if no local server deployment was
-        found.
-    """
-    from zenml.zen_server.deploy.deployer import LocalServerDeployer
-
-    deployer = LocalServerDeployer()
-    try:
-        return deployer.get_server()
-    except ServerDeploymentNotFoundError:
-        return None
-
-
-def connected_to_local_server() -> bool:
-    """Check if the client is connected to a local server.
-
-    Returns:
-        True if the client is connected to a local server, False otherwise.
-    """
-    from zenml.zen_server.deploy.deployer import LocalServerDeployer
-
-    deployer = LocalServerDeployer()
-    return deployer.is_connected_to_server()
-
-
-def show_dashboard(
-    local: bool = False,
-    ngrok_token: Optional[str] = None,
-) -> None:
-    """Show the ZenML dashboard.
-
-    Args:
-        local: Whether to show the dashboard for the local server or the
-            one for the active server.
-        ngrok_token: An ngrok auth token to use for exposing the ZenML
-            dashboard on a public domain. Primarily used for accessing the
-            dashboard in Colab.
-
-    Raises:
-        RuntimeError: If no server is connected.
-    """
-    from zenml.utils.dashboard_utils import show_dashboard
-    from zenml.utils.networking_utils import get_or_create_ngrok_tunnel
-
-    url: Optional[str] = None
-    if not local:
-        gc = GlobalConfiguration()
-        if gc.store_configuration.type == StoreType.REST:
-            url = gc.store_configuration.url
-
-    if not url:
-        # Else, check for local servers
-        server = get_local_server()
-        if server and server.status and server.status.url:
-            url = server.status.url
-
-    if not url:
-        raise RuntimeError(
-            "ZenML is not connected to any server right now. Please use "
-            "`zenml login` to connect to a server or spin up a new local server "
-            "via `zenml login --local`."
-        )
-
-    if ngrok_token:
-        parsed_url = urlparse(url)
-
-        ngrok_url = get_or_create_ngrok_tunnel(
-            ngrok_token=ngrok_token, port=parsed_url.port or 80
-        )
-        logger.debug(f"Tunneling dashboard from {url} to {ngrok_url}.")
-        url = ngrok_url
-
-    show_dashboard(url)
-
-
 F = TypeVar("F", bound=Callable[..., Any])
 
 

zenml/zen_stores/migrations/versions/0.80.0_release.py

@@ -0,0 +1,23 @@
+"""Release [0.80.0].
+
+Revision ID: 0.80.0
+Revises: 12eff0206201
+Create Date: 2025-03-19 10:07:36.959628
+
+"""
+
+# revision identifiers, used by Alembic.
+revision = "0.80.0"
+down_revision = "12eff0206201"
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    """Upgrade database schema and/or data, creating a new revision."""
+    pass
+
+
+def downgrade() -> None:
+    """Downgrade database schema and/or data back to the previous revision."""
+    pass

zenml/zen_stores/schemas/artifact_visualization_schemas.py

@@ -43,7 +43,7 @@ class ArtifactVisualizationSchema(BaseSchema, table=True):
     # Foreign Keys
     artifact_version_id: UUID = build_foreign_key_field(
         source=__tablename__,
-        target=ArtifactVersionSchema.__tablename__,  # type: ignore[has-type]
+        target=ArtifactVersionSchema.__tablename__,
         source_column="artifact_version_id",
         target_column="id",
         ondelete="CASCADE",

zenml/zen_stores/schemas/model_schemas.py

@@ -552,7 +552,7 @@ class ModelVersionArtifactSchema(BaseSchema, table=True):
     )
     artifact_version_id: UUID = build_foreign_key_field(
         source=__tablename__,
-        target=ArtifactVersionSchema.__tablename__,  # type: ignore[has-type]
+        target=ArtifactVersionSchema.__tablename__,
         source_column="artifact_version_id",
         target_column="id",
         ondelete="CASCADE",

zenml/zen_stores/schemas/service_schemas.py

@@ -88,7 +88,7 @@ class ServiceSchema(NamedSchema, table=True):
     )
     model_version_id: Optional[UUID] = build_foreign_key_field(
         source=__tablename__,
-        target=ModelVersionSchema.__tablename__,  # type: ignore[has-type]
+        target=ModelVersionSchema.__tablename__,
         source_column="model_version_id",
         target_column="id",
         ondelete="SET NULL",

zenml/zen_stores/schemas/step_run_schemas.py

@@ -112,7 +112,7 @@ class StepRunSchema(NamedSchema, RunMetadataInterface, table=True):
     )
     pipeline_run_id: UUID = build_foreign_key_field(
         source=__tablename__,
-        target=PipelineRunSchema.__tablename__,  # type: ignore[has-type]
+        target=PipelineRunSchema.__tablename__,
         source_column="pipeline_run_id",
         target_column="id",
         ondelete="CASCADE",

zenml/zen_stores/schemas/trigger_schemas.py

@@ -84,7 +84,7 @@ class TriggerSchema(NamedSchema, table=True):
 
     event_source_id: Optional[UUID] = build_foreign_key_field(
         source=__tablename__,
-        target=EventSourceSchema.__tablename__,  # type: ignore[has-type]
+        target=EventSourceSchema.__tablename__,
         source_column="event_source_id",
         target_column="id",
         # This won't happen because the SQL zen store prevents an event source

zenml/zen_stores/sql_zen_store.py

@@ -9330,6 +9330,11 @@ class SqlZenStore(BaseZenStore):
                 include_metadata=True, include_resources=True
             )
 
+            self._update_onboarding_state(
+                completed_steps={OnboardingStep.PROJECT_CREATED},
+                session=session,
+            )
+
         return project_model
 
     def get_project(

{zenml_nightly-0.75.0.dev20250318.dist-info → zenml_nightly-0.80.0.dev20250321.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: zenml-nightly
-Version: 0.75.0.dev20250318
+Version: 0.80.0.dev20250321
 Summary: ZenML: Write production-ready ML code.
 License: Apache-2.0
 Keywords: machine learning,production,pipeline,mlops,devops
@@ -531,7 +531,7 @@ the Apache License Version 2.0.
     <a href="https://github.com/zenml-io/zenml-projects">Projects Showcase</a>
     <br />
     <br />
-    🎉 Version 0.75.0 is out. Check out the release notes
+    🎉 Version 0.80.0 is out. Check out the release notes
     <a href="https://github.com/zenml-io/zenml/releases">here</a>.
     <br />
     🖥️ Download our VS Code Extension <a href="https://marketplace.visualstudio.com/items?itemName=ZenML.zenml-vscode">here</a>.