| from pydantic import BaseModel, field_validator
|
| from typing import Dict, Optional
|
| from .user_settings import ApiKeyItem
|
|
|
|
|
| class SummaryResponse(BaseModel):
|
| arxiv_id: str
|
| summary: str
|
|
|
|
|
| class SummaryGenerateRequest(BaseModel):
|
| arxiv_id: Optional[str] = None
|
| pdf_url: Optional[str] = None
|
| api_keys_encrypted: Optional[list[ApiKeyItem]] = None
|
|
|
| @field_validator("arxiv_id", "pdf_url")
|
| @classmethod
|
| def check_at_least_one(cls, v, info):
|
| return v
|
|
|
| def model_post_init(self, __context) -> None:
|
| if not self.arxiv_id and not self.pdf_url:
|
| raise ValueError("Either arxiv_id or pdf_url must be provided")
|
|
|
|
|
| class UsabilityGenerateRequest(BaseModel):
|
| arxiv_id: Optional[str] = None
|
| pdf_url: Optional[str] = None
|
| api_keys_encrypted: Optional[list[ApiKeyItem]] = None
|
|
|
| @field_validator("arxiv_id", "pdf_url")
|
| @classmethod
|
| def check_at_least_one(cls, v, info):
|
| return v
|
|
|
| def model_post_init(self, __context) -> None:
|
| if not self.arxiv_id and not self.pdf_url:
|
| raise ValueError("Either arxiv_id or pdf_url must be provided")
|
|
|
|
|
| class UsabilityResponse(BaseModel):
|
| arxiv_id: Optional[str] = None
|
| pdf_url: Optional[str] = None
|
| domain_applicability: Dict[str, float]
|
| reproducibility_score: Dict[str, float]
|
| new_tech_applicability: Dict[str, float]
|
| impact_score: Optional[float]
|
|
|
