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

zenml/zen_stores/schemas/schedule_schema.py

@@ -32,10 +32,10 @@ from zenml.utils.time_utils import utc_now
 from zenml.zen_stores.schemas.base_schemas import NamedSchema
 from zenml.zen_stores.schemas.component_schemas import StackComponentSchema
 from zenml.zen_stores.schemas.pipeline_schemas import PipelineSchema
+from zenml.zen_stores.schemas.project_schemas import ProjectSchema
 from zenml.zen_stores.schemas.schema_utils import build_foreign_key_field
 from zenml.zen_stores.schemas.user_schemas import UserSchema
 from zenml.zen_stores.schemas.utils import RunMetadataInterface
-from zenml.zen_stores.schemas.workspace_schemas import WorkspaceSchema
 
 if TYPE_CHECKING:
     from zenml.zen_stores.schemas.pipeline_deployment_schemas import (
@@ -53,20 +53,20 @@ class ScheduleSchema(NamedSchema, RunMetadataInterface, table=True):
     __table_args__ = (
         UniqueConstraint(
             "name",
-            "workspace_id",
-            name="unique_schedule_name_in_workspace",
+            "project_id",
+            name="unique_schedule_name_in_project",
         ),
     )
 
-    workspace_id: UUID = build_foreign_key_field(
+    project_id: UUID = build_foreign_key_field(
         source=__tablename__,
-        target=WorkspaceSchema.__tablename__,
-        source_column="workspace_id",
+        target=ProjectSchema.__tablename__,
+        source_column="project_id",
         target_column="id",
         ondelete="CASCADE",
         nullable=False,
     )
-    workspace: "WorkspaceSchema" = Relationship(back_populates="schedules")
+    project: "ProjectSchema" = Relationship(back_populates="schedules")
 
     user_id: Optional[UUID] = build_foreign_key_field(
         source=__tablename__,
@@ -138,7 +138,7 @@ class ScheduleSchema(NamedSchema, RunMetadataInterface, table=True):
             interval_second = None
         return cls(
             name=schedule_request.name,
-            workspace_id=schedule_request.workspace,
+            project_id=schedule_request.project,
             user_id=schedule_request.user,
             pipeline_id=schedule_request.pipeline_id,
             orchestrator_id=schedule_request.orchestrator_id,
@@ -203,7 +203,7 @@ class ScheduleSchema(NamedSchema, RunMetadataInterface, table=True):
         metadata = None
         if include_metadata:
             metadata = ScheduleResponseMetadata(
-                workspace=self.workspace.to_model(),
+                project=self.project.to_model(),
                 pipeline_id=self.pipeline_id,
                 orchestrator_id=self.orchestrator_id,
                 run_metadata=self.fetch_metadata(),

zenml/zen_stores/schemas/service_schemas.py

@@ -35,9 +35,9 @@ from zenml.utils.time_utils import utc_now
 from zenml.zen_stores.schemas.base_schemas import NamedSchema
 from zenml.zen_stores.schemas.model_schemas import ModelVersionSchema
 from zenml.zen_stores.schemas.pipeline_run_schemas import PipelineRunSchema
+from zenml.zen_stores.schemas.project_schemas import ProjectSchema
 from zenml.zen_stores.schemas.schema_utils import build_foreign_key_field
 from zenml.zen_stores.schemas.user_schemas import UserSchema
-from zenml.zen_stores.schemas.workspace_schemas import WorkspaceSchema
 
 
 class ServiceSchema(NamedSchema, table=True):
@@ -45,15 +45,15 @@ class ServiceSchema(NamedSchema, table=True):
 
     __tablename__ = "service"
 
-    workspace_id: UUID = build_foreign_key_field(
+    project_id: UUID = build_foreign_key_field(
         source=__tablename__,
-        target=WorkspaceSchema.__tablename__,
-        source_column="workspace_id",
+        target=ProjectSchema.__tablename__,
+        source_column="project_id",
         target_column="id",
         ondelete="CASCADE",
         nullable=False,
     )
-    workspace: "WorkspaceSchema" = Relationship(back_populates="services")
+    project: "ProjectSchema" = Relationship(back_populates="services")
 
     user_id: Optional[UUID] = build_foreign_key_field(
         source=__tablename__,
@@ -146,7 +146,7 @@ class ServiceSchema(NamedSchema, table=True):
         metadata = None
         if include_metadata:
             metadata = ServiceResponseMetadata(
-                workspace=self.workspace.to_model(),
+                project=self.project.to_model(),
                 service_source=self.service_source,
                 config=json.loads(base64.b64decode(self.config).decode()),
                 status=json.loads(base64.b64decode(self.status).decode())
@@ -226,7 +226,7 @@ class ServiceSchema(NamedSchema, table=True):
         """
         return cls(
             name=service_request.name,
-            workspace_id=service_request.workspace,
+            project_id=service_request.project,
             user_id=service_request.user,
             service_source=service_request.service_source,
             service_type=service_request.service_type.model_dump_json(),

zenml/zen_stores/schemas/step_run_schemas.py

@@ -50,10 +50,10 @@ from zenml.zen_stores.schemas.pipeline_deployment_schemas import (
     PipelineDeploymentSchema,
 )
 from zenml.zen_stores.schemas.pipeline_run_schemas import PipelineRunSchema
+from zenml.zen_stores.schemas.project_schemas import ProjectSchema
 from zenml.zen_stores.schemas.schema_utils import build_foreign_key_field
 from zenml.zen_stores.schemas.user_schemas import UserSchema
 from zenml.zen_stores.schemas.utils import RunMetadataInterface
-from zenml.zen_stores.schemas.workspace_schemas import WorkspaceSchema
 
 if TYPE_CHECKING:
     from zenml.zen_stores.schemas.artifact_schemas import ArtifactVersionSchema
@@ -126,10 +126,10 @@ class StepRunSchema(NamedSchema, RunMetadataInterface, table=True):
         ondelete="SET NULL",
         nullable=True,
     )
-    workspace_id: UUID = build_foreign_key_field(
+    project_id: UUID = build_foreign_key_field(
         source=__tablename__,
-        target=WorkspaceSchema.__tablename__,
-        source_column="workspace_id",
+        target=ProjectSchema.__tablename__,
+        source_column="project_id",
         target_column="id",
         ondelete="CASCADE",
         nullable=False,
@@ -144,7 +144,7 @@ class StepRunSchema(NamedSchema, RunMetadataInterface, table=True):
     )
 
     # Relationships
-    workspace: "WorkspaceSchema" = Relationship(back_populates="step_runs")
+    project: "ProjectSchema" = Relationship(back_populates="step_runs")
     user: Optional["UserSchema"] = Relationship(back_populates="step_runs")
     deployment: Optional["PipelineDeploymentSchema"] = Relationship(
         back_populates="step_runs"
@@ -200,7 +200,7 @@ class StepRunSchema(NamedSchema, RunMetadataInterface, table=True):
         """
         return cls(
             name=request.name,
-            workspace_id=request.workspace,
+            project_id=request.project,
             user_id=request.user,
             start_time=request.start_time,
             end_time=request.end_time,
@@ -310,7 +310,7 @@ class StepRunSchema(NamedSchema, RunMetadataInterface, table=True):
         metadata = None
         if include_metadata:
             metadata = StepRunResponseMetadata(
-                workspace=self.workspace.to_model(),
+                project=self.project.to_model(),
                 config=full_step_config.config,
                 spec=full_step_config.spec,
                 cache_key=self.cache_key,

zenml/zen_stores/schemas/trigger_schemas.py

@@ -41,10 +41,10 @@ from zenml.utils.time_utils import utc_now
 from zenml.zen_stores.schemas.action_schemas import ActionSchema
 from zenml.zen_stores.schemas.base_schemas import BaseSchema, NamedSchema
 from zenml.zen_stores.schemas.event_source_schemas import EventSourceSchema
+from zenml.zen_stores.schemas.project_schemas import ProjectSchema
 from zenml.zen_stores.schemas.schema_utils import build_foreign_key_field
 from zenml.zen_stores.schemas.user_schemas import UserSchema
 from zenml.zen_stores.schemas.utils import get_page_from_list
-from zenml.zen_stores.schemas.workspace_schemas import WorkspaceSchema
 
 
 class TriggerSchema(NamedSchema, table=True):
@@ -54,20 +54,20 @@ class TriggerSchema(NamedSchema, table=True):
     __table_args__ = (
         UniqueConstraint(
             "name",
-            "workspace_id",
-            name="unique_trigger_name_in_workspace",
+            "project_id",
+            name="unique_trigger_name_in_project",
         ),
     )
 
-    workspace_id: UUID = build_foreign_key_field(
+    project_id: UUID = build_foreign_key_field(
         source=__tablename__,
-        target=WorkspaceSchema.__tablename__,
-        source_column="workspace_id",
+        target=ProjectSchema.__tablename__,
+        source_column="project_id",
         target_column="id",
         ondelete="CASCADE",
         nullable=False,
     )
-    workspace: "WorkspaceSchema" = Relationship(back_populates="triggers")
+    project: "ProjectSchema" = Relationship(back_populates="triggers")
 
     user_id: Optional[UUID] = build_foreign_key_field(
         source=__tablename__,
@@ -155,7 +155,7 @@ class TriggerSchema(NamedSchema, table=True):
         """
         return cls(
             name=request.name,
-            workspace_id=request.workspace,
+            project_id=request.project,
             user_id=request.user,
             action_id=request.action_id,
             event_source_id=request.event_source_id,
@@ -208,7 +208,7 @@ class TriggerSchema(NamedSchema, table=True):
         metadata = None
         if include_metadata:
             metadata = TriggerResponseMetadata(
-                workspace=self.workspace.to_model(),
+                project=self.project.to_model(),
                 event_filter=json.loads(
                     base64.b64decode(self.event_filter).decode()
                 ),

zenml/zen_stores/schemas/user_schemas.py

@@ -34,8 +34,6 @@ from zenml.models import (
 )
 from zenml.utils.time_utils import utc_now
 from zenml.zen_stores.schemas.base_schemas import NamedSchema
-from zenml.zen_stores.schemas.schema_utils import build_foreign_key_field
-from zenml.zen_stores.schemas.workspace_schemas import WorkspaceSchema
 
 if TYPE_CHECKING:
     from zenml.zen_stores.schemas import (
@@ -84,14 +82,6 @@ class UserSchema(NamedSchema, table=True):
     external_user_id: Optional[UUID] = Field(nullable=True)
     is_admin: bool = Field(default=False)
     user_metadata: Optional[str] = Field(nullable=True)
-    default_workspace_id: Optional[UUID] = build_foreign_key_field(
-        source=__tablename__,
-        target=WorkspaceSchema.__tablename__,
-        source_column="default_workspace_id",
-        target_column="id",
-        ondelete="SET NULL",
-        nullable=True,
-    )
 
     stacks: List["StackSchema"] = Relationship(back_populates="user")
     components: List["StackComponentSchema"] = Relationship(
@@ -195,7 +185,6 @@ class UserSchema(NamedSchema, table=True):
             user_metadata=json.dumps(model.user_metadata)
             if model.user_metadata
             else None,
-            default_workspace_id=model.default_workspace_id,
         )
 
     @classmethod
@@ -310,7 +299,6 @@ class UserSchema(NamedSchema, table=True):
                 created=self.created,
                 updated=self.updated,
                 is_admin=self.is_admin,
-                default_workspace_id=self.default_workspace_id,
             ),
             metadata=metadata,
         )