Update models/entities.py

models/entities.py

@@ -32,7 +32,7 @@ class Note(BaseModel):
     context: Optional[str] = ""
     author: Optional[str] = "System"
 
-class PatientCreate(BaseModel):
+class PatientBase(BaseModel):
     full_name: str
     gender: str
     date_of_birth: str
@@ -43,7 +43,21 @@ class PatientCreate(BaseModel):
     country: Optional[str] = "US"
     marital_status: Optional[str] = "Never Married"
     language: Optional[str] = "en"
+
+class PatientCreate(PatientBase):
     conditions: Optional[List[Condition]] = []
     medications: Optional[List[Medication]] = []
     encounters: Optional[List[Encounter]] = []
-    notes: Optional[List[Note]] = []
+    notes: Optional[List[Note]] = []
+
+class Patient(PatientBase):
+    id: str
+    fhir_id: Optional[str] = None
+    source: Optional[str] = None
+    conditions: List[Condition] = []
+    medications: List[Medication] = []
+    encounters: List[Encounter] = []
+    notes: List[Note] = []
+    
+    class Config:
+        orm_mode = True