Spaces:
Sleeping
Sleeping
File size: 1,469 Bytes
e9ea7c0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 |
from pydantic import BaseModel
from typing import Optional, List
from datetime import datetime
class UserBase(BaseModel):
email: str
full_name: Optional[str] = None
company_name: Optional[str] = None
class UserCreate(UserBase):
password: str
class User(UserBase):
id: int
is_admin: bool
class Config:
orm_mode = True
class Token(BaseModel):
access_token: str
token_type: str
class TokenData(BaseModel):
email: Optional[str] = None
class UserActivity(User):
message_count: int
prompt_count: int
class ConversationBase(BaseModel):
title: str
class ConversationCreate(ConversationBase):
pass
class Conversation(ConversationBase):
id: int
user_id: int
created_at: datetime
updated_at: datetime
class Config:
orm_mode = True
class ChatMessageBase(BaseModel):
role: str
content: str
conversation_id: Optional[int] = None
class ChatMessageCreate(ChatMessageBase):
pass
class ChatMessage(ChatMessageBase):
id: int
user_id: int
timestamp: datetime
class Config:
orm_mode = True
class UserActivity(User):
message_count: int
prompt_count: int
class SavedPromptBase(BaseModel):
title: str
content: str
is_public: bool = False
class SavedPromptCreate(SavedPromptBase):
pass
class SavedPrompt(SavedPromptBase):
id: int
user_id: int
created_at: datetime
class Config:
orm_mode = True
|