1 files changed, 5 insertions, 0 deletions
diff --git a/src/main/scala/xyz/driver/pdsuidomain/formats/json/extracteddata.scala b/src/main/scala/xyz/driver/pdsuidomain/formats/json/extracteddata.scala
index c9ac55b..232bb39 100644
--- a/src/main/scala/xyz/driver/pdsuidomain/formats/json/extracteddata.scala
+++ b/src/main/scala/xyz/driver/pdsuidomain/formats/json/extracteddata.scala
@@ -110,6 +110,10 @@ object extracteddata {
 
     override def read(json: JsValue): RichExtractedData = json match {
       case JsObject(fields) =>
+        val id = fields
+          .get("id")
+          .flatMap(_.convertTo[Option[LongId[ExtractedData]]])
+
         val documentId = fields
           .get("documentId")
           .map(_.convertTo[LongId[Document]])
@@ -135,6 +139,7 @@ object extracteddata {
           .map(l => applyLabelsForExtractedData(l, LongId(0)))
 
         val extractedData = ExtractedData(
+          id = id.getOrElse(LongId(0L)),
           documentId = documentId,
           keywordId = keywordId,
           evidenceText = evidence,