File size: 880 Bytes
220d87f
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
from pydantic import BaseModel, EmailStr, Field
from typing import List, Optional
from datetime import datetime

class UserCreate(BaseModel):
    email: EmailStr
    password: str = Field(min_length=6)

class UserOut(BaseModel):
    id: int
    email: EmailStr
    created_at: datetime
    class Config:
        from_attributes = True

class Token(BaseModel):
    access_token: str
    token_type: str = "bearer"

class LoginIn(BaseModel):
    email: EmailStr
    password: str

class APIKeyOut(BaseModel):
    key: str
    revoked: bool
    created_at: datetime
    class Config:
        from_attributes = True

class KeyRevokeIn(BaseModel):
    key: str

class ChatIn(BaseModel):
    message: str

class ChatOut(BaseModel):
    reply: str

class ChatMessageOut(BaseModel):
    role: str
    content: str
    created_at: datetime
    class Config:
        from_attributes = True