File size: 718 Bytes
b36498c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
# 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
|