sunlabuiuc · dsimanis · Apr 13, 2026
diff --git a/docs/api/datasets.rst b/docs/api/datasets.rst
@@ -225,6 +225,7 @@ Available Datasets
     datasets/pyhealth.datasets.MIMIC3Dataset
     datasets/pyhealth.datasets.MIMIC4Dataset
     datasets/pyhealth.datasets.MedicalTranscriptionsDataset
+    datasets/pyhealth.datasets.MimicIVNoteExtDIDataset
     datasets/pyhealth.datasets.CardiologyDataset
     datasets/pyhealth.datasets.eICUDataset
     datasets/pyhealth.datasets.ISRUCDataset

diff --git a/docs/api/datasets/pyhealth.datasets.MimicIVNoteExtDIDataset.rst b/docs/api/datasets/pyhealth.datasets.MimicIVNoteExtDIDataset.rst
@@ -0,0 +1,9 @@
+pyhealth.datasets.MimicIVNoteExtDIDataset
+==========================================
+
+MIMIC-IV-Note-Ext-DI dataset for patient summary generation, refer to `PhysioNet <https://doi.org/10.13026/m6hf-dq94>`_ and `paper <https://proceedings.mlr.press/v248/hegselmann24a.html>`_.
+
+.. autoclass:: pyhealth.datasets.MimicIVNoteExtDIDataset
+    :members:
+    :undoc-members:
+    :show-inheritance:
diff --git a/docs/api/tasks.rst b/docs/api/tasks.rst
@@ -214,6 +214,7 @@ Available Tasks
     Drug Recommendation <tasks/pyhealth.tasks.drug_recommendation>
     Length of Stay Prediction <tasks/pyhealth.tasks.length_of_stay_prediction>
     Medical Transcriptions Classification <tasks/pyhealth.tasks.MedicalTranscriptionsClassification>
+    Patient Summary Generation <tasks/pyhealth.tasks.PatientSummaryGeneration>
     Mortality Prediction (Next Visit) <tasks/pyhealth.tasks.mortality_prediction>
     Mortality Prediction (StageNet MIMIC-IV) <tasks/pyhealth.tasks.mortality_prediction_stagenet_mimic4>
     Patient Linkage (MIMIC-III) <tasks/pyhealth.tasks.patient_linkage_mimic3_fn>

diff --git a/docs/api/tasks/pyhealth.tasks.PatientSummaryGeneration.rst b/docs/api/tasks/pyhealth.tasks.PatientSummaryGeneration.rst
@@ -0,0 +1,7 @@
+pyhealth.tasks.PatientSummaryGeneration
+========================================
+
+.. autoclass:: pyhealth.tasks.patient_summary_generation.PatientSummaryGeneration
+    :members:
+    :undoc-members:
+    :show-inheritance: