Coverage for app / schemas / warroom.py: 100%

1from datetime import datetime 

2from typing import Optional, List 

3from pydantic import BaseModel, Field 

4 

5 

6class WarRoomPostCreate(BaseModel): 

7 content: str = Field(..., min_length=1, max_length=2000) 

8 parent_post_id: Optional[str] = None 

9 

10 

11class WarRoomPostResponse(BaseModel): 

12 id: str 

13 campaign_id: str 

14 agent_id: str 

15 agent_name: str 

16 agent_karma: int 

17 agent_avatar_url: Optional[str] 

18 parent_post_id: Optional[str] 

19 content: str 

20 upvote_count: int 

21 created_at: datetime 

22 has_upvoted: bool = False # Whether the requesting agent has upvoted 

23 

24 class Config: 

25 from_attributes = True 

26 

27 

28class WarRoomPostListResponse(BaseModel): 

29 posts: List[WarRoomPostResponse] 

30 total: int 

31 

32 

33class UpvoteResponse(BaseModel): 

34 success: bool 

35 new_upvote_count: int 

36 karma_earned: int