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

zenml/zen_stores/schemas/user_schemas.py

@@ -39,6 +39,7 @@ if TYPE_CHECKING:
     from zenml.zen_stores.schemas import (
         ActionSchema,
         APIKeySchema,
+        ArtifactSchema,
         ArtifactVersionSchema,
         CodeRepositorySchema,
         EventSourceSchema,
@@ -51,6 +52,7 @@ if TYPE_CHECKING:
         PipelineRunSchema,
         PipelineSchema,
         RunMetadataSchema,
+        RunTemplateSchema,
         ScheduleSchema,
         SecretSchema,
         ServiceConnectorSchema,
@@ -58,6 +60,7 @@ if TYPE_CHECKING:
         StackComponentSchema,
         StackSchema,
         StepRunSchema,
+        TagSchema,
         TriggerSchema,
     )
 
@@ -100,8 +103,12 @@ class UserSchema(NamedSchema, table=True):
         back_populates="user",
     )
     runs: List["PipelineRunSchema"] = Relationship(back_populates="user")
+    run_templates: List["RunTemplateSchema"] = Relationship(
+        back_populates="user",
+    )
     step_runs: List["StepRunSchema"] = Relationship(back_populates="user")
     builds: List["PipelineBuildSchema"] = Relationship(back_populates="user")
+    artifacts: List["ArtifactSchema"] = Relationship(back_populates="user")
     artifact_versions: List["ArtifactVersionSchema"] = Relationship(
         back_populates="user"
     )
@@ -150,6 +157,9 @@ class UserSchema(NamedSchema, table=True):
         back_populates="service_account",
         sa_relationship_kwargs={"cascade": "delete"},
     )
+    tags: List["TagSchema"] = Relationship(
+        back_populates="user",
+    )
 
     @classmethod
     def from_user_request(cls, model: UserRequest) -> "UserSchema":
@@ -262,8 +272,8 @@ class UserSchema(NamedSchema, table=True):
             include_resources: Whether the resources will be filled.
             **kwargs: Keyword arguments to allow schema specific logic
             include_private: Whether to include the user private information
-                             this is to limit the amount of data one can get
-                             about other users
+                this is to limit the amount of data one can get about other
+                users.
 
         Returns:
             The converted `UserResponse`.

zenml/zen_stores/schemas/utils.py

@@ -110,3 +110,19 @@ class RunMetadataInterface:
             k: sorted(v, key=lambda x: x.created, reverse=True)[0].value
             for k, v in metadata_collection.items()
         }
+
+
+def get_resource_type_name(schema_class: Type[BaseSchema]) -> str:
+    """Get the name of a resource from a schema class.
+
+    Args:
+        schema_class: The schema class to get the name of.
+
+    Returns:
+        The name of the resource.
+    """
+    entity_name = schema_class.__tablename__
+    assert isinstance(entity_name, str)
+    # Some entities are plural, some are singular, some have multiple words
+    # in their table name connected by underscores (e.g. pipeline_run)
+    return entity_name.replace("_", " ").rstrip("s")

zenml/zen_stores/secrets_stores/service_connector_secrets_store.py

@@ -23,7 +23,6 @@ from typing import (
     Optional,
     Type,
 )
-from uuid import uuid4
 
 from pydantic import Field, model_validator
 
@@ -133,8 +132,6 @@ class ServiceConnectorSecretsStore(BaseSecretsStore):
                 name="secrets-store",
                 connector_type=self.SERVICE_CONNECTOR_TYPE,
                 resource_types=[self.SERVICE_CONNECTOR_RESOURCE_TYPE],
-                user=uuid4(),  # Use a fake user ID
-                workspace=uuid4(),  # Use a fake workspace ID
                 auth_method=self.config.auth_method,
                 configuration=self.config.auth_config,
             )