File size: 1,068 Bytes
20b15f3 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 | from typing import List, Literal, Optional
from pydantic import BaseModel, Field
class ResearchPaper(BaseModel):
title: str = Field(description="The title of the paper")
authors: Optional[List[str]] = Field(description="The list of the authors of the paper mentioned.", default=None)
abstract: Optional[str] = Field(description="The abstract or the summary of the mentioned research paper.", default=None)
year: Optional[int] = Field(description="The year in which the paper was published, leave if not mentioned", default=None)
citation_count: Optional[int] = Field(description="The citation count of the paper, leave if not mentioned", default=None)
url: Optional[str] = Field(description="The url of the research given on the source", default=None)
pdf_url: Optional[str] = Field(description="The pdf url of the research paper if mentioned", default=None)
source: Literal["arXiv", "SemanticScholar", "OpenAlex"] = Field(
description="The source from which it was extracted 'arXiv' , 'SemanticScholar' or 'OpenAlex'"
)
|