Handle missing metadata

climatepolicyradar · Aug 29, 2024 · 42309a7 · 42309a7
1 parent b622092
commit 42309a7
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/src/index/vespa_.py b/src/index/vespa_.py
@@ -117,12 +117,15 @@ class MetadataItem(BaseModel):
     corpus_type_name: Optional[str] = None
     collection_title: Optional[str] = None
     collection_summary: Optional[str] = None
-    metadata: list[MetadataItem]
+    metadata: Optional[list[MetadataItem]] = None
 
 
 def reshape_metadata(
-    metadata: dict[str, list[str]]
-) -> list[VespaFamilyDocument.MetadataItem]:
+    metadata: Optional[dict[str, list[str]]]
+) -> Optional[list[VespaFamilyDocument.MetadataItem]]:
+    if metadata is None:
+        return None
+
     metadata_items = []
     for key, values in metadata.items():
         metadata_items.extend(

diff --git a/tests/src/test_vespa.py b/tests/src/test_vespa.py
@@ -49,6 +49,7 @@
                 VespaFamilyDocument.MetadataItem(name="sector", value="Water"),
             ],
         ),
+        (None, None),
     ],
 )
 def test_reshape_metadata(metadata, expected):