Update mcp/pubmed.py

mcp/pubmed.py

@@ -30,17 +30,52 @@ async def fetch_pubmed(query: str, max_results: int = 5):
         }
         efetch_resp = await client.get(PUBMED_EFETCH, params=efetch_params)
         articles = xmltodict.parse(efetch_resp.text)["PubmedArticleSet"].get("PubmedArticle", [])
-        if not isinstance(articles, list):  # Single article
+        if not isinstance(articles, list):  # Single article edge case
             articles = [articles]
         results = []
         for a in articles:
             art = a["MedlineCitation"]["Article"]
+
+            # Robustly extract publication year or date
+            published = ""
+            article_date = art.get("ArticleDate")
+            if isinstance(article_date, list) and article_date:
+                published = article_date[0].get("Year", "")
+            elif isinstance(article_date, dict):
+                published = article_date.get("Year", "")
+            else:
+                # Fallback to PubDate in Journal > JournalIssue > PubDate
+                pubdate = art.get("Journal", {}).get("JournalIssue", {}).get("PubDate", {})
+                published = pubdate.get("Year", "") or pubdate.get("MedlineDate", "")
+
+            # Robustly extract authors
+            authors_raw = art.get("AuthorList", {}).get("Author", [])
+            if isinstance(authors_raw, dict):
+                authors_raw = [authors_raw]
+            authors = ", ".join([
+                f"{a.get('LastName', '')} {a.get('ForeName', '')}".strip()
+                for a in authors_raw if a.get("LastName") and a.get("ForeName")
+            ]) if authors_raw else "Unknown"
+
+            # Robustly extract summary/abstract
+            abstract = art.get("Abstract", {}).get("AbstractText", "")
+            if isinstance(abstract, list):
+                summary = " ".join(abstract)
+            elif isinstance(abstract, dict):
+                summary = abstract.get("#text", "")
+            else:
+                summary = abstract or ""
+
+            pmid = a["MedlineCitation"]["PMID"]
+            if isinstance(pmid, dict):
+                pmid = pmid.get("#text", "")
+
             results.append({
                 "title": art["ArticleTitle"],
-                "authors": ", ".join([a["LastName"] + " " + a["ForeName"] for a in art.get("AuthorList", {}).get("Author", []) if "LastName" in a and "ForeName" in a]),
-                "summary": art.get("Abstract", {}).get("AbstractText", [""])[0],
-                "link": f"https://pubmed.ncbi.nlm.nih.gov/{a['MedlineCitation']['PMID']['#text']}/",
-                "published": art.get("ArticleDate", [{}])[0].get("Year", ""),
+                "authors": authors,
+                "summary": summary,
+                "link": f"https://pubmed.ncbi.nlm.nih.gov/{pmid}/",
+                "published": published,
                 "source": "PubMed"
             })
         return results