| """Shared Pydantic models. |
| |
| ``PaperRecord`` is the normalized in-memory shape produced by the fetch sources and stored |
| in the ``papers`` table. LLM output schemas (summary, analysis, scores, ...) are added in |
| later phases next to the stages that use them. |
| """ |
|
|
| from __future__ import annotations |
|
|
| from pydantic import BaseModel, Field |
|
|
|
|
| class Links(BaseModel): |
| abs: str | None = None |
| pdf: str | None = None |
| project_page: str | None = None |
| code: str | None = None |
| doi: str | None = None |
|
|
|
|
| class PaperRecord(BaseModel): |
| """A fetched, source-normalized item (paper or news) before any LLM processing.""" |
|
|
| id: str = Field(description="stable id, e.g. 'arxiv:2506.01234' or 'news:<sha1>'") |
| source: str |
| title: str |
| authors: list[str] = Field(default_factory=list) |
| published: str | None = None |
| abstract: str | None = None |
| categories: list[str] = Field(default_factory=list) |
| links: Links = Field(default_factory=Links) |
| citations: int = 0 |
| influential_citations: int = 0 |
| has_code: bool = False |
|
|
| @property |
| def arxiv_id(self) -> str | None: |
| return self.id.split("arxiv:", 1)[1] if self.id.startswith("arxiv:") else None |
|
|
