zenml-nightly 0.84.0.dev20250729__py3-none-any.whl → 0.84.1.dev20250731__py3-none-any.whl

zenml/zen_stores/base_zen_store.py

@@ -52,8 +52,6 @@ from zenml.models import (
     ServerModel,
     StackFilter,
     StackResponse,
-    UserFilter,
-    UserResponse,
 )
 from zenml.utils.pydantic_utils import before_validator_handler
 from zenml.zen_stores.zen_store_interface import ZenStoreInterface
@@ -470,23 +468,6 @@ class BaseZenStore(
             raise KeyError("No default stack found.")
         return default_stacks.items[0]
 
-    def get_external_user(self, user_id: UUID) -> UserResponse:
-        """Get a user by external ID.
-
-        Args:
-            user_id: The external ID of the user.
-
-        Returns:
-            The user with the supplied external ID.
-
-        Raises:
-            KeyError: If the user doesn't exist.
-        """
-        users = self.list_users(UserFilter(external_user_id=user_id))
-        if users.total == 0:
-            raise KeyError(f"User with external ID '{user_id}' not found.")
-        return users.items[0]
-
     model_config = ConfigDict(
         # Validate attributes when assigning them. We need to set this in order
         # to have a mix of mutable and immutable attributes

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

@@ -0,0 +1,23 @@
+"""Release [0.84.1].
+
+Revision ID: 0.84.1
+Revises: 8d4b9ba22c1f
+Create Date: 2025-07-29 08:56:44.756559
+
+"""
+
+# revision identifiers, used by Alembic.
+revision = "0.84.1"
+down_revision = "8d4b9ba22c1f"
+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/migrations/versions/8d4b9ba22c1f_add_user_avatar.py

@@ -0,0 +1,39 @@
+"""add user avatar [8d4b9ba22c1f].
+
+Revision ID: 8d4b9ba22c1f
+Revises: d4591f95ac07
+Create Date: 2025-07-18 17:13:54.513273
+
+"""
+
+import sqlalchemy as sa
+import sqlmodel
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision = "8d4b9ba22c1f"
+down_revision = "d4591f95ac07"
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    """Upgrade database schema and/or data, creating a new revision."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table("user", schema=None) as batch_op:
+        batch_op.add_column(
+            sa.Column(
+                "avatar_url", sqlmodel.sql.sqltypes.AutoString(), nullable=True
+            )
+        )
+
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade database schema and/or data back to the previous revision."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table("user", schema=None) as batch_op:
+        batch_op.drop_column("avatar_url")
+
+    # ### end Alembic commands ###

zenml/zen_stores/schemas/user_schemas.py

@@ -14,13 +14,14 @@
 """SQLModel implementation of user tables."""
 
 import json
-from typing import TYPE_CHECKING, Any, List, Optional
+from typing import TYPE_CHECKING, Any, List, Optional, Union
 from uuid import UUID
 
 from sqlalchemy import TEXT, Column, UniqueConstraint
 from sqlmodel import Field, Relationship
 
 from zenml.models import (
+    ServiceAccountInternalRequest,
     ServiceAccountRequest,
     ServiceAccountResponse,
     ServiceAccountResponseBody,
@@ -76,6 +77,7 @@ class UserSchema(NamedSchema, table=True):
     full_name: str
     description: Optional[str] = Field(sa_column=Column(TEXT, nullable=True))
     email: Optional[str] = Field(nullable=True)
+    avatar_url: Optional[str] = Field(default=None, nullable=True)
     active: bool
     password: Optional[str] = Field(nullable=True)
     activation_token: Optional[str] = Field(nullable=True)
@@ -184,6 +186,7 @@ class UserSchema(NamedSchema, table=True):
         return cls(
             name=model.name,
             full_name=model.full_name,
+            avatar_url=model.avatar_url,
             active=model.active,
             password=model.create_hashed_password(),
             activation_token=model.create_hashed_activation_token(),
@@ -199,25 +202,29 @@ class UserSchema(NamedSchema, table=True):
 
     @classmethod
     def from_service_account_request(
-        cls, model: ServiceAccountRequest
+        cls, model: Union[ServiceAccountRequest, ServiceAccountInternalRequest]
     ) -> "UserSchema":
         """Create a `UserSchema` from a Service Account request.
 
         Args:
-            model: The `ServiceAccountRequest` from which to create the
-                schema.
+            model: The `ServiceAccountRequest` or `ServiceAccountInternalRequest`
+                from which to create the schema.
 
         Returns:
             The created `UserSchema`.
         """
         return cls(
             name=model.name,
+            full_name=model.full_name,
             description=model.description or "",
+            external_user_id=model.external_user_id
+            if isinstance(model, ServiceAccountInternalRequest)
+            else None,
             active=model.active,
             is_service_account=True,
             email_opted_in=False,
-            full_name="",
             is_admin=False,
+            avatar_url=model.avatar_url,
         )
 
     def update_user(self, user_update: UserUpdate) -> "UserSchema":
@@ -310,6 +317,7 @@ class UserSchema(NamedSchema, table=True):
                 updated=self.updated,
                 is_admin=self.is_admin,
                 default_project_id=self.default_project_id,
+                avatar_url=self.avatar_url,
             ),
             metadata=metadata,
         )
@@ -330,12 +338,15 @@ class UserSchema(NamedSchema, table=True):
         if include_metadata:
             metadata = ServiceAccountResponseMetadata(
                 description=self.description or "",
+                external_user_id=self.external_user_id,
             )
 
         body = ServiceAccountResponseBody(
+            full_name=self.full_name,
             created=self.created,
             updated=self.updated,
             active=self.active,
+            avatar_url=self.avatar_url,
         )
 
         return ServiceAccountResponse(

zenml/zen_stores/sql_zen_store.py

@@ -260,6 +260,7 @@ from zenml.models import (
     ServerSettingsResponse,
     ServerSettingsUpdate,
     ServiceAccountFilter,
+    ServiceAccountInternalRequest,
     ServiceAccountRequest,
     ServiceAccountResponse,
     ServiceAccountUpdate,
@@ -7296,7 +7297,10 @@ class SqlZenStore(BaseZenStore):
 
     @track_decorator(AnalyticsEvent.CREATED_SERVICE_ACCOUNT)
     def create_service_account(
-        self, service_account: ServiceAccountRequest
+        self,
+        service_account: Union[
+            ServiceAccountRequest, ServiceAccountInternalRequest
+        ],
     ) -> ServiceAccountResponse:
         """Creates a new service account.
 

{zenml_nightly-0.84.0.dev20250729.dist-info → zenml_nightly-0.84.1.dev20250731.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: zenml-nightly
-Version: 0.84.0.dev20250729
+Version: 0.84.1.dev20250731
 Summary: ZenML: Write production-ready ML code.
 License: Apache-2.0
 Keywords: machine learning,production,pipeline,mlops,devops