Corin1998 commited on
Commit
0e6eb22
·
verified ·
1 Parent(s): d2a463b

Update app/schemas.py

Browse files
Files changed (1) hide show
  1. 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 # pdf/url/text
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] # ["email","x","note"]
 
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"]