Add corpus items to passages in post processing

climatepolicyradar · Sep 18, 2024 · 857e252 · 857e252
1 parent cf3f76b
commit 857e252
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/cpr_sdk/models/search.py b/src/cpr_sdk/models/search.py
@@ -397,6 +397,8 @@ def from_vespa_response(cls, response_hit: dict) -> "Passage":
             document_content_type=fields.get("document_content_type"),
             document_cdn_object=fields.get("document_cdn_object"),
             document_source_url=fields.get("document_source_url"),
+            corpus_type_name=fields.get("corpus_type_name"),
+            corpus_import_id=fields.get("corpus_import_id"),
             text_block=fields["text_block"],
             text_block_id=fields["text_block_id"],
             text_block_type=fields["text_block_type"],