Add some descriptions for readability

jyhsu2000 · May 3, 2024 · e1eb11f · e1eb11f
1 parent 068a79d
commit e1eb11f
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/app/models.py b/app/models.py
@@ -3,14 +3,14 @@
 
 class Segment(BaseModel):
     word: str
-    pos: str
+    pos: str = Field(description='Part of speech')
 
 
 class NamedEntity(BaseModel):
     word: str
-    type: str
-    start: int
-    end: int
+    type: str = Field(description='Entity type')
+    start: int = Field(description='Start position')
+    end: int = Field(description='End position')
 
 
 class Sentence(BaseModel):