feat: avatar on team and agent

l3vels · Sep 19, 2023 · f9eb5f0 · f9eb5f0
1 parent 872907c
commit f9eb5f0
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 1 deletion.
diff --git a/apps/server/models/agent.py b/apps/server/models/agent.py
@@ -33,6 +33,7 @@ class AgentModel(BaseModel):
     __tablename__ = 'agent'
     id = Column(UUID, primary_key=True, index=True, default=uuid.uuid4)
     name = Column(String)
+    avatar = Column(String(300), default=None)
     role = Column(String) 
     parent_id = Column(UUID, ForeignKey('agent.id', ondelete='CASCADE'), nullable=True, index=True) 
     workspace_id = Column(UUID, ForeignKey('workspace.id', ondelete='CASCADE'), nullable=True, index=True) 

diff --git a/apps/server/models/team.py b/apps/server/models/team.py
@@ -25,6 +25,7 @@ class TeamModel(BaseModel):
 
     id = Column(UUID, primary_key=True, index=True, default=uuid.uuid4)
     name = Column(String)
+    avatar = Column(String(300), default=None)
     team_type = Column(String) #todo replace as enum (Debates, Plan_Execute, Authoritarian_Speaker, Decentralized_speaker)
     description = Column(String, nullable=True) 
     is_deleted = Column(Boolean, default=False, index=True)

diff --git a/apps/server/typings/team.py b/apps/server/typings/team.py
@@ -36,6 +36,7 @@ class TeamOutput(BaseModel):
     creator: Optional[UserOutput]
     modified_by: Optional[UUID4]
     team_agents: Optional[List[TeamAgentOutput]]
+    avatar: Optional[str]
 
 class QueryParams(BaseModel):
     id: Optional[str]

diff --git a/apps/server/typings/user.py b/apps/server/typings/user.py
@@ -10,7 +10,8 @@ class UserOutput(BaseModel):
     id: UUID4
     name: str
     email: str
-    avatar: str
+    avatar: Optional[str]
+
 @strawberry.type
 class User:
     id: strawberry.ID