# mcp/schemas.py

from pydantic import BaseModel
from typing import List, Optional

class Paper(BaseModel):
    title: str
    authors: str
    summary: str
    link: str
    published: str
    source: str

class UMLSConcept(BaseModel):
    term: str
    cui: Optional[str]
    name: Optional[str]
    definition: Optional[str]

class DrugSafety(BaseModel):
    safety_report_id: Optional[str]
    serious: Optional[str]
    reactions: Optional[List[str]]
    receivedate: Optional[str]

class UnifiedSearchResult(BaseModel):
    papers: List[Paper]
    umls: List[UMLSConcept]
    drug_safety: List[DrugSafety]
    ai_summary: str
    suggested_reading: List[str]

class UnifiedSearchInput(BaseModel):
    query: str