Bills: fix related entity person/org fields

openstates · Jan 29, 2024 · 094c2f9 · 094c2f9
1 parent edbf970
commit 094c2f9
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/api/schemas.py b/api/schemas.py
@@ -259,6 +259,8 @@ class Config:
 class BillActionRelatedEntity(BaseModel):
     name: str = Field(..., example="Senate Committee of the Whole")
     entity_type: str = Field(..., example="organization")
+    organization: Optional[Organization] = Field(None, example=None)
+    person: Optional[CompactPerson]
 
     class Config:
         orm_mode = True
@@ -280,12 +282,6 @@ class Config:
 class BillDocumentLink(BaseModel):
     url: str = Field(..., example="https://example.com/doc.pdf")
     media_type: str = Field(..., example="application/pdf")
-    organization_id: str = Field(
-        ..., example="ocd-organization/fce467b7-470b-41c2-be23-0ed00804b512"
-    )
-    person_id: str = Field(
-        ..., example="ocd-person/5a327e3b-ae91-4279-ae17-9214aad32fa9"
-    )
 
     class Config:
         orm_mode = True