Spaces:
Sleeping
Sleeping
Update app/schemas.py
Browse files- app/schemas.py +2 -7
app/schemas.py
CHANGED
|
@@ -2,15 +2,13 @@ from pydantic import BaseModel, Field
|
|
| 2 |
from typing import Optional, List
|
| 3 |
from app.models import Tone, ContentType, DraftStatus
|
| 4 |
|
| 5 |
-
|
| 6 |
class DraftCreate(BaseModel):
|
| 7 |
-
source_type: str
|
| 8 |
source_ref: Optional[str] = None
|
| 9 |
raw_text: Optional[str] = None
|
| 10 |
content_type: ContentType
|
| 11 |
tone: Tone = Tone.neutral
|
| 12 |
|
| 13 |
-
|
| 14 |
class DraftOut(BaseModel):
|
| 15 |
id: int
|
| 16 |
title: Optional[str]
|
|
@@ -19,11 +17,9 @@ class DraftOut(BaseModel):
|
|
| 19 |
content_type: ContentType
|
| 20 |
status: DraftStatus
|
| 21 |
|
| 22 |
-
|
| 23 |
class Config:
|
| 24 |
from_attributes = True
|
| 25 |
|
| 26 |
-
|
| 27 |
class ApproveInput(BaseModel):
|
| 28 |
approve: bool = True
|
| 29 |
subject_a: Optional[str] = None
|
|
@@ -32,6 +28,5 @@ class ApproveInput(BaseModel):
|
|
| 32 |
deliver_x: Optional[bool] = False
|
| 33 |
deliver_note: Optional[bool] = False
|
| 34 |
|
| 35 |
-
|
| 36 |
class DeliverInput(BaseModel):
|
| 37 |
-
channels: List[str]
|
|
|
|
| 2 |
from typing import Optional, List
|
| 3 |
from app.models import Tone, ContentType, DraftStatus
|
| 4 |
|
|
|
|
| 5 |
class DraftCreate(BaseModel):
|
| 6 |
+
source_type: str # pdf/url/text
|
| 7 |
source_ref: Optional[str] = None
|
| 8 |
raw_text: Optional[str] = None
|
| 9 |
content_type: ContentType
|
| 10 |
tone: Tone = Tone.neutral
|
| 11 |
|
|
|
|
| 12 |
class DraftOut(BaseModel):
|
| 13 |
id: int
|
| 14 |
title: Optional[str]
|
|
|
|
| 17 |
content_type: ContentType
|
| 18 |
status: DraftStatus
|
| 19 |
|
|
|
|
| 20 |
class Config:
|
| 21 |
from_attributes = True
|
| 22 |
|
|
|
|
| 23 |
class ApproveInput(BaseModel):
|
| 24 |
approve: bool = True
|
| 25 |
subject_a: Optional[str] = None
|
|
|
|
| 28 |
deliver_x: Optional[bool] = False
|
| 29 |
deliver_note: Optional[bool] = False
|
| 30 |
|
|
|
|
| 31 |
class DeliverInput(BaseModel):
|
| 32 |
+
channels: List[str] # ["email","x","note"]
|