Oxenstierna

""" Data models for Riksarkivet MCP server. """ from __future__ import annotations from typing import Dict, List, Optional from pydantic import BaseModel class SearchHit(BaseModel): pid: str title: str reference_code: str page_number: str snippet_text: str full_page_text: Optional[str] = None alto_url: Optional[str] = None image_url: Optional[str] = None bildvisning_url: Optional[str] = None score: float = 0.0 hierarchy: Optional[List[Dict[str, str]]] = None note: Optional[str] = None collection_url: Optional[str] = None manifest_url: Optional[str] = None archival_institution: Optional[List[Dict[str, str]]] = None date: Optional[str] = None class PageContext(BaseModel): page_number: int page_id: str reference_code: str full_text: str alto_url: str image_url: str bildvisning_url: str = "" class DocumentMetadata(BaseModel): """Document metadata containing archival information.""" title: Optional[str] = None hierarchy: Optional[List[Dict[str, str]]] = None archival_institution: Optional[List[Dict[str, str]]] = None date: Optional[str] = None note: Optional[str] = None collection_url: Optional[str] = None manifest_url: Optional[str] = None class SearchResult(BaseModel): hits: List[SearchHit] total_hits: int keyword: str offset: int enriched: bool = False def extract_summary(self) -> "SearchSummary": """Extract summary information from this search result. Groups hits by document and creates a SearchSummary with metadata. Returns: SearchSummary containing grouped hits and statistics. """ # Group hits by document (reference code or PID) grouped_hits: Dict[str, List[SearchHit]] = {} for hit in self.hits: document_id = hit.reference_code or hit.pid if document_id not in grouped_hits: grouped_hits[document_id] = [] grouped_hits[document_id].append(hit) return SearchSummary( keyword=self.keyword, total_hits=self.total_hits, page_hits_returned=len(self.hits), documents_returned=len(grouped_hits), enriched=self.enriched, offset=self.offset, grouped_hits=grouped_hits, ) class BrowseResult(BaseModel): contexts: List[PageContext] reference_code: str pages_requested: str # pid: Optional[str] = None manifest_id: Optional[str] = None document_metadata: Optional[DocumentMetadata] = None class SearchSummary(BaseModel): """Summary information from a search operation.""" keyword: str total_hits: int page_hits_returned: int documents_returned: int enriched: bool offset: int grouped_hits: Dict[str, List[SearchHit]]