Create schemas.py

mcp/schemas.py

@@ -0,0 +1,34 @@
+# 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