diff --git a/ingest.py b/ingest.py index ec61478..6a2b9b8 100644 --- a/ingest.py +++ b/ingest.py @@ -90,6 +90,10 @@ def build_bulk_actions( "transcript_full": transcript_full, "transcript_secondary_full": doc.get("transcript_secondary_full"), "transcript_parts": parts, + "internal_references": doc.get("internal_references", []), + "internal_references_count": doc.get("internal_references_count", 0), + "referenced_by": doc.get("referenced_by", []), + "referenced_by_count": doc.get("referenced_by_count", 0), }, } @@ -121,6 +125,10 @@ def ensure_index(client: "Elasticsearch", index: str) -> None: "text": {"type": "text"}, }, }, + "internal_references": {"type": "keyword"}, + "internal_references_count": {"type": "integer"}, + "referenced_by": {"type": "keyword"}, + "referenced_by_count": {"type": "integer"}, } }, ) diff --git a/search_app.py b/search_app.py index c0ecd39..00f2da1 100644 --- a/search_app.py +++ b/search_app.py @@ -400,7 +400,7 @@ def build_query_payload( elif sort == "older": body["sort"] = [{"date": {"order": "asc"}}] elif sort == "referenced": - body["sort"] = [{"referenced_by_count": {"order": "desc"}}] + body["sort"] = [{"referenced_by_count": {"order": "desc", "unmapped_type": "long"}}] return body if query: @@ -505,7 +505,7 @@ def build_query_payload( elif sort == "older": body["sort"] = [{"date": {"order": "asc"}}] elif sort == "referenced": - body["sort"] = [{"referenced_by_count": {"order": "desc"}}] + body["sort"] = [{"referenced_by_count": {"order": "desc", "unmapped_type": "long"}}] return body