Update mcp/clinicaltrials.py

mcp/clinicaltrials.py

@@ -1,34 +1,47 @@
-# mcp/clinicaltrials.py
+#!/usr/bin/env python3
+"""
+Fetch clinical trials from ClinicalTrials.gov without external clients.
+"""
 
-from pytrials.client import ClinicalTrials
-import asyncio
+import httpx
+import urllib.parse
 
-async def fetch_clinical_trials(query: str, max_studies: int = 10) -> list[dict]:
+async def fetch_clinical_trials(
+    query: str,
+    max_studies: int = 10
+) -> list[dict]:
     """
-    Pulls NCTId, Title, Phase, and Status for studies matching the given query.
-    Uses the pytrials wrapper over ClinicalTrials.gov.    
+    Query ClinicalTrials.gov study_fields API for NCTId, Title, Phase, Status.
+    Returns up to `max_studies` results.
     """
-    # pytrials expects a direct search expression, e.g. disease MeSH term or drug
-    client = ClinicalTrials()
-    # These are the fields you want from the API 
-    study_fields = ["NCTId", "BriefTitle", "Phase", "OverallStatus"]
-    # Wrap in a ThreadPool so we can call it in asyncio
-    loop = asyncio.get_event_loop()
-    records = await loop.run_in_executor(
-        None,
-        lambda: client.get_study_fields(
-            search_expr=query,
-            fields=study_fields,
-            max_studies=max_studies
-        )
-    )
-    # Normalize the field names for your UI
-    return [
-        {
-            "nctId": rec.get("NCTId"),
-            "briefTitle": rec.get("BriefTitle"),
-            "phase": rec.get("Phase"),
-            "status": rec.get("OverallStatus"),
-        }
-        for rec in records
-    ]
+    # Build the API URL
+    base = "https://clinicaltrials.gov/api/query/study_fields"
+    params = {
+        "expr": query,
+        "fields": ",".join(["NCTId", "BriefTitle", "Phase", "OverallStatus"]),
+        "max_rnk": max_studies,
+        "fmt": "json",
+    }
+    url = f"{base}?{urllib.parse.urlencode(params)}"
+
+    try:
+        async with httpx.AsyncClient(timeout=10) as client:
+            resp = await client.get(url)
+            resp.raise_for_status()
+            data = resp.json()
+    except Exception:
+        # In case of any network / parse error
+        return []
+
+    # Navigate JSON response
+    # API docs: https://clinicaltrials.gov/api/gui/ref/study_fields
+    studies = data.get("StudyFieldsResponse", {}).get("StudyFields", [])
+    output = []
+    for s in studies:
+        output.append({
+            "nctId": s.get("NCTId", [""])[0],
+            "briefTitle": s.get("BriefTitle", [""])[0],
+            "phase": s.get("Phase", [""])[0],
+            "status": s.get("OverallStatus", [""])[0],
+        })
+    return output