m00913563 commited on
Commit
06c8bab
·
1 Parent(s): 75ab853

fix data type

Browse files
Files changed (1) hide show
  1. models.py +6 -5
models.py CHANGED
@@ -3,18 +3,18 @@ from typing import List, Optional, Any
3
 
4
 
5
  class Experience(BaseModel):
6
- start: Optional[str] = Field(None, description="Start date in YYYY-MM-DD format")
7
- end: Optional[str] = Field(None, description="End date in YYYY-MM-DD format")
8
  designation: str = Field(...)
9
  company: str = Field(...)
10
  experience_description: str = Field(...)
11
 
12
  class Education(BaseModel):
13
- start: Optional[str] = Field(None, description="Start date in YYYY-MM-DD format")
14
- end: Optional[str] = Field(None, description="End date in YYYY-MM-DD format")
15
  major: str = Field(...)
16
  campus: str = Field(...)
17
- GPA: float = Field(...)
18
 
19
  class CVExtracted(BaseModel):
20
  name: str = Field(...)
@@ -23,6 +23,7 @@ class CVExtracted(BaseModel):
23
  achievements: List[str] = Field(...)
24
  experiences: List[Experience] = Field(...)
25
  educations: List[Education] = Field(...)
 
26
  class InsertedText(BaseModel):
27
  text: str
28
 
 
3
 
4
 
5
  class Experience(BaseModel):
6
+ start: Optional[str] = Field(None, description="Start date in ISO String format")
7
+ end: Optional[str] = Field(None, description="End date in ISO String format")
8
  designation: str = Field(...)
9
  company: str = Field(...)
10
  experience_description: str = Field(...)
11
 
12
  class Education(BaseModel):
13
+ start: Optional[str] = Field(None, description="Start date in ISO String format")
14
+ end: Optional[str] = Field(None, description="End date in ISO String format")
15
  major: str = Field(...)
16
  campus: str = Field(...)
17
+ GPA: int = Field(None, description="Get the two most significant digit then times by 100, for example 3.41 become 341")
18
 
19
  class CVExtracted(BaseModel):
20
  name: str = Field(...)
 
23
  achievements: List[str] = Field(...)
24
  experiences: List[Experience] = Field(...)
25
  educations: List[Education] = Field(...)
26
+
27
  class InsertedText(BaseModel):
28
  text: str
29