Update document_generator.py

document_generator.py

@@ -527,6 +527,9 @@ from uuid import uuid4
 import csv
 from io import StringIO
 
+class ObservationResponse(BaseModel):
+    observations: List[Dict]
+    
 def create_csv_response(observations: List[Dict]) -> StreamingResponse:
     def iter_csv(data):
         output = StringIO()