Add email in User model

evidentlyai · Apr 8, 2024 · 7cfe0be · 7cfe0be
1 parent 9d48bc2
commit 7cfe0be
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/src/evidently/ui/api/models.py b/src/evidently/ui/api/models.py
@@ -120,7 +120,8 @@ def from_team(cls, team: Team):
 class UserModel(BaseModel):
     id: UserID
     name: str
+    email: str
 
     @classmethod
     def from_user(cls, user: User):
-        return TeamModel(id=user.id, name=user.name)
+        return UserModel(id=user.id, name=user.name, email=user.email)
diff --git a/src/evidently/ui/base.py b/src/evidently/ui/base.py
@@ -103,6 +103,7 @@ class Team(BaseModel):
 class User(BaseModel):
     id: UserID = Field(default_factory=uuid.uuid4)
     name: str
+    email: str = ""
 
 
 def _default_dashboard():
@@ -406,10 +407,10 @@ def delete_team(self, user_id: UserID, team_id: TeamID):
         self._delete_team(team_id)
 
     @abstractmethod
-    def _list_team_users(self, team_id: TeamID) -> List[User]:
+    def _list_team_users(self, team_id: TeamID) -> List[UserID]:
         raise NotImplementedError
 
-    def list_team_users(self, user_id: UserID, team_id: TeamID) -> List[User]:
+    def list_team_users(self, user_id: UserID, team_id: TeamID) -> List[UserID]:
         if not self.check_team_permission(user_id, team_id, TeamPermission.READ):
             raise TeamNotFound()
         return self._list_team_users(team_id)

diff --git a/src/evidently/ui/storage/common.py b/src/evidently/ui/storage/common.py
@@ -73,7 +73,7 @@ def _list_user_teams(self, user_id: UserID, include_virtual: bool) -> List[Team]
     def _delete_team(self, team_id: TeamID):
         pass
 
-    def _list_team_users(self, team_id: TeamID) -> List[User]:
+    def _list_team_users(self, team_id: TeamID) -> List[UserID]:
         return []