zenml-nightly 0.75.0.dev20250313__py3-none-any.whl → 0.75.0.dev20250315__py3-none-any.whl

zenml/zen_server/routers/code_repositories_endpoints.py

@@ -36,9 +36,7 @@ from zenml.zen_server.rbac.endpoint_utils import (
     verify_permissions_and_update_entity,
 )
 from zenml.zen_server.rbac.models import ResourceType
-from zenml.zen_server.routers.workspaces_endpoints import (
-    router as workspace_router,
-)
+from zenml.zen_server.routers.projects_endpoints import workspace_router
 from zenml.zen_server.utils import (
     handle_exceptions,
     make_dependable,
@@ -59,7 +57,7 @@ router = APIRouter(
 # TODO: the workspace scoped endpoint is only kept for dashboard compatibility
 # and can be removed after the migration
 @workspace_router.post(
-    "/{workspace_name_or_id}" + CODE_REPOSITORIES,
+    "/{project_name_or_id}" + CODE_REPOSITORIES,
     responses={401: error_response, 409: error_response, 422: error_response},
     deprecated=True,
     tags=["code_repositories"],
@@ -67,21 +65,21 @@ router = APIRouter(
 @handle_exceptions
 def create_code_repository(
     code_repository: CodeRepositoryRequest,
-    workspace_name_or_id: Optional[Union[str, UUID]] = None,
+    project_name_or_id: Optional[Union[str, UUID]] = None,
     _: AuthContext = Security(authorize),
 ) -> CodeRepositoryResponse:
     """Creates a code repository.
 
     Args:
         code_repository: Code repository to create.
-        workspace_name_or_id: Optional name or ID of the workspace.
+        project_name_or_id: Optional name or ID of the project.
 
     Returns:
         The created code repository.
     """
-    if workspace_name_or_id:
-        workspace = zen_store().get_workspace(workspace_name_or_id)
-        code_repository.workspace = workspace.id
+    if project_name_or_id:
+        project = zen_store().get_project(project_name_or_id)
+        code_repository.project = project.id
 
     return verify_permissions_and_create_entity(
         request_model=code_repository,
@@ -96,7 +94,7 @@ def create_code_repository(
 # TODO: the workspace scoped endpoint is only kept for dashboard compatibility
 # and can be removed after the migration
 @workspace_router.get(
-    "/{workspace_name_or_id}" + CODE_REPOSITORIES,
+    "/{project_name_or_id}" + CODE_REPOSITORIES,
     responses={401: error_response, 404: error_response, 422: error_response},
     deprecated=True,
     tags=["code_repositories"],
@@ -106,7 +104,7 @@ def list_code_repositories(
     filter_model: CodeRepositoryFilter = Depends(
         make_dependable(CodeRepositoryFilter)
     ),
-    workspace_name_or_id: Optional[Union[str, UUID]] = None,
+    project_name_or_id: Optional[Union[str, UUID]] = None,
     hydrate: bool = False,
     _: AuthContext = Security(authorize),
 ) -> Page[CodeRepositoryResponse]:
@@ -115,15 +113,15 @@ def list_code_repositories(
     Args:
         filter_model: Filter model used for pagination, sorting,
             filtering.
-        workspace_name_or_id: Optional name or ID of the workspace.
+        project_name_or_id: Optional name or ID of the project.
         hydrate: Flag deciding whether to hydrate the output model(s)
             by including metadata fields in the response.
 
     Returns:
         Page of code repository objects.
     """
-    if workspace_name_or_id:
-        filter_model.workspace = workspace_name_or_id
+    if project_name_or_id:
+        filter_model.project = project_name_or_id
 
     return verify_permissions_and_list_entities(
         filter_model=filter_model,

zenml/zen_server/routers/model_versions_endpoints.py

@@ -56,13 +56,11 @@ from zenml.zen_server.rbac.utils import (
 from zenml.zen_server.routers.models_endpoints import (
     router as model_router,
 )
-from zenml.zen_server.routers.workspaces_endpoints import (
-    router as workspace_router,
-)
+from zenml.zen_server.routers.projects_endpoints import workspace_router
 from zenml.zen_server.utils import (
     handle_exceptions,
     make_dependable,
-    set_filter_workspace_scope,
+    set_filter_project_scope,
     zen_store,
 )
 
@@ -85,7 +83,7 @@ router = APIRouter(
 # TODO: the workspace scoped endpoint is only kept for dashboard compatibility
 # and can be removed after the migration
 @workspace_router.post(
-    "/{workspace_name_or_id}" + MODELS + "/{model_id}" + MODEL_VERSIONS,
+    "/{project_name_or_id}" + MODELS + "/{model_id}" + MODEL_VERSIONS,
     responses={401: error_response, 409: error_response, 422: error_response},
     deprecated=True,
     tags=["model_versions"],
@@ -94,7 +92,7 @@ router = APIRouter(
 def create_model_version(
     model_version: ModelVersionRequest,
     model_id: Optional[UUID] = None,
-    workspace_name_or_id: Optional[Union[str, UUID]] = None,
+    project_name_or_id: Optional[Union[str, UUID]] = None,
     _: AuthContext = Security(authorize),
 ) -> ModelVersionResponse:
     """Creates a model version.
@@ -102,14 +100,14 @@ def create_model_version(
     Args:
         model_version: Model version to create.
         model_id: Optional ID of the model.
-        workspace_name_or_id: Optional name or ID of the workspace.
+        project_name_or_id: Optional name or ID of the project.
 
     Returns:
         The created model version.
     """
-    if workspace_name_or_id:
-        workspace = zen_store().get_workspace(workspace_name_or_id)
-        model_version.workspace = workspace.id
+    if project_name_or_id:
+        project = zen_store().get_project(project_name_or_id)
+        model_version.project = project.id
 
     if model_id:
         model_version.model = model_id
@@ -159,14 +157,14 @@ def list_model_versions(
     if not model_version_filter_model.model:
         raise ValueError("Model missing from the filter")
 
-    # A workspace scoped request must always be scoped to a specific
-    # workspace. This is required for the RBAC check to work.
-    set_filter_workspace_scope(model_version_filter_model)
-    assert isinstance(model_version_filter_model.workspace, UUID)
+    # A project scoped request must always be scoped to a specific
+    # project. This is required for the RBAC check to work.
+    set_filter_project_scope(model_version_filter_model)
+    assert isinstance(model_version_filter_model.project, UUID)
 
     model = zen_store().get_model_by_name_or_id(
         model_version_filter_model.model,
-        workspace=model_version_filter_model.workspace,
+        project=model_version_filter_model.project,
     )
 
     # Check read permissions on the model

zenml/zen_server/routers/models_endpoints.py

@@ -41,9 +41,7 @@ from zenml.zen_server.rbac.endpoint_utils import (
     verify_permissions_and_update_entity,
 )
 from zenml.zen_server.rbac.models import ResourceType
-from zenml.zen_server.routers.workspaces_endpoints import (
-    router as workspace_router,
-)
+from zenml.zen_server.routers.projects_endpoints import workspace_router
 from zenml.zen_server.utils import (
     handle_exceptions,
     make_dependable,
@@ -69,7 +67,7 @@ router = APIRouter(
 # TODO: the workspace scoped endpoint is only kept for dashboard compatibility
 # and can be removed after the migration
 @workspace_router.post(
-    "/{workspace_name_or_id}" + MODELS,
+    "/{project_name_or_id}" + MODELS,
     responses={401: error_response, 409: error_response, 422: error_response},
     deprecated=True,
     tags=["models"],
@@ -77,21 +75,21 @@ router = APIRouter(
 @handle_exceptions
 def create_model(
     model: ModelRequest,
-    workspace_name_or_id: Optional[Union[str, UUID]] = None,
+    project_name_or_id: Optional[Union[str, UUID]] = None,
     _: AuthContext = Security(authorize),
 ) -> ModelResponse:
     """Creates a model.
 
     Args:
         model: Model to create.
-        workspace_name_or_id: Optional name or ID of the workspace.
+        project_name_or_id: Optional name or ID of the project.
 
     Returns:
         The created model.
     """
-    if workspace_name_or_id:
-        workspace = zen_store().get_workspace(workspace_name_or_id)
-        model.workspace = workspace.id
+    if project_name_or_id:
+        project = zen_store().get_project(project_name_or_id)
+        model.project = project.id
 
     return verify_permissions_and_create_entity(
         request_model=model,

zenml/zen_server/routers/pipeline_builds_endpoints.py

@@ -34,9 +34,7 @@ from zenml.zen_server.rbac.endpoint_utils import (
     verify_permissions_and_list_entities,
 )
 from zenml.zen_server.rbac.models import ResourceType
-from zenml.zen_server.routers.workspaces_endpoints import (
-    router as workspace_router,
-)
+from zenml.zen_server.routers.projects_endpoints import workspace_router
 from zenml.zen_server.utils import (
     handle_exceptions,
     make_dependable,
@@ -57,7 +55,7 @@ router = APIRouter(
 # TODO: the workspace scoped endpoint is only kept for dashboard compatibility
 # and can be removed after the migration
 @workspace_router.post(
-    "/{workspace_name_or_id}" + PIPELINE_BUILDS,
+    "/{project_name_or_id}" + PIPELINE_BUILDS,
     responses={401: error_response, 409: error_response, 422: error_response},
     deprecated=True,
     tags=["builds"],
@@ -65,21 +63,21 @@ router = APIRouter(
 @handle_exceptions
 def create_build(
     build: PipelineBuildRequest,
-    workspace_name_or_id: Optional[Union[str, UUID]] = None,
+    project_name_or_id: Optional[Union[str, UUID]] = None,
     _: AuthContext = Security(authorize),
 ) -> PipelineBuildResponse:
-    """Creates a build, optionally in a specific workspace.
+    """Creates a build, optionally in a specific project.
 
     Args:
         build: Build to create.
-        workspace_name_or_id: Optional name or ID of the workspace.
+        project_name_or_id: Optional name or ID of the project.
 
     Returns:
         The created build.
     """
-    if workspace_name_or_id:
-        workspace = zen_store().get_workspace(workspace_name_or_id)
-        build.workspace = workspace.id
+    if project_name_or_id:
+        project = zen_store().get_project(project_name_or_id)
+        build.project = project.id
 
     return verify_permissions_and_create_entity(
         request_model=build,
@@ -94,7 +92,7 @@ def create_build(
 # TODO: the workspace scoped endpoint is only kept for dashboard compatibility
 # and can be removed after the migration
 @workspace_router.get(
-    "/{workspace_name_or_id}" + PIPELINE_BUILDS,
+    "/{project_name_or_id}" + PIPELINE_BUILDS,
     responses={401: error_response, 404: error_response, 422: error_response},
     deprecated=True,
     tags=["builds"],
@@ -104,24 +102,24 @@ def list_builds(
     build_filter_model: PipelineBuildFilter = Depends(
         make_dependable(PipelineBuildFilter)
     ),
-    workspace_name_or_id: Optional[Union[str, UUID]] = None,
+    project_name_or_id: Optional[Union[str, UUID]] = None,
     hydrate: bool = False,
     _: AuthContext = Security(authorize),
 ) -> Page[PipelineBuildResponse]:
-    """Gets a list of builds, optionally filtered by workspace.
+    """Gets a list of builds.
 
     Args:
         build_filter_model: Filter model used for pagination, sorting,
             filtering.
-        workspace_name_or_id: Optional name or ID of the workspace to filter by.
+        project_name_or_id: Optional name or ID of the project to filter by.
         hydrate: Flag deciding whether to hydrate the output model(s)
             by including metadata fields in the response.
 
     Returns:
         List of build objects matching the filter criteria.
     """
-    if workspace_name_or_id:
-        build_filter_model.workspace = workspace_name_or_id
+    if project_name_or_id:
+        build_filter_model.project = project_name_or_id
 
     return verify_permissions_and_list_entities(
         filter_model=build_filter_model,

zenml/zen_server/routers/pipeline_deployments_endpoints.py

@@ -34,9 +34,7 @@ from zenml.zen_server.rbac.endpoint_utils import (
     verify_permissions_and_list_entities,
 )
 from zenml.zen_server.rbac.models import ResourceType
-from zenml.zen_server.routers.workspaces_endpoints import (
-    router as workspace_router,
-)
+from zenml.zen_server.routers.projects_endpoints import workspace_router
 from zenml.zen_server.utils import (
     handle_exceptions,
     make_dependable,
@@ -59,7 +57,7 @@ router = APIRouter(
 # TODO: the workspace scoped endpoint is only kept for dashboard compatibility
 # and can be removed after the migration
 @workspace_router.post(
-    "/{workspace_name_or_id}" + PIPELINE_DEPLOYMENTS,
+    "/{project_name_or_id}" + PIPELINE_DEPLOYMENTS,
     responses={401: error_response, 409: error_response, 422: error_response},
     deprecated=True,
     tags=["deployments"],
@@ -67,21 +65,21 @@ router = APIRouter(
 @handle_exceptions
 def create_deployment(
     deployment: PipelineDeploymentRequest,
-    workspace_name_or_id: Optional[Union[str, UUID]] = None,
+    project_name_or_id: Optional[Union[str, UUID]] = None,
     _: AuthContext = Security(authorize),
 ) -> PipelineDeploymentResponse:
     """Creates a deployment.
 
     Args:
         deployment: Deployment to create.
-        workspace_name_or_id: Optional name or ID of the workspace.
+        project_name_or_id: Optional name or ID of the project.
 
     Returns:
         The created deployment.
     """
-    if workspace_name_or_id:
-        workspace = zen_store().get_workspace(workspace_name_or_id)
-        deployment.workspace = workspace.id
+    if project_name_or_id:
+        project = zen_store().get_project(project_name_or_id)
+        deployment.project = project.id
 
     return verify_permissions_and_create_entity(
         request_model=deployment,
@@ -96,7 +94,7 @@ def create_deployment(
 # TODO: the workspace scoped endpoint is only kept for dashboard compatibility
 # and can be removed after the migration
 @workspace_router.get(
-    "/{workspace_name_or_id}" + PIPELINE_DEPLOYMENTS,
+    "/{project_name_or_id}" + PIPELINE_DEPLOYMENTS,
     responses={401: error_response, 404: error_response, 422: error_response},
     deprecated=True,
     tags=["deployments"],
@@ -106,24 +104,24 @@ def list_deployments(
     deployment_filter_model: PipelineDeploymentFilter = Depends(
         make_dependable(PipelineDeploymentFilter)
     ),
-    workspace_name_or_id: Optional[Union[str, UUID]] = None,
+    project_name_or_id: Optional[Union[str, UUID]] = None,
     hydrate: bool = False,
     _: AuthContext = Security(authorize),
 ) -> Page[PipelineDeploymentResponse]:
-    """Gets a list of deployments, optionally filtered by workspace.
+    """Gets a list of deployments.
 
     Args:
         deployment_filter_model: Filter model used for pagination, sorting,
             filtering.
-        workspace_name_or_id: Optional name or ID of the workspace to filter by.
+        project_name_or_id: Optional name or ID of the project to filter by.
         hydrate: Flag deciding whether to hydrate the output model(s)
             by including metadata fields in the response.
 
     Returns:
         List of deployment objects matching the filter criteria.
     """
-    if workspace_name_or_id:
-        deployment_filter_model.workspace = workspace_name_or_id
+    if project_name_or_id:
+        deployment_filter_model.project = project_name_or_id
 
     return verify_permissions_and_list_entities(
         filter_model=deployment_filter_model,

zenml/zen_server/routers/pipelines_endpoints.py

@@ -46,9 +46,7 @@ from zenml.zen_server.rbac.endpoint_utils import (
     verify_permissions_and_update_entity,
 )
 from zenml.zen_server.rbac.models import ResourceType
-from zenml.zen_server.routers.workspaces_endpoints import (
-    router as workspace_router,
-)
+from zenml.zen_server.routers.projects_endpoints import workspace_router
 from zenml.zen_server.utils import (
     handle_exceptions,
     make_dependable,
@@ -70,7 +68,7 @@ router = APIRouter(
 # TODO: the workspace scoped endpoint is only kept for dashboard compatibility
 # and can be removed after the migration
 @workspace_router.post(
-    "/{workspace_name_or_id}" + PIPELINES,
+    "/{project_name_or_id}" + PIPELINES,
     responses={401: error_response, 409: error_response, 422: error_response},
     deprecated=True,
     tags=["pipelines"],
@@ -78,26 +76,26 @@ router = APIRouter(
 @handle_exceptions
 def create_pipeline(
     pipeline: PipelineRequest,
-    workspace_name_or_id: Optional[Union[str, UUID]] = None,
+    project_name_or_id: Optional[Union[str, UUID]] = None,
     _: AuthContext = Security(authorize),
 ) -> PipelineResponse:
-    """Creates a pipeline, optionally in a specific workspace.
+    """Creates a pipeline.
 
     Args:
         pipeline: Pipeline to create.
-        workspace_name_or_id: Optional name or ID of the workspace.
+        project_name_or_id: Optional name or ID of the project.
 
     Returns:
         The created pipeline.
     """
-    if workspace_name_or_id:
-        workspace = zen_store().get_workspace(workspace_name_or_id)
-        pipeline.workspace = workspace.id
+    if project_name_or_id:
+        project = zen_store().get_project(project_name_or_id)
+        pipeline.project = project.id
 
     # We limit pipeline namespaces, not pipeline versions
     skip_entitlements = (
         zen_store().count_pipelines(
-            PipelineFilter(name=pipeline.name, workspace=pipeline.workspace)
+            PipelineFilter(name=pipeline.name, project=pipeline.project)
         )
         > 0
     )
@@ -116,7 +114,7 @@ def create_pipeline(
 # TODO: the workspace scoped endpoint is only kept for dashboard compatibility
 # and can be removed after the migration
 @workspace_router.get(
-    "/{workspace_name_or_id}" + PIPELINES,
+    "/{project_name_or_id}" + PIPELINES,
     responses={401: error_response, 404: error_response, 422: error_response},
     deprecated=True,
     tags=["pipelines"],
@@ -126,24 +124,24 @@ def list_pipelines(
     pipeline_filter_model: PipelineFilter = Depends(
         make_dependable(PipelineFilter)
     ),
-    workspace_name_or_id: Optional[Union[str, UUID]] = None,
+    project_name_or_id: Optional[Union[str, UUID]] = None,
     hydrate: bool = False,
     _: AuthContext = Security(authorize),
 ) -> Page[PipelineResponse]:
-    """Gets a list of pipelines, optionally filtered by workspace.
+    """Gets a list of pipelines.
 
     Args:
         pipeline_filter_model: Filter model used for pagination, sorting,
             filtering.
-        workspace_name_or_id: Optional name or ID of the workspace to filter by.
+        project_name_or_id: Optional name or ID of the project to filter by.
         hydrate: Flag deciding whether to hydrate the output model(s)
             by including metadata fields in the response.
 
     Returns:
         List of pipeline objects matching the filter criteria.
     """
-    if workspace_name_or_id:
-        pipeline_filter_model.workspace = workspace_name_or_id
+    if project_name_or_id:
+        pipeline_filter_model.project = project_name_or_id
 
     return verify_permissions_and_list_entities(
         filter_model=pipeline_filter_model,
@@ -228,7 +226,7 @@ def delete_pipeline(
     should_decrement = (
         ResourceType.PIPELINE in server_config().reportable_resources
         and zen_store().count_pipelines(
-            PipelineFilter(name=pipeline.name, workspace=pipeline.workspace.id)
+            PipelineFilter(name=pipeline.name, project=pipeline.project.id)
         )
         == 0
     )